From 0f3a091767b7b16f9ec570c699641dc229e54856 Mon Sep 17 00:00:00 2001
From: Ian Jackson <ijackson@chiark.greenend.org.uk>
Date: Sat, 12 Oct 2019 13:09:37 +0100
Subject: [PATCH] test: udp-preload: Provide close

Signed-off-by: Ian Jackson <ijackson@chiark.greenend.org.uk>
---
 test/udp-preload.c | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/test/udp-preload.c b/test/udp-preload.c
index c4bf49d..653dfa5 100644
--- a/test/udp-preload.c
+++ b/test/udp-preload.c
@@ -58,12 +58,14 @@ static anyfn_type *find_any(const char *name) {
 }
 
 #define socket_args int domain, int type, int protocol
+#define close_args  int fd
 #define bind_args   int fd, const struct sockaddr *addr, socklen_t addrlen
 #define setsockopt_args  int fd, int level, int optname, \
                          const void *optval, socklen_t optlen
 #define getsockname_args int fd, struct sockaddr *addr, socklen_t *addrlen
 #define WRAPS(X)					\
     X(socket,     (domain,type,protocol))		\
+    X(close,      (fd))					\
     X(bind,       (fd,addr,addrlen))			\
     X(setsockopt, (fd,level,optname,optval,optlen))	\
     X(getsockname,(fd,addr,addrlen))
@@ -180,6 +182,14 @@ WRAP(socket) {
     return -1;
 }
 
+WRAP(close) {
+    if (fd<tablesz) {
+	free(table[fd]);
+	table[fd]=0;
+    }
+    return old_close(fd);
+}
+
 WRAP(bind) {
     fdinfo *ent=lookup(fd);
     if (!ent) return old_bind(fd,addr,addrlen);
-- 
2.30.2