2007-03-22 10:30:00 -07:00
|
|
|
//* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
|
|
/* ***** BEGIN LICENSE BLOCK *****
|
|
|
|
* Version: MPL 1.1/GPL 2.0/LGPL 2.1
|
|
|
|
*
|
|
|
|
* The contents of this file are subject to the Mozilla Public License Version
|
|
|
|
* 1.1 (the "License"); you may not use this file except in compliance with
|
|
|
|
* the License. You may obtain a copy of the License at
|
|
|
|
* http://www.mozilla.org/MPL/
|
|
|
|
*
|
|
|
|
* Software distributed under the License is distributed on an "AS IS" basis,
|
|
|
|
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
|
|
|
|
* for the specific language governing rights and limitations under the
|
|
|
|
* License.
|
|
|
|
*
|
|
|
|
* The Original Code is Url Classifier code
|
|
|
|
*
|
|
|
|
* The Initial Developer of the Original Code is
|
|
|
|
* Google Inc.
|
|
|
|
* Portions created by the Initial Developer are Copyright (C) 2006
|
|
|
|
* the Initial Developer. All Rights Reserved.
|
|
|
|
*
|
|
|
|
* Contributor(s):
|
|
|
|
* Tony Chang <tony@ponderer.org> (original author)
|
|
|
|
*
|
|
|
|
* Alternatively, the contents of this file may be used under the terms of
|
|
|
|
* either the GNU General Public License Version 2 or later (the "GPL"), or
|
|
|
|
* the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
|
|
|
|
* in which case the provisions of the GPL or the LGPL are applicable instead
|
|
|
|
* of those above. If you wish to allow use of your version of this file only
|
|
|
|
* under the terms of either the GPL or the LGPL, and not to allow others to
|
|
|
|
* use your version of this file under the terms of the MPL, indicate your
|
|
|
|
* decision by deleting the provisions above and replace them with the notice
|
|
|
|
* and other provisions required by the GPL or the LGPL. If you do not delete
|
|
|
|
* the provisions above, a recipient may use your version of this file under
|
|
|
|
* the terms of any one of the MPL, the GPL or the LGPL.
|
|
|
|
*
|
|
|
|
* ***** END LICENSE BLOCK ***** */
|
|
|
|
|
|
|
|
#include "nsCRT.h"
|
2007-07-25 23:38:43 -07:00
|
|
|
#include "nsIHttpChannel.h"
|
2007-03-22 10:30:00 -07:00
|
|
|
#include "nsIObserverService.h"
|
2007-07-25 23:38:43 -07:00
|
|
|
#include "nsIStringStream.h"
|
|
|
|
#include "nsIUploadChannel.h"
|
2007-03-22 10:30:00 -07:00
|
|
|
#include "nsIURI.h"
|
|
|
|
#include "nsIUrlClassifierDBService.h"
|
|
|
|
#include "nsStreamUtils.h"
|
2007-07-25 23:38:43 -07:00
|
|
|
#include "nsStringStream.h"
|
2007-03-22 10:30:00 -07:00
|
|
|
#include "nsToolkitCompsCID.h"
|
|
|
|
#include "nsUrlClassifierStreamUpdater.h"
|
|
|
|
#include "prlog.h"
|
|
|
|
|
|
|
|
static const char* gQuitApplicationMessage = "quit-application";
|
|
|
|
|
|
|
|
// NSPR_LOG_MODULES=UrlClassifierStreamUpdater:5
|
|
|
|
#if defined(PR_LOGGING)
|
|
|
|
static const PRLogModuleInfo *gUrlClassifierStreamUpdaterLog = nsnull;
|
|
|
|
#define LOG(args) PR_LOG(gUrlClassifierStreamUpdaterLog, PR_LOG_DEBUG, args)
|
|
|
|
#else
|
|
|
|
#define LOG(args)
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
// nsIUrlClassiferStreamUpdater implementation
|
|
|
|
// Handles creating/running the stream listener
|
|
|
|
|
|
|
|
nsUrlClassifierStreamUpdater::nsUrlClassifierStreamUpdater()
|
2011-10-17 07:59:28 -07:00
|
|
|
: mIsUpdating(false), mInitialized(false), mDownloadError(false),
|
|
|
|
mBeganStream(false), mUpdateUrl(nsnull), mChannel(nsnull)
|
2007-03-22 10:30:00 -07:00
|
|
|
{
|
|
|
|
#if defined(PR_LOGGING)
|
|
|
|
if (!gUrlClassifierStreamUpdaterLog)
|
|
|
|
gUrlClassifierStreamUpdaterLog = PR_NewLogModule("UrlClassifierStreamUpdater");
|
|
|
|
#endif
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2008-05-07 13:33:45 -07:00
|
|
|
NS_IMPL_THREADSAFE_ISUPPORTS9(nsUrlClassifierStreamUpdater,
|
2008-01-29 12:57:18 -08:00
|
|
|
nsIUrlClassifierStreamUpdater,
|
|
|
|
nsIUrlClassifierUpdateObserver,
|
|
|
|
nsIRequestObserver,
|
|
|
|
nsIStreamListener,
|
|
|
|
nsIObserver,
|
|
|
|
nsIBadCertListener2,
|
|
|
|
nsISSLErrorListener,
|
2008-05-07 13:33:45 -07:00
|
|
|
nsIInterfaceRequestor,
|
|
|
|
nsITimerCallback)
|
2007-03-22 10:30:00 -07:00
|
|
|
|
|
|
|
/**
|
2008-01-12 14:22:03 -08:00
|
|
|
* Clear out the update.
|
2007-03-22 10:30:00 -07:00
|
|
|
*/
|
|
|
|
void
|
|
|
|
nsUrlClassifierStreamUpdater::DownloadDone()
|
|
|
|
{
|
2008-01-12 14:22:03 -08:00
|
|
|
LOG(("nsUrlClassifierStreamUpdater::DownloadDone [this=%p]", this));
|
2011-10-17 07:59:28 -07:00
|
|
|
mIsUpdating = false;
|
2008-01-12 14:22:03 -08:00
|
|
|
|
2008-01-29 12:57:18 -08:00
|
|
|
mPendingUpdates.Clear();
|
2011-10-17 07:59:28 -07:00
|
|
|
mDownloadError = false;
|
2008-01-12 14:22:03 -08:00
|
|
|
mSuccessCallback = nsnull;
|
|
|
|
mUpdateErrorCallback = nsnull;
|
|
|
|
mDownloadErrorCallback = nsnull;
|
2007-03-22 10:30:00 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
// nsIUrlClassifierStreamUpdater implementation
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsUrlClassifierStreamUpdater::GetUpdateUrl(nsACString & aUpdateUrl)
|
|
|
|
{
|
|
|
|
if (mUpdateUrl) {
|
|
|
|
mUpdateUrl->GetSpec(aUpdateUrl);
|
|
|
|
} else {
|
|
|
|
aUpdateUrl.Truncate();
|
|
|
|
}
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsUrlClassifierStreamUpdater::SetUpdateUrl(const nsACString & aUpdateUrl)
|
|
|
|
{
|
2007-07-25 23:38:43 -07:00
|
|
|
LOG(("Update URL is %s\n", PromiseFlatCString(aUpdateUrl).get()));
|
|
|
|
|
2007-03-22 10:30:00 -07:00
|
|
|
nsresult rv = NS_NewURI(getter_AddRefs(mUpdateUrl), aUpdateUrl);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
2008-01-12 14:22:03 -08:00
|
|
|
nsresult
|
|
|
|
nsUrlClassifierStreamUpdater::FetchUpdate(nsIURI *aUpdateUrl,
|
2008-01-29 12:57:18 -08:00
|
|
|
const nsACString & aRequestBody,
|
2008-02-27 00:51:02 -08:00
|
|
|
const nsACString & aStreamTable,
|
|
|
|
const nsACString & aServerMAC)
|
2008-01-12 14:22:03 -08:00
|
|
|
{
|
|
|
|
nsresult rv;
|
2009-10-13 06:41:34 -07:00
|
|
|
PRUint32 loadFlags = nsIChannel::INHIBIT_CACHING |
|
|
|
|
nsIChannel::LOAD_BYPASS_CACHE;
|
|
|
|
rv = NS_NewChannel(getter_AddRefs(mChannel), aUpdateUrl, nsnull, nsnull, this,
|
|
|
|
loadFlags);
|
2008-01-12 14:22:03 -08:00
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
2011-10-17 07:59:28 -07:00
|
|
|
mBeganStream = false;
|
2008-05-02 02:54:01 -07:00
|
|
|
|
2008-01-12 14:22:03 -08:00
|
|
|
if (!aRequestBody.IsEmpty()) {
|
|
|
|
rv = AddRequestBody(aRequestBody);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
}
|
|
|
|
|
2008-10-19 20:48:25 -07:00
|
|
|
// Set the appropriate content type for file/data URIs, for unit testing
|
|
|
|
// purposes.
|
2011-09-28 23:19:26 -07:00
|
|
|
bool match;
|
2008-10-19 20:48:25 -07:00
|
|
|
if ((NS_SUCCEEDED(aUpdateUrl->SchemeIs("file", &match)) && match) ||
|
|
|
|
(NS_SUCCEEDED(aUpdateUrl->SchemeIs("data", &match)) && match)) {
|
|
|
|
mChannel->SetContentType(NS_LITERAL_CSTRING("application/vnd.google.safebrowsing-update"));
|
|
|
|
}
|
|
|
|
|
2008-01-12 14:22:03 -08:00
|
|
|
// Make the request
|
|
|
|
rv = mChannel->AsyncOpen(this, nsnull);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
2008-01-29 12:57:18 -08:00
|
|
|
mStreamTable = aStreamTable;
|
2008-02-27 00:51:02 -08:00
|
|
|
mServerMAC = aServerMAC;
|
2008-01-29 12:57:18 -08:00
|
|
|
|
2008-01-12 14:22:03 -08:00
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
nsresult
|
|
|
|
nsUrlClassifierStreamUpdater::FetchUpdate(const nsACString & aUpdateUrl,
|
2008-01-29 12:57:18 -08:00
|
|
|
const nsACString & aRequestBody,
|
2008-02-27 00:51:02 -08:00
|
|
|
const nsACString & aStreamTable,
|
|
|
|
const nsACString & aServerMAC)
|
2008-01-12 14:22:03 -08:00
|
|
|
{
|
|
|
|
nsCOMPtr<nsIURI> uri;
|
|
|
|
nsresult rv = NS_NewURI(getter_AddRefs(uri), aUpdateUrl);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
2008-01-29 12:57:18 -08:00
|
|
|
LOG(("Fetching update from %s\n", PromiseFlatCString(aUpdateUrl).get()));
|
|
|
|
|
2008-02-27 00:51:02 -08:00
|
|
|
return FetchUpdate(uri, aRequestBody, aStreamTable, aServerMAC);
|
2008-01-12 14:22:03 -08:00
|
|
|
}
|
|
|
|
|
2007-03-22 10:30:00 -07:00
|
|
|
NS_IMETHODIMP
|
|
|
|
nsUrlClassifierStreamUpdater::DownloadUpdates(
|
2008-04-15 15:39:44 -07:00
|
|
|
const nsACString &aRequestTables,
|
2007-07-25 23:38:43 -07:00
|
|
|
const nsACString &aRequestBody,
|
2008-02-27 00:51:02 -08:00
|
|
|
const nsACString &aClientKey,
|
2007-07-25 23:38:43 -07:00
|
|
|
nsIUrlClassifierCallback *aSuccessCallback,
|
|
|
|
nsIUrlClassifierCallback *aUpdateErrorCallback,
|
|
|
|
nsIUrlClassifierCallback *aDownloadErrorCallback,
|
2011-09-28 23:19:26 -07:00
|
|
|
bool *_retval)
|
2007-03-22 10:30:00 -07:00
|
|
|
{
|
2008-01-23 11:30:54 -08:00
|
|
|
NS_ENSURE_ARG(aSuccessCallback);
|
|
|
|
NS_ENSURE_ARG(aUpdateErrorCallback);
|
|
|
|
NS_ENSURE_ARG(aDownloadErrorCallback);
|
|
|
|
|
2007-03-22 10:30:00 -07:00
|
|
|
if (mIsUpdating) {
|
|
|
|
LOG(("already updating, skipping update"));
|
2011-10-17 07:59:28 -07:00
|
|
|
*_retval = false;
|
2007-03-22 10:30:00 -07:00
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!mUpdateUrl) {
|
|
|
|
NS_ERROR("updateUrl not set");
|
|
|
|
return NS_ERROR_NOT_INITIALIZED;
|
|
|
|
}
|
|
|
|
|
2008-01-12 14:22:03 -08:00
|
|
|
nsresult rv;
|
|
|
|
|
2007-03-22 10:30:00 -07:00
|
|
|
if (!mInitialized) {
|
|
|
|
// Add an observer for shutdown so we can cancel any pending list
|
|
|
|
// downloads. quit-application is the same event that the download
|
|
|
|
// manager listens for and uses to cancel pending downloads.
|
|
|
|
nsCOMPtr<nsIObserverService> observerService =
|
Bug 560095 - Use mozilla::services::GetObserverService(). r=biesi,dveditz,gavin,josh,jst,mrbkap,roc,sdwilsh,shaver,sicking,smontagu,surkov
2010-04-29 09:59:13 -07:00
|
|
|
mozilla::services::GetObserverService();
|
2007-03-22 10:30:00 -07:00
|
|
|
if (!observerService)
|
|
|
|
return NS_ERROR_FAILURE;
|
|
|
|
|
2011-10-17 07:59:28 -07:00
|
|
|
observerService->AddObserver(this, gQuitApplicationMessage, false);
|
2008-01-12 14:22:03 -08:00
|
|
|
|
|
|
|
mDBService = do_GetService(NS_URLCLASSIFIERDBSERVICE_CONTRACTID, &rv);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
2011-10-17 07:59:28 -07:00
|
|
|
mInitialized = true;
|
2007-03-22 10:30:00 -07:00
|
|
|
}
|
|
|
|
|
2008-04-15 15:39:44 -07:00
|
|
|
rv = mDBService->BeginUpdate(this, aRequestTables, aClientKey);
|
2008-01-23 11:30:54 -08:00
|
|
|
if (rv == NS_ERROR_NOT_AVAILABLE) {
|
|
|
|
LOG(("already updating, skipping update"));
|
2011-10-17 07:59:28 -07:00
|
|
|
*_retval = false;
|
2008-01-23 11:30:54 -08:00
|
|
|
return NS_OK;
|
|
|
|
} else if (NS_FAILED(rv)) {
|
|
|
|
return rv;
|
|
|
|
}
|
|
|
|
|
2008-01-12 14:22:03 -08:00
|
|
|
mSuccessCallback = aSuccessCallback;
|
|
|
|
mUpdateErrorCallback = aUpdateErrorCallback;
|
|
|
|
mDownloadErrorCallback = aDownloadErrorCallback;
|
|
|
|
|
2011-10-17 07:59:28 -07:00
|
|
|
mIsUpdating = true;
|
|
|
|
*_retval = true;
|
2008-01-12 14:22:03 -08:00
|
|
|
|
|
|
|
|
2008-02-27 00:51:02 -08:00
|
|
|
return FetchUpdate(mUpdateUrl, aRequestBody, EmptyCString(), EmptyCString());
|
2008-01-12 14:22:03 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
// nsIUrlClassifierUpdateObserver implementation
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
2008-01-29 12:57:18 -08:00
|
|
|
nsUrlClassifierStreamUpdater::UpdateUrlRequested(const nsACString &aUrl,
|
2008-02-27 00:51:02 -08:00
|
|
|
const nsACString &aTable,
|
|
|
|
const nsACString &aServerMAC)
|
2008-01-12 14:22:03 -08:00
|
|
|
{
|
|
|
|
LOG(("Queuing requested update from %s\n", PromiseFlatCString(aUrl).get()));
|
|
|
|
|
2008-01-29 12:57:18 -08:00
|
|
|
PendingUpdate *update = mPendingUpdates.AppendElement();
|
|
|
|
if (!update)
|
|
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
|
|
|
2008-10-19 20:48:25 -07:00
|
|
|
// Allow data: and file: urls for unit testing purposes, otherwise assume http
|
|
|
|
if (StringBeginsWith(aUrl, NS_LITERAL_CSTRING("data:")) ||
|
|
|
|
StringBeginsWith(aUrl, NS_LITERAL_CSTRING("file:"))) {
|
2008-01-29 12:57:18 -08:00
|
|
|
update->mUrl = aUrl;
|
2008-01-12 14:22:03 -08:00
|
|
|
} else {
|
2008-01-29 12:57:18 -08:00
|
|
|
update->mUrl = NS_LITERAL_CSTRING("http://") + aUrl;
|
2008-01-12 14:22:03 -08:00
|
|
|
}
|
2008-01-29 12:57:18 -08:00
|
|
|
update->mTable = aTable;
|
2008-02-27 00:51:02 -08:00
|
|
|
update->mServerMAC = aServerMAC;
|
2008-01-12 14:22:03 -08:00
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
2008-02-27 00:51:02 -08:00
|
|
|
NS_IMETHODIMP
|
|
|
|
nsUrlClassifierStreamUpdater::RekeyRequested()
|
|
|
|
{
|
|
|
|
nsCOMPtr<nsIObserverService> observerService =
|
Bug 560095 - Use mozilla::services::GetObserverService(). r=biesi,dveditz,gavin,josh,jst,mrbkap,roc,sdwilsh,shaver,sicking,smontagu,surkov
2010-04-29 09:59:13 -07:00
|
|
|
mozilla::services::GetObserverService();
|
|
|
|
|
|
|
|
if (!observerService)
|
|
|
|
return NS_ERROR_FAILURE;
|
2008-02-27 00:51:02 -08:00
|
|
|
|
|
|
|
return observerService->NotifyObservers(static_cast<nsIUrlClassifierStreamUpdater*>(this),
|
|
|
|
"url-classifier-rekey-requested",
|
|
|
|
nsnull);
|
|
|
|
}
|
|
|
|
|
2008-05-07 13:33:45 -07:00
|
|
|
nsresult
|
|
|
|
nsUrlClassifierStreamUpdater::FetchNext()
|
2008-05-07 03:21:11 -07:00
|
|
|
{
|
2008-05-07 13:33:45 -07:00
|
|
|
if (mPendingUpdates.Length() == 0) {
|
|
|
|
return NS_OK;
|
|
|
|
}
|
2008-05-07 03:21:11 -07:00
|
|
|
|
2008-05-07 13:33:45 -07:00
|
|
|
PendingUpdate &update = mPendingUpdates[0];
|
|
|
|
LOG(("Fetching update url: %s\n", update.mUrl.get()));
|
|
|
|
nsresult rv = FetchUpdate(update.mUrl, EmptyCString(),
|
|
|
|
update.mTable, update.mServerMAC);
|
|
|
|
if (NS_FAILED(rv)) {
|
|
|
|
LOG(("Error fetching update url: %s\n", update.mUrl.get()));
|
|
|
|
// We can commit the urls that we've applied so far. This is
|
|
|
|
// probably a transient server problem, so trigger backoff.
|
|
|
|
mDownloadErrorCallback->HandleEvent(EmptyCString());
|
2011-10-17 07:59:28 -07:00
|
|
|
mDownloadError = true;
|
2008-05-07 13:33:45 -07:00
|
|
|
mDBService->FinishUpdate();
|
|
|
|
return rv;
|
|
|
|
}
|
2008-05-07 06:18:38 -07:00
|
|
|
|
2008-05-07 13:33:45 -07:00
|
|
|
mPendingUpdates.RemoveElementAt(0);
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsUrlClassifierStreamUpdater::StreamFinished(nsresult status,
|
|
|
|
PRUint32 requestedDelay)
|
|
|
|
{
|
|
|
|
LOG(("nsUrlClassifierStreamUpdater::StreamFinished [%x, %d]", status, requestedDelay));
|
|
|
|
if (NS_FAILED(status) || mPendingUpdates.Length() == 0) {
|
|
|
|
// We're done.
|
2008-05-07 06:18:38 -07:00
|
|
|
mDBService->FinishUpdate();
|
2008-05-07 13:33:45 -07:00
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Wait the requested amount of time before starting a new stream.
|
|
|
|
nsresult rv;
|
|
|
|
mTimer = do_CreateInstance("@mozilla.org/timer;1", &rv);
|
|
|
|
if (NS_SUCCEEDED(rv)) {
|
|
|
|
rv = mTimer->InitWithCallback(this, requestedDelay,
|
|
|
|
nsITimer::TYPE_ONE_SHOT);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (NS_FAILED(rv)) {
|
|
|
|
NS_WARNING("Unable to initialize timer, fetching next safebrowsing item immediately");
|
|
|
|
return FetchNext();
|
2008-01-12 14:22:03 -08:00
|
|
|
}
|
2007-03-22 10:30:00 -07:00
|
|
|
|
2008-01-12 14:22:03 -08:00
|
|
|
return NS_OK;
|
|
|
|
}
|
2007-03-22 10:30:00 -07:00
|
|
|
|
2008-01-12 14:22:03 -08:00
|
|
|
NS_IMETHODIMP
|
|
|
|
nsUrlClassifierStreamUpdater::UpdateSuccess(PRUint32 requestedTimeout)
|
|
|
|
{
|
|
|
|
LOG(("nsUrlClassifierStreamUpdater::UpdateSuccess [this=%p]", this));
|
2008-05-02 02:54:01 -07:00
|
|
|
if (mPendingUpdates.Length() != 0) {
|
|
|
|
NS_WARNING("Didn't fetch all safebrowsing update redirects");
|
|
|
|
}
|
2008-01-12 14:22:03 -08:00
|
|
|
|
|
|
|
// DownloadDone() clears mSuccessCallback, so we save it off here.
|
2008-05-05 12:57:55 -07:00
|
|
|
nsCOMPtr<nsIUrlClassifierCallback> successCallback = mDownloadError ? nsnull : mSuccessCallback.get();
|
2008-01-12 14:22:03 -08:00
|
|
|
DownloadDone();
|
|
|
|
|
|
|
|
nsCAutoString strTimeout;
|
|
|
|
strTimeout.AppendInt(requestedTimeout);
|
|
|
|
if (successCallback) {
|
|
|
|
successCallback->HandleEvent(strTimeout);
|
|
|
|
}
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsUrlClassifierStreamUpdater::UpdateError(PRUint32 result)
|
|
|
|
{
|
2008-01-23 11:30:54 -08:00
|
|
|
LOG(("nsUrlClassifierStreamUpdater::UpdateError [this=%p]", this));
|
2008-01-12 14:22:03 -08:00
|
|
|
|
|
|
|
// DownloadDone() clears mUpdateErrorCallback, so we save it off here.
|
2008-05-05 12:57:55 -07:00
|
|
|
nsCOMPtr<nsIUrlClassifierCallback> errorCallback = mDownloadError ? nsnull : mUpdateErrorCallback.get();
|
2008-05-02 02:54:01 -07:00
|
|
|
|
2008-01-12 14:22:03 -08:00
|
|
|
DownloadDone();
|
|
|
|
|
|
|
|
nsCAutoString strResult;
|
|
|
|
strResult.AppendInt(result);
|
|
|
|
if (errorCallback) {
|
|
|
|
errorCallback->HandleEvent(strResult);
|
|
|
|
}
|
2007-03-22 10:30:00 -07:00
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
2007-07-25 23:38:43 -07:00
|
|
|
nsresult
|
|
|
|
nsUrlClassifierStreamUpdater::AddRequestBody(const nsACString &aRequestBody)
|
|
|
|
{
|
|
|
|
nsresult rv;
|
|
|
|
nsCOMPtr<nsIStringInputStream> strStream =
|
|
|
|
do_CreateInstance(NS_STRINGINPUTSTREAM_CONTRACTID, &rv);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
rv = strStream->SetData(aRequestBody.BeginReading(),
|
|
|
|
aRequestBody.Length());
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
nsCOMPtr<nsIUploadChannel> uploadChannel = do_QueryInterface(mChannel, &rv);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
rv = uploadChannel->SetUploadStream(strStream,
|
|
|
|
NS_LITERAL_CSTRING("text/plain"),
|
|
|
|
-1);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
nsCOMPtr<nsIHttpChannel> httpChannel = do_QueryInterface(mChannel, &rv);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
rv = httpChannel->SetRequestMethod(NS_LITERAL_CSTRING("POST"));
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
2008-01-12 14:22:03 -08:00
|
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
// nsIStreamListenerObserver implementation
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
2008-01-29 12:57:18 -08:00
|
|
|
nsUrlClassifierStreamUpdater::OnStartRequest(nsIRequest *request,
|
|
|
|
nsISupports* context)
|
2008-01-12 14:22:03 -08:00
|
|
|
{
|
|
|
|
nsresult rv;
|
2011-09-28 23:19:26 -07:00
|
|
|
bool downloadError = false;
|
2008-05-02 02:54:01 -07:00
|
|
|
nsCAutoString strStatus;
|
|
|
|
nsresult status = NS_OK;
|
2008-01-12 14:22:03 -08:00
|
|
|
|
2008-05-02 02:54:01 -07:00
|
|
|
// Only update if we got http success header
|
2008-01-12 14:22:03 -08:00
|
|
|
nsCOMPtr<nsIHttpChannel> httpChannel = do_QueryInterface(request);
|
|
|
|
if (httpChannel) {
|
|
|
|
rv = httpChannel->GetStatus(&status);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
if (NS_ERROR_CONNECTION_REFUSED == status ||
|
|
|
|
NS_ERROR_NET_TIMEOUT == status) {
|
2008-05-02 02:54:01 -07:00
|
|
|
// Assume we're overloading the server and trigger backoff.
|
2011-10-17 07:59:28 -07:00
|
|
|
downloadError = true;
|
2008-05-02 02:54:01 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
if (NS_SUCCEEDED(status)) {
|
2011-09-28 23:19:26 -07:00
|
|
|
bool succeeded = false;
|
2008-05-02 02:54:01 -07:00
|
|
|
rv = httpChannel->GetRequestSucceeded(&succeeded);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
if (!succeeded) {
|
|
|
|
// 404 or other error, pass error status back
|
|
|
|
LOG(("HTTP request returned failure code."));
|
|
|
|
|
|
|
|
rv = httpChannel->GetResponseStatus(&status);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
strStatus.AppendInt(status);
|
2011-10-17 07:59:28 -07:00
|
|
|
downloadError = true;
|
2008-05-02 02:54:01 -07:00
|
|
|
}
|
2008-01-12 14:22:03 -08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2008-05-02 02:54:01 -07:00
|
|
|
if (downloadError) {
|
|
|
|
mDownloadErrorCallback->HandleEvent(strStatus);
|
2011-10-17 07:59:28 -07:00
|
|
|
mDownloadError = true;
|
2008-05-02 02:54:01 -07:00
|
|
|
status = NS_ERROR_ABORT;
|
|
|
|
} else if (NS_SUCCEEDED(status)) {
|
2011-10-17 07:59:28 -07:00
|
|
|
mBeganStream = true;
|
2008-05-02 02:54:01 -07:00
|
|
|
rv = mDBService->BeginStream(mStreamTable, mServerMAC);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
}
|
|
|
|
|
|
|
|
mStreamTable.Truncate();
|
|
|
|
mServerMAC.Truncate();
|
|
|
|
|
|
|
|
return status;
|
2008-01-12 14:22:03 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsUrlClassifierStreamUpdater::OnDataAvailable(nsIRequest *request,
|
|
|
|
nsISupports* context,
|
|
|
|
nsIInputStream *aIStream,
|
|
|
|
PRUint32 aSourceOffset,
|
|
|
|
PRUint32 aLength)
|
|
|
|
{
|
|
|
|
if (!mDBService)
|
|
|
|
return NS_ERROR_NOT_INITIALIZED;
|
|
|
|
|
|
|
|
LOG(("OnDataAvailable (%d bytes)", aLength));
|
|
|
|
|
|
|
|
nsresult rv;
|
|
|
|
|
|
|
|
// Copy the data into a nsCString
|
|
|
|
nsCString chunk;
|
|
|
|
rv = NS_ConsumeStream(aIStream, aLength, chunk);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
//LOG(("Chunk (%d): %s\n\n", chunk.Length(), chunk.get()));
|
|
|
|
|
|
|
|
rv = mDBService->UpdateStream(chunk);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsUrlClassifierStreamUpdater::OnStopRequest(nsIRequest *request, nsISupports* context,
|
|
|
|
nsresult aStatus)
|
|
|
|
{
|
|
|
|
if (!mDBService)
|
|
|
|
return NS_ERROR_NOT_INITIALIZED;
|
|
|
|
|
|
|
|
LOG(("OnStopRequest (status %x)", aStatus));
|
|
|
|
|
|
|
|
nsresult rv;
|
|
|
|
|
2008-05-02 02:54:01 -07:00
|
|
|
if (NS_SUCCEEDED(aStatus)) {
|
|
|
|
// Success, finish this stream and move on to the next.
|
2008-01-12 14:22:03 -08:00
|
|
|
rv = mDBService->FinishStream();
|
2008-05-02 02:54:01 -07:00
|
|
|
} else if (mBeganStream) {
|
|
|
|
// We began this stream and couldn't finish it. We have to cancel the
|
|
|
|
// update, it's not in a consistent state.
|
2008-01-12 14:22:03 -08:00
|
|
|
rv = mDBService->CancelUpdate();
|
2008-05-02 02:54:01 -07:00
|
|
|
} else {
|
|
|
|
// The fetch failed, but we didn't start the stream (probably a
|
|
|
|
// server or connection error). We can commit what we've applied
|
|
|
|
// so far, and request again later.
|
|
|
|
rv = mDBService->FinishUpdate();
|
|
|
|
}
|
2008-01-12 14:22:03 -08:00
|
|
|
|
|
|
|
mChannel = nsnull;
|
|
|
|
|
|
|
|
return rv;
|
|
|
|
}
|
|
|
|
|
2007-03-22 10:30:00 -07:00
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
// nsIObserver implementation
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsUrlClassifierStreamUpdater::Observe(nsISupports *aSubject, const char *aTopic,
|
|
|
|
const PRUnichar *aData)
|
|
|
|
{
|
|
|
|
if (nsCRT::strcmp(aTopic, gQuitApplicationMessage) == 0) {
|
|
|
|
if (mIsUpdating && mChannel) {
|
|
|
|
LOG(("Cancel download"));
|
|
|
|
nsresult rv;
|
|
|
|
rv = mChannel->Cancel(NS_ERROR_ABORT);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
2011-10-17 07:59:28 -07:00
|
|
|
mIsUpdating = false;
|
2007-03-22 10:30:00 -07:00
|
|
|
mChannel = nsnull;
|
|
|
|
}
|
2008-05-07 13:33:45 -07:00
|
|
|
if (mTimer) {
|
|
|
|
mTimer->Cancel();
|
|
|
|
mTimer = nsnull;
|
|
|
|
}
|
2007-03-22 10:30:00 -07:00
|
|
|
}
|
|
|
|
return NS_OK;
|
|
|
|
}
|
2008-01-15 07:06:34 -08:00
|
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
// nsIBadCertListener2 implementation
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsUrlClassifierStreamUpdater::NotifyCertProblem(nsIInterfaceRequestor *socketInfo,
|
|
|
|
nsISSLStatus *status,
|
|
|
|
const nsACString &targetSite,
|
2011-09-28 23:19:26 -07:00
|
|
|
bool *_retval)
|
2008-01-15 07:06:34 -08:00
|
|
|
{
|
2011-10-17 07:59:28 -07:00
|
|
|
*_retval = true;
|
2008-01-15 07:06:34 -08:00
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
// nsISSLErrorListener implementation
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsUrlClassifierStreamUpdater::NotifySSLError(nsIInterfaceRequestor *socketInfo,
|
|
|
|
PRInt32 error,
|
|
|
|
const nsACString &targetSite,
|
2011-09-28 23:19:26 -07:00
|
|
|
bool *_retval)
|
2008-01-15 07:06:34 -08:00
|
|
|
{
|
2011-10-17 07:59:28 -07:00
|
|
|
*_retval = true;
|
2008-01-15 07:06:34 -08:00
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
// nsIInterfaceRequestor implementation
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsUrlClassifierStreamUpdater::GetInterface(const nsIID & eventSinkIID, void* *_retval)
|
|
|
|
{
|
|
|
|
return QueryInterface(eventSinkIID, _retval);
|
|
|
|
}
|
2008-05-07 13:33:45 -07:00
|
|
|
|
|
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
// nsITimerCallback implementation
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsUrlClassifierStreamUpdater::Notify(nsITimer *timer)
|
|
|
|
{
|
|
|
|
LOG(("nsUrlClassifierStreamUpdater::Notify [%p]", this));
|
|
|
|
|
|
|
|
mTimer = nsnull;
|
|
|
|
|
|
|
|
// Start the update process up again.
|
|
|
|
FetchNext();
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|