Bommels05/Mozilla
Bommels05/Mozilla
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Mozilla/mozilla/dom/tools/JSStubGen.cpp
norris%netscape.com 2429ef124a Fix 18553 [DOGFOOD] addEventListener allows sniffing keystrokes 
Add checks to nsScriptSecurityManager::CheckCanListenTo that take
a principal and ensure that the currently executing script code
either is from the same origin as that principal or has the
UniversalBrowserRead privilege enabled. (chrome code has all
privileges enabled by default.) It's okay for the principal passed in
to be null. That just signifies a privileged window/document that only
can be listened to with privileges.

I added GetPrincipal/SetPrincipal methods to nsIEventListenerManager.
nsDocument::GetNewListenerManager sets a principal on the listener
manager when it creates one. Obviously there are other places that
create listener managers, but scripts seem to go through this one.

Another change is to save some memory usage. Currently I allocate an
array of PolicyType that is NS_DOM_PROP_MAX elements long.
Unfortunately, compilers appear to allocate four bytes for each
PolicyType, so the array takes around 2400 bytes. I've added changes
to use two bit vectors that should consume about 1/16 that space.

r=joki

There are also changes that push nsnull onto the JSContext stack when
entering a nested event loop.

r=jband


git-svn-id: svn://10.0.0.236/trunk@54477 18797224-902f-48f8-a5cc-f745e15eee43
1999-11-25 05:28:18 +00:00

1876 lines
53 KiB
C++
Raw Blame History

