mirror of
https://github.com/flutter/flutter.git
synced 2025-06-03 00:51:18 +00:00
93e662abaa
This moves the various copies of port forwarding code in the Device subclasses into the ProtocolDiscovery class. * move port forwarding to a common location * throw exception if protocol Uri is not discovered or port forwarding fails * cancel discovery protocol subscriptions on iOS launches (wasn't happening before) * fix iOS port forwarding to match other implementations * add tests
104 lines
3.4 KiB
Dart
104 lines
3.4 KiB
Dart
// 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.
|
|
|
|
import 'dart:async';
|
|
|
|
import 'base/common.dart';
|
|
import 'base/os.dart';
|
|
import 'device.dart';
|
|
import 'globals.dart';
|
|
|
|
/// Discover service protocol on a device
|
|
/// and forward the service protocol device port to the host.
|
|
class ProtocolDiscovery {
|
|
/// [logReader] - a [DeviceLogReader] to look for service messages in.
|
|
ProtocolDiscovery(DeviceLogReader logReader, String serviceName,
|
|
{this.portForwarder, this.hostPort, this.defaultHostPort})
|
|
: _logReader = logReader, _serviceName = serviceName {
|
|
assert(_logReader != null);
|
|
_subscription = _logReader.logLines.listen(_onLine);
|
|
assert(portForwarder == null || defaultHostPort != null);
|
|
}
|
|
|
|
factory ProtocolDiscovery.observatory(DeviceLogReader logReader,
|
|
{DevicePortForwarder portForwarder, int hostPort}) =>
|
|
new ProtocolDiscovery(logReader, kObservatoryService,
|
|
portForwarder: portForwarder,
|
|
hostPort: hostPort,
|
|
defaultHostPort: kDefaultObservatoryPort);
|
|
|
|
factory ProtocolDiscovery.diagnosticService(DeviceLogReader logReader,
|
|
{DevicePortForwarder portForwarder, int hostPort}) =>
|
|
new ProtocolDiscovery(logReader, kDiagnosticService,
|
|
portForwarder: portForwarder,
|
|
hostPort: hostPort,
|
|
defaultHostPort: kDefaultDiagnosticPort);
|
|
|
|
static const String kObservatoryService = 'Observatory';
|
|
static const String kDiagnosticService = 'Diagnostic server';
|
|
|
|
final DeviceLogReader _logReader;
|
|
final String _serviceName;
|
|
final DevicePortForwarder portForwarder;
|
|
int hostPort;
|
|
final int defaultHostPort;
|
|
|
|
Completer<Uri> _completer = new Completer<Uri>();
|
|
StreamSubscription<String> _subscription;
|
|
|
|
/// The [Future] returned by this function will complete when the next service
|
|
/// Uri is found.
|
|
Future<Uri> nextUri() async {
|
|
Uri deviceUri = await _completer.future.timeout(
|
|
const Duration(seconds: 60), onTimeout: () {
|
|
throwToolExit('Timeout while attempting to retrieve Uri for $_serviceName');
|
|
}
|
|
);
|
|
printTrace('$_serviceName Uri on device: $deviceUri');
|
|
Uri hostUri;
|
|
if (portForwarder != null) {
|
|
int devicePort = deviceUri.port;
|
|
hostPort ??= await findPreferredPort(defaultHostPort);
|
|
hostPort = await portForwarder
|
|
.forward(devicePort, hostPort: hostPort)
|
|
.timeout(const Duration(seconds: 60), onTimeout: () {
|
|
throwToolExit('Timeout while atempting to foward device port $devicePort');
|
|
});
|
|
printTrace('Forwarded host port $hostPort to device port $devicePort');
|
|
hostUri = deviceUri.replace(port: hostPort);
|
|
} else {
|
|
hostUri = deviceUri;
|
|
}
|
|
printStatus('$_serviceName listening on $hostUri');
|
|
return hostUri;
|
|
}
|
|
|
|
void cancel() {
|
|
_subscription.cancel();
|
|
}
|
|
|
|
void _onLine(String line) {
|
|
Uri uri;
|
|
String prefix = '$_serviceName listening on ';
|
|
int index = line.indexOf(prefix + 'http://');
|
|
if (index >= 0) {
|
|
try {
|
|
uri = Uri.parse(line.substring(index + prefix.length));
|
|
} catch (_) {
|
|
// Ignore errors.
|
|
}
|
|
}
|
|
if (uri != null)
|
|
_located(uri);
|
|
}
|
|
|
|
void _located(Uri uri) {
|
|
assert(_completer != null);
|
|
assert(!_completer.isCompleted);
|
|
|
|
_completer.complete(uri);
|
|
_completer = new Completer<Uri>();
|
|
}
|
|
}
|