[agroal#662] Add PostgreSQL message logging

Author: ankushT369

doc/manual/en/71-git.md

@@ -78,14 +78,12 @@ and then create a pull request for it
 
 **Format source code**
 
-Use
+Use the `clang-format.sh` script from the root directory of the project to apply consistent formatting:
 
-``` sh
+```sh
 ./clang-format.sh
 ```
 
-to format the source code
-
 **Repeat**
 
 Based on feedback keep making changes, squashing, rebasing and force pushing

src/include/logging.h

@@ -65,6 +65,12 @@ extern "C" {
 #define pgagroal_log_error(...)                  pgagroal_log_line(PGAGROAL_LOGGING_LEVEL_ERROR, __FILE__, __LINE__, __VA_ARGS__)
 #define pgagroal_log_fatal(...)                  pgagroal_log_line(PGAGROAL_LOGGING_LEVEL_FATAL, __FILE__, __LINE__, __VA_ARGS__)
 
+#ifdef DEBUG
+#define PGAGROAL_LOG_POSTGRES(x) pgagroal_log_postgres(x)
+#else
+#define PGAGROAL_LOG_POSTGRES(x) ((void)(x))
+#endif
+
 /**
  * Initialize the logging system
  * @return 0 upon success, otherwise 1
@@ -184,6 +190,14 @@ log_file_open(void);
 void
 log_file_rotate(void);
 
+/**
+ * The function parses PostgreSQL wire-protocol from
+ * raw data stream and logs message type and length information.
+ * @param msg The message containing raw PostgreSQL wire protocol data
+ */
+void
+pgagroal_log_postgres(struct message* msg);
+
 #ifdef __cplusplus
 }
 #endif

src/libpgagroal/art.c

@@ -1749,4 +1749,4 @@ static int
 art_iterate(struct art* t, art_callback cb, void* data)
 {
    return art_node_iterate(t->root, cb, data);
-}
+}

src/libpgagroal/logging.c

@@ -632,6 +632,98 @@ pgagroal_log_mem(void* data, size_t size)
    }
 }
 
+static void*
+dynamic_buffer_append(void* orig, size_t orig_size,
+                      void* append, size_t append_size,
+                      size_t* new_size)
+{
+   void* d = NULL;
+   size_t s;
+
+   if (append != NULL)
+   {
+      s = orig_size + append_size;
+      d = realloc(orig, s);
+      memcpy((char*)d + orig_size, append, append_size);
+   }
+   else
+   {
+      s = orig_size;
+      d = orig;
+   }
+
+   *new_size = s;
+   return d;
+}
+
+static void* partial_buf = NULL;
+static size_t partial_buf_size = 0;
+
+void
+pgagroal_log_postgres(struct message* msg)
+{
+   size_t offset = 0;
+   size_t remaining = 0;
+   uint8_t* buf = NULL;
+   void* new_buf = NULL;
+
+   uint32_t msg_len = 0;
+   uint8_t msg_type = ' ';
+
+   if (!msg)
+   {
+      return;
+   }
+
+   partial_buf = dynamic_buffer_append(partial_buf, partial_buf_size, msg->data, msg->length, &partial_buf_size);
+
+   buf = (uint8_t*)partial_buf;
+
+   while (offset + 5 <= partial_buf_size)
+   {
+      memcpy(&msg_len, buf + offset + 1, 4);
+      msg_len = ntohl(msg_len);
+
+      if (msg_len < 4)
+      {
+         partial_buf_size = 0;
+         break;
+      }
+
+      if (offset + 1 + msg_len > partial_buf_size)
+      {
+         break;
+      }
+
+      msg_type = buf[offset];
+      pgagroal_log_trace("Message type: %c len: %u", msg_type, msg_len);
+
+      offset += (1 + msg_len);
+   }
+
+   if (offset > 0)
+   {
+      remaining = partial_buf_size - offset;
+      if (remaining > 0)
+      {
+         memmove(partial_buf, buf + offset, remaining);
+
+         new_buf = realloc(partial_buf, remaining);
+         if (new_buf)
+         {
+            partial_buf = new_buf;
+         }
+      }
+      else
+      {
+         free(partial_buf);
+         partial_buf = NULL;
+      }
+
+      partial_buf_size = remaining;
+   }
+}
+
 static void
 output_log_line(char* l)
 {

src/libpgagroal/pipeline_perf.c

@@ -120,6 +120,7 @@ performance_client(struct io_watcher* watcher)
    wi = (struct worker_io*)watcher;
 
    status = pgagroal_recv_message(watcher, &msg);
+   PGAGROAL_LOG_POSTGRES(msg);
 
    if (likely(status == MESSAGE_STATUS_OK))
    {

src/libpgagroal/status.c

@@ -264,6 +264,7 @@ status_details(bool details, struct json* response)
 
          pgagroal_json_create(&js);
 
+         pgagroal_json_put(js, MANAGEMENT_ARGUMENT_DATABASE, (uintptr_t)"*", ValueString);
          pgagroal_json_put(js, MANAGEMENT_ARGUMENT_DATABASE, (uintptr_t)"*", ValueString);
          pgagroal_json_put(js, MANAGEMENT_ARGUMENT_ENABLED, (uintptr_t)!config->all_disabled, ValueBool);
 

src/libpgagroal/utils.c

@@ -328,6 +328,7 @@ uint8_t
 pgagroal_read_uint8(void* data)
 {
    return (uint8_t)*((char*)data);
+   return (uint8_t)*((char*)data);
 }
 
 int16_t