mirror of
https://github.com/flutter/flutter.git
synced 2025-06-03 00:51:18 +00:00
92 lines
2.8 KiB
Python
92 lines
2.8 KiB
Python
# 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)
|
|
|