Changeset 2380 for pjproject/trunk/pjsip/src/pjsip/sip_transport.c

Timestamp:

Dec 17, 2008 2:28:18 PM (15 years ago)

Author:

bennylp

Message:

Ticket 684: protect the memory allocation for TX packet with try/catch, and fixed various transmit data buffer leaks when transmission fails immediately

File:

• pjproject/trunk/pjsip/src/pjsip/sip_transport.c (modified) (1 diff)

pjproject/trunk/pjsip/src/pjsip/sip_transport.c

@@ -569 +569 @@
     /* Allocate buffer if necessary. */
     if (tdata->buf.start == NULL) {
-        tdata->buf.start = (char*) 
-                           pj_pool_alloc( tdata->pool, PJSIP_MAX_PKT_LEN);
+        PJ_USE_EXCEPTION;
+
+        PJ_TRY {
+            tdata->buf.start = (char*) 
+                               pj_pool_alloc(tdata->pool, PJSIP_MAX_PKT_LEN);
+        }
+        PJ_CATCH_ANY {
+            return PJ_ENOMEM;
+        }
+        PJ_END
+
         tdata->buf.cur = tdata->buf.start;
         tdata->buf.end = tdata->buf.start + PJSIP_MAX_PKT_LEN;