// Copyright 2021 Jeisson Hidalgo-Cespedes. Universidad de Costa Rica. CC BY 4.0 #include #include #include #include #include #include "NetworkAddress.hpp" #include "Socket.hpp" #include "WebServer.hpp" const char* const usage = "Usage: webserv [port] [max_connections]\n" "\n" " port Network port to listen incoming HTTP requests, default " DEFAULT_PORT "\n" " max_connections Maximum number of allowed client connections\n"; // TODO(you) Make WebServer a singleton class. See the Log class WebServer::WebServer() { } WebServer::~WebServer() { } int WebServer::start(int argc, char* argv[]) { try { if (this->analyzeArguments(argc, argv)) { // TODO(you) Handle signal 2 (SIGINT) and 15 (SIGTERM), see man signal // Signal handler should call WebServer::stopListening(), send stop // conditions and wait for all secondary threads that it created this->listenForConnections(this->port); const NetworkAddress& address = this->getNetworkAddress(); std::cout << "web server listening on " << address.getIP() << " port " << address.getPort() << "...\n"; this->acceptAllConnections(); } } catch (const std::runtime_error& error) { std::cerr << "error: " << error.what() << std::endl; } return EXIT_SUCCESS; } bool WebServer::analyzeArguments(int argc, char* argv[]) { // Traverse all arguments for (int index = 1; index < argc; ++index) { const std::string argument = argv[index]; if (argument.find("help") != std::string::npos) { std::cout << usage; return false; } } if (argc >= 2) { this->port = argv[1]; } return true; } bool WebServer::handleHttpRequest(HttpRequest& httpRequest, HttpResponse& httpResponse) { // Print IP and port from client const NetworkAddress& address = httpRequest.getNetworkAddress(); std::cout << "connection established with client " << address.getIP() << " port " << address.getPort() << std::endl; // Print HTTP request std::cout << " " << httpRequest.getMethod() << ' ' << httpRequest.getURI() << ' ' << httpRequest.getHttpVersion() << std::endl; return this->route(httpRequest, httpResponse); } bool WebServer::route(HttpRequest& httpRequest, HttpResponse& httpResponse) { // If the home page was asked if (httpRequest.getMethod() == "GET" && httpRequest.getURI() == "/") { return this->serveHomepage(httpRequest, httpResponse); } // TODO(you): URI can be a multi-value list, e.g: 100,2784,-53,200771728 // TODO(you): change for sendGoldbachSums() if you prefer it std::smatch matches; // TODO(you): Numbers given by user may be larger than int64_t, reject them // If a number was asked in the form "/goldbach/1223" // or "/goldbach?number=1223" std::regex inQuery("^/goldbach(/|\\?number=)(\\d+)$"); if (std::regex_search(httpRequest.getURI(), matches, inQuery)) { assert(matches.length() >= 3); const int64_t number = std::stoll(matches[2]); return this->serveGoldbachSums(httpRequest, httpResponse, number); } // Unrecognized request return this->serveNotFound(httpRequest, httpResponse); } // TODO(you): Fix code redundancy in the following methods bool WebServer::serveHomepage(HttpRequest& httpRequest , HttpResponse& httpResponse) { (void)httpRequest; // TODO(you) Move form to your view class, e.g GoldbachWebApp // Set HTTP response metadata (headers) httpResponse.setHeader("Server", "AttoServer v1.0"); httpResponse.setHeader("Content-type", "text/html; charset=ascii"); // Build the body of the response std::string title = "Goldbach sums"; httpResponse.body() << "\n" << "\n" << " \n" << " " << title << "\n" << " \n" << "

" << title << "

\n" << " \n" << "\n"; // Send the response to the client (user agent) return httpResponse.send(); } bool WebServer::serveNotFound(HttpRequest& httpRequest , HttpResponse& httpResponse) { (void)httpRequest; // Set HTTP response metadata (headers) httpResponse.setStatusCode(404); httpResponse.setHeader("Server", "AttoServer v1.0"); httpResponse.setHeader("Content-type", "text/html; charset=ascii"); // Build the body of the response std::string title = "Not found"; httpResponse.body() << "\n" << "\n" << " \n" << " " << title << "\n" << " \n" << "

" << title << "

\n" << "

The requested resouce was not found in this server.

\n" << "

Homepage

\n" << "\n"; // Send the response to the client (user agent) return httpResponse.send(); } // TODO(you) Move domain-logic from WebServer controller to a view class // e.g GoldbachWebApp, and a model class e.g GoldbachCalculator bool WebServer::serveGoldbachSums(HttpRequest& httpRequest , HttpResponse& httpResponse, int64_t number) { (void)httpRequest; // Set HTTP response metadata (headers) httpResponse.setHeader("Server", "AttoServer v1.0"); httpResponse.setHeader("Content-type", "text/html; charset=ascii"); // Build the body of the response std::string title = "Goldbach sums for " + std::to_string(number); httpResponse.body() << "\n" << "\n" << " \n" << " " << title << "\n" << " \n" << "

" << title << "

\n" << "

25

\n" << "

25: 5 sums

\n" << "

-3

\n" << "

-3: NA

\n" << "

-13

\n" << "

-13: 2 sums

\n" << "

3 + 3 + 7
3 + 5 + 5

\n" << "

Back

\n" << "\n"; // Send the response to the client (user agent) return httpResponse.send(); }