summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--browse.c36
-rw-r--r--cgi.h41
-rw-r--r--main.c12
3 files changed, 63 insertions, 26 deletions
diff --git a/browse.c b/browse.c
index 65e3c5a..577183f 100644
--- a/browse.c
+++ b/browse.c
@@ -35,15 +35,16 @@
#include <string.h>
#include "browse.h"
+#include "cgi.h"
#include "config.h"
-#include "error.h"
#include "http.h"
#include "url.h"
#define URL_LENGTH_MAX 8192
/*
- * Checks that the path can be safely processed. Namely, it should not contain "..", which denotes an attempt to get out of the DATA_DIR root folder.
+ * Checks that the path can be safely processed. Namely, it should not contain
+ * "..", which denotes an attempt to get out of the DATA_DIR root folder.
*/
static bool
check_request_path(char *path)
@@ -105,20 +106,19 @@ browse(struct kreq * r)
}
http_open(r, KHTTP_200);
- if (KCGI_OK != khtml_open(&html, r, 0))
- return VAULT_CGI_ERROR;
+ K_OK(khtml_open(&html, r, 0));
/* build basic html page */
- khtml_elem(&html, KELEM_DOCTYPE);
- khtml_elem(&html, KELEM_HEAD);
- khtml_attr(&html, KELEM_META, KATTR_CHARSET, "utf-8", KATTR__MAX);
- khtml_elem(&html, KELEM_HTML);
- khtml_elem(&html, KELEM_BODY);
- khtml_elem(&html, KELEM_P);
- khtml_puts(&html, "/");
- khtml_closeelem(&html, 1);
+ K_OK(khtml_elem(&html, KELEM_DOCTYPE));
+ K_OK(khtml_elem(&html, KELEM_HEAD));
+ K_OK(khtml_attr(&html, KELEM_META, KATTR_CHARSET, "utf-8", KATTR__MAX));
+ K_OK(khtml_elem(&html, KELEM_HTML));
+ K_OK(khtml_elem(&html, KELEM_BODY));
+ K_OK(khtml_elem(&html, KELEM_P));
+ K_OK(khtml_puts(&html, "/"));
+ K_OK(khtml_closeelem(&html, 1));
- khtml_elem(&html, KELEM_UL);
+ K_OK(khtml_elem(&html, KELEM_UL));
while ((dir = readdir(data_dir)) != NULL) {
/* ignore special . and .. folders */
file_name = dir->d_name;
@@ -127,12 +127,12 @@ browse(struct kreq * r)
url_size = build_browse_url(r, url, URL_LENGTH_MAX, file_name);
if (url_size == 0 || url_size >= URL_LENGTH_MAX)
continue;
- khtml_elem(&html, KELEM_LI);
- khtml_attr(&html, KELEM_A, KATTR_HREF, url, KATTR__MAX);
- khtml_puts(&html, dir->d_name);
- khtml_closeelem(&html, 2);
+ K_OK(khtml_elem(&html, KELEM_LI));
+ K_OK(khtml_attr(&html, KELEM_A, KATTR_HREF, url, KATTR__MAX));
+ K_OK(khtml_puts(&html, dir->d_name));
+ K_OK(khtml_closeelem(&html, 2));
}
- khtml_close(&html);
+ K_OK(khtml_close(&html));
closedir(data_dir);
return VAULT_SUCCESS;
diff --git a/cgi.h b/cgi.h
new file mode 100644
index 0000000..25988f3
--- /dev/null
+++ b/cgi.h
@@ -0,0 +1,41 @@
+/*
+ * Copyright 2023, Vincent Douillet <vincent@vdouillet.fr>
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright notice,
+ * this list of conditions and the following disclaimer.
+ *
+ * 2. Redistributions in binary form must reproduce the above copyright notice,
+ * this list of conditions and the following disclaimer in the documentation
+ * and/or other materials provided with the distribution.
+ *
+ * 3. Neither the name of the copyright holder nor the names of its contributors
+ * may be used to endorse or promote products derived from this software without
+ * specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
+ * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+ * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+ * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+ * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+ * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+ * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+ * POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#ifndef VAULT_CGI_H
+#define VAULT_CGI_H
+
+#include <kcgi.h>
+
+#include "error.h"
+
+#define K_OK(kcgi_call) if(kcgi_call != KCGI_OK) \
+ return VAULT_CGI_ERROR
+
+#endif /* VAULT_CGI_H */
diff --git a/main.c b/main.c
index 6abcbc5..6b54a5e 100644
--- a/main.c
+++ b/main.c
@@ -29,10 +29,10 @@
*/
#include <stdlib.h>
-#include <kcgi.h>
#include <unistd.h>
#include "browse.h"
+#include "cgi.h"
#include "http.h"
enum page {
@@ -47,14 +47,10 @@ static const char *const pages[PAGE__MAX] = {
int
main(void)
{
- enum kcgi_err err;
- int page_ret_val;
- struct kreq r;
+ int page_ret_val;
+ struct kreq r;
- err = khttp_parse(&r, NULL, 0, pages, PAGE__MAX, PAGE_BROWSE);
-
- if (KCGI_OK != err)
- return EXIT_FAILURE;
+ K_OK(khttp_parse(&r, NULL, 0, pages, PAGE__MAX, PAGE_BROWSE));
/*
* Make sure basic request parameters are as expected : GET or POST,