2012-09-24 11:53:49 -07:00
|
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this file,
|
|
|
|
* You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
|
|
|
|
#include "TCPSocketParent.h"
|
|
|
|
#include "jsapi.h"
|
|
|
|
#include "jsfriendapi.h"
|
|
|
|
#include "nsJSUtils.h"
|
|
|
|
#include "nsIDOMTCPSocket.h"
|
2013-05-10 19:39:45 -07:00
|
|
|
#include "nsContentUtils.h"
|
2013-05-22 09:05:26 -07:00
|
|
|
#include "nsCxPusher.h"
|
2012-09-24 11:53:49 -07:00
|
|
|
#include "mozilla/unused.h"
|
2012-12-22 03:53:38 -08:00
|
|
|
#include "mozilla/AppProcessChecker.h"
|
2012-09-24 11:53:49 -07:00
|
|
|
|
|
|
|
namespace IPC {
|
|
|
|
|
|
|
|
//Defined in TCPSocketChild.cpp
|
|
|
|
extern bool
|
2013-05-10 19:39:45 -07:00
|
|
|
DeserializeArrayBuffer(JS::Handle<JSObject*> aObj,
|
2013-04-19 02:25:33 -07:00
|
|
|
const InfallibleTArray<uint8_t>& aBuffer,
|
2013-05-10 19:39:45 -07:00
|
|
|
JS::MutableHandle<JS::Value> aVal);
|
2012-09-24 11:53:49 -07:00
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
namespace mozilla {
|
|
|
|
namespace dom {
|
|
|
|
|
|
|
|
static void
|
|
|
|
FireInteralError(mozilla::net::PTCPSocketParent* aActor, uint32_t aLineNo)
|
|
|
|
{
|
|
|
|
mozilla::unused <<
|
|
|
|
aActor->SendCallback(NS_LITERAL_STRING("onerror"),
|
2013-05-02 00:51:54 -07:00
|
|
|
TCPError(NS_LITERAL_STRING("InvalidStateError")),
|
|
|
|
NS_LITERAL_STRING("connecting"), 0);
|
2012-09-24 11:53:49 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMPL_CYCLE_COLLECTION_2(TCPSocketParent, mSocket, mIntermediary)
|
|
|
|
NS_IMPL_CYCLE_COLLECTING_ADDREF(TCPSocketParent)
|
|
|
|
NS_IMPL_CYCLE_COLLECTING_RELEASE(TCPSocketParent)
|
|
|
|
|
|
|
|
NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(TCPSocketParent)
|
|
|
|
NS_INTERFACE_MAP_ENTRY(nsITCPSocketParent)
|
|
|
|
NS_INTERFACE_MAP_ENTRY(nsISupports)
|
|
|
|
NS_INTERFACE_MAP_END
|
|
|
|
|
|
|
|
bool
|
|
|
|
TCPSocketParent::Init(const nsString& aHost, const uint16_t& aPort, const bool& aUseSSL,
|
2013-02-26 10:57:39 -08:00
|
|
|
const nsString& aBinaryType)
|
2012-09-24 11:53:49 -07:00
|
|
|
{
|
|
|
|
nsresult rv;
|
|
|
|
mIntermediary = do_CreateInstance("@mozilla.org/tcp-socket-intermediary;1", &rv);
|
|
|
|
if (NS_FAILED(rv)) {
|
|
|
|
FireInteralError(this, __LINE__);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
rv = mIntermediary->Open(this, aHost, aPort, aUseSSL, aBinaryType, getter_AddRefs(mSocket));
|
|
|
|
if (NS_FAILED(rv) || !mSocket) {
|
|
|
|
FireInteralError(this, __LINE__);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
TCPSocketParent::InitJS(const JS::Value& aIntermediary, JSContext* aCx)
|
|
|
|
{
|
|
|
|
MOZ_ASSERT(aIntermediary.isObject());
|
|
|
|
mIntermediaryObj = &aIntermediary.toObject();
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
TCPSocketParent::RecvSuspend()
|
|
|
|
{
|
|
|
|
NS_ENSURE_TRUE(mSocket, true);
|
|
|
|
nsresult rv = mSocket->Suspend();
|
|
|
|
NS_ENSURE_SUCCESS(rv, true);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
TCPSocketParent::RecvResume()
|
|
|
|
{
|
|
|
|
NS_ENSURE_TRUE(mSocket, true);
|
|
|
|
nsresult rv = mSocket->Resume();
|
|
|
|
NS_ENSURE_SUCCESS(rv, true);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
TCPSocketParent::RecvData(const SendableData& aData)
|
|
|
|
{
|
|
|
|
NS_ENSURE_TRUE(mIntermediary, true);
|
|
|
|
|
|
|
|
nsresult rv;
|
|
|
|
switch (aData.type()) {
|
|
|
|
case SendableData::TArrayOfuint8_t: {
|
2013-05-10 19:39:45 -07:00
|
|
|
AutoSafeJSContext cx;
|
|
|
|
JS::Rooted<JS::Value> val(cx);
|
|
|
|
JS::Rooted<JSObject*> obj(cx, mIntermediaryObj);
|
|
|
|
IPC::DeserializeArrayBuffer(obj, aData.get_ArrayOfuint8_t(), &val);
|
2012-09-24 11:53:49 -07:00
|
|
|
rv = mIntermediary->SendArrayBuffer(val);
|
|
|
|
NS_ENSURE_SUCCESS(rv, true);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
case SendableData::TnsString:
|
|
|
|
rv = mIntermediary->SendString(aData.get_nsString());
|
|
|
|
NS_ENSURE_SUCCESS(rv, true);
|
|
|
|
break;
|
|
|
|
|
|
|
|
default:
|
2013-04-20 13:36:43 -07:00
|
|
|
MOZ_NOT_REACHED("unexpected SendableData type");
|
2012-09-24 11:53:49 -07:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
TCPSocketParent::RecvClose()
|
|
|
|
{
|
|
|
|
NS_ENSURE_TRUE(mSocket, true);
|
|
|
|
nsresult rv = mSocket->Close();
|
|
|
|
NS_ENSURE_SUCCESS(rv, true);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
TCPSocketParent::SendCallback(const nsAString& aType, const JS::Value& aDataVal,
|
|
|
|
const nsAString& aReadyState, uint32_t aBuffered,
|
|
|
|
JSContext* aCx)
|
|
|
|
{
|
|
|
|
if (!mIPCOpen) {
|
|
|
|
NS_WARNING("Dropping callback due to no IPC connection");
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
CallbackData data;
|
|
|
|
if (aDataVal.isString()) {
|
|
|
|
JSString* jsstr = aDataVal.toString();
|
|
|
|
nsDependentJSString str;
|
|
|
|
if (!str.init(aCx, jsstr)) {
|
|
|
|
FireInteralError(this, __LINE__);
|
|
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
|
|
}
|
2013-05-02 00:51:54 -07:00
|
|
|
data = SendableData(str);
|
2012-09-24 11:53:49 -07:00
|
|
|
|
|
|
|
} else if (aDataVal.isUndefined() || aDataVal.isNull()) {
|
|
|
|
data = mozilla::void_t();
|
|
|
|
|
|
|
|
} else if (aDataVal.isObject()) {
|
|
|
|
JSObject* obj = &aDataVal.toObject();
|
2013-04-19 02:25:33 -07:00
|
|
|
if (JS_IsArrayBufferObject(obj)) {
|
|
|
|
uint32_t nbytes = JS_GetArrayBufferByteLength(obj);
|
|
|
|
uint8_t* buffer = JS_GetArrayBufferData(obj);
|
2012-09-24 11:53:49 -07:00
|
|
|
if (!buffer) {
|
|
|
|
FireInteralError(this, __LINE__);
|
|
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
|
|
}
|
|
|
|
FallibleTArray<uint8_t> fallibleArr;
|
|
|
|
if (!fallibleArr.InsertElementsAt(0, buffer, nbytes)) {
|
|
|
|
FireInteralError(this, __LINE__);
|
|
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
|
|
}
|
|
|
|
InfallibleTArray<uint8_t> arr;
|
|
|
|
arr.SwapElements(fallibleArr);
|
|
|
|
data = SendableData(arr);
|
|
|
|
|
|
|
|
} else {
|
2013-05-02 00:51:54 -07:00
|
|
|
nsDependentJSString name;
|
2012-09-24 11:53:49 -07:00
|
|
|
|
2013-05-10 19:39:45 -07:00
|
|
|
JS::Rooted<JS::Value> val(aCx);
|
|
|
|
if (!JS_GetProperty(aCx, obj, "name", val.address())) {
|
2013-05-02 00:51:54 -07:00
|
|
|
NS_ERROR("No name property on supposed error object");
|
2012-09-24 11:53:49 -07:00
|
|
|
} else if (JSVAL_IS_STRING(val)) {
|
2013-05-02 00:51:54 -07:00
|
|
|
if (!name.init(aCx, JSVAL_TO_STRING(val))) {
|
2012-09-24 11:53:49 -07:00
|
|
|
NS_WARNING("couldn't initialize string");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-05-02 00:51:54 -07:00
|
|
|
data = TCPError(name);
|
2012-09-24 11:53:49 -07:00
|
|
|
}
|
|
|
|
} else {
|
|
|
|
NS_ERROR("Unexpected JS value encountered");
|
|
|
|
FireInteralError(this, __LINE__);
|
|
|
|
return NS_ERROR_FAILURE;
|
|
|
|
}
|
|
|
|
mozilla::unused <<
|
|
|
|
PTCPSocketParent::SendCallback(nsString(aType), data,
|
|
|
|
nsString(aReadyState), aBuffered);
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
TCPSocketParent::ActorDestroy(ActorDestroyReason why)
|
|
|
|
{
|
|
|
|
MOZ_ASSERT(mIPCOpen);
|
|
|
|
mIPCOpen = false;
|
|
|
|
if (mSocket) {
|
|
|
|
mSocket->Close();
|
|
|
|
}
|
|
|
|
mSocket = nullptr;
|
|
|
|
mIntermediaryObj = nullptr;
|
|
|
|
mIntermediary = nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
TCPSocketParent::RecvRequestDelete()
|
|
|
|
{
|
|
|
|
mozilla::unused << Send__delete__(this);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
} // namespace dom
|
|
|
|
} // namespace mozilla
|