📄 gen_rpc.awk
字号:
printf("\n\t/*\n\t * XXX Code goes here\n\t */\n\n") >> PFILE printf("\treplyp->status = ret;\n") >> PFILE printf("\treturn;\n") >> PFILE printf("}\n\n") >> PFILE # # ===================================================== # Generate Client code # # Spit out PUBLIC prototypes. # pi = 1; p[pi++] = sprintf("int __dbcl_%s __P((", name); p[pi++] = ""; for (i = 0; i < nvars; ++i) { p[pi++] = pr_type[i]; p[pi++] = ", "; } p[pi - 1] = ""; p[pi++] = "));"; p[pi] = ""; proto_format(p, 0, CFILE); # # Spit out function name/args. # printf("int\n") >> CFILE printf("__dbcl_%s(", name) >> CFILE sep = ""; for (i = 0; i < nvars; ++i) { printf("%s%s", sep, args[i]) >> CFILE sep = ", "; } printf(")\n") >> CFILE for (i = 0; i < nvars; ++i) if (func_arg[i] == 0) printf("\t%s %s;\n", c_type[i], args[i]) >> CFILE else printf("\t%s;\n", c_type[i]) >> CFILE printf("{\n") >> CFILE printf("\tCLIENT *cl;\n") >> CFILE printf("\t__%s_msg msg;\n", name) >> CFILE printf("\t__%s_reply *replyp = NULL;\n", name) >> CFILE; printf("\tint ret;\n") >> CFILE if (!env_handle) printf("\tDB_ENV *dbenv;\n") >> CFILE # # If we are managing a list, we need a few more vars. # for (i = 0; i < nvars; ++i) { if (rpc_type[i] == "LIST") { printf("\t%s %sp;\n", c_type[i], args[i]) >> CFILE printf("\tint %si;\n", args[i]) >> CFILE if (list_type[i] == "GID") printf("\tu_int8_t ** %sq;\n", args[i]) >> CFILE else printf("\tu_int32_t * %sq;\n", args[i]) >> CFILE } } printf("\n") >> CFILE printf("\tret = 0;\n") >> CFILE if (!env_handle) { if (db_handle) printf("\tdbenv = %s->dbenv;\n", args[db_idx]) >> CFILE else if (dbc_handle) printf("\tdbenv = %s->dbp->dbenv;\n", \ args[dbc_idx]) >> CFILE else if (txn_handle) printf("\tdbenv = %s->mgrp->dbenv;\n", \ args[txn_idx]) >> CFILE else printf("\tdbenv = NULL;\n") >> CFILE printf("\tif (dbenv == NULL || !RPC_ON(dbenv))\n") \ >> CFILE printf("\t\treturn (__dbcl_noserver(NULL));\n") >> CFILE } else { printf("\tif (%s == NULL || !RPC_ON(%s))\n", \ args[env_idx], args[env_idx]) >> CFILE printf("\t\treturn (__dbcl_noserver(%s));\n", \ args[env_idx]) >> CFILE } printf("\n") >> CFILE if (!env_handle) printf("\tcl = (CLIENT *)dbenv->cl_handle;\n") >> CFILE else printf("\tcl = (CLIENT *)%s->cl_handle;\n", \ args[env_idx]) >> CFILE printf("\n") >> CFILE # # If there is a function arg, check that it is NULL # for (i = 0; i < nvars; ++i) { if (func_arg[i] != 1) continue; printf("\tif (%s != NULL) {\n", args[i]) >> CFILE if (!env_handle) { printf("\t\t__db_err(dbenv, ") >> CFILE } else { printf("\t\t__db_err(%s, ", args[env_idx]) >> CFILE } printf("\"User functions not supported in RPC\");\n") >> CFILE printf("\t\treturn (EINVAL);\n\t}\n") >> CFILE } # # Compose message components # for (i = 0; i < nvars; ++i) { if (rpc_type[i] == "ID") { printf("\tif (%s == NULL)\n", args[i]) >> CFILE printf("\t\tmsg.%scl_id = 0;\n\telse\n", \ args[i]) >> CFILE if (c_type[i] == "DB_TXN *") { printf("\t\tmsg.%scl_id = %s->txnid;\n", \ args[i], args[i]) >> CFILE } else { printf("\t\tmsg.%scl_id = %s->cl_id;\n", \ args[i], args[i]) >> CFILE } } if (rpc_type[i] == "GID") { printf("\tmemcpy(msg.%s, %s, %d);\n", \ args[i], args[i], xidsize) >> CFILE } if (rpc_type[i] == "INT") { printf("\tmsg.%s = %s;\n", args[i], args[i]) >> CFILE } if (rpc_type[i] == "STRING") { printf("\tif (%s == NULL)\n", args[i]) >> CFILE printf("\t\tmsg.%s = \"\";\n", args[i]) >> CFILE printf("\telse\n") >> CFILE printf("\t\tmsg.%s = (char *)%s;\n", \ args[i], args[i]) >> CFILE } if (rpc_type[i] == "DBT") { printf("\tmsg.%sdlen = %s->dlen;\n", \ args[i], args[i]) >> CFILE printf("\tmsg.%sdoff = %s->doff;\n", \ args[i], args[i]) >> CFILE printf("\tmsg.%sulen = %s->ulen;\n", \ args[i], args[i]) >> CFILE printf("\tmsg.%sflags = %s->flags;\n", \ args[i], args[i]) >> CFILE printf("\tmsg.%sdata.%sdata_val = %s->data;\n", \ args[i], args[i], args[i]) >> CFILE printf("\tmsg.%sdata.%sdata_len = %s->size;\n", \ args[i], args[i], args[i]) >> CFILE } if (rpc_type[i] == "LIST") { printf("\tfor (%si = 0, %sp = %s; *%sp != 0; ", \ args[i], args[i], args[i], args[i]) >> CFILE printf(" %si++, %sp++)\n\t\t;\n", args[i], args[i]) \ >> CFILE # # If we are an array of ints, *_len is how many # elements. If we are a GID, *_len is total bytes. # printf("\tmsg.%s.%s_len = %si",args[i], args[i], \ args[i]) >> CFILE if (list_type[i] == "GID") printf(" * %d;\n", xidsize) >> CFILE else printf(";\n") >> CFILE printf("\tif ((ret = __os_calloc(") >> CFILE if (!env_handle) printf("dbenv,\n") >> CFILE else printf("%s,\n", args[env_idx]) >> CFILE printf("\t msg.%s.%s_len,", \ args[i], args[i]) >> CFILE if (list_type[i] == "GID") printf(" 1,") >> CFILE else printf(" sizeof(u_int32_t),") >> CFILE printf(" &msg.%s.%s_val)) != 0)\n",\ args[i], args[i], args[i], args[i]) >> CFILE printf("\t\treturn (ret);\n") >> CFILE printf("\tfor (%sq = msg.%s.%s_val, %sp = %s; ", \ args[i], args[i], args[i], \ args[i], args[i]) >> CFILE printf("%si--; %sq++, %sp++)\n", \ args[i], args[i], args[i]) >> CFILE printf("\t\t*%sq = ", args[i]) >> CFILE if (list_type[i] == "GID") printf("*%sp;\n", args[i]) >> CFILE if (list_type[i] == "ID") printf("(*%sp)->cl_id;\n", args[i]) >> CFILE if (list_type[i] == "INT") printf("*%sp;\n", args[i]) >> CFILE } } printf("\n") >> CFILE printf("\treplyp = __db_%s_%d%03d(&msg, cl);\n", name, major, minor) \ >> CFILE for (i = 0; i < nvars; ++i) { if (rpc_type[i] == "LIST") { printf("\t__os_free(") >> CFILE if (!env_handle) printf("dbenv, ") >> CFILE else printf("%s, ", args[env_idx]) >> CFILE printf("msg.%s.%s_val);\n", args[i], args[i]) >> CFILE } } printf("\tif (replyp == NULL) {\n") >> CFILE if (!env_handle) { printf("\t\t__db_err(dbenv, ") >> CFILE printf("clnt_sperror(cl, \"Berkeley DB\"));\n") >> CFILE } else { printf("\t\t__db_err(%s, ", args[env_idx]) >> CFILE printf("clnt_sperror(cl, \"Berkeley DB\"));\n") >> CFILE } printf("\t\tret = DB_NOSERVER;\n") >> CFILE printf("\t\tgoto out;\n") >> CFILE printf("\t}\n") >> CFILE if (ret_code == 0) { printf("\tret = replyp->status;\n") >> CFILE } else { printf("\tret = __dbcl_%s_ret(", name) >> CFILE sep = ""; for (i = 0; i < nvars; ++i) { printf("%s%s", sep, args[i]) >> CFILE sep = ", "; } printf("%sreplyp);\n", sep) >> CFILE } printf("out:\n") >> CFILE # # Free reply if there was one. # printf("\tif (replyp != NULL)\n") >> CFILE printf("\t\txdr_free((xdrproc_t)xdr___%s_reply,",name) >> CFILE printf(" (void *)replyp);\n") >> CFILE printf("\treturn (ret);\n") >> CFILE printf("}\n\n") >> CFILE # # Generate Client Template code # if (ret_code) { # # If we are doing a list, write prototypes # pi = 1; p[pi++] = sprintf("int __dbcl_%s_ret __P((", name); p[pi++] = ""; for (i = 0; i < nvars; ++i) { p[pi++] = pr_type[i]; p[pi++] = ", "; } p[pi++] = sprintf("__%s_reply *));", name); p[pi++] = ""; proto_format(p, 0, TFILE); printf("int\n") >> TFILE printf("__dbcl_%s_ret(", name) >> TFILE sep = ""; for (i = 0; i < nvars; ++i) { printf("%s%s", sep, args[i]) >> TFILE sep = ", "; } printf("%sreplyp)\n",sep) >> TFILE for (i = 0; i < nvars; ++i) if (func_arg[i] == 0) printf("\t%s %s;\n", c_type[i], args[i]) \ >> TFILE else printf("\t%s;\n", c_type[i]) >> TFILE printf("\t__%s_reply *replyp;\n", name) >> TFILE; printf("{\n") >> TFILE printf("\tint ret;\n") >> TFILE # # Local vars in template # for (i = 0; i < rvars; ++i) { if (ret_type[i] == "ID" || ret_type[i] == "STRING" || ret_type[i] == "INT" || ret_type[i] == "DBL") { printf("\t%s %s;\n", \ retc_type[i], retargs[i]) >> TFILE } else if (ret_type[i] == "LIST") { if (retlist_type[i] == "GID") printf("\tu_int8_t *__db_%s;\n", \ retargs[i]) >> TFILE if (retlist_type[i] == "ID" || retlist_type[i] == "INT") printf("\tu_int32_t *__db_%s;\n", \ retargs[i]) >> TFILE } else { printf("\t/* %s %s; */\n", \ ret_type[i], retargs[i]) >> TFILE } } # # Client return code # printf("\n") >> TFILE printf("\tif (replyp->status != 0)\n") >> TFILE printf("\t\treturn (replyp->status);\n") >> TFILE for (i = 0; i < rvars; ++i) { varname = ""; if (ret_type[i] == "ID") { varname = sprintf("%scl_id", retargs[i]); } if (ret_type[i] == "STRING") { varname = retargs[i]; } if (ret_type[i] == "INT" || ret_type[i] == "DBL") { varname = retargs[i]; } if (ret_type[i] == "DBT") { varname = sprintf("%sdata", retargs[i]); } if (ret_type[i] == "ID" || ret_type[i] == "STRING" || ret_type[i] == "INT" || ret_type[i] == "DBL") { printf("\t%s = replyp->%s;\n", \ retargs[i], varname) >> TFILE } else if (ret_type[i] == "LIST") { printf("\n\t/*\n") >> TFILE printf("\t * XXX Handle list\n") >> TFILE printf("\t */\n\n") >> TFILE } else { printf("\t/* Handle replyp->%s; */\n", \ varname) >> TFILE } } printf("\n\t/*\n\t * XXX Code goes here\n\t */\n\n") >> TFILE printf("\treturn (replyp->status);\n") >> TFILE printf("}\n\n") >> TFILE }}## split_lines --# Add line separators to pretty-print the output.function split_lines() { if (argcount > 3) { # Reset the counter, remove any trailing whitespace from # the separator. argcount = 0; sub("[ ]$", "", sep) printf("%s\n\t\t", sep) >> PFILE printf("%s\\\n\\\t\\\t", sep) >> SEDFILE }}# proto_format --# Pretty-print a function prototype.function proto_format(p, sedfile, OUTPUT){ if (sedfile) printf("/*\\\n") >> OUTPUT; else printf("/*\n") >> OUTPUT; s = ""; for (i = 1; i in p; ++i) s = s p[i]; if (sedfile) t = "\\ * PUBLIC: " else t = " * PUBLIC: " if (length(s) + length(t) < 80) if (sedfile) printf("%s%s", t, s) >> OUTPUT; else printf("%s%s", t, s) >> OUTPUT; else { split(s, p, "__P"); len = length(t) + length(p[1]); printf("%s%s", t, p[1]) >> OUTPUT n = split(p[2], comma, ","); comma[1] = "__P" comma[1]; for (i = 1; i <= n; i++) { if (len + length(comma[i]) > 75) { if (sedfile) printf(\ "\\\n\\ * PUBLIC: ") >> OUTPUT; else printf("\n * PUBLIC: ") >> OUTPUT; len = 0; } printf("%s%s", comma[i], i == n ? "" : ",") >> OUTPUT; len += length(comma[i]); } } if (sedfile) printf("\\\n\\ */\\\n") >> OUTPUT; else printf("\n */\n") >> OUTPUT; delete p;}
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -