pion-net  4.0.9
net/services/EchoService.cpp
00001 // ------------------------------------------------------------------
00002 // pion-net: a C++ framework for building lightweight HTTP interfaces
00003 // ------------------------------------------------------------------
00004 // Copyright (C) 2007-2008 Atomic Labs, Inc.  (http://www.atomiclabs.com)
00005 //
00006 // Distributed under the Boost Software License, Version 1.0.
00007 // See http://www.boost.org/LICENSE_1_0.txt
00008 //
00009 
00010 #include "EchoService.hpp"
00011 #include <boost/bind.hpp>
00012 #include <pion/PionAlgorithms.hpp>
00013 #include <pion/net/HTTPResponseWriter.hpp>
00014 #include <pion/net/PionUser.hpp>
00015 
00016 using namespace pion;
00017 using namespace pion::net;
00018 
00019 namespace pion {        // begin namespace pion
00020 namespace plugins {     // begin namespace plugins
00021 
00022     
00024 void writeDictionaryTerm(HTTPResponseWriterPtr& writer,
00025                          const HTTPTypes::QueryParams::value_type& val,
00026                          const bool decode)
00027 {
00028     // text is copied into writer text cache
00029     writer << val.first << HTTPTypes::HEADER_NAME_VALUE_DELIMITER
00030     << (decode ? algo::url_decode(val.second) : val.second)
00031     << HTTPTypes::STRING_CRLF;
00032 }
00033 
00034 
00035 // EchoService member functions
00036 
00038 void EchoService::operator()(HTTPRequestPtr& request, TCPConnectionPtr& tcp_conn)
00039 {
00040     // this web service uses static text to test the mixture of "copied" with
00041     // "static" (no-copy) text
00042     static const std::string REQUEST_ECHO_TEXT("[Request Echo]");
00043     static const std::string REQUEST_HEADERS_TEXT("[Request Headers]");
00044     static const std::string QUERY_PARAMS_TEXT("[Query Parameters]");
00045     static const std::string COOKIE_PARAMS_TEXT("[Cookie Parameters]");
00046     static const std::string POST_CONTENT_TEXT("[POST Content]");
00047     static const std::string USER_INFO_TEXT("[USER Info]");
00048     
00049     // Set Content-type to "text/plain" (plain ascii text)
00050     HTTPResponseWriterPtr writer(HTTPResponseWriter::create(tcp_conn, *request,
00051                                                             boost::bind(&TCPConnection::finish, tcp_conn)));
00052     writer->getResponse().setContentType(HTTPTypes::CONTENT_TYPE_TEXT);
00053     
00054     // write request information
00055     writer->writeNoCopy(REQUEST_ECHO_TEXT);
00056     writer->writeNoCopy(HTTPTypes::STRING_CRLF);
00057     writer->writeNoCopy(HTTPTypes::STRING_CRLF);
00058     writer
00059         << "Request method: "
00060         << request->getMethod()
00061         << HTTPTypes::STRING_CRLF
00062         << "Resource originally requested: "
00063         << request->getOriginalResource()
00064         << HTTPTypes::STRING_CRLF
00065         << "Resource delivered: "
00066         << request->getResource()
00067         << HTTPTypes::STRING_CRLF
00068         << "Query string: "
00069         << request->getQueryString()
00070         << HTTPTypes::STRING_CRLF
00071         << "HTTP version: "
00072         << request->getVersionMajor() << '.' << request->getVersionMinor()
00073         << HTTPTypes::STRING_CRLF
00074         << "Content length: "
00075         << (unsigned long)request->getContentLength()
00076         << HTTPTypes::STRING_CRLF
00077         << HTTPTypes::STRING_CRLF;
00078              
00079     // write request headers
00080     writer->writeNoCopy(REQUEST_HEADERS_TEXT);
00081     writer->writeNoCopy(HTTPTypes::STRING_CRLF);
00082     writer->writeNoCopy(HTTPTypes::STRING_CRLF);
00083     std::for_each(request->getHeaders().begin(), request->getHeaders().end(),
00084                   boost::bind(&writeDictionaryTerm, writer, _1, false));
00085     writer->writeNoCopy(HTTPTypes::STRING_CRLF);
00086 
00087     // write query parameters
00088     writer->writeNoCopy(QUERY_PARAMS_TEXT);
00089     writer->writeNoCopy(HTTPTypes::STRING_CRLF);
00090     writer->writeNoCopy(HTTPTypes::STRING_CRLF);
00091     std::for_each(request->getQueryParams().begin(), request->getQueryParams().end(),
00092                   boost::bind(&writeDictionaryTerm, writer, _1, true));
00093     writer->writeNoCopy(HTTPTypes::STRING_CRLF);
00094     
00095     // write cookie parameters
00096     writer->writeNoCopy(COOKIE_PARAMS_TEXT);
00097     writer->writeNoCopy(HTTPTypes::STRING_CRLF);
00098     writer->writeNoCopy(HTTPTypes::STRING_CRLF);
00099     std::for_each(request->getCookieParams().begin(), request->getCookieParams().end(),
00100                   boost::bind(&writeDictionaryTerm, writer, _1, false));
00101     writer->writeNoCopy(HTTPTypes::STRING_CRLF);
00102     
00103     // write POST content
00104     writer->writeNoCopy(POST_CONTENT_TEXT);
00105     writer->writeNoCopy(HTTPTypes::STRING_CRLF);
00106     writer->writeNoCopy(HTTPTypes::STRING_CRLF);
00107     if (request->getContentLength() != 0) {
00108         writer->write(request->getContent(), request->getContentLength());
00109         writer->writeNoCopy(HTTPTypes::STRING_CRLF);
00110         writer->writeNoCopy(HTTPTypes::STRING_CRLF);
00111     }
00112     
00113     // if authenticated, write user info
00114     PionUserPtr user = request->getUser();
00115     if (user) {
00116         writer->writeNoCopy(USER_INFO_TEXT);
00117         writer->writeNoCopy(HTTPTypes::STRING_CRLF);
00118         writer->writeNoCopy(HTTPTypes::STRING_CRLF);
00119         writer << "User authenticated, username: " << user->getUsername();
00120         writer->writeNoCopy(HTTPTypes::STRING_CRLF);
00121     }
00122     
00123     // send the writer
00124     writer->send();
00125 }
00126 
00127 
00128 }   // end namespace plugins
00129 }   // end namespace pion
00130 
00131 
00133 extern "C" PION_SERVICE_API pion::plugins::EchoService *pion_create_EchoService(void)
00134 {
00135     return new pion::plugins::EchoService();
00136 }
00137 
00139 extern "C" PION_SERVICE_API void pion_destroy_EchoService(pion::plugins::EchoService *service_ptr)
00140 {
00141     delete service_ptr;
00142 }