1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #include "base/memory/scoped_ptr.h" 6 #include "net/base/mock_host_resolver.h" 7 #include "net/base/net_errors.h" 8 #include "net/http/http_auth_handler.h" 9 #include "net/http/http_auth_handler_factory.h" 10 #include "net/http/mock_allow_url_security_manager.h" 11 #include "net/http/url_security_manager.h" 12 #include "testing/gtest/include/gtest/gtest.h" 13 14 namespace net { 15 16 namespace { 17 18 class MockHttpAuthHandlerFactory : public HttpAuthHandlerFactory { 19 public: 20 explicit MockHttpAuthHandlerFactory(int return_code) : 21 return_code_(return_code) {} 22 virtual ~MockHttpAuthHandlerFactory() {} 23 24 virtual int CreateAuthHandler(HttpAuth::ChallengeTokenizer* challenge, 25 HttpAuth::Target target, 26 const GURL& origin, 27 CreateReason reason, 28 int nonce_count, 29 const BoundNetLog& net_log, 30 scoped_ptr<HttpAuthHandler>* handler) { 31 handler->reset(); 32 return return_code_; 33 } 34 35 private: 36 int return_code_; 37 }; 38 39 } // namespace 40 41 TEST(HttpAuthHandlerFactoryTest, RegistryFactory) { 42 HttpAuthHandlerRegistryFactory registry_factory; 43 GURL gurl("www.google.com"); 44 const int kBasicReturnCode = ERR_INVALID_SPDY_STREAM; 45 MockHttpAuthHandlerFactory* mock_factory_basic = 46 new MockHttpAuthHandlerFactory(kBasicReturnCode); 47 48 const int kDigestReturnCode = ERR_PAC_SCRIPT_FAILED; 49 MockHttpAuthHandlerFactory* mock_factory_digest = 50 new MockHttpAuthHandlerFactory(kDigestReturnCode); 51 52 const int kDigestReturnCodeReplace = ERR_SYN_REPLY_NOT_RECEIVED; 53 MockHttpAuthHandlerFactory* mock_factory_digest_replace = 54 new MockHttpAuthHandlerFactory(kDigestReturnCodeReplace); 55 56 scoped_ptr<HttpAuthHandler> handler; 57 58 // No schemes should be supported in the beginning. 59 EXPECT_EQ(ERR_UNSUPPORTED_AUTH_SCHEME, 60 registry_factory.CreateAuthHandlerFromString( 61 "Basic", HttpAuth::AUTH_SERVER, gurl, BoundNetLog(), &handler)); 62 63 // Test what happens with a single scheme. 64 registry_factory.RegisterSchemeFactory("Basic", mock_factory_basic); 65 EXPECT_EQ(kBasicReturnCode, 66 registry_factory.CreateAuthHandlerFromString( 67 "Basic", HttpAuth::AUTH_SERVER, gurl, BoundNetLog(), &handler)); 68 EXPECT_EQ(ERR_UNSUPPORTED_AUTH_SCHEME, 69 registry_factory.CreateAuthHandlerFromString( 70 "Digest", HttpAuth::AUTH_SERVER, gurl, BoundNetLog(), 71 &handler)); 72 73 // Test multiple schemes 74 registry_factory.RegisterSchemeFactory("Digest", mock_factory_digest); 75 EXPECT_EQ(kBasicReturnCode, 76 registry_factory.CreateAuthHandlerFromString( 77 "Basic", HttpAuth::AUTH_SERVER, gurl, BoundNetLog(), &handler)); 78 EXPECT_EQ(kDigestReturnCode, 79 registry_factory.CreateAuthHandlerFromString( 80 "Digest", HttpAuth::AUTH_SERVER, gurl, BoundNetLog(), 81 &handler)); 82 83 // Test case-insensitivity 84 EXPECT_EQ(kBasicReturnCode, 85 registry_factory.CreateAuthHandlerFromString( 86 "basic", HttpAuth::AUTH_SERVER, gurl, BoundNetLog(), &handler)); 87 88 // Test replacement of existing auth scheme 89 registry_factory.RegisterSchemeFactory("Digest", mock_factory_digest_replace); 90 EXPECT_EQ(kBasicReturnCode, 91 registry_factory.CreateAuthHandlerFromString( 92 "Basic", HttpAuth::AUTH_SERVER, gurl, BoundNetLog(), &handler)); 93 EXPECT_EQ(kDigestReturnCodeReplace, 94 registry_factory.CreateAuthHandlerFromString( 95 "Digest", HttpAuth::AUTH_SERVER, gurl, BoundNetLog(), 96 &handler)); 97 } 98 99 TEST(HttpAuthHandlerFactoryTest, DefaultFactory) { 100 scoped_ptr<HostResolver> host_resolver(new MockHostResolver()); 101 MockAllowURLSecurityManager url_security_manager; 102 scoped_ptr<HttpAuthHandlerRegistryFactory> http_auth_handler_factory( 103 HttpAuthHandlerFactory::CreateDefault(host_resolver.get())); 104 http_auth_handler_factory->SetURLSecurityManager( 105 "negotiate", &url_security_manager); 106 GURL server_origin("http://www.example.com"); 107 GURL proxy_origin("http://cache.example.com:3128"); 108 { 109 scoped_ptr<HttpAuthHandler> handler; 110 int rv = http_auth_handler_factory->CreateAuthHandlerFromString( 111 "Basic realm=\"FooBar\"", 112 HttpAuth::AUTH_SERVER, 113 server_origin, 114 BoundNetLog(), 115 &handler); 116 EXPECT_EQ(OK, rv); 117 ASSERT_FALSE(handler.get() == NULL); 118 EXPECT_EQ(HttpAuth::AUTH_SCHEME_BASIC, handler->auth_scheme()); 119 EXPECT_STREQ("FooBar", handler->realm().c_str()); 120 EXPECT_EQ(HttpAuth::AUTH_SERVER, handler->target()); 121 EXPECT_FALSE(handler->encrypts_identity()); 122 EXPECT_FALSE(handler->is_connection_based()); 123 } 124 { 125 scoped_ptr<HttpAuthHandler> handler; 126 int rv = http_auth_handler_factory->CreateAuthHandlerFromString( 127 "UNSUPPORTED realm=\"FooBar\"", 128 HttpAuth::AUTH_SERVER, 129 server_origin, 130 BoundNetLog(), 131 &handler); 132 EXPECT_EQ(ERR_UNSUPPORTED_AUTH_SCHEME, rv); 133 EXPECT_TRUE(handler.get() == NULL); 134 } 135 { 136 scoped_ptr<HttpAuthHandler> handler; 137 int rv = http_auth_handler_factory->CreateAuthHandlerFromString( 138 "Digest realm=\"FooBar\", nonce=\"xyz\"", 139 HttpAuth::AUTH_PROXY, 140 proxy_origin, 141 BoundNetLog(), 142 &handler); 143 EXPECT_EQ(OK, rv); 144 ASSERT_FALSE(handler.get() == NULL); 145 EXPECT_EQ(HttpAuth::AUTH_SCHEME_DIGEST, handler->auth_scheme()); 146 EXPECT_STREQ("FooBar", handler->realm().c_str()); 147 EXPECT_EQ(HttpAuth::AUTH_PROXY, handler->target()); 148 EXPECT_TRUE(handler->encrypts_identity()); 149 EXPECT_FALSE(handler->is_connection_based()); 150 } 151 { 152 scoped_ptr<HttpAuthHandler> handler; 153 int rv = http_auth_handler_factory->CreateAuthHandlerFromString( 154 "NTLM", 155 HttpAuth::AUTH_SERVER, 156 server_origin, 157 BoundNetLog(), 158 &handler); 159 EXPECT_EQ(OK, rv); 160 ASSERT_FALSE(handler.get() == NULL); 161 EXPECT_EQ(HttpAuth::AUTH_SCHEME_NTLM, handler->auth_scheme()); 162 EXPECT_STREQ("", handler->realm().c_str()); 163 EXPECT_EQ(HttpAuth::AUTH_SERVER, handler->target()); 164 EXPECT_TRUE(handler->encrypts_identity()); 165 EXPECT_TRUE(handler->is_connection_based()); 166 } 167 { 168 scoped_ptr<HttpAuthHandler> handler; 169 int rv = http_auth_handler_factory->CreateAuthHandlerFromString( 170 "Negotiate", 171 HttpAuth::AUTH_SERVER, 172 server_origin, 173 BoundNetLog(), 174 &handler); 175 EXPECT_EQ(OK, rv); 176 ASSERT_FALSE(handler.get() == NULL); 177 EXPECT_EQ(HttpAuth::AUTH_SCHEME_NEGOTIATE, handler->auth_scheme()); 178 EXPECT_STREQ("", handler->realm().c_str()); 179 EXPECT_EQ(HttpAuth::AUTH_SERVER, handler->target()); 180 EXPECT_TRUE(handler->encrypts_identity()); 181 EXPECT_TRUE(handler->is_connection_based()); 182 } 183 } 184 185 } // namespace net 186
