Блог

Я хочу внедрить c-dll в процесс. В c-dll он загрузит другую dll, написанную golang. Вот мой C-код:

когда загрузчик.dll загружена, она автоматически загрузит рабочий файл.dll, которая записывается в golang.

 // loader.c  

#include <windows.h>
#include <stdio.h>
#include <memory.h>

typedef void (*StartWorker)();
HMODULE hWorker = NULL;


BOOL APIENTRY DllMain(HMODULE hModule,
                      DWORD ul_reason_for_call,
                      LPVOID lpReserved)
{
    switch (ul_reason_for_call)
    {
    case DLL_PROCESS_ATTACH:
    {

        hWorker = LoadLibrary("D:\code\toys\worker.dll");
        if (hWorker == NULL)
        {
            exit(1);
        }
        
        StartWorker startWorker = (StartWorker)GetProcAddress(hWorker, "StartWorker");
        MessageBox(NULL, "worker starting", TEXT("Warning:"), MB_OK);
        if (startWorker == NULL)
        {
            MessageBox(NULL, "error", TEXT("Warning:"), MB_OK);
            exit(1);
        }
        startWorker();
        MessageBox(NULL, "worker started", TEXT("Warning:"), MB_OK);
        FreeLibrary(hWorker);
        break;
    }
    case DLL_PROCESS_DETACH:
    case DLL_THREAD_ATTACH:
    case DLL_THREAD_DETACH:
    default:
        break;
    }
    return TRUE;
}

 

и код go:

 // worker.go

package main

/*
#include <stdlib.h>
#include <windows.h>
*/
import "C"

//export StartWorker
func StartWorker {
    C.MessageBox(nil, C.CString("Hello Worker"), C.CString("Warning:"), 0)
}
func main() {
}
 

Я скомпилировал их MinGW-w64 . Когда загрузчик.dll пытается вызвать StartWorker() worker.dll, процесс не показал окно сообщения. После того, как я перепишу рабочий.dll в c, все работает хорошо. И это также нормально, когда я StartWorker() вызываю следующий код:

 #include <windows.h>
#include <stdio.h>
#include <memory.h>

typedef void (*StartWorker)();

int main()
{
    char *dllPath = "D:\code\toys\loader.dll";
    HMODULE hWorker = NULL;
    hWorker = LoadLibrary(dllPath);
    StartWorker startWorker = (StartWorker)GetProcAddress(hWorker, "StartWorker");
    if (startWorker == NULL)
    {
        MessageBox(NULL, "dllPath", TEXT("Warning:"), MB_OK);
        exit(1);
    }
    startWorker();
    MessageBox(NULL, "target", TEXT("Warning:"), MB_OK);
    FreeLibrary(hWorker);
    return 0;
}

 

Интересно, есть ли какой-либо конфликт с go-runtime?

Тест отлично, возможно, вы вызвали Microsoft-Windows-Only api. Попробуйте изменить MessageBox() (с помощью win32 user32.dll ) Sum() , чтобы проверить c-dll , который вызывается go-dll .

Вот файл go в gen go-dll // gosum.go

 package main

/*
#include <stdlib.h>
*/
import "C"

//export Sum
func Sum(a int32, b int32) int32 {
    return a   b
}

func main() {
}
 

Перейдите к созданию c-shared Windows dll:

 go build -ldflags "-s -w" -buildmode=c-shared -o gosum.dll
 

вы можете написать тестовый файл на c / cpp:

 #include <stdio.h>
#include <malloc.h>
#include <string.h>
#include "gosum.h"

int main() {
    printf("Sum(1,2)=%dn", Sum(1,2));
    return 1;
}
 

Скомпилировал его в MinGW-w64

 g   -c test_gosum.cpp -o test_gosum.o -g -std=c  11
g   test_gosum.o gosum.dll -o test_gosum.exe -g -std=c  11
.test_gosum.exe
 

Затем для сборки loader.dll , в Visual Studio которой вызывается gosum.dll

// скопируйте go-dll gosum.h файл CGO gen в проект, возможно, вам нужно удалить некоторый код ошибки

/файл / gosum.h:

 /* Code generated by cmd/cgo; DO NOT EDIT. */

/* package gosum */


#line 1 "cgo-builtin-export-prolog"

#include <stddef.h> /* for ptrdiff_t below */

