Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(92)

Side by Side Diff: build/android/pylib/utils/md5sum.py

Issue 1108173002: Roll //build, //native_client, and a few more targets of opportunity. Base URL: git@github.com:domokit/mojo.git@master
Patch Set: Test fix Created 5 years, 8 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « build/android/pylib/utils/isolator.py ('k') | build/android/pylib/utils/md5sum_test.py » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 # Copyright 2014 The Chromium Authors. All rights reserved. 1 # Copyright 2014 The Chromium Authors. All rights reserved.
2 # Use of this source code is governed by a BSD-style license that can be 2 # Use of this source code is governed by a BSD-style license that can be
3 # found in the LICENSE file. 3 # found in the LICENSE file.
4 4
5 import collections 5 import collections
6 import logging 6 import logging
7 import os 7 import os
8 import re
8 import tempfile 9 import tempfile
9 import types 10 import types
10 11
11 from pylib import cmd_helper 12 from pylib import cmd_helper
12 from pylib import constants 13 from pylib import constants
13 from pylib.utils import device_temp_file 14 from pylib.utils import device_temp_file
14 15
15 HashAndPath = collections.namedtuple('HashAndPath', ['hash', 'path'])
16
17 MD5SUM_DEVICE_LIB_PATH = '/data/local/tmp/md5sum/' 16 MD5SUM_DEVICE_LIB_PATH = '/data/local/tmp/md5sum/'
18 MD5SUM_DEVICE_BIN_PATH = MD5SUM_DEVICE_LIB_PATH + 'md5sum_bin' 17 MD5SUM_DEVICE_BIN_PATH = MD5SUM_DEVICE_LIB_PATH + 'md5sum_bin'
19 18
20 MD5SUM_DEVICE_SCRIPT_FORMAT = ( 19 MD5SUM_DEVICE_SCRIPT_FORMAT = (
21 'test -f {path} -o -d {path} ' 20 'test -f {path} -o -d {path} '
22 '&& LD_LIBRARY_PATH={md5sum_lib} {device_pie_wrapper} {md5sum_bin} {path}') 21 '&& LD_LIBRARY_PATH={md5sum_lib} {device_pie_wrapper} {md5sum_bin} {path}')
23 22
23 _STARTS_WITH_CHECKSUM_RE = re.compile(r'^\s*[0-9a-fA-f]{32}\s+')
24
24 25
25 def CalculateHostMd5Sums(paths): 26 def CalculateHostMd5Sums(paths):
26 """Calculates the MD5 sum value for all items in |paths|. 27 """Calculates the MD5 sum value for all items in |paths|.
27 28
28 Args: 29 Args:
29 paths: A list of host paths to md5sum. 30 paths: A list of host paths to md5sum.
30 Returns: 31 Returns:
31 A list of named tuples with 'hash' and 'path' attributes. 32 A dict mapping paths to their respective md5sum checksums.
32 """ 33 """
33 if isinstance(paths, basestring): 34 if isinstance(paths, basestring):
34 paths = [paths] 35 paths = [paths]
35 36
36 out = cmd_helper.GetCmdOutput( 37 out = cmd_helper.GetCmdOutput(
37 [os.path.join(constants.GetOutDirectory(), 'md5sum_bin_host')] + 38 [os.path.join(constants.GetOutDirectory(), 'md5sum_bin_host')] +
38 [p for p in paths]) 39 [p for p in paths])
39 return [HashAndPath(*l.split(None, 1)) for l in out.splitlines()] 40
41 return _ParseMd5SumOutput(out.splitlines())
40 42
41 43
42 def CalculateDeviceMd5Sums(paths, device): 44 def CalculateDeviceMd5Sums(paths, device):
43 """Calculates the MD5 sum value for all items in |paths|. 45 """Calculates the MD5 sum value for all items in |paths|.
44 46
45 Args: 47 Args:
46 paths: A list of device paths to md5sum. 48 paths: A list of device paths to md5sum.
47 Returns: 49 Returns:
48 A list of named tuples with 'hash' and 'path' attributes. 50 A dict mapping paths to their respective md5sum checksums.
49 """ 51 """
50 if isinstance(paths, basestring): 52 if isinstance(paths, basestring):
51 paths = [paths] 53 paths = [paths]
52 54
53 if not device.FileExists(MD5SUM_DEVICE_BIN_PATH): 55 if not device.FileExists(MD5SUM_DEVICE_BIN_PATH):
54 device.adb.Push( 56 device.adb.Push(
55 os.path.join(constants.GetOutDirectory(), 'md5sum_dist'), 57 os.path.join(constants.GetOutDirectory(), 'md5sum_dist'),
56 MD5SUM_DEVICE_LIB_PATH) 58 MD5SUM_DEVICE_LIB_PATH)
57 59
58 out = [] 60 out = []
59 61
60 with tempfile.NamedTemporaryFile() as md5sum_script_file: 62 with tempfile.NamedTemporaryFile() as md5sum_script_file:
61 with device_temp_file.DeviceTempFile( 63 with device_temp_file.DeviceTempFile(
62 device.adb) as md5sum_device_script_file: 64 device.adb) as md5sum_device_script_file:
63 device_pie_wrapper = device.GetDevicePieWrapper() 65 device_pie_wrapper = device.GetDevicePieWrapper()
64 md5sum_script = ( 66 md5sum_script = (
65 MD5SUM_DEVICE_SCRIPT_FORMAT.format( 67 MD5SUM_DEVICE_SCRIPT_FORMAT.format(
66 path=p, md5sum_lib=MD5SUM_DEVICE_LIB_PATH, 68 path=p, md5sum_lib=MD5SUM_DEVICE_LIB_PATH,
67 device_pie_wrapper=device_pie_wrapper, 69 device_pie_wrapper=device_pie_wrapper,
68 md5sum_bin=MD5SUM_DEVICE_BIN_PATH) 70 md5sum_bin=MD5SUM_DEVICE_BIN_PATH)
69 for p in paths) 71 for p in paths)
70 md5sum_script_file.write('; '.join(md5sum_script)) 72 md5sum_script_file.write('; '.join(md5sum_script))
71 md5sum_script_file.flush() 73 md5sum_script_file.flush()
72 device.adb.Push(md5sum_script_file.name, md5sum_device_script_file.name) 74 device.adb.Push(md5sum_script_file.name, md5sum_device_script_file.name)
73 out = device.RunShellCommand(['sh', md5sum_device_script_file.name]) 75 out = device.RunShellCommand(['sh', md5sum_device_script_file.name])
74 76
75 return [HashAndPath(*l.split(None, 1)) for l in out if l] 77 return _ParseMd5SumOutput(out)
76 78
79
80 def _ParseMd5SumOutput(out):
81 hash_and_path = (l.split(None, 1) for l in out
82 if l and _STARTS_WITH_CHECKSUM_RE.match(l))
83 return dict((p, h) for h, p in hash_and_path)
84
OLDNEW
« no previous file with comments | « build/android/pylib/utils/isolator.py ('k') | build/android/pylib/utils/md5sum_test.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698