Changeset 2187 for pjproject/trunk/pjlib/src/pj/activesock.c

Timestamp:

Aug 4, 2008 9:59:02 AM (16 years ago)

Author:

bennylp

Message:

Added new active socket API's to specify application buffers in start_read() and start_recv() functions

File:

• pjproject/trunk/pjlib/src/pj/activesock.c (modified) (6 diffs)

pjproject/trunk/pjlib/src/pj/activesock.c

@@ -18 +18 @@
  */
 #include <pj/activesock.h>
+#include <pj/compat/socket.h>
 #include <pj/assert.h>
 #include <pj/errno.h>
@@ -242 +243 @@
                                              pj_uint32_t flags)
 {
+    void **readbuf;
+    unsigned i;
+
+    PJ_ASSERT_RETURN(asock && pool && buff_size, PJ_EINVAL);
+
+    readbuf = (void**) pj_pool_calloc(pool, asock->async_count, 
+                                      sizeof(void*));
+
+    for (i=0; i<asock->async_count; ++i) {
+        readbuf[i] = pj_pool_alloc(pool, buff_size);
+    }
+
+    return pj_activesock_start_read2(asock, pool, buff_size, readbuf, flags);
+}
+
+
+PJ_DEF(pj_status_t) pj_activesock_start_read2( pj_activesock_t *asock,
+                                               pj_pool_t *pool,
+                                               unsigned buff_size,
+                                               void *readbuf[],
+                                               pj_uint32_t flags)
+{
     unsigned i;
     pj_status_t status;
@@ -259 +282 @@
         pj_ssize_t size_to_read;
 
-        r->pkt = (pj_uint8_t*)pj_pool_alloc(pool, buff_size);
+        r->pkt = (pj_uint8_t*)readbuf[i];
         r->max_size = size_to_read = buff_size;
 
@@ -278 +301 @@
                                                  unsigned buff_size,
                                                  pj_uint32_t flags)
+{
+    void **readbuf;
+    unsigned i;
+
+    PJ_ASSERT_RETURN(asock && pool && buff_size, PJ_EINVAL);
+
+    readbuf = (void**) pj_pool_calloc(pool, asock->async_count, 
+                                      sizeof(void*));
+
+    for (i=0; i<asock->async_count; ++i) {
+        readbuf[i] = pj_pool_alloc(pool, buff_size);
+    }
+
+    return pj_activesock_start_recvfrom2(asock, pool, buff_size, 
+                                         readbuf, flags);
+}
+
+
+PJ_DEF(pj_status_t) pj_activesock_start_recvfrom2( pj_activesock_t *asock,
+                                                   pj_pool_t *pool,
+                                                   unsigned buff_size,
+                                                   void *readbuf[],
+                                                   pj_uint32_t flags)
 {
     unsigned i;
@@ -295 +341 @@
         pj_ssize_t size_to_read;
 
-        r->pkt = (pj_uint8_t*) pj_pool_alloc(pool, buff_size);
+        r->pkt = (pj_uint8_t*) readbuf[i];
         r->max_size = size_to_read = buff_size;
         r->src_addr_len = sizeof(r->src_addr);
@@ -372 +418 @@
             }
 
-        } else if (bytes_read <= 0) {
-
+        } else if (bytes_read <= 0 &&
+                   -bytes_read != PJ_STATUS_FROM_OS(OSERR_EWOULDBLOCK) &&
+                   -bytes_read != PJ_STATUS_FROM_OS(OSERR_EINPROGRESS) && 
+                   -bytes_read != PJ_STATUS_FROM_OS(OSERR_ECONNRESET)) 
+        {
             pj_size_t remainder;
             pj_bool_t ret;