#ifndef GO_CGO_EXPORT_PROLOGUE_H
#define GO_CGO_EXPORT_PROLOGUE_H

#ifndef GO_CGO_GOSTRING_TYPEDEF
typedef struct { const char* p; ptrdiff_t n; } _GoString_;
#endif

#endif

/* Start of preamble from import "C" comments.  */


#line 3 "gosum.go"

#include <stdlib.h>

#line 1 "cgo-generated-wrapper"


/* End of preamble from import "C" comments.  */


/* Start of boilerplate cgo prologue.  */
#line 1 "cgo-gcc-export-header-prolog"

#ifndef GO_CGO_PROLOGUE_H
#define GO_CGO_PROLOGUE_H

typedef signed char GoInt8;
typedef unsigned char GoUint8;
typedef short GoInt16;
typedef unsigned short GoUint16;
typedef int GoInt32;
typedef unsigned int GoUint32;
typedef long long GoInt64;
typedef unsigned long long GoUint64;
typedef GoInt64 GoInt;
typedef GoUint64 GoUint;
typedef float GoFloat32;
typedef double GoFloat64;

/*
  static assertion to make sure the file is being used on architecture
  at least with matching size of GoInt.
*/
typedef char _check_for_64_bit_pointer_matching_GoInt[sizeof(void*) == 64 / 8 ? 1 : -1];

#ifndef GO_CGO_GOSTRING_TYPEDEF
typedef _GoString_ GoString;
#endif
typedef void* GoMap;
typedef void* GoChan;
typedef struct { void* t; void* v; } GoInterface;
typedef struct { void* data; GoInt len; GoInt cap; } GoSlice;

#endif

/* End of boilerplate cgo prologue.  */

#ifdef __cplusplus
extern "C" {
#endif
    extern GoInt32 Sum(GoInt32 a, GoInt32 b);
#ifdef __cplusplus
}
#endif
 

/файл / pch.h (созданный vs, нужно что-то добавить):

 #define PCH_H
#include "framework.h"
#endif

// ----- added
#ifdef IMPORT_DLL
#else
#define IMPORT_DLL extern "C" _declspec(dllimport)
#endif

IMPORT_DLL int get_sum_by_another_dll(int a, int b);

// ----- added
 

// dllmain.cpp файл (gen by vs):

 #include "pch.h"

BOOL APIENTRY DllMain( HMODULE hModule,
                       DWORD  ul_reason_for_call,
                       LPVOID lpReserved
                     )
{
    switch (ul_reason_for_call)
    {
    case DLL_PROCESS_ATTACH:
    case DLL_THREAD_ATTACH:
    case DLL_THREAD_DETACH:
    case DLL_PROCESS_DETACH:
        break;
    }
    return TRUE;
}
 

// loader.cpp файл:

 #include "pch.h"
#include <stdlib.h>
#include <windows.h>
#include "gosum.h"  // also need add `gosum.dll` to VS project file
using namespace std;

typedef int (CALLBACK* GOSUMFUNC)(GoInt32 a, GoInt32 b);

int __stdcall get_sum_by_another_dll(int a, int b)
{
    HMODULE h = LoadLibrary(TEXT("gosum.dll"));
    if (NULL == h || INVALID_HANDLE_VALUE == h) {
        return 0;
    }
    GOSUMFUNC go_dll_sum = (GOSUMFUNC)GetProcAddress(h, "Sum");
    if (go_dll_sum) {
        return go_dll_sum(a, b);
    }
    return 1;
}
 

затем скомпилируйте их в VS, вы получите loader.dll .

существует gosum.dll go-dll , loader.dll c-dll компилируется VisualStudio, который вызывается gosum.dll .

вы можете попробовать вызвать их в golang или clang.

// demo.go тестовый загрузчик.dll и gosum.dll с golang, также необходимо скопировать gosum.dll и loader.dll в тот же каталог.

 package main

import (
    "fmt"
    "log"
    "syscall"
)

func main() {
    h, e := syscall.LoadLibrary("loader.dll")   //Make sure this DLL follows Golang machine bit architecture (64-bit in my case)
    if e != nil {
        log.Fatal(e)
    }
    defer syscall.FreeLibrary(h)
    proc, e := syscall.GetProcAddress(h, "get_sum_by_another_dll") //One of the functions in the DLL
    if e != nil {
        log.Fatal(e)
    }
    var a int32 = 2
    var b int32 = 5
    ret, _, err := syscall.Syscall(proc, 2, uintptr(a), uintptr(b), 0)  //Pay attention to the positioning of the parameter
    fmt.Println(ret)
    fmt.Println(err)
}
 

