mirror of
https://github.com/overcuriousity/autopsy-flatpak.git
synced 2025-07-06 21:00:22 +00:00
82 lines
2.3 KiB
Python
82 lines
2.3 KiB
Python
# This python script sets the sleuthkit branch based on the autopsy build branch
|
|
# in appveyor and travis
|
|
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
import xml.etree.ElementTree as ET
|
|
|
|
TSK_HOME=os.getenv("TSK_HOME",False)
|
|
passed = 1
|
|
|
|
def getSleuthkitBranchList():
|
|
# Returns the list of sleuthkit branches
|
|
cmd = ['git','branch','-a']
|
|
retdir = os.getcwd()
|
|
os.chdir(TSK_HOME)
|
|
output = subprocess.check_output(cmd)
|
|
branches = []
|
|
for branch in output.strip().split():
|
|
if branch.startswith('remotes/origin'):
|
|
branches.append(branch.split('/')[2])
|
|
os.chdir(retdir)
|
|
return branches
|
|
|
|
def gitSleuthkitCheckout(branch):
|
|
'''
|
|
Checksout sleuthkit branch
|
|
Args:
|
|
branch: String, which branch to checkout
|
|
'''
|
|
# passed is a global variable that gets set to non-zero integer
|
|
# When an error occurs
|
|
global passed
|
|
cmd = ['git','checkout',branch]
|
|
passed = subprocess.call(cmd,stdout=sys.stdout,cwd=TSK_HOME)
|
|
|
|
def parseXML(xmlFile):
|
|
'''
|
|
parses the TSKVersion.xml file for sleuthkit version
|
|
Args:
|
|
xmlFile: String, xml file to parse
|
|
'''
|
|
tree = ET.parse(xmlFile)
|
|
root = tree.getroot()
|
|
for child in root:
|
|
if child.attrib['name']=='TSK_VERSION':
|
|
return child.attrib['value']
|
|
return None
|
|
|
|
def main():
|
|
global passed
|
|
|
|
if not TSK_HOME:
|
|
sys.exit(1)
|
|
print('Please set TSK_HOME env variable')
|
|
|
|
TRAVIS=os.getenv("TRAVIS",False)
|
|
APPVEYOR=os.getenv("APPVEYOR",False)
|
|
if TRAVIS == "true":
|
|
CURRENT_BRANCH=os.getenv("TRAVIS_BRANCH",False)
|
|
elif APPVEYOR:
|
|
CURRENT_BRANCH=os.getenv("APPVEYOR_REPO_BRANCH",False)
|
|
else:
|
|
cmd=['git','rev-parse','--abbrev-ref','HEAD']
|
|
output = subprocess.check_output(cmd)
|
|
CURRENT_BRANCH=output.strip()
|
|
#check if sleuthkit has release branch
|
|
if CURRENT_BRANCH.startswith('release'):
|
|
version = parseXML('TSKVersion.xml')
|
|
RELEASE_BRANCH = "release-"+version
|
|
gitSleuthkitCheckout(RELEASE_BRANCH)
|
|
elif CURRENT_BRANCH in getSleuthkitBranchList(): # check sleuthkit has same branch as autopsy
|
|
gitSleuthkitCheckout(CURRENT_BRANCH)
|
|
|
|
if passed != 0:
|
|
gitSleuthkitCheckout('develop')
|
|
if passed != 0:
|
|
print('Something gone wrong')
|
|
sys.exit(1)
|
|
if __name__ == '__main__':
|
|
main()
|