Python cffi学习

cffi是连接Python与c的桥梁,可实现在Python中调用c文件。cffi为c语言的外部接口,在Python中使用该接口可以实现在Python中使用外部c文件的数据结构及函数。

由于资料较少,所以对cffi的原理还不是十分熟悉,此处仅从使用进行简单介绍。

所了解的使用方式有以下几种:1)直接在Python文件中通过cffi调用库函数或自定义函数。2)在一个Python文件中进行函数的定义,生成扩展模块后在其他Python中使用定义的函数。3)使用cffi在Python中调用已定义好的函数库中的c文件。

以下分别对这几种方式进行说明:

(一)直接在Python中通过cffi定义c函数并使用

通过二进制文件调用库函数,如:

from cffi import FFI
ffi = FFI()
#cdef用来定义结构体,变量,或者方法的声明
ffi.cdef("""
    int printf(const char *format, ...); //库函数
""")
#dlopen是ABI模式的的基本读取方式
C = ffi.dlopen(None) # 加载整个C命名空间
arg = ffi.new("char[]", b"world") # 等于C代码: char arg[] = "world";
C.printf(b"hello %s!\n", arg)

运行:

定义结构体变量的方式如下:

from cffi import FFI
ffi = FFI()
#cdef用来定义结构体,变量,或者方法的声明
ffi.cdef("""
    typedef struct{
        int x;
        int y;
    } te;
    """) 
test = ffi.new("te *",[1,2]) #定义一个结构体变量并赋值
test = ffi.new("te *", {'x':1, 'y':2}) #另一种赋值方式

也可自定义函数并调用:

from cffi import FFI
ffi = FFI()
ffi.cdef("""
    int add(int a, int b);
    int sub(int a, int b);
""")
#verify是在线api模式的基本方法它里面直接写C代码即可
lib = ffi.verify("""
    int add(int a,int b){
        return a+b;
    }
     int sub(int a,int b){
        return a-b;
    }

""")
print(lib.add(10,2))
print(lib.sub(10,2))

运行:

若c的源码独立成文件,也可以使用ffi.verify(sources=['file.c'])指定,如在当前文件夹下定义一个c文件test.c如下:

#include <stdio.h>

int add(int a, int b)
{
    int c;
    c = a+b;
    return c;
}

void cprint(void)
{
    printf("hello world!\n");
}

int mul(int a,int b)
{
    return a*b;
}

定义test2.py调用该c文件中的函数:

from cffi import FFI
ffi = FFI()
ffi.cdef("""
    int add(int a, int b);
    void cprint(void);
    int mul(int a,int b);
""")
#verify是在线api模式的基本方法它里面直接写C代码即可
lib = ffi.verify(sources=['test.c'])
print(lib.add(10,2))
lib.cprint()
print(lib.mul(3,5))

运行:

其会生成文件__pycache__如下:

(二)生成扩展模块提供给其他Python程序使用

该种方式也称为离线API方式。相关代码参考链接: https://www.jianshu.com/p/4f60eea33d68

首先建立ext_build.py用于生成扩展模块

# ext_build.py
import cffi

ffi = cffi.FFI() #生成cffi实例

ffi.cdef("""int add(int a, int b);""") #函数声明
ffi.cdef("""int sub(int a, int b);""")

ffi.set_source('_ext', """
    int add(int a, int b) //函数定义
    {
        return a + b;
    }
    int sub(int a, int b)
    {
        return a - b;
    }
""")

if __name__ == '__main__':
    ffi.compile(verbose=True)

ffi.cdef表示声明一个函数,ffi.set_source中第一个参数为要生成的扩展模块的名字,第二个参数为函数的具体定义。

执行python ext_build.py,会生成下面三个文件

生成模块_ext用于其他Python文件使用。新建一个test.py使用_ext模块中的c函数:

#test.py
from _ext import ffi, lib #此处import的ffi即为_ext中的ffi

print(lib.add(2, 3))
print(lib.sub(2, 3))

输出:

如此即可在Python文件中使用定义的C函数。

(三)在Python中使用cffi调用外部c库函数

使用cffi还可以在Python中使用外部已经定义好的c库函数。代码参考:https://github.com/wolever/python-cffi-example

对该例子中使用fnmatch的部分代码进行解释,fnmatch用于判断字符串是不是符合pattern所指的结构。需要将库函数中fnmatch函数的声明写入一个头文件中,该例子中建立了一个fnmatch.h,复制函数声明如下,此处根据测试中出现的问题进行了相关修改,头文件中的FNM_PATHNAME参考/usr/include/fnmatch.h进行复制,由于不支持原本的移位运算方式,直接定义成16进制数。

