| OLD | NEW |
| 1 // Copyright 2007, Google Inc. | 1 // Copyright 2007, Google Inc. |
| 2 // All rights reserved. | 2 // All rights reserved. |
| 3 // | 3 // |
| 4 // Redistribution and use in source and binary forms, with or without | 4 // Redistribution and use in source and binary forms, with or without |
| 5 // modification, are permitted provided that the following conditions are | 5 // modification, are permitted provided that the following conditions are |
| 6 // met: | 6 // met: |
| 7 // | 7 // |
| 8 // * Redistributions of source code must retain the above copyright | 8 // * Redistributions of source code must retain the above copyright |
| 9 // notice, this list of conditions and the following disclaimer. | 9 // notice, this list of conditions and the following disclaimer. |
| 10 // * Redistributions in binary form must reproduce the above | 10 // * Redistributions in binary form must reproduce the above |
| (...skipping 16 matching lines...) Expand all Loading... |
| 27 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | 27 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| 28 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | 28 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| 29 | 29 |
| 30 // This header file defines a canonicalizer output method class for STL | 30 // This header file defines a canonicalizer output method class for STL |
| 31 // strings. Because the canonicalizer tries not to be dependent on the STL, | 31 // strings. Because the canonicalizer tries not to be dependent on the STL, |
| 32 // we have segregated it here. | 32 // we have segregated it here. |
| 33 | 33 |
| 34 #ifndef GOOGLEURL_SRC_URL_CANON_STDSTRING_H__ | 34 #ifndef GOOGLEURL_SRC_URL_CANON_STDSTRING_H__ |
| 35 #define GOOGLEURL_SRC_URL_CANON_STDSTRING_H__ | 35 #define GOOGLEURL_SRC_URL_CANON_STDSTRING_H__ |
| 36 | 36 |
| 37 #include <string> | 37 #include "url/url_canon_stdstring.h" |
| 38 #include "googleurl/src/url_canon.h" | |
| 39 | |
| 40 namespace url_canon { | |
| 41 | |
| 42 // Write into a std::string given in the constructor. This object does not own | |
| 43 // the string itself, and the user must ensure that the string stays alive | |
| 44 // throughout the lifetime of this object. | |
| 45 // | |
| 46 // The given string will be appended to; any existing data in the string will | |
| 47 // be preserved. The caller should reserve() the amount of data in the string | |
| 48 // they expect to be written. We will resize if necessary, but that's slow. | |
| 49 // | |
| 50 // Note that when canonicalization is complete, the string will likely have | |
| 51 // unused space at the end because we make the string very big to start out | |
| 52 // with (by |initial_size|). This ends up being important because resize | |
| 53 // operations are slow, and because the base class needs to write directly | |
| 54 // into the buffer. | |
| 55 // | |
| 56 // Therefore, the user should call Complete() before using the string that | |
| 57 // this class wrote into. | |
| 58 class StdStringCanonOutput : public CanonOutput { | |
| 59 public: | |
| 60 StdStringCanonOutput(std::string* str) | |
| 61 : CanonOutput(), | |
| 62 str_(str) { | |
| 63 cur_len_ = static_cast<int>(str_->size()); // Append to existing data. | |
| 64 str_->resize(str_->capacity()); | |
| 65 buffer_ = str_->empty() ? NULL : &(*str_)[0]; | |
| 66 buffer_len_ = static_cast<int>(str_->size()); | |
| 67 } | |
| 68 virtual ~StdStringCanonOutput() { | |
| 69 // Nothing to do, we don't own the string. | |
| 70 } | |
| 71 | |
| 72 // Must be called after writing has completed but before the string is used. | |
| 73 void Complete() { | |
| 74 str_->resize(cur_len_); | |
| 75 buffer_len_ = cur_len_; | |
| 76 } | |
| 77 | |
| 78 virtual void Resize(int sz) { | |
| 79 str_->resize(sz); | |
| 80 buffer_ = str_->empty() ? NULL : &(*str_)[0]; | |
| 81 buffer_len_ = sz; | |
| 82 } | |
| 83 | |
| 84 protected: | |
| 85 std::string* str_; | |
| 86 }; | |
| 87 | |
| 88 // An extension of the Replacements class that allows the setters to use | |
| 89 // standard strings. | |
| 90 // | |
| 91 // The strings passed as arguments are not copied and must remain valid until | |
| 92 // this class goes out of scope. | |
| 93 template<typename STR> | |
| 94 class StdStringReplacements : | |
| 95 public url_canon::Replacements<typename STR::value_type> { | |
| 96 public: | |
| 97 void SetSchemeStr(const STR& s) { | |
| 98 this->SetScheme(s.data(), | |
| 99 url_parse::Component(0, static_cast<int>(s.length()))); | |
| 100 } | |
| 101 void SetUsernameStr(const STR& s) { | |
| 102 this->SetUsername(s.data(), | |
| 103 url_parse::Component(0, static_cast<int>(s.length()))); | |
| 104 } | |
| 105 void SetPasswordStr(const STR& s) { | |
| 106 this->SetPassword(s.data(), | |
| 107 url_parse::Component(0, static_cast<int>(s.length()))); | |
| 108 } | |
| 109 void SetHostStr(const STR& s) { | |
| 110 this->SetHost(s.data(), | |
| 111 url_parse::Component(0, static_cast<int>(s.length()))); | |
| 112 } | |
| 113 void SetPortStr(const STR& s) { | |
| 114 this->SetPort(s.data(), | |
| 115 url_parse::Component(0, static_cast<int>(s.length()))); | |
| 116 } | |
| 117 void SetPathStr(const STR& s) { | |
| 118 this->SetPath(s.data(), | |
| 119 url_parse::Component(0, static_cast<int>(s.length()))); | |
| 120 } | |
| 121 void SetQueryStr(const STR& s) { | |
| 122 this->SetQuery(s.data(), | |
| 123 url_parse::Component(0, static_cast<int>(s.length()))); | |
| 124 } | |
| 125 void SetRefStr(const STR& s) { | |
| 126 this->SetRef(s.data(), | |
| 127 url_parse::Component(0, static_cast<int>(s.length()))); | |
| 128 } | |
| 129 }; | |
| 130 | |
| 131 } // namespace url_canon | |
| 132 | 38 |
| 133 #endif // GOOGLEURL_SRC_URL_CANON_STDSTRING_H__ | 39 #endif // GOOGLEURL_SRC_URL_CANON_STDSTRING_H__ |
| 134 | 40 |
| OLD | NEW |