Added Date header to all responses, and Connection:close to final responses

RamseyK · RamseyK · commit 8b0eb993cf9e · 2012-05-12T02:40:11.000-04:00
diff --git a/HTTPServer.cpp b/HTTPServer.cpp
@@ -355,6 +355,22 @@ HTTPResponse* HTTPServer::handleHead(Client *cl, HTTPRequest *req) {
  * @param disconnect Should the server disconnect the client after sending (Optional, default = false)
  */
 void HTTPServer::sendResponse(Client* cl, HTTPResponse* res, bool disconnect) {
+	// Time stamp the response with the Date header
+	string tstr;
+	char tbuf[36];
+	time_t rawtime;
+	struct tm* ptm;
+	time(&rawtime);
+	ptm = gmtime(&rawtime);
+	// Ex: Fri, 31 Dec 1999 23:59:59 GMT
+	strftime(tbuf, 36, "%a, %d %b %Y %H:%M:%S GMT", ptm);
+	tstr = tbuf;
+	res->addHeader("Date", tstr);
+	
+	// Include a Connection: close header if this is the final response sent by the server
+	if(disconnect)
+		res->addHeader("Connection", "close");
+	
 	// Get raw data by creating the response (pData will be cleaned up by the response obj)
 	byte* pData = res->create();
 	
@@ -378,6 +394,8 @@ void HTTPServer::sendResponse(Client* cl, HTTPResponse* res, bool disconnect) {
 		bytesLeft -= n;
 	}
 	
+	cout << "[" << cl->getClientIP() << "] was sent " << totalSent << " bytes" << endl;
+	
 	if(disconnect)
 		disconnectClient(cl);
 }
diff --git a/HTTPServer.h b/HTTPServer.h
@@ -4,6 +4,7 @@
 #include <map>
 #include <string>
 
+#include <time.h>
 #include <unistd.h>
 #include <sys/types.h>
 #include <sys/socket.h>
