Argument parsing
This commit is contained in:
@@ -0,0 +1,3 @@
|
||||
httpserver
|
||||
compile_commands.json
|
||||
bin/
|
||||
@@ -0,0 +1,22 @@
|
||||
CC=gcc
|
||||
CFLAGS=-Wall -Wextra -Wpedantic -std=c23 -D_POSIX_C_SOURCE=200112L -O2
|
||||
CFLAGS+=-Og -g -fsanitize=address,undefined
|
||||
SRCS=main.c threadpool.c util.c
|
||||
|
||||
OBJS=$(SRCS:%.c=bin/%.o)
|
||||
|
||||
all: httpserver
|
||||
|
||||
httpserver: $(OBJS)
|
||||
$(CC) $(CFLAGS) -o $@ $^
|
||||
|
||||
bin/%.o: src/%.c
|
||||
@mkdir -p bin/deps/
|
||||
$(CC) $(CFLAGS) -MD -MF $(patsubst src/%.c,bin/deps/%.d,$^) -c -o $@ $^
|
||||
|
||||
include $(SRCS:%.c/bin/deps/%.d)
|
||||
|
||||
clean:
|
||||
rm -rf bin/ httpserver
|
||||
|
||||
.PHONY: all clean
|
||||
+110
@@ -0,0 +1,110 @@
|
||||
#include "util.h"
|
||||
|
||||
#include <ctype.h>
|
||||
#include <errno.h>
|
||||
#include <limits.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <unistd.h>
|
||||
|
||||
typedef struct {
|
||||
bool opt_error;
|
||||
bool help_flag;
|
||||
int port;
|
||||
int parallelism;
|
||||
} GlobalFlags;
|
||||
|
||||
static const GlobalFlags DEFAULT_FLAGS = {
|
||||
.opt_error = false,
|
||||
.help_flag = false,
|
||||
.port = 0,
|
||||
.parallelism = 1,
|
||||
};
|
||||
|
||||
// Return false on failure, true on success
|
||||
static bool parse_int(const char *str, int min, int max, int *output) {
|
||||
errno = 0;
|
||||
char *endptr;
|
||||
long conv = strtol(str, &endptr, 10);
|
||||
if (!*str || *endptr) {
|
||||
log_error("malformed number: \"%s\"", str);
|
||||
return false;
|
||||
} else if (((conv == LONG_MIN || conv == LONG_MAX) && errno == ERANGE)
|
||||
|| conv < min || conv > max) {
|
||||
log_error("out of range: %s", str);
|
||||
return false;
|
||||
}
|
||||
*output = conv;
|
||||
return true;
|
||||
}
|
||||
|
||||
void parse_cli_options(int argc, const char **argv, GlobalFlags *flags) {
|
||||
constexpr unsigned long MAX_PARALLELISM = INT_MAX;
|
||||
constexpr unsigned long MAX_PORT = 65535;
|
||||
opterr = false;
|
||||
int c;
|
||||
char pretty_flag[8];
|
||||
while ((c = getopt(argc, (char *const *) argv, "hp:")) >= 0) {
|
||||
if (isprint(c)) {
|
||||
snprintf(pretty_flag, sizeof(pretty_flag), "%c", c);
|
||||
} else {
|
||||
#pragma GCC diagnostic push
|
||||
#pragma GCC diagnostic ignored "-Wformat-truncation"
|
||||
// `c` is actually in the range [0,255], and so will be at most 2
|
||||
// hex digits
|
||||
snprintf(pretty_flag, sizeof(pretty_flag), "0x%02x", c);
|
||||
#pragma GCC diagnostic pop
|
||||
}
|
||||
switch (c) {
|
||||
case 'h':
|
||||
flags->help_flag = true;
|
||||
return;
|
||||
case 'p':
|
||||
if (!parse_int(optarg, 1, MAX_PARALLELISM, &flags->parallelism)) {
|
||||
flags->opt_error = true;
|
||||
}
|
||||
break;
|
||||
case ':':
|
||||
flags->opt_error = true;
|
||||
log_error("flag requires argument: '%s'", pretty_flag);
|
||||
break;
|
||||
case '?':
|
||||
flags->opt_error = true;
|
||||
log_error("unknown flag: '%s'", pretty_flag);
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (optind == argc) {
|
||||
flags->opt_error = true;
|
||||
log_error("no port provided");
|
||||
} else if (optind != argc - 1) {
|
||||
flags->opt_error = true;
|
||||
log_error("extra positional arguments after port");
|
||||
} else if (!parse_int(argv[optind], 1, MAX_PORT, &flags->port)) {
|
||||
flags->opt_error = true;
|
||||
}
|
||||
}
|
||||
|
||||
void print_help(FILE *file) {
|
||||
fprintf(file, "usage: httpserver [-h] [-p PARALLELISM] <PORT>\n");
|
||||
fprintf(file, " -h print this message, then exit\n");
|
||||
fprintf(
|
||||
file,
|
||||
" -p use PARALLELISM threads for processing requests (default: 1)\n");
|
||||
}
|
||||
|
||||
int main(int argc, const char **argv) {
|
||||
setenv("POSIXLY_CORRECT", "1", true);
|
||||
set_exec_name(argv[0]);
|
||||
GlobalFlags flags = DEFAULT_FLAGS;
|
||||
parse_cli_options(argc, argv, &flags);
|
||||
if (flags.help_flag) {
|
||||
print_help(stdout);
|
||||
return flags.opt_error ? EXIT_FAILURE : 0;
|
||||
} else if (flags.opt_error) {
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
printf("Port: %d\n", flags.port);
|
||||
printf("Parallelism: %d\n", flags.parallelism);
|
||||
return 0;
|
||||
}
|
||||
@@ -0,0 +1,26 @@
|
||||
#include "threadpool.h"
|
||||
|
||||
#include <stdlib.h>
|
||||
#include <threads.h>
|
||||
|
||||
struct thread_pool_queue {
|
||||
Task task;
|
||||
void *arg;
|
||||
struct thread_pool_queue *next;
|
||||
};
|
||||
|
||||
struct _ThreadPool {
|
||||
size_t nthreads;
|
||||
thrd_t *threads;
|
||||
|
||||
struct thread_pool_queue *queue;
|
||||
};
|
||||
|
||||
ThreadPool *make_thread_pool(int parallelism) {
|
||||
ThreadPool *pool = malloc(sizeof(ThreadPool));
|
||||
pool->nthreads = parallelism;
|
||||
}
|
||||
|
||||
void destroy_thread_pool(ThreadPool *pool) {}
|
||||
|
||||
void thread_pool_enqueue(Task task, void *arg) {}
|
||||
@@ -0,0 +1,13 @@
|
||||
#ifndef INCLUDED_THREAD_POOL_H
|
||||
#define INCLUDED_THREAD_POOL_H
|
||||
|
||||
typedef struct _ThreadPool ThreadPool;
|
||||
typedef void (*Task)(void *);
|
||||
|
||||
ThreadPool *make_thread_pool(int parallelism);
|
||||
|
||||
void destroy_thread_pool(ThreadPool *pool);
|
||||
|
||||
void thread_pool_enqueue(Task task, void *arg);
|
||||
|
||||
#endif
|
||||
+35
@@ -0,0 +1,35 @@
|
||||
#include "util.h"
|
||||
|
||||
#include <stdarg.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
#define DEFAULT_EXEC_NAME "httpserver"
|
||||
|
||||
const char *EXEC_NAME = DEFAULT_EXEC_NAME;
|
||||
static bool need_free_out_of_memory_msg = false;
|
||||
NO_SANITIZE
|
||||
static const char *out_of_memory_msg =
|
||||
DEFAULT_EXEC_NAME ": error: out of memory\n";
|
||||
|
||||
void set_exec_name(const char *argv0) {
|
||||
const char *slash = strrchr(argv0, '/');
|
||||
if (!slash) {
|
||||
EXEC_NAME = argv0;
|
||||
} else {
|
||||
EXEC_NAME = slash + 1;
|
||||
}
|
||||
if (need_free_out_of_memory_msg) {
|
||||
free((char *) out_of_memory_msg);
|
||||
}
|
||||
}
|
||||
|
||||
void log_error(const char *fmt, ...) {
|
||||
fprintf(stderr, "%s: error: ", EXEC_NAME);
|
||||
va_list args;
|
||||
va_start(args, fmt);
|
||||
vfprintf(stderr, fmt, args);
|
||||
va_end(args);
|
||||
fputc('\n', stderr);
|
||||
}
|
||||
+23
@@ -0,0 +1,23 @@
|
||||
#ifndef INCLUDED_UTIL_H
|
||||
#define INCLUDED_UTIL_H
|
||||
|
||||
#if __has_attribute(format)
|
||||
# define PRINTF_LIKE(i, j) __attribute__((format(printf, i, j)))
|
||||
#else
|
||||
# define PRINTF_LIKE(i, j)
|
||||
#endif
|
||||
|
||||
#if __has_attribute(no_sanitize_address)
|
||||
# define NO_SANITIZE __attribute__((no_sanitize_address))
|
||||
#else
|
||||
# define NO_SANITIZE
|
||||
#endif
|
||||
|
||||
extern const char *EXEC_NAME;
|
||||
|
||||
void set_exec_name(const char *argv0);
|
||||
|
||||
PRINTF_LIKE(1, 2)
|
||||
void log_error(const char *fmt, ...);
|
||||
|
||||
#endif
|
||||
Reference in New Issue
Block a user