#if !defined(DCONNECT_SINGLETHREADED) && !defined(DCONNECT_MULTITHREADED)

#define DCONNECT_MULTITHREADED

#endif

#include "ipcIDConnectService.h"

#include "ipcdclient.h"

#include "nsIInterfaceInfo.h"

#include "nsIInterfaceInfoManager.h"

#include "nsCOMPtr.h"

#include "nsAutoPtr.h"

#include "nsDataHashtable.h"

#include "nsHashKeys.h"

#include "nsHashSets.h"

#include "nsVoidArray.h"

#include "nsAutoLock.h"

#include "xptcall.h"

#include "xptinfo.h"

#if defined(DCONNECT_MULTITHREADED)

#include "ipcList.h"

struct DConnectOp;

struct DConnectRequest : public ipcListNode<DConnectRequest>

{

DConnectRequest (PRUint32 aPeer, const DConnectOp *aOp, PRUint32 aOpLen)

: peer(aPeer)

, opLen(aOpLen)

{

op = (const DConnectOp *) malloc(aOpLen);

memcpy ((void *) op, aOp, aOpLen);

}

~DConnectRequest() { free((void *) op); }

const PRUint32 peer;

const DConnectOp *op;

const PRUint32 opLen;

};

#endif

class ipcMessageWriter;

class DConnectInstanceKey : public PLDHashEntryHdr

{

public:

struct Key

{

Key(PRUint32 aPeer, const nsISupports *aObj, const nsID *aIID)

: mPeer (aPeer), mObj (aObj), mIID (aIID) {}

const PRUint32 mPeer;

const nsISupports *mObj;

const nsIID *mIID;

};

typedef const Key &KeyType;

typedef const Key *KeyTypePointer;

DConnectInstanceKey(const Key *aKey) : mKey (*aKey) {}

DConnectInstanceKey(const DConnectInstanceKey &toCopy) : mKey (toCopy.mKey) {}

~DConnectInstanceKey() {}

KeyType GetKey() const { return mKey; }

KeyTypePointer GetKeyPointer() const { return &mKey; }

PRBool KeyEquals(KeyTypePointer aKey) const {

return mKey.mPeer == aKey->mPeer &&

mKey.mObj == aKey->mObj &&

mKey.mIID->Equals(*aKey->mIID);

}

static KeyTypePointer KeyToPointer(KeyType aKey) { return &aKey; }

static PLDHashNumber HashKey(KeyTypePointer aKey) {

return aKey->mPeer ^

(NS_PTR_TO_INT32(aKey->mObj) >> 2) ^

nsIDHashKey::HashKey(aKey->mIID);

}

enum { ALLOW_MEMMOVE = PR_TRUE };

private:

const Key mKey;

};

class DConnectInstance;

typedef nsDataHashtable<DConnectInstanceKey, DConnectInstance *> DConnectInstanceMap;

class DConnectInstanceSet : public nsVoidHashSet

{

public:

PRBool Init(PRUint32 initSize = PL_DHASH_MIN_SIZE) {

return nsVoidHashSet::Init(initSize) == NS_OK;

}

void Clear() {

PL_DHashTableEnumerate(&mHashTable, PL_DHashStubEnumRemove, nsnull);

}

};

typedef class DConnectInstance* DConAddr;

class DConnectStubKey : public PLDHashEntryHdr

{

public:

struct Key

{

Key(PRUint32 aPeer, const DConAddr aInstance)

: mPeer (aPeer), mInstance (aInstance) {}

const PRUint32 mPeer;

const DConAddr mInstance;

};

typedef const Key &KeyType;

typedef const Key *KeyTypePointer;

DConnectStubKey(const Key *aKey) : mKey (*aKey) {}

DConnectStubKey(const DConnectStubKey &toCopy) : mKey (toCopy.mKey) {}

~DConnectStubKey() {}

KeyType GetKey() const { return mKey; }

KeyTypePointer GetKeyPointer() const { return &mKey; }

PRBool KeyEquals(KeyTypePointer aKey) const {

return mKey.mPeer == aKey->mPeer &&

mKey.mInstance == aKey->mInstance;

}

static KeyTypePointer KeyToPointer(KeyType aKey) { return &aKey; }

static PLDHashNumber HashKey(KeyTypePointer aKey) {

return aKey->mPeer ^

(NS_PTR_TO_INT32(aKey->mInstance) >> 2);

}

enum { ALLOW_MEMMOVE = PR_TRUE };

private:

const Key mKey;

};

class DConnectStub;

typedef nsDataHashtable<DConnectStubKey, DConnectStub *> DConnectStubMap;

class ipcDConnectService : public ipcIDConnectService

, public ipcIMessageObserver

, public ipcIClientObserver

