From f95f1746bb6b07594da6b1ac0cd43d3a1f74cd3d Mon Sep 17 00:00:00 2001
From: Logan Chien <loganchien@google.com>
Date: Tue, 22 Aug 2017 08:32:04 +0800
Subject: [PATCH] vndk-def: Add eligible list update tool

This commit adds a tool to update eligible list datasets from
make_vars-$(TARGET).mk and module-info.json.

Bug: 64503245
Test: ./tools/update_dataset.py \
  -o out.csv \
  --make-vars ${OUT_DIR}/soong/make_vars-$(TARGET).mk \
  --module-info ${ANDROID_PRODUCT_OUT}/module-info.json \
  old-eligible-list.csv

Change-Id: I8c1673046864661f8e0bfac5876b496a5e634dbe
Merged-In: Ic6e79b835528703f164db81e8545a90651bbbdc7
(cherry picked from commit 551b5fdc69083ff1b191fca6644f8335e2c6c341)
---
 .../definition-tool/tools/update_dataset.py   | 110 ++++++++++++++++++
 1 file changed, 110 insertions(+)
 create mode 100755 vndk/tools/definition-tool/tools/update_dataset.py

diff --git a/vndk/tools/definition-tool/tools/update_dataset.py b/vndk/tools/definition-tool/tools/update_dataset.py
new file mode 100755
index 000000000..63053a64c
--- /dev/null
+++ b/vndk/tools/definition-tool/tools/update_dataset.py
@@ -0,0 +1,110 @@
+#!/usr/bin/env python3
+
+# This tool updates extracts the information from Android.bp and updates the
+# datasets for eligible VNDK libraries.
+
+import argparse
+import collections
+import csv
+import json
+import os.path
+import posixpath
+import re
+import sys
+
+def load_make_vars(path):
+    result = collections.OrderedDict([
+        ('SOONG_LLNDK_LIBRARIES', set()),
+        ('SOONG_VNDK_SAMEPROCESS_LIBRARIES', set()),
+        ('SOONG_VNDK_CORE_LIBRARIES', set()),
+    ])
+
+    assign_len = len(' := ')
+
+    with open(path, 'r') as fp:
+        for line in fp:
+            for key, value in result.items():
+                if line.startswith(key):
+                    mod_names = line[len(key) + assign_len:].strip().split(' ')
+                    value.update(mod_names)
+
+    return result.values()
+
+def load_install_paths(module_info_path):
+    with open(module_info_path, 'r') as fp:
+        data = json.load(fp)
+
+    result = set()
+    patt = re.compile(
+            '.*[\\\\/]target[\\\\/]product[\\\\/][^\\\\/]+([\\\\/].*)$')
+    for name, module in data.items():
+        for path in module['installed']:
+            match = patt.match(path)
+            if not match:
+                continue
+            path = match.group(1)
+            path = path.replace(os.path.sep, '/')
+            path = path.replace('/lib/', '/${LIB}/')
+            path = path.replace('/lib64/', '/${LIB}/')
+            result.add(path)
+    return result
+
+def main():
+    parser =argparse.ArgumentParser()
+    parser.add_argument('tag_file')
+    parser.add_argument('-o', '--output', required=True)
+    parser.add_argument('--make-vars', required=True,
+                        help='out/soong/make_vars-$(TARGET).mk')
+    parser.add_argument('--module-info', required=True,
+                        help='out/target/product/$(TARGET)/module-info.json')
+    args = parser.parse_args()
+
+    # Load libraries from `out/soong/make_vars-$(TARGET).mk`.
+    llndk, vndk_sp, vndk = load_make_vars(args.make_vars)
+
+    # Load eligible list csv file.
+    with open(args.tag_file, 'r') as fp:
+        reader = csv.reader(fp)
+        header = next(reader)
+        data = dict()
+        for path, tag, comments in reader:
+            data[path] = [path, tag, comments]
+
+    # Delete non-existing libraries.
+    installed_paths = load_install_paths(args.module_info)
+    data = {
+        path: row
+        for path, row in data.items()
+        if path in installed_paths
+    }
+
+    # Reset all /system/${LIB} libraries to FWK-ONLY.
+    for path, row in data.items():
+        if posixpath.dirname(path) == '/system/${LIB}':
+            row[1] = 'FWK-ONLY'
+
+    # Update tags.
+    def update_tag(path, tag):
+        try:
+            data[path][1] = tag
+        except KeyError:
+            data[path] = [path, tag, '']
+
+    for name in llndk:
+        update_tag('/system/${LIB}/' + name + '.so', 'LL-NDK')
+
+    for name in vndk_sp:
+        update_tag('/system/${LIB}/vndk-sp/' + name + '.so', 'VNDK-SP')
+
+    for name in vndk:
+        update_tag('/system/${LIB}/' + name + '.so', 'VNDK')
+        update_tag('/system/${LIB}/vndk/' + name + '.so', 'VNDK')
+
+    # Write updated eligible list file.
+    with open(args.output, 'w') as fp:
+        writer = csv.writer(fp, lineterminator='\n')
+        writer.writerow(header)
+        writer.writerows(sorted(data.values()))
+
+if __name__ == '__main__':
+    sys.exit(main())