1#!/bin/bash 2# 3# Copyright 2016 The Android Open Source Project 4# 5# Licensed under the Apache License, Version 2.0 (the "License"); 6# you may not use this file except in compliance with the License. 7# You may obtain a copy of the License at 8# 9# http://www.apache.org/licenses/LICENSE-2.0 10# 11# Unless required by applicable law or agreed to in writing, software 12# distributed under the License is distributed on an "AS IS" BASIS, 13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14# See the License for the specific language governing permissions and 15# limitations under the License. 16# 17 18set -euo pipefail 19 20function set_route { 21 local mode="${1}" 22 local ip="${2}" 23 local host="${3}" 24 local gw="${4}" 25 26 if [ "${mode}" = "ethernet" ]; then 27 ${ip} route del "${host}" via "${gw}" 28 else 29 # FIXME: This *should* work for IPv6, but it returns EINVAL. 30 ${ip} route add "${host}" via "${gw}" 31 dbus-send --system --dest=org.chromium.flimflam --print-reply / \ 32 org.chromium.flimflam.Manager.SetServiceOrder \ 33 string:"vpn,wifi,ethernet,wimax,cellular" 34 fi 35} 36 37function find_route { 38 local mode="${1}" 39 local host="${2}" 40 local ip="ip -4" 41 42 if [[ "${host}" = *:* ]]; then 43 ip="ip -6" 44 fi 45 46 set_route "${mode}" "${ip}" "${host}" 192.168.231.254 47 exit 0 48} 49 50function parse_netstat { 51 local mode="${1}" 52 53 while read -r proto recv_q send_q local foreign state; do 54 if [[ "${proto}" = tcp* && \ 55 ("${local}" = *:22 || "${local}" = *:2222) && \ 56 "${state}" == ESTABLISHED ]]; then 57 find_route "${mode}" "${foreign%:*}" 58 exit 0 59 fi 60 done 61 62 echo "Could not find ssh connection in netstat" 63 exit 1 64} 65 66mode="${1:-}" 67if [ "${mode}" != "wifi" -a "${mode}" != "ethernet" ]; then 68 echo "Tells shill to prioritize ethernet or wifi, and adds a route" 69 echo "back to the ssh/adb host so that the device can still be controlled" 70 echo "remotely." 71 echo "" 72 echo "usage: ${0} { ethernet | wifi }" 73 exit 1 74fi 75 76if [ "${mode}" = "ethernet" ]; then 77 # Switch the service order first, because the IP lookup might fail. 78 dbus-send --system --dest=org.chromium.flimflam --print-reply / \ 79 org.chromium.flimflam.Manager.SetServiceOrder \ 80 string:"vpn,ethernet,wifi,wimax,cellular" 81fi 82 83# Find the first connection to our local port 22 (ssh), then use it to 84# set a static route via eth0. 85# This should ideally use $SSH_CLIENT instead, but that will require enabling 86# transparent mode in sslh because $SSH_CLIENT currently points to 87# 127.0.0.1. 88netstat --tcp --numeric --wide | parse_netstat "${mode}" 89exit 0 90 91