Files
pcp/bindings/py/pypcp/dll.py
2014-12-24 11:17:03 +01:00

65 lines
1.9 KiB
Python

#
# 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))