86 lines
2.7 KiB
Python
86 lines
2.7 KiB
Python
|
# Copyright 2016 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.
|
||
|
|
|
||
|
|
|
||
|
|
from recipe_engine import recipe_api
|
||
|
|
|
||
|
|
import default
|
||
|
|
|
||
|
|
|
||
|
|
"""SSH flavor, used for running code on a remote device via SSH.
|
||
|
|
|
||
|
|
Must be subclassed to set self.device_dirs. The default implementation assumes
|
||
|
|
a Linux-based device.
|
||
|
|
"""
|
||
|
|
|
||
|
|
|
||
|
|
class SSHFlavor(default.DefaultFlavor):
|
||
|
|
|
||
|
|
def __init__(self, m):
|
||
|
|
super(SSHFlavor, self).__init__(m)
|
||
|
|
self._user_ip = ''
|
||
|
|
|
||
|
|
@property
|
||
|
|
def user_ip(self):
|
||
|
|
if not self._user_ip:
|
||
|
|
path = self.m.path.expanduser('~/ssh_machine.json')
|
||
|
|
ssh_info = self.m.file.read_json('read ssh_machine.json', path,
|
||
|
|
test_data={'user_ip':'foo@127.0.0.1'})
|
||
|
|
self._user_ip = ssh_info.get(u'user_ip')
|
||
|
|
return self._user_ip
|
||
|
|
|
||
|
|
def ssh(self, title, *cmd, **kwargs):
|
||
|
|
if 'infra_step' not in kwargs:
|
||
|
|
kwargs['infra_step'] = True
|
||
|
|
|
||
|
|
ssh_cmd = ['ssh', '-oConnectTimeout=15', '-oBatchMode=yes',
|
||
|
|
'-t', '-t', self.user_ip] + list(cmd)
|
||
|
|
|
||
|
|
return self._run(title, ssh_cmd, **kwargs)
|
||
|
|
|
||
|
|
def ensure_device_dir(self, path):
|
||
|
|
self.ssh('mkdir %s' % path, 'mkdir', '-p', path)
|
||
|
|
|
||
|
|
def install(self):
|
||
|
|
self.ensure_device_dir(self.device_dirs.resource_dir)
|
||
|
|
self.create_clean_device_dir(self.device_dirs.bin_dir)
|
||
|
|
|
||
|
|
def create_clean_device_dir(self, path):
|
||
|
|
# use -f to silently return if path doesn't exist
|
||
|
|
self.ssh('rm %s' % path, 'rm', '-rf', path)
|
||
|
|
self.ensure_device_dir(path)
|
||
|
|
|
||
|
|
def read_file_on_device(self, path, **kwargs):
|
||
|
|
rv = self.ssh('read %s' % path,
|
||
|
|
'cat', path, stdout=self.m.raw_io.output(),
|
||
|
|
**kwargs)
|
||
|
|
return rv.stdout.rstrip() if rv and rv.stdout else None
|
||
|
|
|
||
|
|
def remove_file_on_device(self, path):
|
||
|
|
# use -f to silently return if path doesn't exist
|
||
|
|
self.ssh('rm %s' % path, 'rm', '-f', path)
|
||
|
|
|
||
|
|
def scp_device_path(self, device_path):
|
||
|
|
return '%s:%s' % (self.user_ip, device_path)
|
||
|
|
|
||
|
|
def copy_file_to_device(self, host_path, device_path):
|
||
|
|
device_path = self.scp_device_path(device_path)
|
||
|
|
self._run('scp %s %s' % (host_path, device_path),
|
||
|
|
['scp', host_path, device_path], infra_step=True)
|
||
|
|
|
||
|
|
# TODO(benjaminwagner): implement with rsync
|
||
|
|
#def copy_directory_contents_to_device(self, host_path, device_path):
|
||
|
|
|
||
|
|
# TODO(benjaminwagner): implement with rsync
|
||
|
|
#def copy_directory_contents_to_host(self, device_path, host_path):
|
||
|
|
|
||
|
|
def step(self, name, cmd, **kwargs):
|
||
|
|
# Push and run cmd
|
||
|
|
host_path = self.host_dirs.bin_dir.join(cmd[0])
|
||
|
|
cmd[0] = self.device_path_join(self.device_dirs.bin_dir, cmd[0])
|
||
|
|
self.copy_file_to_device(host_path, cmd[0])
|
||
|
|
|
||
|
|
self.ssh('make %s executable' % name, 'chmod', '+x', cmd[0])
|
||
|
|
self.ssh(str(name), *cmd)
|