Fixed KSA arguments

v-p-b · Dec 17, 2017 · e1569ae · e1569ae
1 parent 5cdefff
commit e1569ae
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/sample/rc4/prga.py b/sample/rc4/prga.py
@@ -100,15 +100,16 @@ def run(self, arg_0, arg_1):
         self.mu.mem_write(argAddr_1, arg_1)
         self.mu.reg_write(UC_X86_REG_RSI, argAddr_1)
         self._start_unicorn(0x40065a)
-        return self.mu.reg_read(UC_X86_REG_RAX)
+        return self.mu.mem_read(argAddr_1,256)
+        # return self.mu.reg_read(UC_X86_REG_RAX)
 
 key="key"
-S=""
+S=" "*256
 cipher=""
 plain="test"
 
 ksa=KSA()
-ksa.run(key,S)
+S=str(ksa.run(key,S))
 print repr(S)
 prga=PRGA()
 prga.run(S,plain,cipher)