Reboot-Launcher/lib/cli.dart

import 'dart:collection';
import 'dart:convert';
import 'dart:io';

import 'package:args/args.dart';
import 'package:process_run/shell.dart';
import 'package:reboot_launcher/src/model/fortnite_version.dart';
import 'package:reboot_launcher/src/model/game_type.dart';
import 'package:reboot_launcher/src/model/server_type.dart';
import 'package:reboot_launcher/src/util/os.dart';
import 'package:reboot_launcher/src/util/injector.dart';
import 'package:reboot_launcher/src/util/patcher.dart';
import 'package:reboot_launcher/src/util/reboot.dart';
import 'package:reboot_launcher/src/util/server.dart';
import 'package:shelf_proxy/shelf_proxy.dart';
import 'package:win32_suspend_process/win32_suspend_process.dart';
import 'dart:ffi';

import 'package:ffi/ffi.dart';
import 'package:win32/win32.dart';
import 'package:shelf/shelf_io.dart' as shelf_io;
import 'package:http/http.dart' as http;

// Needed because binaries can't be loaded in any other way
const String _craniumDownload = "https://cdn.discordapp.com/attachments/1026121175878881290/1031230848323825675/cranium.dll";
const String _consoleDownload = "https://cdn.discordapp.com/attachments/1026121175878881290/1031230848005046373/console.dll";

Process? _gameProcess;
Process? _eacProcess;
Process? _launcherProcess;

void main(List<String> args){
  handleCLI(args);
}

Future<Map<String, dynamic>> _getControllerJson(String name) async {
  var folder = await _getWindowsPath(FOLDERID_Documents);
  if(folder == null){
    throw Exception("Missing documents folder");
  }

  var file = File("$folder/$name.gs");
  if(!file.existsSync()){
    return HashMap();
  }

  return jsonDecode(file.readAsStringSync());
}

Future<String?> _getWindowsPath(String folderID) {
  final Pointer<Pointer<Utf16>> pathPtrPtr = calloc<Pointer<Utf16>>();
  final Pointer<GUID> knownFolderID = calloc<GUID>()..ref.setGUID(folderID);

  try {
    final int hr = SHGetKnownFolderPath(
      knownFolderID,
      KF_FLAG_DEFAULT,
      NULL,
      pathPtrPtr,
    );

    if (FAILED(hr)) {
      if (hr == E_INVALIDARG || hr == E_FAIL) {
        throw WindowsException(hr);
      }
      return Future<String?>.value();
    }

    final String path = pathPtrPtr.value.toDartString();
    return Future<String>.value(path);
  } finally {
    calloc.free(pathPtrPtr);
    calloc.free(knownFolderID);
  }
}

Future<void> handleCLI(List<String> args) async {
  stdout.writeln("Reboot Launcher");
  stdout.writeln("Wrote by Auties00");
  stdout.writeln("Version 3.13");

  _killOld();

  var gameJson = await _getControllerJson("game");
  var serverJson = await _getControllerJson("server");
  var settingsJson = await _getControllerJson("settings");
  var versions = _getVersions(gameJson);
  var parser = ArgParser()
    ..addCommand("list")
    ..addCommand("launch")
    ..addOption("version", defaultsTo: gameJson["version"])
    ..addOption("username")
    ..addOption("server-type", allowed: _getServerTypes(), defaultsTo: _getDefaultServerType(serverJson))
    ..addOption("server-host")
    ..addOption("server-port")
    ..addOption("dll", defaultsTo: settingsJson["reboot"] ?? (await loadBinary("reboot.dll", true)).path)
    ..addOption("type", allowed: _getTypes(), defaultsTo: _getDefaultType(gameJson))
    ..addFlag("update", defaultsTo: settingsJson["auto_update"] ?? true, negatable: true)
    ..addFlag("log", defaultsTo: false);
  var result = parser.parse(args);
  if (result.command?.name == "list") {
    stdout.writeln("Versions list: ");
    versions.map((entry) => "${entry.location.path}(${entry.name})")
        .forEach((element) => stdout.writeln(element));
    return;
  }

  var dll = result["dll"];
  var type = _getType(result);
  var username = result["username"];
  username ??= gameJson["${type == GameType.client ? "game" : "server"}_username"];
  var verbose = result["log"];

  var dummyVersion = _createVersion(gameJson["version"], result["version"], versions);
  await _updateDLLs();
  if(result["update"]) {
    stdout.writeln("Updating reboot dll...");
    await downloadRebootDll(0);
  }

  stdout.writeln("Launching game(type: ${type.name})...");
  await _startLauncherProcess(dummyVersion);
  if (result["type"] == "headless_server") {
    if(dummyVersion.executable == null){
      throw Exception("Missing game executable at: ${dummyVersion.location.path}");
    }

    await patch(dummyVersion.executable!);
  }

  var serverType = _getServerType(result);
  var host = result["server-host"] ?? serverJson["${serverType.id}_host"];
  var port = result["server-port"] ?? serverJson["${serverType.id}_port"];
  var started = await _startServerIfNeeded(host, port, serverType);
  if(!started){
    stderr.writeln("Cannot start server!");
    return;
  }

  await _startGameProcess(username, type, verbose, dll, dummyVersion);
}

