| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | /* GDBM module using dictionary interface */ | 
					
						
							| 
									
										
										
										
											1998-03-03 22:02:24 +00:00
										 |  |  | /* Author: Anthony Baxter, after dbmmodule.c */ | 
					
						
							|  |  |  | /* Doc strings: Mitch Chapman */ | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-03-20 19:01:55 +09:00
										 |  |  | #define PY_SSIZE_T_CLEAN
 | 
					
						
							| 
									
										
										
										
											1996-12-17 19:55:33 +00:00
										 |  |  | #include "Python.h"
 | 
					
						
							| 
									
										
										
										
											2021-10-13 19:25:53 +02:00
										 |  |  | #include "gdbm.h"
 | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | #include <fcntl.h>
 | 
					
						
							| 
									
										
										
										
											2021-10-13 19:25:53 +02:00
										 |  |  | #include <stdlib.h>               // free()
 | 
					
						
							|  |  |  | #include <sys/stat.h>
 | 
					
						
							|  |  |  | #include <sys/types.h>
 | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2001-02-28 16:44:18 +00:00
										 |  |  | #if defined(WIN32) && !defined(__CYGWIN__)
 | 
					
						
							| 
									
										
										
										
											1998-10-03 05:13:27 +00:00
										 |  |  | #include "gdbmerrno.h"
 | 
					
						
							|  |  |  | extern const char * gdbm_strerror(gdbm_error); | 
					
						
							|  |  |  | #endif
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | typedef struct { | 
					
						
							|  |  |  |     PyTypeObject *gdbm_type; | 
					
						
							|  |  |  |     PyObject *gdbm_error; | 
					
						
							|  |  |  | } _gdbm_state; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static inline _gdbm_state* | 
					
						
							|  |  |  | get_gdbm_state(PyObject *module) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  |     void *state = PyModule_GetState(module); | 
					
						
							|  |  |  |     assert(state != NULL); | 
					
						
							|  |  |  |     return (_gdbm_state *)state; | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  | /*[clinic input]
 | 
					
						
							|  |  |  | module _gdbm | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | class _gdbm.gdbm "gdbmobject *" "&Gdbmtype" | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  | [clinic start generated code]*/ | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | /*[clinic end generated code: output=da39a3ee5e6b4b0d input=38ae71cedfc7172b]*/ | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2002-06-13 20:33:02 +00:00
										 |  |  | PyDoc_STRVAR(gdbmmodule__doc__, | 
					
						
							|  |  |  | "This module provides an interface to the GNU DBM (GDBM) library.\n\
 | 
					
						
							| 
									
										
										
										
											1998-03-03 22:02:24 +00:00
										 |  |  | \n\ | 
					
						
							|  |  |  | This module is quite similar to the dbm module, but uses GDBM instead to\n\ | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  | provide some additional functionality.  Please note that the file formats\n\ | 
					
						
							|  |  |  | created by GDBM and dbm are incompatible.\n\ | 
					
						
							| 
									
										
										
										
											1998-03-03 22:02:24 +00:00
										 |  |  | \n\ | 
					
						
							|  |  |  | GDBM objects behave like mappings (dictionaries), except that keys and\n\ | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  | values are always immutable bytes-like objects or strings.  Printing\n\ | 
					
						
							|  |  |  | a GDBM object doesn't print the keys and values, and the items() and\n\ | 
					
						
							|  |  |  | values() methods are not supported."); | 
					
						
							| 
									
										
										
										
											1998-03-03 22:02:24 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | typedef struct { | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |     PyObject_HEAD | 
					
						
							| 
									
										
										
										
											2020-05-01 21:15:35 +09:00
										 |  |  |     Py_ssize_t di_size;        /* -1 means recompute */ | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |     GDBM_FILE di_dbm; | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | } gdbmobject; | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  | #include "clinic/_gdbmmodule.c.h"
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | #define check_gdbmobject_open(v, err)                                 \
 | 
					
						
							|  |  |  |     if ((v)->di_dbm == NULL) {                                       \ | 
					
						
							|  |  |  |         PyErr_SetString(err, "GDBM object has already been closed"); \ | 
					
						
							|  |  |  |         return NULL;                                                 \ | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2002-06-13 20:33:02 +00:00
										 |  |  | PyDoc_STRVAR(gdbm_object__doc__, | 
					
						
							|  |  |  | "This object represents a GDBM database.\n\
 | 
					
						
							| 
									
										
										
										
											1998-03-03 22:02:24 +00:00
										 |  |  | GDBM objects behave like mappings (dictionaries), except that keys and\n\ | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  | values are always immutable bytes-like objects or strings.  Printing\n\ | 
					
						
							|  |  |  | a GDBM object doesn't print the keys and values, and the items() and\n\ | 
					
						
							|  |  |  | values() methods are not supported.\n\ | 
					
						
							| 
									
										
										
										
											1998-03-03 22:02:24 +00:00
										 |  |  | \n\ | 
					
						
							|  |  |  | GDBM objects also support additional operations such as firstkey,\n\ | 
					
						
							| 
									
										
										
										
											2002-06-13 20:33:02 +00:00
										 |  |  | nextkey, reorganize, and sync."); | 
					
						
							| 
									
										
										
										
											1998-03-03 22:02:24 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											1996-12-17 19:55:33 +00:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | newgdbmobject(_gdbm_state *state, const char *file, int flags, int mode) | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2021-05-27 09:29:00 +02:00
										 |  |  |     gdbmobject *dp = PyObject_GC_New(gdbmobject, state->gdbm_type); | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     if (dp == NULL) { | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |         return NULL; | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     } | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |     dp->di_size = -1; | 
					
						
							|  |  |  |     errno = 0; | 
					
						
							| 
									
										
										
										
											2021-05-27 09:29:00 +02:00
										 |  |  |     PyObject_GC_Track(dp); | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  |     if ((dp->di_dbm = gdbm_open((char *)file, 0, flags, mode, NULL)) == 0) { | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |         if (errno != 0) { | 
					
						
							|  |  |  |             PyErr_SetFromErrnoWithFilename(state->gdbm_error, file); | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         else { | 
					
						
							|  |  |  |             PyErr_SetString(state->gdbm_error, gdbm_strerror(gdbm_errno)); | 
					
						
							|  |  |  |         } | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |         Py_DECREF(dp); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     return (PyObject *)dp; | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* Methods */ | 
					
						
							| 
									
										
										
										
											2021-05-27 09:29:00 +02:00
										 |  |  | static int | 
					
						
							|  |  |  | gdbm_traverse(gdbmobject *dp, visitproc visit, void *arg) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  |     Py_VISIT(Py_TYPE(dp)); | 
					
						
							|  |  |  |     return 0; | 
					
						
							|  |  |  | } | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | static void | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | gdbm_dealloc(gdbmobject *dp) | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2021-05-27 09:29:00 +02:00
										 |  |  |     PyObject_GC_UnTrack(dp); | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     if (dp->di_dbm) { | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |         gdbm_close(dp->di_dbm); | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     } | 
					
						
							|  |  |  |     PyTypeObject *tp = Py_TYPE(dp); | 
					
						
							|  |  |  |     tp->tp_free(dp); | 
					
						
							|  |  |  |     Py_DECREF(tp); | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2006-02-15 17:27:45 +00:00
										 |  |  | static Py_ssize_t | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | gdbm_length(gdbmobject *dp) | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     _gdbm_state *state = PyType_GetModuleState(Py_TYPE(dp)); | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |     if (dp->di_dbm == NULL) { | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |         PyErr_SetString(state->gdbm_error, "GDBM object has already been closed"); | 
					
						
							| 
									
										
										
										
											2007-05-23 20:51:02 +00:00
										 |  |  |         return -1; | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |     } | 
					
						
							|  |  |  |     if (dp->di_size < 0) { | 
					
						
							| 
									
										
										
										
											2020-05-01 21:15:35 +09:00
										 |  |  | #if GDBM_VERSION_MAJOR >= 1 && GDBM_VERSION_MINOR >= 11
 | 
					
						
							|  |  |  |         errno = 0; | 
					
						
							|  |  |  |         gdbm_count_t count; | 
					
						
							|  |  |  |         if (gdbm_count(dp->di_dbm, &count) == -1) { | 
					
						
							|  |  |  |             if (errno != 0) { | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |                 PyErr_SetFromErrno(state->gdbm_error); | 
					
						
							| 
									
										
										
										
											2020-05-01 21:15:35 +09:00
										 |  |  |             } | 
					
						
							|  |  |  |             else { | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |                 PyErr_SetString(state->gdbm_error, gdbm_strerror(gdbm_errno)); | 
					
						
							| 
									
										
										
										
											2020-05-01 21:15:35 +09:00
										 |  |  |             } | 
					
						
							|  |  |  |             return -1; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         if (count > PY_SSIZE_T_MAX) { | 
					
						
							|  |  |  |             PyErr_SetString(PyExc_OverflowError, "count exceeds PY_SSIZE_T_MAX"); | 
					
						
							|  |  |  |             return -1; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         dp->di_size = count; | 
					
						
							|  |  |  | #else
 | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |         datum key,okey; | 
					
						
							|  |  |  |         okey.dsize=0; | 
					
						
							| 
									
										
										
										
											2006-03-01 22:54:36 +00:00
										 |  |  |         okey.dptr=NULL; | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-05-01 21:15:35 +09:00
										 |  |  |         Py_ssize_t size = 0; | 
					
						
							|  |  |  |         for (key = gdbm_firstkey(dp->di_dbm); key.dptr; | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |              key = gdbm_nextkey(dp->di_dbm,okey)) { | 
					
						
							|  |  |  |             size++; | 
					
						
							| 
									
										
										
										
											2020-05-01 21:15:35 +09:00
										 |  |  |             if (okey.dsize) { | 
					
						
							|  |  |  |                 free(okey.dptr); | 
					
						
							|  |  |  |             } | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |             okey=key; | 
					
						
							| 
									
										
										
										
											1997-07-17 22:56:01 +00:00
										 |  |  |         } | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |         dp->di_size = size; | 
					
						
							| 
									
										
										
										
											2020-05-01 21:15:35 +09:00
										 |  |  | #endif
 | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |     } | 
					
						
							|  |  |  |     return dp->di_size; | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-09-08 19:32:40 -07:00
										 |  |  | static int | 
					
						
							|  |  |  | gdbm_bool(gdbmobject *dp) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  |     _gdbm_state *state = PyType_GetModuleState(Py_TYPE(dp)); | 
					
						
							|  |  |  |     if (dp->di_dbm == NULL) { | 
					
						
							|  |  |  |         PyErr_SetString(state->gdbm_error, "GDBM object has already been closed"); | 
					
						
							|  |  |  |         return -1; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     if (dp->di_size > 0) { | 
					
						
							|  |  |  |         /* Known non-zero size. */ | 
					
						
							|  |  |  |         return 1; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     if (dp->di_size == 0) { | 
					
						
							|  |  |  |         /* Known zero size. */ | 
					
						
							|  |  |  |         return 0; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     /* Unknown size.  Ensure DBM object has an entry. */ | 
					
						
							|  |  |  |     datum key = gdbm_firstkey(dp->di_dbm); | 
					
						
							|  |  |  |     if (key.dptr == NULL) { | 
					
						
							|  |  |  |         /* Empty. Cache this fact. */ | 
					
						
							|  |  |  |         dp->di_size = 0; | 
					
						
							|  |  |  |         return 0; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* Non-empty. Don't cache the length since we don't know. */ | 
					
						
							|  |  |  |     free(key.dptr); | 
					
						
							|  |  |  |     return 1; | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-03-20 19:01:55 +09:00
										 |  |  | // Wrapper function for PyArg_Parse(o, "s#", &d.dptr, &d.size).
 | 
					
						
							|  |  |  | // This function is needed to support PY_SSIZE_T_CLEAN.
 | 
					
						
							|  |  |  | // Return 1 on success, same to PyArg_Parse().
 | 
					
						
							|  |  |  | static int | 
					
						
							|  |  |  | parse_datum(PyObject *o, datum *d, const char *failmsg) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  |     Py_ssize_t size; | 
					
						
							|  |  |  |     if (!PyArg_Parse(o, "s#", &d->dptr, &size)) { | 
					
						
							|  |  |  |         if (failmsg != NULL) { | 
					
						
							|  |  |  |             PyErr_SetString(PyExc_TypeError, failmsg); | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         return 0; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     if (INT_MAX < size) { | 
					
						
							|  |  |  |         PyErr_SetString(PyExc_OverflowError, "size does not fit in an int"); | 
					
						
							|  |  |  |         return 0; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     d->dsize = size; | 
					
						
							|  |  |  |     return 1; | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											1996-12-17 19:55:33 +00:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | gdbm_subscript(gdbmobject *dp, PyObject *key) | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |     PyObject *v; | 
					
						
							|  |  |  |     datum drec, krec; | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     _gdbm_state *state = PyType_GetModuleState(Py_TYPE(dp)); | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-03-20 19:01:55 +09:00
										 |  |  |     if (!parse_datum(key, &krec, NULL)) { | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |         return NULL; | 
					
						
							| 
									
										
										
										
											2019-03-20 19:01:55 +09:00
										 |  |  |     } | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |     if (dp->di_dbm == NULL) { | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |         PyErr_SetString(state->gdbm_error, | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |                         "GDBM object has already been closed"); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     drec = gdbm_fetch(dp->di_dbm, krec); | 
					
						
							|  |  |  |     if (drec.dptr == 0) { | 
					
						
							| 
									
										
										
										
											2007-05-23 20:51:02 +00:00
										 |  |  |         PyErr_SetObject(PyExc_KeyError, key); | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2008-05-26 13:28:38 +00:00
										 |  |  |     v = PyBytes_FromStringAndSize(drec.dptr, drec.dsize); | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |     free(drec.dptr); | 
					
						
							|  |  |  |     return v; | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  | /*[clinic input]
 | 
					
						
							|  |  |  | _gdbm.gdbm.get | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     key: object | 
					
						
							|  |  |  |     default: object = None | 
					
						
							|  |  |  |     / | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | Get the value for key, or default if not present. | 
					
						
							|  |  |  | [clinic start generated code]*/ | 
					
						
							| 
									
										
										
										
											2010-12-04 09:14:36 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | _gdbm_gdbm_get_impl(gdbmobject *self, PyObject *key, PyObject *default_value) | 
					
						
							|  |  |  | /*[clinic end generated code: output=92421838f3a852f4 input=a9c20423f34c17b6]*/ | 
					
						
							| 
									
										
										
										
											2010-12-04 09:14:36 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  |     PyObject *res; | 
					
						
							| 
									
										
										
										
											2010-12-04 09:14:36 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     res = gdbm_subscript(self, key); | 
					
						
							| 
									
										
										
										
											2010-12-04 09:14:36 +00:00
										 |  |  |     if (res == NULL && PyErr_ExceptionMatches(PyExc_KeyError)) { | 
					
						
							|  |  |  |         PyErr_Clear(); | 
					
						
							| 
									
										
										
										
											2022-11-14 13:08:15 +01:00
										 |  |  |         return Py_NewRef(default_value); | 
					
						
							| 
									
										
										
										
											2010-12-04 09:14:36 +00:00
										 |  |  |     } | 
					
						
							|  |  |  |     return res; | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | static int | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | gdbm_ass_sub(gdbmobject *dp, PyObject *v, PyObject *w) | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |     datum krec, drec; | 
					
						
							| 
									
										
										
										
											2019-03-20 19:01:55 +09:00
										 |  |  |     const char *failmsg = "gdbm mappings have bytes or string indices only"; | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     _gdbm_state *state = PyType_GetModuleState(Py_TYPE(dp)); | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-03-20 19:01:55 +09:00
										 |  |  |     if (!parse_datum(v, &krec, failmsg)) { | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |         return -1; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     if (dp->di_dbm == NULL) { | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |         PyErr_SetString(state->gdbm_error, | 
					
						
							| 
									
										
										
										
											2007-05-23 20:51:02 +00:00
										 |  |  |                         "GDBM object has already been closed"); | 
					
						
							|  |  |  |         return -1; | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |     } | 
					
						
							|  |  |  |     dp->di_size = -1; | 
					
						
							|  |  |  |     if (w == NULL) { | 
					
						
							|  |  |  |         if (gdbm_delete(dp->di_dbm, krec) < 0) { | 
					
						
							| 
									
										
										
										
											2018-12-12 20:46:55 +08:00
										 |  |  |             if (gdbm_errno == GDBM_ITEM_NOT_FOUND) { | 
					
						
							|  |  |  |                 PyErr_SetObject(PyExc_KeyError, v); | 
					
						
							|  |  |  |             } | 
					
						
							|  |  |  |             else { | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |                 PyErr_SetString(state->gdbm_error, gdbm_strerror(gdbm_errno)); | 
					
						
							| 
									
										
										
										
											2018-12-12 20:46:55 +08:00
										 |  |  |             } | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |             return -1; | 
					
						
							| 
									
										
										
										
											1997-07-17 22:56:01 +00:00
										 |  |  |         } | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |     } | 
					
						
							|  |  |  |     else { | 
					
						
							| 
									
										
										
										
											2019-03-20 19:01:55 +09:00
										 |  |  |         if (!parse_datum(w, &drec, failmsg)) { | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |             return -1; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         errno = 0; | 
					
						
							|  |  |  |         if (gdbm_store(dp->di_dbm, krec, drec, GDBM_REPLACE) < 0) { | 
					
						
							|  |  |  |             if (errno != 0) | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |                 PyErr_SetFromErrno(state->gdbm_error); | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |             else | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |                 PyErr_SetString(state->gdbm_error, | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |                                 gdbm_strerror(gdbm_errno)); | 
					
						
							|  |  |  |             return -1; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     return 0; | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  | /*[clinic input]
 | 
					
						
							|  |  |  | _gdbm.gdbm.setdefault | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     key: object | 
					
						
							|  |  |  |     default: object = None | 
					
						
							|  |  |  |     / | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | Get value for key, or set it to default and return default if not present. | 
					
						
							|  |  |  | [clinic start generated code]*/ | 
					
						
							| 
									
										
										
										
											2010-12-04 09:14:36 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | _gdbm_gdbm_setdefault_impl(gdbmobject *self, PyObject *key, | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  |                            PyObject *default_value) | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | /*[clinic end generated code: output=f3246e880509f142 input=0db46b69e9680171]*/ | 
					
						
							| 
									
										
										
										
											2010-12-04 09:14:36 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  |     PyObject *res; | 
					
						
							| 
									
										
										
										
											2010-12-04 09:14:36 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     res = gdbm_subscript(self, key); | 
					
						
							| 
									
										
										
										
											2010-12-04 09:14:36 +00:00
										 |  |  |     if (res == NULL && PyErr_ExceptionMatches(PyExc_KeyError)) { | 
					
						
							|  |  |  |         PyErr_Clear(); | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |         if (gdbm_ass_sub(self, key, default_value) < 0) | 
					
						
							| 
									
										
										
										
											2010-12-04 09:14:36 +00:00
										 |  |  |             return NULL; | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |         return gdbm_subscript(self, key); | 
					
						
							| 
									
										
										
										
											2010-12-04 09:14:36 +00:00
										 |  |  |     } | 
					
						
							|  |  |  |     return res; | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  | /*[clinic input]
 | 
					
						
							|  |  |  | _gdbm.gdbm.close | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | Close the database. | 
					
						
							|  |  |  | [clinic start generated code]*/ | 
					
						
							| 
									
										
										
										
											1998-03-03 22:02:24 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											1996-12-17 19:55:33 +00:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | _gdbm_gdbm_close_impl(gdbmobject *self) | 
					
						
							|  |  |  | /*[clinic end generated code: output=f5abb4d6bb9e52d5 input=0a203447379b45fd]*/ | 
					
						
							| 
									
										
										
										
											1995-07-07 22:37:11 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     if (self->di_dbm) { | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  |         gdbm_close(self->di_dbm); | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     } | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  |     self->di_dbm = NULL; | 
					
						
							| 
									
										
										
										
											2017-01-23 09:47:21 +02:00
										 |  |  |     Py_RETURN_NONE; | 
					
						
							| 
									
										
										
										
											1995-07-07 22:37:11 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2007-05-23 20:51:02 +00:00
										 |  |  | /* XXX Should return a set or a set view */ | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  | /*[clinic input]
 | 
					
						
							|  |  |  | _gdbm.gdbm.keys | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     cls: defining_class | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  | Get a list of all keys in the database. | 
					
						
							|  |  |  | [clinic start generated code]*/ | 
					
						
							| 
									
										
										
										
											1998-03-03 22:02:24 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											1996-12-17 19:55:33 +00:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | _gdbm_gdbm_keys_impl(gdbmobject *self, PyTypeObject *cls) | 
					
						
							|  |  |  | /*[clinic end generated code: output=c24b824e81404755 input=1428b7c79703d7d5]*/ | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2013-08-13 20:18:52 +02:00
										 |  |  |     PyObject *v, *item; | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |     datum key, nextkey; | 
					
						
							|  |  |  |     int err; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     _gdbm_state *state = PyType_GetModuleState(cls); | 
					
						
							|  |  |  |     assert(state != NULL); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if (self == NULL || !Py_IS_TYPE(self, state->gdbm_type)) { | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |         PyErr_BadInternalCall(); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     check_gdbmobject_open(self, state->gdbm_error); | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  |     v = PyList_New(0); | 
					
						
							|  |  |  |     if (v == NULL) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  |     key = gdbm_firstkey(self->di_dbm); | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |     while (key.dptr) { | 
					
						
							| 
									
										
										
										
											2008-05-26 13:28:38 +00:00
										 |  |  |         item = PyBytes_FromStringAndSize(key.dptr, key.dsize); | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |         if (item == NULL) { | 
					
						
							|  |  |  |             free(key.dptr); | 
					
						
							|  |  |  |             Py_DECREF(v); | 
					
						
							|  |  |  |             return NULL; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         err = PyList_Append(v, item); | 
					
						
							|  |  |  |         Py_DECREF(item); | 
					
						
							|  |  |  |         if (err != 0) { | 
					
						
							|  |  |  |             free(key.dptr); | 
					
						
							|  |  |  |             Py_DECREF(v); | 
					
						
							|  |  |  |             return NULL; | 
					
						
							|  |  |  |         } | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  |         nextkey = gdbm_nextkey(self->di_dbm, key); | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |         free(key.dptr); | 
					
						
							|  |  |  |         key = nextkey; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     return v; | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2006-08-19 23:18:48 +00:00
										 |  |  | static int | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | gdbm_contains(PyObject *self, PyObject *arg) | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     gdbmobject *dp = (gdbmobject *)self; | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |     datum key; | 
					
						
							| 
									
										
										
										
											2013-10-25 00:06:52 +03:00
										 |  |  |     Py_ssize_t size; | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     _gdbm_state *state = PyType_GetModuleState(Py_TYPE(dp)); | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2006-08-19 23:18:48 +00:00
										 |  |  |     if ((dp)->di_dbm == NULL) { | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |         PyErr_SetString(state->gdbm_error, | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |                         "GDBM object has already been closed"); | 
					
						
							|  |  |  |         return -1; | 
					
						
							| 
									
										
										
										
											2006-08-19 23:18:48 +00:00
										 |  |  |     } | 
					
						
							| 
									
										
										
										
											2013-10-25 00:06:52 +03:00
										 |  |  |     if (PyUnicode_Check(arg)) { | 
					
						
							| 
									
										
										
										
											2017-01-22 23:07:07 +02:00
										 |  |  |         key.dptr = (char *)PyUnicode_AsUTF8AndSize(arg, &size); | 
					
						
							| 
									
										
										
										
											2013-10-25 00:06:52 +03:00
										 |  |  |         key.dsize = size; | 
					
						
							|  |  |  |         if (key.dptr == NULL) | 
					
						
							|  |  |  |             return -1; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     else if (!PyBytes_Check(arg)) { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         PyErr_Format(PyExc_TypeError, | 
					
						
							| 
									
										
										
										
											2013-10-25 00:06:52 +03:00
										 |  |  |                      "gdbm key must be bytes or string, not %.100s", | 
					
						
							| 
									
										
										
										
											2020-02-07 03:37:06 +01:00
										 |  |  |                      Py_TYPE(arg)->tp_name); | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         return -1; | 
					
						
							| 
									
										
										
										
											2006-08-19 23:18:48 +00:00
										 |  |  |     } | 
					
						
							| 
									
										
										
										
											2013-10-25 00:06:52 +03:00
										 |  |  |     else { | 
					
						
							|  |  |  |         key.dptr = PyBytes_AS_STRING(arg); | 
					
						
							|  |  |  |         key.dsize = PyBytes_GET_SIZE(arg); | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2006-08-19 23:18:48 +00:00
										 |  |  |     return gdbm_exists(dp->di_dbm, key); | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  | /*[clinic input]
 | 
					
						
							|  |  |  | _gdbm.gdbm.firstkey | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     cls: defining_class | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  | Return the starting key for the traversal. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | It's possible to loop over every key in the database using this method | 
					
						
							|  |  |  | and the nextkey() method.  The traversal is ordered by GDBM's internal | 
					
						
							|  |  |  | hash values, and won't be sorted by the key values. | 
					
						
							|  |  |  | [clinic start generated code]*/ | 
					
						
							| 
									
										
										
										
											1998-03-03 22:02:24 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											1996-12-17 19:55:33 +00:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | _gdbm_gdbm_firstkey_impl(gdbmobject *self, PyTypeObject *cls) | 
					
						
							|  |  |  | /*[clinic end generated code: output=139275e9c8b60827 input=ed8782a029a5d299]*/ | 
					
						
							| 
									
										
										
										
											1995-03-16 16:07:34 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2013-08-13 20:18:52 +02:00
										 |  |  |     PyObject *v; | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |     datum key; | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     _gdbm_state *state = PyType_GetModuleState(cls); | 
					
						
							|  |  |  |     assert(state != NULL); | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     check_gdbmobject_open(self, state->gdbm_error); | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  |     key = gdbm_firstkey(self->di_dbm); | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |     if (key.dptr) { | 
					
						
							| 
									
										
										
										
											2008-05-26 13:28:38 +00:00
										 |  |  |         v = PyBytes_FromStringAndSize(key.dptr, key.dsize); | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |         free(key.dptr); | 
					
						
							|  |  |  |         return v; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     else { | 
					
						
							| 
									
										
										
										
											2017-01-23 09:47:21 +02:00
										 |  |  |         Py_RETURN_NONE; | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |     } | 
					
						
							| 
									
										
										
										
											1995-03-16 16:07:34 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  | /*[clinic input]
 | 
					
						
							|  |  |  | _gdbm.gdbm.nextkey | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     cls: defining_class | 
					
						
							| 
									
										
										
										
											2015-05-07 23:30:09 -07:00
										 |  |  |     key: str(accept={str, robuffer}, zeroes=True) | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  |     / | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | Returns the key that follows key in the traversal. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | The following code prints every key in the database db, without having | 
					
						
							|  |  |  | to create a list in memory that contains them all: | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |       k = db.firstkey() | 
					
						
							| 
									
										
										
										
											2021-08-31 16:59:52 +03:00
										 |  |  |       while k is not None: | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  |           print(k) | 
					
						
							|  |  |  |           k = db.nextkey(k) | 
					
						
							|  |  |  | [clinic start generated code]*/ | 
					
						
							| 
									
										
										
										
											1998-03-03 22:02:24 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											1996-12-17 19:55:33 +00:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | _gdbm_gdbm_nextkey_impl(gdbmobject *self, PyTypeObject *cls, const char *key, | 
					
						
							| 
									
										
										
										
											2021-05-08 10:17:37 +09:00
										 |  |  |                         Py_ssize_t key_length) | 
					
						
							| 
									
										
										
										
											2021-08-31 16:59:52 +03:00
										 |  |  | /*[clinic end generated code: output=c81a69300ef41766 input=365e297bc0b3db48]*/ | 
					
						
							| 
									
										
										
										
											1995-03-16 16:07:34 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2013-08-13 20:18:52 +02:00
										 |  |  |     PyObject *v; | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  |     datum dbm_key, nextkey; | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     _gdbm_state *state = PyType_GetModuleState(cls); | 
					
						
							|  |  |  |     assert(state != NULL); | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  |     dbm_key.dptr = (char *)key; | 
					
						
							|  |  |  |     dbm_key.dsize = key_length; | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     check_gdbmobject_open(self, state->gdbm_error); | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  |     nextkey = gdbm_nextkey(self->di_dbm, dbm_key); | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |     if (nextkey.dptr) { | 
					
						
							| 
									
										
										
										
											2008-05-26 13:28:38 +00:00
										 |  |  |         v = PyBytes_FromStringAndSize(nextkey.dptr, nextkey.dsize); | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |         free(nextkey.dptr); | 
					
						
							|  |  |  |         return v; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     else { | 
					
						
							| 
									
										
										
										
											2017-01-23 09:47:21 +02:00
										 |  |  |         Py_RETURN_NONE; | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |     } | 
					
						
							| 
									
										
										
										
											1995-03-16 16:07:34 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  | /*[clinic input]
 | 
					
						
							|  |  |  | _gdbm.gdbm.reorganize | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     cls: defining_class | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  | Reorganize the database. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | If you have carried out a lot of deletions and would like to shrink | 
					
						
							|  |  |  | the space used by the GDBM file, this routine will reorganize the | 
					
						
							|  |  |  | database.  GDBM will not shorten the length of a database file except | 
					
						
							|  |  |  | by using this reorganization; otherwise, deleted file space will be | 
					
						
							|  |  |  | kept and reused as new (key,value) pairs are added. | 
					
						
							|  |  |  | [clinic start generated code]*/ | 
					
						
							| 
									
										
										
										
											1998-03-03 22:02:24 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											1996-12-17 19:55:33 +00:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | _gdbm_gdbm_reorganize_impl(gdbmobject *self, PyTypeObject *cls) | 
					
						
							|  |  |  | /*[clinic end generated code: output=d77c69e8e3dd644a input=e1359faeef844e46]*/ | 
					
						
							| 
									
										
										
										
											1995-03-16 16:07:34 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     _gdbm_state *state = PyType_GetModuleState(cls); | 
					
						
							|  |  |  |     assert(state != NULL); | 
					
						
							|  |  |  |     check_gdbmobject_open(self, state->gdbm_error); | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |     errno = 0; | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  |     if (gdbm_reorganize(self->di_dbm) < 0) { | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |         if (errno != 0) | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |             PyErr_SetFromErrno(state->gdbm_error); | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |         else | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |             PyErr_SetString(state->gdbm_error, gdbm_strerror(gdbm_errno)); | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2017-01-23 09:47:21 +02:00
										 |  |  |     Py_RETURN_NONE; | 
					
						
							| 
									
										
										
										
											1995-03-16 16:07:34 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  | /*[clinic input]
 | 
					
						
							|  |  |  | _gdbm.gdbm.sync | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     cls: defining_class | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  | Flush the database to the disk file. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | When the database has been opened in fast mode, this method forces | 
					
						
							|  |  |  | any unwritten data to be written to the disk. | 
					
						
							|  |  |  | [clinic start generated code]*/ | 
					
						
							| 
									
										
										
										
											1998-03-03 22:02:24 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											1997-03-25 17:39:56 +00:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | _gdbm_gdbm_sync_impl(gdbmobject *self, PyTypeObject *cls) | 
					
						
							|  |  |  | /*[clinic end generated code: output=bb680a2035c3f592 input=3d749235f79b6f2a]*/ | 
					
						
							| 
									
										
										
										
											1997-03-25 17:39:56 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     _gdbm_state *state = PyType_GetModuleState(cls); | 
					
						
							|  |  |  |     assert(state != NULL); | 
					
						
							|  |  |  |     check_gdbmobject_open(self, state->gdbm_error); | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  |     gdbm_sync(self->di_dbm); | 
					
						
							| 
									
										
										
										
											2017-01-23 09:47:21 +02:00
										 |  |  |     Py_RETURN_NONE; | 
					
						
							| 
									
										
										
										
											1997-03-25 17:39:56 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2013-11-17 15:59:51 +10:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | gdbm__enter__(PyObject *self, PyObject *args) | 
					
						
							| 
									
										
										
										
											2013-11-17 15:59:51 +10:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2022-11-14 13:08:15 +01:00
										 |  |  |     return Py_NewRef(self); | 
					
						
							| 
									
										
										
										
											2013-11-17 15:59:51 +10:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | gdbm__exit__(PyObject *self, PyObject *args) | 
					
						
							| 
									
										
										
										
											2013-11-17 15:59:51 +10:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2022-02-16 18:57:28 +09:00
										 |  |  |     return _gdbm_gdbm_close_impl((gdbmobject *)self); | 
					
						
							| 
									
										
										
										
											2013-11-17 15:59:51 +10:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | static PyMethodDef gdbm_methods[] = { | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  |     _GDBM_GDBM_CLOSE_METHODDEF | 
					
						
							|  |  |  |     _GDBM_GDBM_KEYS_METHODDEF | 
					
						
							|  |  |  |     _GDBM_GDBM_FIRSTKEY_METHODDEF | 
					
						
							|  |  |  |     _GDBM_GDBM_NEXTKEY_METHODDEF | 
					
						
							|  |  |  |     _GDBM_GDBM_REORGANIZE_METHODDEF | 
					
						
							|  |  |  |     _GDBM_GDBM_SYNC_METHODDEF | 
					
						
							|  |  |  |     _GDBM_GDBM_GET_METHODDEF | 
					
						
							|  |  |  |     _GDBM_GDBM_SETDEFAULT_METHODDEF | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     {"__enter__", gdbm__enter__, METH_NOARGS, NULL}, | 
					
						
							|  |  |  |     {"__exit__",  gdbm__exit__, METH_VARARGS, NULL}, | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     {NULL,              NULL}           /* sentinel */ | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | }; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | static PyType_Slot gdbmtype_spec_slots[] = { | 
					
						
							|  |  |  |     {Py_tp_dealloc, gdbm_dealloc}, | 
					
						
							| 
									
										
										
										
											2021-05-27 09:29:00 +02:00
										 |  |  |     {Py_tp_traverse, gdbm_traverse}, | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     {Py_tp_methods, gdbm_methods}, | 
					
						
							|  |  |  |     {Py_sq_contains, gdbm_contains}, | 
					
						
							|  |  |  |     {Py_mp_length, gdbm_length}, | 
					
						
							|  |  |  |     {Py_mp_subscript, gdbm_subscript}, | 
					
						
							|  |  |  |     {Py_mp_ass_subscript, gdbm_ass_sub}, | 
					
						
							| 
									
										
										
										
											2022-09-08 19:32:40 -07:00
										 |  |  |     {Py_nb_bool, gdbm_bool}, | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     {Py_tp_doc, (char*)gdbm_object__doc__}, | 
					
						
							|  |  |  |     {0, 0} | 
					
						
							|  |  |  | }; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static PyType_Spec gdbmtype_spec = { | 
					
						
							|  |  |  |     .name = "_gdbm.gdbm", | 
					
						
							|  |  |  |     .basicsize = sizeof(gdbmobject), | 
					
						
							|  |  |  |     // Calling PyType_GetModuleState() on a subclass is not safe.
 | 
					
						
							|  |  |  |     // dbmtype_spec does not have Py_TPFLAGS_BASETYPE flag
 | 
					
						
							|  |  |  |     // which prevents to create a subclass.
 | 
					
						
							|  |  |  |     // So calling PyType_GetModuleState() in this file is always safe.
 | 
					
						
							| 
									
										
										
										
											2021-05-27 09:29:00 +02:00
										 |  |  |     .flags = (Py_TPFLAGS_DEFAULT | Py_TPFLAGS_DISALLOW_INSTANTIATION | | 
					
						
							| 
									
										
										
										
											2021-06-17 12:06:09 +02:00
										 |  |  |               Py_TPFLAGS_HAVE_GC | Py_TPFLAGS_IMMUTABLETYPE), | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     .slots = gdbmtype_spec_slots, | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | }; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* ----------------------------------------------------------------- */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  | /*[clinic input]
 | 
					
						
							|  |  |  | _gdbm.open as dbmopen | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-09-10 14:26:16 +02:00
										 |  |  |     filename: object | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  |     flags: str="r" | 
					
						
							|  |  |  |     mode: int(py_default="0o666") = 0o666 | 
					
						
							|  |  |  |     / | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | Open a dbm database and return a dbm object. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | The filename argument is the name of the database file. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | The optional flags argument can be 'r' (to open an existing database | 
					
						
							|  |  |  | for reading only -- default), 'w' (to open an existing database for | 
					
						
							|  |  |  | reading and writing), 'c' (which creates the database if it doesn't | 
					
						
							|  |  |  | exist), or 'n' (which always creates a new empty database). | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | Some versions of gdbm support additional flags which must be | 
					
						
							|  |  |  | appended to one of the flags described above.  The module constant | 
					
						
							|  |  |  | 'open_flags' is a string of valid additional flags.  The 'f' flag | 
					
						
							|  |  |  | opens the database in fast mode; altered data will not automatically | 
					
						
							|  |  |  | be written to the disk after every change.  This results in faster | 
					
						
							|  |  |  | writes to the database, but may result in an inconsistent database | 
					
						
							|  |  |  | if the program crashes while the database is still open.  Use the | 
					
						
							|  |  |  | sync() method to force any unwritten data to be written to the disk. | 
					
						
							|  |  |  | The 's' flag causes all database operations to be synchronized to | 
					
						
							|  |  |  | disk.  The 'u' flag disables locking of the database file. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | The optional mode argument is the Unix mode of the file, used only | 
					
						
							|  |  |  | when the database has to be created.  It defaults to octal 0o666. | 
					
						
							|  |  |  | [clinic start generated code]*/ | 
					
						
							| 
									
										
										
										
											1998-03-03 22:02:24 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											1996-12-17 19:55:33 +00:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2018-02-26 16:02:22 +02:00
										 |  |  | dbmopen_impl(PyObject *module, PyObject *filename, const char *flags, | 
					
						
							|  |  |  |              int mode) | 
					
						
							| 
									
										
										
										
											2021-09-10 14:26:16 +02:00
										 |  |  | /*[clinic end generated code: output=9527750f5df90764 input=bca6ec81dc49292c]*/ | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |     int iflags; | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     _gdbm_state *state = get_gdbm_state(module); | 
					
						
							|  |  |  |     assert(state != NULL); | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  |     switch (flags[0]) { | 
					
						
							|  |  |  |     case 'r': | 
					
						
							|  |  |  |         iflags = GDBM_READER; | 
					
						
							|  |  |  |         break; | 
					
						
							|  |  |  |     case 'w': | 
					
						
							|  |  |  |         iflags = GDBM_WRITER; | 
					
						
							|  |  |  |         break; | 
					
						
							|  |  |  |     case 'c': | 
					
						
							|  |  |  |         iflags = GDBM_WRCREAT; | 
					
						
							|  |  |  |         break; | 
					
						
							|  |  |  |     case 'n': | 
					
						
							|  |  |  |         iflags = GDBM_NEWDB; | 
					
						
							|  |  |  |         break; | 
					
						
							|  |  |  |     default: | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |         PyErr_SetString(state->gdbm_error, | 
					
						
							| 
									
										
										
										
											2000-12-17 07:14:13 +00:00
										 |  |  |                         "First flag must be one of 'r', 'w', 'c' or 'n'"); | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2000-12-17 07:14:13 +00:00
										 |  |  |     for (flags++; *flags != '\0'; flags++) { | 
					
						
							|  |  |  |         switch (*flags) { | 
					
						
							|  |  |  | #ifdef GDBM_FAST
 | 
					
						
							|  |  |  |             case 'f': | 
					
						
							|  |  |  |                 iflags |= GDBM_FAST; | 
					
						
							|  |  |  |                 break; | 
					
						
							|  |  |  | #endif
 | 
					
						
							|  |  |  | #ifdef GDBM_SYNC
 | 
					
						
							|  |  |  |             case 's': | 
					
						
							|  |  |  |                 iflags |= GDBM_SYNC; | 
					
						
							|  |  |  |                 break; | 
					
						
							|  |  |  | #endif
 | 
					
						
							|  |  |  | #ifdef GDBM_NOLOCK
 | 
					
						
							|  |  |  |             case 'u': | 
					
						
							|  |  |  |                 iflags |= GDBM_NOLOCK; | 
					
						
							|  |  |  |                 break; | 
					
						
							|  |  |  | #endif
 | 
					
						
							|  |  |  |             default: | 
					
						
							| 
									
										
										
										
											2022-11-14 15:25:34 +02:00
										 |  |  |                 PyErr_Format(state->gdbm_error, | 
					
						
							|  |  |  |                              "Flag '%c' is not supported.", (unsigned char)*flags); | 
					
						
							| 
									
										
										
										
											2000-12-17 07:14:13 +00:00
										 |  |  |                 return NULL; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-09-10 14:26:16 +02:00
										 |  |  |     PyObject *filenamebytes; | 
					
						
							|  |  |  |     if (!PyUnicode_FSConverter(filename, &filenamebytes)) { | 
					
						
							| 
									
										
										
										
											2018-02-26 16:02:22 +02:00
										 |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2021-09-10 14:26:16 +02:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2018-02-26 16:02:22 +02:00
										 |  |  |     const char *name = PyBytes_AS_STRING(filenamebytes); | 
					
						
							|  |  |  |     if (strlen(name) != (size_t)PyBytes_GET_SIZE(filenamebytes)) { | 
					
						
							|  |  |  |         Py_DECREF(filenamebytes); | 
					
						
							|  |  |  |         PyErr_SetString(PyExc_ValueError, "embedded null character"); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     PyObject *self = newgdbmobject(state, name, iflags, mode); | 
					
						
							| 
									
										
										
										
											2018-02-26 16:02:22 +02:00
										 |  |  |     Py_DECREF(filenamebytes); | 
					
						
							|  |  |  |     return self; | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | static const char gdbmmodule_open_flags[] = "rwcn" | 
					
						
							| 
									
										
										
										
											2000-12-17 07:14:13 +00:00
										 |  |  | #ifdef GDBM_FAST
 | 
					
						
							|  |  |  |                                      "f" | 
					
						
							|  |  |  | #endif
 | 
					
						
							|  |  |  | #ifdef GDBM_SYNC
 | 
					
						
							|  |  |  |                                      "s" | 
					
						
							|  |  |  | #endif
 | 
					
						
							|  |  |  | #ifdef GDBM_NOLOCK
 | 
					
						
							|  |  |  |                                      "u" | 
					
						
							|  |  |  | #endif
 | 
					
						
							|  |  |  |                                      ; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | static PyMethodDef _gdbm_module_methods[] = { | 
					
						
							| 
									
										
										
										
											2015-04-17 21:05:18 +03:00
										 |  |  |     DBMOPEN_METHODDEF | 
					
						
							| 
									
										
										
										
											2000-07-08 05:00:07 +00:00
										 |  |  |     { 0, 0 }, | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | }; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | static int | 
					
						
							|  |  |  | _gdbm_exec(PyObject *module) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  |     _gdbm_state *state = get_gdbm_state(module); | 
					
						
							|  |  |  |     state->gdbm_type = (PyTypeObject *)PyType_FromModuleAndSpec(module, | 
					
						
							|  |  |  |                                                         &gdbmtype_spec, NULL); | 
					
						
							|  |  |  |     if (state->gdbm_type == NULL) { | 
					
						
							|  |  |  |         return -1; | 
					
						
							| 
									
										
										
										
											2018-06-19 23:29:22 +02:00
										 |  |  |     } | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     state->gdbm_error = PyErr_NewException("_gdbm.error", PyExc_OSError, NULL); | 
					
						
							|  |  |  |     if (state->gdbm_error == NULL) { | 
					
						
							|  |  |  |         return -1; | 
					
						
							| 
									
										
										
										
											2018-06-19 23:29:22 +02:00
										 |  |  |     } | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     if (PyModule_AddType(module, (PyTypeObject *)state->gdbm_error) < 0) { | 
					
						
							|  |  |  |         return -1; | 
					
						
							| 
									
										
										
										
											2018-06-19 23:29:22 +02:00
										 |  |  |     } | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     if (PyModule_AddStringConstant(module, "open_flags", | 
					
						
							|  |  |  |                                    gdbmmodule_open_flags) < 0) { | 
					
						
							|  |  |  |         return -1; | 
					
						
							| 
									
										
										
										
											2000-12-17 07:14:13 +00:00
										 |  |  |     } | 
					
						
							| 
									
										
										
										
											2018-06-19 23:29:22 +02:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2018-06-20 21:23:30 +08:00
										 |  |  | #if defined(GDBM_VERSION_MAJOR) && defined(GDBM_VERSION_MINOR) && \
 | 
					
						
							|  |  |  |     defined(GDBM_VERSION_PATCH) | 
					
						
							| 
									
										
										
										
											2018-06-19 23:29:22 +02:00
										 |  |  |     PyObject *obj = Py_BuildValue("iii", GDBM_VERSION_MAJOR, | 
					
						
							|  |  |  |                                   GDBM_VERSION_MINOR, GDBM_VERSION_PATCH); | 
					
						
							|  |  |  |     if (obj == NULL) { | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |         return -1; | 
					
						
							| 
									
										
										
										
											2018-06-19 23:29:22 +02:00
										 |  |  |     } | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     if (PyModule_AddObject(module, "_GDBM_VERSION", obj) < 0) { | 
					
						
							| 
									
										
										
										
											2018-06-19 23:29:22 +02:00
										 |  |  |         Py_DECREF(obj); | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |         return -1; | 
					
						
							| 
									
										
										
										
											2018-06-19 23:29:22 +02:00
										 |  |  |     } | 
					
						
							| 
									
										
										
										
											2018-06-20 21:23:30 +08:00
										 |  |  | #endif
 | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  |     return 0; | 
					
						
							|  |  |  | } | 
					
						
							| 
									
										
										
										
											2018-06-19 23:29:22 +02:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | static int | 
					
						
							|  |  |  | _gdbm_module_traverse(PyObject *module, visitproc visit, void *arg) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  |     _gdbm_state *state = get_gdbm_state(module); | 
					
						
							|  |  |  |     Py_VISIT(state->gdbm_error); | 
					
						
							|  |  |  |     Py_VISIT(state->gdbm_type); | 
					
						
							|  |  |  |     return 0; | 
					
						
							|  |  |  | } | 
					
						
							| 
									
										
										
										
											2018-06-19 23:29:22 +02:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-06-17 01:41:23 +09:00
										 |  |  | static int | 
					
						
							|  |  |  | _gdbm_module_clear(PyObject *module) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  |     _gdbm_state *state = get_gdbm_state(module); | 
					
						
							|  |  |  |     Py_CLEAR(state->gdbm_error); | 
					
						
							|  |  |  |     Py_CLEAR(state->gdbm_type); | 
					
						
							|  |  |  |     return 0; | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static void | 
					
						
							|  |  |  | _gdbm_module_free(void *module) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  |     _gdbm_module_clear((PyObject *)module); | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static PyModuleDef_Slot _gdbm_module_slots[] = { | 
					
						
							|  |  |  |     {Py_mod_exec, _gdbm_exec}, | 
					
						
							|  |  |  |     {0, NULL} | 
					
						
							|  |  |  | }; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static struct PyModuleDef _gdbmmodule = { | 
					
						
							|  |  |  |     PyModuleDef_HEAD_INIT, | 
					
						
							|  |  |  |     .m_name = "_gdbm", | 
					
						
							|  |  |  |     .m_doc = gdbmmodule__doc__, | 
					
						
							|  |  |  |     .m_size = sizeof(_gdbm_state), | 
					
						
							|  |  |  |     .m_methods = _gdbm_module_methods, | 
					
						
							|  |  |  |     .m_slots = _gdbm_module_slots, | 
					
						
							|  |  |  |     .m_traverse = _gdbm_module_traverse, | 
					
						
							|  |  |  |     .m_clear = _gdbm_module_clear, | 
					
						
							|  |  |  |     .m_free = _gdbm_module_free, | 
					
						
							|  |  |  | }; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | PyMODINIT_FUNC | 
					
						
							|  |  |  | PyInit__gdbm(void) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  |     return PyModuleDef_Init(&_gdbmmodule); | 
					
						
							| 
									
										
										
										
											1994-08-08 08:06:37 +00:00
										 |  |  | } |