sockscape/server/Socks/PlayerConnection.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Net.Sockets;
using System.Text;
using System.Threading.Tasks;
using Kneesocks;
using Glove;
using SockScape.Encryption;

namespace SockScape {
    class PlayerConnection : Connection {
        private DateTime ConnectionOpened;
        private Key Key;
        public Cipher Encryptor { get; private set; } = null;

        protected override void OnOpen() {
            ConnectionOpened = DateTime.UtcNow;
            Key = new Key();

            Send(Key.GenerateRequestPacket().GetBytes());
        }

        protected override void OnParse() {
            if((DateTime.UtcNow - ConnectionOpened).Seconds > 60) {
                Disconnect(Frame.kClosingReason.ProtocolError, "Logon request timed out.");
            }
        }

        protected override void OnReceive(byte[] data) {
            Packet packet = 
                Encryptor == null ? Packet.FromBytes(data)
                                  : Packet.FromBytes(Encryptor.Parse(data));

            if(packet == null) {
                Disconnect(Frame.kClosingReason.ProtocolError, "Packet received was not legal.");
                return;
            }

            switch(packet.Id) {
                case Packet.kId.KeyExchange:
                    Key.ParseResponsePacket(packet);
                    if(!Key.Succeeded) {
                        Disconnect(Frame.kClosingReason.ProtocolError, "Could not exchange keys.");
                        return;
                    }

                    Encryptor = new Cipher(Key.PrivateKey);
                    break;
                case Packet.kId.LoginAttempt:

                    break;
                default:
                    Disconnect(Frame.kClosingReason.ProtocolError, "Packet ID could not be understood at this time.");
                    break;
            }

            Console.WriteLine(Id + " says " + data.GetString());
        }
    }
}