Commit: 5479c0912351f56c0a87244b3028a5446c1164a4 - Osdn (git)

Loweynet

Commit MetaInfo

• Revision: 5479c0912351f56c0a87244b3028a5446c1164a4 (tree)
• Zeit: 2012-01-21 07:58:27
• Autor: Moriguchi, Hirokazu <umorigu@gmai...>
• Commiter: Moriguchi, Hirokazu

Log Message

MessageUtil_GetUTF8StaticBinaryBlock function: Protect string memory area by using VirtualProtect API

Ändern Zusammenfassung

• modified: msgutil.c

Diff

--- a/msgutil.c
+++ b/msgutil.c
@@ -71,12 +71,17 @@ const char* const MessageUtil_GetUTF8StaticBinaryBlock(const wchar_t* const ws,
 			int index = strMapCount;
 			char *beginPos = 0;
 			int postSize = 0;
+			DWORD flPrevProtect = 0;
 			strMapCount++;
 			pStrMap[index].ws = ws;
 			pStrMap[index].u8size = newSize;
-			beginPos = (char*)malloc(newSize);
+			beginPos = VirtualAlloc(NULL, newSize, MEM_COMMIT, PAGE_READWRITE);
 			pStrMap[index].u8s = beginPos;
 			postSize = WideCharToMultiByte(CP_UTF8, 0, ws, ws_area_length, beginPos, newSize, NULL, NULL);
+			if (!VirtualProtect(beginPos, newSize, PAGE_READONLY, &flPrevProtect))
+			{
+				// 失敗した
+			}
 			pResult = beginPos;
 
 			// pStrMap ソートする
@@ -106,7 +111,7 @@ void MessageUtil_FreeUTF8StaticBinaryBlocks()
 	EnterCriticalSection(&g_msgUtilLLock);
 	for (i = 0; i < strMapCount; i++)
 	{
-		free(pStrMap[i].u8s);
+		VirtualFree(pStrMap[i].u8s, pStrMap[i].u8size, MEM_DECOMMIT);
 	}
 	if (pStrMap)
 	{