tcl_util.c
上传用户:romrleung
上传日期:2022-05-23
资源大小:18897k
文件大小:9k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. /*-
  2.  * See the file LICENSE for redistribution information.
  3.  *
  4.  * Copyright (c) 1999-2001
  5.  * Sleepycat Software.  All rights reserved.
  6.  */
  7. #include "db_config.h"
  8. #ifndef lint
  9. static const char revid[] = "$Id: tcl_util.c,v 11.35 2002/08/06 06:21:42 bostic Exp $";
  10. #endif /* not lint */
  11. #ifndef NO_SYSTEM_INCLUDES
  12. #include <sys/types.h>
  13. #include <fcntl.h>
  14. #include <stdlib.h>
  15. #include <string.h>
  16. #include <tcl.h>
  17. #endif
  18. #include "db_int.h"
  19. #include "dbinc/tcl_db.h"
  20. /*
  21.  * Prototypes for procedures defined later in this file:
  22.  */
  23. static int mutex_Cmd __P((ClientData, Tcl_Interp *, int, Tcl_Obj * CONST*));
  24. /*
  25.  * bdb_RandCommand --
  26.  * Implements rand* functions.
  27.  *
  28.  * PUBLIC: int bdb_RandCommand __P((Tcl_Interp *, int, Tcl_Obj * CONST*));
  29.  */
  30. int
  31. bdb_RandCommand(interp, objc, objv)
  32. Tcl_Interp *interp; /* Interpreter */
  33. int objc; /* How many arguments? */
  34. Tcl_Obj *CONST objv[]; /* The argument objects */
  35. {
  36. static char *rcmds[] = {
  37. "rand", "random_int", "srand",
  38. NULL
  39. };
  40. enum rcmds {
  41. RRAND, RRAND_INT, RSRAND
  42. };
  43. long t;
  44. int cmdindex, hi, lo, result, ret;
  45. Tcl_Obj *res;
  46. char msg[MSG_SIZE];
  47. result = TCL_OK;
  48. /*
  49.  * Get the command name index from the object based on the cmds
  50.  * defined above.  This SHOULD NOT fail because we already checked
  51.  * in the 'berkdb' command.
  52.  */
  53. if (Tcl_GetIndexFromObj(interp,
  54.     objv[1], rcmds, "command", TCL_EXACT, &cmdindex) != TCL_OK)
  55. return (IS_HELP(objv[1]));
  56. res = NULL;
  57. switch ((enum rcmds)cmdindex) {
  58. case RRAND:
  59. /*
  60.  * Must be 0 args.  Error if different.
  61.  */
  62. if (objc != 2) {
  63. Tcl_WrongNumArgs(interp, 2, objv, NULL);
  64. return (TCL_ERROR);
  65. }
  66. ret = rand();
  67. res = Tcl_NewIntObj(ret);
  68. break;
  69. case RRAND_INT:
  70. /*
  71.  * Must be 4 args.  Error if different.
  72.  */
  73. if (objc != 4) {
  74. Tcl_WrongNumArgs(interp, 2, objv, "lo hi");
  75. return (TCL_ERROR);
  76. }
  77. result = Tcl_GetIntFromObj(interp, objv[2], &lo);
  78. if (result != TCL_OK)
  79. break;
  80. result = Tcl_GetIntFromObj(interp, objv[3], &hi);
  81. if (result == TCL_OK) {
  82. #ifndef RAND_MAX
  83. #define RAND_MAX 0x7fffffff
  84. #endif
  85. t = rand();
  86. if (t > RAND_MAX) {
  87. snprintf(msg, MSG_SIZE,
  88.     "Max random is higher than %ldn",
  89.     (long)RAND_MAX);
  90. Tcl_SetResult(interp, msg, TCL_VOLATILE);
  91. result = TCL_ERROR;
  92. break;
  93. }
  94. _debug_check();
  95. ret = (int)(((double)t / ((double)(RAND_MAX) + 1)) *
  96.     (hi - lo + 1));
  97. ret += lo;
  98. res = Tcl_NewIntObj(ret);
  99. }
  100. break;
  101. case RSRAND:
  102. /*
  103.  * Must be 1 arg.  Error if different.
  104.  */
  105. if (objc != 3) {
  106. Tcl_WrongNumArgs(interp, 2, objv, "seed");
  107. return (TCL_ERROR);
  108. }
  109. result = Tcl_GetIntFromObj(interp, objv[2], &lo);
  110. if (result == TCL_OK) {
  111. srand((u_int)lo);
  112. res = Tcl_NewIntObj(0);
  113. }
  114. break;
  115. }
  116. /*
  117.  * Only set result if we have a res.  Otherwise, lower
  118.  * functions have already done so.
  119.  */
  120. if (result == TCL_OK && res)
  121. Tcl_SetObjResult(interp, res);
  122. return (result);
  123. }
  124. /*
  125.  *
  126.  * tcl_Mutex --
  127.  * Opens an env mutex.
  128.  *
  129.  * PUBLIC: int tcl_Mutex __P((Tcl_Interp *, int, Tcl_Obj * CONST*, DB_ENV *,
  130.  * PUBLIC:    DBTCL_INFO *));
  131.  */
  132. int
  133. tcl_Mutex(interp, objc, objv, envp, envip)
  134. Tcl_Interp *interp; /* Interpreter */
  135. int objc; /* How many arguments? */
  136. Tcl_Obj *CONST objv[]; /* The argument objects */
  137. DB_ENV *envp; /* Environment pointer */
  138. DBTCL_INFO *envip; /* Info pointer */
  139. {
  140. DBTCL_INFO *ip;
  141. Tcl_Obj *res;
  142. _MUTEX_DATA *md;
  143. int i, mode, nitems, result, ret;
  144. char newname[MSG_SIZE];
  145. md = NULL;
  146. result = TCL_OK;
  147. mode = nitems = ret = 0;
  148. memset(newname, 0, MSG_SIZE);
  149. if (objc != 4) {
  150. Tcl_WrongNumArgs(interp, 2, objv, "mode nitems");
  151. return (TCL_ERROR);
  152. }
  153. result = Tcl_GetIntFromObj(interp, objv[2], &mode);
  154. if (result != TCL_OK)
  155. return (TCL_ERROR);
  156. result = Tcl_GetIntFromObj(interp, objv[3], &nitems);
  157. if (result != TCL_OK)
  158. return (TCL_ERROR);
  159. snprintf(newname, sizeof(newname),
  160.     "%s.mutex%d", envip->i_name, envip->i_envmutexid);
  161. ip = _NewInfo(interp, NULL, newname, I_MUTEX);
  162. if (ip == NULL) {
  163. Tcl_SetResult(interp, "Could not set up info",
  164.     TCL_STATIC);
  165. return (TCL_ERROR);
  166. }
  167. /*
  168.  * Set up mutex.
  169.  */
  170. /*
  171.  * Map in the region.
  172.  *
  173.  * XXX
  174.  * We don't bother doing this "right", i.e., using the shalloc
  175.  * functions, just grab some memory knowing that it's correctly
  176.  * aligned.
  177.  */
  178. _debug_check();
  179. if (__os_calloc(NULL, 1, sizeof(_MUTEX_DATA), &md) != 0)
  180. goto posixout;
  181. md->env = envp;
  182. md->n_mutex = nitems;
  183. md->size = sizeof(_MUTEX_ENTRY) * nitems;
  184. md->reginfo.type = REGION_TYPE_MUTEX;
  185. md->reginfo.id = INVALID_REGION_TYPE;
  186. md->reginfo.mode = mode;
  187. md->reginfo.flags = REGION_CREATE_OK | REGION_JOIN_OK;
  188. if ((ret = __db_r_attach(envp, &md->reginfo, md->size)) != 0)
  189. goto posixout;
  190. md->marray = md->reginfo.addr;
  191. /* Initialize a created region. */
  192. if (F_ISSET(&md->reginfo, REGION_CREATE))
  193. for (i = 0; i < nitems; i++) {
  194. md->marray[i].val = 0;
  195. if ((ret = __db_mutex_init_int(envp,
  196.     &md->marray[i].m, i, 0)) != 0)
  197. goto posixout;
  198. }
  199. R_UNLOCK(envp, &md->reginfo);
  200. /*
  201.  * Success.  Set up return.  Set up new info
  202.  * and command widget for this mutex.
  203.  */
  204. envip->i_envmutexid++;
  205. ip->i_parent = envip;
  206. _SetInfoData(ip, md);
  207. Tcl_CreateObjCommand(interp, newname,
  208.     (Tcl_ObjCmdProc *)mutex_Cmd, (ClientData)md, NULL);
  209. res = Tcl_NewStringObj(newname, strlen(newname));
  210. Tcl_SetObjResult(interp, res);
  211. return (TCL_OK);
  212. posixout:
  213. if (ret > 0)
  214. Tcl_PosixError(interp);
  215. result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret), "mutex");
  216. _DeleteInfo(ip);
  217. if (md != NULL) {
  218. if (md->reginfo.addr != NULL)
  219. (void)__db_r_detach(md->env,
  220.     &md->reginfo, F_ISSET(&md->reginfo, REGION_CREATE));
  221. __os_free(md->env, md);
  222. }
  223. return (result);
  224. }
  225. /*
  226.  * mutex_Cmd --
  227.  * Implements the "mutex" widget.
  228.  */
  229. static int
  230. mutex_Cmd(clientData, interp, objc, objv)
  231. ClientData clientData; /* Mutex handle */
  232. Tcl_Interp *interp; /* Interpreter */
  233. int objc; /* How many arguments? */
  234. Tcl_Obj *CONST objv[]; /* The argument objects */
  235. {
  236. static char *mxcmds[] = {
  237. "close",
  238. "get",
  239. "getval",
  240. "release",
  241. "setval",
  242. NULL
  243. };
  244. enum mxcmds {
  245. MXCLOSE,
  246. MXGET,
  247. MXGETVAL,
  248. MXRELE,
  249. MXSETVAL
  250. };
  251. DB_ENV *dbenv;
  252. DBTCL_INFO *envip, *mpip;
  253. _MUTEX_DATA *mp;
  254. Tcl_Obj *res;
  255. int cmdindex, id, result, newval;
  256. Tcl_ResetResult(interp);
  257. mp = (_MUTEX_DATA *)clientData;
  258. mpip = _PtrToInfo((void *)mp);
  259. envip = mpip->i_parent;
  260. dbenv = envip->i_envp;
  261. result = TCL_OK;
  262. if (mp == NULL) {
  263. Tcl_SetResult(interp, "NULL mp pointer", TCL_STATIC);
  264. return (TCL_ERROR);
  265. }
  266. if (mpip == NULL) {
  267. Tcl_SetResult(interp, "NULL mp info pointer", TCL_STATIC);
  268. return (TCL_ERROR);
  269. }
  270. /*
  271.  * Get the command name index from the object based on the dbcmds
  272.  * defined above.
  273.  */
  274. if (Tcl_GetIndexFromObj(interp,
  275.     objv[1], mxcmds, "command", TCL_EXACT, &cmdindex) != TCL_OK)
  276. return (IS_HELP(objv[1]));
  277. res = NULL;
  278. switch ((enum mxcmds)cmdindex) {
  279. case MXCLOSE:
  280. if (objc != 2) {
  281. Tcl_WrongNumArgs(interp, 1, objv, NULL);
  282. return (TCL_ERROR);
  283. }
  284. _debug_check();
  285. (void)__db_r_detach(mp->env, &mp->reginfo, 0);
  286. res = Tcl_NewIntObj(0);
  287. (void)Tcl_DeleteCommand(interp, mpip->i_name);
  288. _DeleteInfo(mpip);
  289. __os_free(mp->env, mp);
  290. break;
  291. case MXRELE:
  292. /*
  293.  * Check for 1 arg.  Error if different.
  294.  */
  295. if (objc != 3) {
  296. Tcl_WrongNumArgs(interp, 2, objv, "id");
  297. return (TCL_ERROR);
  298. }
  299. result = Tcl_GetIntFromObj(interp, objv[2], &id);
  300. if (result != TCL_OK)
  301. break;
  302. MUTEX_UNLOCK(dbenv, &mp->marray[id].m);
  303. res = Tcl_NewIntObj(0);
  304. break;
  305. case MXGET:
  306. /*
  307.  * Check for 1 arg.  Error if different.
  308.  */
  309. if (objc != 3) {
  310. Tcl_WrongNumArgs(interp, 2, objv, "id");
  311. return (TCL_ERROR);
  312. }
  313. result = Tcl_GetIntFromObj(interp, objv[2], &id);
  314. if (result != TCL_OK)
  315. break;
  316. MUTEX_LOCK(dbenv, &mp->marray[id].m);
  317. res = Tcl_NewIntObj(0);
  318. break;
  319. case MXGETVAL:
  320. /*
  321.  * Check for 1 arg.  Error if different.
  322.  */
  323. if (objc != 3) {
  324. Tcl_WrongNumArgs(interp, 2, objv, "id");
  325. return (TCL_ERROR);
  326. }
  327. result = Tcl_GetIntFromObj(interp, objv[2], &id);
  328. if (result != TCL_OK)
  329. break;
  330. res = Tcl_NewLongObj((long)mp->marray[id].val);
  331. break;
  332. case MXSETVAL:
  333. /*
  334.  * Check for 2 args.  Error if different.
  335.  */
  336. if (objc != 4) {
  337. Tcl_WrongNumArgs(interp, 2, objv, "id val");
  338. return (TCL_ERROR);
  339. }
  340. result = Tcl_GetIntFromObj(interp, objv[2], &id);
  341. if (result != TCL_OK)
  342. break;
  343. result = Tcl_GetIntFromObj(interp, objv[3], &newval);
  344. if (result != TCL_OK)
  345. break;
  346. mp->marray[id].val = newval;
  347. res = Tcl_NewIntObj(0);
  348. break;
  349. }
  350. /*
  351.  * Only set result if we have a res.  Otherwise, lower
  352.  * functions have already done so.
  353.  */
  354. if (result == TCL_OK && res)
  355. Tcl_SetObjResult(interp, res);
  356. return (result);
  357. }