forked from Telodendria/Telodendria
Make the debug server more accurately mimic the actual server.
This commit is contained in:
parent
971f099bb6
commit
af15234799
1 changed files with 132 additions and 17 deletions
|
@ -23,10 +23,80 @@
|
||||||
*/
|
*/
|
||||||
#include <stdio.h>
|
#include <stdio.h>
|
||||||
#include <signal.h>
|
#include <signal.h>
|
||||||
|
#include <string.h>
|
||||||
|
|
||||||
|
#include <Telodendria.h>
|
||||||
#include <HttpServer.h>
|
#include <HttpServer.h>
|
||||||
#include <Util.h>
|
#include <Util.h>
|
||||||
|
|
||||||
|
#include <Memory.h>
|
||||||
|
#include <Log.h>
|
||||||
|
#include <Db.h>
|
||||||
|
#include <Json.h>
|
||||||
|
|
||||||
|
static void
|
||||||
|
HexDump(size_t off, char *hexBuf , char *asciiBuf, void *args)
|
||||||
|
{
|
||||||
|
char *fmt;
|
||||||
|
if (hexBuf && asciiBuf)
|
||||||
|
{
|
||||||
|
fmt = "%04lx: | %s | %s |";
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
fmt = "%04lx";
|
||||||
|
}
|
||||||
|
|
||||||
|
Log(LOG_DEBUG, fmt, off, hexBuf, asciiBuf);
|
||||||
|
}
|
||||||
|
|
||||||
|
void
|
||||||
|
TelodendriaMemoryHook(MemoryAction a, MemoryInfo * i, void *args)
|
||||||
|
{
|
||||||
|
char *action;
|
||||||
|
int err = 0;
|
||||||
|
|
||||||
|
switch (a)
|
||||||
|
{
|
||||||
|
case MEMORY_ALLOCATE:
|
||||||
|
action = "Allocated";
|
||||||
|
break;
|
||||||
|
case MEMORY_REALLOCATE:
|
||||||
|
action = "Re-allocated";
|
||||||
|
break;
|
||||||
|
case MEMORY_FREE:
|
||||||
|
action = "Freed";
|
||||||
|
break;
|
||||||
|
case MEMORY_BAD_POINTER:
|
||||||
|
err = 1;
|
||||||
|
action = "Bad pointer to";
|
||||||
|
break;
|
||||||
|
case MEMORY_CORRUPTED:
|
||||||
|
err = 1;
|
||||||
|
action = "Corrupted block of";
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
action = "Unknown operation on";
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
Log(err ? LOG_ERR : LOG_DEBUG,
|
||||||
|
"%s:%d: %s %lu bytes of memory at %p.",
|
||||||
|
MemoryInfoGetFile(i), MemoryInfoGetLine(i),
|
||||||
|
action, MemoryInfoGetSize(i),
|
||||||
|
MemoryInfoGetPointer(i));
|
||||||
|
|
||||||
|
if (a != MEMORY_ALLOCATE && a != MEMORY_REALLOCATE)
|
||||||
|
{
|
||||||
|
MemoryHexDump(i, HexDump, NULL);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (err)
|
||||||
|
{
|
||||||
|
raise(SIGINT);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
static HttpServer *server = NULL;
|
static HttpServer *server = NULL;
|
||||||
|
|
||||||
static void
|
static void
|
||||||
|
@ -35,38 +105,66 @@ SignalHandle(int signal)
|
||||||
(void) signal;
|
(void) signal;
|
||||||
HttpServerStop(server);
|
HttpServerStop(server);
|
||||||
}
|
}
|
||||||
|
struct Args
|
||||||
static void
|
|
||||||
HttpHandle(HttpServerContext * cx, void *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);
|
HashMap *headers = HttpRequestHeaders(cx);
|
||||||
HttpRequestMethod method = HttpRequestMethodGet(cx);
|
HttpRequestMethod method = HttpRequestMethodGet(cx);
|
||||||
|
Db *db = args->db;
|
||||||
|
|
||||||
char *key;
|
char *key;
|
||||||
char *val;
|
char *val;
|
||||||
|
|
||||||
size_t bytes;
|
size_t bytes;
|
||||||
|
|
||||||
|
DbRef *ref;
|
||||||
|
|
||||||
(void) args;
|
(void) args;
|
||||||
|
|
||||||
StreamPrintf(StreamStdout(), "%s %s\n", HttpRequestMethodToString(method),
|
Log(LOG_INFO, "%s %s", HttpRequestMethodToString(method),
|
||||||
HttpRequestPath(cx));
|
HttpRequestPath(cx));
|
||||||
|
|
||||||
while (HashMapIterate(headers, &key, (void **) &val))
|
while (HashMapIterate(headers, &key, (void **) &val))
|
||||||
{
|
{
|
||||||
StreamPrintf(StreamStdout(), "%s: %s\n", key, val);
|
Log(LOG_INFO, "%s: %s", key, val);
|
||||||
}
|
}
|
||||||
|
|
||||||
StreamPutc(StreamStdout(), '\n');
|
Log(LOG_INFO, "");
|
||||||
|
|
||||||
bytes = StreamCopy(HttpServerStream(cx), StreamStdout());
|
bytes = StreamCopy(HttpServerStream(cx), StreamStdout());
|
||||||
|
|
||||||
StreamPutc(StreamStdout(), '\n');
|
StreamPutc(StreamStdout(), '\n');
|
||||||
StreamPrintf(StreamStdout(), "(%lu bytes)\n", bytes);
|
|
||||||
|
Log(LOG_DEBUG, "(%lu bytes)", bytes);
|
||||||
|
|
||||||
|
ref = DbLock(db, 1, "test");
|
||||||
|
|
||||||
|
|
||||||
HttpSendHeaders(cx);
|
HttpSendHeaders(cx);
|
||||||
|
|
||||||
StreamPuts(HttpServerStream(cx), "{}\n");
|
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
|
int
|
||||||
|
@ -75,22 +173,43 @@ Main(void)
|
||||||
struct sigaction sa;
|
struct sigaction sa;
|
||||||
HttpServerConfig cfg;
|
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.flags = HTTP_FLAG_NONE;
|
||||||
cfg.port = 8008;
|
cfg.port = 8008;
|
||||||
cfg.threads = 1;
|
cfg.threads = 1;
|
||||||
cfg.maxConnections = 1;
|
cfg.maxConnections = 1;
|
||||||
cfg.handler = HttpHandle;
|
cfg.handler = HttpHandle;
|
||||||
cfg.handlerArgs = NULL;
|
|
||||||
|
cfg.handlerArgs = &args;
|
||||||
|
|
||||||
|
args.db = DbOpen("data", 0);
|
||||||
|
args.router = HttpRouterCreate();
|
||||||
|
|
||||||
|
HttpRouterAdd(args.router, "/test", TestFunc);
|
||||||
|
|
||||||
|
Log(LOG_DEBUG, "Creating server...");
|
||||||
|
|
||||||
server = HttpServerCreate(&cfg);
|
server = HttpServerCreate(&cfg);
|
||||||
|
|
||||||
|
Log(LOG_DEBUG, "Starting server...");
|
||||||
|
|
||||||
if (!HttpServerStart(server))
|
if (!HttpServerStart(server))
|
||||||
{
|
{
|
||||||
StreamPuts(StreamStderr(), "Unable to start HTTP server.\n");
|
Log(LOG_ERR, "Unable to start HTTP server.");
|
||||||
HttpServerFree(server);
|
HttpServerFree(server);
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
StreamPuts(StreamStdout(), "Listening on port 8008.\n");
|
Log(LOG_INFO, "Listening on port 8008.");
|
||||||
|
|
||||||
sa.sa_handler = SignalHandle;
|
sa.sa_handler = SignalHandle;
|
||||||
sigfillset(&sa.sa_mask);
|
sigfillset(&sa.sa_mask);
|
||||||
|
@ -98,7 +217,7 @@ Main(void)
|
||||||
|
|
||||||
if (sigaction(SIGINT, &sa, NULL) < 0)
|
if (sigaction(SIGINT, &sa, NULL) < 0)
|
||||||
{
|
{
|
||||||
StreamPuts(StreamStderr(), "Unable to install signal handler.\n");
|
Log(LOG_ERR, "Unable to install signal handler.");
|
||||||
HttpServerStop(server);
|
HttpServerStop(server);
|
||||||
HttpServerJoin(server);
|
HttpServerJoin(server);
|
||||||
HttpServerFree(server);
|
HttpServerFree(server);
|
||||||
|
@ -107,12 +226,8 @@ Main(void)
|
||||||
|
|
||||||
HttpServerJoin(server);
|
HttpServerJoin(server);
|
||||||
|
|
||||||
StreamPuts(StreamStdout(), "Shutting down.\n");
|
Log(LOG_INFO, "Shutting down.");
|
||||||
HttpServerStop(server);
|
HttpServerStop(server);
|
||||||
|
|
||||||
StreamClose(StreamStdout());
|
|
||||||
StreamClose(StreamStderr());
|
|
||||||
StreamClose(StreamStdin());
|
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue