BodyPart::ptr Multipart::nextPart() { if (m_stream->supportsWrite()) { MORDOR_ASSERT(!m_finished); MORDOR_ASSERT(!m_currentPart); m_currentPart.reset(new BodyPart(shared_from_this())); std::string boundary = m_boundary + "\r\n"; m_stream->write(boundary.c_str(), boundary.size()); return m_currentPart; } else { if (m_finished) { MORDOR_ASSERT(!m_currentPart); return m_currentPart; } if (m_currentPart) { transferStream(m_currentPart->stream(), NullStream::get()); // Changed by the notification callback MORDOR_ASSERT(!m_currentPart); } size_t offsetToBoundary = m_stream->find(m_boundary); Buffer b; size_t result = m_stream->read(b, offsetToBoundary + m_boundary.size()); MORDOR_ASSERT(result == offsetToBoundary + m_boundary.size()); b.clear(); result = m_stream->read(b, 2); if (b == "--") { m_finished = true; } if (b == "\n") { m_stream->unread(b, 1); } if (b != "\r\n") { std::string restOfLine = m_stream->getDelimited(); MORDOR_ASSERT(!restOfLine.empty()); restOfLine.resize(restOfLine.size() - 1); if (restOfLine.find_first_not_of(" \r\t") != std::string::npos) { MORDOR_THROW_EXCEPTION(InvalidMultipartBoundaryException()); } } if (m_finished) { if (multipartFinished) multipartFinished(); return m_currentPart; } m_currentPart.reset(new BodyPart(shared_from_this())); return m_currentPart; } }
void Config::request(ServerRequest::ptr request, Access access) { const std::string &method = request->request().requestLine.method; if (method == POST) { if (access != READWRITE) { respondError(request, FORBIDDEN); return; } if (request->request().entity.contentType.type != "application" || request->request().entity.contentType.subtype != "x-www-form-urlencoded") { respondError(request, UNSUPPORTED_MEDIA_TYPE); return; } Stream::ptr requestStream = request->requestStream(); requestStream.reset(new LimitedStream(requestStream, 65536)); MemoryStream requestBody; transferStream(requestStream, requestBody); std::string queryString; queryString.resize(requestBody.buffer().readAvailable()); requestBody.buffer().copyOut(&queryString[0], requestBody.buffer().readAvailable()); bool failed = false; URI::QueryString qs(queryString); for (URI::QueryString::const_iterator it = qs.begin(); it != qs.end(); ++it) { ConfigVarBase::ptr var = Mordor::Config::lookup(it->first); if (var && !var->fromString(it->second)) failed = true; } if (failed) { respondError(request, HTTP::FORBIDDEN, "One or more new values were not accepted"); return; } // Fall through } if (method == GET || method == HEAD || method == POST) { Format format = HTML; URI::QueryString qs; if (request->request().requestLine.uri.queryDefined()) qs = request->request().requestLine.uri.queryString(); URI::QueryString::const_iterator it = qs.find("alt"); if (it != qs.end() && it->second == "json") format = JSON; // TODO: use Accept to indicate JSON switch (format) { case HTML: { request->response().status.status = OK; request->response().entity.contentType = MediaType("text", "html"); if (method == HEAD) { if (request->request().requestLine.ver == Version(1, 1) && isAcceptable(request->request().request.te, "chunked", true)) { request->response().general.transferEncoding.push_back("chunked"); } return; } Stream::ptr response = request->responseStream(); response.reset(new BufferedStream(response)); response->write("<html><body><table>\n", 20); Mordor::Config::visit(boost::bind(access == READWRITE ? &eachConfigVarHTMLWrite : &eachConfigVarHTML, _1, response)); response->write("</table></body></html>", 22); response->close(); break; } case JSON: { JSON::Object root; Mordor::Config::visit(boost::bind(&eachConfigVarJSON, _1, boost::ref(root))); std::ostringstream os; os << root; std::string str = os.str(); request->response().status.status = OK; request->response().entity.contentType = MediaType("application", "json"); request->response().entity.contentLength = str.size(); if (method != HEAD) { request->responseStream()->write(str.c_str(), str.size()); request->responseStream()->close(); } break; } default: MORDOR_NOTREACHED(); } } else { respondError(request, METHOD_NOT_ALLOWED); } }