2007-03-22 10:30:00 -07:00
|
|
|
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
|
|
|
|
/* ***** 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 mozilla.org code.
|
|
|
|
*
|
|
|
|
* The Initial Developer of the Original Code is
|
|
|
|
* Netscape Communications Corporation.
|
|
|
|
* Portions created by the Initial Developer are Copyright (C) 1998
|
|
|
|
* the Initial Developer. All Rights Reserved.
|
|
|
|
*
|
|
|
|
* Contributor(s):
|
|
|
|
*
|
|
|
|
* 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 "nspr.h"
|
|
|
|
#include "nsDataChannel.h"
|
|
|
|
#include "nsDataHandler.h"
|
|
|
|
#include "nsIURL.h"
|
|
|
|
#include "nsCRT.h"
|
|
|
|
#include "nsIComponentManager.h"
|
|
|
|
#include "nsIServiceManager.h"
|
|
|
|
#include "nsIInterfaceRequestor.h"
|
|
|
|
#include "nsIInterfaceRequestorUtils.h"
|
|
|
|
#include "nsIProgressEventSink.h"
|
|
|
|
#include "nsNetCID.h"
|
2007-08-21 10:28:22 -07:00
|
|
|
#include "nsNetError.h"
|
2007-03-22 10:30:00 -07:00
|
|
|
|
|
|
|
static NS_DEFINE_CID(kSimpleURICID, NS_SIMPLEURI_CID);
|
|
|
|
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
|
|
|
|
nsDataHandler::nsDataHandler() {
|
|
|
|
}
|
|
|
|
|
|
|
|
nsDataHandler::~nsDataHandler() {
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMPL_ISUPPORTS1(nsDataHandler, nsIProtocolHandler)
|
|
|
|
|
2010-06-10 11:11:40 -07:00
|
|
|
nsresult
|
2007-03-22 10:30:00 -07:00
|
|
|
nsDataHandler::Create(nsISupports* aOuter, const nsIID& aIID, void* *aResult) {
|
|
|
|
|
|
|
|
nsDataHandler* ph = new nsDataHandler();
|
|
|
|
if (ph == nsnull)
|
|
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
|
|
NS_ADDREF(ph);
|
|
|
|
nsresult rv = ph->QueryInterface(aIID, aResult);
|
|
|
|
NS_RELEASE(ph);
|
|
|
|
return rv;
|
|
|
|
}
|
|
|
|
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
// nsIProtocolHandler methods:
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsDataHandler::GetScheme(nsACString &result) {
|
|
|
|
result.AssignLiteral("data");
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsDataHandler::GetDefaultPort(PRInt32 *result) {
|
|
|
|
// no ports for data protocol
|
|
|
|
*result = -1;
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsDataHandler::GetProtocolFlags(PRUint32 *result) {
|
|
|
|
*result = URI_NORELATIVE | URI_NOAUTH | URI_INHERITS_SECURITY_CONTEXT |
|
2008-12-09 13:27:42 -08:00
|
|
|
URI_LOADABLE_BY_ANYONE | URI_NON_PERSISTABLE | URI_IS_LOCAL_RESOURCE;
|
2007-03-22 10:30:00 -07:00
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsDataHandler::NewURI(const nsACString &aSpec,
|
|
|
|
const char *aCharset, // ignore charset info
|
|
|
|
nsIURI *aBaseURI,
|
|
|
|
nsIURI **result) {
|
|
|
|
nsresult rv;
|
2011-05-21 18:12:46 -07:00
|
|
|
nsRefPtr<nsIURI> uri;
|
2007-03-22 10:30:00 -07:00
|
|
|
|
2007-08-21 10:28:22 -07:00
|
|
|
nsCString spec(aSpec);
|
|
|
|
|
2011-05-21 18:12:46 -07:00
|
|
|
if (aBaseURI && !spec.IsEmpty() && spec[0] == '#') {
|
|
|
|
// Looks like a reference instead of a fully-specified URI.
|
|
|
|
// --> initialize |uri| as a clone of |aBaseURI|, with ref appended.
|
|
|
|
rv = aBaseURI->Clone(getter_AddRefs(uri));
|
|
|
|
if (NS_FAILED(rv))
|
|
|
|
return rv;
|
|
|
|
rv = uri->SetRef(spec);
|
|
|
|
} else {
|
|
|
|
// Otherwise, we'll assume |spec| is a fully-specified data URI
|
2011-05-24 08:40:10 -07:00
|
|
|
nsCAutoString contentType, contentCharset, dataBuffer, hashRef;
|
2011-09-28 23:19:26 -07:00
|
|
|
bool base64;
|
2011-05-24 08:40:10 -07:00
|
|
|
rv = ParseURI(spec, contentType, contentCharset, base64, dataBuffer, hashRef);
|
2011-05-21 18:12:46 -07:00
|
|
|
if (NS_FAILED(rv))
|
|
|
|
return rv;
|
|
|
|
|
|
|
|
// Strip whitespace unless this is text, where whitespace is important
|
|
|
|
// Don't strip escaped whitespace though (bug 391951)
|
|
|
|
if (base64 || (strncmp(contentType.get(),"text/",5) != 0 &&
|
|
|
|
contentType.Find("xml") == kNotFound)) {
|
|
|
|
// it's ascii encoded binary, don't let any spaces in
|
|
|
|
spec.StripWhitespace();
|
|
|
|
}
|
|
|
|
|
|
|
|
uri = do_CreateInstance(kSimpleURICID, &rv);
|
|
|
|
if (NS_FAILED(rv))
|
|
|
|
return rv;
|
|
|
|
rv = uri->SetSpec(spec);
|
2007-08-21 10:28:22 -07:00
|
|
|
}
|
2007-03-22 10:30:00 -07:00
|
|
|
|
2011-05-21 18:12:46 -07:00
|
|
|
if (NS_FAILED(rv))
|
2007-03-22 10:30:00 -07:00
|
|
|
return rv;
|
|
|
|
|
2011-05-21 18:12:46 -07:00
|
|
|
uri.forget(result);
|
2007-03-22 10:30:00 -07:00
|
|
|
return rv;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
2007-08-21 10:28:22 -07:00
|
|
|
nsDataHandler::NewChannel(nsIURI* uri, nsIChannel* *result) {
|
2007-03-22 10:30:00 -07:00
|
|
|
NS_ENSURE_ARG_POINTER(uri);
|
|
|
|
nsDataChannel* channel = new nsDataChannel(uri);
|
|
|
|
if (!channel)
|
|
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
|
|
NS_ADDREF(channel);
|
|
|
|
|
|
|
|
nsresult rv = channel->Init();
|
|
|
|
if (NS_FAILED(rv)) {
|
|
|
|
NS_RELEASE(channel);
|
|
|
|
return rv;
|
|
|
|
}
|
|
|
|
|
|
|
|
*result = channel;
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
2011-09-28 23:19:26 -07:00
|
|
|
nsDataHandler::AllowPort(PRInt32 port, const char *scheme, bool *_retval) {
|
2007-03-22 10:30:00 -07:00
|
|
|
// don't override anything.
|
2011-10-17 07:59:28 -07:00
|
|
|
*_retval = false;
|
2007-03-22 10:30:00 -07:00
|
|
|
return NS_OK;
|
|
|
|
}
|
2007-08-21 10:28:22 -07:00
|
|
|
|
|
|
|
nsresult
|
|
|
|
nsDataHandler::ParseURI(nsCString& spec,
|
|
|
|
nsCString& contentType,
|
|
|
|
nsCString& contentCharset,
|
2011-09-28 23:19:26 -07:00
|
|
|
bool& isBase64,
|
2011-05-24 08:40:10 -07:00
|
|
|
nsCString& dataBuffer,
|
|
|
|
nsCString& hashRef) {
|
2011-10-17 07:59:28 -07:00
|
|
|
isBase64 = false;
|
2007-08-21 10:28:22 -07:00
|
|
|
|
|
|
|
// move past "data:"
|
2009-06-23 04:44:37 -07:00
|
|
|
char *buffer = (char *) PL_strcasestr(spec.BeginWriting(), "data:");
|
2007-08-21 10:28:22 -07:00
|
|
|
if (!buffer) {
|
|
|
|
// malformed uri
|
|
|
|
return NS_ERROR_MALFORMED_URI;
|
|
|
|
}
|
|
|
|
buffer += 5;
|
|
|
|
|
|
|
|
// First, find the start of the data
|
|
|
|
char *comma = strchr(buffer, ',');
|
|
|
|
if (!comma)
|
|
|
|
return NS_ERROR_MALFORMED_URI;
|
|
|
|
|
|
|
|
*comma = '\0';
|
|
|
|
|
|
|
|
// determine if the data is base64 encoded.
|
2009-06-23 04:44:37 -07:00
|
|
|
char *base64 = PL_strcasestr(buffer, ";base64");
|
2007-08-21 10:28:22 -07:00
|
|
|
if (base64) {
|
2011-10-17 07:59:28 -07:00
|
|
|
isBase64 = true;
|
2007-08-21 10:28:22 -07:00
|
|
|
*base64 = '\0';
|
|
|
|
}
|
|
|
|
|
|
|
|
if (comma == buffer) {
|
|
|
|
// nothing but data
|
|
|
|
contentType.AssignLiteral("text/plain");
|
|
|
|
contentCharset.AssignLiteral("US-ASCII");
|
|
|
|
} else {
|
|
|
|
// everything else is content type
|
|
|
|
char *semiColon = (char *) strchr(buffer, ';');
|
|
|
|
if (semiColon)
|
|
|
|
*semiColon = '\0';
|
|
|
|
|
|
|
|
if (semiColon == buffer || base64 == buffer) {
|
|
|
|
// there is no content type, but there are other parameters
|
|
|
|
contentType.AssignLiteral("text/plain");
|
|
|
|
} else {
|
|
|
|
contentType = buffer;
|
|
|
|
ToLowerCase(contentType);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (semiColon) {
|
|
|
|
char *charset = PL_strcasestr(semiColon + 1, "charset=");
|
|
|
|
if (charset)
|
|
|
|
contentCharset = charset + sizeof("charset=") - 1;
|
|
|
|
|
|
|
|
*semiColon = ';';
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
*comma = ',';
|
|
|
|
if (isBase64)
|
|
|
|
*base64 = ';';
|
|
|
|
|
|
|
|
contentType.StripWhitespace();
|
|
|
|
contentCharset.StripWhitespace();
|
|
|
|
|
2011-05-24 08:40:10 -07:00
|
|
|
// Split encoded data from terminal "#ref" (if present)
|
|
|
|
char *data = comma + 1;
|
|
|
|
char *hash = strchr(data, '#');
|
|
|
|
if (!hash) {
|
|
|
|
dataBuffer.Assign(data);
|
|
|
|
hashRef.Truncate();
|
|
|
|
} else {
|
|
|
|
dataBuffer.Assign(data, hash - data);
|
|
|
|
hashRef.Assign(hash);
|
|
|
|
}
|
2007-08-21 10:28:22 -07:00
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|