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(COCOA)
#include "ObjCObjectGraphCoders.h"
#endif
namespace WebKit {
class InjectedBundleUserMessageEncoder : public UserMessageEncoder<InjectedBundleUserMessageEncoder> {
public:
typedef UserMessageEncoder<InjectedBundleUserMessageEncoder> Base;
InjectedBundleUserMessageEncoder(API::Object* root)
: Base(root)
{
}
InjectedBundleUserMessageEncoder(const InjectedBundleUserMessageEncoder&, API::Object* root)
: Base(root)
{
}
void encode(IPC::ArgumentEncoder& encoder) const
{
API::Object::Type type = API::Object::Type::Null;
if (baseEncode(encoder, *this, type))
return;
switch (type) {
case API::Object::Type::BundlePage: {
WebPage* page = static_cast<WebPage*>(m_root);
encoder << page->pageID();
break;
}
case API::Object::Type::BundleFrame: {
WebFrame* frame = static_cast<WebFrame*>(m_root);
encoder << frame->frameID();
break;
}
case API::Object::Type::BundlePageGroup: {
WebPageGroupProxy* pageGroup = static_cast<WebPageGroupProxy*>(m_root);
encoder << pageGroup->pageGroupID();
break;
}
#if PLATFORM(COCOA)
case API::Object::Type::ObjCObjectGraph: {
ObjCObjectGraph* objectGraph = static_cast<ObjCObjectGraph*>(m_root);
encoder << InjectedBundleObjCObjectGraphEncoder(objectGraph, WebProcess::shared());
break;
}
#endif
default:
ASSERT_NOT_REACHED();
break;
}
}
};
class InjectedBundleUserMessageDecoder : public UserMessageDecoder<InjectedBundleUserMessageDecoder> {
public:
typedef UserMessageDecoder<InjectedBundleUserMessageDecoder> Base;
InjectedBundleUserMessageDecoder(RefPtr<API::Object>& root)
: Base(root)
{
}
InjectedBundleUserMessageDecoder(InjectedBundleUserMessageDecoder&, RefPtr<API::Object>& root)
: Base(root)
{
}
static bool decode(IPC::ArgumentDecoder& decoder, InjectedBundleUserMessageDecoder& coder)
{
API::Object::Type type = API::Object::Type::Null;
if (!Base::baseDecode(decoder, coder, type))
return false;
if (coder.m_root || type == API::Object::Type::Null)
return true;
switch (type) {
case API::Object::Type::Page: {
uint64_t pageID;
if (!decoder.decode(pageID))
return false;
coder.m_root = WebProcess::shared().webPage(pageID);
break;
}
case API::Object::Type::Frame: {
uint64_t frameID;
if (!decoder.decode(frameID))
return false;
coder.m_root = WebProcess::shared().webFrame(frameID);
break;
}
case API::Object::Type::PageGroup: {
WebPageGroupData pageGroupData;
if (!decoder.decode(pageGroupData))
return false;
coder.m_root = WebProcess::shared().webPageGroup(pageGroupData);
break;
}
#if PLATFORM(COCOA)
case API::Object::Type::ObjCObjectGraph: {
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