native_http_log.c

Go to the documentation of this file.

/**
 * @file
 * @brief Supplies MIME lookup, status parsing, client IP extraction, and log-safe HTTP formatting helpers.
 */
 
#include "native_http_log.h"
 
#include "native_support.h"
 
#include <ctype.h>
#include <stdlib.h>
#include <string.h>
#ifndef _WIN32
#include <strings.h>
#endif
 
#ifdef _WIN32
#define pharos_http_log_strdup _strdup
#define pharos_http_log_strcasecmp _stricmp
#else
#define pharos_http_log_strdup strdup
#define pharos_http_log_strcasecmp strcasecmp
#endif
 
char *mime_type_for_path(const char *path) {
  const char *ext = strrchr(path, '.');
  if (ext == NULL) {
    return pharos_http_log_strdup("application/octet-stream");
  }
  if (strcmp(ext, ".html") == 0) return pharos_http_log_strdup("text/html; charset=utf-8");
  if (strcmp(ext, ".json") == 0) return pharos_http_log_strdup("application/json");
  if (strcmp(ext, ".css") == 0) return pharos_http_log_strdup("text/css; charset=utf-8");
  if (strcmp(ext, ".js") == 0) return pharos_http_log_strdup("application/javascript; charset=utf-8");
  if (strcmp(ext, ".svg") == 0) return pharos_http_log_strdup("image/svg+xml");
  if (strcmp(ext, ".png") == 0) return pharos_http_log_strdup("image/png");
  if (strcmp(ext, ".jpg") == 0 || strcmp(ext, ".jpeg") == 0) return pharos_http_log_strdup("image/jpeg");
  return pharos_http_log_strdup("application/octet-stream");
}
 
char *compact_log_text_native(const char *value) {
  Buffer buffer;
  const unsigned char *cursor;
  int pending_space = 0;
  buffer_init(&buffer);
  if (value == NULL) {
    return pharos_http_log_strdup("");
  }
  for (cursor = (const unsigned char *)value; *cursor != '\0'; cursor++) {
    if (*cursor == '\r' || *cursor == '\n' || isspace(*cursor)) {
      pending_space = 1;
      continue;
    }
    if (pending_space && buffer.len > 0) {
      if (buffer_append(&buffer, " ", 1) != 0) {
        buffer_free(&buffer);
        return NULL;
      }
      pending_space = 0;
    }
    {
      char single = (char)*cursor;
      if (buffer_append(&buffer, &single, 1) != 0) {
        buffer_free(&buffer);
        return NULL;
      }
    }
  }
  if (buffer.len > 0 && buffer.data[buffer.len - 1] == ' ') {
    buffer.data[buffer.len - 1] = '\0';
    buffer.len--;
  }
  return buffer.data != NULL ? buffer.data : pharos_http_log_strdup("");
}
 
char *request_real_ip_native(const char *headers) {
  char *copy = pharos_http_log_strdup(headers != NULL ? headers : "");
  char *cursor;
  if (copy == NULL) {
    return pharos_http_log_strdup("unknown");
  }
  cursor = copy;
  while (cursor != NULL && *cursor != '\0') {
    char *line_end = strchr(cursor, '\n');
    char *line = cursor;
    char *colon;
    if (line_end != NULL) {
      *line_end = '\0';
    }
    trim_in_place(line);
    colon = strchr(line, ':');
    if (colon != NULL) {
      char *value;
      *colon = '\0';
      value = colon + 1;
      trim_in_place(line);
      trim_in_place(value);
      if (pharos_http_log_strcasecmp(line, "X-Forwarded-For") == 0) {
        char *comma = strchr(value, ',');
        if (comma != NULL) {
          *comma = '\0';
          trim_in_place(value);
        }
        free(copy);
        return pharos_http_log_strdup(value[0] != '\0' ? value : "unknown");
      }
    }
    if (line_end == NULL) {
      break;
    }
    cursor = line_end + 1;
  }
  free(copy);
  copy = pharos_http_log_strdup(headers != NULL ? headers : "");
  if (copy == NULL) {
    return pharos_http_log_strdup("unknown");
  }
  cursor = copy;
  while (cursor != NULL && *cursor != '\0') {
    char *line_end = strchr(cursor, '\n');
    char *line = cursor;
    char *colon;
    if (line_end != NULL) {
      *line_end = '\0';
    }
    trim_in_place(line);
    colon = strchr(line, ':');
    if (colon != NULL) {
      char *value;
      *colon = '\0';
      value = colon + 1;
      trim_in_place(line);
      trim_in_place(value);
      if (pharos_http_log_strcasecmp(line, "X-Real-IP") == 0) {
        free(copy);
        return pharos_http_log_strdup(value[0] != '\0' ? value : "unknown");
      }
    }
    if (line_end == NULL) {
      break;
    }
    cursor = line_end + 1;
  }
  free(copy);
  return pharos_http_log_strdup("unknown");
}
 
