ReceiveTempSMS
/
Sweden-Number
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

oleview: Always check return value of SendMessage(TVM_GETITEM). 

Signed-off-by: Dmitry Timoshkov <dmitry@baikal.ru>
This commit is contained in:
Dmitry Timoshkov 2015-10-07 18:27:49 +08:00 committed by Alexandre Julliard
parent 65187a0b99
commit 26db80b685
1 changed files with 11 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
programs/oleview/tree.c
View File

@ -82,7 +82,7 @@ void CreateInst(HTREEITEM item, WCHAR *wszMachineName)
tvis.hParent = item; tvis.hParent = item;
tvis.hInsertAfter = TVI_LAST; tvis.hInsertAfter = TVI_LAST;
SendMessageW(globals.hTree, TVM_GETITEMW, 0, (LPARAM)&tvi); if (!SendMessageW(globals.hTree, TVM_GETITEMW, 0, (LPARAM)&tvi)) return;
if(!tvi.lParam || ((ITEM_INFO *)tvi.lParam)->loaded if(!tvi.lParam || ((ITEM_INFO *)tvi.lParam)->loaded
|| !(((ITEM_INFO *)tvi.lParam)->cFlag&SHOWALL)) return; || !(((ITEM_INFO *)tvi.lParam)->cFlag&SHOWALL)) return;
@ -154,9 +154,7 @@ void CreateInst(HTREEITEM item, WCHAR *wszMachineName)
while(hCur) while(hCur)
{ {
tvi.hItem = hCur; tvi.hItem = hCur;
SendMessageW(globals.hTree, TVM_GETITEMW, 0, (LPARAM)&tvi); if(!SendMessageW(globals.hTree, TVM_GETITEMW, 0, (LPARAM)&tvi) || !tvi.lParam)
if(!tvi.lParam)
{ {
hCur = (HTREEITEM)SendMessageW(globals.hTree, TVM_GETNEXTITEM, hCur = (HTREEITEM)SendMessageW(globals.hTree, TVM_GETNEXTITEM,
TVGN_NEXT, (LPARAM)hCur); TVGN_NEXT, (LPARAM)hCur);
@ -192,9 +190,7 @@ void ReleaseInst(HTREEITEM item)
memset(&tvi, 0, sizeof(TVITEMW)); memset(&tvi, 0, sizeof(TVITEMW));
tvi.hItem = item; tvi.hItem = item;
SendMessageW(globals.hTree, TVM_GETITEMW, 0, (LPARAM)&tvi); if(!SendMessageW(globals.hTree, TVM_GETITEMW, 0, (LPARAM)&tvi) || !tvi.lParam) return;
if(!tvi.lParam) return;
pU = ((ITEM_INFO *)tvi.lParam)->pU; pU = ((ITEM_INFO *)tvi.lParam)->pU;
@ -223,18 +219,18 @@ BOOL CreateRegPath(HTREEITEM item, WCHAR *buffer, int bufSize)
{ {
TVITEMW tvi; TVITEMW tvi;
int bufLen; int bufLen;
BOOL ret; BOOL ret = FALSE;
memset(buffer, 0, bufSize * sizeof(WCHAR)); memset(buffer, 0, bufSize * sizeof(WCHAR));
memset(&tvi, 0, sizeof(TVITEMW)); memset(&tvi, 0, sizeof(TVITEMW));
tvi.hItem = item; tvi.hItem = item;
SendMessageW(globals.hTree, TVM_GETITEMW, 0, (LPARAM)&tvi); if (SendMessageW(globals.hTree, TVM_GETITEMW, 0, (LPARAM)&tvi))
ret = (tvi.lParam && ((ITEM_INFO *)tvi.lParam)->cFlag & REGPATH); ret = (tvi.lParam && ((ITEM_INFO *)tvi.lParam)->cFlag & REGPATH);
while(TRUE) while(TRUE)
{ {
SendMessageW(globals.hTree, TVM_GETITEMW, 0, (LPARAM)&tvi); if(!SendMessageW(globals.hTree, TVM_GETITEMW, 0, (LPARAM)&tvi)) break;
if(tvi.lParam && (((ITEM_INFO *)tvi.lParam)->cFlag & (REGPATH|REGTOP))) if(tvi.lParam && (((ITEM_INFO *)tvi.lParam)->cFlag & (REGPATH|REGTOP)))
{ {
@ -318,7 +314,7 @@ static void AddCOMandAll(void)
while(curSearch) while(curSearch)
{ {
tvi.hItem = curSearch; tvi.hItem = curSearch;
SendMessageW(globals.hTree, TVM_GETITEMW, 0, (LPARAM)&tvi); if(!SendMessageW(globals.hTree, TVM_GETITEMW, 0, (LPARAM)&tvi)) break;
if(tvi.lParam && !lstrcmpW(((ITEM_INFO *)tvi.lParam)->info, wszComp)) if(tvi.lParam && !lstrcmpW(((ITEM_INFO *)tvi.lParam)->info, wszComp))
{ {
@ -631,15 +627,13 @@ void EmptyTree(void)
TVM_GETNEXTITEM, TVGN_PARENT, (LPARAM)del); TVM_GETNEXTITEM, TVGN_PARENT, (LPARAM)del);
tvi.hItem = del; tvi.hItem = del;
SendMessageW(globals.hTree, TVM_GETITEMW, 0, (LPARAM)&tvi); if(SendMessageW(globals.hTree, TVM_GETITEMW, 0, (LPARAM)&tvi) && tvi.lParam)
if(tvi.lParam)
{ {
if(((ITEM_INFO *)tvi.lParam)->loaded) ReleaseInst(del); if(((ITEM_INFO *)tvi.lParam)->loaded) ReleaseInst(del);
HeapFree(GetProcessHeap(), 0, (ITEM_INFO *)tvi.lParam); HeapFree(GetProcessHeap(), 0, (ITEM_INFO *)tvi.lParam);
}
SendMessageW(globals.hTree, TVM_DELETEITEM, 0, (LPARAM)del); SendMessageW(globals.hTree, TVM_DELETEITEM, 0, (LPARAM)del);
}
if(!cur) break; if(!cur) break;
} }