javaembed.cpp 121 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306
  1. /*##############################################################################
  2. HPCC SYSTEMS software Copyright (C) 2012 HPCC Systems®.
  3. Licensed under the Apache License, Version 2.0 (the "License");
  4. you may not use this file except in compliance with the License.
  5. You may obtain a copy of the License at
  6. http://www.apache.org/licenses/LICENSE-2.0
  7. Unless required by applicable law or agreed to in writing, software
  8. distributed under the License is distributed on an "AS IS" BASIS,
  9. WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  10. See the License for the specific language governing permissions and
  11. limitations under the License.
  12. ############################################################################## */
  13. #include "platform.h"
  14. #include <jni.h>
  15. #include "jexcept.hpp"
  16. #include "jthread.hpp"
  17. #include "hqlplugins.hpp"
  18. #include "deftype.hpp"
  19. #include "eclhelper.hpp"
  20. #include "eclrtl.hpp"
  21. #include "eclrtl_imp.hpp"
  22. #include "rtlfield.hpp"
  23. #include "rtlds_imp.hpp"
  24. #include "jprop.hpp"
  25. #include "build-config.h"
  26. #include "roxiemem.hpp"
  27. #include "nbcd.hpp"
  28. #include "rtlformat.hpp"
  29. #include "esdl_def.hpp"
  30. #ifndef _WIN32
  31. #include <sys/resource.h>
  32. #endif
  33. static const char * compatibleVersions[] = {
  34. "Java Embed Helper 1.0.0",
  35. NULL };
  36. static const char *version = "Java Embed Helper 1.0.0";
  37. extern "C" DECL_EXPORT bool getECLPluginDefinition(ECLPluginDefinitionBlock *pb)
  38. {
  39. if (pb->size == sizeof(ECLPluginDefinitionBlockEx))
  40. {
  41. ECLPluginDefinitionBlockEx * pbx = (ECLPluginDefinitionBlockEx *) pb;
  42. pbx->compatibleVersions = compatibleVersions;
  43. }
  44. else if (pb->size != sizeof(ECLPluginDefinitionBlock))
  45. return false;
  46. pb->magicVersion = PLUGIN_VERSION;
  47. pb->version = version;
  48. pb->moduleName = "java";
  49. pb->ECL = NULL;
  50. pb->flags = PLUGIN_MULTIPLE_VERSIONS;
  51. pb->description = "Java Embed Helper";
  52. return true;
  53. }
  54. __declspec(noreturn) static void UNSUPPORTED(const char *feature) __attribute__((noreturn));
  55. static void UNSUPPORTED(const char *feature)
  56. {
  57. throw MakeStringException(-1, "UNSUPPORTED feature: %s not supported in java plugin", feature);
  58. }
  59. namespace javaembed {
  60. // Use a global object to ensure that the Java VM is initialized once only.
  61. // We would like to create it lazily for two reasons:
  62. // 1. So that we only get a JVM if we need one (even if we have loaded the plugin)
  63. // 2. It's important for the JVM to be initialized AFTER we have set up signal handlers, as it
  64. // likes to set its own (in particular, it seems to intercept and ignore some SIGSEGV during the
  65. // garbage collection).
  66. // Unfortunately, it seems that the design of the JNI interface is such that JNI_CreateJavaVM has to be called on the 'main thread'.
  67. // So we can't achieve 1, and 2 requires that we create via the INIT_MODULE mechanism (rather than just a static object), and that
  68. // any engines that call InitModuleObjects() or load plugins dynamically do so AFTER setting any signal handlers or calling
  69. // EnableSEHtoExceptionMapping
  70. //
  71. static class JavaGlobalState
  72. {
  73. public:
  74. JavaGlobalState()
  75. {
  76. JavaVMInitArgs vm_args; /* JDK/JRE 6 VM initialization arguments */
  77. StringArray optionStrings;
  78. const char* origPath = getenv("CLASSPATH");
  79. StringBuffer newPath;
  80. newPath.append("-Djava.class.path=");
  81. if (origPath && *origPath)
  82. {
  83. newPath.append(origPath).append(ENVSEPCHAR);
  84. }
  85. const IProperties &conf = queryEnvironmentConf();
  86. if (conf.hasProp("classpath"))
  87. {
  88. conf.getProp("classpath", newPath);
  89. newPath.append(ENVSEPCHAR);
  90. }
  91. else
  92. {
  93. newPath.append(INSTALL_DIR).append(PATHSEPCHAR).append("classes").append(ENVSEPCHAR);
  94. }
  95. newPath.append(".");
  96. optionStrings.append(newPath);
  97. if (conf.hasProp("jvmlibpath"))
  98. {
  99. StringBuffer libPath;
  100. libPath.append("-Djava.library.path=");
  101. conf.getProp("jvmlibpath", libPath);
  102. optionStrings.append(libPath);
  103. }
  104. // Options we should set (but allow for override with jvmoptions below)
  105. optionStrings.append("-XX:-UseLargePages");
  106. if (conf.hasProp("jvmoptions"))
  107. {
  108. // Use space as field sep as ':' and ';' are valid
  109. optionStrings.appendList(conf.queryProp("jvmoptions"), " ");
  110. }
  111. // Options we know we always want set
  112. optionStrings.append("-Xrs");
  113. #ifdef RLIMIT_STACK
  114. // JVM has a habit of reducing the stack limit on main thread to 1M - probably dates back to when it was actually an increase...
  115. StringBuffer stackOption("-Xss");
  116. struct rlimit limit;
  117. rlim_t slim = 0;
  118. if (getrlimit (RLIMIT_STACK, &limit)==0)
  119. slim = limit.rlim_cur;
  120. if (!slim)
  121. slim = 8*1024*1024;
  122. if (slim >= 1*1024*1024)
  123. {
  124. stackOption.append((__uint64) slim);
  125. optionStrings.append(stackOption);
  126. }
  127. #endif
  128. // These may be useful for debugging
  129. #ifdef _DEBUG
  130. // optionStrings.append("-Xcheck:jni");
  131. // optionStrings.append("-verbose:jni");
  132. #endif
  133. JavaVMOption* options = new JavaVMOption[optionStrings.length()];
  134. ForEachItemIn(idx, optionStrings)
  135. {
  136. // DBGLOG("javaembed: Setting JVM option: %s",(char *)optionStrings.item(idx));
  137. options[idx].optionString = (char *) optionStrings.item(idx);
  138. options[idx].extraInfo = NULL;
  139. }
  140. vm_args.nOptions = optionStrings.length();
  141. vm_args.options = options;
  142. vm_args.ignoreUnrecognized = true;
  143. vm_args.version = JNI_VERSION_1_6;
  144. /* load and initialize a Java VM, return a JNI interface pointer in env */
  145. JNIEnv *env; /* receives pointer to native method interface */
  146. int createResult = JNI_CreateJavaVM(&javaVM, (void**)&env, &vm_args);
  147. delete [] options;
  148. if (createResult != 0)
  149. throw MakeStringException(0, "javaembed: Unable to initialize JVM (%d)",createResult);
  150. }
  151. ~JavaGlobalState()
  152. {
  153. // We don't attempt to destroy the Java VM, as it's buggy...
  154. }
  155. JavaVM *javaVM; /* denotes a Java VM */
  156. } *globalState;
  157. static StringBuffer helperLibraryName;
  158. #ifdef _WIN32
  159. EXTERN_C IMAGE_DOS_HEADER __ImageBase;
  160. #endif
  161. MODULE_INIT(INIT_PRIORITY_STANDARD)
  162. {
  163. globalState = new JavaGlobalState;
  164. // Make sure we are never unloaded (as JVM does not support it)
  165. // we do this by doing a dynamic load of the javaembed library
  166. #ifdef _WIN32
  167. char ln[_MAX_PATH];
  168. ::GetModuleFileName((HINSTANCE)&__ImageBase, ln, _MAX_PATH);
  169. if (strstr(path, "javaembed"))
  170. {
  171. HINSTANCE h = LoadSharedObject(ln, false, false);
  172. helperLibraryName.set(ln);
  173. DBGLOG("LoadSharedObject returned %p", h);
  174. }
  175. #else
  176. if (findLoadedModule(helperLibraryName, "javaembed"))
  177. {
  178. HINSTANCE h = LoadSharedObject(helperLibraryName, false, false);
  179. // Deliberately leak this handle
  180. }
  181. #endif
  182. return true;
  183. }
  184. MODULE_EXIT()
  185. {
  186. // We don't attempt to destroy the Java VM, as it's buggy...
  187. // delete globalState;
  188. // globalState = NULL;
  189. }
  190. static void checkType(type_t javatype, size32_t javasize, type_t ecltype, size32_t eclsize)
  191. {
  192. if (javatype != ecltype || javasize != eclsize)
  193. throw MakeStringException(0, "javaembed: Type mismatch"); // MORE - could provide some details!
  194. }
  195. static void checkException(JNIEnv *JNIenv, bool fail=true)
  196. {
  197. if (JNIenv->ExceptionCheck())
  198. {
  199. jthrowable exception = JNIenv->ExceptionOccurred();
  200. JNIenv->ExceptionClear();
  201. jclass throwableClass = JNIenv->FindClass("java/lang/Throwable");
  202. jmethodID throwableToString = JNIenv->GetMethodID(throwableClass, "toString", "()Ljava/lang/String;");
  203. jstring cause = (jstring) JNIenv->CallObjectMethod(exception, throwableToString);
  204. const char *text = JNIenv->GetStringUTFChars(cause, 0);
  205. VStringBuffer message("javaembed: %s", text);
  206. JNIenv->ReleaseStringUTFChars(cause, text);
  207. if (fail)
  208. rtlFail(0, message);
  209. throw MakeStringExceptionDirect(0, message);
  210. }
  211. }
  212. //-------------------------------------------
  213. // A JavaObject accessor has common functionality shared by both the builders below (Java-> ECL and ECL->Java)
  214. class JavaObjectAccessor : public CInterface
  215. {
  216. protected:
  217. JavaObjectAccessor(JNIEnv *_JNIenv, const RtlFieldInfo *_outerRow, jobject _row)
  218. : JNIenv(_JNIenv), row(_row), outerRow(_outerRow), idx(0), limit(0), inSet(false), inDataSet(false)
  219. {
  220. Class = (jclass) JNIenv->NewGlobalRef(JNIenv->GetObjectClass(row));
  221. }
  222. JavaObjectAccessor(JNIEnv *_JNIenv, const RtlFieldInfo *_outerRow)
  223. : JNIenv(_JNIenv), outerRow(_outerRow), idx(0), limit(0), inSet(false), inDataSet(false)
  224. {
  225. row = NULL;
  226. Class = NULL;
  227. }
  228. ~JavaObjectAccessor()
  229. {
  230. // Unwind anything left on the stack (in case we had exceptions), to make sure the Class we release is the global one
  231. if (stack.length())
  232. Class = (jclass) stack.item(0);
  233. if (Class)
  234. JNIenv->DeleteGlobalRef(Class);
  235. }
  236. void push()
  237. {
  238. stack.append(Class);
  239. stack.append(row);
  240. }
  241. void pop()
  242. {
  243. row = (jobject) stack.popGet();
  244. Class = (jclass) stack.popGet();
  245. }
  246. jfieldID getFieldId(const RtlFieldInfo * field, const char *sig, const char *expected)
  247. {
  248. // MORE - if we are going to stream a dataset we really should be caching these somehow
  249. JNIenv->ExceptionClear();
  250. jfieldID fieldId = 0;
  251. if (sig)
  252. {
  253. if (inSet)
  254. {
  255. VStringBuffer arraySig("[%s", sig);
  256. fieldId = JNIenv->GetFieldID(Class, field->name, arraySig.str());
  257. }
  258. else
  259. fieldId = JNIenv->GetFieldID(Class, field->name, sig);
  260. }
  261. else
  262. {
  263. // Do it the hard way via reflection API
  264. // Equivalent java:
  265. // Field field = object.getClass().getDeclaredField(fieldName);
  266. jclass classClass =JNIenv->GetObjectClass(Class);
  267. checkException();
  268. jmethodID getDeclaredField = JNIenv->GetMethodID(classClass, "getDeclaredField", "(Ljava/lang/String;)Ljava/lang/reflect/Field;" );
  269. checkException();
  270. jstring fieldName = JNIenv->NewStringUTF(field->name);
  271. checkException();
  272. jobject reflectedField = JNIenv->CallObjectMethod(Class, getDeclaredField, fieldName);
  273. checkException();
  274. fieldId = JNIenv->FromReflectedField(reflectedField);
  275. }
  276. if (!fieldId && expected)
  277. throw MakeStringException(0, "javaembed: Unable to retrieve field %s of type %s", field->name, expected);
  278. if (expected)
  279. checkException();
  280. else
  281. JNIenv->ExceptionClear();
  282. return fieldId;
  283. }
  284. void checkException()
  285. {
  286. javaembed::checkException(JNIenv);
  287. }
  288. JNIEnv *JNIenv;
  289. jobject row;
  290. const RtlFieldInfo *outerRow;
  291. jclass Class;
  292. ConstPointerArray stack;
  293. unsigned idx;
  294. UnsignedArray idxStack;
  295. unsigned limit;
  296. bool inSet;
  297. bool inDataSet;
  298. };
  299. // A JavaRowBuilder object is used to construct an ECL row from a Java object
  300. class JavaRowBuilder : public JavaObjectAccessor, implements IFieldSource
  301. {
  302. public:
  303. IMPLEMENT_IINTERFACE;
  304. JavaRowBuilder(JNIEnv *_JNIenv, const RtlFieldInfo *_outerRow, jobject _row)
  305. : JavaObjectAccessor(_JNIenv, _outerRow, _row)
  306. {
  307. }
  308. virtual bool getBooleanResult(const RtlFieldInfo *field)
  309. {
  310. jboolean b;
  311. if (inSet)
  312. {
  313. JNIenv->GetBooleanArrayRegion((jbooleanArray) row, idx, 1, &b);
  314. }
  315. else
  316. {
  317. jfieldID fieldId = getFieldId(field, "Z", "boolean");
  318. b = JNIenv->GetBooleanField(row, fieldId);
  319. }
  320. checkException();
  321. return b;
  322. }
  323. virtual void getDataResult(const RtlFieldInfo *field, size32_t &__len, void * &__result)
  324. {
  325. jbyteArray array;
  326. if (inSet)
  327. {
  328. array = (jbyteArray) JNIenv->GetObjectArrayElement((jobjectArray) row, idx);
  329. }
  330. else
  331. {
  332. jfieldID fieldId = getFieldId(field, "[B", "DATA");
  333. array = (jbyteArray) JNIenv->GetObjectField(row, fieldId);
  334. }
  335. checkException();
  336. __len = (array != NULL ? JNIenv->GetArrayLength(array) : 0);
  337. __result = (__len > 0 ? rtlMalloc(__len) : NULL);
  338. if (__result)
  339. JNIenv->GetByteArrayRegion(array, 0, __len, (jbyte *) __result);
  340. checkException();
  341. }
  342. virtual double getRealResult(const RtlFieldInfo *field)
  343. {
  344. double d;
  345. if (inSet)
  346. {
  347. float f;
  348. switch (field->size(NULL, NULL))
  349. {
  350. case 4:
  351. JNIenv->GetFloatArrayRegion((jfloatArray) row, idx, 1, &f);
  352. d = f;
  353. break;
  354. case 8:
  355. JNIenv->GetDoubleArrayRegion((jdoubleArray) row, idx, 1, &d);
  356. break;
  357. default:
  358. throwUnexpected();
  359. }
  360. }
  361. else
  362. {
  363. jfieldID fieldId;
  364. switch (field->size(NULL, NULL))
  365. {
  366. case 4:
  367. fieldId = getFieldId(field, "F", "float");
  368. d = JNIenv->GetFloatField(row, fieldId);
  369. break;
  370. case 8:
  371. fieldId = getFieldId(field, "D", "double");
  372. d = JNIenv->GetDoubleField(row, fieldId);
  373. break;
  374. default:
  375. throwUnexpected();
  376. }
  377. }
  378. checkException();
  379. return d;
  380. }
  381. virtual __int64 getSignedResult(const RtlFieldInfo *field)
  382. {
  383. __int64 ret;
  384. if (inSet)
  385. {
  386. jbyte b;
  387. jshort s;
  388. jint i;
  389. jlong l;
  390. switch (field->size(NULL, NULL))
  391. {
  392. case 1:
  393. JNIenv->GetByteArrayRegion((jbyteArray) row, idx, 1, &b);
  394. ret = b;
  395. break;
  396. case 2:
  397. JNIenv->GetShortArrayRegion((jshortArray) row, idx, 1, &s);
  398. ret = s;
  399. break;
  400. case 4:
  401. JNIenv->GetIntArrayRegion((jintArray) row, idx, 1, &i);
  402. ret = i;
  403. break;
  404. case 8:
  405. JNIenv->GetLongArrayRegion((jlongArray) row, idx, 1, &l);
  406. ret = l;
  407. break;
  408. default:
  409. UNSUPPORTED("non-standard integer sizes");
  410. }
  411. }
  412. else
  413. {
  414. jfieldID fieldId;
  415. switch (field->size(NULL, NULL))
  416. {
  417. case 1:
  418. fieldId = getFieldId(field, "B", "byte");
  419. ret = JNIenv->GetByteField(row, fieldId);
  420. break;
  421. case 2:
  422. fieldId = getFieldId(field, "S", "short");
  423. ret = JNIenv->GetShortField(row, fieldId);
  424. break;
  425. case 4:
  426. fieldId = getFieldId(field, "I", "int");
  427. ret = JNIenv->GetIntField(row, fieldId);
  428. break;
  429. case 8:
  430. fieldId = getFieldId(field, "J", "long");
  431. ret = JNIenv->GetLongField(row, fieldId);
  432. break;
  433. default:
  434. UNSUPPORTED("non-standard integer sizes");
  435. }
  436. }
  437. checkException();
  438. return ret;
  439. }
  440. virtual unsigned __int64 getUnsignedResult(const RtlFieldInfo *field)
  441. {
  442. UNSUPPORTED("unsigned fields"); // No unsigned types in Java
  443. }
  444. virtual void getStringResult(const RtlFieldInfo *field, size32_t &__len, char * &__result)
  445. {
  446. jstring result;
  447. if (inSet)
  448. {
  449. // MORE - set of string1 mapping to Java array of char ? Not sure it's worth it.
  450. result = (jstring) JNIenv->GetObjectArrayElement((jobjectArray) row, idx);
  451. }
  452. else
  453. {
  454. if (field->isFixedSize() && field->size(NULL, NULL)==1)
  455. {
  456. // See if there's a char field
  457. jfieldID charFieldId = getFieldId(field, "C", NULL);
  458. if (charFieldId)
  459. {
  460. jchar resultChar = JNIenv->GetCharField(row, charFieldId);
  461. rtlUnicodeToStrX(__len, __result, 1, &resultChar);
  462. return;
  463. }
  464. }
  465. jfieldID fieldId = getFieldId(field, "Ljava/lang/String;", "String");
  466. result = (jstring) JNIenv->GetObjectField(row, fieldId);
  467. }
  468. if (!result)
  469. {
  470. __len = 0;
  471. __result = NULL;
  472. return;
  473. }
  474. size_t size = JNIenv->GetStringUTFLength(result); // in bytes
  475. const char *text = JNIenv->GetStringUTFChars(result, NULL);
  476. size32_t chars = rtlUtf8Length(size, text);
  477. rtlUtf8ToStrX(__len, __result, chars, text);
  478. JNIenv->ReleaseStringUTFChars(result, text);
  479. JNIenv->DeleteLocalRef(result);
  480. }
  481. virtual void getUTF8Result(const RtlFieldInfo *field, size32_t &__len, char * &__result)
  482. {
  483. jstring result;
  484. if (inSet)
  485. {
  486. // MORE - set of string1 mapping to Java array of char ? Not sure it's worth it.
  487. result = (jstring) JNIenv->GetObjectArrayElement((jobjectArray) row, idx);
  488. }
  489. else
  490. {
  491. if (field->isFixedSize() && field->size(NULL, NULL)==1)
  492. {
  493. // See if there's a char field
  494. jfieldID charFieldId = getFieldId(field, "C", NULL);
  495. if (charFieldId)
  496. {
  497. jchar resultChar = JNIenv->GetCharField(row, charFieldId);
  498. rtlUnicodeToUtf8X(__len, __result, 1, &resultChar);
  499. return;
  500. }
  501. }
  502. jfieldID fieldId = getFieldId(field, "Ljava/lang/String;", "String");
  503. result = (jstring) JNIenv->GetObjectField(row, fieldId);
  504. }
  505. if (!result)
  506. {
  507. __len = 0;
  508. __result = NULL;
  509. return;
  510. }
  511. size_t size = JNIenv->GetStringUTFLength(result); // in bytes
  512. const char *text = JNIenv->GetStringUTFChars(result, NULL);
  513. size32_t chars = rtlUtf8Length(size, text);
  514. rtlUtf8ToUtf8X(__len, __result, chars, text);
  515. JNIenv->ReleaseStringUTFChars(result, text);
  516. JNIenv->DeleteLocalRef(result);
  517. }
  518. virtual void getUnicodeResult(const RtlFieldInfo *field, size32_t &__len, UChar * &__result)
  519. {
  520. jstring result;
  521. if (inSet)
  522. {
  523. // MORE - set of string1 mapping to Java array of char ? Not sure it's worth it.
  524. result = (jstring) JNIenv->GetObjectArrayElement((jobjectArray) row, idx);
  525. }
  526. else
  527. {
  528. if (field->isFixedSize() && field->size(NULL, NULL)==1)
  529. {
  530. // See if there's a char field
  531. jfieldID charFieldId = getFieldId(field, "C", NULL);
  532. if (charFieldId)
  533. {
  534. jchar resultChar = JNIenv->GetCharField(row, charFieldId);
  535. rtlUnicodeToUnicodeX(__len, __result, 1, &resultChar);
  536. return;
  537. }
  538. }
  539. jfieldID fieldId = getFieldId(field, "Ljava/lang/String;", "String");
  540. result = (jstring) JNIenv->GetObjectField(row, fieldId);
  541. }
  542. if (!result)
  543. {
  544. __len = 0;
  545. __result = NULL;
  546. return;
  547. }
  548. size_t size = JNIenv->GetStringUTFLength(result); // in bytes
  549. const char *text = JNIenv->GetStringUTFChars(result, NULL);
  550. size32_t chars = rtlUtf8Length(size, text);
  551. rtlUtf8ToUnicodeX(__len, __result, chars, text);
  552. JNIenv->ReleaseStringUTFChars(result, text);
  553. JNIenv->DeleteLocalRef(result);
  554. }
  555. virtual void getDecimalResult(const RtlFieldInfo *field, Decimal &value)
  556. {
  557. double ret = getRealResult(field);
  558. value.setReal(ret);
  559. }
  560. virtual void processBeginSet(const RtlFieldInfo * field, bool &isAll)
  561. {
  562. isAll = false; // No concept of an 'all' set in Java
  563. push();
  564. jfieldID fieldId = getFieldId(field, NULL, "object"); // We assume it will be an array, but not sure of what...
  565. row = JNIenv->GetObjectField(row, fieldId);
  566. inSet = true;
  567. idx = -1; // First call to next() increments it to 0
  568. limit = row != NULL ? JNIenv->GetArrayLength((jarray) row) : 0;
  569. checkException();
  570. }
  571. virtual bool processNextSet(const RtlFieldInfo * field)
  572. {
  573. assertex(inSet);
  574. idx++;
  575. return idx < limit;
  576. }
  577. virtual void processBeginDataset(const RtlFieldInfo * field)
  578. {
  579. push();
  580. jfieldID fieldId = getFieldId(field, NULL, "object"); // We assume it will be an array, but not sure of what...
  581. row = JNIenv->GetObjectField(row, fieldId);
  582. inDataSet = true;
  583. idx = -1; // First call to next() increments it to 0
  584. limit = row != NULL ? JNIenv->GetArrayLength((jarray) row) : 0;
  585. checkException();
  586. }
  587. virtual void processBeginRow(const RtlFieldInfo * field)
  588. {
  589. if (field != outerRow)
  590. {
  591. push();
  592. if (inDataSet)
  593. {
  594. row = JNIenv->GetObjectArrayElement((jobjectArray) row, idx);
  595. }
  596. else
  597. {
  598. jfieldID fieldId = getFieldId(field, NULL, "object");
  599. row = JNIenv->GetObjectField(row, fieldId);
  600. }
  601. if (!row)
  602. rtlFail(0, "javaembed: child dataset object should not be NULL");
  603. Class = JNIenv->GetObjectClass(row);
  604. }
  605. }
  606. virtual bool processNextRow(const RtlFieldInfo * field)
  607. {
  608. assertex(inDataSet);
  609. idx++;
  610. return idx < limit;
  611. }
  612. virtual void processEndSet(const RtlFieldInfo * field)
  613. {
  614. inSet = false;
  615. JNIenv->DeleteLocalRef(row);
  616. pop();
  617. }
  618. virtual void processEndDataset(const RtlFieldInfo * field)
  619. {
  620. inDataSet = false;
  621. JNIenv->DeleteLocalRef(row);
  622. pop();
  623. }
  624. virtual void processEndRow(const RtlFieldInfo * field)
  625. {
  626. if (field != outerRow)
  627. {
  628. JNIenv->DeleteLocalRef(row);
  629. JNIenv->DeleteLocalRef(Class);
  630. pop();
  631. }
  632. }
  633. };
  634. //-------------------------------------------
  635. // A JavaObjectBuilder object is used to construct a Java object from an ECL row
  636. class JavaObjectBuilder : public JavaObjectAccessor, implements IFieldProcessor
  637. {
  638. public:
  639. IMPLEMENT_IINTERFACE;
  640. JavaObjectBuilder(JNIEnv *_JNIenv, const RtlFieldInfo *_outerRow, const char *className)
  641. : JavaObjectAccessor(_JNIenv, _outerRow)
  642. {
  643. JNIenv->ExceptionClear();
  644. Class = (jclass) JNIenv->NewGlobalRef(JNIenv->FindClass(className)); // MORE - should use the custom classloader, once that fix is merged
  645. checkException();
  646. setConstructor();
  647. }
  648. virtual void processString(unsigned numchars, const char *text, const RtlFieldInfo * field)
  649. {
  650. if (field->isFixedSize() && field->size(NULL, NULL)==1 && !inSet) // SET OF STRING1 is not mapped to array of char...
  651. {
  652. // See if there's a char field
  653. jfieldID charFieldId = getFieldId(field, "C", NULL);
  654. if (charFieldId)
  655. {
  656. assertex(numchars==1);
  657. jchar c;
  658. rtlStrToUnicode(1, &c, 1, text);
  659. JNIenv->SetCharField(row, charFieldId, c);
  660. checkException();
  661. return;
  662. }
  663. }
  664. jfieldID fieldId = getFieldId(field, "Ljava/lang/String;", "String");
  665. size32_t numchars16;
  666. rtlDataAttr unicode16;
  667. rtlStrToUnicodeX(numchars16, unicode16.refustr(), numchars, text);
  668. jstring value = JNIenv->NewString(unicode16.getustr(), numchars16);
  669. checkException();
  670. if (inSet)
  671. JNIenv->SetObjectArrayElement((jobjectArray) row, idx, value);
  672. else
  673. JNIenv->SetObjectField(row, fieldId, value);
  674. JNIenv->DeleteLocalRef(value);
  675. checkException();
  676. }
  677. virtual void processBool(bool value, const RtlFieldInfo * field)
  678. {
  679. jfieldID fieldId = getFieldId(field, "Z", "boolean");
  680. JNIenv->SetBooleanField(row, fieldId, value);
  681. checkException();
  682. }
  683. virtual void processData(unsigned len, const void *value, const RtlFieldInfo * field)
  684. {
  685. jfieldID fieldId = getFieldId(field, "[B", "data");
  686. jbyteArray javaData = JNIenv->NewByteArray(len);
  687. JNIenv->SetByteArrayRegion(javaData, 0, len, (jbyte *) value);
  688. checkException();
  689. if (inSet)
  690. JNIenv->SetObjectArrayElement((jobjectArray) row, idx, javaData);
  691. else
  692. JNIenv->SetObjectField(row, fieldId, javaData);
  693. checkException();
  694. }
  695. virtual void processInt(__int64 value, const RtlFieldInfo * field)
  696. {
  697. jfieldID fieldId;
  698. switch (field->size(NULL, NULL))
  699. {
  700. case 1:
  701. fieldId = getFieldId(field, "B", "byte");
  702. JNIenv->SetByteField(row, fieldId, value);
  703. break;
  704. case 2:
  705. fieldId = getFieldId(field, "S", "short");
  706. JNIenv->SetShortField(row, fieldId, value);
  707. break;
  708. case 4:
  709. fieldId = getFieldId(field, "I", "int");
  710. JNIenv->SetIntField(row, fieldId, value);
  711. break;
  712. case 8:
  713. fieldId = getFieldId(field, "J", "long");
  714. JNIenv->SetLongField(row, fieldId, value);
  715. break;
  716. default:
  717. UNSUPPORTED("non-standard integer sizes");
  718. break;
  719. }
  720. checkException();
  721. }
  722. virtual void processUInt(unsigned __int64 value, const RtlFieldInfo * field)
  723. {
  724. UNSUPPORTED("unsigned fields"); // No unsigned types in Java
  725. }
  726. virtual void processReal(double value, const RtlFieldInfo * field)
  727. {
  728. jfieldID fieldId;
  729. switch (field->size(NULL, NULL))
  730. {
  731. case 4:
  732. fieldId = getFieldId(field, "F", "float");
  733. JNIenv->SetFloatField(row, fieldId, (float) value);
  734. break;
  735. case 8:
  736. fieldId = getFieldId(field, "D", "double");
  737. JNIenv->SetDoubleField(row, fieldId, value);
  738. break;
  739. default:
  740. throwUnexpected();
  741. }
  742. checkException();
  743. }
  744. virtual void processDecimal(const void *value, unsigned digits, unsigned precision, const RtlFieldInfo * field)
  745. {
  746. // we could map to doubles, but probably better to let the ECL programmer do that themselves
  747. UNSUPPORTED("DECIMAL fields");
  748. }
  749. virtual void processUDecimal(const void *value, unsigned digits, unsigned precision, const RtlFieldInfo * field)
  750. {
  751. UNSUPPORTED("UDECIMAL fields");
  752. }
  753. virtual void processUnicode(unsigned numchars, const UChar *text, const RtlFieldInfo * field)
  754. {
  755. jfieldID fieldId = getFieldId(field, "Ljava/lang/String;", "String");
  756. jstring value = JNIenv->NewString(text, numchars);
  757. checkException();
  758. if (inSet)
  759. JNIenv->SetObjectArrayElement((jobjectArray) row, idx, value);
  760. else
  761. JNIenv->SetObjectField(row, fieldId, value);
  762. JNIenv->DeleteLocalRef(value);
  763. checkException();
  764. }
  765. virtual void processQString(unsigned len, const char *value, const RtlFieldInfo * field)
  766. {
  767. size32_t charCount;
  768. rtlDataAttr text;
  769. rtlQStrToStrX(charCount, text.refstr(), len, value);
  770. processString(charCount, text.getstr(), field);
  771. }
  772. virtual void processUtf8(unsigned numchars, const char *text, const RtlFieldInfo * field)
  773. {
  774. jfieldID fieldId = getFieldId(field, "Ljava/lang/String;", "String");
  775. size32_t numchars16;
  776. rtlDataAttr unicode16;
  777. rtlUtf8ToUnicodeX(numchars16, unicode16.refustr(), numchars, text);
  778. jstring value = JNIenv->NewString(unicode16.getustr(), numchars16);
  779. checkException();
  780. if (inSet)
  781. JNIenv->SetObjectArrayElement((jobjectArray) row, idx, value);
  782. else
  783. JNIenv->SetObjectField(row, fieldId, value);
  784. JNIenv->DeleteLocalRef(value);
  785. checkException();
  786. }
  787. virtual bool processBeginSet(const RtlFieldInfo * field, unsigned numElems, bool isAll, const byte *data)
  788. {
  789. push();
  790. idx = 0;
  791. limit = numElems;
  792. const char *javaTypeSignature = NULL;
  793. bool processElements = false;
  794. // row needs to be created as an array of <whatever>
  795. if (isAll)
  796. UNSUPPORTED("ALL sets");
  797. const RtlTypeInfo *childType = field->type->queryChildType();
  798. jobject newRow;
  799. switch(childType->fieldType & RFTMkind)
  800. {
  801. case type_boolean:
  802. newRow = JNIenv->NewBooleanArray(numElems);
  803. JNIenv->SetBooleanArrayRegion((jbooleanArray) newRow, 0, numElems, (jboolean *) data);
  804. javaTypeSignature = "[Z";
  805. break;
  806. case type_int:
  807. if (childType->fieldType & RFTMunsigned)
  808. UNSUPPORTED("unsigned integers");
  809. switch (childType->length)
  810. {
  811. case 1:
  812. newRow = JNIenv->NewByteArray(numElems);
  813. JNIenv->SetByteArrayRegion((jbyteArray) newRow, 0, numElems, (jbyte *) data);
  814. javaTypeSignature = "[B";
  815. break;
  816. case 2:
  817. newRow = JNIenv->NewShortArray(numElems);
  818. JNIenv->SetShortArrayRegion((jshortArray) newRow, 0, numElems, (jshort *) data);
  819. javaTypeSignature = "[S";
  820. break;
  821. case 4:
  822. newRow = JNIenv->NewIntArray(numElems);
  823. JNIenv->SetIntArrayRegion((jintArray) newRow, 0, numElems, (jint *) data);
  824. javaTypeSignature = "[I";
  825. break;
  826. case 8:
  827. newRow = JNIenv->NewLongArray(numElems);
  828. JNIenv->SetLongArrayRegion((jlongArray) newRow, 0, numElems, (jlong *) data);
  829. javaTypeSignature = "[J";
  830. break;
  831. default:
  832. UNSUPPORTED("non-standard integer sizes");
  833. break;
  834. }
  835. break;
  836. case type_real:
  837. switch (childType->length)
  838. {
  839. case 4:
  840. newRow = JNIenv->NewFloatArray(numElems);
  841. JNIenv->SetFloatArrayRegion((jfloatArray) newRow, 0, numElems, (float *) data);
  842. javaTypeSignature = "[F";
  843. break;
  844. case 8:
  845. newRow = JNIenv->NewDoubleArray(numElems);
  846. JNIenv->SetDoubleArrayRegion((jdoubleArray) newRow, 0, numElems, (double *) data);
  847. javaTypeSignature = "[D";
  848. break;
  849. default:
  850. throwUnexpected();
  851. break;
  852. }
  853. break;
  854. case type_string:
  855. case type_varstring:
  856. case type_unicode:
  857. case type_utf8:
  858. newRow = JNIenv->NewObjectArray(numElems, JNIenv->FindClass("java/lang/String"), NULL);
  859. javaTypeSignature = "[Ljava/lang/String;";
  860. processElements = true;
  861. break;
  862. case type_data:
  863. newRow = JNIenv->NewObjectArray(numElems, JNIenv->FindClass("[B"), NULL);
  864. javaTypeSignature = "[[B";
  865. processElements = true;
  866. break;
  867. default:
  868. throwUnexpected();
  869. }
  870. checkException();
  871. jfieldID fieldId = getFieldId(field, javaTypeSignature, "Array");
  872. JNIenv->SetObjectField(row, fieldId, newRow);
  873. row = newRow;
  874. inSet = true;
  875. return processElements;
  876. }
  877. virtual bool processBeginDataset(const RtlFieldInfo * field, unsigned numRows)
  878. {
  879. push();
  880. idxStack.append(idx);
  881. idx = 0;
  882. inDataSet = true;
  883. // Create an empty array
  884. jfieldID childId = getFieldId( field, NULL, "RECORD");
  885. jobject newRow = NULL;
  886. if (numRows)
  887. {
  888. jclass arrayClass = getClassForChild(childId);
  889. jmethodID isArrayMethod = JNIenv->GetMethodID(JNIenv->GetObjectClass(arrayClass), "isArray", "()Z" );
  890. checkException();
  891. if (!JNIenv->CallBooleanMethod(arrayClass, isArrayMethod))
  892. {
  893. JNIenv->ExceptionClear();
  894. VStringBuffer message("javaembed: Array expected for field %s", field->name);
  895. rtlFail(0, message.str());
  896. }
  897. // Set up constructor etc for the child rows, so we don't do it per row
  898. jmethodID getTypeMethod = JNIenv->GetMethodID(JNIenv->GetObjectClass(arrayClass), "getComponentType", "()Ljava/lang/Class;" );
  899. checkException();
  900. Class = (jclass) JNIenv->CallObjectMethod(arrayClass, getTypeMethod);
  901. checkException();
  902. setConstructor();
  903. // Now we need to create the array
  904. newRow = JNIenv->NewObjectArray(numRows, Class, NULL);
  905. checkException();
  906. }
  907. JNIenv->SetObjectField(row, childId, newRow);
  908. checkException();
  909. row = newRow;
  910. return true;
  911. }
  912. virtual bool processBeginRow(const RtlFieldInfo * field)
  913. {
  914. if (field == outerRow)
  915. row = JNIenv->NewObject(Class, constructor);
  916. else
  917. {
  918. push();
  919. stack.append(constructor);
  920. // Now we have to create the child object
  921. jobject newRow = NULL;
  922. if (inDataSet)
  923. {
  924. newRow = JNIenv->NewObject(Class, constructor);
  925. checkException();
  926. JNIenv->SetObjectArrayElement((jobjectArray) row, idx++, newRow);
  927. }
  928. else
  929. {
  930. // All this is done once per dataset in the nested dataset case. But for embedded record case we have to do it here
  931. jfieldID childId = getFieldId( field, NULL, "RECORD");
  932. Class = getClassForChild(childId);
  933. setConstructor();
  934. newRow = JNIenv->NewObject(Class, constructor);
  935. checkException();
  936. JNIenv->SetObjectField(row, childId, newRow);
  937. }
  938. row = newRow;
  939. }
  940. checkException();
  941. return true;
  942. }
  943. virtual void processEndSet(const RtlFieldInfo * field)
  944. {
  945. JNIenv->DeleteLocalRef(row);
  946. pop();
  947. inSet = false;
  948. }
  949. virtual void processEndDataset(const RtlFieldInfo * field)
  950. {
  951. inDataSet = false;
  952. idx = idxStack.popGet();
  953. pop();
  954. }
  955. virtual void processEndRow(const RtlFieldInfo * field)
  956. {
  957. if (field != outerRow)
  958. {
  959. constructor = (jmethodID) stack.popGet();
  960. JNIenv->DeleteLocalRef(row);
  961. pop();
  962. }
  963. }
  964. inline jobject getObject()
  965. {
  966. return row;
  967. }
  968. protected:
  969. jclass getClassForChild(jfieldID childId)
  970. {
  971. jobject reflectedField = JNIenv->ToReflectedField(Class, childId, false);
  972. checkException();
  973. jclass fieldClass =JNIenv->GetObjectClass(reflectedField);
  974. checkException();
  975. jmethodID getTypeMethod = JNIenv->GetMethodID(fieldClass, "getType", "()Ljava/lang/Class;" );
  976. checkException();
  977. jclass result = (jclass) JNIenv->CallObjectMethod(reflectedField, getTypeMethod);
  978. checkException();
  979. JNIenv->DeleteLocalRef(reflectedField);
  980. JNIenv->DeleteLocalRef(fieldClass);
  981. return result;
  982. }
  983. void setConstructor()
  984. {
  985. constructor = JNIenv->GetMethodID(Class, "<init>", "()V");
  986. if (!constructor)
  987. {
  988. JNIenv->ExceptionClear();
  989. jmethodID getNameMethod = JNIenv->GetMethodID(JNIenv->GetObjectClass(Class), "getName", "()Ljava/lang/String;" );
  990. checkException();
  991. jstring name = (jstring) JNIenv->CallObjectMethod(Class, getNameMethod);
  992. checkException();
  993. const char *nameText = JNIenv->GetStringUTFChars(name, NULL);
  994. VStringBuffer message("javaembed: no suitable constructor for field %s", nameText);
  995. JNIenv->ReleaseStringUTFChars(name, nameText);
  996. rtlFail(0, message.str());
  997. }
  998. }
  999. jmethodID constructor;
  1000. };
  1001. //----------------------------------------------------------------------
  1002. // Wrap an IRowStream into a Java Iterator
  1003. class ECLDatasetIterator : public CInterfaceOf<IInterface>
  1004. {
  1005. public:
  1006. ECLDatasetIterator(JNIEnv *JNIenv, const RtlTypeInfo *_typeInfo, const char *className, IRowStream * _val)
  1007. : typeInfo(_typeInfo), val(_val),
  1008. dummyField("<row>", NULL, typeInfo),
  1009. javaBuilder(JNIenv, &dummyField, className)
  1010. {
  1011. nextRead = false;
  1012. nextPending = NULL;
  1013. }
  1014. bool hasNext()
  1015. {
  1016. if (!nextRead)
  1017. {
  1018. nextPending = (const byte *) val->ungroupedNextRow();
  1019. nextRead = true;
  1020. if (!nextPending)
  1021. val->stop();
  1022. }
  1023. return nextPending != NULL;
  1024. }
  1025. jobject next()
  1026. {
  1027. if (!hasNext())
  1028. return NULL;
  1029. typeInfo->process(nextPending, nextPending, &dummyField, javaBuilder); // Creates a java object from the incoming ECL row
  1030. nextRead = false;
  1031. return javaBuilder.getObject();
  1032. }
  1033. protected:
  1034. const RtlTypeInfo *typeInfo; // Not linked (or linkable)
  1035. Linked<IRowStream> val;
  1036. RtlFieldStrInfo dummyField;
  1037. JavaObjectBuilder javaBuilder;
  1038. const byte *nextPending;
  1039. bool nextRead;
  1040. };
  1041. //-------------------------------------------
  1042. // A Java function that returns a dataset will return a JavaRowStream object that can be
  1043. // interrogated to return each row of the result in turn
  1044. static JNIEnv *queryJNIEnv();
  1045. class JavaLocalFrame
  1046. {
  1047. public:
  1048. JavaLocalFrame(JNIEnv *_JNIenv, unsigned size = 16) : JNIenv(_JNIenv)
  1049. {
  1050. JNIenv->PushLocalFrame(size);
  1051. }
  1052. ~JavaLocalFrame()
  1053. {
  1054. JNIenv->PopLocalFrame(NULL);
  1055. }
  1056. private:
  1057. JNIEnv *JNIenv;
  1058. };
  1059. class JavaRowStream : public CInterfaceOf<IRowStream>
  1060. {
  1061. public:
  1062. JavaRowStream(jobject _iterator, IEngineRowAllocator *_resultAllocator)
  1063. : resultAllocator(_resultAllocator)
  1064. {
  1065. JNIEnv *JNIenv = queryJNIEnv();
  1066. iterator = JNIenv->NewGlobalRef(_iterator);
  1067. // Note that we can't cache the JNIEnv, iterClass, or methodIds here - calls may be made on different threads (though not at the same time).
  1068. }
  1069. ~JavaRowStream()
  1070. {
  1071. stop();
  1072. }
  1073. virtual const void *nextRow()
  1074. {
  1075. if (!iterator)
  1076. return NULL;
  1077. JNIEnv *JNIenv = queryJNIEnv();
  1078. JavaLocalFrame lf(JNIenv);
  1079. // Java code would be
  1080. // if (!iterator.hasNext)
  1081. // {
  1082. // stop();
  1083. // return NULL;
  1084. // }
  1085. // result = iterator.next();
  1086. jclass iterClass =JNIenv->GetObjectClass(iterator);
  1087. javaembed::checkException(JNIenv);
  1088. jmethodID hasNextMethod = JNIenv->GetMethodID(iterClass, "hasNext", "()Z" );
  1089. javaembed::checkException(JNIenv);
  1090. jboolean hasNext = JNIenv->CallBooleanMethod(iterator, hasNextMethod);
  1091. if (!hasNext)
  1092. {
  1093. stop();
  1094. return NULL;
  1095. }
  1096. jmethodID nextMethod = JNIenv->GetMethodID(iterClass, "next", "()Ljava/lang/Object;" );
  1097. javaembed::checkException(JNIenv);
  1098. jobject result = JNIenv->CallObjectMethod(iterator, nextMethod);
  1099. RtlDynamicRowBuilder rowBuilder(resultAllocator);
  1100. const RtlTypeInfo *typeInfo = resultAllocator->queryOutputMeta()->queryTypeInfo();
  1101. assertex(typeInfo);
  1102. RtlFieldStrInfo dummyField("<row>", NULL, typeInfo);
  1103. JavaRowBuilder javaRowBuilder(queryJNIEnv(), &dummyField, result);
  1104. size32_t len = typeInfo->build(rowBuilder, 0, &dummyField, javaRowBuilder);
  1105. return rowBuilder.finalizeRowClear(len);
  1106. }
  1107. virtual void stop()
  1108. {
  1109. resultAllocator.clear();
  1110. JNIEnv *JNIenv = queryJNIEnv();
  1111. if (JNIenv)
  1112. {
  1113. if (iterator)
  1114. {
  1115. JNIenv->DeleteGlobalRef(iterator);
  1116. iterator = NULL;
  1117. }
  1118. }
  1119. }
  1120. protected:
  1121. Linked<IEngineRowAllocator> resultAllocator;
  1122. jobject iterator;
  1123. };
  1124. const char *esdl2JavaSig(IEsdlDefinition &esdl, const char *esdlType)
  1125. {
  1126. EsdlBasicElementType t = esdl.translateSimpleType(esdlType);
  1127. switch (t)
  1128. {
  1129. case ESDLT_INT16:
  1130. case ESDLT_UINT16:
  1131. return "Ljava/lang/Short;";
  1132. case ESDLT_INT32:
  1133. case ESDLT_UINT32:
  1134. return "Ljava/lang/Integer;";
  1135. case ESDLT_INT64:
  1136. case ESDLT_UINT64:
  1137. return "Ljava/math/BigInteger;";
  1138. case ESDLT_BOOL:
  1139. return "Ljava/lang/Boolean;";
  1140. case ESDLT_FLOAT:
  1141. return "Ljava/lang/Float;";
  1142. case ESDLT_DOUBLE:
  1143. return "Ljava/lang/Double;";
  1144. case ESDLT_INT8:
  1145. case ESDLT_UINT8:
  1146. case ESDLT_BYTE:
  1147. case ESDLT_UBYTE:
  1148. return "Ljava/lang/Byte;";
  1149. case ESDLT_STRING:
  1150. return "Ljava/lang/String;";
  1151. case ESDLT_UNKOWN:
  1152. case ESDLT_STRUCT:
  1153. case ESDLT_REQUEST:
  1154. case ESDLT_RESPONSE:
  1155. case ESDLT_COMPLEX:
  1156. default:
  1157. return NULL;
  1158. }
  1159. }
  1160. const char *esdl2JavaFullClassName(IEsdlDefinition &esdl, const char *esdlType)
  1161. {
  1162. EsdlBasicElementType t = esdl.translateSimpleType(esdlType);
  1163. switch (t)
  1164. {
  1165. case ESDLT_INT16:
  1166. case ESDLT_UINT16:
  1167. return "java/lang/Short";
  1168. case ESDLT_INT32:
  1169. case ESDLT_UINT32:
  1170. return "java/lang/Integer";
  1171. case ESDLT_INT64:
  1172. case ESDLT_UINT64:
  1173. return "java/math/BigInteger";
  1174. case ESDLT_BOOL:
  1175. return "java/lang/Boolean";
  1176. case ESDLT_FLOAT:
  1177. return "java/lang/Float";
  1178. case ESDLT_DOUBLE:
  1179. return "java/lang/Double";
  1180. case ESDLT_INT8:
  1181. case ESDLT_UINT8:
  1182. case ESDLT_BYTE:
  1183. case ESDLT_UBYTE:
  1184. return "java/lang/Byte";
  1185. case ESDLT_STRING:
  1186. return "java/lang/String";
  1187. case ESDLT_UNKOWN:
  1188. case ESDLT_STRUCT:
  1189. case ESDLT_REQUEST:
  1190. case ESDLT_RESPONSE:
  1191. case ESDLT_COMPLEX:
  1192. default:
  1193. return NULL;
  1194. }
  1195. }
  1196. class JavaObjectXmlWriter : public CInterface
  1197. {
  1198. public:
  1199. JavaObjectXmlWriter(JNIEnv *_JNIenv, jobject _obj, const char *_reqType, IEsdlDefinition &_esdl, const char *_esdlService, IXmlWriter &_writer)
  1200. : JNIenv(_JNIenv), obj(_obj), writer(_writer), esdl(_esdl), esdlService(_esdlService), reqType(_reqType)
  1201. {
  1202. Class = (jclass) JNIenv->NewGlobalRef(JNIenv->GetObjectClass(obj));
  1203. langObjectClass = FindClass("java/lang/Object");
  1204. objToString = JNIenv->GetMethodID(langObjectClass, "toString", "()Ljava/lang/String;");
  1205. }
  1206. ~JavaObjectXmlWriter()
  1207. {
  1208. if (Class)
  1209. JNIenv->DeleteGlobalRef(Class);
  1210. HashIterator it(javaClasses);
  1211. ForEach(it)
  1212. {
  1213. IMapping &entry = it.query();
  1214. jclass *pClass = javaClasses.mapToValue(&entry);
  1215. if (pClass)
  1216. JNIenv->DeleteGlobalRef(*pClass);
  1217. }
  1218. }
  1219. void writeSimpleType(const char *fieldname, jobject fieldObj)
  1220. {
  1221. jstring fieldStr = (jstring) JNIenv->CallObjectMethod(fieldObj, objToString);
  1222. if (!fieldStr)
  1223. return;
  1224. const char *text = JNIenv->GetStringUTFChars(fieldStr, NULL);
  1225. if (text)
  1226. writer.outputCString(text, fieldname);
  1227. JNIenv->DeleteLocalRef(fieldStr);
  1228. }
  1229. void writeSimpleType(jclass parentClass, jobject parentObject, const char *fieldname, const char *javaSig)
  1230. {
  1231. if (!fieldname || !*fieldname)
  1232. return;
  1233. if (!javaSig || !*javaSig)
  1234. return;
  1235. jfieldID fieldId = JNIenv->GetFieldID(parentClass, fieldname, javaSig);
  1236. if (!fieldId)
  1237. return;
  1238. jobject fieldObj = (jobject) JNIenv->GetObjectField(parentObject, fieldId);
  1239. if (!fieldObj)
  1240. return;
  1241. writeSimpleType(fieldname, fieldObj);
  1242. JNIenv->DeleteLocalRef(fieldObj);
  1243. }
  1244. void writeSimpleType(jclass parentClass, jobject parentObject, IEsdlDefObject &defObject)
  1245. {
  1246. const char *fieldname = defObject.queryName();
  1247. const char *javaSig = esdl2JavaSig(esdl, defObject.queryProp("type"));
  1248. writeSimpleType(parentClass, parentObject, fieldname, javaSig);
  1249. }
  1250. void writeEnumType(jclass parentClass, jobject parentObject, IEsdlDefObject &defObject)
  1251. {
  1252. const char *fieldname = defObject.queryName();
  1253. VStringBuffer javaSig("L%s/%s;", esdlService.str(), defObject.queryProp("enum_type"));
  1254. jfieldID fieldId = JNIenv->GetFieldID(parentClass, fieldname, javaSig);
  1255. if (!fieldId)
  1256. return;
  1257. jobject fieldObj = (jobject) JNIenv->GetObjectField(parentObject, fieldId);
  1258. if (!fieldObj)
  1259. return;
  1260. jstring fieldStr = (jstring) JNIenv->CallObjectMethod(fieldObj, objToString);
  1261. const char *text = JNIenv->GetStringUTFChars(fieldStr, NULL);
  1262. if (!text)
  1263. return;
  1264. writer.outputCString(text, defObject.queryName());
  1265. }
  1266. void writeComplexType(jclass parentClass, jobject parentObject, IEsdlDefObject &defObject)
  1267. {
  1268. IEsdlDefStruct *defStruct = esdl.queryStruct(defObject.queryProp("complex_type"));
  1269. if (!defStruct)
  1270. return;
  1271. const char *fieldname = defObject.queryName();
  1272. VStringBuffer javaSig("L%s/%s;", esdlService.str(), defObject.queryProp("complex_type"));
  1273. jfieldID fieldId = JNIenv->GetFieldID(parentClass, fieldname, javaSig); //tbd cache this
  1274. if (!fieldId)
  1275. return;
  1276. jobject fieldObj = (jobject) JNIenv->GetObjectField(parentObject, fieldId);
  1277. if (!fieldObj)
  1278. return;
  1279. writer.outputBeginNested(fieldname, true);
  1280. writeChildren(JNIenv->GetObjectClass(fieldObj), fieldObj, defStruct);
  1281. writer.outputEndNested(fieldname);
  1282. }
  1283. void writeSimpleArray(jobjectArray arrayObj, jint count, const char *name, const char *item_tag)
  1284. {
  1285. writer.outputBeginNested(name, true);
  1286. writer.outputBeginArray(item_tag);
  1287. for (jint i=0; i < count; i++)
  1288. {
  1289. jobject elementObj = JNIenv->GetObjectArrayElement(arrayObj, i);
  1290. if(JNIenv->ExceptionOccurred())
  1291. break;
  1292. writeSimpleType(item_tag, elementObj);
  1293. JNIenv->DeleteLocalRef(elementObj);
  1294. }
  1295. writer.outputEndArray(item_tag);
  1296. writer.outputEndNested(name);
  1297. }
  1298. void writeComplexArray(jobjectArray arrayObj, jint count, const char *name, const char *item_tag, const char *itemTypeName)
  1299. {
  1300. writer.outputBeginNested(name, true);
  1301. writer.outputBeginArray(item_tag);
  1302. {
  1303. VStringBuffer javaClassName("%s/%s", esdlService.str(), itemTypeName);
  1304. jclass elementClass = FindClass(javaClassName);
  1305. if (!elementClass)
  1306. return;
  1307. IEsdlDefStruct *defStruct = esdl.queryStruct(itemTypeName);
  1308. if (!defStruct)
  1309. return;
  1310. for (jint i=0; i < count; i++)
  1311. {
  1312. jobject elementObj = JNIenv->GetObjectArrayElement(arrayObj, i);
  1313. javaembed::checkException(JNIenv, false);
  1314. writer.outputBeginNested(item_tag, true);
  1315. writeChildren(elementClass, elementObj, defStruct);
  1316. writer.outputEndNested(item_tag);
  1317. JNIenv->DeleteLocalRef(elementObj);
  1318. }
  1319. }
  1320. writer.outputEndArray(item_tag);
  1321. writer.outputEndNested(name);
  1322. }
  1323. void writeArray(jclass parentClass, jobject parentObject, IEsdlDefObject &defObject)
  1324. {
  1325. const char *itemTypeName = defObject.queryProp("type");
  1326. if (!itemTypeName)
  1327. return;
  1328. const char *item_tag = defObject.queryProp("item_tag");
  1329. if (!item_tag)
  1330. return;
  1331. const char *fieldname = defObject.queryName();
  1332. jclass arrayListClass = FindClass("java/util/ArrayList");
  1333. if (!arrayListClass)
  1334. return;
  1335. jmethodID toArrayMethod = JNIenv->GetMethodID(arrayListClass, "toArray", "()[Ljava/lang/Object;" );
  1336. if (!toArrayMethod)
  1337. return;
  1338. jfieldID fieldId = JNIenv->GetFieldID(parentClass, fieldname, "Ljava/util/ArrayList;");
  1339. if (!fieldId)
  1340. return;
  1341. jobject arrayListObj = (jobject) JNIenv->GetObjectField(parentObject, fieldId);
  1342. if (!arrayListObj)
  1343. return;
  1344. javaembed::checkException(JNIenv, false);
  1345. jobjectArray arrayObj = (jobjectArray) JNIenv->CallObjectMethod(arrayListObj, toArrayMethod);
  1346. if (arrayObj)
  1347. {
  1348. jint count = JNIenv->GetArrayLength(arrayObj);
  1349. if (count)
  1350. {
  1351. if (esdl2JavaSig(esdl, itemTypeName))
  1352. writeSimpleArray(arrayObj, count, defObject.queryName(), item_tag);
  1353. else
  1354. writeComplexArray(arrayObj, count, defObject.queryName(), item_tag, itemTypeName);
  1355. }
  1356. JNIenv->DeleteLocalRef(arrayObj);
  1357. }
  1358. JNIenv->DeleteLocalRef(arrayListObj);
  1359. }
  1360. void writeChildren(jclass javaClass, jobject javaObject, IEsdlDefStruct *defStruct)
  1361. {
  1362. Owned<IEsdlDefObjectIterator> children = defStruct->getChildren();
  1363. ForEach (*children)
  1364. {
  1365. IEsdlDefObject &child = children->query();
  1366. if (child.getEsdlType()==EsdlTypeElement)
  1367. {
  1368. if (child.hasProp("type"))
  1369. writeSimpleType(javaClass, javaObject, child);
  1370. else if (child.hasProp("complex_type"))
  1371. writeComplexType(javaClass, javaObject, child);
  1372. }
  1373. else if (child.getEsdlType()==EsdlTypeEnumRef)
  1374. {
  1375. writeEnumType(javaClass, javaObject, child);
  1376. }
  1377. else if (child.getEsdlType()==EsdlTypeArray)
  1378. {
  1379. writeArray(javaClass, javaObject, child);
  1380. }
  1381. }
  1382. }
  1383. void write()
  1384. {
  1385. IEsdlDefStruct *reqStruct = esdl.queryStruct(reqType);
  1386. const char *name = reqStruct->queryName();
  1387. writer.outputBeginNested("Response", true);
  1388. writer.outputBeginNested("Results", true);
  1389. writer.outputBeginNested("Result", true);
  1390. writer.outputBeginDataset(name, true);
  1391. writer.outputBeginArray("Row");
  1392. writer.outputBeginNested("Row", true);
  1393. writeChildren(Class, obj, reqStruct);
  1394. writer.outputEndNested("Row");
  1395. writer.outputEndArray("Row");
  1396. writer.outputEndDataset(name);
  1397. writer.outputEndNested("Result");
  1398. writer.outputEndNested("Results");
  1399. writer.outputEndNested("Response");
  1400. }
  1401. void checkException()
  1402. {
  1403. javaembed::checkException(JNIenv, false);
  1404. }
  1405. jclass FindClass(const char *name)
  1406. {
  1407. jclass *pClass = javaClasses.getValue(name);
  1408. if (pClass)
  1409. return *pClass;
  1410. jclass localClass = JNIenv->FindClass(name);
  1411. if (!localClass)
  1412. return 0;
  1413. jclass Class = (jclass) JNIenv->NewGlobalRef(localClass);
  1414. javaClasses.setValue(name, Class);
  1415. JNIenv->DeleteLocalRef(localClass);
  1416. return Class;
  1417. }
  1418. JNIEnv *JNIenv;
  1419. MapStringTo<jclass> javaClasses;
  1420. jclass Class;
  1421. jobject obj;
  1422. jclass langObjectClass;
  1423. jmethodID objToString;
  1424. IXmlWriter &writer;
  1425. IEsdlDefinition &esdl;
  1426. StringAttr reqType;
  1427. StringAttr esdlService;
  1428. };
  1429. //-------------------------------------------
  1430. // There is a singleton JavaThreadContext per thread. This allows us to
  1431. // ensure that we can make repeated calls to a Java function efficiently.
  1432. class JavaThreadContext
  1433. {
  1434. public:
  1435. JNIEnv *JNIenv; /* receives pointer to native method interface */
  1436. public:
  1437. JavaThreadContext()
  1438. {
  1439. jint res = globalState->javaVM->AttachCurrentThread((void **) &JNIenv, NULL);
  1440. assertex(res >= 0);
  1441. javaClass = NULL;
  1442. javaMethodID = NULL;
  1443. prevClassPath.set("dummy"); // Forces the call below to actually do something...
  1444. setThreadClassLoader("");
  1445. }
  1446. ~JavaThreadContext()
  1447. {
  1448. if (javaClass)
  1449. JNIenv->DeleteGlobalRef(javaClass);
  1450. // According to the Java VM 1.7 docs, "A native thread attached to
  1451. // the VM must call DetachCurrentThread() to detach itself before
  1452. // exiting."
  1453. globalState->javaVM->DetachCurrentThread();
  1454. }
  1455. void checkException()
  1456. {
  1457. if (JNIenv->ExceptionCheck())
  1458. {
  1459. jthrowable exception = JNIenv->ExceptionOccurred();
  1460. JNIenv->ExceptionClear();
  1461. jclass throwableClass = JNIenv->FindClass("java/lang/Throwable");
  1462. jmethodID throwableToString = JNIenv->GetMethodID(throwableClass, "toString", "()Ljava/lang/String;");
  1463. jstring cause = (jstring) JNIenv->CallObjectMethod(exception, throwableToString);
  1464. const char *text = JNIenv->GetStringUTFChars(cause, 0);
  1465. VStringBuffer message("javaembed: In method %s: %s", prevtext.get(), text);
  1466. JNIenv->ReleaseStringUTFChars(cause, text);
  1467. rtlFail(0, message.str());
  1468. }
  1469. }
  1470. bool checkException(StringBuffer &message)
  1471. {
  1472. if (JNIenv->ExceptionCheck())
  1473. {
  1474. jthrowable exception = JNIenv->ExceptionOccurred();
  1475. JNIenv->ExceptionClear();
  1476. jclass throwableClass = JNIenv->FindClass("java/lang/Throwable");
  1477. jmethodID throwableToString = JNIenv->GetMethodID(throwableClass, "toString", "()Ljava/lang/String;");
  1478. jstring cause = (jstring) JNIenv->CallObjectMethod(exception, throwableToString);
  1479. const char *text = JNIenv->GetStringUTFChars(cause, 0);
  1480. message.append(text);
  1481. JNIenv->ReleaseStringUTFChars(cause, text);
  1482. return true;
  1483. }
  1484. return false;
  1485. }
  1486. jobject getSystemClassLoader()
  1487. {
  1488. JNIenv->ExceptionClear();
  1489. jclass javaLangClassLoaderClass = JNIenv->FindClass("java/lang/ClassLoader");
  1490. checkException();
  1491. jmethodID getSystemClassLoaderMethod = JNIenv->GetStaticMethodID(javaLangClassLoaderClass, "getSystemClassLoader", "()Ljava/lang/ClassLoader;");
  1492. checkException();
  1493. jobject systemClassLoaderObj = JNIenv->CallStaticObjectMethod(javaLangClassLoaderClass, getSystemClassLoaderMethod);
  1494. checkException();
  1495. assertex(systemClassLoaderObj);
  1496. return systemClassLoaderObj;
  1497. }
  1498. void setThreadClassLoader(jobject classLoader)
  1499. {
  1500. JNIenv->ExceptionClear();
  1501. jclass javaLangThreadClass = JNIenv->FindClass("java/lang/Thread");
  1502. checkException();
  1503. jmethodID currentThreadMethod = JNIenv->GetStaticMethodID(javaLangThreadClass, "currentThread", "()Ljava/lang/Thread;");
  1504. checkException();
  1505. jobject threadObj = JNIenv->CallStaticObjectMethod(javaLangThreadClass, currentThreadMethod);
  1506. checkException();
  1507. jmethodID setContextClassLoaderMethod = JNIenv->GetMethodID(javaLangThreadClass, "setContextClassLoader", "(Ljava/lang/ClassLoader;)V");
  1508. checkException();
  1509. JNIenv->CallObjectMethod(threadObj, setContextClassLoaderMethod, classLoader);
  1510. checkException();
  1511. }
  1512. jobject getThreadClassLoader()
  1513. {
  1514. JNIenv->ExceptionClear();
  1515. jclass javaLangThreadClass = JNIenv->FindClass("java/lang/Thread");
  1516. checkException();
  1517. jmethodID currentThreadMethod = JNIenv->GetStaticMethodID(javaLangThreadClass, "currentThread", "()Ljava/lang/Thread;");
  1518. checkException();
  1519. jobject threadObj = JNIenv->CallStaticObjectMethod(javaLangThreadClass, currentThreadMethod);
  1520. checkException();
  1521. jmethodID getContextClassLoaderMethod = JNIenv->GetMethodID(javaLangThreadClass, "getContextClassLoader", "()Ljava/lang/ClassLoader;");
  1522. checkException();
  1523. jobject contextClassLoaderObj = JNIenv->CallObjectMethod(threadObj, getContextClassLoaderMethod);
  1524. checkException();
  1525. assertex(contextClassLoaderObj);
  1526. return contextClassLoaderObj;
  1527. }
  1528. void setThreadClassLoader(const char *classPath)
  1529. {
  1530. if (classPath && *classPath)
  1531. {
  1532. if (prevClassPath && strcmp(classPath, prevClassPath) == 0)
  1533. return;
  1534. jclass URLcls = JNIenv->FindClass("java/net/URL");
  1535. checkException();
  1536. jmethodID URLclsMid = JNIenv->GetMethodID(URLcls, "<init>","(Ljava/lang/String;)V");
  1537. checkException();
  1538. StringArray paths;
  1539. paths.appendList(classPath, ";"); // NOTE - as we need to be able to include : in the urls, we can't use ENVSEP here
  1540. jobjectArray URLArray = JNIenv->NewObjectArray(paths.length(), URLcls, NULL);
  1541. ForEachItemIn(idx, paths)
  1542. {
  1543. StringBuffer usepath;
  1544. const char *path = paths.item(idx);
  1545. if (!strchr(path, ':'))
  1546. usepath.append("file:");
  1547. usepath.append(path);
  1548. jstring jstr = JNIenv->NewStringUTF(usepath.str());
  1549. checkException();
  1550. jobject URLobj = JNIenv->NewObject(URLcls, URLclsMid, jstr);
  1551. checkException();
  1552. JNIenv->SetObjectArrayElement(URLArray, idx, URLobj);
  1553. JNIenv->DeleteLocalRef(URLobj);
  1554. JNIenv->DeleteLocalRef(jstr);
  1555. }
  1556. checkException();
  1557. jclass customLoaderClass = JNIenv->FindClass("java/net/URLClassLoader");
  1558. checkException();
  1559. jmethodID newInstance = JNIenv->GetStaticMethodID(customLoaderClass, "newInstance","([Ljava/net/URL;Ljava/lang/ClassLoader;)Ljava/net/URLClassLoader;");
  1560. checkException();
  1561. jobject contextClassLoaderObj = JNIenv->NewGlobalRef(JNIenv->CallStaticObjectMethod(customLoaderClass, newInstance, URLArray, getSystemClassLoader()));
  1562. checkException();
  1563. assertex(contextClassLoaderObj);
  1564. setThreadClassLoader(contextClassLoaderObj);
  1565. prevClassPath.set(classPath);
  1566. }
  1567. else
  1568. {
  1569. if (prevClassPath)
  1570. setThreadClassLoader(getSystemClassLoader());
  1571. prevClassPath.clear();
  1572. }
  1573. }
  1574. inline void importFunction(size32_t lenChars, const char *utf, const char *options, jobject instance)
  1575. {
  1576. size32_t bytes = rtlUtf8Size(lenChars, utf);
  1577. StringBuffer text(bytes, utf);
  1578. setThreadClassLoader(options);
  1579. if (!prevtext || strcmp(text, prevtext) != 0)
  1580. {
  1581. prevtext.clear();
  1582. // Name should be in the form class.method:signature
  1583. const char *funcname = strrchr(text, '.');
  1584. if (!funcname)
  1585. throw MakeStringException(MSGAUD_user, 0, "javaembed: Invalid import name %s - Expected classname.methodname:signature", text.str());
  1586. const char *signature = strchr(funcname, ':');
  1587. if (!signature)
  1588. throw MakeStringException(MSGAUD_user, 0, "javaembed: Invalid import name %s - Expected classname.methodname:signature", text.str());
  1589. StringBuffer classname(funcname-text, text);
  1590. // While it's probably preferred for people to use . as the separator in nested classes (to match java import statement),
  1591. // we accept / too (to match what you would see in the jar)
  1592. classname.replace('/', '.');
  1593. funcname++; // skip the '.'
  1594. StringBuffer methodname(signature-funcname, funcname);
  1595. signature++; // skip the ':'
  1596. // We need to patch up the provided signature - any instances of <classname; need to be replaced by Ljava.utils.iterator
  1597. StringBuffer javaSignature;
  1598. const char *finger = signature;
  1599. while (*finger)
  1600. {
  1601. if (*finger == '<')
  1602. {
  1603. // If there is a corresponding >, assume it's the 'extended' form and just strip out the bit from < to >
  1604. const char *close = strchr(finger, '>');
  1605. if (close)
  1606. finger = close;
  1607. else
  1608. {
  1609. javaSignature.append("Ljava/util/Iterator;");
  1610. finger = strchr(finger, ';');
  1611. if (!finger)
  1612. throw MakeStringException(MSGAUD_user, 0, "javaembed: Invalid java function signature %s", signature);
  1613. }
  1614. }
  1615. else
  1616. javaSignature.append(*finger);
  1617. finger++;
  1618. }
  1619. if (javaClass)
  1620. JNIenv->DeleteGlobalRef(javaClass);
  1621. jobject classLoader = getThreadClassLoader();
  1622. jmethodID loadClassMethod = JNIenv->GetMethodID(JNIenv->GetObjectClass(classLoader), "loadClass","(Ljava/lang/String;)Ljava/lang/Class;");
  1623. jstring methodString = JNIenv->NewStringUTF(classname);
  1624. javaClass = (jclass) JNIenv->CallObjectMethod(classLoader, loadClassMethod, methodString);
  1625. StringBuffer message;
  1626. if (checkException(message) || !javaClass)
  1627. throw MakeStringException(MSGAUD_user, 0, "javaembed: Failed to resolve class name %s: %s", classname.str(), message.str());
  1628. javaClass = (jclass) JNIenv->NewGlobalRef(javaClass);
  1629. if (instance)
  1630. javaMethodID = JNIenv->GetMethodID(javaClass, methodname, javaSignature);
  1631. else
  1632. javaMethodID = JNIenv->GetStaticMethodID(javaClass, methodname, javaSignature);
  1633. if (checkException(message) || !javaMethodID)
  1634. throw MakeStringException(MSGAUD_user, 0, "javaembed: Failed to resolve method name %s with signature %s: %s", methodname.str(), signature, message.str());
  1635. const char *returnSig = strrchr(signature, ')');
  1636. assertex(returnSig); // Otherwise how did Java accept it??
  1637. returnSig++;
  1638. returnType.set(returnSig);
  1639. argsig.set(signature);
  1640. prevtext.set(text);
  1641. }
  1642. }
  1643. inline void callFunction(jvalue &result, const jvalue * args, jobject instance)
  1644. {
  1645. JNIenv->ExceptionClear();
  1646. switch (returnType.get()[0])
  1647. {
  1648. case 'C': result.c = JNIenv->CallCharMethodA(instance, javaMethodID, args); break;
  1649. case 'Z': result.z = JNIenv->CallBooleanMethodA(instance, javaMethodID, args); break;
  1650. case 'J': result.j = JNIenv->CallLongMethodA(instance, javaMethodID, args); break;
  1651. case 'F': result.f = JNIenv->CallFloatMethodA(instance, javaMethodID, args); break;
  1652. case 'D': result.d = JNIenv->CallDoubleMethodA(instance, javaMethodID, args); break;
  1653. case 'I': result.i = JNIenv->CallIntMethodA(instance, javaMethodID, args); break;
  1654. case 'S': result.s = JNIenv->CallShortMethodA(instance, javaMethodID, args); break;
  1655. case 'B': result.s = JNIenv->CallByteMethodA(instance, javaMethodID, args); break;
  1656. case '[':
  1657. case 'L': result.l = JNIenv->CallObjectMethodA(instance, javaMethodID, args); break;
  1658. default: throwUnexpected();
  1659. }
  1660. checkException();
  1661. }
  1662. inline void callFunction(jvalue &result, const jvalue * args)
  1663. {
  1664. JNIenv->ExceptionClear();
  1665. switch (returnType.get()[0])
  1666. {
  1667. case 'C': result.c = JNIenv->CallStaticCharMethodA(javaClass, javaMethodID, args); break;
  1668. case 'Z': result.z = JNIenv->CallStaticBooleanMethodA(javaClass, javaMethodID, args); break;
  1669. case 'J': result.j = JNIenv->CallStaticLongMethodA(javaClass, javaMethodID, args); break;
  1670. case 'F': result.f = JNIenv->CallStaticFloatMethodA(javaClass, javaMethodID, args); break;
  1671. case 'D': result.d = JNIenv->CallStaticDoubleMethodA(javaClass, javaMethodID, args); break;
  1672. case 'I': result.i = JNIenv->CallStaticIntMethodA(javaClass, javaMethodID, args); break;
  1673. case 'S': result.s = JNIenv->CallStaticShortMethodA(javaClass, javaMethodID, args); break;
  1674. case 'B': result.s = JNIenv->CallStaticByteMethodA(javaClass, javaMethodID, args); break;
  1675. case '[':
  1676. case 'L': result.l = JNIenv->CallStaticObjectMethodA(javaClass, javaMethodID, args); break;
  1677. default: throwUnexpected();
  1678. }
  1679. checkException();
  1680. }
  1681. inline __int64 getSignedResult(jvalue & result)
  1682. {
  1683. switch (returnType.get()[0])
  1684. {
  1685. case 'B': return result.b;
  1686. case 'S': return result.s;
  1687. case 'I': return result.i;
  1688. case 'J': return result.j;
  1689. case 'L':
  1690. {
  1691. // Result should be of class 'Number'
  1692. if (!result.l)
  1693. return 0;
  1694. jmethodID getVal = JNIenv->GetMethodID(JNIenv->GetObjectClass(result.l), "longValue", "()J");
  1695. if (!getVal)
  1696. throw MakeStringException(MSGAUD_user, 0, "javaembed: Type mismatch on result");
  1697. return JNIenv->CallLongMethod(result.l, getVal);
  1698. }
  1699. default:
  1700. throw MakeStringException(MSGAUD_user, 0, "javaembed: Type mismatch on result");
  1701. }
  1702. }
  1703. inline double getDoubleResult(jvalue &result)
  1704. {
  1705. switch (returnType.get()[0])
  1706. {
  1707. case 'D': return result.d;
  1708. case 'F': return result.f;
  1709. case 'L':
  1710. {
  1711. // Result should be of class 'Number'
  1712. if (!result.l)
  1713. return 0;
  1714. jmethodID getVal = JNIenv->GetMethodID(JNIenv->GetObjectClass(result.l), "doubleValue", "()D");
  1715. if (!getVal)
  1716. throw MakeStringException(MSGAUD_user, 0, "javaembed: Type mismatch on result");
  1717. return JNIenv->CallDoubleMethod(result.l, getVal);
  1718. }
  1719. default:
  1720. throw MakeStringException(MSGAUD_user, 0, "javaembed: Type mismatch on result");
  1721. }
  1722. }
  1723. bool getBooleanResult(jvalue &result)
  1724. {
  1725. switch (returnType.get()[0])
  1726. {
  1727. case 'Z': return result.z;
  1728. case 'L':
  1729. {
  1730. // Result should be of class 'Boolean'
  1731. if (!result.l)
  1732. return false;
  1733. jmethodID getVal = JNIenv->GetMethodID(JNIenv->GetObjectClass(result.l), "booleanValue", "()Z");
  1734. if (!getVal)
  1735. throw MakeStringException(MSGAUD_user, 0, "javaembed: Type mismatch on result");
  1736. return JNIenv->CallBooleanMethod(result.l, getVal);
  1737. }
  1738. default:
  1739. throw MakeStringException(MSGAUD_user, 0, "javaembed: Type mismatch on result");
  1740. }
  1741. }
  1742. inline void getDataResult(jvalue &result, size32_t &__len, void * &__result)
  1743. {
  1744. if (strcmp(returnType, "[B")!=0)
  1745. throw MakeStringException(MSGAUD_user, 0, "javaembed: Type mismatch on result");
  1746. jbyteArray array = (jbyteArray) result.l;
  1747. __len = (array != NULL ? JNIenv->GetArrayLength(array) : 0);
  1748. __result = (__len > 0 ? rtlMalloc(__len) : NULL);
  1749. if (__result)
  1750. JNIenv->GetByteArrayRegion(array, 0, __len, (jbyte *) __result);
  1751. }
  1752. inline void getStringResult(jvalue &result, size32_t &__len, char * &__result)
  1753. {
  1754. switch (returnType.get()[0])
  1755. {
  1756. case 'C': // Single char returned, prototyped as STRING or STRING1 in ECL
  1757. rtlUnicodeToStrX(__len, __result, 1, &result.c);
  1758. break;
  1759. case 'L':
  1760. {
  1761. jstring sresult = (jstring) result.l;
  1762. if (sresult)
  1763. {
  1764. size_t size = JNIenv->GetStringUTFLength(sresult); // in bytes
  1765. const char *text = JNIenv->GetStringUTFChars(sresult, NULL);
  1766. size32_t chars = rtlUtf8Length(size, text);
  1767. rtlUtf8ToStrX(__len, __result, chars, text);
  1768. JNIenv->ReleaseStringUTFChars(sresult, text);
  1769. }
  1770. else
  1771. {
  1772. __len = 0;
  1773. __result = NULL;
  1774. }
  1775. break;
  1776. }
  1777. default:
  1778. throw MakeStringException(MSGAUD_user, 0, "javaembed: Type mismatch on result");
  1779. }
  1780. }
  1781. inline void getUTF8Result(jvalue &result, size32_t &__chars, char * &__result)
  1782. {
  1783. switch (returnType.get()[0])
  1784. {
  1785. case 'C': // Single jchar returned, prototyped as UTF8 in ECL
  1786. rtlUnicodeToUtf8X(__chars, __result, 1, &result.c);
  1787. break;
  1788. case 'L':
  1789. {
  1790. jstring sresult = (jstring) result.l;
  1791. if (sresult)
  1792. {
  1793. size_t size = JNIenv->GetStringUTFLength(sresult); // Returns length in bytes (not chars)
  1794. const char * text = JNIenv->GetStringUTFChars(sresult, NULL);
  1795. rtlUtf8ToUtf8X(__chars, __result, rtlUtf8Length(size, text), text);
  1796. JNIenv->ReleaseStringUTFChars(sresult, text);
  1797. }
  1798. else
  1799. {
  1800. __chars = 0;
  1801. __result = NULL;
  1802. }
  1803. break;
  1804. }
  1805. default:
  1806. throw MakeStringException(MSGAUD_user, 0, "javaembed: Type mismatch on result");
  1807. }
  1808. }
  1809. inline void getUnicodeResult(jvalue &result, size32_t &__chars, UChar * &__result)
  1810. {
  1811. switch (returnType.get()[0])
  1812. {
  1813. case 'C': // Single jchar returned, prototyped as UNICODE or UNICODE1 in ECL
  1814. rtlUnicodeToUnicodeX(__chars, __result, 1, &result.c);
  1815. break;
  1816. case 'L':
  1817. {
  1818. jstring sresult = (jstring) result.l;
  1819. if (sresult)
  1820. {
  1821. size_t size = JNIenv->GetStringUTFLength(sresult); // in bytes
  1822. const char *text = JNIenv->GetStringUTFChars(sresult, NULL);
  1823. size32_t chars = rtlUtf8Length(size, text);
  1824. rtlUtf8ToUnicodeX(__chars, __result, chars, text);
  1825. JNIenv->ReleaseStringUTFChars(sresult, text);
  1826. }
  1827. else
  1828. {
  1829. __chars = 0;
  1830. __result = NULL;
  1831. }
  1832. break;
  1833. }
  1834. default:
  1835. throw MakeStringException(MSGAUD_user, 0, "javaembed: Type mismatch on result");
  1836. }
  1837. }
  1838. inline void getSetResult(jvalue &result, bool & __isAllResult, size32_t & __resultBytes, void * & __result, int _elemType, size32_t elemSize)
  1839. {
  1840. if (returnType.get()[0] != '[')
  1841. throw MakeStringException(MSGAUD_user, 0, "javaembed: Type mismatch on result (array expected)");
  1842. type_t elemType = (type_t) _elemType;
  1843. jarray array = (jarray) result.l;
  1844. int numResults = (array != NULL ? JNIenv->GetArrayLength(array) : 0);
  1845. rtlRowBuilder out;
  1846. byte *outData = NULL;
  1847. size32_t outBytes = 0;
  1848. if (numResults > 0)
  1849. {
  1850. if (elemSize != UNKNOWN_LENGTH)
  1851. {
  1852. out.ensureAvailable(numResults * elemSize); // MORE - check for overflow?
  1853. outData = out.getbytes();
  1854. }
  1855. switch(returnType.get()[1])
  1856. {
  1857. case 'Z':
  1858. checkType(type_boolean, sizeof(jboolean), elemType, elemSize);
  1859. JNIenv->GetBooleanArrayRegion((jbooleanArray) array, 0, numResults, (jboolean *) outData);
  1860. break;
  1861. case 'B':
  1862. checkType(type_int, sizeof(jbyte), elemType, elemSize);
  1863. JNIenv->GetByteArrayRegion((jbyteArray) array, 0, numResults, (jbyte *) outData);
  1864. break;
  1865. case 'C':
  1866. // we COULD map to a set of string1, but is there any point?
  1867. throw MakeStringException(0, "javaembed: Return type mismatch (char[] not supported)");
  1868. break;
  1869. case 'S':
  1870. checkType(type_int, sizeof(jshort), elemType, elemSize);
  1871. JNIenv->GetShortArrayRegion((jshortArray) array, 0, numResults, (jshort *) outData);
  1872. break;
  1873. case 'I':
  1874. checkType(type_int, sizeof(jint), elemType, elemSize);
  1875. JNIenv->GetIntArrayRegion((jintArray) array, 0, numResults, (jint *) outData);
  1876. break;
  1877. case 'J':
  1878. checkType(type_int, sizeof(jlong), elemType, elemSize);
  1879. JNIenv->GetLongArrayRegion((jlongArray) array, 0, numResults, (jlong *) outData);
  1880. break;
  1881. case 'F':
  1882. checkType(type_real, sizeof(jfloat), elemType, elemSize);
  1883. JNIenv->GetFloatArrayRegion((jfloatArray) array, 0, numResults, (jfloat *) outData);
  1884. break;
  1885. case 'D':
  1886. checkType(type_real, sizeof(jdouble), elemType, elemSize);
  1887. JNIenv->GetDoubleArrayRegion((jdoubleArray) array, 0, numResults, (jdouble *) outData);
  1888. break;
  1889. case 'L':
  1890. if (strcmp(returnType, "[Ljava/lang/String;") == 0)
  1891. {
  1892. for (int i = 0; i < numResults; i++)
  1893. {
  1894. jstring elem = (jstring) JNIenv->GetObjectArrayElement((jobjectArray) array, i);
  1895. size_t lenBytes = JNIenv->GetStringUTFLength(elem); // in bytes
  1896. const char *text = JNIenv->GetStringUTFChars(elem, NULL);
  1897. switch (elemType)
  1898. {
  1899. case type_string:
  1900. if (elemSize == UNKNOWN_LENGTH)
  1901. {
  1902. out.ensureAvailable(outBytes + lenBytes + sizeof(size32_t));
  1903. outData = out.getbytes() + outBytes;
  1904. * (size32_t *) outData = lenBytes;
  1905. rtlStrToStr(lenBytes, outData+sizeof(size32_t), lenBytes, text);
  1906. outBytes += lenBytes + sizeof(size32_t);
  1907. }
  1908. else
  1909. rtlStrToStr(elemSize, outData, lenBytes, text);
  1910. break;
  1911. case type_varstring:
  1912. if (elemSize == UNKNOWN_LENGTH)
  1913. {
  1914. out.ensureAvailable(outBytes + lenBytes + 1);
  1915. outData = out.getbytes() + outBytes;
  1916. rtlStrToVStr(0, outData, lenBytes, text);
  1917. outBytes += lenBytes + 1;
  1918. }
  1919. else
  1920. rtlStrToVStr(elemSize, outData, lenBytes, text); // Fixed size null terminated strings... weird.
  1921. break;
  1922. case type_utf8:
  1923. case type_unicode:
  1924. {
  1925. size32_t numchars = rtlUtf8Length(lenBytes, text);
  1926. if (elemType == type_utf8)
  1927. {
  1928. assertex (elemSize == UNKNOWN_LENGTH);
  1929. out.ensureAvailable(outBytes + lenBytes + sizeof(size32_t));
  1930. outData = out.getbytes() + outBytes;
  1931. * (size32_t *) outData = numchars;
  1932. rtlStrToStr(lenBytes, outData+sizeof(size32_t), lenBytes, text);
  1933. outBytes += lenBytes + sizeof(size32_t);
  1934. }
  1935. else
  1936. {
  1937. if (elemSize == UNKNOWN_LENGTH)
  1938. {
  1939. // You can't assume that number of chars in utf8 matches number in unicode16 ...
  1940. size32_t numchars16;
  1941. rtlDataAttr unicode16;
  1942. rtlUtf8ToUnicodeX(numchars16, unicode16.refustr(), numchars, text);
  1943. out.ensureAvailable(outBytes + numchars16*sizeof(UChar) + sizeof(size32_t));
  1944. outData = out.getbytes() + outBytes;
  1945. * (size32_t *) outData = numchars16;
  1946. rtlUnicodeToUnicode(numchars16, (UChar *) (outData+sizeof(size32_t)), numchars16, unicode16.getustr());
  1947. outBytes += numchars16*sizeof(UChar) + sizeof(size32_t);
  1948. }
  1949. else
  1950. rtlUtf8ToUnicode(elemSize / sizeof(UChar), (UChar *) outData, numchars, text);
  1951. }
  1952. break;
  1953. }
  1954. default:
  1955. JNIenv->ReleaseStringUTFChars(elem, text);
  1956. throw MakeStringException(0, "javaembed: Return type mismatch (ECL string type expected)");
  1957. }
  1958. JNIenv->ReleaseStringUTFChars(elem, text);
  1959. JNIenv->DeleteLocalRef(elem);
  1960. if (elemSize != UNKNOWN_LENGTH)
  1961. outData += elemSize;
  1962. }
  1963. }
  1964. else
  1965. throw MakeStringException(0, "javaembed: Return type mismatch (%s[] not supported)", returnType.get()+2);
  1966. break;
  1967. }
  1968. }
  1969. __isAllResult = false;
  1970. __resultBytes = elemSize == UNKNOWN_LENGTH ? outBytes : elemSize * numResults;
  1971. __result = out.detachdata();
  1972. }
  1973. inline const char *querySignature()
  1974. {
  1975. return argsig.get();
  1976. }
  1977. size32_t getRowResult(jobject result, ARowBuilder &builder)
  1978. {
  1979. const RtlTypeInfo *typeInfo = builder.queryAllocator()->queryOutputMeta()->queryTypeInfo();
  1980. assertex(typeInfo);
  1981. RtlFieldStrInfo dummyField("<row>", NULL, typeInfo);
  1982. JavaRowBuilder javaRowBuilder(JNIenv, &dummyField, result);
  1983. return typeInfo->build(builder, 0, &dummyField, javaRowBuilder);
  1984. }
  1985. void writeObjectResult(jobject result, IEsdlDefinition *esdl, const char *esdlservice, const char *name, IXmlWriter *writer)
  1986. {
  1987. JavaObjectXmlWriter x(JNIenv, result, name, *esdl, esdlservice, *writer);
  1988. x.write();
  1989. }
  1990. private:
  1991. StringAttr returnType;
  1992. StringAttr argsig;
  1993. StringAttr prevtext;
  1994. StringAttr prevClassPath;
  1995. jclass javaClass;
  1996. jmethodID javaMethodID;
  1997. };
  1998. class JavaXmlBuilder : implements IXmlWriterExt, public CInterface
  1999. {
  2000. public:
  2001. IMPLEMENT_IINTERFACE;
  2002. JavaXmlBuilder(JNIEnv *_JNIenv, IEsdlDefinition *esdl_, const char *esdlservice, const char *esdltype_)
  2003. : JNIenv(_JNIenv), esdl(esdl_), javaPackage(esdlservice), esdlType(esdltype_)
  2004. {
  2005. }
  2006. ~JavaXmlBuilder()
  2007. {
  2008. while (defStack.length())
  2009. popDefStackEntry(JNIenv);
  2010. HashIterator it(javaClasses);
  2011. ForEach(it)
  2012. {
  2013. IMapping &entry = it.query();
  2014. jclass *pClass = javaClasses.mapToValue(&entry);
  2015. if (pClass)
  2016. JNIenv->DeleteGlobalRef(*pClass);
  2017. }
  2018. }
  2019. void checkException()
  2020. {
  2021. javaembed::checkException(JNIenv, false);
  2022. }
  2023. void initWriter()
  2024. {
  2025. }
  2026. IXmlWriterExt & clear()
  2027. {
  2028. throwUnexpected();
  2029. }
  2030. virtual size32_t length() const
  2031. {
  2032. return 0;
  2033. }
  2034. virtual const char *str() const
  2035. {
  2036. throwUnexpected();
  2037. }
  2038. virtual IInterface *saveLocation() const {return nullptr;}
  2039. virtual void rewindTo(IInterface *loc)
  2040. {
  2041. //needs to be a no-op because it is used, but the way its used to trim empty xml sections I think we're fairly safe.
  2042. //revisit cleaning up any empty objects later.
  2043. }
  2044. inline IEsdlDefStruct *queryCurrentEsdlStruct()
  2045. {
  2046. if (!defStack.length() || !defStack.tos().defType)
  2047. return NULL;
  2048. return dynamic_cast<IEsdlDefStruct*>(defStack.tos().defType.get());
  2049. }
  2050. inline jobject getObject()
  2051. {
  2052. if (!defStack.length())
  2053. return 0;
  2054. return defStack.item(0).obj;
  2055. }
  2056. inline jobject getCurJavaObject()
  2057. {
  2058. if (!defStack.length())
  2059. return 0;
  2060. return defStack.tos().obj;
  2061. }
  2062. inline jclass getCurJavaClass()
  2063. {
  2064. if (!defStack.length())
  2065. return 0;
  2066. return defStack.tos().Class;
  2067. }
  2068. inline jmethodID getCurJavaConstructor()
  2069. {
  2070. if (!defStack.length())
  2071. return 0;
  2072. return defStack.tos().constructor;
  2073. }
  2074. virtual void outputEnumString(unsigned size, const char *text, const char *fieldname, IEsdlDefObject *defField)
  2075. {
  2076. const char *enum_type = defField->queryProp("enum_type");
  2077. if (!enum_type || !*enum_type)
  2078. return;
  2079. VStringBuffer enumClassName("%s/%s", javaPackage.str(), enum_type);
  2080. VStringBuffer enumSig("L%s;", enumClassName.str());
  2081. jfieldID fieldId = JNIenv->GetFieldID(getCurJavaClass(), fieldname, enumSig);
  2082. if (!fieldId)
  2083. return;
  2084. jclass enumClass = FindClass(enumClassName);
  2085. jmethodID fromString = JNIenv->GetStaticMethodID(enumClass, "fromString", "(Ljava/lang/String;)LEsdlExample/AddressType;"); //All types currently used for ESDL mapping have string constructors
  2086. StringAttr s(text, size);
  2087. jstring strvalue = JNIenv->NewStringUTF(s);
  2088. jobject value = JNIenv->CallStaticObjectMethod(enumClass, fromString, strvalue);
  2089. JNIenv->DeleteLocalRef(strvalue);
  2090. checkException();
  2091. JNIenv->SetObjectField(getCurJavaObject(), fieldId, value);
  2092. JNIenv->DeleteLocalRef(value);
  2093. checkException();
  2094. }
  2095. virtual void outputString(unsigned size, const char *text, const char *fieldname)
  2096. {
  2097. DefStackEntry *parent = defStack.length() ? &defStack.tos() : NULL;
  2098. if (!parent)
  2099. return;
  2100. const char *defTypeName = NULL;
  2101. bool isArray = (parent->defObj && parent->defObj->getEsdlType()==EsdlTypeArray);
  2102. if (isArray)
  2103. defTypeName = parent->defObj->queryProp("type");
  2104. else
  2105. {
  2106. IEsdlDefStruct *defStruct = queryCurrentEsdlStruct();
  2107. if (!defStruct)
  2108. return;
  2109. IEsdlDefObject *defField = defStruct->queryChild(fieldname);
  2110. if (!defField)
  2111. return;
  2112. if (defField->getEsdlType()==EsdlTypeEnumRef)
  2113. return outputEnumString(size, text, fieldname, defField);
  2114. defTypeName = defField->queryProp("type");
  2115. }
  2116. if (!defTypeName)
  2117. return;
  2118. const char *javaSig = esdl2JavaSig(*esdl, defTypeName);
  2119. if (!javaSig)
  2120. return;
  2121. const char *fieldClassName = esdl2JavaFullClassName(*esdl, defTypeName);
  2122. jclass typeClass = FindClass(fieldClassName);
  2123. jmethodID typeStringConstructor = JNIenv->GetMethodID(typeClass, "<init>", "(Ljava/lang/String;)V"); //All types currently used for ESDL mapping have string constructors
  2124. StringAttr s(text, size);
  2125. jstring strvalue = JNIenv->NewStringUTF(s);
  2126. jobject value = JNIenv->NewObject(typeClass, typeStringConstructor, strvalue);
  2127. JNIenv->DeleteLocalRef(strvalue);
  2128. checkException();
  2129. if (!value)
  2130. return;
  2131. if (isArray)
  2132. JNIenv->CallObjectMethod(parent->obj, parent->append, value);
  2133. else
  2134. {
  2135. jfieldID fieldId = JNIenv->GetFieldID(getCurJavaClass(), fieldname, javaSig);
  2136. if (fieldId)
  2137. JNIenv->SetObjectField(getCurJavaObject(), fieldId, value);
  2138. }
  2139. JNIenv->DeleteLocalRef(value);
  2140. checkException();
  2141. }
  2142. void outputString(const char *text, const char *fieldname)
  2143. {
  2144. outputString((unsigned)strlen(text), text, fieldname);
  2145. }
  2146. virtual void outputNumericString(const char *field, const char *fieldname)
  2147. {
  2148. outputString(field, fieldname);
  2149. }
  2150. virtual void outputBool(bool value, const char *fieldname)
  2151. {
  2152. outputString(value ? "true" : "false", fieldname);
  2153. }
  2154. virtual void outputUInt(unsigned __int64 field, unsigned size, const char *fieldname)
  2155. {
  2156. StringBuffer value;
  2157. value.append(field);
  2158. outputString(value.length(), value, fieldname);
  2159. }
  2160. virtual void outputInt(__int64 field, unsigned size, const char *fieldname)
  2161. {
  2162. StringBuffer value;
  2163. value.append(field);
  2164. outputString(value.length(), value, fieldname);
  2165. }
  2166. virtual void outputReal(double field, const char *fieldname)
  2167. {
  2168. StringBuffer value;
  2169. value.append(field);
  2170. outputString(value.length(), value, fieldname);
  2171. }
  2172. virtual void outputDecimal(const void *field, unsigned size, unsigned precision, const char *fieldname)
  2173. {
  2174. Decimal d;
  2175. d.setDecimal(size, precision, field);
  2176. outputString(d.getCString(), fieldname);
  2177. }
  2178. virtual void outputUDecimal(const void *field, unsigned size, unsigned precision, const char *fieldname)
  2179. {
  2180. Decimal d;
  2181. d.setUDecimal(size, precision, field);
  2182. outputString(d.getCString(), fieldname);
  2183. }
  2184. virtual void outputQString(unsigned len, const char *field, const char *fieldname)
  2185. {
  2186. MemoryAttr tempBuffer;
  2187. char * temp;
  2188. if (len <= 100)
  2189. temp = (char *)alloca(len);
  2190. else
  2191. temp = (char *)tempBuffer.allocate(len);
  2192. rtlQStrToStr(len, temp, len, field);
  2193. outputString(len, temp, fieldname);
  2194. }
  2195. virtual void outputUnicode(unsigned len, const UChar *field, const char *fieldname)
  2196. {
  2197. char * buff = 0;
  2198. unsigned bufflen = 0;
  2199. rtlUnicodeToCodepageX(bufflen, buff, len, field, "utf-8");
  2200. outputString(bufflen, buff, fieldname);
  2201. rtlFree(buff);
  2202. }
  2203. virtual void outputUtf8(unsigned len, const char *field, const char *fieldname)
  2204. {
  2205. outputString(len, field, fieldname);
  2206. }
  2207. virtual void outputData(unsigned len, const void *value, const char *fieldname)
  2208. {
  2209. }
  2210. virtual void outputQuoted(const char *text) //would have to let beginNested represent simple types with content set using this?
  2211. {
  2212. }
  2213. virtual void outputBeginDataset(const char *dsname, bool nestChildren) //not used by ESDL engine
  2214. {
  2215. }
  2216. virtual void outputEndDataset(const char *dsname)
  2217. {
  2218. }
  2219. inline IEsdlDefObject *queryChildStructDefObj(IEsdlDefObject *child)
  2220. {
  2221. if (child)
  2222. {
  2223. switch (child->getEsdlType())
  2224. {
  2225. case EsdlTypeArray:
  2226. {
  2227. const char *structType = child->queryProp("type");
  2228. if (structType)
  2229. return esdl->queryObj(structType);
  2230. break;
  2231. }
  2232. case EsdlTypeElement:
  2233. {
  2234. const char *structType = child->queryProp("complex_type");
  2235. if (structType)
  2236. return esdl->queryObj(structType);
  2237. break;
  2238. }
  2239. default:
  2240. break;
  2241. }
  2242. }
  2243. return NULL;
  2244. }
  2245. virtual void outputBeginNested(const char *fieldname, bool nestChildren)
  2246. {
  2247. IEsdlDefStruct *defStruct = NULL;
  2248. IEsdlDefObject *defField = NULL;
  2249. IEsdlDefObject *defType = NULL;
  2250. if (!defStack.length())
  2251. {
  2252. defType = esdl->queryObj(fieldname);
  2253. }
  2254. else
  2255. {
  2256. DefStackEntry &parent = defStack.tos();
  2257. if (parent.defObj && parent.defObj->getEsdlType()==EsdlTypeArray)
  2258. {
  2259. defType = parent.defType;
  2260. }
  2261. else
  2262. {
  2263. defStruct = queryCurrentEsdlStruct();
  2264. if (defStruct)
  2265. {
  2266. defField = defStruct->queryChild(fieldname);
  2267. if (defField)
  2268. defType = queryChildStructDefObj(defField);
  2269. }
  2270. }
  2271. }
  2272. pushDefStackEntry(JNIenv, javaPackage, fieldname, defType, defField);
  2273. }
  2274. virtual void outputEndNested(const char *fieldname)
  2275. {
  2276. if (defStack.length()<=1) //don't destroy root object yet
  2277. return;
  2278. if (!streq(fieldname, defStack.tos().name)) //should be exception? or forgive and forget?
  2279. return;
  2280. popDefStackEntry(JNIenv);
  2281. }
  2282. virtual void outputSetAll()
  2283. {
  2284. }
  2285. virtual void outputBeginArray(const char *fieldname)
  2286. {
  2287. }
  2288. virtual void outputEndArray(const char *fieldname)
  2289. {
  2290. }
  2291. virtual void outputInlineXml(const char *text)
  2292. {
  2293. }
  2294. virtual void outputXmlns(const char *name, const char *uri)
  2295. {
  2296. }
  2297. virtual void cutFrom(IInterface *location, StringBuffer& databuf)
  2298. {
  2299. }
  2300. virtual void outputInline(const char* text)
  2301. {
  2302. }
  2303. public:
  2304. JNIEnv *JNIenv;
  2305. Linked<IEsdlDefinition> esdl;
  2306. StringAttr javaPackage;
  2307. StringAttr esdlType;
  2308. class DefStackEntry : public CInterface
  2309. {
  2310. public:
  2311. DefStackEntry(const char *fieldname, IEsdlDefObject *_defType, IEsdlDefObject *_defObj) : name(fieldname), defType(_defType), defObj(_defObj), Class(0), obj(0), constructor(0), append(0), fieldId(0)
  2312. {
  2313. }
  2314. ~DefStackEntry()
  2315. {
  2316. }
  2317. public:
  2318. Linked<IEsdlDefObject> defType;
  2319. Linked<IEsdlDefObject> defObj;
  2320. StringAttr name;
  2321. jclass Class;
  2322. jmethodID constructor;
  2323. jmethodID append;
  2324. jfieldID fieldId;
  2325. jobject obj;
  2326. };
  2327. jobject MakeObjectGlobal(jobject local)
  2328. {
  2329. if (!local)
  2330. return 0;
  2331. jobject global = JNIenv->NewGlobalRef(local);
  2332. JNIenv->DeleteLocalRef(local);
  2333. return global;
  2334. }
  2335. jclass FindClass(const char *name)
  2336. {
  2337. jclass *pClass = javaClasses.getValue(name);
  2338. if (pClass)
  2339. return *pClass;
  2340. jclass Class = (jclass) MakeObjectGlobal(JNIenv->FindClass(name));
  2341. javaClasses.setValue(name, Class); //even save if result has no class
  2342. return Class;
  2343. }
  2344. void popDefStackEntry(JNIEnv *JNIenv)
  2345. {
  2346. if (!defStack.length())
  2347. return;
  2348. Owned<DefStackEntry> entry = &defStack.popGet();
  2349. if (entry->obj)
  2350. JNIenv->DeleteGlobalRef(entry->obj);
  2351. }
  2352. void pushDefStackEntry(JNIEnv *JNIenv, const char *package, const char *fieldname, IEsdlDefObject *defType, IEsdlDefObject *defObject)
  2353. {
  2354. DefStackEntry *parent = defStack.length() ? &defStack.tos() : NULL;
  2355. Owned<DefStackEntry> entry = new DefStackEntry(fieldname, defType, defObject);
  2356. JNIenv->ExceptionClear();
  2357. if (defObject && defObject->getEsdlType()==EsdlTypeArray)
  2358. {
  2359. const char *javaClassName = "java/util/ArrayList";
  2360. entry->Class = FindClass(javaClassName);
  2361. if (entry->Class)
  2362. {
  2363. entry->constructor = JNIenv->GetMethodID(entry->Class, "<init>", "()V");
  2364. entry->append = JNIenv->GetMethodID(entry->Class, "add", "(Ljava/lang/Object;)Z");
  2365. entry->obj = MakeObjectGlobal(JNIenv->NewObject(entry->Class, entry->constructor));
  2366. javaembed::checkException(JNIenv, false);
  2367. if (entry->obj)
  2368. {
  2369. if (parent && parent->Class)
  2370. {
  2371. VStringBuffer javaSig("L%s;", javaClassName);
  2372. entry->fieldId = JNIenv->GetFieldID(parent->Class, fieldname, javaSig);
  2373. if (parent->obj && entry->fieldId)
  2374. JNIenv->SetObjectField(parent->obj, entry->fieldId, entry->obj);
  2375. }
  2376. }
  2377. }
  2378. }
  2379. else if (defType)
  2380. {
  2381. VStringBuffer javaClassName("%s/%s", package, defType->queryName());
  2382. entry->Class = FindClass(javaClassName);
  2383. if (entry->Class)
  2384. {
  2385. entry->constructor = JNIenv->GetMethodID(entry->Class, "<init>", "()V");
  2386. entry->obj = MakeObjectGlobal(JNIenv->NewObject(entry->Class, entry->constructor));
  2387. javaembed::checkException(JNIenv, false);
  2388. if (entry->obj)
  2389. {
  2390. if (parent)
  2391. {
  2392. if (parent->defObj && parent->defObj->getEsdlType()==EsdlTypeArray)
  2393. JNIenv->CallObjectMethod(parent->obj, parent->append, entry->obj);
  2394. else if (parent->Class)
  2395. {
  2396. VStringBuffer javaSig("L%s;", javaClassName.str());
  2397. entry->fieldId = JNIenv->GetFieldID(parent->Class, fieldname, javaSig);
  2398. if (parent->obj && entry->fieldId)
  2399. JNIenv->SetObjectField(parent->obj, entry->fieldId, entry->obj);
  2400. }
  2401. javaembed::checkException(JNIenv, false);
  2402. }
  2403. }
  2404. }
  2405. }
  2406. defStack.append(*entry.getClear());
  2407. }
  2408. CIArrayOf<DefStackEntry> defStack;
  2409. MapStringTo<jclass> javaClasses;
  2410. };
  2411. // Each call to a Java function will use a new JavaEmbedScriptContext object
  2412. #define MAX_JNI_ARGS 10
  2413. class JavaEmbedImportContext : public CInterfaceOf<IEmbedFunctionContext>
  2414. {
  2415. public:
  2416. JavaEmbedImportContext(JavaThreadContext *_sharedCtx, jobject _instance, const char *options)
  2417. : sharedCtx(_sharedCtx), instance(_instance)
  2418. {
  2419. argcount = 0;
  2420. argsig = NULL;
  2421. StringArray opts;
  2422. opts.appendList(options, ",");
  2423. ForEachItemIn(idx, opts)
  2424. {
  2425. const char *opt = opts.item(idx);
  2426. const char *val = strchr(opt, '=');
  2427. if (val)
  2428. {
  2429. StringBuffer optName(val-opt, opt);
  2430. val++;
  2431. if (stricmp(optName, "classpath")==0)
  2432. classpath.set(val);
  2433. else
  2434. throw MakeStringException(0, "javaembed: Unknown option %s", optName.str());
  2435. }
  2436. }
  2437. // Create a new frame for local references and increase the capacity
  2438. // of those references to 64 (default is 16)
  2439. sharedCtx->JNIenv->PushLocalFrame(64);
  2440. }
  2441. ~JavaEmbedImportContext()
  2442. {
  2443. // Pop local reference frame; explicitly frees all local
  2444. // references made during that frame's lifetime
  2445. sharedCtx->JNIenv->PopLocalFrame(NULL);
  2446. }
  2447. virtual bool getBooleanResult()
  2448. {
  2449. return sharedCtx->getBooleanResult(result);
  2450. }
  2451. virtual void getDataResult(size32_t &__len, void * &__result)
  2452. {
  2453. sharedCtx->getDataResult(result, __len, __result);
  2454. }
  2455. virtual double getRealResult()
  2456. {
  2457. return sharedCtx->getDoubleResult(result);
  2458. }
  2459. virtual __int64 getSignedResult()
  2460. {
  2461. return sharedCtx->getSignedResult(result);
  2462. }
  2463. virtual unsigned __int64 getUnsignedResult()
  2464. {
  2465. throw MakeStringException(MSGAUD_user, 0, "javaembed: Unsigned results not supported"); // Java doesn't support unsigned
  2466. }
  2467. virtual void getStringResult(size32_t &__len, char * &__result)
  2468. {
  2469. sharedCtx->getStringResult(result, __len, __result);
  2470. }
  2471. virtual void getUTF8Result(size32_t &__chars, char * &__result)
  2472. {
  2473. sharedCtx->getUTF8Result(result, __chars, __result);
  2474. }
  2475. virtual void getUnicodeResult(size32_t &__chars, UChar * &__result)
  2476. {
  2477. sharedCtx->getUnicodeResult(result, __chars, __result);
  2478. }
  2479. virtual void getSetResult(bool & __isAllResult, size32_t & __resultBytes, void * & __result, int elemType, size32_t elemSize)
  2480. {
  2481. sharedCtx->getSetResult(result, __isAllResult, __resultBytes, __result, elemType, elemSize);
  2482. }
  2483. virtual IRowStream *getDatasetResult(IEngineRowAllocator * _resultAllocator)
  2484. {
  2485. return new JavaRowStream(result.l, _resultAllocator);
  2486. }
  2487. virtual byte * getRowResult(IEngineRowAllocator * _resultAllocator)
  2488. {
  2489. RtlDynamicRowBuilder rowBuilder(_resultAllocator);
  2490. size32_t len = sharedCtx->getRowResult(result.l, rowBuilder);
  2491. return (byte *) rowBuilder.finalizeRowClear(len);
  2492. }
  2493. virtual size32_t getTransformResult(ARowBuilder & builder)
  2494. {
  2495. return sharedCtx->getRowResult(result.l, builder);
  2496. }
  2497. virtual void bindBooleanParam(const char *name, bool val)
  2498. {
  2499. if (*argsig != 'Z')
  2500. typeError("BOOLEAN");
  2501. argsig++;
  2502. jvalue v;
  2503. v.z = val;
  2504. addArg(v);
  2505. }
  2506. virtual void bindDataParam(const char *name, size32_t len, const void *val)
  2507. {
  2508. if (argsig[0] != '[' || argsig[1] != 'B')
  2509. typeError("DATA");
  2510. argsig += 2;
  2511. jvalue v;
  2512. jbyteArray javaData = sharedCtx->JNIenv->NewByteArray(len);
  2513. sharedCtx->JNIenv->SetByteArrayRegion(javaData, 0, len, (jbyte *) val);
  2514. v.l = javaData;
  2515. addArg(v);
  2516. }
  2517. virtual void bindFloatParam(const char *name, float val)
  2518. {
  2519. // Could argue that the size should match...
  2520. jvalue v;
  2521. switch(*argsig)
  2522. {
  2523. case 'D':
  2524. v.d = val;
  2525. break;
  2526. case 'F':
  2527. v.f = val;
  2528. break;
  2529. default:
  2530. typeError("REAL");
  2531. break;
  2532. }
  2533. argsig++;
  2534. addArg(v);
  2535. }
  2536. virtual void bindRealParam(const char *name, double val)
  2537. {
  2538. jvalue v;
  2539. switch(*argsig)
  2540. {
  2541. case 'D':
  2542. v.d = val;
  2543. break;
  2544. case 'F':
  2545. v.f = val;
  2546. break;
  2547. default:
  2548. typeError("REAL");
  2549. break;
  2550. }
  2551. argsig++;
  2552. addArg(v);
  2553. }
  2554. virtual void bindSignedSizeParam(const char *name, int size, __int64 val)
  2555. {
  2556. bindSignedParam(name, val);
  2557. }
  2558. virtual void bindSignedParam(const char *name, __int64 val)
  2559. {
  2560. jvalue v;
  2561. switch(*argsig)
  2562. {
  2563. case 'I':
  2564. v.i = val;
  2565. break;
  2566. case 'J':
  2567. v.j = val;
  2568. break;
  2569. case 'S':
  2570. v.s = val;
  2571. break;
  2572. case 'B':
  2573. v.b = val;
  2574. break;
  2575. default:
  2576. typeError("INTEGER");
  2577. break;
  2578. }
  2579. argsig++;
  2580. addArg(v);
  2581. }
  2582. virtual void bindUnsignedSizeParam(const char *name, int size, unsigned __int64 val)
  2583. {
  2584. bindUnsignedParam(name, val);
  2585. }
  2586. virtual void bindUnsignedParam(const char *name, unsigned __int64 val)
  2587. {
  2588. throw MakeStringException(MSGAUD_user, 0, "javaembed: Unsigned parameters not supported"); // Java doesn't support unsigned
  2589. }
  2590. virtual void bindStringParam(const char *name, size32_t len, const char *val)
  2591. {
  2592. jvalue v;
  2593. switch(*argsig)
  2594. {
  2595. case 'C':
  2596. rtlStrToUnicode(1, &v.c, len, val);
  2597. argsig++;
  2598. break;
  2599. case 'L':
  2600. if (strncmp(argsig, "Ljava/lang/String;", 18) == 0)
  2601. {
  2602. argsig += 18;
  2603. unsigned unicodeChars;
  2604. UChar *unicode;
  2605. rtlStrToUnicodeX(unicodeChars, unicode, len, val);
  2606. v.l = sharedCtx->JNIenv->NewString(unicode, unicodeChars);
  2607. rtlFree(unicode);
  2608. break;
  2609. }
  2610. // fall into ...
  2611. default:
  2612. typeError("STRING");
  2613. break;
  2614. }
  2615. addArg(v);
  2616. }
  2617. virtual void bindVStringParam(const char *name, const char *val)
  2618. {
  2619. bindStringParam(name, strlen(val), val);
  2620. }
  2621. virtual void bindUTF8Param(const char *name, size32_t numchars, const char *val)
  2622. {
  2623. jvalue v;
  2624. switch(*argsig)
  2625. {
  2626. case 'C':
  2627. rtlUtf8ToUnicode(1, &v.c, numchars, val);
  2628. argsig++;
  2629. break;
  2630. case 'L':
  2631. if (strncmp(argsig, "Ljava/lang/String;", 18) == 0)
  2632. {
  2633. argsig += 18;
  2634. unsigned unicodeChars;
  2635. UChar *unicode;
  2636. rtlUtf8ToUnicodeX(unicodeChars, unicode, numchars, val);
  2637. v.l = sharedCtx->JNIenv->NewString(unicode, unicodeChars);
  2638. rtlFree(unicode);
  2639. break;
  2640. }
  2641. // fall into ...
  2642. default:
  2643. typeError("UTF8");
  2644. break;
  2645. }
  2646. addArg(v);
  2647. }
  2648. virtual void bindUnicodeParam(const char *name, size32_t numchars, const UChar *val)
  2649. {
  2650. jvalue v;
  2651. switch(*argsig)
  2652. {
  2653. case 'C':
  2654. rtlUnicodeToUnicode(1, &v.c, numchars, val);
  2655. argsig++;
  2656. break;
  2657. case 'L':
  2658. if (strncmp(argsig, "Ljava/lang/String;", 18) == 0)
  2659. {
  2660. argsig += 18;
  2661. v.l = sharedCtx->JNIenv->NewString(val, numchars);
  2662. break;
  2663. }
  2664. // fall into ...
  2665. default:
  2666. typeError("UNICODE");
  2667. break;
  2668. }
  2669. addArg(v);
  2670. }
  2671. virtual void bindSetParam(const char *name, int _elemType, size32_t elemSize, bool isAll, size32_t totalBytes, const void *setData)
  2672. {
  2673. jvalue v;
  2674. if (*argsig != '[')
  2675. typeError("SET");
  2676. argsig++;
  2677. type_t elemType = (type_t) _elemType;
  2678. int numElems = totalBytes / elemSize;
  2679. switch(*argsig)
  2680. {
  2681. case 'Z':
  2682. checkType(type_boolean, sizeof(jboolean), elemType, elemSize);
  2683. v.l = sharedCtx->JNIenv->NewBooleanArray(numElems);
  2684. sharedCtx->JNIenv->SetBooleanArrayRegion((jbooleanArray) v.l, 0, numElems, (jboolean *) setData);
  2685. break;
  2686. case 'B':
  2687. checkType(type_int, sizeof(jbyte), elemType, elemSize);
  2688. v.l = sharedCtx->JNIenv->NewByteArray(numElems);
  2689. sharedCtx->JNIenv->SetByteArrayRegion((jbyteArray) v.l, 0, numElems, (jbyte *) setData);
  2690. break;
  2691. case 'C':
  2692. // we COULD map to a set of string1, but is there any point?
  2693. typeError("");
  2694. break;
  2695. case 'S':
  2696. checkType(type_int, sizeof(jshort), elemType, elemSize);
  2697. v.l = sharedCtx->JNIenv->NewShortArray(numElems);
  2698. sharedCtx->JNIenv->SetShortArrayRegion((jshortArray) v.l, 0, numElems, (jshort *) setData);
  2699. break;
  2700. case 'I':
  2701. checkType(type_int, sizeof(jint), elemType, elemSize);
  2702. v.l = sharedCtx->JNIenv->NewIntArray(numElems);
  2703. sharedCtx->JNIenv->SetIntArrayRegion((jintArray) v.l, 0, numElems, (jint *) setData);
  2704. break;
  2705. case 'J':
  2706. checkType(type_int, sizeof(jlong), elemType, elemSize);
  2707. v.l = sharedCtx->JNIenv->NewLongArray(numElems);
  2708. sharedCtx->JNIenv->SetLongArrayRegion((jlongArray) v.l, 0, numElems, (jlong *) setData);
  2709. break;
  2710. case 'F':
  2711. checkType(type_real, sizeof(jfloat), elemType, elemSize);
  2712. v.l = sharedCtx->JNIenv->NewFloatArray(numElems);
  2713. sharedCtx->JNIenv->SetFloatArrayRegion((jfloatArray) v.l, 0, numElems, (jfloat *) setData);
  2714. break;
  2715. case 'D':
  2716. checkType(type_real, sizeof(jdouble), elemType, elemSize);
  2717. v.l = sharedCtx->JNIenv->NewDoubleArray(numElems);
  2718. sharedCtx->JNIenv->SetDoubleArrayRegion((jdoubleArray) v.l, 0, numElems, (jdouble *) setData);
  2719. break;
  2720. case 'L':
  2721. if (strncmp(argsig, "Ljava/lang/String;", 18) == 0)
  2722. {
  2723. argsig += 17; // Yes, 17, because we increment again at the end of the case
  2724. const byte *inData = (const byte *) setData;
  2725. const byte *endData = inData + totalBytes;
  2726. if (elemSize == UNKNOWN_LENGTH)
  2727. {
  2728. numElems = 0;
  2729. // Will need 2 passes to work out how many elements there are in the set :(
  2730. while (inData < endData)
  2731. {
  2732. int thisSize;
  2733. switch (elemType)
  2734. {
  2735. case type_varstring:
  2736. thisSize = strlen((const char *) inData) + 1;
  2737. break;
  2738. case type_string:
  2739. thisSize = * (size32_t *) inData + sizeof(size32_t);
  2740. break;
  2741. case type_unicode:
  2742. thisSize = (* (size32_t *) inData) * sizeof(UChar) + sizeof(size32_t);
  2743. break;
  2744. case type_utf8:
  2745. thisSize = rtlUtf8Size(* (size32_t *) inData, inData + sizeof(size32_t)) + sizeof(size32_t);;
  2746. break;
  2747. default:
  2748. typeError("STRING");
  2749. }
  2750. inData += thisSize;
  2751. numElems++;
  2752. }
  2753. inData = (const byte *) setData;
  2754. }
  2755. int idx = 0;
  2756. v.l = sharedCtx->JNIenv->NewObjectArray(numElems, sharedCtx->JNIenv->FindClass("java/lang/String"), NULL);
  2757. while (inData < endData)
  2758. {
  2759. jstring thisElem;
  2760. size32_t thisSize = elemSize;
  2761. switch (elemType)
  2762. {
  2763. case type_varstring:
  2764. {
  2765. size32_t numChars = strlen((const char *) inData);
  2766. unsigned unicodeChars;
  2767. rtlDataAttr unicode;
  2768. rtlStrToUnicodeX(unicodeChars, unicode.refustr(), numChars, (const char *) inData);
  2769. thisElem = sharedCtx->JNIenv->NewString(unicode.getustr(), unicodeChars);
  2770. if (elemSize == UNKNOWN_LENGTH)
  2771. thisSize = numChars + 1;
  2772. break;
  2773. }
  2774. case type_string:
  2775. {
  2776. if (elemSize == UNKNOWN_LENGTH)
  2777. {
  2778. thisSize = * (size32_t *) inData;
  2779. inData += sizeof(size32_t);
  2780. }
  2781. unsigned unicodeChars;
  2782. rtlDataAttr unicode;
  2783. rtlStrToUnicodeX(unicodeChars, unicode.refustr(), thisSize, (const char *) inData);
  2784. thisElem = sharedCtx->JNIenv->NewString(unicode.getustr(), unicodeChars);
  2785. break;
  2786. }
  2787. case type_unicode:
  2788. {
  2789. if (elemSize == UNKNOWN_LENGTH)
  2790. {
  2791. thisSize = (* (size32_t *) inData) * sizeof(UChar); // NOTE - it's in chars...
  2792. inData += sizeof(size32_t);
  2793. }
  2794. thisElem = sharedCtx->JNIenv->NewString((const UChar *) inData, thisSize / sizeof(UChar));
  2795. //checkJPythonError();
  2796. break;
  2797. }
  2798. case type_utf8:
  2799. {
  2800. assertex (elemSize == UNKNOWN_LENGTH);
  2801. size32_t numChars = * (size32_t *) inData;
  2802. inData += sizeof(size32_t);
  2803. unsigned unicodeChars;
  2804. rtlDataAttr unicode;
  2805. rtlUtf8ToUnicodeX(unicodeChars, unicode.refustr(), numChars, (const char *) inData);
  2806. thisElem = sharedCtx->JNIenv->NewString(unicode.getustr(), unicodeChars);
  2807. thisSize = rtlUtf8Size(numChars, inData);
  2808. break;
  2809. }
  2810. default:
  2811. typeError("STRING");
  2812. }
  2813. sharedCtx->checkException();
  2814. inData += thisSize;
  2815. sharedCtx->JNIenv->SetObjectArrayElement((jobjectArray) v.l, idx, thisElem);
  2816. sharedCtx->JNIenv->DeleteLocalRef(thisElem);
  2817. idx++;
  2818. }
  2819. }
  2820. else
  2821. typeError("");
  2822. break;
  2823. default:
  2824. throwUnexpected();
  2825. }
  2826. argsig++;
  2827. addArg(v);
  2828. }
  2829. virtual void bindRowParam(const char *name, IOutputMetaData & metaVal, const byte *val) override
  2830. {
  2831. if (*argsig != 'L') // should tell us the type of the object we need to create to pass in
  2832. typeError("RECORD");
  2833. // Class name is from the char after the L up to the first ;
  2834. const char *tail = strchr(argsig, ';');
  2835. if (!tail)
  2836. typeError("RECORD");
  2837. StringAttr className(argsig+1, tail - (argsig+1));
  2838. argsig = tail+1;
  2839. const RtlTypeInfo *typeInfo = metaVal.queryTypeInfo();
  2840. assertex(typeInfo);
  2841. RtlFieldStrInfo dummyField("<row>", NULL, typeInfo);
  2842. JavaObjectBuilder javaBuilder(sharedCtx->JNIenv, &dummyField, className);
  2843. typeInfo->process(val, val, &dummyField, javaBuilder); // Creates a java object from the incoming ECL row
  2844. jvalue v;
  2845. v.l = javaBuilder.getObject();
  2846. addArg(v);
  2847. }
  2848. virtual IInterface *bindParamWriter(IInterface *esdl, const char *esdlservice, const char *esdltype, const char *name)
  2849. {
  2850. if (*argsig != 'L') // should tell us the type of the object we need to create to pass in
  2851. typeError("OBJECT");
  2852. // Class name is from the char after the L up to the first ;
  2853. const char *tail = strchr(argsig, ';');
  2854. if (!tail)
  2855. typeError("OBJECT");
  2856. StringAttr className(argsig+1, tail - (argsig+1));
  2857. argsig = tail+1;
  2858. Owned<JavaXmlBuilder> writer = new JavaXmlBuilder(sharedCtx->JNIenv, dynamic_cast<IEsdlDefinition*>(esdl), esdlservice, esdltype);
  2859. writer->initWriter();
  2860. return (IXmlWriter*)writer.getClear();
  2861. }
  2862. virtual void paramWriterCommit(IInterface *writer)
  2863. {
  2864. JavaXmlBuilder *javaWriter = dynamic_cast<JavaXmlBuilder*>(writer);
  2865. if (!javaWriter)
  2866. throw MakeStringException(0, "javaembed: Invalid object writer for %s", sharedCtx->querySignature());
  2867. jvalue v;
  2868. v.l = javaWriter->getObject();
  2869. addArg(v);
  2870. }
  2871. virtual void bindDatasetParam(const char *name, IOutputMetaData & metaVal, IRowStream * val)
  2872. {
  2873. jvalue v;
  2874. char argsigStart = *argsig;
  2875. switch (argsigStart)
  2876. {
  2877. case '[':
  2878. case '<':
  2879. ++argsig;
  2880. break;
  2881. case 'L':
  2882. if (strncmp(argsig, "Ljava/util/Iterator<", 20) == 0)
  2883. {
  2884. argsig += 20;
  2885. break;
  2886. }
  2887. /* no break */
  2888. default:
  2889. typeError("DATASET");
  2890. }
  2891. if (*argsig != 'L') // should tell us the type of the object we need to create to pass in
  2892. typeError("DATASET");
  2893. // Class name is from the char after the L up to the first ;
  2894. const char *tail = strchr(argsig, ';');
  2895. if (!tail)
  2896. typeError("RECORD");
  2897. StringAttr className(argsig+1, tail - (argsig+1));
  2898. argsig = tail+1;
  2899. if (argsigStart=='L')
  2900. {
  2901. if (argsig[0] != '>' || argsig[1] != ';')
  2902. typeError("DATASET");
  2903. argsig += 2;
  2904. }
  2905. if (argsigStart=='[')
  2906. {
  2907. // Pass in an array of objects
  2908. PointerArrayOf<_jobject> allRows;
  2909. const RtlTypeInfo *typeInfo = metaVal.queryTypeInfo();
  2910. assertex(typeInfo);
  2911. RtlFieldStrInfo dummyField("<row>", NULL, typeInfo);
  2912. JavaObjectBuilder javaBuilder(sharedCtx->JNIenv, &dummyField, className);
  2913. for (;;)
  2914. {
  2915. roxiemem::OwnedConstRoxieRow thisRow = val->ungroupedNextRow();
  2916. if (!thisRow)
  2917. break;
  2918. const byte *brow = (const byte *) thisRow.get();
  2919. typeInfo->process(brow, brow, &dummyField, javaBuilder); // Creates a java object from the incoming ECL row
  2920. allRows.append(javaBuilder.getObject());
  2921. }
  2922. jobjectArray array = sharedCtx->JNIenv->NewObjectArray(allRows.length(), sharedCtx->JNIenv->FindClass(className), NULL);
  2923. ForEachItemIn(idx, allRows)
  2924. {
  2925. sharedCtx->JNIenv->SetObjectArrayElement(array, idx, allRows.item(idx));
  2926. }
  2927. v.l = array;
  2928. }
  2929. else
  2930. {
  2931. // Pass in an iterator
  2932. // Create a java object of type com.HPCCSystems.HpccUtils - this acts as a proxy for the iterator
  2933. sharedCtx->JNIenv->ExceptionClear();
  2934. jclass proxyClass = sharedCtx->JNIenv->FindClass("com/HPCCSystems/HpccUtils");
  2935. sharedCtx->checkException();
  2936. jmethodID constructor = sharedCtx->JNIenv->GetMethodID(proxyClass, "<init>", "(JLjava/lang/String;)V");
  2937. sharedCtx->checkException();
  2938. jvalue param;
  2939. const RtlTypeInfo *typeInfo = metaVal.queryTypeInfo();
  2940. ECLDatasetIterator *iterator = new ECLDatasetIterator(sharedCtx->JNIenv, typeInfo, className, val);
  2941. param.j = (jlong) iterator;
  2942. iterators.append(*iterator);
  2943. jobject proxy = sharedCtx->JNIenv->NewObject(proxyClass, constructor, param, sharedCtx->JNIenv->NewStringUTF(helperLibraryName));
  2944. sharedCtx->checkException();
  2945. v.l = proxy;
  2946. }
  2947. addArg(v);
  2948. }
  2949. virtual void writeResult(IInterface *esdl, const char *esdlservice, const char *esdltype, IInterface *writer)
  2950. {
  2951. return sharedCtx->writeObjectResult(result.l, dynamic_cast<IEsdlDefinition*>(esdl), esdlservice, esdltype, dynamic_cast<IXmlWriter*>(writer));
  2952. }
  2953. virtual void importFunction(size32_t lenChars, const char *utf)
  2954. {
  2955. sharedCtx->importFunction(lenChars, utf, classpath, instance);
  2956. argsig = sharedCtx->querySignature();
  2957. assertex(*argsig == '(');
  2958. argsig++;
  2959. }
  2960. virtual void callFunction()
  2961. {
  2962. if (*argsig != ')')
  2963. throw MakeStringException(0, "javaembed: Too few ECL parameters passed for Java signature %s", sharedCtx->querySignature());
  2964. if (instance)
  2965. sharedCtx->callFunction(result, args, instance);
  2966. else
  2967. sharedCtx->callFunction(result, args);
  2968. }
  2969. virtual void compileEmbeddedScript(size32_t lenChars, const char *script)
  2970. {
  2971. throwUnexpected(); // The java language helper supports only imported functions, not embedding java code in ECL.
  2972. }
  2973. protected:
  2974. JavaThreadContext *sharedCtx;
  2975. jvalue result;
  2976. StringAttr classpath;
  2977. IArrayOf<ECLDatasetIterator> iterators; // to make sure they get freed
  2978. jobject instance; //instance of service object to call methods on
  2979. private:
  2980. __declspec(noreturn) void typeError(const char *ECLtype) __attribute__((noreturn))
  2981. {
  2982. const char *javaType;
  2983. int javaLen = 0;
  2984. switch (*argsig)
  2985. {
  2986. case 'Z': javaType = "boolean"; break;
  2987. case 'B': javaType = "byte"; break;
  2988. case 'C': javaType = "char"; break;
  2989. case 'S': javaType = "short"; break;
  2990. case 'I': javaType = "int"; break;
  2991. case 'J': javaType = "long"; break;
  2992. case 'F': javaType = "float"; break;
  2993. case 'D': javaType = "double"; break;
  2994. case '[': javaType = "array"; break;
  2995. case 'L':
  2996. {
  2997. javaType = argsig+1;
  2998. const char *semi = strchr(argsig, ';');
  2999. if (semi)
  3000. javaLen = semi - javaType;
  3001. break;
  3002. }
  3003. case ')':
  3004. throw MakeStringException(0, "javaembed: Too many ECL parameters passed for Java signature %s", sharedCtx->querySignature());
  3005. default:
  3006. throw MakeStringException(0, "javaembed: Unrecognized character %c in java signature %s", *argsig, sharedCtx->querySignature());
  3007. }
  3008. if (!javaLen)
  3009. javaLen = strlen(argsig);
  3010. throw MakeStringException(0, "javaembed: ECL type %s cannot be passed to Java type %.*s", ECLtype, javaLen, javaType);
  3011. }
  3012. void addArg(jvalue &arg)
  3013. {
  3014. assertex(argcount < MAX_JNI_ARGS);
  3015. args[argcount] = arg;
  3016. argcount++;
  3017. }
  3018. jvalue args[MAX_JNI_ARGS];
  3019. int argcount;
  3020. const char *argsig;
  3021. };
  3022. static __thread JavaThreadContext* threadContext; // We reuse per thread, for speed
  3023. static __thread ThreadTermFunc threadHookChain;
  3024. static void releaseContext()
  3025. {
  3026. if (threadContext)
  3027. {
  3028. delete threadContext;
  3029. threadContext = NULL;
  3030. }
  3031. if (threadHookChain)
  3032. {
  3033. (*threadHookChain)();
  3034. threadHookChain = NULL;
  3035. }
  3036. }
  3037. static JavaThreadContext *queryContext()
  3038. {
  3039. if (!threadContext)
  3040. {
  3041. threadContext = new JavaThreadContext;
  3042. threadHookChain = addThreadTermFunc(releaseContext);
  3043. }
  3044. return threadContext;
  3045. }
  3046. static JNIEnv *queryJNIEnv()
  3047. {
  3048. return queryContext()->JNIenv;
  3049. }
  3050. class JavaEmbedServiceContext : public CInterfaceOf<IEmbedServiceContext>
  3051. {
  3052. public:
  3053. JavaEmbedServiceContext(JavaThreadContext *_sharedCtx, const char *service, const char *_options)
  3054. : sharedCtx(_sharedCtx), Class(0), options(_options), className(service), object(0)
  3055. {
  3056. StringArray opts;
  3057. opts.appendList(options, ",");
  3058. ForEachItemIn(idx, opts)
  3059. {
  3060. const char *opt = opts.item(idx);
  3061. const char *val = strchr(opt, '=');
  3062. if (val)
  3063. {
  3064. StringBuffer optName(val-opt, opt);
  3065. val++;
  3066. if (stricmp(optName, "classpath")==0)
  3067. classpath.set(val);
  3068. else
  3069. throw MakeStringException(0, "javaembed: Unknown option %s", optName.str());
  3070. }
  3071. }
  3072. // Create a new frame for local references and increase the capacity
  3073. // of those references to 64 (default is 16)
  3074. sharedCtx->JNIenv->PushLocalFrame(64);
  3075. }
  3076. ~JavaEmbedServiceContext()
  3077. {
  3078. if (object)
  3079. sharedCtx->JNIenv->DeleteGlobalRef(object);
  3080. if (Class)
  3081. sharedCtx->JNIenv->DeleteGlobalRef(Class);
  3082. // Pop local reference frame; explicitly frees all local
  3083. // references made during that frame's lifetime
  3084. sharedCtx->JNIenv->PopLocalFrame(NULL);
  3085. }
  3086. void init()
  3087. {
  3088. jobject classLoader = sharedCtx->getThreadClassLoader();
  3089. checkException();
  3090. jmethodID loadClassMethod = sharedCtx->JNIenv->GetMethodID(sharedCtx->JNIenv->GetObjectClass(classLoader), "loadClass","(Ljava/lang/String;)Ljava/lang/Class;");
  3091. checkException();
  3092. jstring methodString = sharedCtx->JNIenv->NewStringUTF(className);
  3093. checkException();
  3094. Class = (jclass) sharedCtx->JNIenv->NewGlobalRef(sharedCtx->JNIenv->CallObjectMethod(classLoader, loadClassMethod, methodString));
  3095. checkException();
  3096. jmethodID constructor = sharedCtx->JNIenv->GetMethodID(Class, "<init>", "()V");
  3097. checkException();
  3098. object = sharedCtx->JNIenv->NewGlobalRef(sharedCtx->JNIenv->NewObject(Class, constructor));
  3099. checkException();
  3100. }
  3101. virtual IEmbedFunctionContext *createFunctionContext(const char *function)
  3102. {
  3103. if (!object)
  3104. return NULL;
  3105. Owned<JavaEmbedImportContext> fctx = new JavaEmbedImportContext(queryContext(), object, options);
  3106. fctx->importFunction(rtlUtf8Length(strlen(function), function), function);
  3107. return fctx.getClear();
  3108. }
  3109. void checkException()
  3110. {
  3111. javaembed::checkException(sharedCtx->JNIenv, false);
  3112. }
  3113. protected:
  3114. JavaThreadContext *sharedCtx;
  3115. StringBuffer className;
  3116. jclass Class;
  3117. jobject object;
  3118. StringAttr classpath;
  3119. StringAttr options;
  3120. };
  3121. class JavaEmbedContext : public CInterfaceOf<IEmbedContext>
  3122. {
  3123. public:
  3124. virtual IEmbedFunctionContext *createFunctionContext(unsigned flags, const char *options) override
  3125. {
  3126. return createFunctionContextEx(nullptr, nullptr, flags, options);
  3127. }
  3128. virtual IEmbedFunctionContext *createFunctionContextEx(ICodeContext * ctx, const IThorActivityContext *activityCtx, unsigned flags, const char *options) override
  3129. {
  3130. assertex(flags & EFimport);
  3131. return new JavaEmbedImportContext(queryContext(), NULL, options);
  3132. }
  3133. virtual IEmbedServiceContext *createServiceContext(const char *service, unsigned flags, const char *options) override
  3134. {
  3135. Owned<JavaEmbedServiceContext> serviceContext = new JavaEmbedServiceContext(queryContext(), service, options);
  3136. serviceContext->init();
  3137. return serviceContext.getClear();
  3138. }
  3139. };
  3140. extern DECL_EXPORT IEmbedContext* getEmbedContext()
  3141. {
  3142. return new JavaEmbedContext;
  3143. }
  3144. } // namespace
  3145. // Callbacks from java
  3146. extern "C" {
  3147. JNIEXPORT jboolean JNICALL Java_com_HPCCSystems_HpccUtils__1hasNext (JNIEnv *, jclass, jlong);
  3148. JNIEXPORT jobject JNICALL Java_com_HPCCSystems_HpccUtils__1next (JNIEnv *, jclass, jlong);
  3149. }
  3150. JNIEXPORT jboolean JNICALL Java_com_HPCCSystems_HpccUtils__1hasNext (JNIEnv *JNIenv, jclass, jlong proxy)
  3151. {
  3152. try
  3153. {
  3154. javaembed::ECLDatasetIterator *e = (javaembed::ECLDatasetIterator *) proxy;
  3155. return e->hasNext();
  3156. }
  3157. catch (IException *E)
  3158. {
  3159. StringBuffer msg;
  3160. E->errorMessage(msg);
  3161. E->Release();
  3162. jclass eClass = JNIenv->FindClass("java/lang/IllegalArgumentException");
  3163. if (eClass)
  3164. JNIenv->ThrowNew(eClass, msg.str());
  3165. return false;
  3166. }
  3167. }
  3168. JNIEXPORT jobject JNICALL Java_com_HPCCSystems_HpccUtils__1next (JNIEnv *JNIenv, jclass, jlong proxy)
  3169. {
  3170. try
  3171. {
  3172. javaembed::ECLDatasetIterator *e = (javaembed::ECLDatasetIterator *) proxy;
  3173. return e->next();
  3174. }
  3175. catch (IException *E)
  3176. {
  3177. StringBuffer msg;
  3178. E->errorMessage(msg);
  3179. E->Release();
  3180. jclass eClass = JNIenv->FindClass("java/lang/IllegalArgumentException");
  3181. if (eClass)
  3182. JNIenv->ThrowNew(eClass, msg.str());
  3183. return NULL;
  3184. }
  3185. }
  3186. // Used for dynamically loading in ESDL
  3187. extern "C" DECL_EXPORT IEmbedContext *getEmbedContextDynamic()
  3188. {
  3189. return javaembed::getEmbedContext();
  3190. }