Changes in uspace/srv/net/tcp/pdu.c [2f19103:4a0bc99] in mainline

File:

• uspace/srv/net/tcp/pdu.c (modified) (3 diffs)

uspace/srv/net/tcp/pdu.c

@@ -249 +249 @@
         if (pdu->text != NULL)
                 free(pdu->text);
+        free(pdu);
 
         return NULL;
@@ -323 +324 @@
         size_t text_size;
         uint16_t checksum;
+        int rc;
 
         npdu = tcp_pdu_new();
@@ -330 +332 @@
         npdu->src = epp->local.addr;
         npdu->dest = epp->remote.addr;
-        tcp_header_encode(epp, seg, &npdu->header, &npdu->header_size);
+        rc = tcp_header_encode(epp, seg, &npdu->header, &npdu->header_size);
+        if (rc != EOK) {
+                free(npdu);
+                return rc;
+        }
 
         text_size = tcp_segment_text_size(seg);
         npdu->text = calloc(1, text_size);
-        if (npdu->text == NULL)
+        if (npdu->text == NULL) {
+                free(npdu->header);
+                free(npdu);
                 return ENOMEM;
+        }
 
         npdu->text_size = text_size;