CommandBehavior.cs

using WebSocketSharp;
using WebSocketSharp.Server;
namespace PCRemoteControlServer
{
    public class CommandBehavior : WebSocketBehavior
    {
        public static string? Secret { get; set; }

        private bool _authenticated;
        private string _clientEndpoint = "unknown";

        protected override void OnMessage(MessageEventArgs e)
        {
            var data = e.Data ?? string.Empty;

            if (!_authenticated)
            {
                if (data == Secret)
                {
                    _authenticated = true;
                    KeyCommandExecutor.Log("Client authenticated: " + _clientEndpoint);
                    Send("AUTH_OK");
                }
                else
                {
                    KeyCommandExecutor.Log("Client failed authentication: " + _clientEndpoint);
                    Send("AUTH_FAIL");
                    Context.WebSocket.Close(CloseStatusCode.PolicyViolation, "Invalid secret");
                }
                return;
            }

            KeyCommandExecutor.ExecuteCommand(data);
            Send("OK");
        }

        protected override void OnOpen()
        {
            _clientEndpoint = SafeGetClientEndpoint();
            _authenticated = string.IsNullOrEmpty(Secret);
            var authState = _authenticated ? "(no auth)" : "(awaiting auth)";
            KeyCommandExecutor.Log("Client connected: " + _clientEndpoint + " " + authState);
        }

        protected override void OnClose(CloseEventArgs e)
        {
            KeyCommandExecutor.Log("Client disconnected: " + _clientEndpoint + " Code=" + e.Code);
        }

        protected override void OnError(ErrorEventArgs e)
        {
            if (_clientEndpoint == "unknown")
            {
                _clientEndpoint = SafeGetClientEndpoint();
            }

            KeyCommandExecutor.Log("WebSocket error (" + _clientEndpoint + "): " + e.Message);
        }

        private string SafeGetClientEndpoint()
        {
            try
            {
                return Context?.UserEndPoint?.ToString() ?? "unknown";
            }
            catch (NullReferenceException)
            {
                return "unknown";
            }
            catch (ObjectDisposedException)
            {
                return "unknown";
            }
        }
    }
}