Home | History | Annotate | Download | only in http 
      1 // Copyright (c) 2012 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 "net/http/http_auth_handler.h"
      6 
      7 #include "base/bind.h"
      8 #include "base/bind_helpers.h"
      9 #include "base/logging.h"
     10 #include "net/base/net_errors.h"
     11 
     12 namespace net {
     13 
     14 HttpAuthHandler::HttpAuthHandler()
     15     : auth_scheme_(HttpAuth::AUTH_SCHEME_MAX),
     16       score_(-1),
     17       target_(HttpAuth::AUTH_NONE),
     18       properties_(-1) {
     19 }
     20 
     21 HttpAuthHandler::~HttpAuthHandler() {
     22 }
     23 
     24 bool HttpAuthHandler::InitFromChallenge(
     25     HttpAuth::ChallengeTokenizer* challenge,
     26     HttpAuth::Target target,
     27     const GURL& origin,
     28     const BoundNetLog& net_log) {
     29   origin_ = origin;
     30   target_ = target;
     31   score_ = -1;
     32   properties_ = -1;
     33   net_log_ = net_log;
     34 
     35   auth_challenge_ = challenge->challenge_text();
     36   bool ok = Init(challenge);
     37 
     38   // Init() is expected to set the scheme, realm, score, and properties.  The
     39   // realm may be empty.
     40   DCHECK(!ok || score_ != -1);
     41   DCHECK(!ok || properties_ != -1);
     42   DCHECK(!ok || auth_scheme_ != HttpAuth::AUTH_SCHEME_MAX);
     43 
     44   return ok;
     45 }
     46 
     47 namespace {
     48 
     49 NetLog::EventType EventTypeFromAuthTarget(HttpAuth::Target target) {
     50   switch (target) {
     51     case HttpAuth::AUTH_PROXY:
     52       return NetLog::TYPE_AUTH_PROXY;
     53     case HttpAuth::AUTH_SERVER:
     54       return NetLog::TYPE_AUTH_SERVER;
     55     default:
     56       NOTREACHED();
     57       return NetLog::TYPE_CANCELLED;
     58   }
     59 }
     60 
     61 }  // namespace
     62 
     63 int HttpAuthHandler::GenerateAuthToken(
     64     const AuthCredentials* credentials, const HttpRequestInfo* request,
     65     const CompletionCallback& callback, std::string* auth_token) {
     66   // TODO(cbentzel): Enforce non-NULL callback after cleaning up SocketStream.
     67   DCHECK(request);
     68   DCHECK(credentials != NULL || AllowsDefaultCredentials());
     69   DCHECK(auth_token != NULL);
     70   DCHECK(callback_.is_null());
     71   callback_ = callback;
     72   net_log_.BeginEvent(EventTypeFromAuthTarget(target_));
     73   int rv = GenerateAuthTokenImpl(
     74       credentials, request,
     75       base::Bind(&HttpAuthHandler::OnGenerateAuthTokenComplete,
     76                  base::Unretained(this)),
     77       auth_token);
     78   if (rv != ERR_IO_PENDING)
     79     FinishGenerateAuthToken();
     80   return rv;
     81 }
     82 
     83 bool HttpAuthHandler::NeedsIdentity() {
     84   return true;
     85 }
     86 
     87 bool HttpAuthHandler::AllowsDefaultCredentials() {
     88   return false;
     89 }
     90 
     91 bool HttpAuthHandler::AllowsExplicitCredentials() {
     92   return true;
     93 }
     94 
     95 void HttpAuthHandler::OnGenerateAuthTokenComplete(int rv) {
     96   CompletionCallback callback = callback_;
     97   FinishGenerateAuthToken();
     98   if (!callback.is_null())
     99     callback.Run(rv);
    100 }
    101 
    102 void HttpAuthHandler::FinishGenerateAuthToken() {
    103   // TOOD(cbentzel): Should this be done in OK case only?
    104   net_log_.EndEvent(EventTypeFromAuthTarget(target_));
    105   callback_.Reset();
    106 }
    107 
    108 }  // namespace net
    109