Enforce HTTP server response body size if "Content-Length" is present.

vibe-d · Feb 2, 2016 · 3d85140 · 3d85140
1 parent e0fae09
commit 3d85140
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/source/vibe/http/server.d b/source/vibe/http/server.d
@@ -944,9 +944,10 @@ final class HTTPServerResponse : HTTPResponse {
 			headers.remove("Content-Length");
 		}
 
-		if ("Content-Length" in headers) {
+		if (auto pcl = "Content-Length" in headers) {
 			writeHeader();
-			m_bodyWriter = m_countingWriter; // TODO: LimitedOutputStream(m_conn, content_length)
+			m_countingWriter.writeLimit = (*pcl).to!ulong;
+			m_bodyWriter = m_countingWriter;
 		} else {
 			headers["Transfer-Encoding"] = "chunked";
 			writeHeader();
@@ -979,8 +980,7 @@ final class HTTPServerResponse : HTTPResponse {
 	{
 		statusCode = status;
 		headers["Location"] = url;
-		headers["Content-Length"] = "14";
-		bodyWriter.write("redirecting...");
+		writeBody("redirecting...");
 	}
 	/// ditto
 	void redirect(URL url, int status = HTTPStatus.Found)