вызовите ctypes.execve с динамическим списком аргументов - PullRequest
Новая
       33

вызовите ctypes.execve с динамическим списком аргументов

0 голосов
/ 20 января 2019

Я хочу использовать ctypes.execve в python для вызова внешнего скрипта.Я уже разобрался, как вызвать скрипт и как передать список аргументов.Но, как вы можете видеть, список или аргументы довольно статичны, и я не мог понять, как я могу сделать это динамически.

Список аргументов должен быть списком Python, например: argv_list = ["./tmp/arg_test.sh", "arg_1", "arg_2"] Этот список Python нуждается вчтобы преобразовать как-нибудь в соответствие char * const argv[], чтобы я мог передать его execve.<- Вот где я борюсь. </p> 

cmd = "./tmp/arg_test.sh"
cmd_byte = cmd.encode('utf-8')     # create byte objects from the strings

### works ### ### works ### ### works ### ### works ###

argv_0 = "argv_0"
argv_0_byte = argv_0.encode('utf-8')

argv_1 = "argv_1"
argv_1_byte = argv_1.encode('utf-8')

argv_2 = "argv_2"
argv_2_byte = argv_2.encode('utf-8')

argv_3 = "argv_3"
argv_3_byte = argv_3.encode('utf-8')

b_argv = (ctypes.c_char_p * 5)(argv_0_byte, argv_1_byte, argv_2_byte, argv_3_byte)

### works ### ### works ### ### works ### ### works ###


# int execve(const char * filename, char * const argv[], char * const envp[]);
libc.execve(ctypes.c_char_p(cmd_byte), b_argv, 0)

Спасибо!

1 Ответ

0 голосов
/ 20 января 2019

Вот способ генерировать аргументы динамически. Это написано для Windows msvcrt, но должно работать и для libc:

test.cpp 

#include <stdio.h>
#include <stdlib.h>

int main(int argc, char* argv[])
{
    for(int i = 0; i < argc; ++i)
        printf("argv[%d] = \"%s\"\n",i,argv[i]);
    printf("TEST1=%s\n",getenv("TEST1"));
    printf("TEST2=%s\n",getenv("TEST2"));
    return 0;
}

test.py 

from ctypes import *

dll = CDLL('msvcrt')
dll._execve.argtypes = c_char_p,POINTER(c_char_p),POINTER(c_char_p)
dll._execve.restype = c_ssize_t

cmd = b'test.exe'
args = [b'arg1',b'arg2',b'arg3']
env = [b'TEST1=hello',b'TEST2=goodbye']

# Convert Python lists to c_char_p arrays.
# Arrays should be null-terminated so adds None to the end.
cargs = (c_char_p * (len(args) + 2))(cmd,*args,None)
cenv = (c_char_p * (len(env) + 1))(*env,None)

dll._execve(cmd,cargs,cenv)

Выход: 

argv[0] = "test.exe"
argv[1] = "arg1"
argv[2] = "arg2"
argv[3] = "arg3"
TEST1=hello
TEST2=goodbye
...