First attempt Messages.cs

Messages.cs

@@ -1,3 +1,43 @@
+using System.Collections.Generic;
+using System.IO;
+using System.Net;
+using System.Net.Sockets;
+using System.Threading.Tasks;
+
+public class Client {
+	//public ConnectionMethod connection;
+	public UdpClient udpClient;
+	public string ipAddress;
+	public int port;
+
+	public byte networkId;
+
+	public static Client GetClient(string ipAddress, int port) {
+		foreach (Client c in clients) {
+			if (c.ipAddress == ipAddress && c.port == port)
+				return c;
+		}
+		return null;
+	}
+	static public List<Client> clients = new List<Client>();
+
+	public static Client NewClient() {
+		Client client = new();
+		clients.Add(client);
+		client.networkId = 0;
+
+		return client;
+	}
+
+	public static Client NewUDPClient(UdpClient udpClient, string ipAddress, int port) {
+		Client client = NewClient();
+		client.ipAddress = null;
+		client.port = port;
+		client.udpClient = udpClient;
+		return client;
+	}
+}
+
 public class IMessage {
 	public IMessage() { }
 	public IMessage(byte[] data) {
@@ -6,8 +46,74 @@ public class IMessage {
 
 	public virtual byte[] Serialize() { return null; }
 	public virtual void Deserialize(byte[] data) { }
+	
+	public static void SendMsg(Client client, byte[] data) {
+		if (client.ipAddress != null)
+			client.udpClient.Send(data, data.Length, client.ipAddress, client.port);
+		else
+			;//Debug.Log("Unknown client");
+	}
 }
 
 #region Client
 
-#endregion
+class ClientMsg : IMessage {
+	public const byte length = 2;
+	public byte clientId;
+
+	public ClientMsg(byte[] data) : base(data) { }
+	public override void Deserialize(byte[] data) {
+		base.Deserialize(data);
+		uint ix = 0;
+		clientId = data[ix++];
+	}
+	
+	protected async Task<bool> Receive(Stream dataStream, Client client, byte packetSize) {
+		if (packetSize != ClientMsg.length) {
+			//Debug.LogWarning($"Client msg size invalid: {packetSize}");
+			return false;
+		}
+
+		byte[] buffer = new byte[packetSize - 1]; //  without msgId
+		int byteCount = dataStream.Read(buffer, 0, packetSize - 1);
+		while (byteCount < packetSize - 1) {
+			// not all bytes have been read, wait and try again
+			await Task.Delay(1);
+			byteCount += dataStream.Read(buffer, byteCount, packetSize - 1 - byteCount);
+		}
+
+		ClientMsg msg = new(buffer);
+
+		if (client.networkId == 0) {
+			client.networkId = (byte)(Client.clients.Count);
+			NetworkIdMsg.Send(client);
+			//if (string.IsNullOrEmpty(sceneUrl) == false)
+				//SendModelUrl(client, sceneUrl);
+		}
+		else if (msg.clientId == 0) {
+			NetworkIdMsg.Send(client);
+			//if (string.IsNullOrEmpty(sceneUrl) == false)
+				//SendModelUrl(client, sceneUrl);
+		}
+
+		return true;
+	}
+}
+
+#endregion Client
+
+#region Network Id
+
+class NetworkIdMsg : IMessage {
+	public const byte Id = 0xA1;
+	public const byte length = 2;
+	
+	public static void Send(Client client) {
+		byte[] data = new byte[NetworkIdMsg.length];
+		data[0] = NetworkIdMsg.Id;
+		data[1] = client.networkId;
+		SendMsg(client, data);
+	}
+}
+
+#endregion Network Id