void _killOld() async {
  var shell = Shell(
      commandVerbose: false,
      commentVerbose: false,
      verbose: false
  );
  try {
    await shell.run("taskkill /f /im FortniteLauncher.exe");
    await shell.run("taskkill /f /im FortniteClient-Win64-Shipping_EAC.exe");
  }catch(_){

  }
}

Iterable<String> _getTypes() => GameType.values.map((entry) => entry.id);

Iterable<String> _getServerTypes() => ServerType.values.map((entry) => entry.id);

String _getDefaultServerType(Map<String, dynamic> json) {
  var type = ServerType.values.elementAt(json["type"] ?? 0);
  return type.id;
}

GameType _getType(ArgResults result) {
  var type = GameType.of(result["type"]);
  if(type == null){
    throw Exception("Unknown game type: $result. Use --type only with ${_getTypes().join(", ")}");
  }

  return type;
}

ServerType _getServerType(ArgResults result) {
  var type = ServerType.of(result["server-type"]);
  if(type == null){
    throw Exception("Unknown server type: $result. Use --server-type only with ${_getServerTypes().join(", ")}");
  }

  return type;
}

String _getDefaultType(Map<String, dynamic> json){
  var type = GameType.values.elementAt(json["type"] ?? 0);
  switch(type){
    case GameType.client:
      return "client";
    case GameType.server:
      return "server";
    case GameType.headlessServer:
      return "headless_server";
  }
}

Future<void> _updateDLLs() async {
  stdout.writeln("Downloading necessary components...");
  var consoleDll = await loadBinary("console.dll", true);
  if(!consoleDll.existsSync()){
    var response = await http.get(Uri.parse(_consoleDownload));
    if(response.statusCode != 200){
      throw Exception("Cannot download console.dll");
    }

    await consoleDll.writeAsBytes(response.bodyBytes);
  }

  var craniumDll = await loadBinary("cranium.dll", true);
  if(!craniumDll.existsSync()){
    var response = await http.get(Uri.parse(_craniumDownload));
    if(response.statusCode != 200){
      throw Exception("Cannot download cranium.dll");
    }

    await craniumDll.writeAsBytes(response.bodyBytes);
  }
}

List<FortniteVersion> _getVersions(Map<String, dynamic> gameJson) {
  Iterable iterable = jsonDecode(gameJson["versions"] ?? "[]");
  return iterable.map((entry) => FortniteVersion.fromJson(entry))
      .toList();
}

Future<void> _startGameProcess(String? username, GameType type, bool verbose, String dll, FortniteVersion version) async {
  var gamePath = version.executable?.path;
  if (gamePath == null) {
    throw Exception("${version.location
        .path} no longer contains a Fortnite executable. Did you delete it?");
  }

  var hosting = type != GameType.client;
  if (username == null) {
    username = "Reboot${hosting ? 'Host' : 'Player'}";
    stdout.writeln("No username was specified, using $username by default. Use --username to specify one");
  }

  _gameProcess = await Process.start(gamePath, createRebootArgs(username, type == GameType.headlessServer))
    ..exitCode.then((_) => _onClose())
    ..outLines.forEach((line) => _onGameOutput(line, dll, hosting, verbose));
  await _injectOrShowError("cranium.dll");
}

