| /* |
| * Copyright (c) 2007 Denys Vlasenko <vda.linux@googlemail.com> |
| * |
| * Licensed under GPLv2, see file LICENSE in this source tree. |
| */ |
| |
| /* |
| * This program is a CGI application. It outputs directory index page. |
| * Put it into cgi-bin/index.cgi and chmod 0755. |
| */ |
| |
| /* Build a-la |
| i486-linux-uclibc-gcc \ |
| -static -static-libgcc \ |
| -D_LARGEFILE_SOURCE -D_LARGEFILE64_SOURCE -D_FILE_OFFSET_BITS=64 \ |
| -Wall -Wshadow -Wwrite-strings -Wundef -Wstrict-prototypes -Werror \ |
| -Wold-style-definition -Wdeclaration-after-statement -Wno-pointer-sign \ |
| -Wmissing-prototypes -Wmissing-declarations \ |
| -Os -fno-builtin-strlen -finline-limit=0 -fomit-frame-pointer \ |
| -ffunction-sections -fdata-sections -fno-guess-branch-probability \ |
| -funsigned-char \ |
| -falign-functions=1 -falign-jumps=1 -falign-labels=1 -falign-loops=1 \ |
| -march=i386 -mpreferred-stack-boundary=2 \ |
| -Wl,-Map -Wl,link.map -Wl,--warn-common -Wl,--sort-common -Wl,--gc-sections \ |
| httpd_indexcgi.c -o index.cgi |
| */ |
| |
| /* We don't use printf, as it pulls in >12 kb of code from uclibc (i386). */ |
| /* Currently malloc machinery is the biggest part of libc we pull in. */ |
| /* We have only one realloc and one strdup, any idea how to do without? */ |
| |
| /* Size (i386, static uclibc, approximate): |
| * text data bss dec hex filename |
| * 13036 44 3052 16132 3f04 index.cgi |
| * 2576 4 2048 4628 1214 index.cgi.o |
| */ |
| |
| #define _GNU_SOURCE 1 /* for strchrnul */ |
| #include <sys/types.h> |
| #include <sys/stat.h> |
| #include <errno.h> |
| #include <stdint.h> |
| #include <stdlib.h> |
| #include <string.h> |
| #include <unistd.h> |
| #include <stdio.h> |
| #include <dirent.h> |
| #include <time.h> |
| |
| /* Appearance of the table is controlled by style sheet *ONLY*, |
| * formatting code uses <TAG class=CLASS> to apply style |
| * to elements. Edit stylesheet to your liking and recompile. */ |
| |
| #define STYLE_STR \ |
| "<style>" \ |
| "table {" \ |
| "width:100%;" \ |
| "background-color:#fff5ee;" \ |
| "border-width:1px;" /* 1px 1px 1px 1px; */ \ |
| "border-spacing:2px;" \ |
| "border-style:solid;" /* solid solid solid solid; */ \ |
| "border-color:black;" /* black black black black; */ \ |
| "border-collapse:collapse" \ |
| "}" \ |
| "th {" \ |
| "border-width:1px;" /* 1px 1px 1px 1px; */ \ |
| "padding:1px;" /* 1px 1px 1px 1px; */ \ |
| "border-style:solid;" /* solid solid solid solid; */ \ |
| "border-color:black" /* black black black black; */ \ |
| "}" \ |
| "td {" \ |
| /* top right bottom left */ \ |
| "border-width:0 1px 0 1px;" \ |
| "padding:1px;" /* 1px 1px 1px 1px; */ \ |
| "border-style:solid;" /* solid solid solid solid; */ \ |
| "border-color:black;" /* black black black black; */ \ |
| "white-space:nowrap" \ |
| "}" \ |
| "tr:nth-child(odd) { background-color:#ffffff }" \ |
| "tr.hdr { background-color:#eee5de }" \ |
| "tr.foot { background-color:#eee5de }" \ |
| "th.cnt { text-align:left }" \ |
| "th.sz { text-align:right }" \ |
| "th.dt { text-align:right }" \ |
| "td.sz { text-align:right }" \ |
| "td.dt { text-align:right }" \ |
| "col.nm { width:98% }" \ |
| "col.sz { width:1% }" \ |
| "col.dt { width:1% }" \ |
| "</style>" \ |
| |
| typedef struct dir_list_t { |
| char *dl_name; |
| mode_t dl_mode; |
| off_t dl_size; |
| time_t dl_mtime; |
| } dir_list_t; |
| |
| static int compare_dl(dir_list_t *a, dir_list_t *b) |
| { |
| /* ".." is 'less than' any other dir entry */ |
| if (strcmp(a->dl_name, "..") == 0) { |
| return -1; |
| } |
| if (strcmp(b->dl_name, "..") == 0) { |
| return 1; |
| } |
| if (S_ISDIR(a->dl_mode) != S_ISDIR(b->dl_mode)) { |
| /* 1 if b is a dir (and thus a is 'after' b, a > b), |
| * else -1 (a < b) */ |
| return (S_ISDIR(b->dl_mode) != 0) ? 1 : -1; |
| } |
| return strcmp(a->dl_name, b->dl_name); |
| } |
| |
| static char buffer[2*1024 > sizeof(STYLE_STR) ? 2*1024 : sizeof(STYLE_STR)]; |
| static char *dst = buffer; |
| enum { |
| BUFFER_SIZE = sizeof(buffer), |
| HEADROOM = 64, |
| }; |
| |
| /* After this call, you have at least size + HEADROOM bytes available |
| * ahead of dst */ |
| static void guarantee(int size) |
| { |
| if (buffer + (BUFFER_SIZE-HEADROOM) - dst >= size) |
| return; |
| write(STDOUT_FILENO, buffer, dst - buffer); |
| dst = buffer; |
| } |
| |
| /* NB: formatters do not store terminating NUL! */ |
| |
| /* HEADROOM bytes are available after dst after this call */ |
| static void fmt_str(/*char *dst,*/ const char *src) |
| { |
| unsigned len = strlen(src); |
| guarantee(len); |
| memcpy(dst, src, len); |
| dst += len; |
| } |
| |
| /* HEADROOM bytes after dst are available after this call */ |
| static void fmt_url(/*char *dst,*/ const char *name) |
| { |
| while (*name) { |
| unsigned c = *name++; |
| guarantee(3); |
| *dst = c; |
| if ((c - '0') > 9 /* not a digit */ |
| && ((c|0x20) - 'a') > ('z' - 'a') /* not A-Z or a-z */ |
| && !strchr("._-+@", c) |
| ) { |
| *dst++ = '%'; |
| *dst++ = "0123456789ABCDEF"[c >> 4]; |
| *dst = "0123456789ABCDEF"[c & 0xf]; |
| } |
| dst++; |
| } |
| } |
| |
| /* HEADROOM bytes are available after dst after this call */ |
| static void fmt_html(/*char *dst,*/ const char *name) |
| { |
| while (*name) { |
| char c = *name++; |
| if (c == '<') |
| fmt_str("<"); |
| else if (c == '>') |
| fmt_str(">"); |
| else if (c == '&') { |
| fmt_str("&"); |
| } else { |
| guarantee(1); |
| *dst++ = c; |
| continue; |
| } |
| } |
| } |
| |
| /* HEADROOM bytes are available after dst after this call */ |
| static void fmt_ull(/*char *dst,*/ unsigned long long n) |
| { |
| char buf[sizeof(n)*3 + 2]; |
| char *p; |
| |
| p = buf + sizeof(buf) - 1; |
| *p = '\0'; |
| do { |
| *--p = (n % 10) + '0'; |
| n /= 10; |
| } while (n); |
| fmt_str(/*dst,*/ p); |
| } |
| |
| /* Does not call guarantee - eats into headroom instead */ |
| static void fmt_02u(/*char *dst,*/ unsigned n) |
| { |
| /* n %= 100; - not needed, callers don't pass big n */ |
| dst[0] = (n / 10) + '0'; |
| dst[1] = (n % 10) + '0'; |
| dst += 2; |
| } |
| |
| /* Does not call guarantee - eats into headroom instead */ |
| static void fmt_04u(/*char *dst,*/ unsigned n) |
| { |
| /* n %= 10000; - not needed, callers don't pass big n */ |
| fmt_02u(n / 100); |
| fmt_02u(n % 100); |
| } |
| |
| int main(int argc, char **argv) |
| { |
| dir_list_t *dir_list; |
| dir_list_t *cdir; |
| unsigned dir_list_count; |
| unsigned count_dirs; |
| unsigned count_files; |
| unsigned long long size_total; |
| DIR *dirp; |
| char *location; |
| |
| location = getenv("REQUEST_URI"); |
| if (!location) |
| return 1; |
| |
| /* drop URL arguments if any */ |
| strchrnul(location, '?')[0] = '\0'; |
| |
| if (location[0] != '/' |
| || strstr(location, "//") |
| || strstr(location, "/../") |
| || strcmp(strrchr(location, '/'), "/..") == 0 |
| ) { |
| return 1; |
| } |
| |
| if (chdir("..") |
| || (location[1] && chdir(location + 1)) |
| ) { |
| return 1; |
| } |
| |
| dirp = opendir("."); |
| if (!dirp) |
| return 1; |
| dir_list = NULL; |
| dir_list_count = 0; |
| while (1) { |
| struct dirent *dp; |
| struct stat sb; |
| |
| dp = readdir(dirp); |
| if (!dp) |
| break; |
| if (dp->d_name[0] == '.' && !dp->d_name[1]) |
| continue; |
| if (stat(dp->d_name, &sb) != 0) |
| continue; |
| dir_list = realloc(dir_list, (dir_list_count + 1) * sizeof(dir_list[0])); |
| dir_list[dir_list_count].dl_name = strdup(dp->d_name); |
| dir_list[dir_list_count].dl_mode = sb.st_mode; |
| dir_list[dir_list_count].dl_size = sb.st_size; |
| dir_list[dir_list_count].dl_mtime = sb.st_mtime; |
| dir_list_count++; |
| } |
| closedir(dirp); |
| |
| qsort(dir_list, dir_list_count, sizeof(dir_list[0]), (void*)compare_dl); |
| |
| fmt_str( |
| "" /* Additional headers (currently none) */ |
| "\r\n" /* Mandatory empty line after headers */ |
| "<html><head><title>Index of "); |
| /* Guard against directories with &, > etc */ |
| fmt_html(location); |
| fmt_str( |
| "</title>\n" |
| STYLE_STR |
| "</head>" "\n" |
| "<body>" "\n" |
| "<h1>Index of "); |
| fmt_html(location); |
| fmt_str( |
| "</h1>" "\n" |
| "<table>" "\n" |
| "<col class=nm><col class=sz><col class=dt>" "\n" |
| "<tr class=hdr><th class=cnt>Name<th class=sz>Size<th class=dt>Last modified" "\n"); |
| |
| count_dirs = 0; |
| count_files = 0; |
| size_total = 0; |
| cdir = dir_list; |
| while (dir_list_count--) { |
| struct tm *ptm; |
| |
| if (S_ISDIR(cdir->dl_mode)) { |
| count_dirs++; |
| } else if (S_ISREG(cdir->dl_mode)) { |
| count_files++; |
| size_total += cdir->dl_size; |
| } else |
| goto next; |
| |
| fmt_str("<tr><td class=nm><a href='"); |
| fmt_url(cdir->dl_name); /* %20 etc */ |
| if (S_ISDIR(cdir->dl_mode)) |
| *dst++ = '/'; |
| fmt_str("'>"); |
| fmt_html(cdir->dl_name); /* < etc */ |
| if (S_ISDIR(cdir->dl_mode)) |
| *dst++ = '/'; |
| fmt_str("</a><td class=sz>"); |
| if (S_ISREG(cdir->dl_mode)) |
| fmt_ull(cdir->dl_size); |
| fmt_str("<td class=dt>"); |
| ptm = gmtime(&cdir->dl_mtime); |
| fmt_04u(1900 + ptm->tm_year); *dst++ = '-'; |
| fmt_02u(ptm->tm_mon + 1); *dst++ = '-'; |
| fmt_02u(ptm->tm_mday); *dst++ = ' '; |
| fmt_02u(ptm->tm_hour); *dst++ = ':'; |
| fmt_02u(ptm->tm_min); *dst++ = ':'; |
| fmt_02u(ptm->tm_sec); |
| *dst++ = '\n'; |
| |
| next: |
| cdir++; |
| } |
| |
| fmt_str("<tr class=foot><th class=cnt>Files: "); |
| fmt_ull(count_files); |
| /* count_dirs - 1: we don't want to count ".." */ |
| fmt_str(", directories: "); |
| fmt_ull(count_dirs - 1); |
| fmt_str("<th class=sz>"); |
| fmt_ull(size_total); |
| fmt_str("<th class=dt>\n"); |
| /* "</table></body></html>" - why bother? */ |
| guarantee(BUFFER_SIZE * 2); /* flush */ |
| |
| return 0; |
| } |