fix(client): harden control plane and SOCKS
This commit is contained in:
1 parent
0ed7bf27b5
commit
0692bb83e1
4 files changed
+372
-169
No files matched your search
@@ -20,13 +20,13 @@ abstract final class LocalSocksTester {
|
||||
|
||||
/// Fast local readiness probe used while starting the VPN.
|
||||
///
|
||||
/// This stops after SOCKS5 authentication. Startup must not depend on an
|
||||
/// unrelated external host being reachable.
|
||||
/// This stops after authenticated SOCKS5 negotiation. Startup must not depend
|
||||
/// on an unrelated external host being reachable.
|
||||
static Future<LocalSocksTestResult> testListener({
|
||||
String host = '127.0.0.1',
|
||||
int port = 10807,
|
||||
String? username,
|
||||
String? password,
|
||||
required int port,
|
||||
required String username,
|
||||
required String password,
|
||||
}) async {
|
||||
final stopwatch = Stopwatch()..start();
|
||||
Socket? socket;
|
||||
@@ -86,26 +86,40 @@ abstract final class LocalSocksTester {
|
||||
}
|
||||
}
|
||||
|
||||
/// Full user-facing test: authenticated local SOCKS handshake plus outbound
|
||||
/// CONNECT. Credentials are optional so the helper remains usable for legacy
|
||||
/// no-auth diagnostic builds.
|
||||
/// Full user-facing test: authenticated local SOCKS handshake plus an
|
||||
/// outbound TCP CONNECT through the active ReVolt route.
|
||||
static Future<LocalSocksTestResult> test({
|
||||
String host = '127.0.0.1',
|
||||
int port = 10807,
|
||||
String? username,
|
||||
String? password,
|
||||
String targetHost = 'paladinvpn.duckdns.org',
|
||||
required int port,
|
||||
required String username,
|
||||
required String password,
|
||||
String targetHost = 'example.com',
|
||||
int targetPort = 443,
|
||||
}) async {
|
||||
const totalTimeout = Duration(seconds: 8);
|
||||
final stopwatch = Stopwatch()..start();
|
||||
Socket? socket;
|
||||
_SocketReader? reader;
|
||||
|
||||
Duration remaining() {
|
||||
final micros = totalTimeout.inMicroseconds - stopwatch.elapsedMicroseconds;
|
||||
if (micros <= 0) throw TimeoutException('SOCKS5 test timed out.', totalTimeout);
|
||||
return Duration(microseconds: micros);
|
||||
}
|
||||
|
||||
try {
|
||||
if (targetPort <= 0 || targetPort > 65535) {
|
||||
return const LocalSocksTestResult(
|
||||
ok: false,
|
||||
latencyMs: null,
|
||||
message: 'SOCKS5 test target is invalid.',
|
||||
);
|
||||
}
|
||||
|
||||
socket = await Socket.connect(
|
||||
host,
|
||||
port,
|
||||
timeout: const Duration(seconds: 3),
|
||||
timeout: remaining(),
|
||||
);
|
||||
socket.setOption(SocketOption.tcpNoDelay, true);
|
||||
reader = _SocketReader(socket);
|
||||
@@ -115,6 +129,7 @@ abstract final class LocalSocksTester {
|
||||
socket,
|
||||
username: username,
|
||||
password: password,
|
||||
timeout: remaining(),
|
||||
);
|
||||
if (!authenticated) {
|
||||
return const LocalSocksTestResult(
|
||||
@@ -145,7 +160,7 @@ abstract final class LocalSocksTester {
|
||||
]);
|
||||
await socket.flush();
|
||||
|
||||
final replyHead = await reader.readExactly(4);
|
||||
final replyHead = await reader.readExactly(4, timeout: remaining());
|
||||
if (replyHead[0] != 0x05 || replyHead[1] != 0x00) {
|
||||
return const LocalSocksTestResult(
|
||||
ok: false,
|
||||
@@ -155,7 +170,7 @@ abstract final class LocalSocksTester {
|
||||
);
|
||||
}
|
||||
|
||||
await _consumeAddress(reader, replyHead[3]);
|
||||
await _consumeAddress(reader, replyHead[3], timeout: remaining());
|
||||
stopwatch.stop();
|
||||
return LocalSocksTestResult(
|
||||
ok: true,
|
||||
@@ -189,23 +204,10 @@ abstract final class LocalSocksTester {
|
||||
static Future<bool> _authenticate(
|
||||
_SocketReader reader,
|
||||
Socket socket, {
|
||||
required String? username,
|
||||
required String? password,
|
||||
required String username,
|
||||
required String password,
|
||||
Duration timeout = const Duration(seconds: 4),
|
||||
}) async {
|
||||
final passwordAuth = username != null && password != null;
|
||||
if ((username == null) != (password == null)) return false;
|
||||
|
||||
socket.add(passwordAuth
|
||||
? const <int>[0x05, 0x01, 0x02]
|
||||
: const <int>[0x05, 0x01, 0x00]);
|
||||
await socket.flush();
|
||||
|
||||
final greeting = await reader.readExactly(2, timeout: timeout);
|
||||
if (greeting[0] != 0x05) return false;
|
||||
if (!passwordAuth) return greeting[1] == 0x00;
|
||||
if (greeting[1] != 0x02) return false;
|
||||
|
||||
final userBytes = utf8.encode(username);
|
||||
final passBytes = utf8.encode(password);
|
||||
if (userBytes.isEmpty ||
|
||||
@@ -215,6 +217,15 @@ abstract final class LocalSocksTester {
|
||||
return false;
|
||||
}
|
||||
|
||||
// ReVolt never accepts unauthenticated local SOCKS. Advertising only RFC
|
||||
// 1929 username/password auth makes a future credential-plumbing regression
|
||||
// fail closed instead of silently falling back to method 0x00.
|
||||
socket.add(const <int>[0x05, 0x01, 0x02]);
|
||||
await socket.flush();
|
||||
|
||||
final greeting = await reader.readExactly(2, timeout: timeout);
|
||||
if (greeting[0] != 0x05 || greeting[1] != 0x02) return false;
|
||||
|
||||
socket.add(<int>[
|
||||
0x01,
|
||||
userBytes.length,
|
||||
@@ -229,18 +240,24 @@ abstract final class LocalSocksTester {
|
||||
}
|
||||
|
||||
static Future<void> _consumeAddress(
|
||||
_SocketReader reader, int addressType) async {
|
||||
_SocketReader reader,
|
||||
int addressType, {
|
||||
required Duration timeout,
|
||||
}) async {
|
||||
if (addressType == 0x01) {
|
||||
await reader.readExactly(4 + 2);
|
||||
await reader.readExactly(4 + 2, timeout: timeout);
|
||||
return;
|
||||
}
|
||||
if (addressType == 0x04) {
|
||||
await reader.readExactly(16 + 2);
|
||||
await reader.readExactly(16 + 2, timeout: timeout);
|
||||
return;
|
||||
}
|
||||
if (addressType == 0x03) {
|
||||
final length = (await reader.readExactly(1)).first;
|
||||
await reader.readExactly(length + 2);
|
||||
final length = (await reader.readExactly(1, timeout: timeout)).first;
|
||||
if (length == 0) {
|
||||
throw const FormatException('Invalid SOCKS5 domain length');
|
||||
}
|
||||
await reader.readExactly(length + 2, timeout: timeout);
|
||||
return;
|
||||
}
|
||||
throw const FormatException('Unsupported SOCKS5 address type');
|
||||
@@ -257,8 +274,15 @@ class _SocketReader {
|
||||
int count, {
|
||||
Duration timeout = const Duration(seconds: 4),
|
||||
}) async {
|
||||
final elapsed = Stopwatch()..start();
|
||||
while (_buffer.length < count) {
|
||||
final hasNext = await _iterator.moveNext().timeout(timeout);
|
||||
final remainingMicros = timeout.inMicroseconds - elapsed.elapsedMicroseconds;
|
||||
if (remainingMicros <= 0) {
|
||||
throw TimeoutException('Socket read timed out', timeout);
|
||||
}
|
||||
final hasNext = await _iterator.moveNext().timeout(
|
||||
Duration(microseconds: remainingMicros),
|
||||
);
|
||||
if (!hasNext) {
|
||||
throw const SocketException('Socket closed early');
|
||||
}
|
||||
|
||||
Reference in new issue
Block a user