initial commit

net/UDP.h

@@ -0,0 +1,103 @@
+#ifndef UDP_H
+#define UDP_H
+
+extern "C" {
+	#include "mem.h"
+	#include "espconn.h"
+}
+
+typedef void (*UDPCallback)(void* arg, char* data, unsigned short len);
+
+#include "../Debug.h"
+#include "IP.h"
+
+class UDP {
+
+private:
+
+	static constexpr const char* NAME = "UDP";
+
+	espconn* con;
+
+public:
+
+	UDP() {
+		init();
+	}
+
+	/** dtor */
+	~UDP() {
+		cleanup();
+	}
+
+
+
+	/** bind the socket to the given local port */
+	void bind(const Port localPort) {
+
+		debugMod1(NAME, "binding to local port %d", localPort);
+
+		// set the local port to listen on
+		con->proto.udp->local_port = localPort;
+
+		// todo: check? 0=OK
+		const int res = espconn_create(con);
+		os_printf("create: %d\r\n", res);
+
+	}
+
+	bool send(const IP ip, const Port port, const void* data, const uint16_t dataLen) {
+
+		debugMod1(NAME, "sending packet to remote port %d", port);
+
+		// set remote port and IP
+		con->proto.udp->remote_port = port;
+		os_memcpy(con->proto.udp->remote_ip, &(ip.val), 4);
+
+		//os_printf("send %d bytes\r\n", dataLen);
+		//os_printf("IP: %d.%d.%d.%d\n\r", con->proto.udp->remote_ip[0], con->proto.udp->remote_ip[1], con->proto.udp->remote_ip[2], con->proto.udp->remote_ip[3]);
+
+		// send. TODO: check. 0=OK
+		const int res = espconn_sent(con, (unsigned char*)data, dataLen);
+		return (res == 0);
+
+	}
+
+	/** set the callback to call whenever a packet is received */
+	void setRecvCallback(UDPCallback callback) {
+		espconn_regist_recvcb(con, callback);
+	}
+
+private:
+
+	/** initialize the UDP "connection" */
+	void init() {
+
+		debugMod(NAME, "init()");
+
+		// allocate connection-objects
+		con = (espconn*) os_zalloc(sizeof(espconn));
+		ets_memset( con, 0, sizeof(  espconn ) );
+
+		// configure
+		con->type = ESPCONN_UDP;
+		//con->state = ESPCONN_NONE;
+
+		con->proto.udp = (esp_udp*) os_zalloc(sizeof(esp_udp));
+
+	}
+
+	/** cleanup everything */
+	void cleanup() {
+
+		debugMod(NAME, "cleanup()");
+
+		espconn_delete(con);
+		os_free(con->proto.udp);
+		os_free(con);
+
+	}
+
+};
+
+#endif // UDP_H