native_static_serve.c

Go to the documentation of this file.

/**
 * @file
 * @brief Implements native static artifact serving helpers.
 */
 
#include "native_static_serve.h"
 
#include "native_dispatch_table.h"
#include "native_http_access_log.h"
#include "native_http_log.h"
#include "native_http_response.h"
#include "native_support.h"
 
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
 
#ifdef _WIN32
#define native_static_serve_strdup _strdup
#else
#define native_static_serve_strdup strdup
#endif
 
int native_serve_static_file(NativeConnection *connection, const char *root, const char *path, const HttpRequest *request, const char *log_path) {
  char *relative = NULL;
  char *file_path = NULL;
  char *content_type = NULL;
  FILE *file = NULL;
  int status = 500;
  const char *path_part = path;
  const char *query = strchr(path, '?');
  size_t path_len = query != NULL ? (size_t)(query - path) : strlen(path);
  long long content_length = 0;
  if (strstr(path, "..") != NULL) {
    native_send_text_response(connection, 400, "Bad Request", "blocked path traversal\n");
    status = 400;
    goto cleanup;
  }
  relative = native_dispatch_response_file_for_request(root, request != NULL && request->method != NULL ? request->method : "GET", path_part);
  if (relative == NULL) {
    if (path_len == 0 || (path_len == 1 && path[0] == '/')) {
      relative = native_static_serve_strdup("index.html");
    } else if (path[path_len - 1] == '/') {
      relative = (char *)malloc(path_len + strlen("index.html"));
      if (relative != NULL) {
        memcpy(relative, path_part + 1, path_len - 1);
        memcpy(relative + path_len - 1, "index.html", strlen("index.html") + 1);
      }
    } else {
      relative = dup_range(path_part + 1, path_len - 1);
    }
  }
  if (relative == NULL) {
    native_send_text_response(connection, 500, "Internal Server Error", "memory allocation failed\n");
    status = 500;
    goto cleanup;
  }
  file_path = path_join(root, relative);
  if (file_path == NULL) {
    native_send_text_response(connection, 500, "Internal Server Error", "memory allocation failed\n");
    status = 500;
    goto cleanup;
  }
  if (path_is_directory(file_path)) {
    char *dir_index = path_join(file_path, "index.html");
    free(file_path);
    file_path = dir_index;
  }
  if (file_path == NULL || !path_exists(file_path)) {
    native_send_text_response(connection, 404, "Not Found", "Not Found");
    status = 404;
    goto cleanup;
  }
  file = fopen(file_path, "rb");
  if (file == NULL) {
    native_send_text_response(connection, 404, "Not Found", "Not Found");
    status = 404;
    goto cleanup;
  }
  if (fseek(file, 0, SEEK_END) != 0) {
    native_send_text_response(connection, 500, "Internal Server Error", "failed to read static artifact\n");
    status = 500;
    goto cleanup;
  }
  content_length = (long long)ftell(file);
  if (content_length < 0 || fseek(file, 0, SEEK_SET) != 0) {
    native_send_text_response(connection, 500, "Internal Server Error", "failed to read static artifact\n");
    status = 500;
    goto cleanup;
  }
  content_type = mime_type_for_path(file_path);
  if (content_type == NULL) {
    native_send_text_response(connection, 500, "Internal Server Error", "mime type failure\n");
    status = 500;
    goto cleanup;
  }
  if (starts_with(path_part, "/api/") && strcmp(content_type, "application/octet-stream") == 0) {
    free(content_type);
    content_type = native_static_serve_strdup("application/json");
    if (content_type == NULL) {
      native_send_text_response(connection, 500, "Internal Server Error", "mime type failure\n");
      status = 500;
      goto cleanup;
    }
  }
  if (native_connection_printf(
        connection,
        "HTTP/1.1 200 OK\r\n"
        "Content-Type: %s\r\n"
        "Content-Length: %lld\r\n"
        "Connection: close\r\n"
        "\r\n",
        content_type,
        content_length
      ) != 0) {
    status = 500;
    goto cleanup;
  }
  {
    char chunk[4096];
    size_t bytes_read;
    while ((bytes_read = fread(chunk, 1, sizeof(chunk), file)) > 0) {
      if (native_connection_write(connection, chunk, bytes_read) != 0) {
        status = -1;
        goto cleanup;
      }
    }
  }
  status = 200;
 
cleanup:
  if (status >= 200 && status < 400) {
    native_write_access_log(log_path, request, path, status);
  } else if (status > 0) {
    native_write_error_log(log_path, NULL, "app", request, path, http_status_text_native(status), "http_response", path, NULL);
  }
  free(relative);
  free(file_path);
  free(content_type);
  if (file != NULL) {
    fclose(file);
  }
  return status;
}