запустите его, и вы получите результат: get_sum_by_another_dll(2,5) = 7

// callloader.cpp тестовый погрузчик.dll и gosum.dll с c / cpp, также необходимо скопировать gosum.dll и loader.dll в тот же каталог.

 #include <iostream>
#include <Windows.h>

int main()
{
    HINSTANCE hDllInst;
    hDllInst = LoadLibrary(L"loader.dll");
    typedef int(*SUMFUNC)(int a, int b);
    SUMFUNC dll_sum_fun = (SUMFUNC)GetProcAddress(hDllInst, "get_sum_by_another_dll"); 
    std::cout << dll_sum_fun(1, 2);
}
 

затем скомпилируйте его в VS, запустите. /callloader.exe , вы получите результат: get_sum_by_another_dll(1,2) = 3

Я нашел способ реализовать эту функцию. Я только что написал несколько новых кодов следующим образом. Теперь они работают нормально.

http.go

 package main

import "C"
import (
    "fmt"
    "net/http"
)

func sayHello(w http.ResponseWriter, r *http.Request) {
    _, _ = w.Write([]byte("Hello World!"))
}

//export StartHttp
func StartHttp() {
    http.HandleFunc("/", sayHello)

    err := http.ListenAndServe("127.0.0.1:9999", nil)
    if err != nil {
        fmt.Printf("Error: %vn", err)
        return
    }
    fmt.Printf("Listening: http://127.0.0.1:9000")
}

func main() {
    // StartHttp();
}

 

загрузчик.c

 #include <windows.h>
#include <stdio.h>

// https://forum.pellesc.de/index.php?topic=4725.0
#ifdef __GNUC__
HANDLE k __attribute__((section(".shared"), shared)) = NULL;
#endif
#ifdef _MSC_VER
#pragma data_seg(".shared")
HANDLE k = NULL;
#pragma data_seg()
#pragma comment(linker, "/section:.shared,RWS")
#endif

typedef void (*StartHttp)();

HMODULE hHttp = NULL;

BOOL APIENTRY DllMain(HMODULE hModule,
                      DWORD ul_reason_for_call,
                      LPVOID lpReserved)
{
    switch (ul_reason_for_call)
    {
    case DLL_PROCESS_ATTACH:
    {
        hHttp = LoadLibrary("http.dll");
        StartHttp startWorker = (StartHttp)GetProcAddress(hHttp, "StartHttp");
        k = startWorker;  
        // do not execute the function here!
        break;
    }

    case DLL_PROCESS_DETACH:
    {
        FreeLibrary(hHttp);
        break;
    }
    case DLL_THREAD_ATTACH:
    case DLL_THREAD_DETACH:
    default:
        break;
    }
    return TRUE;
}

HANDLE GetHttpStarter()
{
    return k;
}

 

client.c

 #include <windows.h>
#include <stdio.h>
#include <memory.h>

typedef void (*StartHttp)();
typedef HMODULE (*GetHttpStarter)();

int main()
{
    HMODULE hLoader = NULL;

    hLoader = LoadLibrary("loader.dll");
    GetHttpStarter getHttpStarter = (GetHttpStarter)GetProcAddress(hLoader, "GetHttpStarter");
    StartHttp startHttp = (StartHttp)getHttpStarter();
    startHttp();

    printf("^^n");
    FreeLibrary(hLoader);
    return 0;
}

 

 > go build -buildmode=c-shared -o http.dll .http.go
> gcc .loader.c -shared -o loader.dll
> gcc .client.c -o client.exe
> .client.exe
 

Самое главное, что я изменил, это то, что загружайте только http.dll(go) in’ loader.dll(c) ы DllMain() без функций execute http.dll(go) и сохраняйте адреса этих функций в общей памяти. таким образом, я могу получить адрес функции http.dll(go) , вызвав функцию GetHttpStarter() экспорта loader.dll(c) . Выглядит сложно, но это работает 🙂

Я также протестировал внедрение dll, в результате я могу разместить службу http notepad.exe сейчас: D