| # -*- coding: utf-8 -*- |
| |
| #------------------------------------------------------------------------- |
| # drawElements Quality Program utilities |
| # -------------------------------------- |
| # |
| # Copyright 2015 The Android Open Source Project |
| # |
| # Licensed under the Apache License, Version 2.0 (the "License"); |
| # you may not use this file except in compliance with the License. |
| # You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| # |
| #------------------------------------------------------------------------- |
| |
| import os |
| import sys |
| import shutil |
| import tarfile |
| import zipfile |
| import hashlib |
| import argparse |
| import subprocess |
| import ssl |
| import stat |
| import platform |
| |
| sys.path.append(os.path.join(os.path.dirname(__file__), "..", "scripts")) |
| |
| from ctsbuild.common import * |
| |
| EXTERNAL_DIR = os.path.realpath(os.path.normpath(os.path.dirname(__file__))) |
| |
| SYSTEM_NAME = platform.system() |
| |
| def computeChecksum (data): |
| return hashlib.sha256(data).hexdigest() |
| |
| def onReadonlyRemoveError (func, path, exc_info): |
| os.chmod(path, stat.S_IWRITE) |
| os.unlink(path) |
| |
| class Source: |
| def __init__(self, baseDir, extractDir): |
| self.baseDir = baseDir |
| self.extractDir = extractDir |
| |
| def clean (self): |
| fullDstPath = os.path.join(EXTERNAL_DIR, self.baseDir, self.extractDir) |
| # Remove read-only first |
| readonlydir = os.path.join(fullDstPath, ".git") |
| if os.path.exists(readonlydir): |
| shutil.rmtree(readonlydir, onerror = onReadonlyRemoveError) |
| if os.path.exists(fullDstPath): |
| shutil.rmtree(fullDstPath, ignore_errors=False) |
| |
| class SourcePackage (Source): |
| def __init__(self, url, checksum, baseDir, extractDir = "src", postExtract=None): |
| Source.__init__(self, baseDir, extractDir) |
| self.url = url |
| self.filename = os.path.basename(self.url) |
| self.checksum = checksum |
| self.archiveDir = "packages" |
| self.postExtract = postExtract |
| |
| if SYSTEM_NAME == 'Windows' or SYSTEM_NAME.startswith('CYGWIN') or SYSTEM_NAME.startswith('MINGW'): |
| self.sysNdx = 0 |
| elif SYSTEM_NAME == 'Linux': |
| self.sysNdx = 1 |
| elif SYSTEM_NAME == 'Darwin': |
| self.sysNdx = 2 |
| else: |
| self.sysNdx = -1 # unknown system |
| |
| self.FFmpeg = "FFmpeg" in url |
| |
| def clean (self): |
| Source.clean(self) |
| self.removeArchives() |
| |
| def update (self, cmdProtocol = None, force = False): |
| if self.sysNdx != 2: |
| if not self.isArchiveUpToDate(): |
| self.fetchAndVerifyArchive() |
| |
| if self.getExtractedChecksum() != self.checksum: |
| Source.clean(self) |
| self.extract() |
| self.storeExtractedChecksum(self.checksum) |
| |
| def removeArchives (self): |
| archiveDir = os.path.join(EXTERNAL_DIR, pkg.baseDir, pkg.archiveDir) |
| if os.path.exists(archiveDir): |
| shutil.rmtree(archiveDir, ignore_errors=False) |
| |
| def isArchiveUpToDate (self): |
| archiveFile = os.path.join(EXTERNAL_DIR, pkg.baseDir, pkg.archiveDir, pkg.filename) |
| if os.path.exists(archiveFile): |
| return computeChecksum(readBinaryFile(archiveFile)) == self.checksum |
| else: |
| return False |
| |
| def getExtractedChecksumFilePath (self): |
| return os.path.join(EXTERNAL_DIR, pkg.baseDir, pkg.archiveDir, "extracted") |
| |
| def getExtractedChecksum (self): |
| extractedChecksumFile = self.getExtractedChecksumFilePath() |
| |
| if os.path.exists(extractedChecksumFile): |
| return readFile(extractedChecksumFile) |
| else: |
| return None |
| |
| def storeExtractedChecksum (self, checksum): |
| checksum_bytes = checksum.encode("utf-8") |
| writeBinaryFile(self.getExtractedChecksumFilePath(), checksum_bytes) |
| |
| def connectToUrl (self, url): |
| result = None |
| |
| if sys.version_info < (3, 0): |
| from urllib2 import urlopen |
| else: |
| from urllib.request import urlopen |
| |
| if args.insecure: |
| print("Ignoring certificate checks") |
| ssl_context = ssl._create_unverified_context() |
| result = urlopen(url, context=ssl_context) |
| else: |
| result = urlopen(url) |
| |
| return result |
| |
| def fetchAndVerifyArchive (self): |
| print("Fetching %s" % self.url) |
| |
| req = self.connectToUrl(self.url) |
| data = req.read() |
| checksum = computeChecksum(data) |
| dstPath = os.path.join(EXTERNAL_DIR, self.baseDir, self.archiveDir, self.filename) |
| |
| if checksum != self.checksum: |
| raise Exception("Checksum mismatch for %s, expected %s, got %s" % (self.filename, self.checksum, checksum)) |
| |
| if not os.path.exists(os.path.dirname(dstPath)): |
| os.mkdir(os.path.dirname(dstPath)) |
| |
| writeBinaryFile(dstPath, data) |
| |
| def extract (self): |
| print("Extracting %s to %s/%s" % (self.filename, self.baseDir, self.extractDir)) |
| |
| srcPath = os.path.join(EXTERNAL_DIR, self.baseDir, self.archiveDir, self.filename) |
| tmpPath = os.path.join(EXTERNAL_DIR, ".extract-tmp-%s" % self.baseDir) |
| dstPath = os.path.join(EXTERNAL_DIR, self.baseDir, self.extractDir) |
| |
| if self.filename.endswith(".zip"): |
| archive = zipfile.ZipFile(srcPath) |
| else: |
| archive = tarfile.open(srcPath) |
| |
| if os.path.exists(tmpPath): |
| shutil.rmtree(tmpPath, ignore_errors=False) |
| |
| os.mkdir(tmpPath) |
| |
| archive.extractall(tmpPath) |
| archive.close() |
| |
| extractedEntries = os.listdir(tmpPath) |
| if len(extractedEntries) != 1 or not os.path.isdir(os.path.join(tmpPath, extractedEntries[0])): |
| raise Exception("%s doesn't contain single top-level directory" % self.filename) |
| |
| topLevelPath = os.path.join(tmpPath, extractedEntries[0]) |
| |
| if not os.path.exists(dstPath): |
| os.mkdir(dstPath) |
| |
| for entry in os.listdir(topLevelPath): |
| if os.path.exists(os.path.join(dstPath, entry)): |
| raise Exception("%s exists already" % entry) |
| |
| shutil.move(os.path.join(topLevelPath, entry), dstPath) |
| |
| shutil.rmtree(tmpPath, ignore_errors=True) |
| |
| if self.postExtract != None: |
| self.postExtract(dstPath) |
| |
| class SourceFile (Source): |
| def __init__(self, url, filename, checksum, baseDir, extractDir = "src"): |
| Source.__init__(self, baseDir, extractDir) |
| self.url = url |
| self.filename = filename |
| self.checksum = checksum |
| |
| def update (self, cmdProtocol = None, force = False): |
| if not self.isFileUpToDate(): |
| Source.clean(self) |
| self.fetchAndVerifyFile() |
| |
| def isFileUpToDate (self): |
| file = os.path.join(EXTERNAL_DIR, pkg.baseDir, pkg.extractDir, pkg.filename) |
| if os.path.exists(file): |
| data = readFile(file) |
| return computeChecksum(data.encode('utf-8')) == self.checksum |
| else: |
| return False |
| |
| def connectToUrl (self, url): |
| result = None |
| |
| if sys.version_info < (3, 0): |
| from urllib2 import urlopen |
| else: |
| from urllib.request import urlopen |
| |
| if args.insecure: |
| print("Ignoring certificate checks") |
| ssl_context = ssl._create_unverified_context() |
| result = urlopen(url, context=ssl_context) |
| else: |
| result = urlopen(url) |
| |
| return result |
| |
| def fetchAndVerifyFile (self): |
| print("Fetching %s" % self.url) |
| |
| req = self.connectToUrl(self.url) |
| data = req.read() |
| checksum = computeChecksum(data) |
| dstPath = os.path.join(EXTERNAL_DIR, self.baseDir, self.extractDir, self.filename) |
| |
| if checksum != self.checksum: |
| raise Exception("Checksum mismatch for %s, expected %s, got %s" % (self.filename, self.checksum, checksum)) |
| |
| if not os.path.exists(os.path.dirname(dstPath)): |
| os.mkdir(os.path.dirname(dstPath)) |
| |
| writeBinaryFile(dstPath, data) |
| |
| class GitRepo (Source): |
| def __init__(self, httpsUrl, sshUrl, revision, baseDir, extractDir = "src", removeTags = [], patch = ""): |
| Source.__init__(self, baseDir, extractDir) |
| self.httpsUrl = httpsUrl |
| self.sshUrl = sshUrl |
| self.revision = revision |
| self.removeTags = removeTags |
| self.patch = patch |
| |
| def checkout(self, url, fullDstPath, force): |
| if not os.path.exists(os.path.join(fullDstPath, '.git')): |
| execute(["git", "clone", "--no-checkout", url, fullDstPath]) |
| |
| pushWorkingDir(fullDstPath) |
| try: |
| for tag in self.removeTags: |
| proc = subprocess.Popen(['git', 'tag', '-l', tag], stdout=subprocess.PIPE) |
| (stdout, stderr) = proc.communicate() |
| if len(stdout) > 0: |
| execute(["git", "tag", "-d",tag]) |
| force_arg = ['--force'] if force else [] |
| execute(["git", "fetch"] + force_arg + ["--tags", url, "+refs/heads/*:refs/remotes/origin/*"]) |
| execute(["git", "checkout"] + force_arg + [self.revision]) |
| |
| if(self.patch != ""): |
| patchFile = os.path.join(EXTERNAL_DIR, self.patch) |
| execute(["git", "reset", "--hard", "HEAD"]) |
| execute(["git", "apply", patchFile]) |
| finally: |
| popWorkingDir() |
| |
| def update (self, cmdProtocol, force = False): |
| fullDstPath = os.path.join(EXTERNAL_DIR, self.baseDir, self.extractDir) |
| url = self.httpsUrl |
| backupUrl = self.sshUrl |
| |
| # If url is none then start with ssh |
| if cmdProtocol == 'ssh' or url == None: |
| url = self.sshUrl |
| backupUrl = self.httpsUrl |
| |
| try: |
| self.checkout(url, fullDstPath, force) |
| except: |
| if backupUrl != None: |
| self.checkout(backupUrl, fullDstPath, force) |
| |
| def postExtractLibpng (path): |
| shutil.copy(os.path.join(path, "scripts", "pnglibconf.h.prebuilt"), |
| os.path.join(path, "pnglibconf.h")) |
| |
| if SYSTEM_NAME == 'Windows' or SYSTEM_NAME.startswith('CYGWIN') or SYSTEM_NAME.startswith('MINGW'): |
| ffmpeg_url = "https://github.com/BtbN/FFmpeg-Builds/releases/download/autobuild-2022-05-31-12-34/ffmpeg-n4.4.2-1-g8e98dfc57f-win64-lgpl-shared-4.4.zip" |
| ffmpeg_hash_value = "670df8e9d2ddd5e761459b3538f64b8826566270ef1ed13bcbfc63e73aab3fd9" |
| elif SYSTEM_NAME == 'Linux': |
| ffmpeg_url = "https://github.com/BtbN/FFmpeg-Builds/releases/download/autobuild-2022-05-31-12-34/ffmpeg-n4.4.2-1-g8e98dfc57f-linux64-gpl-shared-4.4.tar.xz" |
| ffmpeg_hash_value = "817f8c93ff1ef7ede3dad15b20415d5e366bcd6848844d55046111fd3de827d0" |
| elif SYSTEM_NAME == 'Darwin': |
| ffmpeg_url = "" |
| ffmpeg_hash_value = "" |
| else: |
| ffmpeg_url = None |
| ffmpeg_hash_value = None |
| |
| PACKAGES = [ |
| SourcePackage( |
| "http://zlib.net/fossils/zlib-1.2.13.tar.gz", |
| "b3a24de97a8fdbc835b9833169501030b8977031bcb54b3b3ac13740f846ab30", |
| "zlib"), |
| SourcePackage( |
| "http://prdownloads.sourceforge.net/libpng/libpng-1.6.27.tar.gz", |
| "c9d164ec247f426a525a7b89936694aefbc91fb7a50182b198898b8fc91174b4", |
| "libpng", |
| postExtract = postExtractLibpng), |
| SourcePackage( |
| ffmpeg_url, |
| ffmpeg_hash_value, |
| "ffmpeg"), |
| SourceFile( |
| "https://raw.githubusercontent.com/baldurk/renderdoc/v1.1/renderdoc/api/app/renderdoc_app.h", |
| "renderdoc_app.h", |
| "e7b5f0aa5b1b0eadc63a1c624c0ca7f5af133aa857d6a4271b0ef3d0bdb6868e", |
| "renderdoc"), |
| GitRepo( |
| "https://github.com/KhronosGroup/SPIRV-Tools.git", |
| "git@github.com:KhronosGroup/SPIRV-Tools.git", |
| "f98473ceeb1d33700d01e20910433583e5256030", |
| "spirv-tools"), |
| GitRepo( |
| "https://github.com/KhronosGroup/glslang.git", |
| "git@github.com:KhronosGroup/glslang.git", |
| "77417d5c9e0a5d4c79ddd0285d530b45f7259f0d", |
| "glslang", |
| removeTags = ["master-tot"]), |
| GitRepo( |
| "https://github.com/KhronosGroup/SPIRV-Headers.git", |
| "git@github.com:KhronosGroup/SPIRV-Headers.git", |
| "87d5b782bec60822aa878941e6b13c0a9a954c9b", |
| "spirv-headers"), |
| GitRepo( |
| "https://github.com/KhronosGroup/Vulkan-Docs.git", |
| "git@github.com:KhronosGroup/Vulkan-Docs.git", |
| "9a2e576a052a1e65a5d41b593e693ff02745604b", |
| "vulkan-docs"), |
| GitRepo( |
| "https://github.com/google/amber.git", |
| "git@github.com:google/amber.git", |
| "8b145a6c89dcdb4ec28173339dd176fb7b6f43ed", |
| "amber"), |
| GitRepo( |
| "https://github.com/open-source-parsers/jsoncpp.git", |
| "git@github.com:open-source-parsers/jsoncpp.git", |
| "9059f5cad030ba11d37818847443a53918c327b1", |
| "jsoncpp"), |
| GitRepo( |
| "https://github.com/nvpro-samples/vk_video_samples.git", |
| None, |
| "7d68747d3524842afaf050c5e00a10f5b8c07904", |
| "video-parser"), |
| ] |
| |
| def parseArgs (): |
| versionsForInsecure = ((2,7,9), (3,4,3)) |
| versionsForInsecureStr = ' or '.join(('.'.join(str(x) for x in v)) for v in versionsForInsecure) |
| |
| parser = argparse.ArgumentParser(description = "Fetch external sources") |
| parser.add_argument('--clean', dest='clean', action='store_true', default=False, |
| help='Remove sources instead of fetching') |
| parser.add_argument('--insecure', dest='insecure', action='store_true', default=False, |
| help="Disable certificate check for external sources." |
| " Minimum python version required " + versionsForInsecureStr) |
| parser.add_argument('--protocol', dest='protocol', default='https', choices=['ssh', 'https'], |
| help="Select protocol to checkout git repositories.") |
| parser.add_argument('--force', dest='force', action='store_true', default=False, |
| help="Pass --force to git fetch and checkout commands") |
| |
| args = parser.parse_args() |
| |
| if args.insecure: |
| for versionItem in versionsForInsecure: |
| if (sys.version_info.major == versionItem[0]): |
| if sys.version_info < versionItem: |
| parser.error("For --insecure minimum required python version is " + |
| versionsForInsecureStr) |
| break; |
| |
| return args |
| |
| def run(*popenargs, **kwargs): |
| process = subprocess.Popen(*popenargs, **kwargs) |
| |
| try: |
| stdout, stderr = process.communicate(None) |
| except: |
| process.kill() |
| process.wait() |
| raise |
| |
| retcode = process.poll() |
| |
| if retcode: |
| raise subprocess.CalledProcessError(retcode, process.args, output=stdout, stderr=stderr) |
| |
| return retcode, stdout, stderr |
| |
| if __name__ == "__main__": |
| # Rerun script with python3 as python2 does not have lzma (xz) decompression support |
| if sys.version_info < (3, 0): |
| if SYSTEM_NAME == 'Windows' or SYSTEM_NAME.startswith('CYGWIN') or SYSTEM_NAME.startswith('MINGW'): |
| cmd = ['py', '-3'] |
| elif SYSTEM_NAME == 'Linux': |
| cmd = ['python3'] |
| elif SYSTEM_NAME == 'Darwin': |
| cmd = ['python3'] |
| else: |
| cmd = None # unknown system |
| |
| cmd = cmd + sys.argv |
| run(cmd) |
| else: |
| args = parseArgs() |
| |
| for pkg in PACKAGES: |
| if args.clean: |
| pkg.clean() |
| else: |
| pkg.update(args.protocol, args.force) |