| 1 | // |
| 2 | // HTTPAuthenticationParams.cpp |
| 3 | // |
| 4 | // Library: Net |
| 5 | // Package: HTTP |
| 6 | // Module: HTTPAuthenticationParams |
| 7 | // |
| 8 | // Copyright (c) 2011, Anton V. Yabchinskiy (arn at bestmx dot ru). |
| 9 | // Copyright (c) 2012, Applied Informatics Software Engineering GmbH. |
| 10 | // and Contributors. |
| 11 | // |
| 12 | // SPDX-License-Identifier: BSL-1.0 |
| 13 | // |
| 14 | |
| 15 | |
| 16 | #include "Poco/Net/HTTPAuthenticationParams.h" |
| 17 | #include "Poco/Net/HTTPRequest.h" |
| 18 | #include "Poco/Net/HTTPResponse.h" |
| 19 | #include "Poco/Net/NetException.h" |
| 20 | #include "Poco/String.h" |
| 21 | #include "Poco/Ascii.h" |
| 22 | #include "Poco/Exception.h" |
| 23 | |
| 24 | |
| 25 | using Poco::icompare; |
| 26 | using Poco::Ascii; |
| 27 | |
| 28 | |
| 29 | namespace |
| 30 | { |
| 31 | bool mustBeQuoted(const std::string& name) |
| 32 | { |
| 33 | return |
| 34 | icompare(name, "cnonce" ) == 0 || |
| 35 | icompare(name, "domain" ) == 0 || |
| 36 | icompare(name, "nonce" ) == 0 || |
| 37 | icompare(name, "opaque" ) == 0 || |
| 38 | icompare(name, "qop" ) == 0 || |
| 39 | icompare(name, "realm" ) == 0 || |
| 40 | icompare(name, "response" ) == 0 || |
| 41 | icompare(name, "uri" ) == 0 || |
| 42 | icompare(name, "username" ) == 0; |
| 43 | } |
| 44 | |
| 45 | |
| 46 | void (std::string& result, const std::string& name, const std::string& value) |
| 47 | { |
| 48 | result += name; |
| 49 | result += '='; |
| 50 | if (mustBeQuoted(name)) |
| 51 | { |
| 52 | result += '"'; |
| 53 | result += value; |
| 54 | result += '"'; |
| 55 | } |
| 56 | else |
| 57 | { |
| 58 | result += value; |
| 59 | } |
| 60 | } |
| 61 | } |
| 62 | |
| 63 | |
| 64 | namespace Poco { |
| 65 | namespace Net { |
| 66 | |
| 67 | |
| 68 | const std::string HTTPAuthenticationParams::REALM("realm" ); |
| 69 | const std::string HTTPAuthenticationParams::NTLM("NTLM" ); |
| 70 | const std::string HTTPAuthenticationParams::WWW_AUTHENTICATE("WWW-Authenticate" ); |
| 71 | const std::string HTTPAuthenticationParams::PROXY_AUTHENTICATE("Proxy-Authenticate" ); |
| 72 | |
| 73 | |
| 74 | HTTPAuthenticationParams::HTTPAuthenticationParams() |
| 75 | { |
| 76 | } |
| 77 | |
| 78 | |
| 79 | HTTPAuthenticationParams::HTTPAuthenticationParams(const std::string& authInfo) |
| 80 | { |
| 81 | fromAuthInfo(authInfo); |
| 82 | } |
| 83 | |
| 84 | |
| 85 | HTTPAuthenticationParams::HTTPAuthenticationParams(const HTTPRequest& request) |
| 86 | { |
| 87 | fromRequest(request); |
| 88 | } |
| 89 | |
| 90 | |
| 91 | HTTPAuthenticationParams::HTTPAuthenticationParams(const HTTPResponse& response, const std::string& ) |
| 92 | { |
| 93 | fromResponse(response, header); |
| 94 | } |
| 95 | |
| 96 | |
| 97 | HTTPAuthenticationParams::~HTTPAuthenticationParams() |
| 98 | { |
| 99 | } |
| 100 | |
| 101 | |
| 102 | HTTPAuthenticationParams& HTTPAuthenticationParams::operator = (const HTTPAuthenticationParams& authParams) |
| 103 | { |
| 104 | NameValueCollection::operator = (authParams); |
| 105 | |
| 106 | return *this; |
| 107 | } |
| 108 | |
| 109 | |
| 110 | void HTTPAuthenticationParams::fromAuthInfo(const std::string& authInfo) |
| 111 | { |
| 112 | parse(authInfo.begin(), authInfo.end()); |
| 113 | } |
| 114 | |
| 115 | |
| 116 | void HTTPAuthenticationParams::fromRequest(const HTTPRequest& request) |
| 117 | { |
| 118 | std::string scheme; |
| 119 | std::string authInfo; |
| 120 | |
| 121 | request.getCredentials(scheme, authInfo); |
| 122 | |
| 123 | if (icompare(scheme, "Digest" ) != 0) |
| 124 | throw InvalidArgumentException("Could not parse non-Digest authentication information" , scheme); |
| 125 | |
| 126 | fromAuthInfo(authInfo); |
| 127 | } |
| 128 | |
| 129 | |
| 130 | void HTTPAuthenticationParams::fromResponse(const HTTPResponse& response, const std::string& ) |
| 131 | { |
| 132 | NameValueCollection::ConstIterator it = response.find(header); |
| 133 | if (it == response.end()) |
| 134 | throw NotAuthenticatedException("HTTP response has no authentication header" ); |
| 135 | |
| 136 | bool found = false; |
| 137 | while (!found && it != response.end() && icompare(it->first, header) == 0) |
| 138 | { |
| 139 | const std::string& = it->second; |
| 140 | if (icompare(headerValue, 0, 6, "Basic " ) == 0) |
| 141 | { |
| 142 | parse(headerValue.begin() + 6, headerValue.end()); |
| 143 | found = true; |
| 144 | } |
| 145 | else if (icompare(headerValue, 0, 7, "Digest " ) == 0) |
| 146 | { |
| 147 | parse(headerValue.begin() + 7, headerValue.end()); |
| 148 | found = true; |
| 149 | } |
| 150 | else if (icompare(headerValue, 0, 5, "NTLM " ) == 0) |
| 151 | { |
| 152 | set(NTLM, headerValue.substr(5)); |
| 153 | found = true; |
| 154 | } |
| 155 | ++it; |
| 156 | } |
| 157 | if (!found) throw NotAuthenticatedException("No Basic, Digest or NTLM authentication header found" ); |
| 158 | } |
| 159 | |
| 160 | |
| 161 | const std::string& HTTPAuthenticationParams::getRealm() const |
| 162 | { |
| 163 | return get(REALM); |
| 164 | } |
| 165 | |
| 166 | |
| 167 | void HTTPAuthenticationParams::setRealm(const std::string& realm) |
| 168 | { |
| 169 | set(REALM, realm); |
| 170 | } |
| 171 | |
| 172 | |
| 173 | std::string HTTPAuthenticationParams::toString() const |
| 174 | { |
| 175 | std::string result; |
| 176 | if (size() == 1 && find(NTLM) != end()) |
| 177 | { |
| 178 | result = get(NTLM); |
| 179 | } |
| 180 | else |
| 181 | { |
| 182 | ConstIterator iter = begin(); |
| 183 | |
| 184 | if (iter != end()) |
| 185 | { |
| 186 | formatParameter(result, iter->first, iter->second); |
| 187 | ++iter; |
| 188 | } |
| 189 | |
| 190 | for (; iter != end(); ++iter) |
| 191 | { |
| 192 | result.append(", " ); |
| 193 | formatParameter(result, iter->first, iter->second); |
| 194 | } |
| 195 | } |
| 196 | return result; |
| 197 | } |
| 198 | |
| 199 | |
| 200 | void HTTPAuthenticationParams::parse(std::string::const_iterator first, std::string::const_iterator last) |
| 201 | { |
| 202 | enum State |
| 203 | { |
| 204 | STATE_INITIAL = 0x0100, |
| 205 | STATE_FINAL = 0x0200, |
| 206 | |
| 207 | STATE_SPACE = STATE_INITIAL | 0, |
| 208 | STATE_TOKEN = 1, |
| 209 | STATE_EQUALS = 2, |
| 210 | STATE_VALUE = STATE_FINAL | 3, |
| 211 | STATE_VALUE_QUOTED = 4, |
| 212 | STATE_VALUE_ESCAPE = 5, |
| 213 | STATE_COMMA = STATE_FINAL | 6 |
| 214 | }; |
| 215 | |
| 216 | int state = STATE_SPACE; |
| 217 | std::string token; |
| 218 | std::string value; |
| 219 | |
| 220 | for (std::string::const_iterator it = first; it != last; ++it) |
| 221 | { |
| 222 | switch (state) |
| 223 | { |
| 224 | case STATE_SPACE: |
| 225 | if (Ascii::isAlphaNumeric(*it) || *it == '_' || *it == '-') |
| 226 | { |
| 227 | token += *it; |
| 228 | state = STATE_TOKEN; |
| 229 | } |
| 230 | else if (Ascii::isSpace(*it)) |
| 231 | { |
| 232 | // Skip |
| 233 | } |
| 234 | else throw SyntaxException("Invalid authentication information" ); |
| 235 | break; |
| 236 | |
| 237 | case STATE_TOKEN: |
| 238 | if (*it == '=') |
| 239 | { |
| 240 | state = STATE_EQUALS; |
| 241 | } |
| 242 | else if (Ascii::isAlphaNumeric(*it) || *it == '_') |
| 243 | { |
| 244 | token += *it; |
| 245 | } |
| 246 | else throw SyntaxException("Invalid authentication information" ); |
| 247 | break; |
| 248 | |
| 249 | case STATE_EQUALS: |
| 250 | if (Ascii::isAlphaNumeric(*it) || *it == '_' || *it == '-') |
| 251 | { |
| 252 | value += *it; |
| 253 | state = STATE_VALUE; |
| 254 | } |
| 255 | else if (*it == '"') |
| 256 | { |
| 257 | state = STATE_VALUE_QUOTED; |
| 258 | } |
| 259 | else throw SyntaxException("Invalid authentication information" ); |
| 260 | break; |
| 261 | |
| 262 | case STATE_VALUE_QUOTED: |
| 263 | if (*it == '\\') |
| 264 | { |
| 265 | state = STATE_VALUE_ESCAPE; |
| 266 | } |
| 267 | else if (*it == '"') |
| 268 | { |
| 269 | add(token, value); |
| 270 | token.clear(); |
| 271 | value.clear(); |
| 272 | state = STATE_COMMA; |
| 273 | } |
| 274 | else |
| 275 | { |
| 276 | value += *it; |
| 277 | } |
| 278 | break; |
| 279 | |
| 280 | case STATE_VALUE_ESCAPE: |
| 281 | value += *it; |
| 282 | state = STATE_VALUE_QUOTED; |
| 283 | break; |
| 284 | |
| 285 | case STATE_VALUE: |
| 286 | if (Ascii::isSpace(*it)) |
| 287 | { |
| 288 | add(token, value); |
| 289 | token.clear(); |
| 290 | value.clear(); |
| 291 | state = STATE_COMMA; |
| 292 | } |
| 293 | else if (*it == ',') |
| 294 | { |
| 295 | add(token, value); |
| 296 | token.clear(); |
| 297 | value.clear(); |
| 298 | state = STATE_SPACE; |
| 299 | } |
| 300 | else |
| 301 | { |
| 302 | value += *it; |
| 303 | } |
| 304 | break; |
| 305 | |
| 306 | case STATE_COMMA: |
| 307 | if (*it == ',') |
| 308 | { |
| 309 | state = STATE_SPACE; |
| 310 | } |
| 311 | else if (Ascii::isSpace(*it)) |
| 312 | { |
| 313 | // Skip |
| 314 | } |
| 315 | else throw SyntaxException("Invalid authentication information" ); |
| 316 | break; |
| 317 | } |
| 318 | } |
| 319 | |
| 320 | if (state == STATE_VALUE) |
| 321 | add(token, value); |
| 322 | |
| 323 | if (!(state & STATE_FINAL)) |
| 324 | throw SyntaxException("Invalid authentication information" ); |
| 325 | } |
| 326 | |
| 327 | |
| 328 | } } // namespace Poco::Net |
| 329 | |