# Copyright 2018 - The Android Open Source Project
|
#
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
# you may not use this file except in compliance with the License.
|
# You may obtain a copy of the License at
|
#
|
# http://www.apache.org/licenses/LICENSE-2.0
|
#
|
# Unless required by applicable law or agreed to in writing, software
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
# See the License for the specific language governing permissions and
|
# limitations under the License.
|
r"""Reconnect entry point.
|
|
Reconnect will:
|
- re-establish ssh tunnels for adb/vnc port forwarding for a remote instance
|
- adb connect to forwarded ssh port for remote instance
|
- restart vnc for remote/local instances
|
"""
|
|
from __future__ import print_function
|
|
import getpass
|
import re
|
|
from acloud.delete import delete
|
from acloud.internal.lib import auth
|
from acloud.internal.lib import android_compute_client
|
from acloud.internal.lib import utils
|
from acloud.internal.lib.adb_tools import AdbTools
|
from acloud.list import list as list_instance
|
from acloud.public import config
|
|
_RE_DISPLAY = re.compile(r"([\d]+)x([\d]+)\s.*")
|
_VNC_STARTED_PATTERN = "ssvnc vnc://127.0.0.1:%(vnc_port)d"
|
|
|
def StartVnc(vnc_port, display):
|
"""Start vnc connect to AVD.
|
|
Confirm whether there is already a connection before VNC connection.
|
If there is a connection, it will not be connected. If not, connect it.
|
Before reconnecting, clear old disconnect ssvnc viewer.
|
|
Args:
|
vnc_port: Integer of vnc port number.
|
display: String, vnc connection resolution. e.g., 1080x720 (240)
|
"""
|
vnc_started_pattern = _VNC_STARTED_PATTERN % {"vnc_port": vnc_port}
|
if not utils.IsCommandRunning(vnc_started_pattern):
|
#clean old disconnect ssvnc viewer.
|
delete.CleanupSSVncviewer(vnc_port)
|
|
match = _RE_DISPLAY.match(display)
|
if match:
|
utils.LaunchVncClient(vnc_port, match.group(1), match.group(2))
|
else:
|
utils.LaunchVncClient(vnc_port)
|
|
|
def AddPublicSshRsaToInstance(cfg, user, instance_name):
|
"""Add the public rsa key to the instance's metadata.
|
|
When the public key doesn't exist in the metadata, it will add it.
|
|
Args:
|
cfg: An AcloudConfig instance.
|
user: String, the ssh username to access instance.
|
instance_name: String, instance name.
|
"""
|
credentials = auth.CreateCredentials(cfg)
|
compute_client = android_compute_client.AndroidComputeClient(
|
cfg, credentials)
|
compute_client.AddSshRsaInstanceMetadata(
|
cfg.zone,
|
user,
|
cfg.ssh_public_key_path,
|
instance_name)
|
|
|
def ReconnectInstance(ssh_private_key_path, instance):
|
"""Reconnect adb/vnc/ssh to the specified instance.
|
|
Args:
|
ssh_private_key_path: Path to the private key file.
|
e.g. ~/.ssh/acloud_rsa
|
instance: list.Instance() object.
|
"""
|
adb_cmd = AdbTools(instance.forwarding_adb_port)
|
vnc_port = instance.forwarding_vnc_port
|
# ssh tunnel is up but device is disconnected on adb
|
if instance.ssh_tunnel_is_connected and not adb_cmd.IsAdbConnectionAlive():
|
adb_cmd.DisconnectAdb()
|
adb_cmd.ConnectAdb()
|
# ssh tunnel is down and it's a remote instance
|
elif not instance.ssh_tunnel_is_connected and not instance.islocal:
|
adb_cmd.DisconnectAdb()
|
forwarded_ports = utils.AutoConnect(
|
instance.ip,
|
ssh_private_key_path,
|
utils.AVD_PORT_DICT[instance.avd_type].vnc_port,
|
utils.AVD_PORT_DICT[instance.avd_type].adb_port,
|
getpass.getuser())
|
vnc_port = forwarded_ports.vnc_port
|
|
if vnc_port:
|
StartVnc(vnc_port, instance.display)
|
|
|
def Run(args):
|
"""Run reconnect.
|
|
Args:
|
args: Namespace object from argparse.parse_args.
|
"""
|
cfg = config.GetAcloudConfig(args)
|
instances_to_reconnect = []
|
if args.instance_names is not None:
|
# user input instance name to get instance object.
|
instances_to_reconnect = list_instance.GetInstancesFromInstanceNames(
|
cfg, args.instance_names)
|
if not instances_to_reconnect:
|
instances_to_reconnect = list_instance.ChooseInstances(cfg, args.all)
|
for instance in instances_to_reconnect:
|
AddPublicSshRsaToInstance(cfg, getpass.getuser(), instance.name)
|
ReconnectInstance(cfg.ssh_private_key_path, instance)
|