/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-
*
* The contents of this file are subject to the Netscape 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/NPL/
*
* 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 Netscape
* Communications Corporation. Portions created by Netscape are
* Copyright (C) 1998 Netscape Communications Corporation. All
* Rights Reserved.
*
* Contributor(s):
*/
#ifndef XP_MAC
#include <sys/types.h>
#include <sys/stat.h>
#else
#include <stat.h>
#include <stddef.h>
#endif
#if !defined XP_UNIX && !defined XP_MAC && ! defined XP_BEOS
#include <direct.h>
#endif
#include <fstream.h>
#include <ctype.h>
#include "nsIPtr.h"
#include "plhash.h"
#include "JSStubGen.h"
#include "Exceptions.h"
#include "IdlSpecification.h"
#include "IdlInterface.h"
#include "IdlVariable.h"
#include "IdlAttribute.h"
#include "IdlFunction.h"
#include "IdlParameter.h"
static const char kFilePrefix[] = "nsJS";
static const char kFileSuffix[] = "cpp";
JSStubGen::JSStubGen()
{
}
JSStubGen::~JSStubGen()
{
}
void
JSStubGen::Generate(char *aFileName,
char *aOutputDirName,
IdlSpecification &aSpec,
int aIsGlobal)
{
IdlInterface *iface = aSpec.GetInterfaceAt(0);
if (!OpenFile(iface->GetName(), aOutputDirName, kFilePrefix, kFileSuffix)) {
throw new CantOpenFileException(aFileName);
}
mIsGlobal = aIsGlobal;
GenerateNPL();
GenerateIncludes(aSpec);
GenerateIIDDefinitions(aSpec);
#ifndef USE_COMPTR
GenerateDefPtrs(aSpec);
#endif
GeneratePropertySlots(aSpec);
GeneratePropertyFunc(aSpec, PR_TRUE);
GeneratePropertyFunc(aSpec, PR_FALSE);
GenerateFinalize(aSpec);
GenerateEnumerate(aSpec);
GenerateResolve(aSpec);
GenerateMethods(aSpec);
GenerateJSClass(aSpec);
GenerateClassProperties(aSpec);
GenerateClassFunctions(aSpec);
GenerateConstructor(aSpec);
GenerateInitClass(aSpec);
GenerateNew(aSpec);
CloseFile();
}
static const char kIncludeDefaultsStr[] = "\n"
"#include \"jsapi.h\"\n"
"#include \"nsJSUtils.h\"\n"
"#include \"nsDOMError.h\"\n"
"#include \"nscore.h\"\n"
"#include \"nsIScriptContext.h\"\n"
"#include \"nsIScriptSecurityManager.h\"\n"
"#include \"nsIJSScriptObject.h\"\n"
"#include \"nsIScriptObjectOwner.h\"\n"
"#include \"nsIScriptGlobalObject.h\"\n"
"#include \"nsCOMPtr.h\"\n"
"#include \"nsDOMPropEnums.h\"\n"
#ifndef USE_COMPTR
"#include \"nsIPtr.h\"\n"
#endif
"#include \"nsString.h\"\n";
static const char kIncludeStr[] = "#include \"nsIDOM%s.h\"\n";
static const char kXPIDLIncludeStr[] = "#include \"%s.h\"\n";
static const char kIncludeConstructorStr[] =
"#include \"nsIScriptNameSpaceManager.h\"\n"
"#include \"nsIComponentManager.h\"\n"
"#include \"nsIJSNativeInitializer.h\"\n"
"#include \"nsDOMCID.h\"\n";
static PRIntn
IncludeEnumerator(PLHashEntry *he, PRIntn i, void *arg)
{
char buf[512];
switch ((Type)(int)(he->value)) {
case TYPE_OBJECT:
case TYPE_FUNC:
sprintf(buf, kIncludeStr, (char *)he->key);
break;
case TYPE_XPIDL_OBJECT:
sprintf(buf, kXPIDLIncludeStr, (char *)he->key);
break;
default:
// uh oh...
break;
}
ofstream *file = (ofstream *)arg;
*file << buf;
return HT_ENUMERATE_NEXT;
}
void
JSStubGen::GenerateIncludes(IdlSpecification &aSpec)
{
ofstream *file = GetFile();
IdlInterface *primary_iface = aSpec.GetInterfaceAt(0);
*file << kIncludeDefaultsStr;
EnumerateAllObjects(aSpec, (PLHashEnumerator)IncludeEnumerator,
file, PR_FALSE);
if (HasConstructor(*primary_iface, NULL)) {
*file << kIncludeConstructorStr;
}
*file << "\n";
}
static const char kIIDDefaultStr[] = "\n"
"static NS_DEFINE_IID(kIScriptObjectOwnerIID, NS_ISCRIPTOBJECTOWNER_IID);\n"
"static NS_DEFINE_IID(kIJSScriptObjectIID, NS_IJSSCRIPTOBJECT_IID);\n"
"static NS_DEFINE_IID(kIScriptGlobalObjectIID, NS_ISCRIPTGLOBALOBJECT_IID);\n";
static const char kIIDStr[] = "static NS_DEFINE_IID(kI%sIID, %s);\n";
PRIntn
JSStubGen_IIDEnumerator(PLHashEntry *he, PRIntn i, void *arg)
{
char buf[512];
char iid_buf[256];
JSStubGen *me = (JSStubGen *)arg;
ofstream *file = me->GetFile();
switch ((Type)(int)(he->value)) {
case TYPE_OBJECT:
case TYPE_FUNC:
me->GetInterfaceIID(iid_buf, (char *)he->key);
sprintf(buf, kIIDStr, (char *)he->key, iid_buf);
break;
case TYPE_XPIDL_OBJECT:
{
// This sucks. I know.
char* p = (char*) he->key;
if (p[0] == 'n' && p[1] == 's' && p[2] == 'I')
p += 3;
me->GetXPIDLInterfaceIID(iid_buf, p);
sprintf(buf, kIIDStr, p, iid_buf);
break;
}
}
*file << buf;
return HT_ENUMERATE_NEXT;
}
void
JSStubGen::GenerateIIDDefinitions(IdlSpecification &aSpec)
{
ofstream *file = GetFile();
*file << kIIDDefaultStr;
EnumerateAllObjects(aSpec, (PLHashEnumerator)JSStubGen_IIDEnumerator,
this, PR_FALSE);
*file << "\n";
}
static const char kDefPtrStr[] =
"NS_DEF_PTR(nsIDOM%s);\n";
static const char kDefXPIDLPtrStr[] =
"NS_DEF_PTR(%s);\n";
PRIntn
JSStubGen_DefPtrEnumerator(PLHashEntry *he, PRIntn i, void *arg)
{
char buf[512];
JSStubGen *me = (JSStubGen *)arg;
ofstream *file = me->GetFile();
switch ((Type)(int)(he->value)) {
case TYPE_OBJECT:
case TYPE_FUNC:
sprintf(buf, kDefPtrStr, (char *)he->key);
break;
case TYPE_XPIDL_OBJECT:
sprintf(buf, kDefXPIDLPtrStr, (char *)he->key);
break;
}
*file << buf;
return HT_ENUMERATE_NEXT;
}
void
JSStubGen::GenerateDefPtrs(IdlSpecification &aSpec)
{
ofstream *file = GetFile();
EnumerateAllObjects(aSpec, (PLHashEnumerator)JSStubGen_DefPtrEnumerator,
this, PR_FALSE);
*file << "\n";
}
static const char kPropEnumStr[] =
"//\n"
"// %s property ids\n"
"//\n"
"enum %s_slots {\n";
static const char kPropSlotStr[] = " %s_%s = -%d";
void
JSStubGen::GeneratePropertySlots(IdlSpecification &aSpec)
{
char buf[512];
ofstream *file = GetFile();
int any_props = 0;
int prop_counter = 0;
int i, icount = aSpec.InterfaceCount();
for (i = 0; i < icount; i++) {
IdlInterface *iface = aSpec.GetInterfaceAt(i);
char iface_name[128];
strcpy(iface_name, iface->GetName());
StrUpr(iface_name);
int a, acount = iface->AttributeCount();
for (a = 0; a < acount; a++) {
IdlAttribute *attr = iface->GetAttributeAt(a);
char attr_name[128];
if (attr->GetIsNoScript()) {
continue;
}
// If this is the first time...
if (!any_props) {
sprintf(buf, kPropEnumStr, iface->GetName(), iface->GetName());
*file << buf;
}
// If this is the first time for this interface...
else if (a == 0) {
*file << ",\n";
}
any_props = 1;
strcpy(attr_name, attr->GetName());
StrUpr(attr_name);
sprintf(buf, kPropSlotStr, iface_name, attr_name, ++prop_counter);
*file << buf;
if (a != acount-1) {
*file << ",\n";
}
}
}
if (any_props) {
*file << "\n};\n";
}
}
static const char kPropFuncBeginStr[] = "\n"
"/***********************************************************************/\n"
"//\n"
"// %s Properties %ster\n"
"//\n"
"PR_STATIC_CALLBACK(JSBool)\n"
"%s%sProperty(JSContext *cx, JSObject *obj, jsval id, jsval *vp)\n"
"{\n"
" nsIDOM%s *a = (nsIDOM%s*)nsJSUtils::nsGetNativeThis(cx, obj);\n"
"\n"
" // If there's no private data, this must be the prototype, so ignore\n"
" if (nsnull == a) {\n"
" return JS_TRUE;\n"
" }\n"
"\n";
static const char kIntCaseStr[] =
" if (JSVAL_IS_INT(id)) {\n"
" nsIScriptContext *scriptCX = (nsIScriptContext *)JS_GetContextPrivate(cx);\n"
" nsCOMPtr<nsIScriptSecurityManager> secMan;\n"
" if (NS_OK != scriptCX->GetSecurityManager(getter_AddRefs(secMan))) {\n"
" return nsJSUtils::nsReportError(cx, NS_ERROR_DOM_SECMAN_ERR);\n"
" }\n"
" switch(JSVAL_TO_INT(id)) {\n";
static const char kIntCaseNamedItemStr[] =
" PRBool checkNamedItem = PR_TRUE;\n"
" if (JSVAL_IS_INT(id)) {\n"
" nsIScriptContext *scriptCX = (nsIScriptContext *)JS_GetContextPrivate(cx);\n"
" nsCOMPtr<nsIScriptSecurityManager> secMan;\n"
" if (NS_OK != scriptCX->GetSecurityManager(getter_AddRefs(secMan))) {\n"
" return nsJSUtils::nsReportError(cx, NS_ERROR_DOM_SECMAN_ERR);\n"
" }\n"
" checkNamedItem = PR_FALSE;\n"
" switch(JSVAL_TO_INT(id)) {\n";
static const char kPropFuncDefaultStr[] =
" default:\n"
" return nsJSUtils::nsCallJSScriptObject%sProperty(a, cx, id, vp);\n"
" }\n"
" }\n";
static const char kPropFuncDefaultNamedItemStr[] =
" default:\n"
" checkNamedItem = PR_TRUE;\n"
" }\n"
" }\n";
static const char kPropFuncDefaultItemStr[] =
" default:\n"
" {\n"
" nsresult result = NS_OK;\n"
" %s prop;\n"
" result = a->Item(JSVAL_TO_INT(id), %sprop);\n"
" if (NS_SUCCEEDED(result)) {\n"
"%s"
" }\n"
" else {\n"
" return nsJSUtils::nsReportError(cx, result);\n"
" }\n"
" }\n"
" }\n"
" }\n";
static const char kPropFuncDefaultItemNonPrimaryStr[] =
" default:\n"
" {\n"
" %s prop;\n"
" nsIDOM%s* b;\n"
" if (NS_OK == a->QueryInterface(kI%sIID, (void **)&b)) {\n"
" nsresult result = NS_OK;\n"
" result = b->Item(JSVAL_TO_INT(id), %sprop);\n"
" if (NS_SUCCEEDED(result)) {\n"
"%s"
" NS_RELEASE(b);\n"
" }\n"
" else {\n"
" NS_RELEASE(b);\n"
" return nsJSUtils::nsReportError(cx, result);\n"
" }\n"
" }\n"
" else {\n"
" return nsJSUtils::nsReportError(cx, NS_ERROR_DOM_WRONG_TYPE_ERR);\n"
" }\n"
" }\n"
" }\n"
" }\n";
static const char kPropFuncEndStr[] =
" else {\n"
" return nsJSUtils::nsCallJSScriptObject%sProperty(a, cx, id, vp);\n"
" }\n"
"\n"
" return PR_TRUE;\n"
"}\n";
static const char kPropFuncNamedItemStr[] =
"\n"
" if (checkNamedItem) {\n"
" %s prop;\n"
" nsAutoString name;\n"
" nsresult result = NS_OK;\n"
"\n"
" JSString *jsstring = JS_ValueToString(cx, id);\n"
" if (nsnull != jsstring) {\n"
" name.SetString(JS_GetStringChars(jsstring));\n"
" }\n"
" else {\n"
" name.SetString(\"\");\n"
" }\n"
"\n"
" result = a->NamedItem(name, %sprop);\n"
" if (NS_SUCCEEDED(result)) {\n"
" if (NULL != prop) {\n"
"%s"
" }\n"
" else {\n"
" return nsJSUtils::nsCallJSScriptObject%sProperty(a, cx, id, vp);\n"
" }\n"
" }\n"
" else {\n"
" return nsJSUtils::nsReportError(cx, result);\n"
" }\n"
" }\n";
static const char kPropFuncNamedItemNonPrimaryStr[] =
"\n"
" if (checkNamedItem) {\n"
" %s prop;\n"
" nsIDOM%s* b;\n"
" nsAutoString name;\n"
"\n"
" JSString *jsstring = JS_ValueToString(cx, id);\n"
" if (nsnull != jsstring) {\n"
" name.SetString(JS_GetStringChars(jsstring));\n"
" }\n"
" else {\n"
" name.SetString(\"\");\n"
" }\n"
"\n"
" if (NS_OK == a->QueryInterface(kI%sIID, (void **)&b)) {\n"
" nsresult result = NS_OK;\n"
" result = b->NamedItem(name, %sprop);\n"
" if (NS_SUCCEEDED(result)) {\n"
" NS_RELEASE(b);\n"
" if (NULL != prop) {\n"
"%s"
" }\n"
" else {\n"
" return nsJSUtils::nsCallJSScriptObject%sProperty(a, cx, id, vp);\n"
" }\n"
" }\n"
" else {\n"
" NS_RELEASE(b);\n"
" return nsJSUtils::nsReportError(cx, result);\n"
" }\n"
" }\n"
" else {\n"
" return nsJSUtils::nsReportError(cx, NS_ERROR_DOM_WRONG_TYPE_ERR);\n"
" }\n"
" }\n";
#define JSGEN_GENERATE_PROPFUNCBEGIN(buffer, op, className) \
sprintf(buffer, kPropFuncBeginStr, className, op, op, className, className, className)
#define JSGEN_GENERATE_PROPFUNCEND(buffer, op) \
sprintf(buffer, kPropFuncEndStr, op)
#define JSGEN_GENERATE_PROPFUNCDEFAULT(buffer, op) \
sprintf(buffer, kPropFuncDefaultStr, op)
static const char kPropCaseBeginStr[] =
" case %s_%s:\n"
" {\n"
" PRBool ok = PR_FALSE;\n"
" secMan->CheckScriptAccess(scriptCX, obj, NS_DOM_PROP_%s_%s, %s, &ok);\n"
" if (!ok) {\n"
" return nsJSUtils::nsReportError(cx, NS_ERROR_DOM_SECURITY_ERR);\n"
" }\n";
static const char kPropCaseEndStr[] =
" break;\n"
" }\n";
static const char kNoAttrStr[] = " case 0:\n";
void
JSStubGen::GeneratePropertyFunc(IdlSpecification &aSpec, PRBool aIsGetter)
{
char buf[1024];
ofstream *file = GetFile();
IdlInterface *primary_iface = aSpec.GetInterfaceAt(0);
PRBool any = PR_FALSE;
JSGEN_GENERATE_PROPFUNCBEGIN(buf, aIsGetter ? "Get" : "Set", primary_iface->GetName());
*file << buf;
IdlFunction *item_func = NULL;
IdlFunction *named_item_func = NULL;
IdlInterface *item_iface = NULL;
IdlInterface *named_item_iface = NULL;
int i, icount = aSpec.InterfaceCount();
for (i = 0; i < icount; i++) {
IdlInterface *iface = aSpec.GetInterfaceAt(i);
int m, mcount = iface->FunctionCount();
for (m = 0; m < mcount; m++) {
IdlFunction *func = iface->GetFunctionAt(m);
if (strcmp(func->GetName(), "item") == 0) {
item_func = func;
item_iface = iface;
}
else if (strcmp(func->GetName(), "namedItem") == 0) {
named_item_func = func;
named_item_iface = iface;
}
}
}
if (NULL == named_item_func) {
*file << kIntCaseStr;
}
else {
*file << kIntCaseNamedItemStr;
}
for (i = 0; i < icount; i++) {
IdlInterface *iface = aSpec.GetInterfaceAt(i);
char iface_name[128];
strcpy(iface_name, iface->GetName());
StrUpr(iface_name);
int a, acount = iface->AttributeCount();
for (a = 0; a < acount; a++) {
IdlAttribute *attr = iface->GetAttributeAt(a);
char attr_name[128];
strcpy(attr_name, attr->GetName());
StrUpr(attr_name);
if (attr->GetIsNoScript() || (!aIsGetter && (attr->GetReadOnly()))) {
continue;
}
any = PR_TRUE;
char upr_attr_name[128];
strcpy(upr_attr_name, attr_name);
StrUpr(upr_attr_name);
char upr_iface_name[128];
strcpy(upr_iface_name, iface_name);
StrUpr(upr_iface_name);
if (aIsGetter) {
sprintf(buf, kPropCaseBeginStr, iface_name, attr_name, upr_iface_name, upr_attr_name, "PR_FALSE");
}
else {
sprintf(buf, kPropCaseBeginStr, iface_name, attr_name, upr_iface_name, upr_attr_name, "PR_TRUE");
}
*file << buf;
if (aIsGetter) {
GeneratePropGetter(file, *iface, *attr,
iface == primary_iface ? JSSTUBGEN_PRIMARY : JSSTUBGEN_NONPRIMARY);
}
else {
GeneratePropSetter(file, *iface, *attr, iface == primary_iface);
}
sprintf(buf, kPropCaseEndStr);
*file << buf;
}
}
if (!any) {
*file << kNoAttrStr;
}
if (aIsGetter) {
if (NULL != item_func) {
IdlVariable *rval = item_func->GetReturnValue();
GeneratePropGetter(file, *item_iface, *rval,
item_iface == primary_iface ?
JSSTUBGEN_DEFAULT : JSSTUBGEN_DEFAULT_NONPRIMARY);
}
else if (NULL != named_item_func) {
*file << kPropFuncDefaultNamedItemStr;
}
else {
JSGEN_GENERATE_PROPFUNCDEFAULT(buf, aIsGetter ? "Get" : "Set");
*file << buf;
}
}
else {
JSGEN_GENERATE_PROPFUNCDEFAULT(buf, aIsGetter ? "Get" : "Set");
*file << buf;
}
if (aIsGetter && (NULL != named_item_func)) {
IdlVariable *rval = named_item_func->GetReturnValue();
GeneratePropGetter(file, *named_item_iface, *rval,
named_item_iface == primary_iface ?
JSSTUBGEN_NAMED_ITEM : JSSTUBGEN_NAMED_ITEM_NONPRIMARY);
}
if (aIsGetter) {
JSGEN_GENERATE_PROPFUNCEND(buf, "Get");
}
else {
JSGEN_GENERATE_PROPFUNCEND(buf, "Set");
}
*file << buf;
}
static const char kGetCaseStr[] =
" %s prop;\n"
" nsresult result = NS_OK;\n"
" result = a->Get%s(%sprop);\n"
" if (NS_SUCCEEDED(result)) {\n"
"%s"
" }\n"
" else {\n"
" return nsJSUtils::nsReportError(cx, result);\n"
" }\n";
static const char kGetCaseNonPrimaryStr[] =
" %s prop;\n"
" nsIDOM%s* b;\n"
" if (NS_OK == a->QueryInterface(kI%sIID, (void **)&b)) {\n"
" nsresult result = NS_OK;\n"
" result = b->Get%s(%sprop);\n"
" if(NS_SUCCEEDED(result)) {\n"
"%s"
" NS_RELEASE(b);\n"
" }\n"
" else {\n"
" NS_RELEASE(b);\n"
" return nsJSUtils::nsReportError(cx, result);\n"
" }\n"
" }\n"
" else {\n"
" return nsJSUtils::nsReportError(cx, NS_ERROR_DOM_WRONG_TYPE_ERR);\n"
" }\n";
static const char kObjectGetCaseStr[] =
" // get the js object\n"
" nsJSUtils::nsConvertObjectToJSVal((nsISupports *)prop, cx, vp);\n";
static const char kXPIDLObjectGetCaseStr[] =
" // get the js object; n.b., this will do a release on 'prop'\n"
" nsJSUtils::nsConvertXPCObjectToJSVal(prop, %s::GetIID(), cx, vp);\n";
static const char kStringGetCaseStr[] =
" nsJSUtils::nsConvertStringToJSVal(prop, cx, vp);\n";
static const char kIntGetCaseStr[] =
" *vp = INT_TO_JSVAL(prop);\n";
static const char kBoolGetCaseStr[] =
" *vp = BOOLEAN_TO_JSVAL(prop);\n";
static const char kJSValGetCaseStr[] =
" *vp = prop;\n";
void
JSStubGen::GeneratePropGetter(ofstream *file,
IdlInterface &aInterface,
IdlVariable &aAttribute,
PRInt32 aType)
{
char buf[2048];
char buf2[1024];
char attr_type[128];
char attr_name[128];
const char *case_str;
GetVariableTypeForLocal(attr_type, aAttribute);
if ((JSSTUBGEN_PRIMARY == aType) || (JSSTUBGEN_NONPRIMARY == aType)) {
GetCapitalizedName(attr_name, aAttribute);
}
switch (aAttribute.GetType()) {
case TYPE_BOOLEAN:
case_str = kBoolGetCaseStr;
break;
case TYPE_LONG:
case TYPE_SHORT:
case TYPE_ULONG:
case TYPE_USHORT:
case TYPE_CHAR:
case TYPE_INT:
case TYPE_UINT:
case_str = kIntGetCaseStr;
break;
case TYPE_JSVAL:
case_str = kJSValGetCaseStr;
break;
case TYPE_STRING:
case_str = kStringGetCaseStr;
break;
case TYPE_OBJECT:
case_str = kObjectGetCaseStr;
break;
case TYPE_XPIDL_OBJECT:
case_str = buf2;
sprintf(buf2, kXPIDLObjectGetCaseStr, aAttribute.GetTypeName(), aAttribute.GetTypeName());
break;
default:
// XXX Fail for other cases
break;
}
if (JSSTUBGEN_PRIMARY == aType) {
sprintf(buf, kGetCaseStr, attr_type, attr_name,
aAttribute.GetType() == TYPE_STRING ? "" : "&",
case_str);
}
else if (JSSTUBGEN_NONPRIMARY == aType) {
sprintf(buf, kGetCaseNonPrimaryStr, attr_type,
aInterface.GetName(), aInterface.GetName(),
attr_name,
aAttribute.GetType() == TYPE_STRING ? "" : "&",
case_str);
}
else if (JSSTUBGEN_DEFAULT == aType) {
sprintf(buf, kPropFuncDefaultItemStr, attr_type,
aAttribute.GetType() == TYPE_STRING ? "" : "&",
case_str);
}
else if (JSSTUBGEN_DEFAULT_NONPRIMARY == aType) {
sprintf(buf, kPropFuncDefaultItemNonPrimaryStr, attr_type,
aInterface.GetName(), aInterface.GetName(),
aAttribute.GetType() == TYPE_STRING ? "" : "&",
case_str);
}
else if (JSSTUBGEN_NAMED_ITEM == aType) {
sprintf(buf, kPropFuncNamedItemStr, attr_type,
aAttribute.GetType() == TYPE_STRING ? "" : "&",
case_str, "Get");
}
else if (JSSTUBGEN_NAMED_ITEM_NONPRIMARY == aType) {
sprintf(buf, kPropFuncNamedItemNonPrimaryStr, attr_type,
aInterface.GetName(), aInterface.GetName(),
aAttribute.GetType() == TYPE_STRING ? "" : "&",
case_str, "Get");
}
*file << buf;
}
static const char kSetCaseStr[] =
" %s prop;\n"
"%s \n"
" a->Set%s(prop);\n"
" %s\n";
static const char kSetCaseNonPrimaryStr[] =
" %s prop;\n"
"%s \n"
" nsIDOM%s *b;\n"
" if (NS_OK == a->QueryInterface(kI%sIID, (void **)&b)) {\n"
" b->Set%s(prop);\n"
" NS_RELEASE(b);\n"
" }\n"
" else {\n"
" %s\n"
" return nsJSUtils::nsReportError(cx, NS_ERROR_DOM_WRONG_TYPE_ERR);\n"
" }\n"
" %s\n";
static const char kObjectSetCaseStr[] =
" if (PR_FALSE == nsJSUtils::nsConvertJSValToObject((nsISupports **)&prop,\n"
" kI%sIID, \"%s\",\n"
" cx, *vp)) {\n"
" return nsJSUtils::nsReportError(cx, NS_ERROR_DOM_NOT_OBJECT_ERR);\n"
" }\n";
static const char kXPIDLObjectSetCaseStr[] =
" if (PR_FALSE == nsJSUtils::nsConvertJSValToXPCObject((nsISupports **) &prop,\n"
" kI%sIID, cx, *vp)) {\n"
" return nsJSUtils::nsReportError(cx, NS_ERROR_DOM_NOT_XPC_OBJECT_ERR);\n"
" }\n";
static const char kObjectSetCaseEndStr[] = "NS_IF_RELEASE(prop);";
static const char* kXPIDLObjectSetCaseEndStr = kObjectSetCaseEndStr;
static const char kStringSetCaseStr[] =
" nsJSUtils::nsConvertJSValToString(prop, cx, *vp);\n";
static const char kIntSetCaseStr[] =
" int32 temp;\n"
" if (JSVAL_IS_NUMBER(*vp) && JS_ValueToInt32(cx, *vp, &temp)) {\n"
" prop = (%s)temp;\n"
" }\n"
" else {\n"
" return nsJSUtils::nsReportError(cx, NS_ERROR_DOM_NOT_NUMBER_ERR);\n"
" }\n";
static const char kBoolSetCaseStr[] =
" if (PR_FALSE == nsJSUtils::nsConvertJSValToBool(&prop, cx, *vp)) {\n"
" return nsJSUtils::nsReportError(cx, NS_ERROR_DOM_NOT_BOOLEAN_ERR);\n"
" }\n";
static const char kJSValSetCaseStr[] =
" prop = *vp;\n";
void
JSStubGen::GeneratePropSetter(ofstream *file,
IdlInterface &aInterface,
IdlAttribute &aAttribute,
PRBool aIsPrimary)
{
char buf[1024];
char attr_type[128];
char attr_name[128];
char case_buf[1024];
const char *end_str;
GetVariableTypeForLocal(attr_type, aAttribute);
GetCapitalizedName(attr_name, aAttribute);
switch (aAttribute.GetType()) {
case TYPE_BOOLEAN:
sprintf(case_buf, kBoolSetCaseStr);
break;
case TYPE_LONG:
case TYPE_SHORT:
case TYPE_ULONG:
case TYPE_USHORT:
case TYPE_CHAR:
case TYPE_INT:
case TYPE_UINT:
sprintf(case_buf, kIntSetCaseStr, attr_type);
break;
case TYPE_JSVAL:
strcpy(case_buf, kJSValSetCaseStr);
break;
case TYPE_STRING:
strcpy(case_buf, kStringSetCaseStr);
break;
case TYPE_OBJECT:
sprintf(case_buf, kObjectSetCaseStr, aAttribute.GetTypeName(), aAttribute.GetTypeName());
break;
case TYPE_XPIDL_OBJECT:
// Yeah, this sucks. That's life.
{
char* p = aAttribute.GetTypeName();
if (p[0] == 'n' && p[1] == 's' && p[2] == 'I')
p += 3;
sprintf(case_buf, kXPIDLObjectSetCaseStr, p);
}
break;
default:
// XXX Fail for other cases
break;
}
switch (aAttribute.GetType()) {
case TYPE_OBJECT: end_str = kObjectSetCaseEndStr; break;
case TYPE_XPIDL_OBJECT: end_str = kXPIDLObjectSetCaseEndStr; break;
default: end_str = ""; break;
}
if (aIsPrimary) {
sprintf(buf, kSetCaseStr, attr_type, case_buf, attr_name, end_str);
}
else {
sprintf(buf, kSetCaseNonPrimaryStr, attr_type, case_buf,
aInterface.GetName(), aInterface.GetName(), attr_name,
end_str, end_str);
}
*file << buf;
}
static const char kFinalizeStr[] =
"\n\n//\n"
"// %s finalizer\n"
"//\n"
"PR_STATIC_CALLBACK(void)\n"
"Finalize%s(JSContext *cx, JSObject *obj)\n"
"{\n"
" nsJSUtils::nsGenericFinalize(cx, obj);\n"
"}\n";
void
JSStubGen::GenerateFinalize(IdlSpecification &aSpec)
{
char buf[512];
ofstream *file = GetFile();
IdlInterface *iface = aSpec.GetInterfaceAt(0);
sprintf(buf, kFinalizeStr, iface->GetName(), iface->GetName());
*file << buf;
}
static const char kEnumerateStr[] =
"\n\n//\n"
"// %s enumerate\n"
"//\n"
"PR_STATIC_CALLBACK(JSBool)\n"
"Enumerate%s(JSContext *cx, JSObject *obj)\n"
"{\n"
" return nsJSUtils::nsGenericEnumerate(cx, obj);\n"
"}\n";
#define JSGEN_GENERATE_ENUMERATE(buf, className) \
sprintf(buf, kEnumerateStr, className, className);
void
JSStubGen::GenerateEnumerate(IdlSpecification &aSpec)
{
char buf[512];
ofstream *file = GetFile();
IdlInterface *iface = aSpec.GetInterfaceAt(0);
char *name = iface->GetName();
JSGEN_GENERATE_ENUMERATE(buf, name);
*file << buf;
}
static const char kResolveStr[] =
"\n\n//\n"
"// %s resolve\n"
"//\n"
"PR_STATIC_CALLBACK(JSBool)\n"
"Resolve%s(JSContext *cx, JSObject *obj, jsval id)\n"
"{\n"
"%s"
"}\n";
static const char* kGenericResolveStr =
" return nsJSUtils::nsGenericResolve(cx, obj, id);\n";
static const char* kGlobalResolveStr =
" return nsJSUtils::nsGlobalResolve(cx, obj, id);\n";
#define JSGEN_GENERATE_RESOLVE(buf, className, str) \
sprintf(buf, kResolveStr, className, className, str);
void
JSStubGen::GenerateResolve(IdlSpecification &aSpec)
{
char buf[512];
ofstream *file = GetFile();
IdlInterface *iface = aSpec.GetInterfaceAt(0);
char *name = iface->GetName();
if (mIsGlobal) {
JSGEN_GENERATE_RESOLVE(buf, name, kGlobalResolveStr);
}
else {
JSGEN_GENERATE_RESOLVE(buf, name, kGenericResolveStr);
}
*file << buf;
}
static const char kMethodBeginStr[] = "\n\n"
"//\n"
"// Native method %s\n"
"//\n"
"PR_STATIC_CALLBACK(JSBool)\n"
"%s%s(JSContext *cx, JSObject *obj, uintN argc, jsval *argv, jsval *rval)\n"
"{\n"
" nsIDOM%s *nativeThis = (nsIDOM%s*)nsJSUtils::nsGetNativeThis(cx, obj);\n"
" nsresult result = NS_OK;\n";
static const char kMethodBeginNonPrimaryStr[] = "\n\n"
"//\n"
"// Native method %s\n"
"//\n"
"PR_STATIC_CALLBACK(JSBool)\n"
"%s%s(JSContext *cx, JSObject *obj, uintN argc, jsval *argv, jsval *rval)\n"
"{\n"
" nsIDOM%s *privateThis = (nsIDOM%s*)nsJSUtils::nsGetNativeThis(cx, obj);\n"
" nsIDOM%s *nativeThis = nsnull;\n"
" nsresult result = NS_OK;\n"
" if (NS_OK != privateThis->QueryInterface(kI%sIID, (void **)&nativeThis)) {\n"
" return nsJSUtils::nsReportError(cx, NS_ERROR_DOM_WRONG_TYPE_ERR);\n"
" }\n"
"\n";
static const char kMethodReturnStr[] =
" %s nativeRet;\n";
static const char kMethodParamStr[] = " %s b%d;\n";
static const char kMethodBodyBeginStr[] = "\n"
" *rval = JSVAL_NULL;\n"
"\n"
" nsIScriptContext *scriptCX = (nsIScriptContext *)JS_GetContextPrivate(cx);\n"
" nsCOMPtr<nsIScriptSecurityManager> secMan;\n"
" if (NS_OK != scriptCX->GetSecurityManager(getter_AddRefs(secMan))) {\n"
" return nsJSUtils::nsReportError(cx, NS_ERROR_DOM_SECMAN_ERR);\n"
" }\n"
" {\n"
" PRBool ok;\n"
" secMan->CheckScriptAccess(scriptCX, obj, NS_DOM_PROP_%s_%s, PR_FALSE, &ok);\n"
" if (!ok) {\n"
" return nsJSUtils::nsReportError(cx, NS_ERROR_DOM_SECURITY_ERR);\n"
" }\n"
" }\n"
"\n"
" // If there's no private data, this must be the prototype, so ignore\n"
" if (nsnull == nativeThis) {\n"
" return JS_TRUE;\n"
" }\n"
"\n"
" {\n";
static const char kMethodObjectParamStr[] =
#ifdef USE_COMPTR
" if (JS_FALSE == nsJSUtils::nsConvertJSValToObject((nsISupports **)(void**)getter_AddRefs(b%d),\n"
#else
" if (JS_FALSE == nsJSUtils::nsConvertJSValToObject((nsISupports **)&b%d,\n"
#endif
" kI%sIID,\n"
" \"%s\",\n"
" cx,\n"
" argv[%d])) {\n"
" return nsJSUtils::nsReportError(cx, NS_ERROR_DOM_NOT_OBJECT_ERR);\n"
" }\n";
#define JSGEN_GENERATE_OBJECTPARAM(buffer, paramNum, paramType) \
sprintf(buffer, kMethodObjectParamStr, paramNum, paramType, \
paramType, paramNum)
static const char kMethodXPIDLObjectParamStr[] =
#ifdef USE_COMPTR
" if (JS_FALSE == nsJSUtils::nsConvertJSValToXPCObject(getter_AddRef(b%d),\n"
#else
" if (JS_FALSE == nsJSUtils::nsConvertJSValToXPCObject((nsISupports**) &b%d,\n"
#endif
" kI%sIID, cx, argv[%d])) {\n"
" return nsJSUtils::nsReportError(cx, NS_ERROR_DOM_NOT_XPC_OBJECT_ERR);\n"
" }\n";
#define JSGEN_GENERATE_XPIDL_OBJECTPARAM(buffer, paramNum, paramType) \
sprintf(buffer, kMethodXPIDLObjectParamStr, paramNum, \
(((paramType)[0] == 'n' && (paramType)[1] == 's' && (paramType)[2] == 'I') \
? ((paramType) + 3) : (paramType)), \
paramNum)
static const char kMethodStringParamStr[] =
" nsJSUtils::nsConvertJSValToString(b%d, cx, argv[%d]);\n";
#define JSGEN_GENERATE_STRINGPARAM(buffer, paramNum) \
sprintf(buffer, kMethodStringParamStr, paramNum, paramNum)
static const char kMethodBoolParamStr[] =
" if (!nsJSUtils::nsConvertJSValToBool(&b%d, cx, argv[%d])) {\n"
" return nsJSUtils::nsReportError(cx, NS_ERROR_DOM_NOT_BOOLEAN_ERR);\n"
" }\n";
#define JSGEN_GENERATE_BOOLPARAM(buffer, paramNum) \
sprintf(buffer, kMethodBoolParamStr, paramNum, paramNum)
static const char kMethodIntParamStr[] =
" if (!JS_ValueToInt32(cx, argv[%d], (int32 *)&b%d)) {\n"
" return nsJSUtils::nsReportError(cx, NS_ERROR_DOM_NOT_NUMBER_ERR);\n"
" }\n";
#define JSGEN_GENERATE_INTPARAM(buffer, paramNum) \
sprintf(buffer, kMethodIntParamStr, paramNum, paramNum)
static const char kMethodFuncParamStr[] =
#ifdef USE_COMPTR
" if (!nsJSUtils::nsConvertJSValToFunc(getter_AddRefs(b%d),\n"
#else
" if (!nsJSUtils::nsConvertJSValToFunc((nsIDOMEventListener**)(nsISupports**) &b%d,\n"
#endif
" cx,\n"
" obj,\n"
" argv[%d])) {\n"
" return nsJSUtils::nsReportError(cx, NS_ERROR_DOM_NOT_FUNCTION_ERR);\n"
" }\n";
#define JSGEN_GENERATE_FUNCPARAM(buffer, paramNum, paramType) \
sprintf(buffer, kMethodFuncParamStr, paramNum, paramNum)
static const char kMethodJSValParamStr[] =
" b%d = argv[%d];\n";
#define JSGEN_GENERATE_JSVALPARAM(buffer, paramNum) \
sprintf(buffer, kMethodJSValParamStr, paramNum, paramNum)
static const char kMethodParamListStr[] = "b%d";
static const char kMethodParamListDelimiterStr[] = ", ";
static const char kMethodParamEllipsisStr[] = "cx, argv+%d, argc-%d";
static const char kMethodBodyMiddleStr[] =
"\n"
" result = nativeThis->%s(%s%snativeRet);\n"
" if (NS_FAILED(result)) {\n"
" return nsJSUtils::nsReportError(cx, result);\n"
" }\n"
"\n";
static const char kMethodBodyMiddleNoReturnStr[] =
"\n"
" result = nativeThis->%s(%s);\n"
" if (NS_FAILED(result)) {\n"
" return nsJSUtils::nsReportError(cx, result);\n"
" }\n"
"\n";
static const char kMethodObjectRetStr[] =
" nsJSUtils::nsConvertObjectToJSVal(nativeRet, cx, rval);\n";
static const char kMethodXPIDLObjectRetStr[] =
" // n.b., this will release nativeRet\n"
" nsJSUtils::nsConvertXPCObjectToJSVal(nativeRet, %s::GetIID(), cx, rval);\n";
static const char kMethodStringRetStr[] =
" nsJSUtils::nsConvertStringToJSVal(nativeRet, cx, rval);\n";
static const char kMethodIntRetStr[] =
" *rval = INT_TO_JSVAL(nativeRet);\n";
static const char kMethodBoolRetStr[] =
" *rval = BOOLEAN_TO_JSVAL(nativeRet);\n";
static const char kMethodVoidRetStr[] =
" *rval = JSVAL_VOID;\n";
static const char kMethodJSValRetStr[] =
" *rval = nativeRet;\n";
static const char kMethodBadParamStr[] =
" if (argc < %d) {\n"
" return nsJSUtils::nsReportError(cx, NS_ERROR_DOM_TOO_FEW_PARAMETERS_ERR);\n"
" }\n"
"\n";
static const char kMethodEndStr[] =
" }\n"
"\n"
" return JS_TRUE;\n"
"}\n";
void
JSStubGen::GenerateMethods(IdlSpecification &aSpec)
{
char buf[1024];
ofstream *file = GetFile();
IdlInterface *primary_iface = aSpec.GetInterfaceAt(0);
int i, icount = aSpec.InterfaceCount();
for (i = 0; i < icount; i++) {
IdlInterface *iface = aSpec.GetInterfaceAt(i);
char iface_name[128];
GetCapitalizedName(iface_name, *iface);
int m, mcount = iface->FunctionCount();
for (m = 0; m < mcount; m++) {
IdlFunction *func = iface->GetFunctionAt(m);
IdlVariable *rval = func->GetReturnValue();
char method_name[128];
char return_type[128];
int p, pcount = func->ParameterCount();
if (func->GetIsNoScript()) {
continue;
}
GetCapitalizedName(method_name, *func);
// If this is a constructor don't have a method for it
if (strcmp(method_name, iface_name) == 0) {
continue;
}
GetVariableTypeForLocal(return_type, *rval);
if (i == 0) {
sprintf(buf, kMethodBeginStr, method_name, iface->GetName(),
method_name, iface->GetName(), iface->GetName());
}
else {
sprintf(buf, kMethodBeginNonPrimaryStr, method_name, iface->GetName(),
method_name, primary_iface->GetName(), primary_iface->GetName(),
iface->GetName(), iface->GetName(), iface->GetName());
}
*file << buf;
if (rval->GetType() != TYPE_VOID) {
sprintf(buf, kMethodReturnStr, return_type);
*file << buf;
}
for (p = 0; p < pcount; p++) {
IdlParameter *param = func->GetParameterAt(p);
GetVariableTypeForMethodLocal(return_type, *param);
sprintf(buf, kMethodParamStr, return_type, p);
*file << buf;
}
char upr_method_name[128];
strcpy(upr_method_name, method_name);
StrUpr(upr_method_name);
char upr_iface_name[128];
strcpy(upr_iface_name, iface->GetName());
StrUpr(upr_iface_name);
sprintf(buf, kMethodBodyBeginStr, upr_iface_name,
upr_method_name);
*file << buf;
if (pcount > 0) {
sprintf(buf, kMethodBadParamStr, pcount);
*file << buf;
}
for (p = 0; p < pcount; p++) {
IdlParameter *param = func->GetParameterAt(p);
switch(param->GetType()) {
case TYPE_BOOLEAN:
JSGEN_GENERATE_BOOLPARAM(buf, p);
break;
case TYPE_LONG:
case TYPE_SHORT:
case TYPE_ULONG:
case TYPE_USHORT:
case TYPE_CHAR:
case TYPE_INT:
case TYPE_UINT:
JSGEN_GENERATE_INTPARAM(buf, p);
break;
case TYPE_JSVAL:
JSGEN_GENERATE_JSVALPARAM(buf, p);
break;
case TYPE_STRING:
JSGEN_GENERATE_STRINGPARAM(buf, p);
break;
case TYPE_OBJECT:
JSGEN_GENERATE_OBJECTPARAM(buf, p, param->GetTypeName());
break;
case TYPE_XPIDL_OBJECT:
JSGEN_GENERATE_XPIDL_OBJECTPARAM(buf, p, param->GetTypeName());
break;
case TYPE_FUNC:
JSGEN_GENERATE_FUNCPARAM(buf, p, param->GetTypeName());
break;
default:
// XXX Fail for other cases
break;
}
*file << buf;
}
char param_buf[512];
char *param_ptr = param_buf;
param_buf[0] = '\0';
for (p = 0; p < pcount; p++) {
if (p > 0) {
strcpy(param_ptr, kMethodParamListDelimiterStr);
param_ptr += strlen(param_ptr);
}
sprintf(param_ptr, kMethodParamListStr, p);
param_ptr += strlen(param_ptr);
}
if (func->GetHasEllipsis()) {
if (pcount > 0) {
strcpy(param_ptr, kMethodParamListDelimiterStr);
param_ptr += strlen(param_ptr);
}
sprintf(param_ptr, kMethodParamEllipsisStr, pcount, pcount);
param_ptr += strlen(param_ptr);
}
if (rval->GetType() != TYPE_VOID) {
if ((pcount > 0) || func->GetHasEllipsis()) {
strcpy(param_ptr, kMethodParamListDelimiterStr);
}
sprintf(buf, kMethodBodyMiddleStr, method_name, param_buf,
rval->GetType() == TYPE_STRING ? "" : "&");
}
else {
sprintf(buf, kMethodBodyMiddleNoReturnStr, method_name, param_buf);
}
*file << buf;
switch(rval->GetType()) {
case TYPE_BOOLEAN:
*file << kMethodBoolRetStr;
break;
case TYPE_LONG:
case TYPE_SHORT:
case TYPE_ULONG:
case TYPE_USHORT:
case TYPE_CHAR:
case TYPE_INT:
case TYPE_UINT:
*file << kMethodIntRetStr;
break;
case TYPE_JSVAL:
*file << kMethodJSValRetStr;
break;
case TYPE_STRING:
*file << kMethodStringRetStr;
break;
case TYPE_OBJECT:
*file << kMethodObjectRetStr;
break;
case TYPE_XPIDL_OBJECT:
{
char buf[1024];
sprintf(buf, kMethodXPIDLObjectRetStr, rval->GetTypeName());
*file << buf;
}
break;
case TYPE_VOID:
*file << kMethodVoidRetStr;
break;
default:
// XXX Fail for other cases
break;
}
*file << kMethodEndStr;
}
}
}
static const char kJSClassStr[] =
"\n\n/***********************************************************************/\n"
"//\n"
"// class for %s\n"
"//\n"
"JSClass %sClass = {\n"
" \"%s\", \n"
" JSCLASS_HAS_PRIVATE | JSCLASS_PRIVATE_IS_NSISUPPORTS,\n"
" JS_PropertyStub,\n"
" JS_PropertyStub,\n"
" Get%sProperty,\n"
" Set%sProperty,\n"
" Enumerate%s,\n"
" Resolve%s,\n"
" JS_ConvertStub,\n"
" Finalize%s\n"
"};\n";
#define JSGEN_GENERATE_JSCLASS(buf, className) \
sprintf(buf, kJSClassStr, className, className, className, className, \
className, className, className, className);
void
JSStubGen::GenerateJSClass(IdlSpecification &aSpec)
{
char buf[512];
ofstream *file = GetFile();
IdlInterface *iface = aSpec.GetInterfaceAt(0);
char *name = iface->GetName();
JSGEN_GENERATE_JSCLASS(buf, name);
*file << buf;
}
static const char kPropSpecBeginStr[] =
"\n\n//\n"
"// %s class properties\n"
"//\n"
"static JSPropertySpec %sProperties[] =\n"
"{\n";
static const char kPropSpecEntryStr[] =
" {\"%s\", %s_%s, JSPROP_ENUMERATE%s},\n";
static const char kPropSpecReadOnlyStr[] = " | JSPROP_READONLY";
static const char kPropSpecEndStr[] =
" {0}\n"
"};\n";
void
JSStubGen::GenerateClassProperties(IdlSpecification &aSpec)
{
char buf[512];
ofstream *file = GetFile();
IdlInterface *primary_iface = aSpec.GetInterfaceAt(0);
sprintf(buf, kPropSpecBeginStr, primary_iface->GetName(),
primary_iface->GetName());
*file << buf;
int i, icount = aSpec.InterfaceCount();
for (i = 0; i < icount; i++) {
IdlInterface *iface = aSpec.GetInterfaceAt(i);
char iface_name[128];
strcpy(iface_name, iface->GetName());
StrUpr(iface_name);
int a, acount = iface->AttributeCount();
for (a = 0; a < acount; a++) {
IdlAttribute *attr = iface->GetAttributeAt(a);
char attr_name[128];
if (attr->GetIsNoScript()) {
continue;
}
strcpy(attr_name, attr->GetName());
StrUpr(attr_name);
sprintf(buf, kPropSpecEntryStr, attr->GetName(),
iface_name, attr_name,
attr->GetReadOnly() ? kPropSpecReadOnlyStr : "");
*file << buf;
}
}
*file << kPropSpecEndStr;
}
static const char kFuncSpecBeginStr[] =
"\n\n//\n"
"// %s class methods\n"
"//\n"
"static JSFunctionSpec %sMethods[] = \n"
"{\n";
static const char kFuncSpecEntryStr[] =
" {\"%s\", %s%s, %d},\n";
static const char kFuncSpecEndStr[] =
" {0}\n"
"};\n";
void
JSStubGen::GenerateClassFunctions(IdlSpecification &aSpec)
{
char buf[512];
ofstream *file = GetFile();
IdlInterface *primary_iface = aSpec.GetInterfaceAt(0);
sprintf(buf, kFuncSpecBeginStr, primary_iface->GetName(),
primary_iface->GetName());
*file << buf;
int i, icount = aSpec.InterfaceCount();
for (i = 0; i < icount; i++) {
IdlInterface *iface = aSpec.GetInterfaceAt(i);
char iface_name[128];
GetCapitalizedName(iface_name, *iface);
int m, mcount = iface->FunctionCount();
for (m = 0; m < mcount; m++) {
char method_name[128];
IdlFunction *func = iface->GetFunctionAt(m);
if (func->GetIsNoScript()) {
continue;
}
GetCapitalizedName(method_name, *func);
// If this is a constructor don't have a method for it.
if (strcmp(method_name, iface_name) == 0) {
continue;
}
sprintf(buf, kFuncSpecEntryStr, func->GetName(),
iface->GetName(), method_name,
func->ParameterCount());
*file << buf;
}
}
*file << kFuncSpecEndStr;
}
static const char kEmptyConstructorStr[] =
"\n\n//\n"
"// %s constructor\n"
"//\n"
"PR_STATIC_CALLBACK(JSBool)\n"
"%s(JSContext *cx, JSObject *obj, uintN argc, jsval *argv, jsval *rval)\n"
"{\n"
" return JS_FALSE;\n"
"}\n";
#define JSGEN_GENERATE_EMPTYCONSTRUCTOR(buf, className) \
sprintf(buf, kEmptyConstructorStr, className, className);
static const char kConstructorBeginStr[] =
"\n\n//\n"
"// %s constructor\n"
"//\n"
"PR_STATIC_CALLBACK(JSBool)\n"
"%s(JSContext *cx, JSObject *obj, uintN argc, jsval *argv, jsval *rval)\n"
"{\n"
" nsresult result;\n"
" nsIID classID;\n"
" nsIScriptContext* context = (nsIScriptContext*)JS_GetContextPrivate(cx);\n"
" nsIScriptNameSpaceManager* manager;\n"
" nsIDOM%s *nativeThis;\n"
" nsIScriptObjectOwner *owner = nsnull;\n"
" nsIJSNativeInitializer* initializer = nsnull;\n"
"\n"
" static NS_DEFINE_IID(kIDOM%sIID, NS_IDOM%s_IID);\n"
" static NS_DEFINE_IID(kIJSNativeInitializerIID, NS_IJSNATIVEINITIALIZER_IID);\n"
"\n"
" result = context->GetNameSpaceManager(&manager);\n"
" if (NS_OK != result) {\n"
" return JS_FALSE;\n"
" }\n"
"\n"
" result = manager->LookupName(\"%s\", PR_TRUE, classID);\n"
" NS_RELEASE(manager);\n"
" if (NS_OK != result) {\n"
" return JS_FALSE;\n"
" }\n"
"\n"
" result = nsComponentManager::CreateInstance(classID,\n"
" nsnull,\n"
" kIDOM%sIID,\n"
" (void **)&nativeThis);\n"
" if (NS_OK != result) {\n"
" return JS_FALSE;\n"
" }\n"
"\n"
" result = nativeThis->QueryInterface(kIJSNativeInitializerIID, (void **)&initializer);\n"
" if (NS_OK == result) {\n"
" result = initializer->Initialize(cx, argc, argv);\n"
" NS_RELEASE(initializer);\n"
"\n"
" if (NS_OK != result) {\n"
" NS_RELEASE(nativeThis);\n"
" return JS_FALSE;\n"
" }\n"
" }\n"
"\n"
" result = nativeThis->QueryInterface(kIScriptObjectOwnerIID, (void **)&owner);\n"
" if (NS_OK != result) {\n"
" NS_RELEASE(nativeThis);\n"
" return JS_FALSE;\n"
" }\n"
"\n"
" owner->SetScriptObject((void *)obj);\n"
" JS_SetPrivate(cx, obj, nativeThis);\n"
"\n"
" NS_RELEASE(owner);\n"
" return JS_TRUE;\n"
"}";
#define JSGEN_GENERATE_CONSTRUCTOR(buf, className, caps) \
sprintf(buf, kConstructorBeginStr, className, className, className, \
className, caps, className, className);
void
JSStubGen::GenerateConstructor(IdlSpecification &aSpec)
{
char buf[2048];
ofstream *file = GetFile();
IdlInterface *primary_iface = aSpec.GetInterfaceAt(0);
IdlFunction *constructor;
char *name = primary_iface->GetName();
char caps_name[128];
strcpy(caps_name, name);
StrUpr(caps_name);
if (HasConstructor(*primary_iface, &constructor)) {
JSGEN_GENERATE_CONSTRUCTOR(buf, name, caps_name);
*file << buf;
}
else if (!mIsGlobal) {
JSGEN_GENERATE_EMPTYCONSTRUCTOR(buf, name);
*file << buf;
}
}
static const char kGlobalInitClassStr[] =
"\n\n//\n"
"// %s class initialization\n"
"//\n"
"nsresult NS_Init%sClass(nsIScriptContext *aContext, \n"
" nsIScriptGlobalObject *aGlobal)\n"
"{\n"
" JSContext *jscontext = (JSContext *)aContext->GetNativeContext();\n"
" JSObject *global = JS_GetGlobalObject(jscontext);\n"
"\n"
" JS_DefineProperties(jscontext, global, %sProperties);\n"
" JS_DefineFunctions(jscontext, global, %sMethods);\n"
"\n"
" return NS_OK;\n"
"}\n";
#define JSGEN_GENERATE_GLOBALINITCLASS(buffer, className) \
sprintf(buffer, kGlobalInitClassStr, className, className, className, \
className)
static const char kInitClassBeginStr[] =
"\n\n//\n"
"// %s class initialization\n"
"//\n"
"extern \"C\" NS_DOM nsresult NS_Init%sClass(nsIScriptContext *aContext, void **aPrototype)\n"
"{\n"
" JSContext *jscontext = (JSContext *)aContext->GetNativeContext();\n"
" JSObject *proto = nsnull;\n"
" JSObject *constructor = nsnull;\n"
" JSObject *parent_proto = nsnull;\n"
" JSObject *global = JS_GetGlobalObject(jscontext);\n"
" jsval vp;\n"
"\n"
" if ((PR_TRUE != JS_LookupProperty(jscontext, global, \"%s\", &vp)) ||\n"
" !JSVAL_IS_OBJECT(vp) ||\n"
" ((constructor = JSVAL_TO_OBJECT(vp)) == nsnull) ||\n"
" (PR_TRUE != JS_LookupProperty(jscontext, JSVAL_TO_OBJECT(vp), \"prototype\", &vp)) || \n"
" !JSVAL_IS_OBJECT(vp)) {\n\n";
#define JSGEN_GENERATE_INITCLASSBEGIN(buffer, className) \
sprintf(buffer, kInitClassBeginStr, className, className, className)
static const char kGetParentProtoStr[] =
" if (NS_OK != NS_Init%sClass(aContext, (void **)&parent_proto)) {\n"
" return NS_ERROR_FAILURE;\n"
" }\n";
static const char kInitClassBodyStr[] =
" proto = JS_InitClass(jscontext, // context\n"
" global, // global object\n"
" parent_proto, // parent proto \n"
" &%sClass, // JSClass\n"
" %s, // JSNative ctor\n"
" 0, // ctor args\n"
" %sProperties, // proto props\n"
" %sMethods, // proto funcs\n"
" nsnull, // ctor props (static)\n"
" nsnull); // ctor funcs (static)\n"
" if (nsnull == proto) {\n"
" return NS_ERROR_FAILURE;\n"
" }\n"
"\n";
#define JSGEN_GENERATE_INITCLASSBODY(buffer, className) \
sprintf(buffer, kInitClassBodyStr, className, className, \
className, className)
static const char kAliasConstructorStr[] =
" JS_AliasProperty(jscontext, global, \"%s\", \"%s\");\n";
static const char kInitStaticBeginStr[] =
" if ((PR_TRUE == JS_LookupProperty(jscontext, global, \"%s\", &vp)) &&\n"
" JSVAL_IS_OBJECT(vp) &&\n"
" ((constructor = JSVAL_TO_OBJECT(vp)) != nsnull)) {\n";
static const char kInitStaticEntryStr[] =
" vp = INT_TO_JSVAL(nsIDOM%s::%s);\n"
" JS_SetProperty(jscontext, constructor, \"%s\", &vp);\n"
"\n";
static const char kInitStaticEndStr[] =
" }\n"
"\n";
static const char kInitClassEndStr[] =
" }\n"
" else if ((nsnull != constructor) && JSVAL_IS_OBJECT(vp)) {\n"
" proto = JSVAL_TO_OBJECT(vp);\n"
" }\n"
" else {\n"
" return NS_ERROR_FAILURE;\n"
" }\n"
"\n"
" if (aPrototype) {\n"
" *aPrototype = proto;\n"
" }\n"
" return NS_OK;\n"
"}\n";
void
JSStubGen::GenerateInitClass(IdlSpecification &aSpec)
{
char buf[2048];
ofstream *file = GetFile();
IdlInterface *primary_iface = aSpec.GetInterfaceAt(0);
char *primary_class = primary_iface->GetName();
if (mIsGlobal) {
JSGEN_GENERATE_GLOBALINITCLASS(buf, primary_class);
*file << buf;
return;
}
JSGEN_GENERATE_INITCLASSBEGIN(buf, primary_class);
*file << buf;
if (primary_iface->BaseClassCount() > 0) {
char *base_class = primary_iface->GetBaseClassAt(0);
sprintf(buf, kGetParentProtoStr, base_class);
*file << buf;
}
JSGEN_GENERATE_INITCLASSBODY(buf, primary_class);
*file << buf;
int i, icount = aSpec.InterfaceCount();
for (i = 0; i < icount; i++) {
IdlInterface *iface = aSpec.GetInterfaceAt(i);
char iface_name[128];
GetCapitalizedName(iface_name, *iface);
int m, mcount = iface->FunctionCount();
for (m = 0; m < mcount; m++) {
char method_name[128];
IdlFunction *func = iface->GetFunctionAt(m);
GetCapitalizedName(method_name, *func);
// If this is a constructor defined in a non-primary interface
// don't have a method for it...we'll alias it to the constructor
// for the primary interface.
if ((strcmp(method_name, iface_name) == 0) &&
(iface != primary_iface)) {
sprintf(buf, kAliasConstructorStr, primary_class, method_name);
*file << buf;
}
}
}
int c, ccount = primary_iface->ConstCount();
if (ccount > 0) {
sprintf(buf, kInitStaticBeginStr, primary_iface->GetName());
*file << buf;
for (c = 0; c < ccount; c++) {
IdlVariable *var = primary_iface->GetConstAt(c);
if (NULL != var) {
sprintf(buf, kInitStaticEntryStr, primary_iface->GetName(),
var->GetName(), var->GetName());
*file << buf;
}
}
*file << kInitStaticEndStr;
}
*file << kInitClassEndStr;
}
static const char kNewGlobalJSObjectStr[] =
"\n\n//\n"
"// Method for creating a new %s JavaScript object\n"
"//\n"
"extern \"C\" NS_DOM nsresult NS_NewScript%s(nsIScriptContext *aContext, nsISupports *aSupports, nsISupports *aParent, void **aReturn)\n"
"{\n"
" NS_PRECONDITION(nsnull != aContext && nsnull != aSupports && nsnull != aReturn, \"null arg\");\n"
" JSContext *jscontext = (JSContext *)aContext->GetNativeContext();\n"
"\n"
" JSObject *global = ::JS_NewObject(jscontext, &%sClass, NULL, NULL);\n"
" if (global) {\n"
" // The global object has a to be defined in two step:\n"
" // 1- create a generic object, with no prototype and no parent which\n"
" // will be passed to JS_InitStandardClasses. JS_InitStandardClasses \n"
" // will make it the global object\n"
" // 2- define the global object to be what you really want it to be.\n"
" //\n"
" // The js runtime is not fully initialized before JS_InitStandardClasses\n"
" // is called, so part of the global object initialization has to be moved \n"
" // after JS_InitStandardClasses\n"
"\n"
" // assign \"this\" to the js object\n"
" ::JS_SetPrivate(jscontext, global, aSupports);\n"
" NS_ADDREF(aSupports);\n"
"\n"
" JS_DefineProperties(jscontext, global, %sProperties);\n"
" JS_DefineFunctions(jscontext, global, %sMethods);\n"
"\n"
" *aReturn = (void*)global;\n"
" return NS_OK;\n"
" }\n"
"\n"
" return NS_ERROR_FAILURE;\n"
"}\n";
#define JSGEN_GENERATE_NEWGLOBALJSOBJECT(buffer, className) \
sprintf(buffer, kNewGlobalJSObjectStr, className, \
className, className, className, className)
static const char kNewJSObjectStr[] =
"\n\n//\n"
"// Method for creating a new %s JavaScript object\n"
"//\n"
"extern \"C\" NS_DOM nsresult NS_NewScript%s(nsIScriptContext *aContext, nsISupports *aSupports, nsISupports *aParent, void **aReturn)\n"
"{\n"
" NS_PRECONDITION(nsnull != aContext && nsnull != aSupports && nsnull != aReturn, \"null argument to NS_NewScript%s\");\n"
" JSObject *proto;\n"
" JSObject *parent;\n"
" nsIScriptObjectOwner *owner;\n"
" JSContext *jscontext = (JSContext *)aContext->GetNativeContext();\n"
" nsresult result = NS_OK;\n"
" nsIDOM%s *a%s;\n"
"\n"
" if (nsnull == aParent) {\n"
" parent = nsnull;\n"
" }\n"
" else if (NS_OK == aParent->QueryInterface(kIScriptObjectOwnerIID, (void**)&owner)) {\n"
" if (NS_OK != owner->GetScriptObject(aContext, (void **)&parent)) {\n"
" NS_RELEASE(owner);\n"
" return NS_ERROR_FAILURE;\n"
" }\n"
" NS_RELEASE(owner);\n"
" }\n"
" else {\n"
" return NS_ERROR_FAILURE;\n"
" }\n"
"\n"
" if (NS_OK != NS_Init%sClass(aContext, (void **)&proto)) {\n"
" return NS_ERROR_FAILURE;\n"
" }\n"
"\n"
" result = aSupports->QueryInterface(kI%sIID, (void **)&a%s);\n"
" if (NS_OK != result) {\n"
" return result;\n"
" }\n"
"\n"
" // create a js object for this class\n"
" *aReturn = JS_NewObject(jscontext, &%sClass, proto, parent);\n"
" if (nsnull != *aReturn) {\n"
" // connect the native object to the js object\n"
" JS_SetPrivate(jscontext, (JSObject *)*aReturn, a%s);\n"
" }\n"
" else {\n"
" NS_RELEASE(a%s);\n"
" return NS_ERROR_FAILURE; \n"
" }\n"
"\n"
" return NS_OK;\n"
"}\n";
#define JSGEN_GENERATE_NEWJSOBJECT(buffer, className) \
sprintf(buffer, kNewJSObjectStr, className, className, \
className, className, className, className, \
className, className, className, className, className)
void
JSStubGen::GenerateNew(IdlSpecification &aSpec)
{
char buf[2048];
ofstream *file = GetFile();
IdlInterface *primary_iface = aSpec.GetInterfaceAt(0);
char *primary_class = primary_iface->GetName();
if (mIsGlobal) {
JSGEN_GENERATE_NEWGLOBALJSOBJECT(buf, primary_class);
}
else {
JSGEN_GENERATE_NEWJSOBJECT(buf, primary_class);
}
*file << buf;
}
Reference in New Issue View Git Blame Copy Permalink