1 files changed, 99 insertions, 0 deletions
diff --git a/core.h b/core.h
new file mode 100644
index 0000000..e38d0f6
--- /dev/null
+++ b/core.h
@@ -0,0 +1,99 @@
+#ifndef CORE_H
+#define CORE_H 
+
+#include "buffer.h"
+
+#define CLIENT_LIMIT 100
+#define SERVER_LIMIT 20
+
+#define SESSION_KEEPALIVE 30
+
+#define SESSION_KEY_SIZE 16
+
+struct SocketRWCommon {
+	Buffer inputBuf, outputBuf;
+
+	enum ConnState {
+		CS_DISCONNECTED = 0,
+		CS_WAITING_DNS = 1, // server only
+		CS_WAITING_CONNECT = 2, // server only
+		CS_TLS_HANDSHAKE = 3,
+		CS_CONNECTED = 4
+	};
+	ConnState state;
+
+	int sock;
+	gnutls_session_t tls;
+	bool tlsActive;
+
+	SocketRWCommon();
+	virtual ~SocketRWCommon();
+
+	void tryTLSHandshake();
+	virtual void close();
+
+	void readAction();
+	void writeAction();
+	bool hasTlsPendingData() const;
+private:
+	virtual void processReadBuffer() = 0;
+};
+
+
+struct Packet {
+	int type;
+	int id;
+	Buffer data;
+};
+
+struct Client : SocketRWCommon {
+	enum AuthState {
+		AS_LOGIN_WAIT = 0,
+		AS_AUTHED = 1
+	};
+
+	AuthState authState;
+	uint8_t sessionKey[SESSION_KEY_SIZE];
+	time_t deadTime;
+
+	std::list<Packet *> packetCache;
+	int nextPacketID, lastReceivedPacketID;
+
+	Client();
+	~Client();
+
+	void startService(int _sock, bool withTls);
+	void close();
+
+private:
+	int readBufPosition;
+	void processReadBuffer();
+	void handleLine(char *line, int size);
+
+	void generateSessionKey();
+
+	void stealConnection(Client *other);
+};
+
+struct Server : SocketRWCommon {
+	char ircHostname[256];
+	int ircPort;
+	int dnsQueryId;
+	bool ircUseTls;
+
+	Server();
+	~Server();
+
+	void beginConnect();
+	void tryConnectPhase();
+	void connectionSuccessful();
+
+	void close();
+
+private:
+	void processReadBuffer();
+	void handleLine(char *line, int size);
+};
+
+
+#endif /* CORE_H */