1# Copyright 2014 The Chromium OS Authors. All rights reserved.
2# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
4"""Module to compare two machines."""
5
6from __future__ import print_function
7
8import os.path
9import sys
10import argparse
11
12from machine_manager import CrosMachine
13
14
15def PrintUsage(msg):
16  print(msg)
17  print('Usage: ')
18  print('\n compare_machines.py --chromeos_root=/path/to/chroot/ '
19        'machine1 machine2 ...')
20
21
22def Main(argv):
23
24  parser = argparse.ArgumentParser()
25  parser.add_argument(
26      '--chromeos_root',
27      default='/path/to/chromeos',
28      dest='chromeos_root',
29      help='ChromeOS root checkout directory')
30  parser.add_argument('remotes', nargs=argparse.REMAINDER)
31
32  options = parser.parse_args(argv)
33
34  machine_list = options.remotes
35  if len(machine_list) < 2:
36    PrintUsage('ERROR: Must specify at least two machines.')
37    return 1
38  elif not os.path.exists(options.chromeos_root):
39    PrintUsage('Error: chromeos_root does not exist %s' % options.chromeos_root)
40    return 1
41
42  chroot = options.chromeos_root
43  cros_machines = []
44  test_machine_checksum = None
45  for m in machine_list:
46    cm = CrosMachine(m, chroot, 'average')
47    cros_machines = cros_machines + [cm]
48    test_machine_checksum = cm.machine_checksum
49
50  ret = 0
51  for cm in cros_machines:
52    print('checksum for %s : %s' % (cm.name, cm.machine_checksum))
53    if cm.machine_checksum != test_machine_checksum:
54      ret = 1
55      print('Machine checksums do not all match')
56
57  if ret == 0:
58    print('Machines all match.')
59
60  return ret
61
62
63if __name__ == '__main__':
64  retval = Main(sys.argv[1:])
65  sys.exit(retval)
66