Unnamed

from pypy.rpython.tool import rffi_platform as platformfrom pypy.interpreter.baseobjspace import ObjSpace, W_Rootfrom pypy.rpython.lltypesystem import rffi, lltypefrom pypy.rpython.lltypesystem.rffi import llexternal, CCHARP, ExternalCompilationInfo, CStructPtr, INTfrom pypy.rpython.lltypesystem.lltype import Signed, Ptr, Char, GcStruct, GcArray, OpaqueType, mallocfrom pypy.rpython.lltypesystem.rstr import STRfrom pypy.rpython.lltypesystem import lltypefrom pypy.rpython.rtyper import RPythonTyperfrom pypy.interpreter.baseobjspace import Wrappablefrom pypy.interpreter.typedef import TypeDeffrom pypy.interpreter.gateway import interp2app, ObjSpace, W_Rootfrom pypy.tool.udir import udirimport py, sys_compilation_info_ = ExternalCompilationInfo(                         includes=['gdbm.h'],                         libraries=['gdbm'])c_source = py.code.Source("""     #include <python/Python.h>     #include <string.h>     char * fetch(GDBM_FILE name, char *key)     {         datum data, d_key;         d_key.dsize = strlen(key) + 1;         d_key.dptr = (char *) PyObject_Malloc(d_key.dsize);         if(d_key.dptr == NULL)            return NULL;         strcpy(d_key.dptr, key);         data = gdbm_fetch(name, d_key);         return data.dptr;     }    """)eci = ExternalCompilationInfo(            includes=['gdbm.h'],            libraries=['gdbm'],            include_dirs=[udir],            separate_module_sources=[c_source],        )datum = GcStruct('datum',('dptr',CCHARP), ('dsize', lltype.Signed))err_func = lltype.Ptr(lltype.FuncType([], lltype.Void))GDBM_FILE = rffi.COpaquePtr('GDBM_FILE', compilation_info=_compilation_info_)open_gdbm = rffi.llexternal('gdbm_open', [CCHARP, INT, INT, INT, err_func], GDBM_FILE, compilation_info = _compilation_info_)store_gdbm = rffi.llexternal('gdbm_store', [GDBM_FILE, datum, datum, INT], INT, compilation_info = _compilation_info_)fetch_gdbm = rffi.llexternal('fetch', [GDBM_FILE, CCHARP], CCHARP, compilation_info = eci)close_gdbm = rffi.llexternal('gdbm_close', [GDBM_FILE], INT, compilation_info = _compilation_info_)class GDBM(Wrappable):    def __init__(self, space):        self.space = space    def gdbm_open(self, name, blocksize, read_write, mode):        c_name = rffi.str2charp(name)        self.struct_gdbm = open_gdbm(c_name, blocksize, read_write, mode, 0)        if not self.struct_gdbm:            return False        return True    def gdbm_store(self, key, content, flag):        s = malloc(datum, zero=True)        s.dptr = rffi.str2charp(key)        s.dsize = len(key)        s2 = malloc(datum, zero=True)        s2.dptr = rffi.str2charp(content)        s2.dsize = len(content)        res_gdbm = store_gdbm(self.struct_gdbm, s, s2, flag)        return self.space.wrap(res_gdbm)    def gdbm_fetch(self, key):        c_key = rffi.str2charp(key)        res = fetch_gdbm(self.struct_gdbm, c_key)        str_res = rffi.charp2str(res)        return self.space.wrap(str_res)    def gdbm_close(self):        close_gdbm(self.struct_gdbm)def GDBM_new(space, w_subtype, args= ''):    w_gdbm = space.allocate_instance(GDBM, w_subtype)    gdbm = space.interp_w(GDBM, w_gdbm)    GDBM.__init__(gdbm, space)    return w_gdbmGDBM.typedef = TypeDef(     'GDBMType',     __new__  = interp2app(GDBM_new, unwrap_spec=[ObjSpace, W_Root]),     open     = interp2app(GDBM.gdbm_open, unwrap_spec=['self', str, int, int, int]),     store    = interp2app(GDBM.gdbm_store, unwrap_spec=['self', str, str, int]),     fetch    = interp2app(GDBM.gdbm_fetch, unwrap_spec=['self', str]),     close    = interp2app(GDBM.gdbm_open, unwrap_spec=['self']))