fix resuming accept() calls after exceeding client limit

diff --git a/client.c b/client.c
index 7920102..dd3f912 100644 (file)
--- a/client.c
+++ b/client.c
@@ -438,7 +438,7 @@ static void set_addr(struct uh_addr *addr, void *src)
        }
 }
 
        }
 }
 

-void uh_accept_client(int fd)
+bool uh_accept_client(int fd)

 {
        static struct client *next_client;
        struct client *cl;
 {
        static struct client *next_client;
        struct client *cl;


@@ -455,7 +455,7 @@ void uh_accept_client(int fd)
        sl = sizeof(addr);
        sfd = accept(fd, (struct sockaddr *) &addr, &sl);
        if (sfd < 0)
        sl = sizeof(addr);
        sfd = accept(fd, (struct sockaddr *) &addr, &sl);
        if (sfd < 0)

-               return;
+               return false;

 
        set_addr(&cl->peer_addr, &addr);
        sl = sizeof(addr);
 
        set_addr(&cl->peer_addr, &addr);
        sl = sizeof(addr);


@@ -476,6 +476,7 @@ void uh_accept_client(int fd)
        next_client = NULL;
        n_clients++;
        cl->id = client_id++;
        next_client = NULL;
        n_clients++;
        cl->id = client_id++;

+       return true;

 }
 
 void uh_close_fds(void)
 }
 
 void uh_close_fds(void)

diff --git a/listen.c b/listen.c
index c4b6b2b..f11768a 100644 (file)
--- a/listen.c
+++ b/listen.c
@@ -56,7 +56,7 @@ void uh_unblock_listeners(void)
 {
        struct listener *l;
 
 {
        struct listener *l;
 

-       if (!n_blocked && conf.max_requests &&
+       if ((!n_blocked && conf.max_requests) ||

            n_clients >= conf.max_requests)
                return;
 
            n_clients >= conf.max_requests)
                return;
 


@@ -64,6 +64,10 @@ void uh_unblock_listeners(void)
                if (!l->blocked)
                        continue;
 
                if (!l->blocked)
                        continue;
 

+               l->fd.cb(&l->fd, ULOOP_READ);
+           if (n_clients >= conf.max_requests)
+                       break;
+

                n_blocked--;
                l->blocked = false;
                uloop_fd_add(&l->fd, ULOOP_READ);
                n_blocked--;
                l->blocked = false;
                uloop_fd_add(&l->fd, ULOOP_READ);


@@ -74,7 +78,10 @@ static void listener_cb(struct uloop_fd *fd, unsigned int events)
 {
        struct listener *l = container_of(fd, struct listener, fd);
 
 {
        struct listener *l = container_of(fd, struct listener, fd);
 

-       uh_accept_client(fd->fd);
+       while (1) {
+               if (!uh_accept_client(fd->fd))
+                       break;
+       }

 
        if (conf.max_requests && n_clients >= conf.max_requests)
                uh_block_listener(l);
 
        if (conf.max_requests && n_clients >= conf.max_requests)
                uh_block_listener(l);

diff --git a/uhttpd.h b/uhttpd.h
index a9777e8..d67e48c 100644 (file)
--- a/uhttpd.h
+++ b/uhttpd.h
@@ -192,7 +192,7 @@ extern struct dispatch_handler cgi_dispatch;
 
 void uh_index_add(const char *filename);
 
 
 void uh_index_add(const char *filename);
 

-void uh_accept_client(int fd);
+bool uh_accept_client(int fd);

 
 void uh_unblock_listeners(void);
 void uh_setup_listeners(void);
 
 void uh_unblock_listeners(void);
 void uh_setup_listeners(void);