Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit d42d283

Browse files
committed
avoid tracebacks because the parameter does not exist
1 parent c9c520a commit d42d283

10 files changed

Lines changed: 12 additions & 13 deletions

File tree

lib/takeover/registry.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ def _createRemoteBatchFile(self):
6565
logger.debug("creating batch file '%s'" % self._batPathRemote)
6666

6767
self._createLocalBatchFile()
68-
self.writeFile(self._batPathLocal, self._batPathRemote, "text")
68+
self.writeFile(self._batPathLocal, self._batPathRemote, "text", forceCheck=True)
6969

7070
os.unlink(self._batPathLocal)
7171

plugins/dbms/access/filesystem.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,6 @@ def readFile(self, rFile):
1616
errMsg = "on Microsoft Access it is not possible to read files"
1717
raise SqlmapUnsupportedFeatureException(errMsg)
1818

19-
def writeFile(self, wFile, dFile, fileType=None):
19+
def writeFile(self, wFile, dFile, fileType=None, forceCheck=False):
2020
errMsg = "on Microsoft Access it is not possible to write files"
2121
raise SqlmapUnsupportedFeatureException(errMsg)

plugins/dbms/firebird/filesystem.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,6 @@ def readFile(self, rFile):
1616
errMsg = "on Firebird it is not possible to read files"
1717
raise SqlmapUnsupportedFeatureException(errMsg)
1818

19-
def writeFile(self, wFile, dFile, fileType=None):
19+
def writeFile(self, wFile, dFile, fileType=None, forceCheck=False):
2020
errMsg = "on Firebird it is not possible to write files"
2121
raise SqlmapUnsupportedFeatureException(errMsg)

plugins/dbms/maxdb/filesystem.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,6 @@ def readFile(self, rFile):
1616
errMsg = "on SAP MaxDB reading of files is not supported"
1717
raise SqlmapUnsupportedFeatureException(errMsg)
1818

19-
def writeFile(self, wFile, dFile, fileType=None):
19+
def writeFile(self, wFile, dFile, fileType=None, forceCheck=False):
2020
errMsg = "on SAP MaxDB writing of files is not supported"
2121
raise SqlmapUnsupportedFeatureException(errMsg)

plugins/dbms/mssqlserver/filesystem.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -158,7 +158,7 @@ def stackedReadFile(self, rFile):
158158

159159
return result
160160

161-
def unionWriteFile(self, wFile, dFile, fileType):
161+
def unionWriteFile(self, wFile, dFile, fileType, forceCheck=False):
162162
errMsg = "Microsoft SQL Server does not support file upload with "
163163
errMsg += "UNION query SQL injection technique"
164164
raise SqlmapUnsupportedFeatureException(errMsg)
@@ -340,8 +340,7 @@ def stackedWriteFile(self, wFile, dFile, fileType, forceCheck=False):
340340
wFileContent = f.read()
341341

342342
self._stackedWriteFileVbs(tmpPath, wFileContent, dFile, fileType)
343-
344-
written = self.askCheckWrittenFile(wFile, dFile)
343+
written = self.askCheckWrittenFile(wFile, dFile, forceCheck)
345344

346345
if written is False:
347346
message = "do you want to try to upload the file with "

plugins/dbms/oracle/filesystem.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ def readFile(self, rFile):
1717
errMsg += "Oracle"
1818
raise SqlmapUnsupportedFeatureException(errMsg)
1919

20-
def writeFile(self, wFile, dFile, fileType=None):
20+
def writeFile(self, wFile, dFile, fileType=None, forceCheck=False):
2121
errMsg = "File system write access not yet implemented for "
2222
errMsg += "Oracle"
2323
raise SqlmapUnsupportedFeatureException(errMsg)

plugins/dbms/postgresql/filesystem.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ def stackedReadFile(self, rFile):
2828

2929
return self.udfEvalCmd(cmd=rFile, udfName="sys_fileread")
3030

31-
def unionWriteFile(self, wFile, dFile, fileType):
31+
def unionWriteFile(self, wFile, dFile, fileType, forceCheck=False):
3232
errMsg = "PostgreSQL does not support file upload with UNION "
3333
errMsg += "query SQL injection technique"
3434
raise SqlmapUnsupportedFeatureException(errMsg)

plugins/dbms/sqlite/filesystem.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,6 @@ def readFile(self, rFile):
1616
errMsg = "on SQLite it is not possible to read files"
1717
raise SqlmapUnsupportedFeatureException(errMsg)
1818

19-
def writeFile(self, wFile, dFile, fileType=None):
19+
def writeFile(self, wFile, dFile, fileType=None, forceCheck=False):
2020
errMsg = "on SQLite it is not possible to write files"
2121
raise SqlmapUnsupportedFeatureException(errMsg)

plugins/dbms/sybase/filesystem.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,6 @@ def readFile(self, rFile):
1616
errMsg = "on Sybase it is not possible to read files"
1717
raise SqlmapUnsupportedFeatureException(errMsg)
1818

19-
def writeFile(self, wFile, dFile, fileType=None):
19+
def writeFile(self, wFile, dFile, fileType=None, forceCheck=False):
2020
errMsg = "on Sybase it is not possible to write files"
2121
raise SqlmapUnsupportedFeatureException(errMsg)

plugins/generic/filesystem.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -171,12 +171,12 @@ def stackedReadFile(self, remoteFile):
171171
errMsg += "into the specific DBMS plugin"
172172
raise SqlmapUndefinedMethod(errMsg)
173173

174-
def unionWriteFile(self, localFile, remoteFile, fileType):
174+
def unionWriteFile(self, localFile, remoteFile, fileType, forceCheck=False):
175175
errMsg = "'unionWriteFile' method must be defined "
176176
errMsg += "into the specific DBMS plugin"
177177
raise SqlmapUndefinedMethod(errMsg)
178178

179-
def stackedWriteFile(self, localFile, remoteFile, fileType):
179+
def stackedWriteFile(self, localFile, remoteFile, fileType, forceCheck=False):
180180
errMsg = "'stackedWriteFile' method must be defined "
181181
errMsg += "into the specific DBMS plugin"
182182
raise SqlmapUndefinedMethod(errMsg)

0 commit comments

Comments
 (0)