InjectedBundleUserMessageCoders.h [plain text]
#ifndef InjectedBundleUserMessageCoders_h
#define InjectedBundleUserMessageCoders_h
#include "UserMessageCoders.h"
#include "WebFrame.h"
#include "WebPage.h"
#include "WebPageGroupData.h"
#include "WebPageGroupProxy.h"
#include "WebProcess.h"
#if PLATFORM(MAC)
#include "ObjCObjectGraphCoders.h"
#endif
namespace WebKit {
class InjectedBundleUserMessageEncoder : public UserMessageEncoder<InjectedBundleUserMessageEncoder> {
public:
typedef UserMessageEncoder<InjectedBundleUserMessageEncoder> Base;
InjectedBundleUserMessageEncoder(APIObject* root)
: Base(root)
{
}
void encode(CoreIPC::ArgumentEncoder& encoder) const
{
APIObject::Type type = APIObject::TypeNull;
if (baseEncode(encoder, type))
return;
switch (type) {
case APIObject::TypeBundlePage: {
WebPage* page = static_cast<WebPage*>(m_root);
encoder << page->pageID();
break;
}
case APIObject::TypeBundleFrame: {
WebFrame* frame = static_cast<WebFrame*>(m_root);
encoder << frame->frameID();
break;
}
case APIObject::TypeBundlePageGroup: {
WebPageGroupProxy* pageGroup = static_cast<WebPageGroupProxy*>(m_root);
encoder << pageGroup->pageGroupID();
break;
}
#if PLATFORM(MAC)
case APIObject::TypeObjCObjectGraph: {
ObjCObjectGraph* objectGraph = static_cast<ObjCObjectGraph*>(m_root);
encoder << InjectedBundleObjCObjectGraphEncoder(objectGraph);
break;
}
#endif
default:
ASSERT_NOT_REACHED();
break;
}
}
};
class InjectedBundleUserMessageDecoder : public UserMessageDecoder<InjectedBundleUserMessageDecoder> {
public:
typedef UserMessageDecoder<InjectedBundleUserMessageDecoder> Base;
InjectedBundleUserMessageDecoder(RefPtr<APIObject>& root)
: Base(root)
{
}
InjectedBundleUserMessageDecoder(InjectedBundleUserMessageDecoder&, RefPtr<APIObject>& root)
: Base(root)
{
}
static bool decode(CoreIPC::ArgumentDecoder& decoder, InjectedBundleUserMessageDecoder& coder)
{
APIObject::Type type = APIObject::TypeNull;
if (!Base::baseDecode(decoder, coder, type))
return false;
if (coder.m_root || type == APIObject::TypeNull)
return true;
switch (type) {
case APIObject::TypePage: {
uint64_t pageID;
if (!decoder.decode(pageID))
return false;
coder.m_root = WebProcess::shared().webPage(pageID);
break;
}
case APIObject::TypeFrame: {
uint64_t frameID;
if (!decoder.decode(frameID))
return false;
coder.m_root = WebProcess::shared().webFrame(frameID);
break;
}
case APIObject::TypePageGroup: {
WebPageGroupData pageGroupData;
if (!decoder.decode(pageGroupData))
return false;
coder.m_root = WebProcess::shared().webPageGroup(pageGroupData);
break;
}
#if PLATFORM(MAC)
case APIObject::TypeObjCObjectGraph: {
RefPtr<ObjCObjectGraph> objectGraph;
InjectedBundleObjCObjectGraphDecoder objectGraphDecoder(objectGraph, &WebProcess::shared());
if (!decoder.decode(objectGraphDecoder))
return false;
coder.m_root = objectGraph.get();
break;
}
#endif
default:
return false;
}
return true;
}
};
}
#endif // InjectedBundleUserMessageCoders_h