build/unification/zbi/compare_zbis.py

#!/usr/bin/env python2.7
# Copyright 2020 The Fuchsia Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

'''Compares the contents of two zbi files, erroring if they differ.'''

from __future__ import print_function

import argparse
import difflib
import subprocess
import sys


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--tool',
                        help='Path to the zbi tool',
                        required=True)
    parser.add_argument('--reference-zbi',
                        help='Path to the original zbi',
                        required=True)
    parser.add_argument('--generated-zbi',
                        help='Path to the generated zbi',
                        required=True)
    parser.add_argument('--stamp',
                        help='Path to the stamp file signalling success',
                        required=True)
    args = parser.parse_args()

    def get_contents(path):
        command = [args.tool, '-tv', path]
        return subprocess.check_output(command).splitlines()

    reference = get_contents(args.reference_zbi)
    generated = get_contents(args.generated_zbi)
    if reference != generated:
        print('Error: zbi\'s don\'t match')
        diff = difflib.Differ().compare(reference, generated)
        print('\n'.join(l.strip() for l in diff))
        return 1

    with open(args.stamp, 'w') as stamp_file:
        stamp_file.write('Success!')
    return 0


if __name__ == '__main__':
    sys.exit(main())