Я пытаюсь передать массивы в C и выполнить некоторые вычисления там.
Я строю исполняемый файл как exe и экспортирую некоторые функции, которые затем вызывает Python. Процедура работает, но я не могу использовать стандартные функции, такие как printf()
, Это работает, если я строю двоичный файл как DLL. Из поиска в Интернете я понял, что загрузка dll и загрузка exe (maincrtstartup?) Делают инициализации для библиотек времени выполнения. Можно ли повторить то, что они делают, чтобы я мог использовать printf()
?
Код Python
import numpy as np
import ctypes
from numpy.ctypeslib import ndpointer
lib = ctypes.cdll.LoadLibrary('pycon.exe')
fun = lib.fun
fun.restype = None
fun.argtypes = [ndpointer(ctypes.c_double),
ctypes.c_size_t]data = np.ones((5,5)).astype('double')
def wfun(d):
fun(d, d.size)
print data
wfun(data)
print data
Код C
#include <stdio.h>
#include <math.h>
extern "C" __declspec(dllexport)
void fun(double *datav, size_t size) {
//printf("crash");
double * data = (double *) datav;
int i;
for (i = 0; i < size; ++i)
data[i] = 1.7159*tanh(0.66666667*data[i]);
}extern "C" __declspec(dllexport)
void init() {
//what to put here?
}int main(int argc, char* argv[])
{
return 0;
}
я понял, что рассматривать exe как dll гораздо сложнее, чем я себе представлял. я закончил со следующим обходным путем.
питон
import numpy as np
import ctypes as C
lib = C.cdll.LoadLibrary('pycon.exe')
lib.init( C.windll.kernel32.GetProcAddress( C.windll.msvcrt._handle,'printf'))
d = np.ones((5)).astype('double')
lib.fun(d.ctypes.data_as(C.POINTER(C.c_double)), C.c_size_t(5))
с
#include <stdio.h>
#include <math.h>
int (*pyprintf)(const char *,...) =0;
extern "C" __declspec(dllexport)
void fun(double *datav, size_t size) {
double * data = (double *) datav;
int i;
for (i = 0; i < size; ++i)
{
data[i] = 1.7159*tanh(0.66666667*data[i]);
pyprintf("workign! %f\n",data[i]);
}
}
extern "C" __declspec(dllexport)
void init(int (*_printfPtr)(const char *,...)) {
pyprintf=_printfPtr;
}
int main(int argc, char* argv[])
{
printf("exe");
return 0;
}
любопытство взяло верх над мной, и кроличья нора растет, теперь она немного менее пышная, чем выше, и, похоже, работает как exe и dll с немного большим количеством загружаемого кода.
/FIXED:NO -entry:_DllMainCRTStartup@12
питон
import numpy as np
import ctypes as C
import pefile
def unprotect( address):
crap = C.byref(C.create_string_buffer("\x00"*4))
res = C.windll.kernel32.VirtualProtect(address, 0x1000, 0x40, crap)
lib = C.cdll.LoadLibrary('pycon.exe')
k32 = C.windll.kernel32
pe = pefile.PE('pycon.exe')for entry in pe.DIRECTORY_ENTRY_IMPORT:
#print entry.dll
#assuming that the exe only has dependency on kernel32
for imp in entry.imports:
diff = imp.address-0x400000
memptr = k32.GetProcAddress(k32._handle,imp.name)
unprotect(lib._handle+diff)
fptr = (C.c_uint).from_address(lib._handle+diff)
fptr.value = memptr
d = np.ones((5)).astype('double')
lib.init()
lib.fun(d.ctypes.data_as(C.POINTER(C.c_double)), C.c_size_t(5))
print d
с
#include <stdio.h>
#include <math.h>
#include <windows.h>
extern "C" BOOL WINAPI _CRT_INIT(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpReserved);
extern "C" __declspec(dllexport)
void fun(double *datav, size_t size) {
double * data = (double *) datav;
int i;
for (i = 0; i < size; ++i)
{
data[i] = 1.7159*tanh(0.66666667*data[i]);
printf("workign! %f\n",data[i]);
}
fflush(stdout);
}
extern "C" __declspec(dllexport)
void init() {
_CRT_INIT(NULL,DLL_PROCESS_ATTACH,NULL);
}BOOL APIENTRY DllMain( HMODULE hModule,
DWORD ul_reason_for_call,
LPVOID lpReserved
)
{
_CRT_INIT(NULL,DLL_PROCESS_ATTACH,NULL);
printf("main exe");
return TRUE;
}
Других решений пока нет …