瀏覽代碼

app_ruby: update to handle xval return for functions with 3 parameters

- fix for functions like KSR.sqlops.sql_result_get()
Daniel-Constantin Mierla 3 年之前
父節點
當前提交
3136d47993
共有 1 個文件被更改,包括 93 次插入66 次删除
  1. 93 66
      src/modules/app_ruby/app_ruby_api.c

+ 93 - 66
src/modules/app_ruby/app_ruby_api.c

@@ -935,77 +935,104 @@ VALUE sr_kemi_ruby_exec_func_ex(ksr_ruby_context_t *R, sr_kemi_t *ket, int argc,
 			}
 		break;
 		case 3:
-			if(ket->ptypes[0]==SR_KEMIP_INT) {
-				if(ket->ptypes[1]==SR_KEMIP_INT) {
-					if(ket->ptypes[2]==SR_KEMIP_INT) {
-						ret = ((sr_kemi_fmnnn_f)(ket->func))(env_R->msg,
-								vps[0].n, vps[1].n, vps[2].n);
-						return sr_kemi_ruby_return_int(ket, ret);
-					} else if(ket->ptypes[2]==SR_KEMIP_STR) {
-						ret = ((sr_kemi_fmnns_f)(ket->func))(env_R->msg,
-								vps[0].n, vps[1].n, &vps[2].s);
-						return sr_kemi_ruby_return_int(ket, ret);
-					} else {
-						LM_ERR("invalid parameters for: %.*s\n",
-								fname->len, fname->s);
-						return Qfalse;
-					}
-				} else if(ket->ptypes[1]==SR_KEMIP_STR) {
-					if(ket->ptypes[2]==SR_KEMIP_INT) {
-						ret = ((sr_kemi_fmnsn_f)(ket->func))(env_R->msg,
-								vps[0].n, &vps[1].s, vps[2].n);
-						return sr_kemi_ruby_return_int(ket, ret);
-					} else if(ket->ptypes[2]==SR_KEMIP_STR) {
-						ret = ((sr_kemi_fmnss_f)(ket->func))(env_R->msg,
-								vps[0].n, &vps[1].s, &vps[2].s);
-						return sr_kemi_ruby_return_int(ket, ret);
-					} else {
-						LM_ERR("invalid parameters for: %.*s\n",
-								fname->len, fname->s);
-						return Qfalse;
-					}
+			if(ket->ptypes[0]==SR_KEMIP_STR
+					&& ket->ptypes[1]==SR_KEMIP_STR
+					&& ket->ptypes[2]==SR_KEMIP_STR) {
+				if(ket->rtype==SR_KEMIP_XVAL) {
+					xret = ((sr_kemi_xfmsss_f)(ket->func))(env_R->msg,
+						&vps[0].s, &vps[1].s, &vps[2].s);
+					return sr_kemi_ruby_return_xval(ket, xret);
 				} else {
-					LM_ERR("invalid parameters for: %.*s\n",
-							fname->len, fname->s);
-					return Qfalse;
+					ret = ((sr_kemi_fmsss_f)(ket->func))(env_R->msg,
+						&vps[0].s, &vps[1].s, &vps[2].s);
+					return sr_kemi_ruby_return_int(ket, ret);
 				}
-			} else if(ket->ptypes[0]==SR_KEMIP_STR) {
-				if(ket->ptypes[1]==SR_KEMIP_INT) {
-					if(ket->ptypes[2]==SR_KEMIP_INT) {
-						ret = ((sr_kemi_fmsnn_f)(ket->func))(env_R->msg,
-								&vps[0].s, vps[1].n, vps[2].n);
-						return sr_kemi_ruby_return_int(ket, ret);
-					} else if(ket->ptypes[2]==SR_KEMIP_STR) {
-						ret = ((sr_kemi_fmsns_f)(ket->func))(env_R->msg,
-								&vps[0].s, vps[1].n, &vps[2].s);
-						return sr_kemi_ruby_return_int(ket, ret);
-					} else {
-						LM_ERR("invalid parameters for: %.*s\n",
-								fname->len, fname->s);
-						return Qfalse;
-					}
-				} else if(ket->ptypes[1]==SR_KEMIP_STR) {
-					if(ket->ptypes[2]==SR_KEMIP_INT) {
-						ret = ((sr_kemi_fmssn_f)(ket->func))(env_R->msg,
-								&vps[0].s, &vps[1].s, vps[2].n);
-						return sr_kemi_ruby_return_int(ket, ret);
-					} else if(ket->ptypes[2]==SR_KEMIP_STR) {
-						ret = ((sr_kemi_fmsss_f)(ket->func))(env_R->msg,
-								&vps[0].s, &vps[1].s, &vps[2].s);
-						return sr_kemi_ruby_return_int(ket, ret);
-					} else {
-						LM_ERR("invalid parameters for: %.*s\n",
-								fname->len, fname->s);
-						return Qfalse;
-					}
+			} else if(ket->ptypes[0]==SR_KEMIP_STR
+					&& ket->ptypes[1]==SR_KEMIP_STR
+					&& ket->ptypes[2]==SR_KEMIP_INT) {
+				if(ket->rtype==SR_KEMIP_XVAL) {
+					xret = ((sr_kemi_xfmssn_f)(ket->func))(env_R->msg,
+						&vps[0].s, &vps[1].s, vps[2].n);
+					return sr_kemi_ruby_return_xval(ket, xret);
 				} else {
-					LM_ERR("invalid parameters for: %.*s\n",
-							fname->len, fname->s);
-					return Qfalse;
+					ret = ((sr_kemi_fmssn_f)(ket->func))(env_R->msg,
+						&vps[0].s, &vps[1].s, vps[2].n);
+					return sr_kemi_ruby_return_int(ket, ret);
+				}
+			} else if(ket->ptypes[0]==SR_KEMIP_STR
+					&& ket->ptypes[1]==SR_KEMIP_INT
+					&& ket->ptypes[2]==SR_KEMIP_STR) {
+				if(ket->rtype==SR_KEMIP_XVAL) {
+					xret = ((sr_kemi_xfmsns_f)(ket->func))(env_R->msg,
+						&vps[0].s, vps[1].n, &vps[2].s);
+					return sr_kemi_ruby_return_xval(ket, xret);
+				} else {
+					ret = ((sr_kemi_fmsns_f)(ket->func))(env_R->msg,
+						&vps[0].s, vps[1].n, &vps[2].s);
+					return sr_kemi_ruby_return_int(ket, ret);
+				}
+			} else if(ket->ptypes[0]==SR_KEMIP_STR
+					&& ket->ptypes[1]==SR_KEMIP_INT
+					&& ket->ptypes[2]==SR_KEMIP_INT) {
+				if(ket->rtype==SR_KEMIP_XVAL) {
+					xret = ((sr_kemi_xfmsnn_f)(ket->func))(env_R->msg,
+						&vps[0].s, vps[1].n, vps[2].n);
+					return sr_kemi_ruby_return_xval(ket, xret);
+				} else {
+					ret = ((sr_kemi_fmsnn_f)(ket->func))(env_R->msg,
+						&vps[0].s, vps[1].n, vps[2].n);
+					return sr_kemi_ruby_return_int(ket, ret);
+				}
+			} else if(ket->ptypes[0]==SR_KEMIP_INT
+					&& ket->ptypes[1]==SR_KEMIP_STR
+					&& ket->ptypes[2]==SR_KEMIP_STR) {
+				if(ket->rtype==SR_KEMIP_XVAL) {
+					xret = ((sr_kemi_xfmnss_f)(ket->func))(env_R->msg,
+						vps[0].n, &vps[1].s, &vps[2].s);
+					return sr_kemi_ruby_return_xval(ket, xret);
+				} else {
+					ret = ((sr_kemi_fmnss_f)(ket->func))(env_R->msg,
+						vps[0].n, &vps[1].s, &vps[2].s);
+					return sr_kemi_ruby_return_int(ket, ret);
+				}
+			} else if(ket->ptypes[0]==SR_KEMIP_INT
+					&& ket->ptypes[1]==SR_KEMIP_STR
+					&& ket->ptypes[2]==SR_KEMIP_INT) {
+				if(ket->rtype==SR_KEMIP_XVAL) {
+					xret = ((sr_kemi_xfmnsn_f)(ket->func))(env_R->msg,
+						vps[0].n, &vps[1].s, vps[2].n);
+					return sr_kemi_ruby_return_xval(ket, xret);
+				} else {
+					ret = ((sr_kemi_fmnsn_f)(ket->func))(env_R->msg,
+						vps[0].n, &vps[1].s, vps[2].n);
+					return sr_kemi_ruby_return_int(ket, ret);
+				}
+			} else if(ket->ptypes[0]==SR_KEMIP_INT
+					&& ket->ptypes[1]==SR_KEMIP_INT
+					&& ket->ptypes[2]==SR_KEMIP_STR) {
+				if(ket->rtype==SR_KEMIP_XVAL) {
+					xret = ((sr_kemi_xfmnns_f)(ket->func))(env_R->msg,
+						vps[0].n, vps[1].n, &vps[2].s);
+					return sr_kemi_ruby_return_xval(ket, xret);
+				} else {
+					ret = ((sr_kemi_fmnns_f)(ket->func))(env_R->msg,
+						vps[0].n, vps[1].n, &vps[2].s);
+					return sr_kemi_ruby_return_int(ket, ret);
+				}
+			} else if(ket->ptypes[0]==SR_KEMIP_INT
+					&& ket->ptypes[1]==SR_KEMIP_INT
+					&& ket->ptypes[2]==SR_KEMIP_INT) {
+				if(ket->rtype==SR_KEMIP_XVAL) {
+					xret = ((sr_kemi_xfmnnn_f)(ket->func))(env_R->msg,
+						vps[0].n, vps[1].n, vps[2].n);
+					return sr_kemi_ruby_return_xval(ket, xret);
+				} else {
+					ret = ((sr_kemi_fmnnn_f)(ket->func))(env_R->msg,
+						vps[0].n, vps[1].n, vps[2].n);
+					return sr_kemi_ruby_return_int(ket, ret);
 				}
 			} else {
-				LM_ERR("invalid parameters for: %.*s\n",
-						fname->len, fname->s);
+				LM_ERR("invalid parameters for: %.*s\n", fname->len, fname->s);
 				return Qfalse;
 			}
 		break;