summaryrefslogtreecommitdiffstats
path: root/net/host_resolver_helper/host_resolver_helper.h
blob: bc18a215ad474de901f1eb01955646bf188b5f1d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
/** ---------------------------------------------------------------------------
 Copyright (c) 2011, Code Aurora Forum. All rights reserved.

 Redistribution and use in source and binary forms, with or without
 modification, are permitted provided that the following conditions are
 met:
     * Redistributions of source code must retain the above copyright
       notice, this list of conditions and the following disclaimer.
     * Redistributions in binary form must reproduce the above
       copyright notice, this list of conditions and the following
       disclaimer in the documentation and/or other materials provided
       with the distribution.
     * Neither the name of Code Aurora Forum, Inc. nor the names of its
       contributors may be used to endorse or promote products derived
       from this software without specific prior written permission.

 THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESS OR IMPLIED
 WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
 MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT
 ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS
 BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
 BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
 WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
 OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
 IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 -----------------------------------------------------------------------------**/
#ifndef __DNSRESOLVERHELPER_H__
#define __DNSRESOLVERHELPER_H__

#include <vector>
#include <string>

#include "config.h"
#include "net/base/host_resolver.h"
#include "net/base/address_list.h"
#include "net/base/net_log.h"
#include "net/base/net_errors.h"
#include "base/message_loop.h"
#include "net/base/completion_callback.h"
#include "hosts_provider.h"

//This class does DNS pre-resolution and should be used by net::HostResolver
//It's lifetime is depending on the host resolver and they should be created and destroyed
// in the correct order
//TODO: make sure the resolver is never destroyed before the pre-resolver
class HostResolverHelper: public net::HostResolver::HostnameResolverExt {
public:
    HostResolverHelper(net::HostResolver* hostresolver);
    virtual ~HostResolverHelper();

    ///////////////////////////////////////////////////////////////////////////////////////
    // net::HostResolver::HostnamePreresolver interface
    virtual void Resolve() {
        MessageLoop::current()->PostDelayedTask(FROM_HERE, NewRunnableFunction(&HostResolverHelper::DoResolve, this),
                500);
    }
    ///////////////////////////////////////////////////////////////////////////////////////

    //task which is executed through the message loop
    static void DoResolve(HostResolverHelper* obj);

    //call it to connect with the hostnames provider
    void Init(HostsProvider* provider);

private:

    //to be called when hosts pre-resolution is requested (worker function)
    bool StartHostsResolution();

private:
    int num_of_hosts_to_resolve;
    net::HostResolver* hostresolver_;
    HostsProvider* hostname_provider_;
    // Delegate interface, for notification when the ResolveRequest completes.

    class HostInfo: public base::RefCounted<HostInfo> {
    public:
        net::AddressList addrlist;
        net::HostResolver::RequestInfo reqinfo;
        net::HostResolver::RequestHandle reqhandle;
        bool pending;
        net::CompletionCallbackImpl<HostInfo> completion_callback_;

        HostInfo(const std::string& hostname);
        ~HostInfo() {
        }

        void OnLookupFinished(int result);

    };

    std::vector<scoped_refptr<HostInfo> > hostinfo_list_;

    //used to cancel all pending requests before issuing new ones
    void CancelAllRequests();
    void PrepareRequestsData(const std::vector<std::string>& hostnames);

};

//intialization for the webkit
net::HostResolver::HostnameResolverExt* CreateResolverIPObserver(net::HostResolver* hostResolver);

#endif