forked from auracaster/bumble_mirror
96 lines
3.9 KiB
Python
96 lines
3.9 KiB
Python
# Copyright 2021-2022 Google LLC
|
|
#
|
|
# 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
|
|
#
|
|
# https://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.
|
|
|
|
# -----------------------------------------------------------------------------
|
|
# Imports
|
|
# -----------------------------------------------------------------------------
|
|
import logging
|
|
|
|
from .common import Transport, AsyncPipeSink
|
|
from ..link import RemoteLink
|
|
from ..controller import Controller
|
|
|
|
# -----------------------------------------------------------------------------
|
|
# Logging
|
|
# -----------------------------------------------------------------------------
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
# -----------------------------------------------------------------------------
|
|
async def open_transport(name):
|
|
'''
|
|
Open a transport by name.
|
|
The name must be <type>:<parameters>
|
|
Where <parameters> depend on the type (and may be empty for some types).
|
|
The supported types are: serial,udp,tcp,pty,usb
|
|
'''
|
|
scheme, *spec = name.split(':', 1)
|
|
if scheme == 'serial' and spec:
|
|
from .serial import open_serial_transport
|
|
return await open_serial_transport(spec[0])
|
|
elif scheme == 'udp' and spec:
|
|
from .udp import open_udp_transport
|
|
return await open_udp_transport(spec[0])
|
|
elif scheme == 'tcp-client' and spec:
|
|
from .tcp_client import open_tcp_client_transport
|
|
return await open_tcp_client_transport(spec[0])
|
|
elif scheme == 'tcp-server' and spec:
|
|
from .tcp_server import open_tcp_server_transport
|
|
return await open_tcp_server_transport(spec[0])
|
|
elif scheme == 'ws-client' and spec:
|
|
from .ws_client import open_ws_client_transport
|
|
return await open_ws_client_transport(spec[0])
|
|
elif scheme == 'ws-server' and spec:
|
|
from .ws_server import open_ws_server_transport
|
|
return await open_ws_server_transport(spec[0])
|
|
elif scheme == 'pty':
|
|
from .pty import open_pty_transport
|
|
return await open_pty_transport(spec[0] if spec else None)
|
|
elif scheme == 'file':
|
|
from .file import open_file_transport
|
|
return await open_file_transport(spec[0] if spec else None)
|
|
elif scheme == 'vhci':
|
|
from .vhci import open_vhci_transport
|
|
return await open_vhci_transport(spec[0] if spec else None)
|
|
elif scheme == 'hci-socket':
|
|
from .hci_socket import open_hci_socket_transport
|
|
return await open_hci_socket_transport(spec[0] if spec else None)
|
|
elif scheme == 'usb':
|
|
from .usb import open_usb_transport
|
|
return await open_usb_transport(spec[0] if spec else None)
|
|
elif scheme == 'pyusb':
|
|
from .pyusb import open_pyusb_transport
|
|
return await open_pyusb_transport(spec[0] if spec else None)
|
|
elif scheme == 'android-emulator':
|
|
from .android_emulator import open_android_emulator_transport
|
|
return await open_android_emulator_transport(spec[0] if spec else None)
|
|
else:
|
|
raise ValueError('unknown transport scheme')
|
|
|
|
|
|
# -----------------------------------------------------------------------------
|
|
async def open_transport_or_link(name):
|
|
if name.startswith('link-relay:'):
|
|
link = RemoteLink(name[11:])
|
|
await link.wait_until_connected()
|
|
controller = Controller('remote', link = link)
|
|
|
|
class LinkTransport(Transport):
|
|
async def close(self):
|
|
link.close()
|
|
|
|
return LinkTransport(controller, AsyncPipeSink(controller))
|
|
else:
|
|
return await open_transport(name)
|