aboutsummaryrefslogtreecommitdiff
path: root/main/sock.c
blob: fe932bb9f95c043e583afb901e31c18bdb3f3f0c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
#include <pico/stdio.h>

#include <lwip/opt.h>
#include <lwip/sys.h>
#include <lwip/api.h>
#include <string.h>

#include "init.h"
#include "config.h"
#include "i2ctcpv1.h"
#include "sock.h"

struct netconn* current_connection = NULL;
i2ctcp_msg_t recv_msg;

void i2c_send(uint16_t addr, const char * data, size_t data_size) {
	if (current_connection == NULL) return;

	i2ctcp_msg_t send_msg = {
		.addr = addr,
		.data = (char *) data,
		.length = data_size,
	};

	char * buf;
	size_t buf_sz;

	if (!i2ctcp_write(&send_msg, &buf, &buf_sz)) return;

	// NOTE: netconn does return an error code, but the data needs to be freed
	// whether netconn throws an error or not, so it remains unused
	netconn_write(current_connection, buf, buf_sz, NETCONN_COPY);

	free(buf);
}

void i2c_recv(uint16_t addr, const char * data, size_t data_size) {
	printf("address: 0x%02x\n", addr);
	printf("data:    \"%.*s\"\n", data_size, data);

	// send message back
	char reply[] = "Test message back!";
	i2c_send(0x69, reply, strlen(reply));

	// TODO: this function should forward the recieved message onto the puzzle
	// bus instead of printing/replying
}

void recv_handler(struct netconn* conn, struct netbuf* buf) {
	i2ctcp_read_reset(&recv_msg);

	do {
		char* data;
		uint16_t len;
		netbuf_data(buf, (void**)&data, &len);

		// continue early if more data is needed to complete message
		if (!i2ctcp_read(&recv_msg, data, len)) continue;

		// forward received message to puzzle bus
		i2c_recv(recv_msg.addr, recv_msg.data, recv_msg.length);
		free(recv_msg.data);
	} while (netbuf_next(buf) >= 0);

	netbuf_delete(buf);
}

void accept_handler(struct netconn* conn) {
	current_connection = conn;

	struct netbuf* buf;
	while (netconn_recv(conn, &buf) == ERR_OK)
		recv_handler(conn, buf);

	netconn_close(conn);
	netconn_delete(conn);

	current_connection = NULL;
}

void serve_task() {
	await_init();

	printf("starting server...\n");
	struct netconn* conn = netconn_new(NETCONN_TCP);
	netconn_bind(conn, IP_ADDR_ANY, CONF_SRV_PORT);
	netconn_listen(conn);

	printf("listening on %s:%d\n", ip4addr_ntoa(netif_ip4_addr(netif_list)), CONF_SRV_PORT);
	while (1) {
		struct netconn* incoming;
		if (netconn_accept(conn, &incoming) == ERR_OK)
			accept_handler(incoming);
	}
}