Telodendria/tools/src/http-debug-server.c

172 lines
4 KiB
C

/*
* Copyright (C) 2022-2024 Jordan Bancino <@jordan:bancino.net> with
* other valuable contributors. See CONTRIBUTORS.txt for the full list.
*
* Permission is hereby granted, free of charge, to any person
* obtaining a copy of this software and associated documentation files
* (the "Software"), to deal in the Software without restriction,
* including without limitation the rights to use, copy, modify, merge,
* publish, distribute, sublicense, and/or sell copies of the Software,
* and to permit persons to whom the Software is furnished to do so,
* subject to the following conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
* BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
* ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
#include <stdio.h>
#include <signal.h>
#include <string.h>
#include <Telodendria.h>
#include <Cytoplasm/HttpServer.h>
#include <Cytoplasm/Util.h>
#include <Cytoplasm/Memory.h>
#include <Cytoplasm/Log.h>
#include <Cytoplasm/Db.h>
#include <Cytoplasm/Json.h>
static HttpServer *server = NULL;
static void
SignalHandle(int signal)
{
(void) signal;
HttpServerStop(server);
}
struct Args
{
Db *db;
HttpRouter *router;
HttpServerContext *cx;
};
static void *
TestFunc(Array * path, void *argp)
{
struct Args *args = argp;
HttpServerContext *cx = args->cx;
HashMap *headers = HttpRequestHeaders(cx);
HttpRequestMethod method = HttpRequestMethodGet(cx);
Db *db = args->db;
char *key;
char *val;
size_t bytes;
DbRef *ref;
(void) args;
(void) path;
Log(LOG_INFO, "%s %s", HttpRequestMethodToString(method),
HttpRequestPath(cx));
while (HashMapIterate(headers, &key, (void **) &val))
{
Log(LOG_INFO, "%s: %s", key, val);
}
Log(LOG_INFO, "");
bytes = StreamCopy(HttpServerStream(cx), StreamStdout());
StreamPutc(StreamStdout(), '\n');
Log(LOG_DEBUG, "(%lu bytes)", bytes);
ref = DbLock(db, 1, "test");
HttpSendHeaders(cx);
JsonEncode(DbJson(ref), HttpServerStream(cx), JSON_DEFAULT);
DbUnlock(db, ref);
return NULL;
}
void
HttpHandle(HttpServerContext * cx, void *argp)
{
struct Args *args = argp;
args->cx = cx;
HttpRouterRoute(args->router, HttpRequestPath(cx), args, NULL);
}
int
Main(void)
{
struct sigaction sa;
HttpServerConfig cfg;
struct Args args;
LogConfigLevelSet(LogConfigGlobal(), LOG_DEBUG);
Log(LOG_INFO, "Setting memory hook...");
MemoryHook(TelodendriaMemoryHook, NULL);
memset(&cfg, 0, sizeof(HttpServerConfig));
cfg.flags = HTTP_FLAG_NONE;
cfg.port = 8008;
cfg.threads = 1;
cfg.maxConnections = 1;
cfg.handler = HttpHandle;
cfg.handlerArgs = &args;
args.db = DbOpen("data", 0);
args.router = HttpRouterCreate();
HttpRouterAdd(args.router, "/test", TestFunc);
Log(LOG_DEBUG, "Creating server...");
server = HttpServerCreate(&cfg);
Log(LOG_DEBUG, "Starting server...");
if (!HttpServerStart(server))
{
Log(LOG_ERR, "Unable to start HTTP server.");
HttpServerFree(server);
return 1;
}
Log(LOG_INFO, "Listening on port 8008.");
sa.sa_handler = SignalHandle;
sigfillset(&sa.sa_mask);
sa.sa_flags = SA_RESTART;
if (sigaction(SIGINT, &sa, NULL) < 0)
{
Log(LOG_ERR, "Unable to install signal handler.");
HttpServerStop(server);
HttpServerJoin(server);
HttpServerFree(server);
return 1;
}
HttpServerJoin(server);
Log(LOG_INFO, "Shutting down.");
HttpServerStop(server);
return 0;
}