/* */ #include "HttpServer.h" #include #include "HttpHeader.h" #include "SocketCore.h" #include "HttpHeaderProcessor.h" #include "DlAbortEx.h" #include "message.h" #include "util.h" #include "LogFactory.h" #include "Logger.h" #include "Base64.h" #include "a2functional.h" namespace aria2 { HttpServer::HttpServer(const SharedHandle& socket, DownloadEngine* e): socket_(socket), socketBuffer_(socket), e_(e), headerProcessor_(new HttpHeaderProcessor()), logger_(LogFactory::getInstance()), keepAlive_(true), gzip_(false), acceptsPersistentConnection_(true), acceptsGZip_(false) {} HttpServer::~HttpServer() {} SharedHandle HttpServer::receiveRequest() { size_t size = 512; unsigned char buf[size]; socket_->peekData(buf, size); if(size == 0 && !(socket_->wantRead() || socket_->wantWrite())) { throw DL_ABORT_EX(EX_EOF_FROM_PEER); } headerProcessor_->update(buf, size); if(!headerProcessor_->eoh()) { socket_->readData(buf, size); return SharedHandle(); } size_t putbackDataLength = headerProcessor_->getPutBackDataLength(); size -= putbackDataLength; socket_->readData(buf, size); SharedHandle header = headerProcessor_->getHttpRequestHeader(); if(!header.isNull()) { logger_->info("HTTP Server received request\n%s", headerProcessor_->getHeaderString().c_str()); lastRequestHeader_ = header; lastBody_.clear(); lastBody_.str(""); lastContentLength_ = lastRequestHeader_->getFirstAsUInt(HttpHeader::CONTENT_LENGTH); headerProcessor_->clear(); std::string connection = util::toLower(lastRequestHeader_->getFirst(HttpHeader::CONNECTION)); acceptsPersistentConnection_ = connection.find(HttpHeader::CLOSE) == std::string::npos && (lastRequestHeader_->getVersion() == HttpHeader::HTTP_1_1 || connection.find("keep-alive") != std::string::npos); std::vector acceptEncodings; util::split(lastRequestHeader_->getFirst(HttpHeader::ACCEPT_ENCODING), std::back_inserter(acceptEncodings), A2STR::COMMA_C, true); acceptsGZip_ = std::find(acceptEncodings.begin(), acceptEncodings.end(), "gzip") != acceptEncodings.end(); } return header; } bool HttpServer::receiveBody() { if(lastContentLength_ == 0) { return true; } const size_t BUFLEN = 4096; char buf[BUFLEN]; size_t length = std::min(BUFLEN, static_cast (lastContentLength_-lastBody_.tellg())); socket_->readData(buf, length); if(length == 0 && !(socket_->wantRead() || socket_->wantWrite())) { throw DL_ABORT_EX(EX_EOF_FROM_PEER); } lastBody_.write(buf, length); return lastContentLength_ == static_cast(lastBody_.tellp()); } std::string HttpServer::getBody() const { return lastBody_.str(); } const std::string& HttpServer::getRequestPath() const { return lastRequestHeader_->getRequestPath(); } void HttpServer::feedResponse(const std::string& text, const std::string& contentType) { feedResponse("200 OK", "", text, contentType); } void HttpServer::feedResponse(const std::string& status, const std::string& headers, const std::string& text, const std::string& contentType) { std::string header = "HTTP/1.1 "; strappend(header, status, "\r\n", "Content-Type: ", contentType, "\r\n", "Content-Length: ", util::uitos(text.size()), "\r\n"); if(supportsGZip()) { header += "Content-Encoding: gzip\r\n"; } if(!supportsPersistentConnection()) { header += "Connection: close\r\n"; } if(!headers.empty()) { header += headers; if(!util::endsWith(headers, "\r\n")) { header += "\r\n"; } } header += "\r\n"; if(logger_->debug()) { logger_->debug("HTTP Server sends response:\n%s", header.c_str()); } socketBuffer_.pushStr(header); socketBuffer_.pushStr(text); } ssize_t HttpServer::sendResponse() { return socketBuffer_.send(); } bool HttpServer::sendBufferIsEmpty() const { return socketBuffer_.sendBufferIsEmpty(); } bool HttpServer::authenticate() { if(username_.empty()) { return true; } std::string authHeader = lastRequestHeader_->getFirst("Authorization"); if(authHeader.empty()) { return false; } std::pair p = util::split(authHeader, " "); if(p.first != "Basic") { return false; } std::string userpass = Base64::decode(p.second); std::pair userpassPair = util::split(userpass, ":"); return username_ == userpassPair.first && password_ == userpassPair.second; } } // namespace aria2