From da1b85df1f4655adcc44636ffa8fabdab24c629a Mon Sep 17 00:00:00 2001
From: Vedant Tewari <vedanttewari541@gmail.com>
Date: Mon, 9 Sep 2024 01:34:31 -0500
Subject: [PATCH] Added checks for return type signature and array dimensions

---
 cobc/codegen.c | 105 ++++++++++++++++++++++++++++++++++++-------------
 libcob/java.c  |   4 ++
 2 files changed, 81 insertions(+), 28 deletions(-)

diff --git a/cobc/codegen.c b/cobc/codegen.c
index d3b758e21..36e4940ed 100644
--- a/cobc/codegen.c
+++ b/cobc/codegen.c
@@ -7100,6 +7100,7 @@ output_java_call (struct cb_call *p)
 	char *last_dot;
 	char *method_name;
 	const char *class_name;
+	char return_type_signature[32];
 	char method_signature[2048] = "(";
 	char* mangled;
 	struct cb_tree_common *ptr;
@@ -7146,34 +7147,50 @@ output_java_call (struct cb_call *p)
                     strcat(method_signature, "Ljava/lang/String;");
                 }
                 break;
-        	case CB_TAG_LIST:
-				{
-					struct cb_tree_common **list_elements = (struct cb_tree_common **) ptr;
-					for (int j = 0; list_elements[j] != NULL; j++) {
-						switch (CB_TREE_TAG(list_elements[j])) {
-							case CB_TAG_INTEGER:
-								strcat(method_signature, "[I"); 
-								break;
-							case CB_USAGE_FLOAT:
-								strcat(method_signature, "[F");
-								break;
-							case CB_USAGE_DOUBLE:
-								strcat(method_signature, "[D");
-								break;
-							case CB_CLASS_BOOLEAN:
-								strcat(method_signature, "[Z"); 
-								break;
-							case CB_TAG_STRING:
-								strcat(method_signature, "[Ljava/lang/String;"); 
-								break;
-							case CB_USAGE_OBJECT:
-								strcat(method_signature, "[Ljava/lang/Object;"); 
-								break;
-							default:
-								cobc_err_msg(_("Unsupported array type in Java method call"));
-								COBC_ABORT();
-						}
-					}
+				case CB_TAG_LIST:
+                {
+                    struct cb_tree_common **list_elements = (struct cb_tree_common **) ptr;
+                    int array_dimension = 1; 
+
+                    while (list_elements != NULL) {
+                        struct cb_tree_common **inner_list = NULL;
+                        for (int j = 0; list_elements[j] != NULL; j++) {
+                            if (CB_TREE_TAG(list_elements[j]) == CB_TAG_LIST) {
+                                array_dimension++;
+                                inner_list = (struct cb_tree_common **) list_elements[j];
+                            } else {
+                                switch (CB_TREE_TAG(list_elements[j])) {
+                                    case CB_TAG_INTEGER:
+                                        strcat(method_signature, "[I");
+                                        break;
+                                    case CB_USAGE_FLOAT:
+                                        strcat(method_signature, "[F");
+                                        break;
+                                    case CB_USAGE_DOUBLE:
+                                        strcat(method_signature, "[D");
+                                        break;
+                                    case CB_CLASS_BOOLEAN:
+                                        strcat(method_signature, "[Z");
+                                        break;
+                                    case CB_TAG_STRING:
+                                        strcat(method_signature, "[Ljava/lang/String;");  
+                                        break;
+                                    case CB_USAGE_OBJECT:
+                                        strcat(method_signature, "[Ljava/lang/Object;");  
+                                        break;
+                                    default:
+                                        cobc_err_msg(_("Unsupported array element type in Java method call"));
+                                        COBC_ABORT();
+                                }
+                            }
+                        }
+                        list_elements = inner_list;
+                    }
+
+					if (array_dimension > 2) {
+                        cobc_err_msg(_("Unsupported array dimension: %d"), array_dimension);
+                        COBC_ABORT();
+                    }
 				}
 				break;
 			default:
@@ -7182,6 +7199,38 @@ output_java_call (struct cb_call *p)
 		}
     }
 
+	if (p->call_returning == NULL) {
+		strcat(method_signature, ")V");
+		strcpy(return_type_signature, "void");
+	} else {
+		switch(CB_TREE_TAG(p->call_returning)) {
+			case CB_TAG_INTEGER:
+				strcat(method_signature, ")I");
+				strcpy(return_type_signature, "jint");
+				break;
+			case CB_TAG_STRING:
+				strcat(method_signature, ")Ljava/lang/String;");
+				strcpy(return_type_signature, "jstring");
+				break;
+			case CB_USAGE_FLOAT:
+				strcat(method_signature, ")F");
+				strcpy(return_type_signature, "jfloat");
+				break;
+			case CB_USAGE_DOUBLE:
+				strcat(method_signature, ")D");
+				strcpy(return_type_signature, "jdouble");
+				break;
+			case CB_CLASS_BOOLEAN:
+				strcat(method_signature, ")Z");
+				strcpy(return_type_signature, "jboolean");
+				break;
+			default:
+				strcat(method_signature, ")V");
+				strcpy(return_type_signature, "void");
+				break;
+		}
+	}
+
 	strcat(method_signature, ")V");
 
 	lookup_java_call(mangled, method_signature);
diff --git a/libcob/java.c b/libcob/java.c
index 9f569aa81..266cf2fc5 100644
--- a/libcob/java.c
+++ b/libcob/java.c
@@ -79,11 +79,15 @@ resolve_java (const char		*class_name,
 	cls = (*env)->FindClass(env, jni_class_name);
 	free(jni_class_name);
 	if (!cls) {
+		(*env)->ExceptionDescribe(env);
+		(*env)->ExceptionClear(env);
 		return NULL;
 	}
 
 	mid = (*env)->GetStaticMethodID(env, cls, method_name, method_signature);
 	if (!mid) {
+		(*env)->ExceptionDescribe(env);
+		(*env)->ExceptionClear(env);
 		(*env)->DeleteLocalRef(env, cls);
 		return NULL;
 	}