patch to Vanilla Tomato 1.28

1 files changed, 183 insertions, 0 deletions

diff --git a/release/src/router/httpd/blackhole.c b/release/src/router/httpd/blackhole.c
new file mode 100644
index 00000000..ab7df705
--- /dev/null
+++ b/release/src/router/httpd/blackhole.c
@@ -0,0 +1,183 @@
+/*
+
+	Tomato Firmware
+	Copyright (C) 2006-2009 Jonathan Zarate
+
+*/
+
+#include "tomato.h"
+
+#include <time.h>
+
+
+#ifdef BLACKHOLE
+void wi_blackhole(char *url, int len, char *boundary)
+{
+	char buf[2048];
+	int size;
+	int n;
+	time_t tick;
+	int foo;
+	const char *error;
+	int blen;
+	FILE *f;
+
+	if (!post) {
+		send_header(200, NULL, mime_html, 0);
+		web_printf(
+			"<h1>Upload Test</h1>"
+			"<form method='post' action='blackhole.cgi?_http_id=%s' encType='multipart/form-data'>"
+			"<input type='file' name='file'><input type='submit'>"
+			"</form>",
+			nvram_safe_get("http_id"));
+		return;
+	}
+
+	check_id();
+
+	cprintf("\nblackhole\n");
+	cprintf("%s<\n", boundary);
+
+	if ((blen = strlen(boundary)) == 0) {
+		error = "no boundary";
+ERROR:
+		cprintf("ERROR: %s\n", error);
+
+		web_eat(len);
+
+		send_header(200, NULL, mime_html, 0);
+		web_printf("ERROR: %s", error);
+		return;
+	}
+
+	if (blen > (sizeof(buf) - 32)) {
+		error = "boundary is too big";
+		goto ERROR;
+	}
+
+	// --b\r\n
+	// <data>\r\n
+	// --b--\r\n
+	if (len < ((blen * 2) + 12)) {
+		error = "not enough data";
+		goto ERROR;
+	}
+
+	foo = 1;
+	while (len > 0) {
+		if (!web_getline(buf, MIN(len, sizeof(buf)))) {
+			break;
+		}
+		n = strlen(buf);
+		len -= n;
+
+		if (n < 2) {
+			error = "n < 2";
+			goto ERROR;
+		}
+		if (buf[--n] != '\n') {
+			error = "\\n not found";
+			goto ERROR;
+		}
+		if (buf[--n] != '\r') {
+			error = "\\n without \\r";
+			goto ERROR;
+		}
+		if (n == 0) break;
+		buf[n] = 0;
+
+		cprintf("%s<\n", buf);
+
+		if (foo) {
+			if ((buf[0] != '-') || (buf[1] != '-') || (strcmp(buf + 2, boundary) != 0)) {
+				error = "boundary not found on first line";
+				goto ERROR;
+			}
+			foo = 0;
+		}
+	}
+
+	if (foo != 0) {
+		error = "boundary not found before reaching end of header";
+		goto ERROR;
+	}
+
+	blen += 6;
+	len -= (blen + 2);
+	if (len < 0) {
+		error = "not enough data for end boundary";
+		goto ERROR;
+	}
+
+	unlink("/tmp/blackhole");
+/*	if (len < (1*1024*1024)) f = fopen("/tmp/blackhole", "w");
+		else*/ f = NULL;
+
+#if 0
+	// read half, trigger tcp reset
+	len >>= 1;
+#endif
+
+	size = len;
+	tick = time(0);
+	while (len > 0) {
+		 if ((n = web_read(buf, MIN(len, sizeof(buf)))) <= 0) {
+			 break;
+		 }
+		 if (f) fwrite(buf, n, 1, f);
+		 len -= n;
+	}
+	tick = time(0) - tick;
+	if (f) fclose(f);
+
+	if (len > 0) {
+		error = "not all data was read";
+		goto ERROR;
+	}
+
+	if (web_read(buf, blen) != blen) {
+		error = "error while reading end boundary";
+		goto ERROR;
+	}
+	buf[blen] = 0;
+	cprintf(">>%s<<\n", buf);
+	if ((strncmp(buf, "\r\n--", 4) != 0) ||
+		(buf[blen - 1] != '-') || (buf[blen - 2] != '-')
+		|| (strncmp(buf + 4, boundary, blen - 6) != 0)) {
+		error = "end boundary not found";
+		goto ERROR;
+	}
+
+	len += 2;
+	if (len > 0) {
+		 if ((n = web_read(buf, MIN(len, sizeof(buf) - 1))) > 0) {
+			len -= n;
+			buf[n] = 0;
+			cprintf("last >>%s<<\n", buf);
+		 }
+	}
+
+	web_eat(len);
+	cprintf("len=%d\n", len);
+
+	if (tick > 0) n = size / tick;
+		else n = 0;
+
+	send_header(200, NULL, mime_html, 0);
+	web_printf(
+		"<pre>"
+		"Size .......: %d bytes\n"
+		"Time .......: %d seconds\n"
+		"Speed ......: %.2f kb/s | %.2f mb/s | %.2f KB/s | %.2f MB/s\n",
+		size,
+		tick,
+		n / 128.0, n / 131072.0,
+		n / 1024.0, n / 1048576.0);
+	if (f) {
+		web_pipecmd("md5sum /tmp/blackhole", WOF_NONE);
+	}
+	web_puts("</pre>");
+
+	cprintf("done...\n");
+}
+#endif