char *request_headers_with_client_ip_native(const char *headers, const char *client_ip) {
  Buffer buffer;
  char *copy;
  char *cursor;
  int replaced = 0;
  buffer_init(&buffer);
  copy = pharos_http_log_strdup(headers != NULL ? headers : "");
  if (copy == NULL) {
    return NULL;
  }
  cursor = copy;
  while (cursor != NULL && *cursor != '\0') {
    char *line_end = strchr(cursor, '\n');
    char *line = cursor;
    char *colon;
    if (line_end != NULL) {
      *line_end = '\0';
    }
    trim_in_place(line);
    colon = strchr(line, ':');
    if (colon != NULL) {
      *colon = '\0';
      trim_in_place(line);
      trim_in_place(colon + 1);
      if (pharos_http_log_strcasecmp(line, "X-Real-IP") == 0) {
        if (buffer_append(&buffer, "X-Real-IP:", 10) != 0 ||
            buffer_append(&buffer, client_ip != NULL ? client_ip : "unknown", strlen(client_ip != NULL ? client_ip : "unknown")) != 0 ||
            buffer_append(&buffer, "\n", 1) != 0) {
          buffer_free(&buffer);
          free(copy);
          return NULL;
        }
        replaced = 1;
      } else {
        if (buffer_append(&buffer, line, strlen(line)) != 0 ||
            buffer_append(&buffer, ":", 1) != 0 ||
            buffer_append(&buffer, colon + 1, strlen(colon + 1)) != 0 ||
            buffer_append(&buffer, "\n", 1) != 0) {
          buffer_free(&buffer);
          free(copy);
          return NULL;
        }
      }
    }
    if (line_end == NULL) {
      break;
    }
    cursor = line_end + 1;
  }
  if (!replaced && client_ip != NULL && client_ip[0] != '\0') {
    if (buffer_append(&buffer, "X-Real-IP:", 10) != 0 ||
        buffer_append(&buffer, client_ip, strlen(client_ip)) != 0 ||
        buffer_append(&buffer, "\n", 1) != 0) {
      buffer_free(&buffer);
      free(copy);
      return NULL;
    }
  }
  free(copy);
  return buffer.data != NULL ? buffer.data : pharos_http_log_strdup("");
}
 
const char *http_status_text_native(int status_code) {
  switch (status_code) {
    case 200: return "200 OK";
    case 302: return "302 Found";
    case 400: return "400 Bad Request";
    case 404: return "404 Not Found";
    case 405: return "405 Method Not Allowed";
    case 500: return "500 Internal Server Error";
    default: return "200 OK";
  }
}
 
char *http_status_line_from_response_native(const char *response) {
  const char *line_end;
  const char *status_start;
  if (response == NULL || response[0] == '\0') {
    return pharos_http_log_strdup("500 Internal Server Error");
  }
  line_end = strstr(response, "\r\n");
  if (line_end == NULL) {
    line_end = strchr(response, '\n');
  }
  if (line_end == NULL) {
    line_end = response + strlen(response);
  }
  status_start = strchr(response, ' ');
  if (status_start != NULL && status_start < line_end) {
    status_start++;
    return dup_range(status_start, (size_t)(line_end - status_start));
  }
  return dup_range(response, (size_t)(line_end - response));
}
 
int http_status_code_from_response_native(const char *response) {
  const char *status_start;
  if (response == NULL || response[0] == '\0') {
    return 500;
  }
  status_start = strchr(response, ' ');
  if (status_start == NULL || status_start[1] == '\0') {
    return 500;
  }
  return atoi(status_start + 1);
}