Refactor HTTP client

This commit is contained in:
Sergey Lyubka
2025-08-12 09:21:38 +01:00
parent 9f76ea8030
commit de0a3b035a
2 changed files with 33 additions and 22 deletions
+2 -13
View File
@@ -2,11 +2,11 @@ PROG ?= example # Program we are building
PACK ?= ./pack # Packing executable
DELETE = rm -rf # Command to remove files
OUT ?= -o $(PROG) # Compiler argument for output file
SOURCES = main.c mongoose.c packed_fs.c # Source code files, packed_fs.c contains ca.pem, which contains CA certs for TLS
SOURCES = main.c mongoose.c # Source code files
CFLAGS = -W -Wall -Wextra -g -I. # Build options
# Mongoose build options. See https://mongoose.ws/documentation/#build-options
CFLAGS_MONGOOSE += -DMG_ENABLE_LINES=1 -DMG_ENABLE_PACKED_FS=1
CFLAGS_MONGOOSE += -DMG_ENABLE_LINES=1
CFLAGS_EXTRA ?= -DMG_TLS=MG_TLS_BUILTIN
ifeq ($(OS),Windows_NT) # Windows settings. Assume MinGW compiler. To use VC: make CC=cl CFLAGS=/MD OUT=/Feprog.exe
@@ -27,14 +27,3 @@ $(PROG): $(SOURCES) # Build program from sources
clean: # Cleanup. Delete built program and all build artifacts
$(DELETE) $(PROG) *.o *.obj *.exe *.dSYM mbedtls $(PACK)
# Generate packed filesystem for serving cert
packed_fs.c: $(wildcard certs/*) Makefile
$(CC) ../../../test/pack.c -o $(PACK)
$(PACK) $(wildcard certs/*) > $@
# see https://mongoose.ws/tutorials/tls/#how-to-build for TLS build options
mbedtls: # Pull and build mbedTLS library
git clone --depth 1 -b v2.28.2 https://github.com/mbed-tls/mbedtls $@
$(MAKE) -C mbedtls/library
+31 -9
View File
@@ -11,6 +11,7 @@
// The very first web page in history. You can replace it from command line
static const char *s_url = "http://info.cern.ch/";
static struct mg_str s_ca_pem; // CA PEM file
static const char *s_post_data = NULL; // POST data
static const uint64_t s_timeout_ms = 1500; // Connect timeout in milliseconds
@@ -29,7 +30,7 @@ static void fn(struct mg_connection *c, int ev, void *ev_data) {
struct mg_str host = mg_url_host(s_url);
if (c->is_tls) {
struct mg_tls_opts opts = {.ca = mg_unpacked("/certs/ca.pem"),
struct mg_tls_opts opts = {.ca = s_ca_pem,
.name = mg_url_host(s_url)};
mg_tls_init(c, &opts);
}
@@ -49,7 +50,7 @@ static void fn(struct mg_connection *c, int ev, void *ev_data) {
// Response is received. Print it
struct mg_http_message *hm = (struct mg_http_message *) ev_data;
printf("%.*s", (int) hm->message.len, hm->message.buf);
c->is_draining = 1; // Tell mongoose to close this connection
c->is_closing = 1; // Tell mongoose to close this connection
*(bool *) c->fn_data = true; // Tell event loop to stop
} else if (ev == MG_EV_ERROR) {
*(bool *) c->fn_data = true; // Error, tell event loop to stop
@@ -57,16 +58,37 @@ static void fn(struct mg_connection *c, int ev, void *ev_data) {
}
int main(int argc, char *argv[]) {
const char *log_level = getenv("LOG_LEVEL"); // Allow user to set log level
if (log_level == NULL) log_level = "4"; // Default is verbose
struct mg_mgr mgr; // Event manager
bool done = false; // Event handler flips it to true
int i, log_level = MG_LL_DEBUG;
struct mg_mgr mgr; // Event manager
bool done = false; // Event handler flips it to true
if (argc > 1) s_url = argv[1]; // Use URL provided in the command line
mg_log_set(atoi(log_level)); // Set to 0 to disable debug
mg_mgr_init(&mgr); // Initialise event manager
// Parse command-line flags
for (i = 1; i + 1 < argc; i++) {
if (strcmp(argv[i], "-ca") == 0) {
s_ca_pem = mg_file_read(&mg_fs_posix, argv[++i]);
} else if (strcmp(argv[i], "-post") == 0) {
s_post_data = argv[++i];
} else if (strcmp(argv[i], "-url") == 0) {
s_url = argv[++i];
} else if (strcmp(argv[i], "-v") == 0) {
log_level = atoi(argv[++i]);
} else {
fprintf(stderr,
"Usage: %s OPTIONS\n"
" -ca PEM - TLS CA PEM file path, default: not set\n"
" -post DATA - data to POST, default: not set\n"
" -url URL - URL to fetch, default: %s\n"
" -v LEVEL - debug level, from 0 to 4, default: %d\n",
argv[0], s_url, log_level);
exit(EXIT_FAILURE);
}
}
mg_mgr_init(&mgr); // Initialise event manager
mg_log_set(log_level); // Set log level
mg_http_connect(&mgr, s_url, fn, &done); // Create client connection
while (!done) mg_mgr_poll(&mgr, 50); // Event manager loops until 'done'
mg_mgr_free(&mgr); // Free resources
return 0;
}