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#define _GNU_SOURCE 1
39#include <sys/types.h>
40#include <sys/stat.h>
41#include <errno.h>
42#include <stdint.h>
43#include <stdlib.h>
44#include <string.h>
45#include <unistd.h>
46#include <stdio.h>
47#include <dirent.h>
48#include <time.h>
49
50
51
52
53
54#define STYLE_STR \
55"<style>" "\n"\
56"table {" "\n"\
57 "width:100%;" "\n"\
58 "background-color:#fff5ee;" "\n"\
59 "border-width:1px;" "\n"\
60 "border-spacing:2px;" "\n"\
61 "border-style:solid;" "\n"\
62 "border-color:black;" "\n"\
63 "border-collapse:collapse;" "\n"\
64"}" "\n"\
65"th {" "\n"\
66 "border-width:1px;" "\n"\
67 "padding:1px;" "\n"\
68 "border-style:solid;" "\n"\
69 "border-color:black;" "\n"\
70"}" "\n"\
71"td {" "\n"\
72 \
73 "border-width:0px 1px 0px 1px;" "\n"\
74 "padding:1px;" "\n"\
75 "border-style:solid;" "\n"\
76 "border-color:black;" "\n"\
77 "white-space:nowrap;" "\n"\
78"}" "\n"\
79"tr.hdr { background-color:#eee5de; }" "\n"\
80"tr.o { background-color:#ffffff; }" "\n"\
81 \
82"tr.foot { background-color:#eee5de; }" "\n"\
83"th.cnt { text-align:left; }" "\n"\
84"th.sz { text-align:right; }" "\n"\
85"th.dt { text-align:right; }" "\n"\
86"td.sz { text-align:right; }" "\n"\
87"td.dt { text-align:right; }" "\n"\
88"col.nm { width:98%; }" "\n"\
89"col.sz { width:1%; }" "\n"\
90"col.dt { width:1%; }" "\n"\
91"</style>" "\n"\
92
93typedef struct dir_list_t {
94 char *dl_name;
95 mode_t dl_mode;
96 off_t dl_size;
97 time_t dl_mtime;
98} dir_list_t;
99
100static int compare_dl(dir_list_t *a, dir_list_t *b)
101{
102
103 if (strcmp(a->dl_name, "..") == 0) {
104 return -1;
105 }
106 if (strcmp(b->dl_name, "..") == 0) {
107 return 1;
108 }
109 if (S_ISDIR(a->dl_mode) != S_ISDIR(b->dl_mode)) {
110
111
112 return (S_ISDIR(b->dl_mode) != 0) ? 1 : -1;
113 }
114 return strcmp(a->dl_name, b->dl_name);
115}
116
117static char buffer[2*1024 > sizeof(STYLE_STR) ? 2*1024 : sizeof(STYLE_STR)];
118static char *dst = buffer;
119enum {
120 BUFFER_SIZE = sizeof(buffer),
121 HEADROOM = 64,
122};
123
124
125
126static void guarantee(int size)
127{
128 if (buffer + (BUFFER_SIZE-HEADROOM) - dst >= size)
129 return;
130 write(STDOUT_FILENO, buffer, dst - buffer);
131 dst = buffer;
132}
133
134
135
136
137static void fmt_str( const char *src)
138{
139 unsigned len = strlen(src);
140 guarantee(len);
141 memcpy(dst, src, len);
142 dst += len;
143}
144
145
146static void fmt_url( const char *name)
147{
148 while (*name) {
149 unsigned c = *name++;
150 guarantee(3);
151 *dst = c;
152 if ((c - '0') > 9
153 && ((c|0x20) - 'a') > ('z' - 'a')
154 && !strchr("._-+@", c)
155 ) {
156 *dst++ = '%';
157 *dst++ = "0123456789ABCDEF"[c >> 4];
158 *dst = "0123456789ABCDEF"[c & 0xf];
159 }
160 dst++;
161 }
162}
163
164
165static void fmt_html( const char *name)
166{
167 while (*name) {
168 char c = *name++;
169 if (c == '<')
170 fmt_str("<");
171 else if (c == '>')
172 fmt_str(">");
173 else if (c == '&') {
174 fmt_str("&");
175 } else {
176 guarantee(1);
177 *dst++ = c;
178 continue;
179 }
180 }
181}
182
183
184static void fmt_ull( unsigned long long n)
185{
186 char buf[sizeof(n)*3 + 2];
187 char *p;
188
189 p = buf + sizeof(buf) - 1;
190 *p = '\0';
191 do {
192 *--p = (n % 10) + '0';
193 n /= 10;
194 } while (n);
195 fmt_str( p);
196}
197
198
199static void fmt_02u( unsigned n)
200{
201
202 dst[0] = (n / 10) + '0';
203 dst[1] = (n % 10) + '0';
204 dst += 2;
205}
206
207
208static void fmt_04u( unsigned n)
209{
210
211 fmt_02u(n / 100);
212 fmt_02u(n % 100);
213}
214
215int main(int argc, char *argv[])
216{
217 dir_list_t *dir_list;
218 dir_list_t *cdir;
219 unsigned dir_list_count;
220 unsigned count_dirs;
221 unsigned count_files;
222 unsigned long long size_total;
223 int odd;
224 DIR *dirp;
225 char *location;
226
227 location = getenv("REQUEST_URI");
228 if (!location)
229 return 1;
230
231
232 strchrnul(location, '?')[0] = '\0';
233
234 if (location[0] != '/'
235 || strstr(location, "//")
236 || strstr(location, "/../")
237 || strcmp(strrchr(location, '/'), "/..") == 0
238 ) {
239 return 1;
240 }
241
242 if (chdir("..")
243 || (location[1] && chdir(location + 1))
244 ) {
245 return 1;
246 }
247
248 dirp = opendir(".");
249 if (!dirp)
250 return 1;
251 dir_list = NULL;
252 dir_list_count = 0;
253 while (1) {
254 struct dirent *dp;
255 struct stat sb;
256
257 dp = readdir(dirp);
258 if (!dp)
259 break;
260 if (dp->d_name[0] == '.' && !dp->d_name[1])
261 continue;
262 if (stat(dp->d_name, &sb) != 0)
263 continue;
264 dir_list = realloc(dir_list, (dir_list_count + 1) * sizeof(dir_list[0]));
265 dir_list[dir_list_count].dl_name = strdup(dp->d_name);
266 dir_list[dir_list_count].dl_mode = sb.st_mode;
267 dir_list[dir_list_count].dl_size = sb.st_size;
268 dir_list[dir_list_count].dl_mtime = sb.st_mtime;
269 dir_list_count++;
270 }
271 closedir(dirp);
272
273 qsort(dir_list, dir_list_count, sizeof(dir_list[0]), (void*)compare_dl);
274
275 fmt_str(
276 ""
277 "\r\n"
278 "<html><head><title>Index of ");
279
280 fmt_html(location);
281 fmt_str(
282 "</title>\n"
283 STYLE_STR
284 "</head>" "\n"
285 "<body>" "\n"
286 "<h1>Index of ");
287 fmt_html(location);
288 fmt_str(
289 "</h1>" "\n"
290 "<table>" "\n"
291 "<col class=nm><col class=sz><col class=dt>" "\n"
292 "<tr class=hdr><th class=cnt>Name<th class=sz>Size<th class=dt>Last modified" "\n");
293
294 odd = 0;
295 count_dirs = 0;
296 count_files = 0;
297 size_total = 0;
298 cdir = dir_list;
299 while (dir_list_count--) {
300 struct tm *ptm;
301
302 if (S_ISDIR(cdir->dl_mode)) {
303 count_dirs++;
304 } else if (S_ISREG(cdir->dl_mode)) {
305 count_files++;
306 size_total += cdir->dl_size;
307 } else
308 goto next;
309
310 fmt_str("<tr class=");
311 *dst++ = (odd ? 'o' : 'e');
312 fmt_str("><td class=nm><a href='");
313 fmt_url(cdir->dl_name);
314 if (S_ISDIR(cdir->dl_mode))
315 *dst++ = '/';
316 fmt_str("'>");
317 fmt_html(cdir->dl_name);
318 if (S_ISDIR(cdir->dl_mode))
319 *dst++ = '/';
320 fmt_str("</a><td class=sz>");
321 if (S_ISREG(cdir->dl_mode))
322 fmt_ull(cdir->dl_size);
323 fmt_str("<td class=dt>");
324 ptm = gmtime(&cdir->dl_mtime);
325 fmt_04u(1900 + ptm->tm_year); *dst++ = '-';
326 fmt_02u(ptm->tm_mon + 1); *dst++ = '-';
327 fmt_02u(ptm->tm_mday); *dst++ = ' ';
328 fmt_02u(ptm->tm_hour); *dst++ = ':';
329 fmt_02u(ptm->tm_min); *dst++ = ':';
330 fmt_02u(ptm->tm_sec);
331 *dst++ = '\n';
332
333 odd = 1 - odd;
334 next:
335 cdir++;
336 }
337
338 fmt_str("<tr class=foot><th class=cnt>Files: ");
339 fmt_ull(count_files);
340
341 fmt_str(", directories: ");
342 fmt_ull(count_dirs - 1);
343 fmt_str("<th class=sz>");
344 fmt_ull(size_total);
345 fmt_str("<th class=dt>\n");
346
347 guarantee(BUFFER_SIZE * 2);
348
349 return 0;
350}
351