| #!/usr/bin/env python3 |
| # |
| # Copyright (C) 2019 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 argparse |
| import json |
| import os |
| import subprocess |
| import sys |
| |
| from collections import defaultdict |
| from glob import glob |
| |
| def parse_args(): |
| """Parse commandline arguments.""" |
| parser = argparse.ArgumentParser(description='Find sharedUserId violators') |
| parser.add_argument('--product_out', help='PRODUCT_OUT directory', |
| default=os.environ.get("PRODUCT_OUT")) |
| parser.add_argument('--aapt', help='Path to aapt or aapt2', |
| default="aapt2") |
| parser.add_argument('--copy_out_system', help='TARGET_COPY_OUT_SYSTEM', |
| default="system") |
| parser.add_argument('--copy_out_vendor', help='TARGET_COPY_OUT_VENDOR', |
| default="vendor") |
| parser.add_argument('--copy_out_product', help='TARGET_COPY_OUT_PRODUCT', |
| default="product") |
| parser.add_argument('--copy_out_system_ext', help='TARGET_COPY_OUT_SYSTEM_EXT', |
| default="system_ext") |
| return parser.parse_args() |
| |
| def execute(cmd): |
| p = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE) |
| out, err = map(lambda b: b.decode('utf-8'), p.communicate()) |
| return p.returncode == 0, out, err |
| |
| def make_aapt_cmds(file): |
| return [aapt + ' dump ' + file + ' --file AndroidManifest.xml', |
| aapt + ' dump xmltree ' + file + ' --file AndroidManifest.xml'] |
| |
| def extract_shared_uid(file): |
| for cmd in make_aapt_cmds(file): |
| success, manifest, error_msg = execute(cmd) |
| if success: |
| break |
| else: |
| print(error_msg, file=sys.stderr) |
| sys.exit() |
| |
| for l in manifest.split('\n'): |
| if "sharedUserId" in l: |
| return l.split('"')[-2] |
| return None |
| |
| |
| args = parse_args() |
| |
| product_out = args.product_out |
| aapt = args.aapt |
| |
| partitions = ( |
| ("system", args.copy_out_system), |
| ("vendor", args.copy_out_vendor), |
| ("product", args.copy_out_product), |
| ("system_ext", args.copy_out_system_ext), |
| ) |
| |
| shareduid_app_dict = defaultdict(list) |
| |
| for part, location in partitions: |
| for f in glob(os.path.join(product_out, location, "*", "*", "*.apk")): |
| apk_file = os.path.basename(f) |
| shared_uid = extract_shared_uid(f) |
| |
| if shared_uid is None: |
| continue |
| shareduid_app_dict[shared_uid].append((part, apk_file)) |
| |
| |
| output = defaultdict(lambda: defaultdict(list)) |
| |
| for uid, app_infos in shareduid_app_dict.items(): |
| partitions = {p for p, _ in app_infos} |
| if len(partitions) > 1: |
| for part in partitions: |
| output[uid][part].extend([a for p, a in app_infos if p == part]) |
| |
| print(json.dumps(output, indent=2, sort_keys=True)) |