/*
 * Definitions for fnmatch, copy+pasted from <fnmatch.h> with some small
 * cleanups by hand.
 */

/* Bits set in the FLAGS argument to `fnmatch'.  */
#define FNM_PATHNAME    0x1 /* No wildcard can ever match `/'.  */
#define FNM_NOESCAPE    0x10 /* Backslashes don't quote special chars.  */
#define FNM_PERIOD  0x100 /* Leading `.' is matched only explicitly.  */

// /* Value returned by `fnmatch' if STRING does not match PATTERN.  */
 #define    FNM_NOMATCH    1

/* Match NAME against the filename pattern PATTERN,
   returning zero if it matches, FNM_NOMATCH if not.  */
extern int fnmatch (const char *__pattern, const char *__name, int __flags);

编写setup.py,其中cffi_modules制定了需要生成ffi实例的文件

#!/usr/bin/env python

#setup.py
import os
import sys

from setuptools import setup, find_packages

os.chdir(os.path.dirname(sys.argv[0]) or ".")

setup(
    name="cffi-example",
    version="0.1",
    classifiers=[
        "Development Status :: 4 - Beta",
        "Programming Language :: Python :: 2",
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: Implementation :: PyPy",
        "License :: OSI Approved :: BSD License",
    ],
    packages=find_packages(),
    install_requires=["cffi>=1.0.0"],
    setup_requires=["cffi>=1.0.0"],
    cffi_modules=[
        "./cffi_example/build_fnmatch.py:ffi",
    ],
)

 build_fnmatch.py如下,用于生成模块,在fnmatch.py中进行使用

import os

from cffi import FFI

ffi = FFI()

ffi.set_source("cffi_example._fnmatch",
    # Since we are calling fnmatch directly no custom source is necessary. We
    # need to #include <fnmatch.h>, though, because behind the scenes cffi
    # generates a .c file which contains a Python-friendly wrapper around
    # ``fnmatch``:
    #    static PyObject *
    #    _cffi_f_fnmatch(PyObject *self, PyObject *args) {
    #        ... setup ...
    #        result = fnmatch(...);
    #        return PyInt_FromLong(result);
    #    }
    "#include <fnmatch.h>", #此处fnmatch.h为动态库中的头文件
    # The important thing is to inclue libc in the list of libraries we're
    # linking against:
    libraries=["c"],
)

with open(os.path.join(os.path.dirname(__file__), "fnmatch.h")) as f: #此处fnmatch.h为根据动态库中的fnmatch.h自己定义的文件,内容参考动态库中的fnmatch.h,选出需要使用的函数的声明等
    ffi.cdef(f.read())

if __name__ == "__main__":
    ffi.compile()

运行 python setup.py build

python setup.py install 

生成如下文件:

进入cffi_example,运行python build_fnmatch.py,与第二种方式相同会生成相关.c,.o,.so文件。

对test文件夹下的test_fnmatch.py做了细微修改,由于提示 AttributeError: 'function' object has no attribute 'FNM_PATHNAME' ,故直接将其换为0x1:

import pytest

from fnmatch import fnmatch

@pytest.mark.parametrize("pattern,name,flags,expected", [
    ("foo", "bar", 0, False),
    ("f**", "foo", 0, False),
    ("f/bar", "f/bar", 0, True),
    ("f*bar", "f/bar", 0x1, False),
])
def test_fnmatch(pattern, name, flags, expected):
    assert fnmatch(pattern, name) == expected

运行py.test test/测试结果如下:

因此对于外部c文件,可以使用cffi.verify()的形式使用,但是该种方式仍然需要进行函数声明。或者将c文件编译成为函数库,如此自定义.h文件声明需要使用的函数,即可在Python中通过cffi使用其中的函数。如libvirt库等。此外开源工具nitro中对于libvmi的使用方式也是如此。

 

参考:https://cffi.readthedocs.io/en/latest/using.html#

https://hsz1273327.gitbooks.io/python_total_tutorial/Python_Other_Topics/%E6%89%A9%E5%B1%95%E4%B8%8E%E6%80%A7%E8%83%BD%E4%BC%98%E5%8C%96/%E9%AB%98%E6%80%A7%E8%83%BD%E8%AF%AD%E8%A8%80%E6%89%A9%E5%B1%95/C%E6%89%A9%E5%B1%95/CFFI.html

https://github.com/wolever/python-cffi-example

https://www.jianshu.com/p/4f60eea33d68

 

posted @ 2018-09-13 22:30  ccxikka  Views(9539)  Comments(0Edit  收藏