X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ian/git?a=blobdiff_plain;f=test%2Fudp-preload.c;h=653dfa57c01f6b74a69e3c57bd2a98a15a8cbc2c;hb=0f3a091767b7b16f9ec570c699641dc229e54856;hp=c4bf49d54a87ddc4c2cc3c3d7ec2859af9af6b40;hpb=9913c83f9120afa7bd10d89f460ccf01ed347dc9;p=secnet.git

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);