Patch for an Issue sqlmapproject#1047

stamparm · stamparm · commit 8947f2df9615 · 2014-12-17T23:07:27.000+01:00
diff --git a/plugins/dbms/postgresql/takeover.py b/plugins/dbms/postgresql/takeover.py
@@ -8,12 +8,14 @@
 import os
 
 from lib.core.common import Backend
+from lib.core.common import checkFile
 from lib.core.common import decloakToTemp
 from lib.core.common import randomStr
 from lib.core.data import kb
 from lib.core.data import logger
 from lib.core.data import paths
 from lib.core.enums import OS
+from lib.core.exception import SqlmapSystemException
 from lib.core.exception import SqlmapUnsupportedFeatureException
 from lib.request import inject
 from plugins.generic.takeover import Takeover as GenericTakeover
@@ -60,14 +62,20 @@ def udfSetLocalPaths(self):
             errMsg = "unsupported feature on versions of PostgreSQL before 8.2"
             raise SqlmapUnsupportedFeatureException(errMsg)
 
-        if Backend.isOs(OS.WINDOWS):
-            _ = os.path.join(self.udfLocalFile, "postgresql", "windows", "%d" % Backend.getArch(), majorVer, "lib_postgresqludf_sys.dll_")
-            self.udfLocalFile = decloakToTemp(_)
-            self.udfSharedLibExt = "dll"
-        else:
-            _ = os.path.join(self.udfLocalFile, "postgresql", "linux", "%d" % Backend.getArch(), majorVer, "lib_postgresqludf_sys.so_")
-            self.udfLocalFile = decloakToTemp(_)
-            self.udfSharedLibExt = "so"
+        try:
+            if Backend.isOs(OS.WINDOWS):
+                _ = os.path.join(self.udfLocalFile, "postgresql", "windows", "%d" % Backend.getArch(), majorVer, "lib_postgresqludf_sys.dll_")
+                checkFile(_)
+                self.udfLocalFile = decloakToTemp(_)
+                self.udfSharedLibExt = "dll"
+            else:
+                _ = os.path.join(self.udfLocalFile, "postgresql", "linux", "%d" % Backend.getArch(), majorVer, "lib_postgresqludf_sys.so_")
+                checkFile(_)
+                self.udfLocalFile = decloakToTemp(_)
+                self.udfSharedLibExt = "so"
+        except SqlmapSystemException:
+            errMsg = "unsupported feature on PostgreSQL %s (%s-bit)" % (majorVer, Backend.getArch())
+            raise SqlmapUnsupportedFeatureException(errMsg)
 
     def udfCreateFromSharedLib(self, udf, inpRet):
         if udf in self.udfToCreate:
