sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2024-11-25 19:13:48 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix for an Issue #121

Browse Source
This commit is contained in:
Miroslav Stampar 2012-07-26 00:02:38 +02:00
parent 4a0b55f651
commit 18b1d1efd6
2 changed files with 16 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
lib/core/revision.py
View File

@ -21,7 +21,7 @@ def getRevisionNumber():
_ = os.path.dirname(__file__)
while True:
filePath = os.path.join(_, ".git/refs/heads/master").replace('/', os.path.sep)
filePath = os.path.join(_, ".git", "HEAD")
if os.path.exists(filePath):
break
else:
@ -30,10 +30,17 @@ def getRevisionNumber():
break
else:
_ = os.path.dirname(_)
if filePath:
while True:
if filePath and os.path.isfile(filePath):
with open(filePath, "r") as f:
match = re.match(r"(?i)[0-9a-f]{32}", f.read())
content = f.read()
filePath = None
if content.startswith("ref: "):
filePath = os.path.join(_, ".git", content.replace("ref: ", "")).strip()
else:
match = re.match(r"(?i)[0-9a-f]{32}", content)
retVal = match.group(0) if match else None
break
if not retVal:
process = execute("git rev-parse --verify HEAD", shell=True, stdout=PIPE, stderr=PIPE)

5
lib/core/update.py
View File

@ -17,6 +17,7 @@ from lib.core.common import getUnicode
from lib.core.data import conf
from lib.core.data import logger
from lib.core.data import paths
from lib.core.revision import getRevisionNumber
from lib.core.settings import GIT_REPOSITORY
from lib.core.settings import IS_WIN
from lib.core.settings import REVISION
@ -49,7 +50,9 @@ def update():
success = not process.returncode
if success:
logger.info("%s the latest revision '%s'" % ("already at" if "Already" in stdout else "updated to", REVISION))
import lib.core.settings
_ = lib.core.settings.REVISION = getRevisionNumber()
logger.info("%s the latest revision '%s'" % ("already at" if "Already" in stdout else "updated to", _))
else:
logger.error("update could not be completed ('%s')" % re.sub(r"\W+", " ", stderr).strip())