Switch to using the new apr_poll() API.


git-svn-id: https://svn.apache.org/repos/asf/httpd/test/trunk/flood@96014 13f79535-47bb-0310-9956-ffa450edef68
diff --git a/CHANGES b/CHANGES
index fad2b8c..b550e11 100644
--- a/CHANGES
+++ b/CHANGES
@@ -1,5 +1,7 @@
 Changes since milestone-03:
 
+* Switch to new apr_poll() API.  [Justin Erenkrantz]
+
 * Stop using the apr_lock.h API and start using apr_thread_mutex.h
   instead.  [Aaron Bannert]
 
diff --git a/flood_net.c b/flood_net.c
index 30d8669..8c10096 100644
--- a/flood_net.c
+++ b/flood_net.c
@@ -103,10 +103,11 @@
     }
 
     apr_setsocketopt(fs->socket, APR_SO_TIMEOUT, LOCAL_SOCKET_TIMEOUT);
-
-    apr_poll_setup(&fs->poll, 1, pool);
-    apr_poll_socket_add(fs->poll, fs->socket, APR_POLLIN);
-
+    fs->read_pollset.desc_type = APR_POLL_SOCKET;
+    fs->read_pollset.desc.s = fs->socket;
+    fs->read_pollset.reqevents = APR_POLLIN;
+    fs->read_pollset.p = pool;
+    
     return fs;
 }
 
@@ -120,9 +121,9 @@
 apr_status_t read_socket(flood_socket_t *s, char *buf, int *buflen)
 {
     apr_status_t e;
-    int socketsRead = 1;
+    apr_int32_t socketsRead;
 
-    e = apr_poll(s->poll, &socketsRead, LOCAL_SOCKET_TIMEOUT);
+    e = apr_poll(&s->read_pollset, 1, &socketsRead, LOCAL_SOCKET_TIMEOUT);
     if (e != APR_SUCCESS)
         return e;
     return apr_recv(s->socket, buf, buflen);
@@ -147,19 +148,18 @@
 apr_status_t check_socket(flood_socket_t *s, apr_pool_t *pool)
 {
     apr_status_t e;
-    int socketsRead = 1;
-    apr_pollfd_t *pout;
+    apr_int32_t socketsRead;
+    apr_pollfd_t pout;
     apr_int16_t event;
 
-    apr_poll_setup(&pout, 1, pool);
-    apr_poll_socket_add(pout, s->socket, APR_POLLIN | APR_POLLPRI | APR_POLLERR | APR_POLLHUP | APR_POLLNVAL);
-
-    e = apr_poll(pout, &socketsRead, 1000);
-    if (socketsRead) {
-        apr_poll_revents_get(&event, s->socket, pout);
-        if (event) {
-            return APR_EGENERAL;
-        }
+    pout.desc_type = APR_POLL_SOCKET;
+    pout.desc.s = s->socket;
+    pout.reqevents = APR_POLLIN | APR_POLLPRI | APR_POLLERR | APR_POLLHUP | APR_POLLNVAL;
+    pout.p = pool;
+    
+    e = apr_poll(&pout, 1, &socketsRead, 1000);
+    if (socketsRead && pout.rtnevents) {
+        return APR_EGENERAL;
     }
     
     return APR_SUCCESS;
diff --git a/flood_net.h b/flood_net.h
index 297d3f8..ae851b0 100644
--- a/flood_net.h
+++ b/flood_net.h
@@ -58,13 +58,14 @@
 #define __flood_socket_h
 
 #include <apr_network_io.h> /* apr_socket_t */
+#include <apr_poll.h>       /* apr_pollfd_t */
 #include <apr_pools.h>      /* apr_pool_t */
 
 #include "flood_profile.h"
 
 typedef struct flood_socket_t {
     apr_socket_t *socket;
-    apr_pollfd_t *poll;
+    apr_pollfd_t read_pollset;
 } flood_socket_t;
 
 flood_socket_t* open_socket(apr_pool_t *pool, request_t *r);
diff --git a/flood_net_ssl.c b/flood_net_ssl.c
index 03979ba..f3c52d8 100644
--- a/flood_net_ssl.c
+++ b/flood_net_ssl.c
@@ -277,12 +277,13 @@
 apr_status_t ssl_read_socket(ssl_socket_t *s, char *buf, int *buflen)
 {
     apr_status_t e;
-    int sslError, socketsRead;
+    int sslError;
+    apr_int32_t socketsRead;
 
     /* Wait until there is something to read. */
     if (SSL_pending(s->ssl_connection) < *buflen) {
-        socketsRead = 1;
-        e = apr_poll(s->socket->poll, &socketsRead, LOCAL_SOCKET_TIMEOUT);
+        e = apr_poll(&s->socket->read_pollset, 1, &socketsRead,
+                     LOCAL_SOCKET_TIMEOUT);
 
         if (socketsRead != 1)
             return APR_TIMEUP;
@@ -319,9 +320,10 @@
     char buf[1];
     int buflen = 1; 
     /* Wait until there is something to read. */
-    int socketsRead = 1;
+    apr_int32_t socketsRead;
     apr_status_t e;
-    e = apr_poll(s->socket->poll, &socketsRead, LOCAL_SOCKET_TIMEOUT);
+    e = apr_poll(&s->socket->read_pollset, 1, &socketsRead,
+                 LOCAL_SOCKET_TIMEOUT);
     e = SSL_read(s->ssl_connection, buf, buflen);
 }