From 115fe1992e2c50fd108f7bc7cc31d7d2b1f5c6b8 Mon Sep 17 00:00:00 2001
From: mrb0nk500 <b0nk@b0nk.xyz>
Date: Sun, 26 Feb 2023 19:20:47 -0400
Subject: TSocket: Start work on, and match most of `TSocket`

Still need to implement, and match `resolve_domain()`, but it's a good
starting point for decompiling the netcode.
---
 context.h | 80 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 79 insertions(+), 1 deletion(-)

(limited to 'context.h')

diff --git a/context.h b/context.h
index f42ba4a..007aa61 100644
--- a/context.h
+++ b/context.h
@@ -485,6 +485,13 @@ struct packet_header {
 	void bswap();
 } __packed__;
 
+// pso/TSocket.h
+// Union defs.
+union ipv4_addr {
+	u32 addr;
+	TArray<u8, 4> addr_bytes;
+};
+
 // pso/TObject.h
 // Enum defs.
 enum object_flags {
@@ -516,8 +523,79 @@ static inline void operator^=(object_flags &a, object_flags b) { a = a ^ b; };
 static inline void operator&=(object_flags &a, object_flags b) { a = a & b; };
 static inline void operator|=(object_flags &a, object_flags b) { a = a | b; };
 
-// pso/TPlyGuildCardTag.h
+// pso/TSocket.h
 // Class defs.
+class TSocket : public TObject {
+private:
+	void set_flags(u8 flags) {
+		m_sock_flags |= flags;
+	};
+
+	void clear_flags(u8 flags) {
+		m_sock_flags &= ~flags;
+	};
+
+	void toggle_flags(u8 flags) {
+		m_sock_flags ^= flags;
+	};
+
+	u8 get_flags(u8 flags) {
+		return m_sock_flags & flags;
+	};
+
+	ipv4_addr m_dst_addr;
+	u16 m_dst_port;
+	u16 m_src_port;
+	ipv4_addr m_src_addr;
+	s16 m_sock_fd;
+	u8 m_sock_flags;
+	bool m_buffer_cleared;
+	s16 m_size;
+	s16 m_buffer_offset;
+	u32 m_unused;
+	TArray<u8, 64> m_unused2;
+	TArray<u8, 2048> m_packet_buffer;
+	s16 m_stat;
+	u16 m_unused3;
+	u32 m_send_window;
+	u32 m_recv_window;
+	void (*m_callback)(TSocket *socket);
+public:
+	TSocket(TObject *parent);
+	virtual ~TSocket();
+
+	virtual int open() = 0;
+	virtual int close() = 0;
+	virtual void recv() = 0;
+	virtual int send(u8 *data) = 0;
+	virtual int send(u8 *data, size_t size) = 0;
+
+	int resolve_domain(char *domain);
+	void set_ip_address(u32 addr);
+	void set_port(u32 port);
+	const u8 next();
+	int is_empty();
+
+	PRIVATE_MEMBER_ACCESSORS(ipv4_addr, dst_addr);
+	PRIVATE_MEMBER_ACCESSORS(u16, dst_port);
+	PRIVATE_MEMBER_ACCESSORS(u16, src_port);
+	PRIVATE_MEMBER_ACCESSORS(ipv4_addr, src_addr);
+	PRIVATE_MEMBER_ACCESSORS(s16, sock_fd);
+	PRIVATE_MEMBER_ACCESSORS(u8, sock_flags);
+	PRIVATE_MEMBER_ACCESSORS(bool, buffer_cleared);
+	PRIVATE_MEMBER_ACCESSORS(s16, size);
+	PRIVATE_MEMBER_ACCESSORS(s16, buffer_offset);
+	PRIVATE_MEMBER_ACCESSORS(u32, unused);
+	PRIVATE_MEMBER_ACCESSORS_ARRAY(u8, unused2, 64);
+	PRIVATE_MEMBER_ACCESSORS_ARRAY(u8, packet_buffer, 2048);
+	PRIVATE_MEMBER_ACCESSORS(s16, stat);
+	PRIVATE_MEMBER_ACCESSORS(u16, unused3);
+	PRIVATE_MEMBER_ACCESSORS(u32, send_window);
+	PRIVATE_MEMBER_ACCESSORS(u32, recv_window);
+	PRIVATE_MEMBER_ACCESSORS_FUNC(void, callback, TSocket *socket);
+};
+
+// pso/TPlyGuildCardTag.h
 class TPlyGuildCardTag {
 	public:
 	u8 tag0;
-- 
cgit v1.2.3-13-gbd6f