[ViewVC] Diff of: repos/Smersh/Daemon.cpp

Comparing Smersh/Daemon.cpp (file contents):
Revision 174 by Douglas Thrift, 2004-06-21T20:16:43-07:00 vs.
Revision 177 by Douglas Thrift, 2004-06-26T01:26:41-07:00

+//
+// $Id$
-+
+#include <arpa/inet.h>
-+
+#include "Daemon.hpp"
-+
+string Daemon::crlf("\r\n");
-+
+void Daemon::serve(int port, bool fork, Daemon* self)
+        api::TcpSocket server;
-+
+Daemon::Status Daemon::request(istream& sin, Environment& env, ostream& post,
-+
+        ostream& log)
-+
+{
-+
+        string line;
-+
+        Matcher request("^([A-Z]+) .*?(\\?.+)? HTTP/(\\d+)\\.(\\d+)$");
-+
-+
+        getline(sin, line);
-+
-+
+        log << '"' << line << "\" ";
-+
-+
+        env.put("HTTP_REFERER=-");
-+
+        env.put("HTTP_USER_AGENT=-");
-+
-+
+        if (line == request)
-+
+        {
-+
+                if (lexical_cast<unsigned>(request[3]) > 1) return version;
-+
-+
+                Matcher method("^GET|HEAD|POST$");
-+
-+
+                if (request[1] != method) return notImplemented;
-+
-+
+                env.set("REQUEST_METHOD", method);
-+
-+
+                if (!request[2].empty()) env.set("QUERY_STRING", request[2].substr(1));
-+
-+
+                headers(sin, env);
-+
-+
+                if (method[0] == "POST") return message(sin, env, post);
-+
-+
+                return ok;
-+
+        }
-+
-+
+        return notFound;
-+
+}
-+
-+
+void Daemon::response(ostream& sout, Status status)
-+
+{
-+
+        sout << "HTTP/1.1 " << status << ' ' << reason(status) << crlf << "Date: "
-+
+                << date() << crlf << "Server: Smersh/0.9" << crlf
-+
+                << "Connection: close" << crlf;
-+
+}
-+
-+
+streamsize Daemon::error(ostream& sout, Status status)
-+
+{
-+
+        string reason(this->reason(status));
-+
+        ostringstream error;
-+
-+
+        error << "<html><head><title>" << status << ' ' << reason << "</title></hea"
-+
+                << "d><body><h1>" << reason << "</h1><p>Mistakes were made, deal with t"
-+
+                << "hem.</p><hr /><address>Smersh/0.9</address></body></html>\n";
-+
+        sout << "Content-Length: " << error.str().length() << crlf
-+
+                << "Content-Type: text/html; charset=UTF-8\r\n\r\n" << error.str();
-+
-+
+        return error.str().size();
-+
+}
-+
-+
+string Daemon::date(bool log)
-+
+{
-+
+        time_t now(time(NULL));
-+
+        tm time;
-+
-+
+        if (log) localtime_r(&now, &time); else gmtime_r(&now, &time);
-+
-+
+        const char* format(log ? "[%m/%d/%Y:%T %z]" : "%a, %d %b %Y %T GMT");
-+
+        char when[log ? 29 : 30];
-+
-+
+        strftime(when, log ? 29 : 30, format, &time);
-+
-+
+        return when;
-+
+}
-+
-+
+string Daemon::ip(const api::TcpSocket& socket)
-+
+{
-+
+        api::InternetAddress address(socket.GetAddress());
-+
-+
+        return inet_ntoa(address->sin_addr);
-+
+}
-+
+int Daemon::handle(api::TcpSocket* client)
+        ios::InputOutputStreamBufAdapter adapter(*client);
-<
+        iostream socket(&adapter);
->
+        iostream sio(&adapter);
->
+        Environment env;
->
+        stringstream post;
->
+        ostringstream log;
->
+        Status code(request(sio, env, post, log));
->
->
+        response(sio, code);
->
->
+        bool head(env.get("REQUEST_METHOD") == "HEAD");
->
+        streamsize sent(0);
->
->
+        if (code == ok && !head)
->
+        {
->
+                ostringstream output;
->
+                Smersh smersh(post, output, env);
->
+                string content(output.str().substr(40));
->
->
+                sio << "Content-Length: " << content.length() << crlf
->
+                        << "Content-Type: text/html; charset=UTF-8\r\n\r\n";
-<
+        //
->
+                sio.write(content.data(), content.size());
->
->
+                sent = content.size();
->
+        }
->
+        else if (head) sio << "Content-Type: text/html; charset=UTF-8\r\n\r\n";
->
+        else sent = error(sio, code);
->
->
+        log << code << ' ' << (sent > 0 ? lexical_cast<string>(sent) : string("0"))
->
+                << " \"" << env.get("HTTP_REFERER") << "\" \""
->
+                << env.get("HTTP_USER_AGENT") << '"';
->
->
+        ofstream fout(this->log.c_str(), ios_base::app);
->
->
+        fout << ip(*client) << " - - " << date(true) << ' ' << log.str() << '\n';
+        delete client;
-+
-+
+void Daemon::headers(istream& sin, Environment& env)
-+
+{
-+
+        do
-+
+        {
-+
+                string line;
-+
-+
+                getline(sin, line);
-+
-+
+                if (line.empty()) break;
-+
-+
+                istringstream input(line);
-+
+                string name, value;
-+
-+
+                ::getline(input, name, ':');
-+
+                getline(input, value);
-+
-+
+                for (char next(sin.peek()); next == ' ' || next == '\t'; next =
-+
+                        sin.peek())
-+
+                {
-+
+                        getline(sin, line);
-+
-+
+                        value += ' ' + line;
-+
+                }
-+
-+
+                Matcher matcher("^\\s*(.+)\\s*$");
-+
-+
+                if (name == "Content-Length")
-+
+                {
-+
+                        if (value == matcher) env.set("CONTENT_LENGTH", matcher[1]);
-+
+                }
-+
+                else if (name == "Content-Type")
-+
+                {
-+
+                        if (value == matcher) env.set("CONTENT_TYPE", matcher[1]);
-+
+                }
-+
+                else if (name == "Referer")
-+
+                {
-+
+                        if (value == matcher) env.set("HTTP_REFERER", matcher[1]);
-+
+                }
-+
+                else if (name == "User-Agent")
-+
+                {
-+
+                        if (value == matcher) env.set("HTTP_USER_AGENT", matcher[1]);
-+
+                }
-+
+        }
-+
+        while (sin.good());
-+
+}
-+
-+
+Daemon::Status Daemon::message(istream& sin, Environment& env, ostream& post)
-+
+{
-+
+        string contentLength(env.get("CONTENT_LENGTH"));
-+
-+
+        if (env.get("CONTENT_TYPE") != "application/x-www-form-urlencoded") return
-+
+                mediaType;
-+
+        if (contentLength.empty()) return lengthRequired;
-+
-+
+        streamsize length(lexical_cast<streamsize>(contentLength));
-+
+        char* content(new char[length]);
-+
-+
+        sin.read(content, length);
-+
+        post.write(content, length);
-+
-+
+        delete [] content;
-+
-+
+        return ok;
-+
+}
-+
-+
+string Daemon::reason(Status status)
-+
+{
-+
+        ostringstream sout;
-+
-+
+        switch (status)
-+
+        {
-+
+        case ok:
-+
+                sout << "OK";
-+
+                break;
-+
+        case found:
-+
+                sout << "Found";
-+
+                break;
-+
+        case seeOther:
-+
+                sout << "See Other";
-+
+                break;
-+
+        case bad:
-+
+                sout << "Bad Request";
-+
+                break;
-+
+        case notFound:
-+
+                sout << "Not Found";
-+
+                break;
-+
+        case lengthRequired:
-+
+                sout << "Length Required";
-+
+                break;
-+
+        case mediaType:
-+
+                sout << "Unsupported Media Type";
-+
+                break;
-+
+        case serverError:
-+
+                sout << "Internal Server Error";
-+
+                break;
-+
+        case notImplemented:
-+
+                sout << "Not Implemented";
-+
+                break;
-+
+        case version:
-+
+                sout << "HTTP Version not supported";
-+
+        }
-+
-+
+        return sout.str();
-+
+}

Diff Legend

-–
+Removed lines
-+
+Added lines
-<
+Changed lines
->
+Changed lines

Comparing Smersh/Daemon.cpp (file contents): Revision 174 by Douglas Thrift, 2004-06-21T20:16:43-07:00 vs. Revision 177 by Douglas Thrift, 2004-06-26T01:26:41-07:00

Diff Legend

Comparing Smersh/Daemon.cpp (file contents):
Revision 174 by Douglas Thrift, 2004-06-21T20:16:43-07:00 vs.
Revision 177 by Douglas Thrift, 2004-06-26T01:26:41-07:00