Commit | Line | Data |
7214af3e |
1 | // License information for this code is in license.txt |
2 | |
3 | #include <windows.h> |
4 | #include "SFmpqapi.h" |
a36705ea |
5 | #include "SFmpqInternal.h" |
7214af3e |
6 | #include "SFUtil.h" |
7 | #include "MpqCrypt.h" |
8 | |
9 | BOOL WriteBlockTable(MPQARCHIVE *mpqOpenArc) |
10 | { |
11 | DWORD tsz; |
12 | |
13 | if (mpqOpenArc->MpqHeader.dwBlockTableSize == 0) return TRUE; |
14 | if (!mpqOpenArc->lpBlockTable) return FALSE; |
15 | |
16 | char *buffer = (char *)SFAlloc(sizeof(BLOCKTABLEENTRY) * mpqOpenArc->MpqHeader.dwBlockTableSize); |
17 | if (buffer) { |
18 | memcpy(buffer,mpqOpenArc->lpBlockTable,sizeof(BLOCKTABLEENTRY) * mpqOpenArc->MpqHeader.dwBlockTableSize); |
19 | EncryptData((LPBYTE)buffer,sizeof(BLOCKTABLEENTRY) * mpqOpenArc->MpqHeader.dwBlockTableSize,dwBlockTableKey); |
a36705ea |
20 | SFSetFilePointer(mpqOpenArc->hFile, mpqOpenArc->qwMPQStart + mpqOpenArc->MpqHeader.dwBlockTableOffset + ((UINT64)mpqOpenArc->MpqHeader_Ex.wBlockTableOffsetHigh << 32), FILE_BEGIN); |
7214af3e |
21 | WriteFile(mpqOpenArc->hFile,buffer,sizeof(BLOCKTABLEENTRY) * mpqOpenArc->MpqHeader.dwBlockTableSize,&tsz,0); |
22 | SFFree(buffer); |
23 | } |
24 | else { |
25 | EncryptData((LPBYTE)mpqOpenArc->lpBlockTable,sizeof(BLOCKTABLEENTRY) * mpqOpenArc->MpqHeader.dwBlockTableSize,dwBlockTableKey); |
a36705ea |
26 | SFSetFilePointer(mpqOpenArc->hFile, mpqOpenArc->qwMPQStart + mpqOpenArc->MpqHeader.dwBlockTableOffset + ((UINT64)mpqOpenArc->MpqHeader_Ex.wBlockTableOffsetHigh << 32), FILE_BEGIN); |
7214af3e |
27 | WriteFile(mpqOpenArc->hFile,mpqOpenArc->lpBlockTable,sizeof(BLOCKTABLEENTRY) * mpqOpenArc->MpqHeader.dwBlockTableSize,&tsz,0); |
28 | DecryptData((LPBYTE)mpqOpenArc->lpBlockTable,sizeof(BLOCKTABLEENTRY) * mpqOpenArc->MpqHeader.dwBlockTableSize,dwBlockTableKey); |
29 | } |
30 | |
31 | return TRUE; |
32 | } |