1 : /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
2 : /* ***** BEGIN LICENSE BLOCK *****
3 : * Version: MPL 1.1/GPL 2.0/LGPL 2.1
4 : *
5 : * The contents of this file are subject to the Mozilla Public License Version
6 : * 1.1 (the "License"); you may not use this file except in compliance with
7 : * the License. You may obtain a copy of the License at
8 : * http://www.mozilla.org/MPL/
9 : *
10 : * Software distributed under the License is distributed on an "AS IS" basis,
11 : * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
12 : * for the specific language governing rights and limitations under the
13 : * License.
14 : *
15 : * The Original Code is Mozilla.
16 : *
17 : * The Initial Developer of the Original Code is
18 : * Mozilla Corporation.
19 : * Portions created by the Initial Developer are Copyright (C) 2008
20 : * the Initial Developer. All Rights Reserved.
21 : *
22 : * Contributor(s):
23 : * Patrick McManus <mcmanus@ducksong.com>
24 : *
25 : * Alternatively, the contents of this file may be used under the terms of
26 : * either of the GNU General Public License Version 2 or later (the "GPL"),
27 : * or the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
28 : * in which case the provisions of the GPL or the LGPL are applicable instead
29 : * of those above. If you wish to allow use of your version of this file only
30 : * under the terms of either the GPL or the LGPL, and not to allow others to
31 : * use your version of this file under the terms of the MPL, indicate your
32 : * decision by deleting the provisions above and replace them with the notice
33 : * and other provisions required by the GPL or the LGPL. If you do not delete
34 : * the provisions above, a recipient may use your version of this file under
35 : * the terms of any one of the MPL, the GPL or the LGPL.
36 : *
37 : * ***** END LICENSE BLOCK ***** */
38 :
39 : #include "nsDNSPrefetch.h"
40 : #include "nsCOMPtr.h"
41 : #include "nsString.h"
42 :
43 : #include "nsNetUtil.h"
44 :
45 : #include "nsIDNSListener.h"
46 : #include "nsIDNSRecord.h"
47 : #include "nsIDNSService.h"
48 : #include "nsICancelable.h"
49 :
50 : static NS_DEFINE_CID(kDNSServiceCID, NS_DNSSERVICE_CID);
51 : static nsIDNSService *sDNSService = nsnull;
52 :
53 : nsresult
54 2863 : nsDNSPrefetch::Initialize(nsIDNSService *aDNSService)
55 : {
56 2863 : NS_IF_RELEASE(sDNSService);
57 2863 : sDNSService = aDNSService;
58 2863 : NS_IF_ADDREF(sDNSService);
59 2863 : return NS_OK;
60 : }
61 :
62 : nsresult
63 1419 : nsDNSPrefetch::Shutdown()
64 : {
65 1419 : NS_IF_RELEASE(sDNSService);
66 1419 : return NS_OK;
67 : }
68 :
69 3471 : nsDNSPrefetch::nsDNSPrefetch(nsIURI *aURI, bool storeTiming)
70 3471 : : mStoreTiming(storeTiming)
71 : {
72 3471 : aURI->GetAsciiHost(mHostname);
73 3471 : }
74 :
75 : nsresult
76 3471 : nsDNSPrefetch::Prefetch(PRUint16 flags)
77 : {
78 3471 : if (mHostname.IsEmpty())
79 0 : return NS_ERROR_NOT_AVAILABLE;
80 :
81 3471 : if (!sDNSService)
82 0 : return NS_ERROR_NOT_AVAILABLE;
83 :
84 6942 : nsCOMPtr<nsICancelable> tmpOutstanding;
85 :
86 3471 : if (mStoreTiming)
87 0 : mStartTimestamp = mozilla::TimeStamp::Now();
88 : // If AsyncResolve fails, for example because prefetching is disabled,
89 : // then our timing will be useless. However, in such a case,
90 : // mEndTimestamp will be a null timestamp and callers should check
91 : // TimingsValid() before using the timing.
92 : return sDNSService->AsyncResolve(mHostname, flags | nsIDNSService::RESOLVE_SPECULATE,
93 3471 : this, nsnull, getter_AddRefs(tmpOutstanding));
94 : }
95 :
96 : nsresult
97 0 : nsDNSPrefetch::PrefetchLow()
98 : {
99 0 : return Prefetch(nsIDNSService::RESOLVE_PRIORITY_LOW);
100 : }
101 :
102 : nsresult
103 0 : nsDNSPrefetch::PrefetchMedium()
104 : {
105 0 : return Prefetch(nsIDNSService::RESOLVE_PRIORITY_MEDIUM);
106 : }
107 :
108 : nsresult
109 3471 : nsDNSPrefetch::PrefetchHigh()
110 : {
111 3471 : return Prefetch(0);
112 : }
113 :
114 :
115 27205 : NS_IMPL_THREADSAFE_ISUPPORTS1(nsDNSPrefetch, nsIDNSListener)
116 :
117 : NS_IMETHODIMP
118 3468 : nsDNSPrefetch::OnLookupComplete(nsICancelable *request,
119 : nsIDNSRecord *rec,
120 : nsresult status)
121 : {
122 3468 : if (mStoreTiming)
123 0 : mEndTimestamp = mozilla::TimeStamp::Now();
124 3468 : return NS_OK;
125 : }
|