mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2025-12-06 04:31:30 +00:00
added support for displaying revision number at unhandled exception message
This commit is contained in:
39
lib/core/revision.py
Normal file
39
lib/core/revision.py
Normal file
@@ -0,0 +1,39 @@
|
||||
import os
|
||||
import re
|
||||
|
||||
from subprocess import PIPE
|
||||
from subprocess import Popen as execute
|
||||
|
||||
def getRevisionNumber():
|
||||
curDir = os.path.dirname(os.path.realpath(__file__))
|
||||
retVal = None
|
||||
|
||||
try:
|
||||
import pysvn
|
||||
|
||||
client = pysvn.Client()
|
||||
if client.info(curDir):
|
||||
retVal = client.info(curDir).revision.number
|
||||
|
||||
except ImportError, _:
|
||||
process = execute("svn info %s" % curDir, shell=True, stdout=PIPE, stderr=PIPE)
|
||||
|
||||
svnStdout, svnStderr = process.communicate()
|
||||
|
||||
if svnStdout:
|
||||
revision = re.search("Revision:\s+([\d]+)", svnStdout)
|
||||
|
||||
if revision:
|
||||
retVal = revision.group(1)
|
||||
|
||||
if retVal:
|
||||
try:
|
||||
retVal = int(retVal)
|
||||
except ValueError:
|
||||
retVal = None
|
||||
|
||||
#if not retVal:
|
||||
#debugMsg = "sqlmap was not able to retrieve the revision number"
|
||||
#logger.debug(debugMsg)
|
||||
|
||||
return retVal
|
||||
Reference in New Issue
Block a user