sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-06-20 13:03:23 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix for an Issue #318

Browse Source
This commit is contained in:
Miroslav Stampar 2012-12-21 11:10:05 +01:00
parent 35728fa443
commit 6c1ec9b54f
3 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lib/takeover/icmpsh.py
View File

@ -30,7 +30,7 @@ class ICMPsh:
self.rhostStr = None self.rhostStr = None
self.localIP = getLocalIP() self.localIP = getLocalIP()
self.remoteIP = getRemoteIP() self.remoteIP = getRemoteIP()
self.__icmpslave = normalizePath(os.path.join(paths.SQLMAP_EXTRAS_PATH, "icmpsh", "icmpsh.exe")) self._icmpslave = normalizePath(os.path.join(paths.SQLMAP_EXTRAS_PATH, "icmpsh", "icmpsh.exe"))
def _selectRhost(self): def _selectRhost(self):
message = "what is the back-end DBMS address? [%s] " % self.remoteIP message = "what is the back-end DBMS address? [%s] " % self.remoteIP
@ -63,7 +63,7 @@ class ICMPsh:
self.execCmd(cmd, silent=True) self.execCmd(cmd, silent=True)
def uploadIcmpshSlave(self, web=False): def uploadIcmpshSlave(self, web=False):
self._initVars() ICMPsh._initVars(self)
self._randStr = randomStr(lowercase=True) self._randStr = randomStr(lowercase=True)
self._icmpslaveRemoteBase = "tmpi%s.exe" % self._randStr self._icmpslaveRemoteBase = "tmpi%s.exe" % self._randStr
@ -77,9 +77,9 @@ class ICMPsh:
logger.info("uploading icmpsh slave to '%s'" % self._icmpslaveRemote) logger.info("uploading icmpsh slave to '%s'" % self._icmpslaveRemote)
if web: if web:
self.webUpload(self._icmpslaveRemote, self.webDirectory, filepath=self.__icmpslave) self.webUpload(self._icmpslaveRemote, self.webDirectory, filepath=self._icmpslave)
else: else:
self.writeFile(self.__icmpslave, self._icmpslaveRemote, "binary") self.writeFile(self._icmpslave, self._icmpslaveRemote, "binary")
def icmpPwn(self): def icmpPwn(self):
self._prepareIngredients() self._prepareIngredients()

4
lib/takeover/metasploit.py
View File

@ -503,7 +503,7 @@ class Metasploit:
self._randStr = randomStr(lowercase=True) self._randStr = randomStr(lowercase=True)
self._shellcodeFilePath = os.path.join(conf.outputPath, "tmpm%s" % self._randStr) self._shellcodeFilePath = os.path.join(conf.outputPath, "tmpm%s" % self._randStr)
self._initVars() Metasploit._initVars(self)
self._prepareIngredients(encode=encode) self._prepareIngredients(encode=encode)
self._forgeMsfPayloadCmd(exitfunc, format, self._shellcodeFilePath, extra) self._forgeMsfPayloadCmd(exitfunc, format, self._shellcodeFilePath, extra)
@ -583,7 +583,7 @@ class Metasploit:
self.delRemoteFile(self.shellcodeexecRemote) self.delRemoteFile(self.shellcodeexecRemote)
def smb(self): def smb(self):
self._initVars() Metasploit._initVars(self)
self._randFile = "tmpu%s.txt" % randomStr(lowercase=True) self._randFile = "tmpu%s.txt" % randomStr(lowercase=True)
self._runMsfCliSmbrelay() self._runMsfCliSmbrelay()

18
lib/takeover/registry.py
View File

@ -49,11 +49,11 @@ class Registry:
def _createLocalBatchFile(self): def _createLocalBatchFile(self):
self._batPathFp = open(self._batPathLocal, "w") self._batPathFp = open(self._batPathLocal, "w")
if self.__operation == "read": if self._operation == "read":
lines = self._batRead lines = self._batRead
elif self.__operation == "add": elif self._operation == "add":
lines = self._batAdd lines = self._batAdd
elif self.__operation == "delete": elif self._operation == "delete":
lines = self._batDel lines = self._batDel
for line in lines: for line in lines:
@ -70,9 +70,9 @@ class Registry:
os.unlink(self._batPathLocal) os.unlink(self._batPathLocal)
def readRegKey(self, regKey, regValue, parse=False): def readRegKey(self, regKey, regValue, parse=False):
self.__operation = "read" self._operation = "read"
self._initVars(regKey, regValue, parse=parse) Registry._initVars(self, regKey, regValue, parse=parse)
self._createRemoteBatchFile() self._createRemoteBatchFile()
logger.debug("reading registry key '%s' value '%s'" % (regKey, regValue)) logger.debug("reading registry key '%s' value '%s'" % (regKey, regValue))
@ -90,9 +90,9 @@ class Registry:
return data return data
def addRegKey(self, regKey, regValue, regType, regData): def addRegKey(self, regKey, regValue, regType, regData):
self.__operation = "add" self._operation = "add"
self._initVars(regKey, regValue, regType, regData) Registry._initVars(self, regKey, regValue, regType, regData)
self._createRemoteBatchFile() self._createRemoteBatchFile()
debugMsg = "adding registry key value '%s' " % self._regValue debugMsg = "adding registry key value '%s' " % self._regValue
@ -103,9 +103,9 @@ class Registry:
self.delRemoteFile(self._batPathRemote) self.delRemoteFile(self._batPathRemote)
def delRegKey(self, regKey, regValue): def delRegKey(self, regKey, regValue):
self.__operation = "delete" self._operation = "delete"
self._initVars(regKey, regValue) Registry._initVars(self, regKey, regValue)
self._createRemoteBatchFile() self._createRemoteBatchFile()
debugMsg = "deleting registry key value '%s' " % self._regValue debugMsg = "deleting registry key value '%s' " % self._regValue