lib/tdb/tdb.i

   1 /*
   2    Unix SMB/CIFS implementation.
   3
   4    Swig interface to tdb.
   5
   6    Copyright (C) 2004-2006 Tim Potter <tpot@samba.org>
   7    Copyright (C) 2007 Jelmer Vernooij <jelmer@samba.org>
   8
   9      ** NOTE! The following LGPL license applies to the tdb
  10      ** library. This does NOT imply that all of Samba is released
  11      ** under the LGPL
  12
  13    This library is free software; you can redistribute it and/or
  14    modify it under the terms of the GNU Lesser General Public
  15    License as published by the Free Software Foundation; either
  16    version 3 of the License, or (at your option) any later version.
  17
  18    This library is distributed in the hope that it will be useful,
  19    but WITHOUT ANY WARRANTY; without even the implied warranty of
  20    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  21    Lesser General Public License for more details.
  22
  23    You should have received a copy of the GNU Lesser General Public
  24    License along with this library; if not, see <http://www.gnu.org/licenses/>.
  25 */
  26
  27 %define DOCSTRING
  28 "TDB is a simple key-value database similar to GDBM that supports multiple writers."
  29 %enddef
  30
  31 %module(docstring=DOCSTRING) tdb
  32
  33 %{
  34
  35 /* This symbol is used in both includes.h and Python.h which causes an
  36    annoying compiler warning. */
  37
  38 #ifdef HAVE_FSTAT
  39 #undef HAVE_FSTAT
  40 #endif
  41
  42 /* Include tdb headers */
  43 #include <stdint.h>
  44 #include <signal.h>
  45 #include <tdb.h>
  46 #include <fcntl.h>
  47
  48 typedef TDB_CONTEXT tdb;
  49 %}
  50
  51 /* The tdb functions will crash if a NULL tdb context is passed */
  52
  53 %import exception.i
  54 %import stdint.i
  55
  56 %typemap(check,noblock=1) TDB_CONTEXT* {
  57         if ($1 == NULL)
  58                 SWIG_exception(SWIG_ValueError,
  59                         "tdb context must be non-NULL");
  60 }
  61
  62 /* In and out typemaps for the TDB_DATA structure.  This is converted to
  63    and from the Python string type which can contain arbitrary binary
  64    data.. */
  65
  66 %typemap(in,noblock=1) TDB_DATA {
  67     if ($input == Py_None) {
  68         $1.dsize = 0;
  69         $1.dptr = NULL;
  70     } else if (!PyString_Check($input)) {
  71                 PyErr_SetString(PyExc_TypeError, "string arg expected");
  72                 return NULL;
  73         } else {
  74         $1.dsize = PyString_Size($input);
  75         $1.dptr = (uint8_t *)PyString_AsString($input);
  76     }
  77 }
  78
  79 %typemap(out,noblock=1) TDB_DATA {
  80         if ($1.dptr == NULL && $1.dsize == 0) {
  81                 $result = Py_None;
  82         } else {
  83                 $result = PyString_FromStringAndSize((const char *)$1.dptr, $1.dsize);
  84                 free($1.dptr);
  85     }
  86 }
  87
  88 /* Treat a mode_t as an unsigned integer */
  89 typedef int mode_t;
  90
  91 /* flags to tdb_store() */
  92 %constant int REPLACE = TDB_REPLACE;
  93 %constant int INSERT = TDB_INSERT;
  94 %constant int MODIFY = TDB_MODIFY;
  95
  96 /* flags for tdb_open() */
  97 %constant int DEFAULT = TDB_DEFAULT;
  98 %constant int CLEAR_IF_FIRST = TDB_CLEAR_IF_FIRST;
  99 %constant int INTERNAL = TDB_INTERNAL;
 100 %constant int NOLOCK = TDB_NOLOCK;
 101 %constant int NOMMAP = TDB_NOMMAP;
 102 %constant int CONVERT = TDB_CONVERT;
 103 %constant int BIGENDIAN = TDB_BIGENDIAN;
 104
 105 enum TDB_ERROR {
 106      TDB_SUCCESS=0,
 107      TDB_ERR_CORRUPT,
 108      TDB_ERR_IO,
 109      TDB_ERR_LOCK,
 110      TDB_ERR_OOM,
 111      TDB_ERR_EXISTS,
 112      TDB_ERR_NOLOCK,
 113      TDB_ERR_LOCK_TIMEOUT,
 114      TDB_ERR_NOEXIST,
 115      TDB_ERR_EINVAL,
 116      TDB_ERR_RDONLY
 117 };
 118
 119 %rename(lock_all) tdb_context::lockall;
 120 %rename(unlock_all) tdb_context::unlockall;
 121
 122 %rename(read_lock_all) tdb_context::lockall_read;
 123 %rename(read_unlock_all) tdb_context::unlockall_read;
 124
 125 %typemap(default,noblock=1) int tdb_flags {
 126     $1 = TDB_DEFAULT;
 127 }
 128
 129 %typemap(default,noblock=1) int flags {
 130     $1 = O_RDWR;
 131 }
 132
 133 %typemap(default,noblock=1) int hash_size {
 134     $1 = 0;
 135 }
 136
 137 %typemap(default,noblock=1) mode_t mode {
 138     $1 = 0600;
 139 }
 140
 141 %typemap(default,noblock=1) int flag {
 142     $1 = TDB_REPLACE;
 143 }
 144
 145 %rename(Tdb) tdb_context;
 146 %feature("docstring") tdb_context "A TDB file.";
 147 %typemap(out,noblock=1) tdb * {
 148     /* Throw an IOError exception from errno if tdb_open() returns NULL */
 149     if ($1 == NULL) {
 150         PyErr_SetFromErrno(PyExc_IOError);
 151         SWIG_fail;
 152     }
 153     $result = SWIG_NewPointerObj($1, $1_descriptor, 0);
 154 }
 155
 156 typedef struct tdb_context {
 157     %extend {
 158         %feature("docstring") tdb "S.__init__(name,hash_size=0,tdb_flags=TDB_DEFAULT,flags=O_RDWR,mode=0600)\n"
 159                                   "Open a TDB file.";
 160         tdb(const char *name, int hash_size, int tdb_flags, int flags, mode_t mode) {
 161             return tdb_open(name, hash_size, tdb_flags, flags, mode);
 162         }
 163         %feature("docstring") error "S.error() -> int\n"
 164                                     "Find last error number returned by operation on this TDB.";
 165         enum TDB_ERROR error();
 166         ~tdb() { tdb_close($self); }
 167         %feature("docstring") close "S.close() -> None\n"
 168                                     "Close the TDB file.";
 169         int close();
 170         int append(TDB_DATA key, TDB_DATA new_dbuf);
 171         %feature("docstring") errorstr "S.errorstr() -> errorstring\n"
 172                                         "Obtain last error message.";
 173         const char *errorstr();
 174         %rename(get) fetch;
 175         %feature("docstring") fetch "S.fetch(key) -> value\n"
 176                                         "Fetch a value.";
 177         TDB_DATA fetch(TDB_DATA key);
 178         %feature("docstring") delete "S.delete(key) -> None\n"
 179                                         "Delete an entry.";
 180         int delete(TDB_DATA key);
 181         %feature("docstring") store "S.store(key, value, flag=TDB_REPLACE) -> None\n"
 182                                         "Store an entry.";
 183         int store(TDB_DATA key, TDB_DATA dbuf, int flag);
 184         %feature("docstring") exists "S.exists(key) -> bool\n"
 185                                         "Check whether key exists in this database.";
 186         int exists(TDB_DATA key);
 187         %feature("docstring") firstkey "S.firstkey() -> data\n"
 188                                         "Return the first key in this database.";
 189         TDB_DATA firstkey();
 190         %feature("docstring") nextkey "S.nextkey(prev) -> data\n"
 191                                         "Return the next key in this database.";
 192         TDB_DATA nextkey(TDB_DATA key);
 193         %feature("docstring") lockall "S.lockall() -> bool";
 194         int lockall();
 195         %feature("docstring") unlockall "S.unlockall() -> bool";
 196         int unlockall();
 197         %feature("docstring") unlockall "S.lockall_read() -> bool";
 198         int lockall_read();
 199         %feature("docstring") unlockall "S.unlockall_read() -> bool";
 200         int unlockall_read();
 201         %feature("docstring") reopen "S.reopen() -> bool\n"
 202                                         "Reopen this file.";
 203         int reopen();
 204         %feature("docstring") transaction_start "S.transaction_start() -> None\n"
 205                                         "Start a new transaction.";
 206         int transaction_start();
 207         %feature("docstring") transaction_commit "S.transaction_commit() -> None\n"
 208                                         "Commit the currently active transaction.";
 209         int transaction_commit();
 210         %feature("docstring") transaction_cancel "S.transaction_cancel() -> None\n"
 211                                         "Cancel the currently active transaction.";
 212         int transaction_cancel();
 213         int transaction_recover();
 214         %feature("docstring") hash_size "S.hash_size() -> int";
 215         int hash_size();
 216         %feature("docstring") map_size "S.map_size() -> int";
 217         size_t map_size();
 218         %feature("docstring") get_flags "S.get_flags() -> int";
 219         int get_flags();
 220         %feature("docstring") set_max_dead "S.set_max_dead(int) -> None";
 221         void set_max_dead(int max_dead);
 222         %feature("docstring") name "S.name() -> path\n" \
 223                                    "Return filename of this TDB file.";
 224         const char *name();
 225     }
 226
 227     %pythoncode {
 228     def __repr__(self):
 229         return "Tdb('%s')" % self.name()
 230
 231     # Random access to keys, values
 232     def __getitem__(self, key):
 233         result = self.get(key)
 234         if result is None:
 235             raise KeyError, '%s: %s' % (key, self.errorstr())
 236         return result
 237
 238     def __setitem__(self, key, item):
 239         if self.store(key, item) == -1:
 240             raise IOError, self.errorstr()
 241
 242     def __delitem__(self, key):
 243         if not self.exists(key):
 244             raise KeyError, '%s: %s' % (key, self.errorstr())
 245         self.delete(key)
 246
 247     def __contains__(self, key):
 248         return self.exists(key) != 0
 249
 250     def has_key(self, key):
 251         return self.exists(key) != 0
 252
 253     def fetch_uint32(self, key):
 254         data = self.get(key)
 255         if data is None:
 256             return None
 257         import struct
 258         return struct.unpack("<L", data)[0]
 259
 260     def fetch_int32(self, key):
 261         data = self.get(key)
 262         if data is None:
 263             return None
 264         import struct
 265         return struct.unpack("<l", data)[0]
 266
 267     # Tdb iterator
 268     class TdbIterator:
 269         def __init__(self, tdb):
 270             self.tdb = tdb
 271             self.key = None
 272
 273         def __iter__(self):
 274             return self
 275
 276         def next(self):
 277             if self.key is None:
 278                 self.key = self.tdb.firstkey()
 279                 if self.key is None:
 280                     raise StopIteration
 281                 return self.key
 282             else:
 283                 self.key = self.tdb.nextkey(self.key)
 284                 if self.key is None:
 285                     raise StopIteration
 286                 return self.key
 287
 288     def __iter__(self):
 289         return self.TdbIterator(self)
 290
 291     # Implement other dict functions using TdbIterator
 292
 293     def keys(self):
 294         return [k for k in iter(self)]
 295
 296     def values(self):
 297         return [self[k] for k in iter(self)]
 298
 299     def items(self):
 300         return [(k, self[k]) for k in iter(self)]
 301
 302     def __len__(self):
 303         return len(self.keys())
 304
 305     def clear(self):
 306         for k in iter(self):
 307             del(self[k])
 308
 309     def iterkeys(self):
 310         for k in iter(self):
 311             yield k
 312
 313     def itervalues(self):
 314         for k in iter(self):
 315             yield self[k]
 316
 317     def iteritems(self):
 318         for k in iter(self):
 319             yield (k, self[k])
 320
 321     # TODO: any other missing methods for container types
 322     }
 323 } tdb;