1 : //* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*-/
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 Url Classifier code
16 : *
17 : * The Initial Developer of the Original Code is
18 : * Google Inc.
19 : * Portions created by the Initial Developer are Copyright (C) 2006
20 : * the Initial Developer. All Rights Reserved.
21 : *
22 : * Contributor(s):
23 : * Tony Chang <tony@ponderer.org> (original author)
24 : *
25 : * Alternatively, the contents of this file may be used under the terms of
26 : * either the GNU General Public License Version 2 or later (the "GPL"), or
27 : * 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 : #ifndef nsUrlClassifierStreamUpdater_h_
40 : #define nsUrlClassifierStreamUpdater_h_
41 :
42 : #include <nsISupportsUtils.h>
43 :
44 : #include "nsCOMPtr.h"
45 : #include "nsIObserver.h"
46 : #include "nsIUrlClassifierStreamUpdater.h"
47 : #include "nsIStreamListener.h"
48 : #include "nsNetUtil.h"
49 : #include "nsTArray.h"
50 : #include "nsIBadCertListener2.h"
51 : #include "nsISSLErrorListener.h"
52 : #include "nsITimer.h"
53 :
54 : // Forward declare pointers
55 : class nsIURI;
56 :
57 : class nsUrlClassifierStreamUpdater : public nsIUrlClassifierStreamUpdater,
58 : public nsIUrlClassifierUpdateObserver,
59 : public nsIStreamListener,
60 : public nsIObserver,
61 : public nsIBadCertListener2,
62 : public nsISSLErrorListener,
63 : public nsIInterfaceRequestor,
64 : public nsITimerCallback
65 : {
66 : public:
67 : nsUrlClassifierStreamUpdater();
68 :
69 : NS_DECL_ISUPPORTS
70 : NS_DECL_NSIURLCLASSIFIERSTREAMUPDATER
71 : NS_DECL_NSIURLCLASSIFIERUPDATEOBSERVER
72 : NS_DECL_NSIINTERFACEREQUESTOR
73 : NS_DECL_NSIREQUESTOBSERVER
74 : NS_DECL_NSISTREAMLISTENER
75 : NS_DECL_NSIBADCERTLISTENER2
76 : NS_DECL_NSISSLERRORLISTENER
77 : NS_DECL_NSIOBSERVER
78 : NS_DECL_NSITIMERCALLBACK
79 :
80 : private:
81 : // No subclassing
82 8 : ~nsUrlClassifierStreamUpdater() {}
83 :
84 : // When the dbservice sends an UpdateComplete or UpdateFailure, we call this
85 : // to reset the stream updater.
86 : void DownloadDone();
87 :
88 : // Disallow copy constructor
89 : nsUrlClassifierStreamUpdater(nsUrlClassifierStreamUpdater&);
90 :
91 : nsresult AddRequestBody(const nsACString &aRequestBody);
92 :
93 : nsresult FetchUpdate(nsIURI *aURI,
94 : const nsACString &aRequestBody,
95 : const nsACString &aTable,
96 : const nsACString &aServerMAC);
97 : nsresult FetchUpdate(const nsACString &aURI,
98 : const nsACString &aRequestBody,
99 : const nsACString &aTable,
100 : const nsACString &aServerMAC);
101 :
102 : nsresult FetchNext();
103 :
104 : bool mIsUpdating;
105 : bool mInitialized;
106 : bool mDownloadError;
107 : bool mBeganStream;
108 : nsCOMPtr<nsIURI> mUpdateUrl;
109 : nsCString mStreamTable;
110 : nsCString mServerMAC;
111 : nsCOMPtr<nsIChannel> mChannel;
112 : nsCOMPtr<nsIUrlClassifierDBService> mDBService;
113 : nsCOMPtr<nsITimer> mTimer;
114 :
115 34 : struct PendingUpdate {
116 : nsCString mUrl;
117 : nsCString mTable;
118 : nsCString mServerMAC;
119 : };
120 : nsTArray<PendingUpdate> mPendingUpdates;
121 :
122 : nsCOMPtr<nsIUrlClassifierCallback> mSuccessCallback;
123 : nsCOMPtr<nsIUrlClassifierCallback> mUpdateErrorCallback;
124 : nsCOMPtr<nsIUrlClassifierCallback> mDownloadErrorCallback;
125 : };
126 :
127 : #endif // nsUrlClassifierStreamUpdater_h_
|