-
Notifications
You must be signed in to change notification settings - Fork 21
/
example_httpuplds.c
139 lines (130 loc) · 5.05 KB
/
example_httpuplds.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
/* _
* ___ __ _ __ _ _ _(_)
* / __|/ _` |/ _` | | | | |
* \__ \ (_| | (_| | |_| | |
* |___/\__,_|\__, |\__,_|_|
* |___/
*
* Cross-platform library which helps to develop web servers or frameworks.
*
* Copyright (C) 2016-2020 Silvio Clecio <[email protected]>
*
* Sagui library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* Sagui library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with Sagui library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/
#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <limits.h>
#include <sagui.h>
/* NOTE: Error checking has been omitted to make it clear. */
#ifdef _WIN32
#define PATH_SEP '\\'
#else /* _WIN32 */
#define PATH_SEP '/'
#endif /* _WIN32 */
#define PAGE_FORM \
"<html>" \
"<body>" \
"<form action=\"\" method=\"post\" enctype=\"multipart/form-data\">" \
"<fieldset>" \
"<legend>Choose the files:</legend>" \
"File 1: <input type=\"file\" name=\"file1\"/><br>" \
"File 2: <input type=\"file\" name=\"file2\"/><br>" \
"<input type=\"submit\"/>" \
"</fieldset>" \
"</form>" \
"</body>" \
"</html>"
#define PAGE_DONE \
"<html>" \
"<head>" \
"<title>Uploads</title>" \
"</head>" \
"<body>" \
"<strong>Uploaded files:</strong><br>" \
"%s" \
"</body>" \
"</html>"
#define CONTENT_TYPE "text/html; charset=utf-8"
static void process_uploads(struct sg_httpreq *req, struct sg_httpres *res) {
struct sg_httpupld *upld;
struct sg_str *body;
const char *name;
char *str;
char errmsg[256];
int errnum;
body = sg_str_new();
sg_str_printf(body, "<ol>");
upld = sg_httpreq_uploads(req);
while (upld) {
name = sg_httpupld_name(upld);
errnum = sg_httpupld_save(upld, true);
if (errnum == 0)
sg_str_printf(body, "<li><a href=\"?file=%s\">%s</a></li>", name, name);
else {
sg_strerror(errnum, errmsg, sizeof(errmsg));
sg_str_printf(body, "<li><font color='red'>%s - failed - %s</font></li>",
name, errmsg);
}
sg_httpuplds_next(&upld);
}
sg_str_printf(body, "</ol>");
str = strdup(sg_str_content(body));
sg_str_clear(body);
sg_str_printf(body, PAGE_DONE, str);
free(str);
sg_httpres_send(res, sg_str_content(body), CONTENT_TYPE, 200);
sg_str_free(body);
}
static void req_cb(__SG_UNUSED void *cls, struct sg_httpreq *req,
struct sg_httpres *res) {
struct sg_strmap **qs;
const char *file;
char path[PATH_MAX];
if (sg_httpreq_is_uploading(req))
process_uploads(req, res);
else {
qs = sg_httpreq_params(req);
if (qs && sg_strmap_count(*qs) > 0) {
file = sg_strmap_get(*qs, "file");
if (file) {
sprintf(path, "%s%c%s", sg_httpsrv_upld_dir(sg_httpreq_srv(req)),
PATH_SEP, file);
sg_httpres_download(res, path, 200);
}
} else
sg_httpres_send(res, PAGE_FORM, CONTENT_TYPE, 200);
}
}
int main(int argc, const char *argv[]) {
struct sg_httpsrv *srv;
uint16_t port;
if (argc != 2) {
printf("%s <PORT>\n", argv[0]);
return EXIT_FAILURE;
}
port = strtol(argv[1], NULL, 10);
srv = sg_httpsrv_new(req_cb, NULL);
if (!sg_httpsrv_listen(srv, port, false)) {
sg_httpsrv_free(srv);
return EXIT_FAILURE;
}
fprintf(stdout, "Server running at https://localhost:%d\n",
sg_httpsrv_port(srv));
fflush(stdout);
getchar();
sg_httpsrv_free(srv);
return EXIT_SUCCESS;
}