X-Git-Url: https://www.chiark.greenend.org.uk/ucgi/~mdw/git/secnet/blobdiff_plain/53f4e66603bcc2d21c70ad7a69db5fda6da5d645..5c679ae0bf87d5d53b65c5e7667a1fa304bdcdeb:/udp.c

diff --git a/udp.c b/udp.c
index 8f0bb8b..a58690e 100644
--- a/udp.c
+++ b/udp.c
@@ -288,7 +288,14 @@ failed:
 void udp_socks_register(struct udpcommon *uc, struct udpsocks *socks)
 {
     socks->uc=uc;
-    register_for_poll(socks,udp_socks_beforepoll,udp_socks_afterpoll,"udp");
+    socks->interest=
+	register_for_poll(socks,udp_socks_beforepoll,udp_socks_afterpoll,"udp");
+}
+
+void udp_socks_deregister(struct udpcommon *uc, struct udpsocks *socks)
+{
+    socks->uc=uc;
+    deregister_for_poll(socks->interest);
 }
 
 static void udp_phase_hook(void *sst, uint32_t new_phase)