Вобщем о том как в итоге проблема была "зарыта в могилу" (может кому пригодится).
Итак с BeginUpdateResource, UpdateResource и EndUpdateResource всё таки ничего не получилось. Пришлось разбираться с форматом Portable Executable ( поиск по строке PE в MSDN выдаёт нужные статьи).
Идея вот в чём :
Сами данные записываем просто в конец файла (В любом виде. Я просто строки писал).
Но возникает проблема - с какого адреса ( с какого смещения в файле ) нужно читать свои данные ? Для определения этого смещения пришлось проанализировать заголовок файла и определить где заканчивается последний блок данных которые были записаны при компиляции.
Вот код ехе2 который модифицируе файл ехе1 дописывая данные в его конец:
| Код |
BOOL CResUpdDlg::OnInitDialog()
{
.................
// Я предварительно добавил в ресурсы ехе2 файл с данными которые хочу добавить в ехе1
// Получаю эти данные
HMODULE hInst = ::AfxGetInstanceHandle();
HRSRC hRes = ::FindResource(hInst,MAKEINTRESOURCE(IDR_XML1),"XML");
HGLOBAL hGlob = (LPVOID)::LoadResource(hInst,hRes);
LPVOID pRes = ::LockResource(hGlob);
// Размер новых данных
DWORD resSize = ::SizeofResource(hInst,hRes);
// Получил сами данные
char * pData = (char*)pRes ;
// Открываю на запись файл в который буду записывать новые данные
CFile file("d:\\My\\MFC\\Test\\Test\\Debug\\Test.exe",CFile::modeWrite);
// Переходим в конец файла
file.SeekToEnd();
// Записываем
file.Write((LPCTSTR)pData,resSize);
::UnlockResource(hGlob);
return TRUE; // ¶З¦^ TRUEЎA°Ј«D±z№п±±Ёо¶µі]©wµJВI
}
|
Вот код который находит "оригинальный " конец файла ехе1 (при старте самого ехе1 )и читает новые данные:
| Код |
BOOL CTestDlg::OnInitDialog()
{
...........
// Находим "оригинальный " конец ехе из которого запустились
DWORD OriginalEndOfFile = GetOriginalEndOfFile();
CFile file("d:\\My\\MFC\\Test\\Test\\Debug\\Test.exe",CFile::modeRead);
//Реальная длина файла
DWORD RealEndOfFile = file.GetLength();
// Размер дописанных данных
DWORD BuffSize = RealEndOfFile - OriginalEndOfFile;
char * str = new char[BuffSize + 1];
// Переходим на свои данные (которые сами записали)
file.Seek(OriginalEndOfFile,CFile::begin);
// Читаем их
file.Read(str,BuffSize);
str[BuffSize] = 0;
// Отображаем
MessageBox(str);
delete[] str;
// Выход
::PostQuitMessage(0);
return TRUE;
}
DWORD CTestDlg::GetOriginalEndOfFile()
{
HANDLE hFile;
HANDLE hFileMapping;
LPVOID lpFileBase;
PIMAGE_DOS_HEADER dosHeader;
PIMAGE_NT_HEADERS pNTHeader;
PIMAGE_SECTION_HEADER pSectionHeader;
PIMAGE_SECTION_HEADER pLastSectionHeader;
DWORD FileAddress;
CString filename;
FileAddress = 0;
// Файл из которого будем читать свои данные (он же в данный момент и запущен)
filename = "d:\\My\\MFC\\Test\\Test\\Debug\\Test.exe";
// Создаём дескриптор файла
hFile = CreateFile(filename, GENERIC_READ, FILE_SHARE_READ, NULL,
OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);
if ( hFile == INVALID_HANDLE_VALUE )
{
//printf("Couldn't open file with CreateFile()\n");
return 0;
}
// Мапим файл в память и получаем дескриптор.
// На диске данные расположены с теми же смещениями что и в памяти.
hFileMapping = CreateFileMapping(hFile, NULL, PAGE_READONLY, 0, 0, NULL);
if ( hFileMapping == 0 )
{
CloseHandle(hFile);
//printf("Couldn't open file mapping with CreateFileMapping()\n");
return 0;
}
// Получаем указатель на базовый адресс по которому расположен файл
lpFileBase = MapViewOfFile(hFileMapping, FILE_MAP_READ, 0, 0, 0);
if ( lpFileBase == 0 )
{
CloseHandle(hFileMapping);
CloseHandle(hFile);
//printf("Couldn't map view of file with MapViewOfFile()\n");
return 0;
}
//printf("Dump of file %s\n\n", filename);
// Получаем указатель на DOS заголовок
dosHeader = (PIMAGE_DOS_HEADER)lpFileBase;
// Получаем указатель на NT заголовок
pNTHeader = (PIMAGE_NT_HEADERS)((DWORD)dosHeader + dosHeader->e_lfanew);
// Получаем указатель на SectionTable.
// Идёт сразу за IMAGE_NT_HEADER. Поэтому увеличиваем указатель pNTHeader (имеет тип PIMAGE_NT_HEADERS)
// на 1. Тем самым перепрыгиваем NT заголовок и попадаем на SectionTable.
pSectionHeader = (PIMAGE_SECTION_HEADER)(pNTHeader + 1);
// Количество секций
WORD nSections = pNTHeader->FileHeader.NumberOfSections;
// Получаем указатель на последнюю секцию т.к. они расположены в порядке увеличения RVA
// (увеличиваем указатель на "количество секций" - 1)
pLastSectionHeader = (PIMAGE_SECTION_HEADER)(pSectionHeader + (nSections - 1 ));
// Прибавляем к смещению начала данных данных последней секции относительно начала файла (PointerToRawData)
// размер данных последней секции (SizeOfRawData) и таким образом получаем
// смещение от начала файла (адресс) первого байта тех данных которые мы записали в конец файла.
FileAddress = pLastSectionHeader->PointerToRawData + pLastSectionHeader ->SizeOfRawData;
UnmapViewOfFile(lpFileBase);
CloseHandle(hFileMapping);
CloseHandle(hFile);
return FileAddress;
}
|
У меня получилось так. Желаю удачи всем тем кто захочет это использовать.
Спасибо всем кто поучаствовал в обсуждении темы. |
)