{

public:

NS_DECL_ISUPPORTS

NS_DECL_IPCIDCONNECTSERVICE

NS_DECL_IPCIMESSAGEOBSERVER

NS_DECL_IPCICLIENTOBSERVER

ipcDConnectService();

NS_HIDDEN_(nsresult) Init();

NS_HIDDEN_(void) Shutdown();

NS_HIDDEN_(nsresult) GetInterfaceInfo(const nsID &iid, nsIInterfaceInfo **);

NS_HIDDEN_(nsresult) GetIIDForMethodParam(nsIInterfaceInfo *iinfo,

const nsXPTMethodInfo *methodInfo,

const nsXPTParamInfo &paramInfo,

const nsXPTType &type,

PRUint16 methodIndex,

PRUint8 paramIndex,

nsXPTCMiniVariant *dispatchParams,

PRBool isFullVariantArray,

nsID &result);

NS_HIDDEN_(nsresult) SerializeInterfaceParam(ipcMessageWriter &writer,

PRUint32 peer, const nsID &iid,

nsISupports *obj,

nsVoidArray &wrappers);

NS_HIDDEN_(void) ReleaseWrappers(nsVoidArray &wrappers);

NS_HIDDEN_(nsresult) CreateStub(const nsID &, PRUint32, DConAddr, DConnectStub **);

NS_HIDDEN_(PRBool) FindStubAndAddRef(PRUint32, const DConAddr, DConnectStub **);

// public only for DConnectStub::~DConnectStub()

NS_HIDDEN_(void) DeleteStub(DConnectStub *);

// public only for DConnectInstance::Release()

NS_HIDDEN_(void) DeleteInstance(DConnectInstance *);

// public only for DConnectStub::CallMethod()

NS_HIDDEN_(PRBool) CheckInstanceAndAddRef(DConnectInstance *);

PRLock *StubLock() { return mStubLock; }

static nsRefPtr <ipcDConnectService> GetInstance() {

return nsRefPtr <ipcDConnectService> (mInstance);

}

private:

NS_HIDDEN ~ipcDConnectService();

NS_HIDDEN_(nsresult) StoreInstance(DConnectInstance *);

NS_HIDDEN_(PRBool) FindInstanceAndAddRef(PRUint32,

const nsISupports *,

const nsIID *,

DConnectInstance **);

NS_HIDDEN_(nsresult) StoreStub(DConnectStub *);

NS_HIDDEN_(void) OnIncomingRequest(PRUint32 peer, const struct DConnectOp *op, PRUint32 opLen);

NS_HIDDEN_(void) OnSetup(PRUint32 peer, const struct DConnectSetup *, PRUint32 opLen);

NS_HIDDEN_(void) OnRelease(PRUint32 peer, const struct DConnectRelease *);

NS_HIDDEN_(void) OnInvoke(PRUint32 peer, const struct DConnectInvoke *, PRUint32 opLen);

#if defined(DCONNECT_MULTITHREADED)

NS_HIDDEN_(nsresult) CreateWorker();

#endif

private:

nsCOMPtr<nsIInterfaceInfoManager> mIIM;

// lock to protect access to instance and stub sets and the disconnected flag

PRLock *mLock;

// table of local object instances allocated on behalf of a peer

// (keys are interface pointers of real objects these instances represent)

DConnectInstanceMap mInstances;

// hashset containing the same instances as above

// (used for quick parameter validity checks)

DConnectInstanceSet mInstanceSet;

// table of remote object stubs allocated to communicate with peer's instances

DConnectStubMap mStubs;

// this is true after IPC_Shutdown() has been called

PRBool mDisconnected;

// our IPC client ID

PRUint32 mSelfID;

// global lock to protect access to DConnectStub internal data

PRLock *mStubLock;

#if defined(DCONNECT_MULTITHREADED)

friend class DConnectWorker;

// pool of worker threads to serve incoming requests

nsVoidArray mWorkers;

// queue of pending requests

ipcList<DConnectRequest> mPendingQ;

// monitor to protect mPendingQ

PRMonitor *mPendingMon;

// number of waiting workers

PRUint32 mWaitingWorkers;

// monitor used to wait on changes in mWaitingWorkers.

PRMonitor *mWaitingWorkersMon;

#endif

// global ipcDConnectService instance for internal usage

static ipcDConnectService *mInstance;

};

#define IPC_DCONNECTSERVICE_CLASSNAME \

"ipcDConnectService"

#define IPC_DCONNECTSERVICE_CONTRACTID \

"@mozilla.org/ipc/dconnect-service;1"

#define IPC_DCONNECTSERVICE_CID \

{ /* 63a5d9dc-4828-425a-bd50-bd10a4b26f2c */ \

0x63a5d9dc, \

0x4828, \

0x425a, \

{0xbd, 0x50, 0xbd, 0x10, 0xa4, 0xb2, 0x6f, 0x2c} \

}