Версия для печати темы
Нажмите сюда для просмотра этой темы в оригинальном формате
Форум программистов > C/C++: Системное программирование и WinAPI > EnumResourceNames и ее коллбэк вешают программу


Автор: Aoizora 17.2.2017, 16:33
Пишу программу, которая извлекает ресурсы из себя. При этом возникли две проблемы:
1. После извлечения ресурсов программа зависает, а в олли дебаггере возникает исключение после возврата из коллбэка.
2. Когда я хочу сформировать строку из имени ресурса при помощи вызова wsprintf, возникает ошибка.

Что у меня неправильно?

Код

#include <windows.h>
#include <shellapi.h>

#pragma comment(lib, "shell32.lib");
#pragma comment(lib, "user32.lib")


BOOL WINAPI DropResource(HMODULE hModule, LPCTSTR lpType, LPTSTR lpName, LONG_PTR lParam)
{
    HRSRC hResource = NULL;
    HANDLE hFile = NULL;
    DWORD dw = 0;
    TCHAR szFileName[MAX_PATH];

    hResource = FindResource(hModule, lpName, lpType);
    if (hResource == NULL)
        return FALSE;

    wsprintf(szFileName, TEXT("%s"), lpName);
    hFile = CreateFile(TEXT("file.exe"),
        GENERIC_WRITE,
        FILE_SHARE_WRITE,
        NULL,
        CREATE_ALWAYS,
        FILE_ATTRIBUTE_NORMAL,
        0);
    if (hFile != INVALID_HANDLE_VALUE)
    {
        WriteFile(hFile,
            LockResource(LoadResource(hModule, hResource)),
            SizeofResource(hModule, hResource),
            &dw,
            NULL);
        CloseHandle(hFile);
        ShellExecute(NULL, "OPEN", "file.exe", NULL, NULL, 0);
    }
    return TRUE;
}

int WINAPI WinMain(HINSTANCE hInst, HINSTANCE hPrev, LPSTR lpCmdLine, int nCmdShow)
{
    EnumResourceNames(NULL, RT_RCDATA, DropResource, 0);

    return 0;
}

Автор: GremlinProg 17.2.2017, 22:45
lpName и lpType вполне могут быть невалидными указателями для wsprintf.
Тут может передаваться и ID'шник, преобразованный в указатель.
В такой ситуации поможет макрос IS_INTRESOURCE.

Автор: _zorn_ 17.2.2017, 23:17
Цитата(Aoizora @  17.2.2017,  23:33 Найти цитируемый пост)
Пишу программу, которая извлекает ресурсы из себя

Ресурсы кто то туда положил, правильно ?
Цитата(Aoizora @  17.2.2017,  23:33 Найти цитируемый пост)
Что у меня неправильно?

А где ресурсы ?
Почему вообще такая уверенность что они корректные ?

Автор: Aoizora 18.2.2017, 19:40
Ресурсы туда положил я при помощи другой программы, так что они корректные.

Код

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


void * ReadImage(LPSTR szPath, DWORD *dwSize);
void AddResource(LPSTR szPath, LPBYTE lpBuffer, DWORD dwSize, LPTSTR lpName);

int main(int argc, char *argv[])
{
    LPVOID lpImage = NULL;
    DWORD dwImageSize = 0;

    if (argc > 2)
    {
        lpImage = ReadImage(argv[2], &dwImageSize);
        AddResource(argv[1], lpImage, dwImageSize, TEXT("Main"));
    }
}

void * ReadImage(LPSTR szPath, DWORD *dwSize)
{
    HANDLE hFile;
    DWORD dwFileSize;
    DWORD dwBytesRead;
    LPBYTE lpBuffer;

    hFile = CreateFileA(szPath,
        GENERIC_READ, 
        0,
        NULL,
        OPEN_EXISTING,
        FILE_ATTRIBUTE_NORMAL,
        NULL);

    if (INVALID_HANDLE_VALUE != hFile)
    {
        dwFileSize = GetFileSize(hFile, NULL);

        lpBuffer = HeapAlloc(GetProcessHeap(), 0, dwFileSize);
        ReadFile(hFile, lpBuffer, dwFileSize, &dwBytesRead, NULL);
        CloseHandle(hFile);

        *dwSize = dwFileSize;
        return lpBuffer;
    }
    return NULL;
}

void AddResource(LPSTR szPath, LPBYTE lpBuffer, DWORD dwSize, LPTSTR lpName)
{
    HANDLE hResource;

    hResource = BeginUpdateResource(szPath, FALSE);
    if (NULL != hResource)
    {
        if (UpdateResource(hResource, 
            RT_RCDATA, 
            lpName, 
            MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), 
            (LPVOID)lpBuffer, 
            dwSize) != FALSE)
        {
            EndUpdateResource(hResource, FALSE);
        }
    }
}

Автор: _zorn_ 20.2.2017, 14:01
reshacker и прочие рестораторы нормально извлекают эти ресурсы ?

Powered by Invision Power Board (http://www.invisionboard.com)
© Invision Power Services (http://www.invisionpower.com)