| Index: build/android/pylib/utils/md5sum.py
|
| diff --git a/build/android/pylib/utils/md5sum.py b/build/android/pylib/utils/md5sum.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..3e61c8facb14f806ec4b4c29f5eb88f084672949
|
| --- /dev/null
|
| +++ b/build/android/pylib/utils/md5sum.py
|
| @@ -0,0 +1,91 @@
|
| +# Copyright 2014 The Chromium Authors. All rights reserved.
|
| +# Use of this source code is governed by a BSD-style license that can be
|
| +# found in the LICENSE file.
|
| +
|
| +import collections
|
| +import logging
|
| +import os
|
| +import re
|
| +import tempfile
|
| +import types
|
| +
|
| +from pylib import cmd_helper
|
| +from pylib import constants
|
| +from pylib.utils import device_temp_file
|
| +
|
| +MD5SUM_DEVICE_LIB_PATH = '/data/local/tmp/md5sum/'
|
| +MD5SUM_DEVICE_BIN_PATH = MD5SUM_DEVICE_LIB_PATH + 'md5sum_bin'
|
| +
|
| +MD5SUM_DEVICE_SCRIPT_FORMAT = (
|
| + 'test -f {path} -o -d {path} '
|
| + '&& LD_LIBRARY_PATH={md5sum_lib} {md5sum_bin} {path}')
|
| +
|
| +_STARTS_WITH_CHECKSUM_RE = re.compile(r'^\s*[0-9a-fA-F]{32}\s+')
|
| +
|
| +
|
| +def CalculateHostMd5Sums(paths):
|
| + """Calculates the MD5 sum value for all items in |paths|.
|
| +
|
| + Directories are traversed recursively and the MD5 sum of each file found is
|
| + reported in the result.
|
| +
|
| + Args:
|
| + paths: A list of host paths to md5sum.
|
| + Returns:
|
| + A dict mapping file paths to their respective md5sum checksums.
|
| + """
|
| + if isinstance(paths, basestring):
|
| + paths = [paths]
|
| +
|
| + md5sum_bin_host_path = os.path.join(
|
| + constants.GetOutDirectory(), 'md5sum_bin_host')
|
| + if not os.path.exists(md5sum_bin_host_path):
|
| + raise IOError('File not built: %s' % md5sum_bin_host_path)
|
| + out = cmd_helper.GetCmdOutput([md5sum_bin_host_path] + [p for p in paths])
|
| +
|
| + return _ParseMd5SumOutput(out.splitlines())
|
| +
|
| +
|
| +def CalculateDeviceMd5Sums(paths, device):
|
| + """Calculates the MD5 sum value for all items in |paths|.
|
| +
|
| + Directories are traversed recursively and the MD5 sum of each file found is
|
| + reported in the result.
|
| +
|
| + Args:
|
| + paths: A list of device paths to md5sum.
|
| + Returns:
|
| + A dict mapping file paths to their respective md5sum checksums.
|
| + """
|
| + if isinstance(paths, basestring):
|
| + paths = [paths]
|
| +
|
| + if not device.FileExists(MD5SUM_DEVICE_BIN_PATH):
|
| + md5sum_dist_path = os.path.join(constants.GetOutDirectory(), 'md5sum_dist')
|
| + if not os.path.exists(md5sum_dist_path):
|
| + raise IOError('File not built: %s' % md5sum_dist_path)
|
| + device.adb.Push(md5sum_dist_path, MD5SUM_DEVICE_LIB_PATH)
|
| +
|
| + out = []
|
| +
|
| + with tempfile.NamedTemporaryFile() as md5sum_script_file:
|
| + with device_temp_file.DeviceTempFile(
|
| + device.adb) as md5sum_device_script_file:
|
| + md5sum_script = (
|
| + MD5SUM_DEVICE_SCRIPT_FORMAT.format(
|
| + path=p, md5sum_lib=MD5SUM_DEVICE_LIB_PATH,
|
| + md5sum_bin=MD5SUM_DEVICE_BIN_PATH)
|
| + for p in paths)
|
| + md5sum_script_file.write('; '.join(md5sum_script))
|
| + md5sum_script_file.flush()
|
| + device.adb.Push(md5sum_script_file.name, md5sum_device_script_file.name)
|
| + out = device.RunShellCommand(['sh', md5sum_device_script_file.name])
|
| +
|
| + return _ParseMd5SumOutput(out)
|
| +
|
| +
|
| +def _ParseMd5SumOutput(out):
|
| + hash_and_path = (l.split(None, 1) for l in out
|
| + if l and _STARTS_WITH_CHECKSUM_RE.match(l))
|
| + return dict((p, h) for h, p in hash_and_path)
|
| +
|
|
|