mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2026-01-28 11:15:50 +03:00
63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
#!/usr/bin/env python
|
|
|
|
"""
|
|
Copyright (c) 2006-2026 sqlmap developers (https://sqlmap.org)
|
|
See the file 'LICENSE' for copying permission
|
|
"""
|
|
|
|
import os
|
|
import re
|
|
import subprocess
|
|
|
|
from lib.core.common import openFile
|
|
from lib.core.convert import getText
|
|
|
|
def getRevisionNumber():
|
|
"""
|
|
Returns abbreviated commit hash number as retrieved with "git rev-parse --short HEAD"
|
|
|
|
>>> len(getRevisionNumber() or (' ' * 7)) == 7
|
|
True
|
|
"""
|
|
|
|
retVal = None
|
|
filePath = None
|
|
directory = os.path.dirname(__file__)
|
|
|
|
while True:
|
|
candidate = os.path.join(directory, ".git", "HEAD")
|
|
if os.path.exists(candidate):
|
|
filePath = candidate
|
|
break
|
|
|
|
parent = os.path.dirname(directory)
|
|
if parent == directory:
|
|
break
|
|
directory = parent
|
|
|
|
if filePath:
|
|
with openFile(filePath, "r") as f:
|
|
content = getText(f.read()).strip()
|
|
|
|
if content.startswith("ref: "):
|
|
ref_path = content.replace("ref: ", "").strip()
|
|
filePath = os.path.join(directory, ".git", ref_path)
|
|
|
|
if os.path.exists(filePath):
|
|
with openFile(filePath, "r") as f_ref:
|
|
content = getText(f_ref.read()).strip()
|
|
|
|
match = re.match(r"(?i)[0-9a-f]{40}", content)
|
|
retVal = match.group(0) if match else None
|
|
|
|
if not retVal:
|
|
try:
|
|
process = subprocess.Popen(["git", "rev-parse", "--verify", "HEAD"], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
stdout, _ = process.communicate()
|
|
match = re.search(r"(?i)[0-9a-f]{40}", getText(stdout or ""))
|
|
retVal = match.group(0) if match else None
|
|
except:
|
|
pass
|
|
|
|
return retVal[:7] if retVal else None
|