Merge pull request #136 from knownsec/bugfix/issue-132

bugfix for issue #132
knownsec · Sep 19, 2017 · ba63dcd · ba63dcd
2 parents 3fee581 + e504c7c
commit ba63dcd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/pocsuite/__init__.py b/pocsuite/__init__.py
@@ -7,7 +7,7 @@
 """
 
 __title__ = 'pocsuite'
-__version__ = '2.0.5'
+__version__ = '2.0.6'
 __author__ = 'Knownsec Security Team'
 __author_email__ = '[email protected]'
 __license__ = 'GPL 2.0'

diff --git a/pocsuite/api/cannon.py b/pocsuite/api/cannon.py
@@ -8,6 +8,7 @@
 
 import time
 import socket
+import hashlib
 from pocsuite.lib.core.data import kb
 from pocsuite.lib.core.data import conf
 from pocsuite.lib.core.data import logger
@@ -53,7 +54,8 @@ def _setHTTPTimeout(self, timeout):
 
     def registerPoc(self):
         pocString = multipleReplace(self.pocString, POC_IMPORTDICT)
-        _, self.moduleName = filepathParser(self.pocName)
+        _, fileName = filepathParser(self.pocName)
+        self.moduleName = "%s%s" %(fileName, hashlib.md5(self.target).hexdigest()[:8])
         try:
             importer = StringImporter(self.moduleName, pocString)
             importer.load_module(self.moduleName)