void _onClose() {
  _kill();
  stdout.writeln("The game was closed");
  exit(0);
}

Future<void> _startLauncherProcess(FortniteVersion dummyVersion) async {
  if (dummyVersion.launcher == null) {
    return;
  }

  _launcherProcess = await Process.start(dummyVersion.launcher!.path, []);
  Win32Process(_launcherProcess!.pid).suspend();
}

Future<bool> _startServerIfNeeded(String? host, String? port, ServerType type) async {
  stdout.writeln("Starting lawin server...");
  switch(type){
    case ServerType.local:
      var result = await ping(host ?? "127.0.0.1", port ?? "3551");
      if(result == null){
        throw Exception("Local lawin server is not running");
      }

      stdout.writeln("Detected local lawin server");
      return true;
    case ServerType.embedded:
      stdout.writeln("Starting an embedded server...");
      return await _changeEmbeddedServerState();
    case ServerType.remote:
      if(host == null){
        throw Exception("Missing host for remote server");
      }

      if(port == null){
        throw Exception("Missing host for remote server");
      }

      stdout.writeln("Starting a reverse proxy to $host:$port");
      return await _changeReverseProxyState(host, port) != null;
  }
}

Future<bool> _changeEmbeddedServerState() async {

  return true;
}

Future<HttpServer?> _changeReverseProxyState(String host, String port) async {
  host = host.trim();
  if(host.isEmpty){
    throw Exception("Missing host name");
  }

  port = port.trim();
  if(port.isEmpty){
    throw Exception("Missing port");
  }

  if(int.tryParse(port) == null){
    throw Exception("Invalid port, use only numbers");
  }

  try{
    var uri = await ping(host, port);
    if(uri == null){
      return null;
    }

    return await shelf_io.serve(proxyHandler(uri), "127.0.0.1", 3551);
  }catch(error){
    throw Exception("Cannot start reverse proxy");
  }
}


FortniteVersion _createVersion(String? versionName, String? versionPath, List<FortniteVersion> versions) {
  if (versionPath != null) {
    return FortniteVersion(name: "dummy", location: Directory(versionPath));
  }

  if(versionName != null){
    try {
      return versions.firstWhere((element) => versionName == element.name);
    }catch(_){
      throw Exception("Cannot find version $versionName");
    }
  }

  throw Exception(
      "Specify a version using --version or open the launcher GUI and select it manually");
}


void _onGameOutput(String line, String rebootDll, bool host, bool verbose) {
  if(verbose) {
    stdout.writeln(line);
  }

  if (line.contains("FOnlineSubsystemGoogleCommon::Shutdown()")) {
    _onClose();
    return;
  }

  if(line.contains("port 3551 failed: Connection refused")){
    stderr.writeln("Connection refused from lawin server");
    _onClose();
    return;
  }

  if(line.contains("HTTP 400 response from ") || line.contains("Unable to login to Fortnite servers")){
    stderr.writeln("Connection refused from lawin server");
    _onClose();
    return;
  }

  if(line.contains("Network failure when attempting to check platform restrictions") || line.contains("UOnlineAccountCommon::ForceLogout")){
    stderr.writeln("Expired token, please reopen the game");
    _kill();
    _onClose();
    return;
  }

  if (line.contains("Game Engine Initialized") && !host) {
    _injectOrShowError("console.dll");
    return;
  }

  if(line.contains("Region") && host){
    _injectOrShowError(rebootDll, false);
  }
}

void _kill() {
  _gameProcess?.kill(ProcessSignal.sigabrt);
  _eacProcess?.kill(ProcessSignal.sigabrt);
  _launcherProcess?.kill(ProcessSignal.sigabrt);
}

Future<void> _injectOrShowError(String binary, [bool locate = true]) async {
  if (_gameProcess == null) {
    return;
  }

  try {
    stdout.writeln("Injecting $binary...");
    var dll = locate ? await loadBinary(binary, true) : File(binary);
    if(!dll.existsSync()){
      throw Exception("Cannot inject $dll: missing file");
    }

    await injectDll(_gameProcess!.pid, dll.path);
  } catch (exception) {
    throw Exception("Cannot inject binary: $binary");
  }
}