Allow changing the accept callbacks.

This is used in the test, in order to stop accepting new connections after
the first one.

diff --git a/test.c b/test.c
index 8ad46c63f49eada35109e303582ed121e8444953..a37f85618db6c2b3eef6200711635fab0d3d1fc1 100644 (file)
--- a/test.c
+++ b/test.c
@@ -39,6 +39,7 @@ ssize_t do_recv(struct utcp_connection *c, const void *data, size_t len) {
 void do_accept(struct utcp_connection *nc, uint16_t port) {
        utcp_accept(nc, do_recv, NULL);
        c = nc;
+       utcp_set_accept_cb(c->utcp, NULL, NULL);
 }
 
 ssize_t do_send(struct utcp *utcp, const void *data, size_t len) {

diff --git a/utcp.c b/utcp.c
index e8755ed14470577b6baafbab0d127476454e4dba..062cc5af09380d4c2fe1d46496dae4200b6bf48b 100644 (file)
--- a/utcp.c
+++ b/utcp.c
@@ -1256,3 +1256,10 @@ void utcp_set_poll_cb(struct utcp_connection *c, utcp_poll_t poll) {
        if(c)
                c->poll = poll;
 }
+
+void utcp_set_accept_cb(struct utcp *utcp, utcp_accept_t accept, utcp_pre_accept_t pre_accept) {
+       if(utcp) {
+               utcp->accept = accept;
+               utcp->pre_accept = pre_accept;
+       }
+}

diff --git a/utcp.h b/utcp.h
index 7ebca734bfa36f650b55f829a3e8f80c2def80d1..8d3874f15476c28cc3b40d949e825f11a4720b71 100644 (file)
--- a/utcp.h
+++ b/utcp.h
@@ -61,6 +61,7 @@ extern int utcp_shutdown(struct utcp_connection *connection, int how);
 extern struct timeval utcp_timeout(struct utcp *utcp);
 extern void utcp_set_recv_cb(struct utcp_connection *connection, utcp_recv_t recv);
 extern void utcp_set_poll_cb(struct utcp_connection *connection, utcp_poll_t poll);
+extern void utcp_set_accept_cb(struct utcp *utcp, utcp_accept_t accept, utcp_pre_accept_t pre_accept);
 
 // Global socket options