Initial commit

nksrv/LobbyServer/LobbyHandler.cs

@@ -0,0 +1,130 @@
+using ASodium;
+using EmbedIO;
+using Google.Protobuf;
+using nksrv.Utils;
+using Swan.Logging;
+
+namespace nksrv.LobbyServer
+{
+    public static class LobbyHandler
+    {
+        public static readonly Dictionary<string, LobbyMsgHandler> Handlers = new Dictionary<string, LobbyMsgHandler>();
+        static LobbyHandler()
+        {
+            foreach (System.Type type in typeof(LobbyMsgHandler).Assembly.GetTypes())
+            {
+                if (type.GetCustomAttributes(typeof(PacketPathAttribute), true).Length > 0)
+                {
+                    var attrib = (PacketPathAttribute?)Attribute.GetCustomAttribute(type, typeof(PacketPathAttribute));
+                    if (attrib == null)
+                    {
+                        Logger.Error("WARNING: Failed to get attribute for " + type.FullName);
+                        continue;
+                    }
+
+
+                    var instance = Activator.CreateInstance(type);
+                    if (instance is LobbyMsgHandler handler)
+                    {
+                        Handlers.Add(attrib.Url, handler);
+                    }
+                    else
+                    {
+                        Logger.Error($"WARNING: Type {type.FullName} has PacketPathAttribute but does not implement LobbyMsgHandler");
+                    }
+                }
+            }
+        }
+        public static async Task DispatchSingle(IHttpContext ctx)
+        {
+            //var x = new RedirectionHandler();
+            //await x.HandleAsync(ctx);
+            //return;
+            LobbyMsgHandler? handler = null;
+            foreach (var item in Handlers)
+            {
+                if (ctx.RequestedPath == item.Key)
+                {
+                    handler = item.Value;
+                }
+            }
+
+            if (handler == null)
+            {
+                ctx.Response.StatusCode = 404;
+                //Logger.Error("HTTPS: No handler for /v1/" + ctx.RequestedPath);
+            }
+            else
+            {
+                // todo move everClass1.csything to its own handler
+                handler.Reset();
+                await handler.HandleAsync(ctx);
+                return;
+            }
+        }
+
+        /// <summary>
+        /// Private key, Token
+        /// </summary>
+        /// <param name="publicKey"></param>
+        /// <returns></returns>
+        public static GameClientInfo GenGameClientTok(ByteString publicKey, string authToken)
+        {
+            var token = Rng.RandomString(381);
+
+            var info = new GameClientInfo() { ClientPublicKey = publicKey.ToArray() };
+
+
+            var box = SodiumKeyExchange.CalculateServerSharedSecret(JsonDb.ServerPublicKey, JsonDb.ServerPrivateKey, publicKey.ToArray());
+
+            info.Keys = box;
+            info.ClientAuthToken = token;
+
+            // look up user id
+            foreach (var user in JsonDb.Instance.LauncherAccessTokens)
+            {
+                if (user.Token == authToken)
+                {
+                    info.UserId = user.UserID;
+                }
+            }
+            if (info.UserId == 0)
+                throw new Exception("expected user account");
+
+            JsonDb.Instance.GameClientTokens.Add(token, info);
+            JsonDb.Save();
+            return info;
+        }
+
+        public static GameClientInfo? GetInfo(string token)
+        {
+            return JsonDb.Instance.GameClientTokens[token];
+        }
+
+        public static void Init()
+        {
+            // By calling this function, we force .NET to initialize handler dictanary to catch errors early on.
+        }
+    }
+
+    public class GameClientInfo
+    {
+        /// <summary>
+        /// Client public key generated by game client
+        /// </summary>
+        public byte[] ClientPublicKey;
+        /// <summary>
+        /// Authentication token
+        /// </summary>
+        public string ClientAuthToken;
+        /// <summary>
+        /// Rx/Tx key pair
+        /// </summary>
+        public SodiumKeyExchangeSharedSecretBox Keys;
+        /// <summary>
+        /// User ID of the user
+        /// </summary>
+        public ulong UserId;
+
+    }
+}