Python shellexec 动态扩展模块

做测试用的，顺手写的一个python的c模块code，放在这里存放下做个sample。

这个模块主要是用c扩展了python，其中存在一个system方法，可以直接执行系统命令。类似于mysql的udf或者php的dl。这个在某些特定的python环境下可能形成突破（例如禁用了python自带的os.system）

找个目录，然后写两个文件：

system.c

#include <Python.h> 
#include <string.h> 

static PyObject * system_system(PyObject *self, PyObject *args) { 

    const char *command; 
    int sts; 

    if (! PyArg_ParseTuple(args, "s", &command)) { 
        return NULL; 
    } 

    sts = system(command); 

    return Py_BuildValue("i", sts); 
} 

// Module's method table and initialization function 
// See: http://docs.python.org/extending/extending.html#the-module-s-method-table-and-initialization-function
 static PyMethodDef SystemMethods[] = { 
    {"system", system_system, METH_VARARGS, "Execute a shell command."}, 
    {NULL, NULL, 0, NULL} 
}; 

void initsystem(void) { 
    PyImport_AddModule("system"); 
    Py_InitModule("system", SystemMethods); 
} 

int main(int argc, char *argv[]) { 
    Py_SetProgramName(argv[0]); 
    Py_Initialize(); 
    initsystem(); 
    return 0; 
}

setup.py

from distutils.core import setup, Extension

ext = Extension('system', sources=['system.c'])

setup(name='system', version='1.0', description='Test description', ext_modules=[ext])

然后在当前目录下执行：

python setup.py build

这时候就会编译一个python的动态模块在build/lib.xxxxxxx/目录下。叫做system.so

然后你只要执行如下代码，即可成功调用：

Python 2.7.3 (default, Aug 30 2012, 23:03:58)
[GCC 4.1.2 20080704 (Red Hat 4.1.2-52)] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import imp
>>> imp.load_dynamic("system", "build/lib.linux-x86_64-2.7/system.so")
<module 'system' from 'build/lib.linux-x86_64-2.7/system.so'>
>>> import system
>>> system.system("whoami")
root
0

转自：http://zone.wooyun.org/content/1019

文章目录