ACE/protocols/ace/INet/HeaderBase.cpp

   1 #include "ace/INet/HeaderBase.h"
   2 #include "ace/OS_NS_stdlib.h"
   3
   4 #if !defined (__ACE_INLINE__)
   5 #include "ace/INet/HeaderBase.inl"
   6 #endif
   7
   8 #include "ace/INet/INet_Log.h"
   9
  10
  11 ACE_BEGIN_VERSIONED_NAMESPACE_DECL
  12
  13 namespace ACE
  14 {
  15   namespace INet
  16   {
  17     const int         HeaderBase::UNKNOWN_CONTENT_LENGTH     = -1;
  18     const ACE_CString HeaderBase::UNKNOWN_CONTENT_TYPE;
  19     const ACE_CString HeaderBase::CONTENT_LENGTH             = "Content-Length";
  20     const ACE_CString HeaderBase::CONTENT_TYPE               = "Content-Type";
  21     const ACE_CString HeaderBase::EMPTY;
  22
  23     const int HeaderBase::eof_ = std::char_traits<char>::eof();
  24
  25     HeaderBase::HeaderBase()
  26       {
  27       }
  28
  29     HeaderBase::~HeaderBase()
  30       {
  31       }
  32
  33     void HeaderBase::set_content_length(int length)
  34       {
  35         if (length == UNKNOWN_CONTENT_LENGTH)
  36           {
  37             this->remove (CONTENT_LENGTH);
  38           }
  39         else
  40           {
  41             char buf[32];
  42             this->set (CONTENT_LENGTH,
  43                        ACE_OS::itoa (length, buf, 10));
  44           }
  45       }
  46
  47     void HeaderBase::write(std::ostream& str) const
  48       {
  49         TNVMap::ITERATOR it (const_cast<TNVMap&> (this->header_values_));
  50         for (it.first (); !it.done () ;it.advance ())
  51         {
  52           str << (*it).first ().c_str () << ": " << (*it).second ().c_str () << "\r\n";
  53
  54           INET_DEBUG (9, (LM_DEBUG, DLINFO
  55                           ACE_TEXT ("ACE_INet_HTTP: +-> %C: %C\n"),
  56                           (*it).first ().c_str (),
  57                           (*it).second ().c_str ()));
  58         }
  59       }
  60
  61     bool HeaderBase::read(std::istream& str)
  62       {
  63         ACE_CString name (64, '\0');
  64         ACE_CString value (128, '\0');
  65         int ch = str.peek ();
  66         while (ch != eof_ && ch != '\r' && ch != '\n')
  67           {
  68             name.fast_clear ();
  69             value.fast_clear ();
  70             // parse name
  71             ch = this->read_field (str, name, MAX_NAME_LENGTH, ':');
  72             if (ch == '\n')
  73               {
  74                 ch = str.get ();
  75                 continue; // ignore invalid headers
  76               }
  77             if (ch != ':')
  78               {
  79                 return false; // name too long/missing colon; cannot continue
  80               }
  81
  82             // skip leading whitespace before next field
  83             while (ACE_OS::ace_isspace (str.peek ()))
  84               {
  85                 ch = str.get ();
  86               }
  87
  88             // parse value
  89             ch = this->read_field (str, value, MAX_VALUE_LENGTH, '\r');
  90             if (ch == '\r')
  91               ch = str.get (); // get lf
  92             if (ch != '\n')
  93               return false; // value too long/no crlf found; cannot continue
  94
  95             // followup lines starting with ws are continuations of the value
  96             // and must be appended
  97             ch = str.peek ();
  98             while (ch == ' ' || ch == '\t')
  99             {
 100               ch = this->read_field (str, value, MAX_VALUE_LENGTH, '\r');
 101               if (ch == '\r')
 102                 ch = str.get (); // get lf
 103               if (ch != '\n')
 104                 return false; // multiline value too long/no crlf; cannot continue
 105
 106               ch = str.peek ();
 107             }
 108
 109             this->add (name, value);
 110
 111             INET_DEBUG (9, (LM_DEBUG, DLINFO
 112                             ACE_TEXT ("ACE_INet_HTTP: <-+ %C: %C\n"),
 113                             name.c_str (),
 114                             value.c_str ()));
 115           }
 116         return true;
 117       }
 118
 119
 120     void HeaderBase::set (const ACE_CString& name, const ACE_CString& value)
 121       {
 122         TNVMap::ITERATOR it (this->header_values_);
 123         if (this->header_values_.find (NVPair (name), it) == 0)
 124           {
 125             (*it).second (value);
 126           }
 127         else
 128           {
 129             this->header_values_.insert (NVPair (name, value));
 130           }
 131       }
 132
 133     void HeaderBase::get_values(const ACE_CString& name, ACE_Array<ACE_CString> & values) const
 134       {
 135         TNVMap::ITERATOR it (const_cast<TNVMap&> (this->header_values_));
 136         if (this->header_values_.find (name, it) == 0)
 137           {
 138             for (; !it.done () && ((*it).second () == name) ;it.advance ())
 139               {
 140                 if (values.size (values.size ()+1) == 0)
 141                   {
 142                     values.set ((*it).second (), values.size ()-1);
 143                   }
 144               }
 145           }
 146       }
 147
 148
 149     NVPair::NVPair () {}
 150
 151     NVPair::NVPair (const ACE_CString& first)
 152       : first_ (first)
 153       {
 154       }
 155
 156     NVPair::NVPair (const ACE_CString& first, const ACE_CString& second)
 157       : first_ (first), second_ (second)
 158       {
 159       }
 160
 161     NVPair::NVPair (const NVPair& pair)
 162       {
 163         *this = pair;
 164       }
 165
 166     NVPair::~NVPair () {}
 167
 168   }
 169 }
 170
 171 ACE_END_VERSIONED_NAMESPACE_DECL