00001 
00002 
00003 
00004 
00005 
00006 
00007 
00008 
00009 
00010 #ifndef __PION_HTTP_RESPONSE_HEADER__
00011 #define __PION_HTTP_RESPONSE_HEADER__
00012 
00013 #include <boost/shared_ptr.hpp>
00014 #include <boost/lexical_cast.hpp>
00015 #include <pion/config.hpp>
00016 #include <pion/http/message.hpp>
00017 #include <pion/http/request.hpp>
00018 
00019 
00020 namespace pion {    
00021 namespace http {    
00022 
00023     
00027 class response
00028     : public http::message
00029 {
00030 public:
00031 
00037     response(const http::request& http_request_ptr)
00038         : m_status_code(RESPONSE_CODE_OK),
00039         m_status_message(RESPONSE_MESSAGE_OK)
00040     {
00041         update_request_info(http_request_ptr);
00042     }
00043 
00049     response(const std::string& request_method)
00050         : m_status_code(RESPONSE_CODE_OK), m_status_message(RESPONSE_MESSAGE_OK),
00051         m_request_method(request_method)
00052     {}
00053     
00055     response(const response& http_response)
00056         : message(http_response),
00057         m_status_code(http_response.m_status_code),
00058         m_status_message(http_response.m_status_message),
00059         m_request_method(http_response.m_request_method)
00060     {}
00061     
00064     response(void)
00065         : m_status_code(RESPONSE_CODE_OK),
00066         m_status_message(RESPONSE_MESSAGE_OK)
00067     {}
00068     
00070     virtual ~response() {}
00071 
00073     virtual void clear(void) {
00074         http::message::clear();
00075         m_status_code = RESPONSE_CODE_OK;
00076         m_status_message = RESPONSE_MESSAGE_OK;
00077         m_request_method.clear();
00078     }
00079 
00081     virtual bool is_content_length_implied(void) const {
00082         return (m_request_method == REQUEST_METHOD_HEAD             
00083                 || (m_status_code >= 100 && m_status_code <= 199)       
00084                 || m_status_code == 204 || m_status_code == 205     
00085                 || m_status_code == 304                             
00086                 );
00087     }
00088 
00095     inline void update_request_info(const http::request& http_request) {
00096         m_request_method = http_request.get_method();
00097         if (http_request.get_version_major() == 1 && http_request.get_version_minor() >= 1)
00098             set_chunks_supported(true);
00099     }
00100     
00102     inline void set_status_code(unsigned int n) {
00103         m_status_code = n;
00104         clear_first_line();
00105     }
00106 
00108     inline void set_status_message(const std::string& msg) {
00109         m_status_message = msg;
00110         clear_first_line();
00111     }
00112     
00114     inline unsigned int get_status_code(void) const { return m_status_code; }
00115     
00117     inline const std::string& get_status_message(void) const { return m_status_message; }
00118     
00119 
00127     inline void set_cookie(const std::string& name, const std::string& value) {
00128         std::string set_cookie_header(make_set_cookie_header(name, value, "/"));
00129         add_header(HEADER_SET_COOKIE, set_cookie_header);
00130     }
00131     
00140     inline void set_cookie(const std::string& name, const std::string& value,
00141                           const std::string& path)
00142     {
00143         std::string set_cookie_header(make_set_cookie_header(name, value, path));
00144         add_header(HEADER_SET_COOKIE, set_cookie_header);
00145     }
00146     
00155     inline void set_cookie(const std::string& name, const std::string& value,
00156                           const std::string& path, const unsigned long max_age)
00157     {
00158         std::string set_cookie_header(make_set_cookie_header(name, value, path, true, max_age));
00159         add_header(HEADER_SET_COOKIE, set_cookie_header);
00160     }
00161     
00169     inline void set_cookie(const std::string& name, const std::string& value,
00170                           const unsigned long max_age)
00171     {
00172         std::string set_cookie_header(make_set_cookie_header(name, value, "/", true, max_age));
00173         add_header(HEADER_SET_COOKIE, set_cookie_header);
00174     }
00175     
00177     inline void delete_cookie(const std::string& name) {
00178         std::string set_cookie_header(make_set_cookie_header(name, "", "/", true, 0));
00179         add_header(HEADER_SET_COOKIE, set_cookie_header);
00180     }
00181     
00183     inline void delete_cookie(const std::string& name, const std::string& path) {
00184         std::string set_cookie_header(make_set_cookie_header(name, "", path, true, 0));
00185         add_header(HEADER_SET_COOKIE, set_cookie_header);
00186     }
00187     
00189     inline void set_last_modified(const unsigned long t) {
00190         change_header(HEADER_LAST_MODIFIED, get_date_string(t));
00191     }
00192     
00193     
00194 protected:
00195     
00197     virtual void update_first_line(void) const {
00198         
00199         m_first_line = get_version_string();
00200         m_first_line += ' ';
00201         
00202         m_first_line +=  boost::lexical_cast<std::string>(m_status_code);
00203         m_first_line += ' ';
00204         
00205         m_first_line += m_status_message;
00206     }
00207     
00209     virtual void append_cookie_headers(void) {
00210         for (ihash_multimap::const_iterator i = get_cookies().begin(); i != get_cookies().end(); ++i) {
00211             set_cookie(i->first, i->second);
00212         }
00213     }
00214 
00215     
00216 private:
00217 
00219     unsigned int            m_status_code;
00220     
00222     std::string             m_status_message;
00223     
00225     std::string             m_request_method;
00226 };
00227 
00228 
00230 typedef boost::shared_ptr<response>     response_ptr;
00231 
00232 
00233 }   
00234 }   
00235 
00236 #endif