mirror of
https://github.com/lollipopkit/flutter_server_box.git
synced 2025-12-17 07:14:28 +01:00
118 lines
3.4 KiB
Dart
118 lines
3.4 KiB
Dart
import 'dart:async';
|
|
|
|
import 'package:dartssh2/dartssh2.dart';
|
|
import 'package:fl_lib/fl_lib.dart';
|
|
import 'package:flutter/foundation.dart';
|
|
import 'package:server_box/data/model/app/error.dart';
|
|
import 'package:server_box/data/model/server/server_private_info.dart';
|
|
import 'package:server_box/data/res/store.dart';
|
|
|
|
/// Must put this func out of any Class.
|
|
///
|
|
/// Because of this function is called by [compute].
|
|
///
|
|
/// https://stackoverflow.com/questions/51998995/invalid-arguments-illegal-argument-in-isolate-message-object-is-a-closure
|
|
List<SSHKeyPair> loadIndentity(String key) {
|
|
return SSHKeyPair.fromPem(key);
|
|
}
|
|
|
|
/// [args] : [key, pwd]
|
|
String decyptPem(List<String> args) {
|
|
/// skip when the key is not encrypted, or will throw exception
|
|
if (!SSHKeyPair.isEncryptedPem(args[0])) return args[0];
|
|
final sshKey = SSHKeyPair.fromPem(args[0], args[1]);
|
|
return sshKey.first.toPem();
|
|
}
|
|
|
|
enum GenSSHClientStatus { socket, key, pwd }
|
|
|
|
String getPrivateKey(String id) {
|
|
final pki = Stores.key.fetchOne(id);
|
|
if (pki == null) {
|
|
throw SSHErr(type: SSHErrType.noPrivateKey, message: 'key [$id] not found');
|
|
}
|
|
return pki.key;
|
|
}
|
|
|
|
Future<SSHClient> genClient(
|
|
Spi spi, {
|
|
void Function(GenSSHClientStatus)? onStatus,
|
|
|
|
/// Only pass this param if using multi-threading and key login
|
|
String? privateKey,
|
|
|
|
/// Only pass this param if using multi-threading and key login
|
|
String? jumpPrivateKey,
|
|
Duration timeout = const Duration(seconds: 5),
|
|
|
|
/// [Spi] of the jump server
|
|
///
|
|
/// Must pass this param if using multi-threading and key login
|
|
Spi? jumpSpi,
|
|
|
|
/// Handle keyboard-interactive authentication
|
|
SSHUserInfoRequestHandler? onKeyboardInteractive,
|
|
}) async {
|
|
onStatus?.call(GenSSHClientStatus.socket);
|
|
|
|
String? alterUser;
|
|
|
|
final socket = await () async {
|
|
// Proxy
|
|
final jumpSpi_ = () {
|
|
// Multi-thread or key login
|
|
if (jumpSpi != null) return jumpSpi;
|
|
// Main thread
|
|
if (spi.jumpId != null) return Stores.server.box.get(spi.jumpId);
|
|
}();
|
|
if (jumpSpi_ != null) {
|
|
final jumpClient = await genClient(jumpSpi_, privateKey: jumpPrivateKey, timeout: timeout);
|
|
|
|
return await jumpClient.forwardLocal(spi.ip, spi.port);
|
|
}
|
|
|
|
// Direct
|
|
try {
|
|
return await SSHSocket.connect(spi.ip, spi.port, timeout: timeout);
|
|
} catch (e) {
|
|
Loggers.app.warning('genClient', e);
|
|
if (spi.alterUrl == null) rethrow;
|
|
try {
|
|
final res = spi.parseAlterUrl();
|
|
alterUser = res.$2;
|
|
return await SSHSocket.connect(res.$1, res.$3, timeout: timeout);
|
|
} catch (e) {
|
|
Loggers.app.warning('genClient alterUrl', e);
|
|
rethrow;
|
|
}
|
|
}
|
|
}();
|
|
|
|
final keyId = spi.keyId;
|
|
if (keyId == null) {
|
|
onStatus?.call(GenSSHClientStatus.pwd);
|
|
return SSHClient(
|
|
socket,
|
|
username: alterUser ?? spi.user,
|
|
onPasswordRequest: () => spi.pwd,
|
|
onUserInfoRequest: onKeyboardInteractive,
|
|
onVerifyHostKey: (type, fingerprint) => true,
|
|
// printDebug: debugPrint,
|
|
// printTrace: debugPrint,
|
|
);
|
|
}
|
|
privateKey ??= getPrivateKey(keyId);
|
|
|
|
onStatus?.call(GenSSHClientStatus.key);
|
|
return SSHClient(
|
|
socket,
|
|
username: spi.user,
|
|
// Must use [compute] here, instead of [Computer.shared.start]
|
|
identities: await compute(loadIndentity, privateKey),
|
|
onUserInfoRequest: onKeyboardInteractive,
|
|
onVerifyHostKey: (type, fingerprint) => true,
|
|
// printDebug: debugPrint,
|
|
// printTrace: debugPrint,
|
|
);
|
|
}
|