rustdesk/lib/models/server_model.dart

540 lines
15 KiB
Dart
Raw Normal View History

import 'dart:async';
2022-03-19 23:28:29 +08:00
import 'dart:convert';
2022-03-25 16:34:27 +08:00
import 'package:dash_chat/dash_chat.dart';
2022-03-19 23:28:29 +08:00
import 'package:flutter/material.dart';
import '../common.dart';
import '../pages/server_page.dart';
import 'model.dart';
2022-04-19 13:07:45 +08:00
const loginDialogTag = "LOGIN";
2022-04-17 11:34:46 +08:00
final _emptyIdShow = translate("Generating ...");
2022-03-19 23:28:29 +08:00
class ServerModel with ChangeNotifier {
Timer? _interval;
2022-04-17 11:34:46 +08:00
bool _isStart = false; // Android MainService status
2022-03-19 23:28:29 +08:00
bool _mediaOk = false;
bool _inputOk = false;
bool _audioOk = false;
bool _fileOk = false;
2022-04-17 11:34:46 +08:00
int _connectStatus = 0; // Rendezvous Server status
final _serverId = TextEditingController(text: _emptyIdShow);
final _serverPasswd = TextEditingController(text: "");
2022-04-13 15:33:23 +08:00
Map<int, Client> _clients = {};
2022-03-19 23:28:29 +08:00
bool get isStart => _isStart;
2022-03-19 23:28:29 +08:00
bool get mediaOk => _mediaOk;
bool get inputOk => _inputOk;
2022-03-22 21:47:42 +08:00
bool get audioOk => _audioOk;
bool get fileOk => _fileOk;
2022-04-17 11:34:46 +08:00
int get connectStatus => _connectStatus;
TextEditingController get serverId => _serverId;
TextEditingController get serverPasswd => _serverPasswd;
2022-04-13 15:33:23 +08:00
Map<int, Client> get clients => _clients;
2022-03-19 23:28:29 +08:00
ServerModel() {
2022-04-13 15:33:23 +08:00
() async {
/**
* 1. check android permission
* 2. check config
2022-04-09 21:38:46 +08:00
* audio true by default (if permission on) (false default < Android 10)
* file true by default (if permission on)
* input false by default (it need turning on manually everytime)
*/
await Future.delayed(Duration(seconds: 1));
// audio
2022-04-13 15:33:23 +08:00
if (androidVersion < 30 || !await PermissionManager.check("audio")) {
_audioOk = false;
2022-04-13 15:33:23 +08:00
FFI.setByName(
'option',
jsonEncode(Map()
..["name"] = "enable-audio"
..["value"] = "N"));
2022-04-13 15:33:23 +08:00
} else {
final audioOption = FFI.getByName('option', 'enable-audio');
_audioOk = audioOption.isEmpty;
}
2022-03-22 21:47:42 +08:00
// file
2022-04-13 15:33:23 +08:00
if (!await PermissionManager.check("file")) {
_fileOk = false;
2022-04-13 15:33:23 +08:00
FFI.setByName(
'option',
jsonEncode(Map()
..["name"] = "enable-file-transfer"
..["value"] = "N"));
2022-04-13 15:33:23 +08:00
} else {
final fileOption = FFI.getByName('option', 'enable-file-transfer');
_fileOk = fileOption.isEmpty;
}
// input (mouse control)
2022-03-22 21:47:42 +08:00
Map<String, String> res = Map()
..["name"] = "enable-keyboard"
..["value"] = 'N';
FFI.setByName('option', jsonEncode(res)); // input false by default
notifyListeners();
}();
2022-04-17 11:34:46 +08:00
Timer.periodic(Duration(seconds: 1), (timer) {
var status = int.tryParse(FFI.getByName('connect_statue')) ?? 0;
if (status > 0) {
status = 1;
}
if (status != _connectStatus) {
_connectStatus = status;
notifyListeners();
}
});
2022-03-22 21:47:42 +08:00
}
toggleAudio() async {
2022-04-13 15:33:23 +08:00
if (!_audioOk && !await PermissionManager.check("audio")) {
final res = await PermissionManager.request("audio");
2022-04-13 15:33:23 +08:00
if (!res) {
// TODO handle fail
return;
}
}
2022-03-22 21:47:42 +08:00
_audioOk = !_audioOk;
Map<String, String> res = Map()
..["name"] = "enable-audio"
..["value"] = _audioOk ? '' : 'N';
FFI.setByName('option', jsonEncode(res));
notifyListeners();
}
toggleFile() async {
2022-04-13 15:33:23 +08:00
if (!_fileOk && !await PermissionManager.check("file")) {
final res = await PermissionManager.request("file");
2022-04-13 15:33:23 +08:00
if (!res) {
// TODO handle fail
return;
}
}
_fileOk = !_fileOk;
Map<String, String> res = Map()
..["name"] = "enable-file-transfer"
2022-03-22 21:47:42 +08:00
..["value"] = _fileOk ? '' : 'N';
FFI.setByName('option', jsonEncode(res));
notifyListeners();
}
2022-04-13 15:33:23 +08:00
toggleInput() {
if (_inputOk) {
2022-03-24 17:58:33 +08:00
FFI.invokeMethod("stop_input");
2022-04-13 15:33:23 +08:00
} else {
2022-03-22 21:47:42 +08:00
showInputWarnAlert();
}
}
toggleService() async {
2022-04-13 15:33:23 +08:00
if (_isStart) {
final res =
await DialogManager.show<bool>((setState, close) => CustomAlertDialog(
title: Row(children: [
Icon(Icons.warning_amber_sharp,
color: Colors.redAccent, size: 28),
SizedBox(width: 10),
Text(translate("Warning")),
]),
content: Text(translate("android_stop_service_tip")),
actions: [
TextButton(
onPressed: () => close(),
child: Text(translate("Cancel"))),
ElevatedButton(
onPressed: () => close(true),
child: Text(translate("OK"))),
],
));
if (res == true) {
2022-03-22 21:47:42 +08:00
stopService();
}
2022-04-13 15:33:23 +08:00
} else {
final res =
await DialogManager.show<bool>((setState, close) => CustomAlertDialog(
title: Row(children: [
Icon(Icons.warning_amber_sharp,
color: Colors.redAccent, size: 28),
SizedBox(width: 10),
Text(translate("Warning")),
]),
content: Text(translate("android_service_will_start_tip")),
actions: [
TextButton(
onPressed: () => close(),
child: Text(translate("Cancel"))),
ElevatedButton(
onPressed: () => close(true),
child: Text(translate("OK"))),
],
));
if (res == true) {
2022-03-22 21:47:42 +08:00
startService();
}
}
}
Future<Null> startService() async {
_isStart = true;
notifyListeners();
FFI.setByName("ensure_init_event_queue");
_interval = Timer.periodic(Duration(milliseconds: 30), (timer) {
FFI.ffiModel.update("");
});
await FFI.invokeMethod("init_service");
FFI.setByName("start_service");
getIDPasswd();
updateClientState();
}
2022-04-13 15:33:23 +08:00
Future<Null> stopService() async {
_isStart = false;
2022-03-22 21:47:42 +08:00
_interval?.cancel();
_interval = null;
FFI.serverModel.closeAll();
await FFI.invokeMethod("stop_service");
FFI.setByName("stop_service");
notifyListeners();
}
Future<Null> initInput() async {
await FFI.invokeMethod("init_input");
}
2022-03-30 23:09:19 +08:00
Future<bool> updatePassword(String pw) async {
final oldPasswd = _serverPasswd.text;
2022-04-13 15:33:23 +08:00
FFI.setByName("update_password", pw);
2022-03-30 23:09:19 +08:00
await Future.delayed(Duration(milliseconds: 500));
await getIDPasswd(force: true);
// check result
2022-04-13 15:33:23 +08:00
if (pw == "") {
if (_serverPasswd.text.isNotEmpty && _serverPasswd.text != oldPasswd) {
2022-03-30 23:09:19 +08:00
return true;
2022-04-13 15:33:23 +08:00
} else {
2022-03-30 23:09:19 +08:00
return false;
}
2022-04-13 15:33:23 +08:00
} else {
if (_serverPasswd.text == pw) {
2022-03-30 23:09:19 +08:00
return true;
2022-04-13 15:33:23 +08:00
} else {
2022-03-30 23:09:19 +08:00
return false;
}
}
}
getIDPasswd({bool force = false}) async {
if (!force && _serverId.text != _emptyIdShow && _serverPasswd.text != "") {
return;
}
var count = 0;
const maxCount = 10;
while (count < maxCount) {
await Future.delayed(Duration(seconds: 1));
final id = FFI.getByName("server_id");
final passwd = FFI.getByName("server_password");
if (id.isEmpty) {
continue;
} else {
_serverId.text = id;
}
if (passwd.isEmpty) {
continue;
} else {
_serverPasswd.text = passwd;
}
debugPrint(
"fetch id & passwd again at $count:id:${_serverId.text},passwd:${_serverPasswd.text}");
count++;
if (_serverId.text != _emptyIdShow && _serverPasswd.text.isNotEmpty) {
break;
}
}
notifyListeners();
}
2022-03-19 23:28:29 +08:00
changeStatue(String name, bool value) {
debugPrint("changeStatue value $value");
2022-03-19 23:28:29 +08:00
switch (name) {
case "media":
_mediaOk = value;
2022-04-13 15:33:23 +08:00
if (value && !_isStart) {
startService();
}
2022-03-19 23:28:29 +08:00
break;
case "input":
2022-04-13 15:33:23 +08:00
if (_inputOk != value) {
2022-03-22 21:47:42 +08:00
Map<String, String> res = Map()
..["name"] = "enable-keyboard"
..["value"] = value ? '' : 'N';
FFI.setByName('option', jsonEncode(res));
}
2022-03-19 23:28:29 +08:00
_inputOk = value;
break;
default:
return;
}
notifyListeners();
}
updateClientState() {
var res = FFI.getByName("clients_state");
try {
final List clientsJson = jsonDecode(res);
2022-04-13 15:33:23 +08:00
for (var clientJson in clientsJson) {
final client = Client.fromJson(clientJson);
2022-03-25 16:34:27 +08:00
_clients[client.id] = client;
}
2022-03-19 23:28:29 +08:00
notifyListeners();
} catch (e) {
debugPrint("Failed to updateClientState:$e");
}
2022-03-19 23:28:29 +08:00
}
2022-04-19 13:07:45 +08:00
void loginRequest(Map<String, dynamic> evt) {
try {
2022-03-19 23:28:29 +08:00
final client = Client.fromJson(jsonDecode(evt["client"]));
2022-04-19 13:07:45 +08:00
if (_clients.containsKey(client.id)) {
return;
}
_clients[client.id] = client;
notifyListeners();
showLoginDialog(client);
} catch (e) {
debugPrint("Failed to call loginRequest,error:$e");
}
}
void showLoginDialog(Client client) {
DialogManager.show(
(setState, close) => CustomAlertDialog(
title: Row(
mainAxisAlignment: MainAxisAlignment.spaceBetween,
2022-03-30 17:31:32 +08:00
children: [
2022-04-19 13:07:45 +08:00
Text(translate(client.isFileTransfer
? "File Connection"
: "Screen Connection")),
IconButton(
onPressed: () {
close();
},
icon: Icon(Icons.close))
]),
content: Column(
mainAxisSize: MainAxisSize.min,
mainAxisAlignment: MainAxisAlignment.center,
crossAxisAlignment: CrossAxisAlignment.start,
children: [
Text(translate("Do you accept?")),
clientInfo(client),
Text(
translate("android_new_connection_tip"),
style: TextStyle(color: Colors.black54),
),
],
),
2022-04-19 13:07:45 +08:00
actions: [
TextButton(
child: Text(translate("Dismiss")),
onPressed: () {
sendLoginResponse(client, false);
close();
}),
ElevatedButton(
child: Text(translate("Accept")),
onPressed: () {
sendLoginResponse(client, true);
close();
}),
],
),
tag: getLoginDialogTag(client.id));
}
void sendLoginResponse(Client client, bool res) {
final Map<String, dynamic> response = Map();
response["id"] = client.id;
response["res"] = res;
if (res) {
FFI.setByName("login_res", jsonEncode(response));
if (!client.isFileTransfer) {
FFI.invokeMethod("start_capture");
}
FFI.invokeMethod("cancel_notification", client.id);
_clients[client.id]?.authorized = true;
notifyListeners();
} else {
FFI.setByName("login_res", jsonEncode(response));
FFI.invokeMethod("cancel_notification", client.id);
_clients.remove(client.id);
2022-03-19 23:28:29 +08:00
}
}
2022-03-22 21:47:42 +08:00
void onClientAuthorized(Map<String, dynamic> evt) {
2022-04-13 15:33:23 +08:00
try {
2022-03-25 16:34:27 +08:00
final client = Client.fromJson(jsonDecode(evt['client']));
2022-04-19 13:07:45 +08:00
DialogManager.dismissByTag(getLoginDialogTag(client.id));
2022-03-25 16:34:27 +08:00
_clients[client.id] = client;
2022-03-22 21:47:42 +08:00
notifyListeners();
2022-04-13 15:33:23 +08:00
} catch (e) {}
2022-03-22 21:47:42 +08:00
}
2022-03-19 23:28:29 +08:00
void onClientRemove(Map<String, dynamic> evt) {
try {
2022-03-19 23:28:29 +08:00
final id = int.parse(evt['id'] as String);
2022-04-13 15:33:23 +08:00
if (_clients.containsKey(id)) {
2022-03-29 23:10:43 +08:00
_clients.remove(id);
2022-04-19 13:07:45 +08:00
DialogManager.dismissByTag(getLoginDialogTag(id));
2022-04-13 15:33:23 +08:00
FFI.invokeMethod("cancel_notification", id);
2022-03-29 23:10:43 +08:00
}
2022-03-19 23:28:29 +08:00
notifyListeners();
} catch (e) {
2022-03-19 23:28:29 +08:00
debugPrint("onClientRemove failed,error:$e");
}
}
closeAll() {
2022-04-13 15:33:23 +08:00
_clients.forEach((id, client) {
2022-03-25 16:34:27 +08:00
FFI.setByName("close_conn", id.toString());
2022-03-19 23:28:29 +08:00
});
2022-03-25 16:34:27 +08:00
_clients.clear();
2022-03-19 23:28:29 +08:00
}
}
class Client {
2022-03-22 21:47:42 +08:00
int id = 0; // client connections inner count id
2022-03-19 23:28:29 +08:00
bool authorized = false;
bool isFileTransfer = false;
String name = "";
2022-03-22 21:47:42 +08:00
String peerId = ""; // peer user's id,show at app
bool keyboard = false;
bool clipboard = false;
bool audio = false;
2022-03-25 16:34:27 +08:00
late ChatUser chatUser;
2022-03-19 23:28:29 +08:00
2022-04-13 15:33:23 +08:00
Client(this.authorized, this.isFileTransfer, this.name, this.peerId,
this.keyboard, this.clipboard, this.audio);
2022-03-19 23:28:29 +08:00
Client.fromJson(Map<String, dynamic> json) {
id = json['id'];
authorized = json['authorized'];
isFileTransfer = json['is_file_transfer'];
name = json['name'];
peerId = json['peer_id'];
2022-04-13 15:33:23 +08:00
keyboard = json['keyboard'];
clipboard = json['clipboard'];
audio = json['audio'];
2022-03-25 16:34:27 +08:00
chatUser = ChatUser(
2022-04-13 15:33:23 +08:00
uid: peerId,
name: name,
2022-03-25 16:34:27 +08:00
);
2022-03-19 23:28:29 +08:00
}
Map<String, dynamic> toJson() {
final Map<String, dynamic> data = new Map<String, dynamic>();
data['id'] = this.id;
data['is_start'] = this.authorized;
data['is_file_transfer'] = this.isFileTransfer;
data['name'] = this.name;
data['peer_id'] = this.peerId;
2022-03-22 21:47:42 +08:00
data['keyboard'] = this.keyboard;
data['clipboard'] = this.clipboard;
data['audio'] = this.audio;
2022-03-19 23:28:29 +08:00
return data;
}
}
2022-03-22 21:47:42 +08:00
2022-04-19 13:07:45 +08:00
String getLoginDialogTag(int id) {
return loginDialogTag + id.toString();
}
showInputWarnAlert() {
DialogManager.show((setState, close) => CustomAlertDialog(
title: Text(translate("How to get Android input permission?")),
content: Column(
mainAxisSize: MainAxisSize.min,
children: [
Text(translate(translate("android_input_permission_tip1"))),
SizedBox(height: 10),
Text(translate(translate("android_input_permission_tip2"))),
2022-03-22 21:47:42 +08:00
],
2022-04-19 13:07:45 +08:00
),
actions: [
TextButton(child: Text(translate("Cancel")), onPressed: close),
ElevatedButton(
child: Text(translate("Open System Setting")),
onPressed: () {
FFI.serverModel.initInput();
close();
}),
],
));
2022-03-22 21:47:42 +08:00
}
class PermissionManager {
static Completer<bool>? _completer;
static Timer? _timer;
static var _current = "";
2022-04-13 15:33:23 +08:00
static final permissions = ["audio", "file"];
2022-04-13 15:33:23 +08:00
static bool isWaitingFile() {
if (_completer != null) {
return !_completer!.isCompleted && _current == "file";
}
return false;
}
2022-04-13 15:33:23 +08:00
static Future<bool> check(String type) {
if (!permissions.contains(type))
return Future.error("Wrong permission!$type");
2022-04-13 15:33:23 +08:00
return FFI.invokeMethod("check_permission", type);
}
2022-04-13 15:33:23 +08:00
static Future<bool> request(String type) {
if (!permissions.contains(type))
return Future.error("Wrong permission!$type");
_current = type;
_completer = Completer<bool>();
2022-04-13 15:33:23 +08:00
FFI.invokeMethod("request_permission", type);
// timeout
_timer?.cancel();
2022-04-13 15:33:23 +08:00
_timer = Timer(Duration(seconds: 60), () {
if (_completer == null) return;
if (!_completer!.isCompleted) {
_completer!.complete(false);
}
_completer = null;
_current = "";
});
return _completer!.future;
}
2022-04-13 15:33:23 +08:00
static complete(String type, bool res) {
if (type != _current) {
res = false;
}
_timer?.cancel();
_completer?.complete(res);
_current = "";
}
}