New serializer APIs ndpi_serialize_start_of_block / ndpi_serialize_end_of_block (JSON only)

1 files changed, 62 insertions, 1 deletions

diff --git a/src/lib/ndpi_serializer.c b/src/lib/ndpi_serializer.c
index 87ce5e364..49f29a4e8 100644
--- a/src/lib/ndpi_serializer.c
+++ b/src/lib/ndpi_serializer.c
@@ -454,7 +454,10 @@ static inline void ndpi_serialize_json_pre(ndpi_serializer *_serializer) {
     if(serializer->status.flags & NDPI_SERIALIZER_STATUS_ARRAY)
       serializer->status.size_used--; /* Remove ']'*/
     serializer->status.size_used--; /* Remove '}'*/
-    if(serializer->status.flags & NDPI_SERIALIZER_STATUS_COMMA)
+
+    if (serializer->status.flags & NDPI_SERIALIZER_STATUS_SOB)
+      serializer->status.flags &= ~NDPI_SERIALIZER_STATUS_SOB;
+    else if(serializer->status.flags & NDPI_SERIALIZER_STATUS_COMMA)
       serializer->buffer[serializer->status.size_used++] = ',';
   }
 }
@@ -1206,6 +1209,64 @@ int ndpi_serialize_string_string(ndpi_serializer *_serializer,
 
 /* ********************************** */
 
+/* Serialize start of nested block (JSON only)*/
+int ndpi_serialize_start_of_block(ndpi_serializer *_serializer,
+				  const char *key) {
+  ndpi_private_serializer *serializer = (ndpi_private_serializer*)_serializer;
+  u_int32_t buff_diff = serializer->buffer_size - serializer->status.size_used;
+  u_int32_t needed, klen = strlen(key);
+
+  if (serializer->fmt != ndpi_serialization_format_json)
+    return -1;
+
+  needed = 16 + klen;
+
+  if (buff_diff < needed) {
+    if (ndpi_extend_serializer_buffer(_serializer, needed - buff_diff) < 0)
+      return(-1);
+    buff_diff = serializer->buffer_size - serializer->status.size_used;
+  }
+
+  ndpi_serialize_json_pre(_serializer);
+  serializer->status.size_used += ndpi_json_string_escape(key, klen,
+    (char *) &serializer->buffer[serializer->status.size_used], buff_diff);
+  buff_diff = serializer->buffer_size - serializer->status.size_used;
+  serializer->status.size_used += snprintf((char *) &serializer->buffer[serializer->status.size_used], buff_diff, ": {");
+  buff_diff = serializer->buffer_size - serializer->status.size_used;
+  ndpi_serialize_json_post(_serializer);
+
+  serializer->status.flags |= NDPI_SERIALIZER_STATUS_SOB;
+
+  return(0);
+}
+
+/* ********************************** */
+
+/* Serialize start of nested block (JSON only)*/
+int ndpi_serialize_end_of_block(ndpi_serializer *_serializer) {
+  ndpi_private_serializer *serializer = (ndpi_private_serializer*)_serializer;
+  u_int32_t buff_diff = serializer->buffer_size - serializer->status.size_used;
+  u_int32_t needed;
+
+  if (serializer->fmt != ndpi_serialization_format_json)
+    return -1;
+
+  needed = 4;
+
+  if (buff_diff < needed) {
+    if (ndpi_extend_serializer_buffer(_serializer, needed - buff_diff) < 0)
+      return(-1);
+    buff_diff = serializer->buffer_size - serializer->status.size_used;
+  }
+
+  buff_diff = serializer->buffer_size - serializer->status.size_used;
+  ndpi_serialize_json_post(_serializer);
+
+  return(0);
+}
+
+/* ********************************** */
+
 void ndpi_serializer_create_snapshot(ndpi_serializer *_serializer) {
   ndpi_private_serializer *serializer = (ndpi_private_serializer*)_serializer;