#define WIN32_LEAN_AND_MEAN

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

#include "../../MemoryModule.h"

typedef int (*addNumberProc)(int, int);

#define DLL_FILE "..\\SampleDLL\\SampleDLL.dll"

void LoadFromFile(void)
{
    addNumberProc addNumber;
    HRSRC resourceInfo;
    DWORD resourceSize;
    LPVOID resourceData;
    _TCHAR buffer[100];
    
    HINSTANCE handle = LoadLibrary(DLL_FILE);
    if (handle == NULL)
        return;

    addNumber = (addNumberProc)GetProcAddress(handle, "addNumbers");
    printf("From file: %d\n", addNumber(1, 2));
    
    resourceInfo = FindResource(handle, MAKEINTRESOURCE(VS_VERSION_INFO), RT_VERSION);
    printf("FindResource returned 0x%p\n", resourceInfo);
    
    resourceSize = SizeofResource(handle, resourceInfo);
    resourceData = LoadResource(handle, resourceInfo);
    printf("Resource data: %ld bytes at 0x%p\n", resourceSize, resourceData);
    
    LoadString(handle, 1, buffer, sizeof(buffer));
    printf("String1: %s\n", buffer);
    
    LoadString(handle, 20, buffer, sizeof(buffer));
    printf("String2: %s\n", buffer);

    FreeLibrary(handle);
}

void LoadFromMemory(void)
{
    FILE *fp;
    unsigned char *data=NULL;
    size_t size;
    HMEMORYMODULE handle;
    addNumberProc addNumber;
    HMEMORYRSRC resourceInfo;
    DWORD resourceSize;
    LPVOID resourceData;
    _TCHAR buffer[100];
    
    fp = fopen(DLL_FILE, "rb");
    if (fp == NULL)
    {
        printf("Can't open DLL file \"%s\".", DLL_FILE);
        goto exit;
    }

    fseek(fp, 0, SEEK_END);
    size = ftell(fp);
    data = (unsigned char *)malloc(size);
    fseek(fp, 0, SEEK_SET);
    fread(data, 1, size, fp);
    fclose(fp);

    handle = MemoryLoadLibrary(data);
    if (handle == NULL)
    {
        printf("Can't load library from memory.\n");
        goto exit;
    }

    addNumber = (addNumberProc)MemoryGetProcAddress(handle, "addNumbers");
    printf("From memory: %d\n", addNumber(1, 2));

    resourceInfo = MemoryFindResource(handle, MAKEINTRESOURCE(VS_VERSION_INFO), RT_VERSION);
    printf("MemoryFindResource returned 0x%p\n", resourceInfo);
    
    resourceSize = MemorySizeofResource(handle, resourceInfo);
    resourceData = MemoryLoadResource(handle, resourceInfo);
    printf("Memory resource data: %ld bytes at 0x%p\n", resourceSize, resourceData);
    
    MemoryLoadString(handle, 1, buffer, sizeof(buffer));
    printf("String1: %s\n", buffer);
    
    MemoryLoadString(handle, 20, buffer, sizeof(buffer));
    printf("String2: %s\n", buffer);

    MemoryFreeLibrary(handle);

exit:
    if (data)
        free(data);
}

int main(int argc, char* argv[])
{
    LoadFromFile();
    printf("\n\n");
    LoadFromMemory();
    return 0;
}