using Rcon; using Rcon.Events; using SanAndreasUnity.Net; using SanAndreasUnity.Utilities; using System; using System.Collections.Concurrent; using System.ComponentModel; using System.Threading.Tasks; using UnityEngine; namespace SanAndreasUnity.RCON { public class RCONManager : MonoBehaviour { // Todo : set these from config private static String password; private static int portNumber; // Objects used to pass commands and responses between threads private static BlockingCollection mainToSec = new BlockingCollection(1); private static BlockingCollection secToMain = new BlockingCollection(1); private static BackgroundWorker workerInstance = null; public static void StartServer() { password = Config.GetString("RCON_password"); portNumber = Config.GetInt("RCON_port"); if (workerInstance != null) return; workerInstance = new BackgroundWorker(); workerInstance.DoWork += new DoWorkEventHandler( worker_doWork ); workerInstance.ProgressChanged += new ProgressChangedEventHandler( worker_progressChanged ); workerInstance.WorkerReportsProgress = true; // workerInstance.WorkerSupportsCancellation = true; workerInstance.RunWorkerAsync(); // Call the background worker } #region Code that runs in the RCON Server Thread private static void worker_doWork(object sender, DoWorkEventArgs e) { using (RconServer server = new RconServer(password, portNumber)) { server.OnClientCommandReceived += Server_OnClientCommandReceived; server.OnClientConnected += Server_OnClientConnected; server.OnClientAuthenticated += Server_OnClientAuthenticated; server.OnClientDisconnected += Server_OnClientDisconnected; server.Start(); } } static void Server_OnClientAuthenticated(object sender, ClientAuthenticatedEventArgs e) { // Console.WriteLine("{0} authenticated", e.Client.Client.LocalEndPoint); } static void Server_OnClientDisconnected(object sender, ClientDisconnectedEventArgs e) { // Console.WriteLine("{0} disconnected", e.EndPoint); } static void Server_OnClientConnected(object sender, ClientConnectedEventArgs e) { // Console.WriteLine("{0} connected", e.Client.Client.LocalEndPoint); } static string Server_OnClientCommandReceived(object sender, ClientSentCommandEventArgs e) { secToMain.Add(e.Command); // Pass the command to the main thread workerInstance.ReportProgress(0); //Report our progress to the main thread String commandResult = "Command didn't process correctly"; // default value commandResult = mainToSec.Take(); return commandResult; } #endregion // Runs in main thread private static void worker_progressChanged(object sender, ProgressChangedEventArgs e) { String command = "unknown"; command = secToMain.Take(); mainToSec.Add(CommandInterpreter.Interpret(command)); } void OnLoaderFinished() { if (NetStatus.IsServer) { if (Config.GetBool("RCON_enabled")) StartServer(); } } } }