-
Notifications
You must be signed in to change notification settings - Fork 195
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
11 changed files
with
964 additions
and
2 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,61 @@ | ||
import asyncio | ||
import logging | ||
|
||
import click | ||
from cryptography.hazmat.primitives.asymmetric import rsa | ||
|
||
from pymobiledevice3.cli.cli_common import print_json | ||
from pymobiledevice3.remote.bonjour import get_iphone_address | ||
from pymobiledevice3.remote.core_device_tunnel_service import create_core_device_tunnel_service | ||
from pymobiledevice3.remote.remote_service_discovery import RSD_PORT, RemoteServiceDiscoveryService | ||
|
||
logger = logging.getLogger(__name__) | ||
|
||
|
||
@click.group() | ||
def cli(): | ||
""" remote cli """ | ||
pass | ||
|
||
|
||
@cli.group('remote') | ||
def remote_cli(): | ||
""" remote options """ | ||
pass | ||
|
||
|
||
@remote_cli.command('rsd-info') | ||
@click.option('--color/--no-color', default=True) | ||
def rsd_info(color: bool): | ||
""" show info extracted from RSD peer """ | ||
hostname = asyncio.run(get_iphone_address()) | ||
|
||
with RemoteServiceDiscoveryService((hostname, RSD_PORT)) as rsd: | ||
print_json(rsd.peer_info, colored=color) | ||
|
||
|
||
@remote_cli.command('create-listener') | ||
@click.option('-p', '--protocol', type=click.Choice(['quic', 'udp'])) | ||
@click.option('--color/--no-color', default=True) | ||
def create_listener(protocol: str, color: bool): | ||
""" start a remote listener """ | ||
hostname = asyncio.run(get_iphone_address()) | ||
|
||
private_key = rsa.generate_private_key(public_exponent=65537, key_size=2048) | ||
with RemoteServiceDiscoveryService((hostname, RSD_PORT)) as rsd: | ||
with create_core_device_tunnel_service(rsd, autopair=True) as service: | ||
print_json(service.create_listener(private_key, protocol=protocol), colored=color) | ||
|
||
|
||
@remote_cli.command('start-quic-tunnel') | ||
@click.option('--color/--no-color', default=True) | ||
def start_quic_tunnel(color: bool): | ||
""" start quic tunnel """ | ||
logger.critical('This is a WIP command. Will only print the required parameters for the quic connection') | ||
|
||
hostname = asyncio.run(get_iphone_address()) | ||
|
||
private_key = rsa.generate_private_key(public_exponent=65537, key_size=2048) | ||
with RemoteServiceDiscoveryService((hostname, RSD_PORT)) as rsd: | ||
with create_core_device_tunnel_service(rsd, autopair=True) as service: | ||
print_json(asyncio.run(service.start_quic_tunnel(private_key)), colored=color) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Empty file.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,40 @@ | ||
import asyncio | ||
from socket import AF_INET6, inet_ntop | ||
|
||
from ifaddr import get_adapters | ||
from zeroconf import ServiceBrowser, ServiceListener, Zeroconf | ||
from zeroconf.const import _TYPE_AAAA | ||
|
||
|
||
class RemotedListener(ServiceListener): | ||
def __init__(self): | ||
super().__init__() | ||
self.is_finished = asyncio.Event() | ||
|
||
def add_service(self, zc: Zeroconf, type_: str, name: str) -> None: | ||
if name == 'ncm._remoted._tcp.local.': | ||
service_info = zc.get_service_info(type_, name) | ||
records = zc.cache.async_entries_with_name(service_info.server) | ||
for record in records: | ||
if record.type == _TYPE_AAAA: | ||
self.record = record | ||
self.is_finished.set() | ||
|
||
|
||
async def try_get_iphone_address(adapter): | ||
ip = adapter.ips[0].ip[0] | ||
zeroconf = Zeroconf(interfaces=[ip]) | ||
waiter_task = asyncio.create_task(zeroconf.notify_event.wait()) | ||
listener = RemotedListener() | ||
ServiceBrowser(zeroconf, '_remoted._tcp.local.', listener) | ||
await waiter_task | ||
await listener.is_finished.wait() | ||
return inet_ntop(AF_INET6, listener.record.address) + '%' + adapter.nice_name | ||
|
||
|
||
async def get_iphone_address(): | ||
adapters = get_adapters() | ||
adapters = [adapter for adapter in adapters if adapter.ips[0].is_IPv6] | ||
tasks = [asyncio.create_task(try_get_iphone_address(adapter)) for adapter in adapters] | ||
finished, unfinished = await asyncio.wait(tasks, return_when=asyncio.FIRST_COMPLETED) | ||
return list(finished)[0].result() |
Oops, something went wrong.