hio/bin/webs.c

384 lines
8.4 KiB
C
Raw Normal View History

#include <hio-http.h>
2022-10-25 22:47:40 +00:00
#include <hio-tar.h>
#include <hio-opt.h>
2022-12-26 17:03:53 +00:00
#include <hio-prv.h>
#include <signal.h>
#include <string.h>
#include <stdio.h>
2022-10-25 22:47:40 +00:00
#include <unistd.h>
#include <dirent.h>
#include <stdlib.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/stat.h>
struct arg_info_t
{
const char* laddrs;
const char* docroot;
int file_list_dir;
int file_load_index_page;
};
typedef struct arg_info_t arg_info_t;
struct htts_ext_t
{
2022-12-31 15:52:38 +00:00
const arg_info_t* ai;
};
typedef struct htts_ext_t htts_ext_t;
2022-10-25 22:47:40 +00:00
void untar (hio_t* hio, hio_dev_thr_iopair_t* iop, hio_svc_htts_thr_func_info_t* tfi, void* ctx)
{
FILE* wfp = HIO_NULL;
2022-10-25 22:47:40 +00:00
htts_ext_t* ext;
hio_tar_t* tar = HIO_NULL;
hio_uint8_t buf[4096];
ssize_t n;
2022-10-25 22:47:40 +00:00
ext = hio_svc_htts_getxtn(tfi->htts);
2022-12-26 17:03:53 +00:00
/* TODO: error handling on write() failure */
2022-10-25 22:47:40 +00:00
wfp = fdopen(iop->wfd, "w");
if (!wfp)
{
write (iop->wfd, "Status: 500\r\n\r\n", 15); /* internal server error */
goto done;
}
tar = hio_tar_open(hio, 0);
if (!tar)
2022-10-25 22:47:40 +00:00
{
write (iop->wfd, "Status: 500\r\n\r\n", 15);
2022-10-25 22:47:40 +00:00
goto done;
}
hio_tar_setxrootwithbcstr (tar, ext->ai->docroot);
while (1)
{
n = read(iop->rfd, buf, HIO_COUNTOF(buf));
if (n <= 0) break; /* eof or error */
if (hio_tar_xfeed(tar, buf, n) <= -1)
{
write (iop->wfd, "Status: 500\r\n\r\n", 20);
goto done;
}
}
hio_tar_endxfeed (tar);
2022-10-25 22:47:40 +00:00
write (iop->wfd, "Status: 200\r\n\r\n", 15);
done:
if (tar)
{
hio_tar_close (tar);
}
2022-10-25 22:47:40 +00:00
if (wfp)
{
iop->wfd = HIO_SYSHND_INVALID; /* prevent double close by the main library since this function closes it with fclose() */
2022-10-25 22:47:40 +00:00
fclose (wfp);
}
}
static const hio_bch_t* file_get_mime_type (hio_svc_htts_t* htts, const hio_bch_t* qpath, const hio_bch_t* file_path, void* ctx)
{
const hio_bch_t* mt = HIO_NULL;
const hio_bch_t* dot;
dot = hio_rfind_bchar_in_bcstr(file_path, '.');
if (dot) mt = hio_get_mime_type_by_ext(dot + 1);
return mt;
}
2022-12-26 17:03:53 +00:00
static int file_open_dir_list (hio_svc_htts_t* htts, const hio_bch_t* qpath, const hio_bch_t* dir_path, const hio_bch_t** res_mime_type, void* ctx)
{
htts_ext_t* ext = hio_svc_htts_getxtn(htts);
hio_t* hio = hio_svc_htts_gethio(htts);
DIR* dp = HIO_NULL;
hio_bch_t file_path[] = "/tmp/.XXXXXX";
int fd = -1;
struct dirent* de;
if (ext->ai->file_load_index_page)
{
2022-12-31 15:52:38 +00:00
hio_bch_t* index_path;
2022-12-26 17:03:53 +00:00
index_path = hio_svc_htts_dupmergepaths(htts, dir_path, "index.html");
if (HIO_UNLIKELY(!index_path)) goto oops;
fd = open(index_path, O_RDONLY, 0644);
if (fd >= 0)
{
if (res_mime_type)
{
const hio_bch_t* mt;
mt = file_get_mime_type(htts, qpath, index_path, ctx);
if (mt) *res_mime_type = mt;
}
hio_freemem (hio, index_path);
return fd;
}
hio_freemem (hio, index_path);
}
if (!ext->ai->file_list_dir) goto oops;
dp = opendir(dir_path);
if (!dp) goto oops;
/* TOOD: mkostemp instead and specify O_CLOEXEC and O_LARGEFILE? */
fd = mkstemp(file_path);
if (fd <= -1) goto oops;
unlink (file_path);
write (fd, "<html><body>", 12);
2022-12-31 15:52:38 +00:00
if (!(qpath[0] == '\0' || (qpath[0] == '/' && qpath[1] == '\0')))
write (fd, "<li><a href=\"..\">..</a>", 23);
/* TODO: sorting, other informatino like size, */
/* TODO: error handling of write() error */
while ((de = readdir(dp)))
{
struct stat st;
2022-12-31 15:52:38 +00:00
hio_bch_t* tmp_path;
int n;
if ((de->d_name[0] == '.' && de->d_name[1] == '\0') ||
(de->d_name[0] == '.' && de->d_name[1] == '.' && de->d_name[2] == '\0')) continue;
2022-12-31 15:52:38 +00:00
tmp_path = hio_svc_htts_dupmergepaths(htts, dir_path, de->d_name);
if (HIO_UNLIKELY(!tmp_path)) continue;
n = stat(tmp_path, &st);
hio_freemem (hio, tmp_path);
if (HIO_UNLIKELY(n <= -1)) continue;
write (fd, "<li><a href=\"", 13);
2022-12-31 15:52:38 +00:00
{
char tmp[1000]; /* TODO:use dynamic buffer?? */
hio_perenc_http_bcstr(0, de->d_name, tmp, HIO_NULL); /* url encoding */
//write (fd, de->d_name, strlen(de->d_name));
write (fd, tmp, strlen(tmp));
}
if (S_ISDIR(st.st_mode)) write (fd, "/", 1);
write (fd, "\">", 2);
2022-12-31 15:52:38 +00:00
write (fd, de->d_name, strlen(de->d_name)); /* TODO: html entity encoding */
if (S_ISDIR(st.st_mode)) write (fd, "/", 1);
write (fd, "</a>", 4);
}
write (fd, "</body></html>\n", 15);
closedir (dp);
lseek (fd, SEEK_SET, 0);
done:
if (res_mime_type) *res_mime_type = "text/html";
return fd;
oops:
if (fd >= 0) close (fd);
if (dp) closedir (dp);
return -1;
}
static int process_http_request (hio_svc_htts_t* htts, hio_dev_sck_t* csck, hio_htre_t* req)
{
htts_ext_t* ext = hio_svc_htts_getxtn(htts);
hio_t* hio = hio_svc_htts_gethio(htts);
hio_http_method_t mth;
const hio_bch_t* qpath;
static hio_svc_htts_file_cbs_t fcbs = { file_get_mime_type, file_open_dir_list, HIO_NULL };
hio_htre_perdecqpath (req);
mth = hio_htre_getqmethodtype(req);
qpath = hio_htre_getqpath(req);
2022-10-25 22:47:40 +00:00
if (mth == HIO_HTTP_OTHER && hio_comp_bcstr(hio_htre_getqmethodname(req), "UNTAR", 1) == 0)
{
/* don't care about the path for now. TODO: make this secure and reasonable */
2022-10-25 22:47:40 +00:00
hio_svc_htts_dothr(htts, csck, req, untar, HIO_NULL, 0);
}
else // if (mth == HIO_HTTP_GET || mth == HIO_HTTP_POST)
{
/* TODO: proper mime-type */
/* TODO: make HIO_SVC_HTTS_FILE_DIR a cli option */
if (hio_svc_htts_dofile(htts, csck, req, ext->ai->docroot, qpath, HIO_NULL, 0, &fcbs) <= -1) goto oops;
}
#if 0
else
{
if (hio_svc_htts_dotxt(htts, csck, req, HIO_HTTP_STATUS_FORBIDDEN, "text/plain", hio_http_status_to_bcstr(403), 0) <= -1) goto oops;
}
#endif
return 0;
2022-10-25 22:47:40 +00:00
oops:
hio_dev_sck_halt (csck);
return 0;
}
int webs_start (hio_t* hio, const arg_info_t* ai)
{
const hio_bch_t* ptr, * end;
hio_bcs_t tok;
hio_dev_sck_bind_t bi[100];
hio_oow_t bic;
hio_svc_htts_t* webs;
htts_ext_t* ext;
bic = 0;
ptr = ai->laddrs;
end = ptr + hio_count_bcstr(ptr);
while (ptr)
{
ptr = hio_tokenize_bchars(ptr, end - ptr, ", ", 2, &tok, 0);
if (tok.len > 0)
{
if (hio_bcharstoskad(hio, tok.ptr, tok.len, &bi[bic].localaddr) <= -1)
{
/* TODO: logging */
continue;
}
bi[bic].options = HIO_DEV_SCK_BIND_REUSEADDR | HIO_DEV_SCK_BIND_REUSEPORT | HIO_DEV_SCK_BIND_IGNERR;
bic++;
if (bic >= HIO_COUNTOF(bi)) break; /* TODO: make 'bi' dynamic */
}
}
webs = hio_svc_htts_start(hio, HIO_SIZEOF(htts_ext_t), bi, bic, process_http_request);
if (!webs) return -1; /* TODO: logging */
ext = hio_svc_htts_getxtn(webs);
ext->ai = ai;
return 0;
}
static hio_t* g_hio = HIO_NULL;
static void handle_sigint (int sig)
{
if (g_hio) hio_stop (g_hio, HIO_STOPREQ_TERMINATION);
}
static int process_args (int argc, char* argv[], arg_info_t* ai)
{
static hio_bopt_lng_t lopt[] =
{
{ "file-no-list-dir", '\0' },
{ "file-no-load-index-page", '\0'},
{ HIO_NULL, '\0'}
};
static hio_bopt_t opt =
{
"",
2022-12-26 17:03:53 +00:00
lopt
};
hio_bci_t c;
if (argc < 3)
{
print_usage:
fprintf (stderr, "Usage: %s [options] listen-address:port docroot-dir\n", argv[0]);
return -1;
}
memset (ai, 0, HIO_SIZEOF(*ai));
ai->file_list_dir = 1;
ai->file_load_index_page = 1;
while ((c = hio_getbopt(argc, argv, &opt)) != HIO_BCI_EOF)
{
switch (c)
{
case '\0':
if (strcasecmp(opt.lngopt, "file-no-list-dir") == 0)
{
ai->file_list_dir = 0;
break;
}
else if (strcasecmp(opt.lngopt, "file-no-load-index-page") == 0)
{
ai->file_load_index_page = 0;
break;
}
goto print_usage;
case ':':
if (opt.lngopt)
fprintf (stderr, "bad argument for '%s'\n", opt.lngopt);
else
fprintf (stderr, "bad argument for '%c'\n", opt.opt);
return -1;
default:
goto print_usage;
}
}
if (argc - opt.ind != 2) goto print_usage;
ai->laddrs = argv[opt.ind++];
ai->docroot = argv[opt.ind++];
return 0;
}
int main (int argc, char* argv[])
{
hio_t* hio = HIO_NULL;
struct sigaction sigact;
arg_info_t ai;
int xret = -1;
if (process_args(argc, argv, &ai) <= -1) return -1;
memset (&sigact, 0, HIO_SIZEOF(sigact));
sigact.sa_handler = SIG_IGN;
sigaction (SIGPIPE, &sigact, HIO_NULL);
memset (&sigact, 0, HIO_SIZEOF(sigact));
sigact.sa_handler = handle_sigint;
sigaction (SIGINT, &sigact, HIO_NULL);
hio = hio_open(HIO_NULL, 0, HIO_NULL, HIO_FEATURE_ALL, 512, HIO_NULL);
if (!hio)
{
printf ("Cannot open hio\n");
goto oops;
}
hio_setoption (hio, HIO_LOG_TARGET_BCSTR, "/dev/stderr");
{
hio_bitmask_t logmask;
hio_getoption (hio, HIO_LOG_MASK, &logmask);
logmask |= HIO_LOG_GUARDED;
hio_setoption (hio, HIO_LOG_MASK, &logmask);
}
g_hio = hio;
if (webs_start(hio, &ai) <= -1) goto oops;
hio_loop (hio);
g_hio = HIO_NULL;
xret = 0;
oops:
memset (&sigact, 0, HIO_SIZEOF(sigact));
sigact.sa_handler = SIG_IGN;
sigaction (SIGINT, &sigact, HIO_NULL);
if (hio) hio_close (hio);
return xret;
}