#include <global_types.h>
#include <stdio.h>
#include <string.h>
#include "pso/macros.h"
#include "pso/TArray.h"
#include "pso/TMainTask.h"
#include "pso/TObject.h"
#include "pso/TSocket.h"
#include "pso/TTcpSocket.h"

OBJECT_NAME(TTcpSocket);
TTcpSocket *tcp_socket_table[16] = {nullptr};

int TTcpSocket::test_connection() {
	for (int i = 0; i < 1800; ++i) {
		if (int status = stat()) {
			log(get_sock_status_name(status));
			set_flags(1);
			return 1;
		} else if (send_window() <= 128) {
			set_flags(1);
			return 1;
		} else if (stat_val() < 0) {
			set_flags(1);
			return 1;
		} else if (!get_link_status()) {
			set_flags(1);
			return 1;
		} else if (send_window() > 4095) {
			return 0;
		} else {
			main_task.some_empty_func();
			main_task.run_tl_camera_tasks();
			main_task.render();
			main_task.render_screen_overlay();
			func_80019aa0();
			render_tick();
		}
	}

	set_flags(1);
	return 1;
}

void TTcpSocket::some_stub() {}

short TTcpSocket::stat() {
	if (sock_fd() != -1) {
		return tcp_stat(sock_fd(), &stat_val(), nullptr, &send_window(), &recv_window());
	} else {
		return 0;
	}
}

short TTcpSocket::close() {
	if (sock_fd() != -1) {
		(short)tcp_abort(sock_fd());
		(short)tcp_delete(sock_fd());
		tcp_socket_table[sock_fd()] = nullptr;
		set_sock_fd(-1);
		m_is_encrypted = 0;
		set_size(0);
		set_buffer_offset(0);
		set_buffer_cleared(true);
	}
	return sock_fd();
}

short TTcpSocket::open() {
	set_sock_fd(-1);
	m_is_encrypted = 0;
	set_size(0);
	set_buffer_offset(0);
	set_buffer_cleared(true);
	set_unused(0);
	set_sock_fd(tcp_create());

	if (sock_fd() < 0) {
		log(get_sock_status_name(sock_fd()));
		return -1;
	} else {
		struct at_ip_option connect_option;
		struct at_ip_addr connect_addr;
		struct at_ip_addr bind_addr;

		u32 opt = 1;

		tcp_get_opt(sock_fd(), 0x2001, &opt);

		bind_addr.type = 4;
		bind_addr.ip46.ip4 = 0;

		tcp_bind(sock_fd(), &bind_addr, src_port());

		connect_addr.type = 4;
		connect_addr.ip46.ip4 = dst_addr().addr;

		connect_option.type = 4;
		connect_option.ip46.ip4.ttl = 120;
		connect_option.ip46.ip4.svctype = 0;
		connect_option.ip46.ip4.df_flag = 0;

		if (tcp_connect(sock_fd(), &connect_addr, dst_port(), &connect_option)) {
			log(get_sock_status_name(sock_fd()));
			return -1;
		} else {
			for (;;) {
				func_803d96a4();
				controller_stuff();
				if (short status = stat()) {
					log(get_sock_status_name(status));
					return -2;
				} else if (stat_val() < 0) {
					return -2;
				} else {
					some_stub();
					if (stat_val() < 4) {
						if (!get_link_status()) {
							return -3;
						} else if (func_801a5d1c()) {
							return -2;
						} else {
							main_task.some_empty_func();
							main_task.run_tl_camera_tasks();
							main_task.render();
							main_task.render_screen_overlay();
							func_80019aa0();
							render_tick();
						}
					} else {
						tcp_socket_table[sock_fd()] = this;
						return sock_fd();
					}
				}
			}
		}
	}
}

WEAK_FUNC TTcpSocket::TTcpSocket(TObject *parent) : TSocket(parent) {
	set_name(TTcpSocket_name);
}

void func_80019aa0() {}
void controller_stuff() {}
char func_801a5d1c() {}
void render_tick() {}
void func_803d96a4() {}
short tcp_abort(short nh) {}
short tcp_bind(short nh, struct at_ip_addr *addr, u16 port) {}
short tcp_connect(short nh, struct at_ip_addr *addr, u16 port, struct at_ip_option *option) {}
short tcp_create() {}
short tcp_delete(short nh) {}
short tcp_get_opt(short nh, short type, u32 *opt) {}
short tcp_send(short nh, void (*notify)(short size, short sock_fd), char bufnum, struct send_buffs *sb) {}
short tcp_stat(short nh, short *stat, short *backlog, u32 *sendwin, u32 *recvwin) {}
short tcp_receive(short nh, void (*notify)(short size, short sock_fd), short len, u8 *buf) {}
int get_link_status() {}
char *get_sock_status_name(short code) {}