libweb/examples/hello/main.c

118 lines
2.6 KiB
C

#include <dynstr.h>
#include <libweb/handler.h>
#include <libweb/html.h>
#include <libweb/http.h>
#include <stddef.h>
#include <stdio.h>
#include <stdlib.h>
static int hello(const struct http_payload *const pl,
struct http_response *const r, void *const user)
{
int ret = -1;
struct html_node *html = html_node_alloc("html"), *body, *p;
struct dynstr d;
dynstr_init(&d);
if (!html)
{
fprintf(stderr, "%s: html_node_alloc failed\n", __func__);
goto end;
}
else if (!(body = html_node_add_child(html, "body")))
{
fprintf(stderr, "%s: html_node_add_child body failed\n", __func__);
goto end;
}
else if (!(p = html_node_add_child(html, "p")))
{
fprintf(stderr, "%s: html_node_add_child p failed\n", __func__);
goto end;
}
else if (html_node_set_value(p, "Hello from libweb!"))
{
fprintf(stderr, "%s: html_node_set_value p failed\n", __func__);
goto end;
}
else if (html_serialize(html, &d))
{
fprintf(stderr, "%s: html_serialize failed\n", __func__);
goto end;
}
*r = (const struct http_response)
{
.status = HTTP_STATUS_OK,
.buf.rw = d.str,
.n = d.len,
.free = free
};
if (http_response_add_header(r, "Content-Type", "text/html"))
{
fprintf(stderr, "%s: http_response_add_header failed\n", __func__);
goto end;
}
ret = 0;
end:
html_node_free(html);
if (ret)
dynstr_free(&d);
return ret;
}
static int on_length(const unsigned long long len,
const struct http_cookie *const c, struct http_response *const r,
void *const user)
{
*r = (const struct http_response)
{
.status = HTTP_STATUS_FORBIDDEN
};
return 1;
}
int main(int argc, char *argv[])
{
int ret = EXIT_FAILURE;
const short port = 8080;
const struct handler_cfg cfg =
{
.length = on_length
};
struct handler *const h = handler_alloc(&cfg);
static const char *const urls[] = {"/", "/index.html"};
if (!h)
{
fprintf(stderr, "%s: handler_alloc failed\n", __func__);
goto end;
}
for (size_t i = 0; i < sizeof urls / sizeof *urls; i++)
if (handler_add(h, urls[i], HTTP_OP_GET, hello, NULL))
{
fprintf(stderr, "%s: handler_add failed\n", __func__);
goto end;
}
if (handler_listen(h, port))
{
fprintf(stderr, "%s: handler_listen failed\n", __func__);
goto end;
}
ret = EXIT_SUCCESS;
end:
handler_free(h);
return ret;
}