зеркало из https://github.com/mozilla/pjs.git
331 строка
9.4 KiB
C++
331 строка
9.4 KiB
C++
/* -*- 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 mozilla.org code.
|
|
*
|
|
* The Initial Developer of the Original Code is
|
|
* the Mozilla Foundation.
|
|
* Portions created by the Initial Developer are Copyright (C) 2010
|
|
* the Initial Developer. All Rights Reserved.
|
|
*
|
|
* Contributor(s):
|
|
* Josh Matthews <josh@joshmatthews.net> (Initial Developer)
|
|
*
|
|
* 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 ***** */
|
|
|
|
#ifndef mozilla_net_NeckoMessageUtils_h
|
|
#define mozilla_net_NeckoMessageUtils_h
|
|
|
|
#include "IPC/IPCMessageUtils.h"
|
|
#include "nsStringGlue.h"
|
|
#include "nsIURI.h"
|
|
#include "nsIIPCSerializable.h"
|
|
#include "nsIClassInfo.h"
|
|
#include "nsComponentManagerUtils.h"
|
|
#include "nsNetUtil.h"
|
|
#include "nsStringStream.h"
|
|
|
|
namespace IPC {
|
|
|
|
// Since IPDL doesn't have any knowledge of pointers, there's no easy way to
|
|
// pass around nsIURI pointers. This is a very thin wrapper that IPDL can
|
|
// easily work with, allowing for conversion to and from an nsIURI pointer.
|
|
|
|
class URI {
|
|
public:
|
|
URI() : mURI(nsnull) {}
|
|
URI(nsIURI* aURI) : mURI(aURI) {}
|
|
operator nsIURI*() const { return mURI.get(); }
|
|
|
|
friend struct ParamTraits<URI>;
|
|
|
|
private:
|
|
// Unimplemented
|
|
URI& operator=(URI&);
|
|
|
|
nsCOMPtr<nsIURI> mURI;
|
|
};
|
|
|
|
template<>
|
|
struct ParamTraits<URI>
|
|
{
|
|
typedef URI paramType;
|
|
|
|
static void Write(Message* aMsg, const paramType& aParam)
|
|
{
|
|
bool isNull = !aParam.mURI;
|
|
WriteParam(aMsg, isNull);
|
|
if (isNull)
|
|
return;
|
|
|
|
nsCOMPtr<nsIIPCSerializable> serializable = do_QueryInterface(aParam.mURI);
|
|
if (!serializable) {
|
|
nsCString scheme;
|
|
aParam.mURI->GetScheme(scheme);
|
|
if (!scheme.EqualsASCII("about:") &&
|
|
!scheme.EqualsASCII("javascript:") &&
|
|
!scheme.EqualsASCII("javascript"))
|
|
NS_WARNING("All IPDL URIs must be serializable or an allowed scheme");
|
|
}
|
|
|
|
bool isSerialized = !!serializable;
|
|
WriteParam(aMsg, isSerialized);
|
|
if (!isSerialized) {
|
|
nsCString spec, charset;
|
|
aParam.mURI->GetSpec(spec);
|
|
aParam.mURI->GetOriginCharset(charset);
|
|
WriteParam(aMsg, spec);
|
|
WriteParam(aMsg, charset);
|
|
return;
|
|
}
|
|
|
|
nsCOMPtr<nsIClassInfo> classInfo = do_QueryInterface(aParam.mURI);
|
|
char cidStr[NSID_LENGTH];
|
|
nsCID cid;
|
|
#ifdef DEBUG
|
|
nsresult rv =
|
|
#endif
|
|
classInfo->GetClassIDNoAlloc(&cid);
|
|
NS_ABORT_IF_FALSE(NS_SUCCEEDED(rv), "All IPDL URIs must report a valid class ID");
|
|
|
|
cid.ToProvidedString(cidStr);
|
|
WriteParam(aMsg, nsCAutoString(cidStr));
|
|
serializable->Write(aMsg);
|
|
}
|
|
|
|
static bool Read(const Message* aMsg, void** aIter, paramType* aResult)
|
|
{
|
|
bool isNull;
|
|
if (!ReadParam(aMsg, aIter, &isNull))
|
|
return false;
|
|
if (isNull) {
|
|
aResult->mURI = nsnull;
|
|
return true;
|
|
}
|
|
|
|
bool isSerialized;
|
|
if (!ReadParam(aMsg, aIter, &isSerialized))
|
|
return false;
|
|
if (!isSerialized) {
|
|
nsCString spec, charset;
|
|
if (!ReadParam(aMsg, aIter, &spec) ||
|
|
!ReadParam(aMsg, aIter, &charset))
|
|
return false;
|
|
|
|
nsCOMPtr<nsIURI> uri;
|
|
nsresult rv = NS_NewURI(getter_AddRefs(uri), spec, charset.get());
|
|
if (NS_FAILED(rv))
|
|
return false;
|
|
|
|
uri.swap(aResult->mURI);
|
|
return true;
|
|
}
|
|
|
|
nsCAutoString cidStr;
|
|
nsCID cid;
|
|
if (!ReadParam(aMsg, aIter, &cidStr) ||
|
|
!cid.Parse(cidStr.get()))
|
|
return false;
|
|
|
|
nsCOMPtr<nsIURI> uri = do_CreateInstance(cid);
|
|
if (!uri)
|
|
return false;
|
|
nsCOMPtr<nsIIPCSerializable> serializable = do_QueryInterface(uri);
|
|
if (!serializable || !serializable->Read(aMsg, aIter))
|
|
return false;
|
|
|
|
uri.swap(aResult->mURI);
|
|
return true;
|
|
}
|
|
|
|
static void Log(const paramType& aParam, std::wstring* aLog)
|
|
{
|
|
nsIURI* uri = aParam.mURI;
|
|
if (uri) {
|
|
nsCString spec;
|
|
uri->GetSpec(spec);
|
|
aLog->append(StringPrintf(L"[%s]", spec.get()));
|
|
}
|
|
else {
|
|
aLog->append(L"[]");
|
|
}
|
|
}
|
|
};
|
|
|
|
class InputStream {
|
|
public:
|
|
InputStream() : mStream(nsnull) {}
|
|
InputStream(nsIInputStream* aStream) : mStream(aStream) {}
|
|
operator nsIInputStream*() const { return mStream.get(); }
|
|
|
|
friend struct ParamTraits<InputStream>;
|
|
|
|
private:
|
|
// Unimplemented
|
|
InputStream& operator=(InputStream&);
|
|
|
|
nsCOMPtr<nsIInputStream> mStream;
|
|
};
|
|
|
|
template<>
|
|
struct ParamTraits<InputStream>
|
|
{
|
|
typedef InputStream paramType;
|
|
|
|
static void Write(Message* aMsg, const paramType& aParam)
|
|
{
|
|
bool isNull = !aParam.mStream;
|
|
aMsg->WriteBool(isNull);
|
|
|
|
if (isNull)
|
|
return;
|
|
|
|
nsCOMPtr<nsIIPCSerializable> serializable = do_QueryInterface(aParam.mStream);
|
|
bool isSerializable = !!serializable;
|
|
WriteParam(aMsg, isSerializable);
|
|
|
|
if (!serializable) {
|
|
NS_WARNING("nsIInputStream implementation doesn't support nsIIPCSerializable; falling back to copying data");
|
|
|
|
nsCString streamString;
|
|
PRUint32 bytes;
|
|
|
|
aParam.mStream->Available(&bytes);
|
|
if (bytes > 0) {
|
|
nsresult rv = NS_ReadInputStreamToString(aParam.mStream, streamString, bytes);
|
|
NS_ABORT_IF_FALSE(NS_SUCCEEDED(rv), "Can't read input stream into a string!");
|
|
}
|
|
|
|
WriteParam(aMsg, streamString);
|
|
return;
|
|
}
|
|
|
|
nsCOMPtr<nsIClassInfo> classInfo = do_QueryInterface(aParam.mStream);
|
|
char cidStr[NSID_LENGTH];
|
|
nsCID cid;
|
|
nsresult rv = classInfo->GetClassIDNoAlloc(&cid);
|
|
NS_ABORT_IF_FALSE(NS_SUCCEEDED(rv), "All IPDL streams must report a valid class ID");
|
|
|
|
cid.ToProvidedString(cidStr);
|
|
WriteParam(aMsg, nsCAutoString(cidStr));
|
|
serializable->Write(aMsg);
|
|
}
|
|
|
|
static bool Read(const Message* aMsg, void** aIter, paramType* aResult)
|
|
{
|
|
bool isNull;
|
|
if (!ReadParam(aMsg, aIter, &isNull))
|
|
return false;
|
|
|
|
if (isNull) {
|
|
aResult->mStream = nsnull;
|
|
return true;
|
|
}
|
|
|
|
bool isSerializable;
|
|
if (!ReadParam(aMsg, aIter, &isSerializable))
|
|
return false;
|
|
|
|
nsCOMPtr<nsIInputStream> stream;
|
|
if (!isSerializable) {
|
|
nsCString streamString;
|
|
if (!ReadParam(aMsg, aIter, &streamString))
|
|
return false;
|
|
|
|
nsresult rv = NS_NewCStringInputStream(getter_AddRefs(stream), streamString);
|
|
if (NS_FAILED(rv))
|
|
return false;
|
|
} else {
|
|
nsCAutoString cidStr;
|
|
nsCID cid;
|
|
if (!ReadParam(aMsg, aIter, &cidStr) ||
|
|
!cid.Parse(cidStr.get()))
|
|
return false;
|
|
|
|
stream = do_CreateInstance(cid);
|
|
if (!stream)
|
|
return false;
|
|
nsCOMPtr<nsIIPCSerializable> serializable = do_QueryInterface(stream);
|
|
if (!serializable || !serializable->Read(aMsg, aIter))
|
|
return false;
|
|
}
|
|
|
|
stream.swap(aResult->mStream);
|
|
return true;
|
|
}
|
|
};
|
|
|
|
// nsIPermissionManager utilities
|
|
|
|
struct Permission
|
|
{
|
|
nsCString host, type;
|
|
PRUint32 capability, expireType;
|
|
PRInt64 expireTime;
|
|
|
|
Permission() { }
|
|
Permission(const nsCString& aHost,
|
|
const nsCString& aType,
|
|
const PRUint32 aCapability,
|
|
const PRUint32 aExpireType,
|
|
const PRInt64 aExpireTime) : host(aHost),
|
|
type(aType),
|
|
capability(aCapability),
|
|
expireType(aExpireType),
|
|
expireTime(aExpireTime) { }
|
|
};
|
|
|
|
template<>
|
|
struct ParamTraits<Permission>
|
|
{
|
|
static void Write(Message* aMsg, const Permission& aParam)
|
|
{
|
|
WriteParam(aMsg, aParam.host);
|
|
WriteParam(aMsg, aParam.type);
|
|
WriteParam(aMsg, aParam.capability);
|
|
WriteParam(aMsg, aParam.expireType);
|
|
WriteParam(aMsg, aParam.expireTime);
|
|
}
|
|
|
|
static bool Read(const Message* aMsg, void** aIter, Permission* aResult)
|
|
{
|
|
return ReadParam(aMsg, aIter, &aResult->host) &&
|
|
ReadParam(aMsg, aIter, &aResult->type) &&
|
|
ReadParam(aMsg, aIter, &aResult->capability) &&
|
|
ReadParam(aMsg, aIter, &aResult->expireType) &&
|
|
ReadParam(aMsg, aIter, &aResult->expireTime);
|
|
}
|
|
|
|
static void Log(const Permission& aParam, std::wstring* aLog)
|
|
{
|
|
aLog->append(StringPrintf(L"[%s]", aParam.host.get()));
|
|
}
|
|
};
|
|
|
|
}
|
|
|
|
#endif // mozilla_net_NeckoMessageUtils_h
|