瀏覽代碼

Increase buffer size if needed when appending data to send

Sylver Bruneau 9 年之前
父節點
當前提交
01ca1dc934
共有 1 個文件被更改,包括 2 次插入1 次删除
  1. 2 1
      src/netbios_session.c

+ 2 - 1
src/netbios_session.c

@@ -180,7 +180,8 @@ int               netbios_session_packet_append(netbios_session *s,
     assert(s && s->packet);
 
     if (s->packet_payload_size - s->packet_cursor < size)
-        return (0);
+        if (!session_buffer_realloc(s, size + s->packet_cursor))
+            return 0;
 
     start = ((char *)&s->packet->payload) + s->packet_cursor;
     memcpy(start, data, size);