1#!/usr/bin/env python 2# 3# Copyright (C) 2018 The Android Open Source Project 4# 5# Licensed under the Apache License, Version 2.0 (the "License"); 6# you may not use this file except in compliance with the License. 7# You may obtain a copy of the License at 8# 9# http://www.apache.org/licenses/LICENSE-2.0 10# 11# Unless required by applicable law or agreed to in writing, software 12# distributed under the License is distributed on an "AS IS" BASIS, 13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14# See the License for the specific language governing permissions and 15# limitations under the License. 16 17import apex_manifest_pb2 18from google.protobuf import message 19from google.protobuf.json_format import MessageToJson 20import zipfile 21 22class ApexManifestError(Exception): 23 24 def __init__(self, errmessage): 25 # Apex Manifest parse error (extra fields) or if required fields not present 26 self.errmessage = errmessage 27 28 29def ValidateApexManifest(manifest_pb): 30 # Checking required fields 31 if manifest_pb.name == "": 32 raise ApexManifestError("'name' field is required.") 33 if manifest_pb.version == 0: 34 raise ApexManifestError("'version' field is required.") 35 if manifest_pb.noCode and (manifest_pb.preInstallHook or 36 manifest_pb.postInstallHook): 37 raise ApexManifestError( 38 "'noCode' can't be true when either preInstallHook or postInstallHook is set" 39 ) 40 41def ParseApexManifest(file): 42 try: 43 with open(file, "rb") as f: 44 manifest_pb = apex_manifest_pb2.ApexManifest() 45 manifest_pb.ParseFromString(f.read()) 46 return manifest_pb 47 except message.DecodeError as err: 48 raise ApexManifestError(err) 49 50def fromApex(apexFilePath): 51 with zipfile.ZipFile(apexFilePath, 'r') as apexFile: 52 with apexFile.open('apex_manifest.pb') as manifestFile: 53 manifest = apex_manifest_pb2.ApexManifest() 54 manifest.ParseFromString(manifestFile.read()) 55 return manifest 56 57def toJsonString(manifest): 58 return MessageToJson(manifest, indent=2) 59