| Index: build/android/pylib/utils/device_temp_file.py
|
| diff --git a/build/android/pylib/utils/device_temp_file.py b/build/android/pylib/utils/device_temp_file.py
|
| index ba95395d8b525d320c1172844ca5bce44b700e99..2a9b45101803e18f2ca8046a25e8bb126e5624b4 100644
|
| --- a/build/android/pylib/utils/device_temp_file.py
|
| +++ b/build/android/pylib/utils/device_temp_file.py
|
| @@ -9,42 +9,34 @@
|
| import random
|
| import time
|
|
|
| -from pylib import cmd_helper
|
| -from pylib.device import device_errors
|
| -
|
| -
|
| class DeviceTempFile(object):
|
| - def __init__(self, adb, prefix='temp_file', suffix=''):
|
| + def __init__(self, device, prefix='temp_file', suffix=''):
|
| """Find an unused temporary file path in the devices external directory.
|
|
|
| When this object is closed, the file will be deleted on the device.
|
|
|
| Args:
|
| - adb: An instance of AdbWrapper
|
| + device: An instance of DeviceUtils
|
| prefix: The prefix of the name of the temp file.
|
| suffix: The suffix of the name of the temp file.
|
| """
|
| - self._adb = adb
|
| - external_storage = self._adb.Shell('echo $EXTERNAL_STORAGE').rstrip()
|
| - # make sure that external storage is readable
|
| - self._adb.Shell('test -d %s' % external_storage)
|
| + self._device = device
|
| while True:
|
| - self.name = '{path}/{prefix}-{time:d}-{nonce:d}{suffix}'.format(
|
| - path=external_storage, prefix=prefix, time=int(time.time()),
|
| - nonce=random.randint(0, 1000000), suffix=suffix)
|
| - self.name_quoted = cmd_helper.SingleQuote(self.name)
|
| - try:
|
| - self._adb.Shell('test -e %s' % self.name_quoted)
|
| - except device_errors.AdbCommandFailedError:
|
| - break # file does not exist
|
| -
|
| - # Immediately touch the file, so other temp files can't get the same name.
|
| - self._adb.Shell('touch %s' % self.name_quoted)
|
| + i = random.randint(0, 1000000)
|
| + self.name = '%s/%s-%d-%010d%s' % (
|
| + self._device.GetExternalStoragePath(),
|
| + prefix, int(time.time()), i, suffix)
|
| + if not self._device.FileExists(self.name):
|
| + break
|
| + # Immediately create an empty file so that other temp files can't
|
| + # be given the same name.
|
| + # |as_root| must be set to False due to the implementation of |WriteFile|.
|
| + # Having |as_root| be True may cause infinite recursion.
|
| + self._device.WriteFile(self.name, '', as_root=False)
|
|
|
| def close(self):
|
| """Deletes the temporary file from the device."""
|
| - # we use -f to ignore errors if the file is already gone
|
| - self._adb.Shell('rm -f %s' % self.name_quoted)
|
| + self._device.RunShellCommand(['rm', self.name])
|
|
|
| def __enter__(self):
|
| return self
|
|
|