2 #include "mit_moira_MoiraConnectInternal.h"
9 static void throwMoiraException(JNIEnv *env, int code)
17 IOE = (*env)->FindClass(env, "mit/moira/MoiraException");
19 fprintf(stderr, "moirai: No Class\n");
22 mid = (*env)->GetMethodID(env, IOE, "<init>", "(Ljava/lang/String;I)V");
24 fprintf(stderr, "moirai: No Method\n");
27 sprintf(buffer, "%s", error_message(code));
28 jmess = (*env)->NewStringUTF(env, buffer);
30 fprintf(stderr, "Cannot get new string\n");
33 tothrow = (*env)->NewObject(env, IOE, mid, jmess, (jint) code);
34 if (tothrow == NULL) {
35 fprintf(stderr, "moirai: No Throw\n");
38 (*env)->Throw(env, (jthrowable) tothrow);
45 static void throwMoiraExceptionMess(JNIEnv *env, char *mess)
47 jclass IOE = (*env)->FindClass(env, "mit/moira/MoiraException");
48 if (IOE == NULL) abort();
49 (*env)->ThrowNew(env, IOE, mess);
53 JNIEXPORT void JNICALL Java_mit_moira_MoiraConnectInternal_connect(JNIEnv *env,
54 jclass Class, jstring server) {
57 const char *aserver = (*env)->GetStringUTFChars(env, server, 0);
58 if (strlen(aserver) > sizeof(buffer)) abort();
59 strcpy(buffer, aserver);
60 (*env)->ReleaseStringUTFChars(env, server, aserver);
61 status = mr_connect(buffer);
62 if (status != MR_SUCCESS) throwMoiraException(env, status);
63 status = mr_version(2);
64 if (status != MR_SUCCESS) {
65 if (status == MR_VERSION_LOW) return; /* This is OK */
68 throwMoiraException(env, status);
74 JNIEXPORT void JNICALL Java_mit_moira_MoiraConnectInternal_proxy(JNIEnv *env,
79 const char *user = (*env)->GetStringUTFChars(env, jUser, 0);
80 if (strlen(user) > sizeof(buffer)) abort();
82 (*env)->ReleaseStringUTFChars(env, jUser, user);
83 status = mr_proxy(buffer, "Java");
84 if (status != MR_SUCCESS) throwMoiraException(env, status);
89 JNIEXPORT void JNICALL Java_mit_moira_MoiraConnectInternal_auth(JNIEnv *env,
92 status = mr_krb5_auth("JavaInterface");
93 if (status != MR_SUCCESS) throwMoiraException(env, status);
97 JNIEXPORT void JNICALL Java_mit_moira_MoiraConnectInternal_disconnect
98 (JNIEnv *env, jclass Class) {
103 typedef struct ArgBlock {
109 typedef struct ListBlock {
115 static jobjectArray ArgBlockToJavaArray(JNIEnv *env, ArgBlock *argb);
116 static jobjectArray ListBlockToJavaArray(JNIEnv *env, ListBlock *lb);
118 extern int mr_query(char *name, int argc, char **argv,
119 int (*proc)(int, char **, void *), void *hint);
120 static ListBlock *NewListBlock();
121 static ArgBlock *NewArgBlock();
122 static void FreeListBlock(ListBlock *);
123 static int StashResults(int argc, char **argv, void *callback);
124 static char **ConvertJavaArgs(JNIEnv *env, jobjectArray jargs);
125 static void FreeArgs(char **args);
127 JNIEXPORT jobjectArray JNICALL Java_mit_moira_MoiraConnectInternal_mr_1query
128 (JNIEnv *env, jclass Class, jstring jcommand, jobjectArray jargs) {
129 ListBlock *listblock = NewListBlock();
134 char **args = ConvertJavaArgs(env, jargs);
135 if (args == NULL) return (NULL); /* It probably thru an exception */
136 command = (*env)->GetStringUTFChars(env, jcommand, 0);
137 strncpy(icommand, command, sizeof(icommand));
139 status = mr_query(icommand, (*env)->GetArrayLength(env, jargs), args,
140 StashResults, listblock);
141 FreeArgs(args); /* Don't need them anymore */
142 if (status != MR_SUCCESS) {
143 FreeListBlock(listblock);
144 throwMoiraException(env, status);
147 /* if (listblock->count == 0) { / * No such list or empty list * /
148 FreeListBlock(listblock);
149 throwMoiraExceptionMess(env, "No Such List or Empty List");
153 if (listblock->count == 0) return (0);
155 retval = ListBlockToJavaArray(env, listblock);
156 FreeListBlock(listblock);
161 static ArgBlock *NewArgBlock() {
162 ArgBlock *argb = (ArgBlock *)malloc(sizeof(ArgBlock));
163 argb->alloccount = 10;
164 argb->stringval = (char **) malloc(10 * sizeof(char *));
169 static void AddArg(ArgBlock *argb, char *arg) {
171 if (argb->alloccount <= (argb->count + 1)) {
172 char **st = malloc((2 * argb->alloccount) * sizeof(char *));
173 argb->alloccount *= 2;
174 for (i = 0; i < argb->count; i++)
175 st[i] = argb->stringval[i];
176 free(argb->stringval);
177 argb->stringval = st;
179 argb->stringval[argb->count++] = arg;
182 static ListBlock *NewListBlock() {
183 ListBlock *list = (ListBlock *)malloc(sizeof(ListBlock));
184 list->alloccount = 10;
185 list->Args = (ArgBlock **) malloc(10 * sizeof(ArgBlock *));
190 static void FreeArgBlock(ArgBlock *argb) {
192 for (i = 0; i < argb->count; i++) free(argb->stringval[i]);
193 free(argb->stringval);
197 static void FreeListBlock(ListBlock *list) {
199 for (i = 0; i < list->count; i++) FreeArgBlock(list->Args[i]);
204 static jobjectArray ArgBlockToJavaArray(JNIEnv *env, ArgBlock *argb) {
207 retval = (*env)->NewObjectArray(env, argb->count,
208 (*env)->FindClass(env, "java/lang/String"),
210 for (i = 0; i < argb->count; i++) {
211 (*env)->SetObjectArrayElement(env, retval, i,
212 (*env)->NewStringUTF(env, argb->stringval[i]));
217 static jobjectArray ListBlockToJavaArray(JNIEnv *env, ListBlock *lb) {
220 retval = (*env)->NewObjectArray(env, lb->count,
221 (*env)->FindClass(env, "java/lang/Object"),
223 for (i = 0; i < lb->count; i++)
224 (*env)->SetObjectArrayElement(env, retval, i,
225 ArgBlockToJavaArray(env, lb->Args[i]));
229 static int StashResults(int argc, char **argv, void *callback) {
230 ListBlock *list = (ListBlock *) callback;
235 /* printf("DEBUG: StashResults argc = %d\n", argc);
236 * for (i = 0; i < argc; i++)
237 * printf("DEBUG: StashResults argv[%d] = %s\n", i, argv[i]);
240 while (list->alloccount <= (list->count + 1)) {
241 ArgBlock **args = (ArgBlock **) malloc(list->alloccount * 2 * sizeof(char *));
242 list->alloccount = list->alloccount * 2;
243 for(i = 0; i < list->count; i++) {
244 args[i] = list->Args[i];
250 argb = NewArgBlock();
252 for (i = 0; i < argc; i++) {
253 arg = (char *) malloc(strlen(argv[i]) + 1);
254 strcpy(arg, argv[i]);
257 list->Args[list->count++] = argb;
261 static char **ConvertJavaArgs(JNIEnv *env, jobjectArray jargs) {
265 int length = (*env)->GetArrayLength(env, jargs);
266 if (length == 0) { /* Does this happen in a non-error situation? */
267 retval = (char **) malloc(sizeof (char *));
271 retval = (char **) malloc((length + 1) * sizeof(char *));
272 for (i = 0; i < length; i++) {
273 jobject jarg = (*env)->GetObjectArrayElement(env, jargs, i);
274 if ((*env)->ExceptionOccurred(env)) {
275 for (j = 0; j < i; j++) free(retval[j]);
279 iarg = (*env)->GetStringUTFChars(env, jarg, 0);
280 if ((*env)->ExceptionOccurred(env)) {
281 for (j = 0; j < i; j++) free(retval[j]);
285 retval[i] = malloc(strlen(iarg) + 1);
286 strcpy(retval[i], iarg);
287 (*env)->ReleaseStringUTFChars(env, jarg, iarg);
289 retval[length] = NULL;
293 static void FreeArgs(char **args) {
296 while (args[i] != NULL) free(args[i++]);