pcp/bindings/py/pypcp/dll.py

#
#    This file is part of Pretty Curved Privacy (pcp1).
#
#    Copyright (C) 2013-2015 T. von Dein.
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU General Public License as published by
#    the Free Software Foundation, either version 3 of the License, or
#    (at your option) any later version.
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU General Public License for more details.
#
#    You should have received a copy of the GNU General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
#    You can contact me by mail: <tlinden AT cpan DOT org>.
#

import os
from raw import *
from static import *

from cffi import FFI

# from:
# https://gist.github.com/inactivist/4ef7058c2132fa16759d
# with fixes

def __convert_struct_field( s, fields ):
    for field,fieldtype in fields:
        if fieldtype.type.kind == 'primitive':
            yield (field,getattr( s, field ))
        else:
            yield (field, convert_to_python( getattr( s, field ) ))

def convert_to_python(s):
    type=ffi.typeof(s)
    if type.item.kind == 'struct':
        return dict(__convert_struct_field( s, type.item.fields ) )
    elif type.kind == 'array':
        if type.item.kind == 'primitive':
            if type.item.cname == 'char':
                return ffi.string(s)
            else:
                return [ s[i] for i in range(type.length) ]
        else:
            return [ convert_to_python(s[i]) for i in range(type.length) ]
    elif type.kind == 'primitive':
        return int(s)

ffi = FFI()
libso = ''

if 'PCPCP_MAKE_TEST' in os.environ:
    libso = "../libpcp/.libs/libpcp1.so.0"
else:
    libso = "libpcp1.so.0"

libpcp = ffi.dlopen(libso)

ffi.cdef("%s\n%s\n" % (STATIC, PCP_RAW_CODE))