sqlmap/lib/core/revision.py

40 lines
972 B
Python

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