VPF::EnumResourceNames и ее коллбэк вешают программу

Aoizora

Дата 17.2.2017, 16:33 (ссылка)

(нет голосов)

Загрузка ...

Новичок

Профиль
Группа: Участник
Сообщений: 24
Регистрация: 18.11.2016

Репутация: нет
Всего: нет

Пишу программу, которая извлекает ресурсы из себя. При этом возникли две проблемы:
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 (ссылка)

(нет голосов)

Загрузка ...

Эксперт

Профиль
Группа: Комодератор
Сообщений: 2706
Регистрация: 9.8.2005
Где: Тюмень

Репутация: 99
Всего: 106

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

Это сообщение отредактировал(а) GremlinProg - 18.2.2017, 01:23

--------------------

"Гений всегда разумнее, чем умнее. Ум — это машина, разум — водитель этой машины."

_zorn_

Дата 17.2.2017, 23:17 (ссылка)

(нет голосов)

Загрузка ...

Эксперт

Профиль
Группа: Завсегдатай
Сообщений: 1077
Регистрация: 21.8.2007

Репутация: нет
Всего: 12

Цитата(Aoizora @ 17.2.2017, 23:33

)

Пишу программу, которая извлекает ресурсы из себя

Ресурсы кто то туда положил, правильно ?

Цитата(Aoizora @ 17.2.2017, 23:33

)

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

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

Aoizora

Дата 18.2.2017, 19:40 (ссылка)

(голосов:1)

Загрузка ...

Новичок

Профиль
Группа: Участник
Сообщений: 24
Регистрация: 18.11.2016

Репутация: нет
Всего: нет

Ресурсы туда положил я при помощи другой программы, так что они корректные.

Код


#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 (ссылка)

(нет голосов)

Загрузка ...

Эксперт

Профиль
Группа: Завсегдатай
Сообщений: 1077
Регистрация: 21.8.2007

Репутация: нет
Всего: 12

reshacker и прочие рестораторы нормально извлекают эти ресурсы ?

1 Пользователей читают эту тему (1 Гостей и 0 Скрытых Пользователей)
0 Пользователей:
« Предыдущая тема \| C/C++: Системное программирование и WinAPI \| Следующая тема »