python c函数_python 调用 c 函数

python 调用 c 函数

最近对lua加密，用到了xxtea_encrypt，打算用python写个工具，直接python调用c.

// xxtea_Encrypt.cpp : 定义控制台应用程序的入口点。

//

#include "xxtea.h"

#include "Python.h"

PyObject* xxtea_encrypt(PyObject* self, PyObject* args)

{

unsigned char *data = NULL;

unsigned char *key = NULL;

unsigned int dataLen = 0;

unsigned int keyLen = 0;

unsigned int retLen = 0;

if (! PyArg_ParseTuple(args, "s#s#", &data,&dataLen,&key,&keyLen))

return NULL;

unsigned char * encrypt = xxtea_encrypt(data,dataLen,key,keyLen,&retLen);

return Py_BuildValue("s#",encrypt,retLen);

}

PyObject* xxtea_decrypt(PyObject* self, PyObject* args)

{

unsigned char *data = NULL;

unsigned char *key = NULL;

unsigned int dataLen = 0;

unsigned int keyLen = 0;

unsigned int retLen = 0;

if (! PyArg_ParseTuple(args, "s#s#", &data,&dataLen,&key,&keyLen))

return NULL;

unsigned char * decrypt = xxtea_decrypt(data,dataLen,key,keyLen,&retLen);

return Py_BuildValue("s#",decrypt,retLen);

}

static PyMethodDef xxteaMethods[] =

{

{"encrypt", xxtea_encrypt, METH_VARARGS, "xxteaEncrypt"},

{"decrypt", xxtea_decrypt, METH_VARARGS, "xxteaDecrypt"},

{NULL, NULL}

};

void initxxteaModule()

{

PyObject* m;

m = Py_InitModule("xxteaModule", xxteaMethods);

}

initxxteaModule python封装头，函数名为init+module名(Py_InitModule的第一个参数)。python导包时，会用到 import xxteaModule.

Py_InitModule第二个参数，c函数的相关描述信息。

struct PyMethodDef {

const char  *ml_name;   /* The name of the built-in function/method */

PyCFunction ml_meth;    /* The C function that implements it */

int         ml_flags;   /* Combination of METH_xxx flags, which mostly

describe the args expected by the C func */

const char  *ml_doc;    /* The __doc__ attribute, or NULL */

};

然后就是对c函数的封装，主要是c到python变量的一些转换.

最后说一下编译，这个网上很多都没有说清楚。写一个setup.py.

from distutils.core import setup, Extension

MOD = 'xxteaModule'

setup (name = MOD,ext_modules = [Extension(name=MOD, sources = ['xxtea_Encrypt.cpp','xxtea.cpp'])])

cmd 下执行命令：python setup.py build

将生成的build\lib.win32-2.7\xxteaModule.pyd拷贝到python DLLs目录中。

来看调用的例子。

import xxteaModule

def EncodeWithXxteaModule(filePath,key,signment):

all_the_text = ReadFile(filePath)

if all_the_text[:len(signment)] == signment :

return

#bak lua

BakFile(filePath,all_the_text)

encrypt = xxteaModule.encrypt(all_the_text,key)

signment = signment + encrypt

WriteFile(filePath,signment)

要注意的是，编译pyd时，机器必须安装vs2008,如果要使用其它编译器，改Python27\Lib\distutils\msvc9compiler.py 。关键代码：vc_env = query_vcvarsall(VERSION, plat_spec)