| // Copyright (c) 2010 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "net/http/http_auth_handler_ntlm.h" |
| |
| #include "base/base64.h" |
| #include "base/logging.h" |
| #include "base/string_util.h" |
| #include "base/utf_string_conversions.h" |
| #include "net/base/net_errors.h" |
| #include "net/base/net_util.h" |
| |
| namespace net { |
| |
| int HttpAuthHandlerNTLM::GenerateAuthToken( |
| const std::wstring& username, |
| const std::wstring& password, |
| const HttpRequestInfo* request, |
| const ProxyInfo* proxy, |
| std::string* auth_token) { |
| #if defined(NTLM_SSPI) |
| return auth_sspi_.GenerateAuthToken( |
| &username, |
| &password, |
| CreateSPN(origin_), |
| request, |
| proxy, |
| auth_token); |
| #else // !defined(NTLM_SSPI) |
| // TODO(wtc): See if we can use char* instead of void* for in_buf and |
| // out_buf. This change will need to propagate to GetNextToken, |
| // GenerateType1Msg, and GenerateType3Msg, and perhaps further. |
| const void* in_buf; |
| void* out_buf; |
| uint32 in_buf_len, out_buf_len; |
| std::string decoded_auth_data; |
| |
| // |username| may be in the form "DOMAIN\user". Parse it into the two |
| // components. |
| std::wstring domain; |
| std::wstring user; |
| size_t backslash_idx = username.find(L'\\'); |
| if (backslash_idx == std::wstring::npos) { |
| user = username; |
| } else { |
| domain = username.substr(0, backslash_idx); |
| user = username.substr(backslash_idx + 1); |
| } |
| domain_ = WideToUTF16(domain); |
| username_ = WideToUTF16(user); |
| password_ = WideToUTF16(password); |
| |
| // Initial challenge. |
| if (auth_data_.empty()) { |
| in_buf_len = 0; |
| in_buf = NULL; |
| int rv = InitializeBeforeFirstChallenge(); |
| if (rv != OK) |
| return rv; |
| } else { |
| if (!base::Base64Decode(auth_data_, &decoded_auth_data)) { |
| LOG(ERROR) << "Unexpected problem Base64 decoding."; |
| return ERR_UNEXPECTED; |
| } |
| in_buf_len = decoded_auth_data.length(); |
| in_buf = decoded_auth_data.data(); |
| } |
| |
| int rv = GetNextToken(in_buf, in_buf_len, &out_buf, &out_buf_len); |
| if (rv != OK) |
| return rv; |
| |
| // Base64 encode data in output buffer and prepend "NTLM ". |
| std::string encode_input(static_cast<char*>(out_buf), out_buf_len); |
| std::string encode_output; |
| bool base64_rv = base::Base64Encode(encode_input, &encode_output); |
| // OK, we are done with |out_buf| |
| free(out_buf); |
| if (!base64_rv) { |
| LOG(ERROR) << "Unexpected problem Base64 encoding."; |
| return ERR_UNEXPECTED; |
| } |
| *auth_token = std::string("NTLM ") + encode_output; |
| return OK; |
| #endif |
| } |
| |
| // The NTLM challenge header looks like: |
| // WWW-Authenticate: NTLM auth-data |
| bool HttpAuthHandlerNTLM::ParseChallenge( |
| HttpAuth::ChallengeTokenizer* tok) { |
| scheme_ = "ntlm"; |
| score_ = 3; |
| properties_ = ENCRYPTS_IDENTITY | IS_CONNECTION_BASED; |
| |
| #if defined(NTLM_SSPI) |
| return auth_sspi_.ParseChallenge(tok); |
| #else |
| auth_data_.clear(); |
| |
| // Verify the challenge's auth-scheme. |
| if (!tok->valid() || !LowerCaseEqualsASCII(tok->scheme(), "ntlm")) |
| return false; |
| |
| tok->set_expect_base64_token(true); |
| if (tok->GetNext()) |
| auth_data_.assign(tok->value_begin(), tok->value_end()); |
| return true; |
| #endif // defined(NTLM_SSPI) |
| } |
| |
| // static |
| std::wstring HttpAuthHandlerNTLM::CreateSPN(const GURL& origin) { |
| // The service principal name of the destination server. See |
| // http://msdn.microsoft.com/en-us/library/ms677949%28VS.85%29.aspx |
| std::wstring target(L"HTTP/"); |
| target.append(ASCIIToWide(GetHostAndPort(origin))); |
| return target; |
| } |
| |
| } // namespace net |