Feat/client (#16)

* feat: client drafy

* fix: client

* feat: username

* feat: client drafy

* fix: client

* feat: username

* fix: messages are sent encrypted using RSA

* chore: formating

* chore: delete Database class

* feat: add username to Client constructor

* feat: make username public

* fix: copilot nitpicking

---------

Co-authored-by: IhtDzenda <[email protected]>

Author: JerrySB03

sutor-aes/Client.cs

@@ -7,17 +7,18 @@ namespace SutorAes
     class Client
     {
         Guid Id { get; set; }
-        string Name { get; set; }
+        public string Name { get; set; }
 
-        byte[] RsaPub { get; set; }
+        public byte[] RsaPub { get; set; }
         byte[] RsaPriv { get; set; }
         Dictionary<Guid, byte[]> KnownHosts { get; set; } = new();
         private event Action<string>? Handler;
 
         RSAEncryptionPadding padding = RSAEncryptionPadding.Pkcs1;
 
-        public Client()
+        public Client(string _name)
         {
+            this.Name = _name;
             GenerateRSAKeys();
         }
 
@@ -105,19 +106,21 @@ public void EstablishConnection(Client other)
             byte[] key = new byte[16];
 
             System.Security.Cryptography.RandomNumberGenerator.Fill(key);
-            other.RecieveConnection(this.Id, key);
+            byte[] encryptedKey = this.EncryptRSA(key, other.RsaPub);
+            other.RecieveConnection(this.Id, encryptedKey);
             this.KnownHosts.Add(other.Id, key);
         }
-        public void RecieveConnection(Guid id, byte[] key)
+        public void RecieveConnection(Guid id, byte[] encryptedKey)
         {
+            byte[] key = this.DecryptRSA(encryptedKey);
             this.KnownHosts.Add(id, key);
         }
 
-        public byte[] EncryptRSA(byte[] data)
+        public byte[] EncryptRSA(byte[] data, byte[] pubKey)
         {
             using (RSA rsa = RSA.Create())
             {
-                rsa.ImportRSAPublicKey(RsaPub, out _);
+                rsa.ImportRSAPublicKey(pubKey, out _);
                 return rsa.Encrypt(data, padding);
             }
         }
@@ -131,24 +134,4 @@ public byte[] DecryptRSA(byte[] encryptedData)
             }
         }
     }
-
-    class Database
-    {
-        private Dictionary<Guid, byte[]> User2Pubkey { get; set; }
-
-        public Database()
-        {
-            User2Pubkey = new Dictionary<Guid, byte[]>();
-        }
-
-        public void RegisterUser(Guid id, byte[] rsaPub)
-        {
-            User2Pubkey.Add(id, rsaPub);
-        }
-
-        public Dictionary<Guid, byte[]> ListUsers()
-        {
-            return User2Pubkey;
-        }
-    }
 }