archive
/
windows-server-2003
mirror of https://github.com/selfrender/Windows-Server-2003
2
0
Fork 0
Code Issues Projects Releases Wiki Activity
Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
4 Commits
1 Branch
0 Tags
1.5 GiB
 
 
 
 
 
 
Tree: 5c6fe3db62
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from '5c6fe3db62'
${ noResults }
windows-server-2003/sdktools/debuggers/oca/cer/cw15/main.cpp

2442 lines
60 KiB
Raw Blame History

#include "Main.h"
#include "KernelMode.h"
#include "UserMode.h"
#include "WinMessages.h"
#include "resource.h"
#include "clist.h"
#include "cuserlist.h"
#include "reportFault.h"
#include "CERHELP.h"
HWND g_hWnd;
HINSTANCE g_hinst;
HWND hKrnlMode = NULL;
HWND hUserMode = NULL;
KMODE_DATA KModeData;
TCHAR CerRoot[MAX_PATH];
HWND LoadThreadParam = NULL;
MRU_LIST MruList;
Clist CsvContents;
CUserList cUserData;
GLOBAL_POLICY GlobalPolicy;
BOOL g_bFirstBucket = TRUE;
BOOL g_bAdminAccess = TRUE;
TCHAR szUserColumnHeaders[][100] =
{ _T("Bucket ID"),
_T("App. Name"),
_T("App. Ver"),
_T("Module Name"),
_T("Module Version"),
_T("Offset"),
_T("Hits"),
_T("Cabs Collected"),
_T("Cabs Not Reported"),
_T("Requested Collection"),
_T("Allowed Collection"),
_T("Microsoft Response"),
_T("URL to Launch")
};
TCHAR szKerenelColumnHeaders[][100] =
{ _T("BucketID"),
_T("Cabs Collected"),
_T("Microsoft Response")
};
BOOL GetFileName(HWND hwnd, TCHAR *FileName, int iType)
{
OPENFILENAME opfn;
// if we are running on NT4.0 use sizeof (OPENFILENAME_SIZE_VERSION_400)
ZeroMemory (&opfn, sizeof OPENFILENAME );
opfn.lStructSize = sizeof OPENFILENAME;
opfn.hwndOwner = hwnd;
opfn.nMaxFileTitle = MAX_PATH * sizeof TCHAR;
opfn.lpstrFileTitle = FileName;
opfn.lpstrFilter=_T("*.TXT");
opfn.nFilterIndex = iType;
opfn.Flags = OFN_CREATEPROMPT | OFN_OVERWRITEPROMPT | OFN_OVERWRITEPROMPT;
if (GetSaveFileName((LPOPENFILENAME) &opfn))
return TRUE;
else
return FALSE;
}
void ExportUserModeData(HWND hwnd)
{
TCHAR szFileName[MAX_PATH];
int iFileType = 1;
TCHAR tchDelimiter = _T(',');
HANDLE hFile = INVALID_HANDLE_VALUE;
USER_DATA UserData;
DWORD dwWritten = 0;
BOOL bEOF = FALSE;
ZeroMemory (szFileName, sizeof szFileName);
if ( !GetFileName(hwnd, szFileName, iFileType))
{
goto ERRORS;
}
if (iFileType == 3)
{
//determine the max column widths
goto ERRORS;
}
if (iFileType == 1)
{
tchDelimiter = _T(',');
}
if (iFileType == 2)
{
tchDelimiter = _T('\t');
}
// Write out the data
hFile = CreateFile (szFileName,
GENERIC_WRITE,
FILE_SHARE_READ,
NULL,
CREATE_ALWAYS,
FILE_ATTRIBUTE_NORMAL,
NULL);
if (hFile == INVALID_HANDLE_VALUE)
goto ERRORS;
for (int i = 0; i < USER_COL_COUNT; i++)
{
WriteFile(hFile, szUserColumnHeaders[i], _tcslen(szUserColumnHeaders[i]) *sizeof TCHAR, &dwWritten, NULL);
}
CloseHandle (hFile);
//loop through the linked list and write the data to the output file
cUserData.ResetCurrPos();
while (cUserData.GetNextEntry(&UserData, &bEOF))
{
// write the output buffer to the file
;
}
ERRORS:
;
}
void ExportKernelModeData(HWND hwnd)
{
TCHAR FileName[MAX_PATH];
int iType = 0;
GetFileName(hwnd, FileName, iType);
}
typedef enum REPORTING_MODES {USER_MODE, KERNEL_MODE,GENERAL_MODE};
REPORTING_MODES ReportMode = USER_MODE;
DLGTEMPLATE * WINAPI DoLockDlgRes(LPCSTR lpszResName)
{
HRSRC hrsrc = FindResource(NULL, lpszResName, RT_DIALOG);
if (hrsrc)
{
HGLOBAL hglb = LoadResource(g_hinst, hrsrc);
return (DLGTEMPLATE *) LockResource(hglb);
}
else
{
return NULL;
}
}
void DoKernelMode(HWND hwnd)
{
TCHAR DialogTitle[255];
DLGTEMPLATE *Temp = NULL;
if (_tcscmp(CerRoot, _T("\0")))
{
if (StringCbPrintf(DialogTitle, sizeof DialogTitle, _T("Corporate Error Reporting - KERNEL MODE - %s"), CerRoot)!= S_OK)
{
goto ERRORS;
}
}
else
{
if (StringCbPrintf(DialogTitle, sizeof DialogTitle, _T("Corporate Error Reporting - KERNEL MODE")) != S_OK)
{
goto ERRORS;
}
}
SendMessage(hwnd, WM_SETTEXT, 0, (LPARAM)DialogTitle);
if (ReportMode != KERNEL_MODE)
{
ReportMode = KERNEL_MODE;
if (hKrnlMode == NULL)
{
Temp = DoLockDlgRes(MAKEINTRESOURCE(IDD_KERNEL_MODE));
if (Temp)
{
hKrnlMode = CreateDialogIndirect(g_hinst,Temp , hwnd, (DLGPROC) KrnlDlgProc);
}
}
else
{
HMENU hMenu = GetMenu(hwnd);
if (hMenu)
{
EnableMenuItem (hMenu, ID_REPORT_SELECTEDCRASHES , MF_BYCOMMAND| MF_GRAYED);
}
ShowWindow(hKrnlMode,1);
}
if (hUserMode != NULL)
{
ShowWindow(hUserMode,0);
}
}
ERRORS:
return;
}
void DoUserMode(HWND hwnd)
{
TCHAR DialogTitle[255];
DLGTEMPLATE *Temp = NULL;
if (_tcscmp(CerRoot, _T("\0")))
{
if (StringCbPrintf(DialogTitle, sizeof DialogTitle, _T("Corporate Error Reporting - USER MODE - %s"), CerRoot)!= S_OK)
{
goto ERRORS;
}
}
else
{
if (StringCbPrintf(DialogTitle, sizeof DialogTitle, _T("Corporate Error Reporting - USER MODE")) != S_OK)
{
goto ERRORS;
}
}
SendMessage(hwnd, WM_SETTEXT, 0, (LPARAM)DialogTitle);
if (ReportMode != USER_MODE)
{
ReportMode = USER_MODE;
if (hUserMode == NULL)
{
Temp = DoLockDlgRes(MAKEINTRESOURCE(IDD_USER_MODE));
if (Temp)
{
hUserMode = CreateDialogIndirect(g_hinst,Temp , hwnd, (DLGPROC) UserDlgProc);
}
}
else
{
HMENU hMenu = GetMenu(hwnd);
if (hMenu)
{
if (_tcscmp(CerRoot, _T("\0")))
{
EnableMenuItem (hMenu, ID_REPORT_SELECTEDCRASHES , MF_BYCOMMAND| MF_ENABLED);
}
}
ShowWindow(hUserMode,1);
}
if (hKrnlMode != NULL)
{
ShowWindow(hKrnlMode,0);
}
}
ERRORS:
return;
}
BOOL FileSystemIsNTFS(TCHAR *szPath)
{
TCHAR szPathToValidate[MAX_PATH];
TCHAR szType[50];
ZeroMemory (szPathToValidate, sizeof szPathToValidate);
ZeroMemory (szType, sizeof szType);
if (StringCbCopy (szPathToValidate, sizeof szPathToValidate, szPath) != S_OK)
goto ERRORS;
PathStripToRoot(szPathToValidate);
// add a \ if one is not found
if (_tcslen (szPathToValidate ) > 1)
{
if (szPathToValidate[_tcslen(szPathToValidate)-1] != _T('\\'))
{
if (StringCbCat(szPathToValidate, sizeof szPathToValidate, _T("\\"))!= S_OK)
goto ERRORS;
}
if (GetVolumeInformation(szPathToValidate, NULL, 0, NULL, 0, NULL, szType, sizeof szType / sizeof TCHAR))
{
if (!_tcscmp(szType, _T("NTFS")))
{
return TRUE;
}
}
}
ERRORS:
return FALSE;
}
int CALLBACK BrowseCallbackProc(
HWND hwnd,
UINT uMsg,
LPARAM lParam,
LPARAM lpData
)
{
TCHAR Msg [100];
switch (uMsg)
{
case BFFM_VALIDATEFAILEDW:
{
if (StringCbPrintf(Msg, sizeof Msg, _T("The selected directory is not valid.\r\nPlease select another directory."))!= S_OK)
{
goto ERRORS;
}
MessageBox(hwnd, Msg, NULL,MB_OK);
return 1;
}
default:
return 0;
}
ERRORS:
return 0;
}
void DoCreateFileTree(HWND hwnd)
/*
*/
{
TCHAR szSubDir[MAX_PATH];
DWORD ErrorCode;
WCHAR wzPath[MAX_PATH];
BROWSEINFOW bi;
ACL *NewAcl;
EXPLICIT_ACCESS eAccess[3];
TCHAR szPath[MAX_PATH];
ULARGE_INTEGER FreeBytesAvailable;
ULARGE_INTEGER TotalFreeBytes ;
ULARGE_INTEGER TotalBytes ;
LPMALLOC pMalloc = NULL;
PSID pSidAdministrators = NULL;
PSID pSidEveryone = NULL;
ZeroMemory (&bi, sizeof BROWSEINFOW);
bi.hwndOwner = hwnd;
bi.pidlRoot = NULL;
bi.pszDisplayName = wzPath;
bi.lpszTitle = L"Select a file share to be used as the root of the CER file tree...";
bi.ulFlags = BIF_USENEWUI | BIF_RETURNONLYFSDIRS | BIF_EDITBOX | BIF_VALIDATE;
bi.lpfn = BrowseCallbackProc;
SID_IDENTIFIER_AUTHORITY pSIANTAuth = SECURITY_NT_AUTHORITY;
SID_IDENTIFIER_AUTHORITY pSIAWorld = SECURITY_WORLD_SID_AUTHORITY;
CoInitialize(NULL);
ZeroMemory (wzPath, sizeof wzPath);
ITEMIDLIST *pidl = SHBrowseForFolderW(&bi);
if (SHGetMalloc(&pMalloc) == E_FAIL)
{
// we couldn't get to the shell
MessageBox(hwnd, _T("Failed to open the selected directory"), NULL,MB_OK);
goto DONE;
}
if (!wcscmp(wzPath, L"My Documents"))
{
MessageBox(hwnd, _T("Failed to open the selected directory"), NULL,MB_OK);
goto DONE;
}
ZeroMemory(&FreeBytesAvailable, sizeof ULARGE_INTEGER);
ZeroMemory(&TotalFreeBytes, sizeof ULARGE_INTEGER);
ZeroMemory(&TotalBytes, sizeof ULARGE_INTEGER);
if (pidl != NULL)
{
if (SHGetPathFromIDList(pidl, szPath))
{
if (StringCbCopy (CerRoot,sizeof CerRoot, szPath) != S_OK)
{
goto ERRORS;
}
if (PathIsDirectory(szPath))
{
if (!FileSystemIsNTFS(szPath))
{
MessageBox(hwnd, _T("The CER tree can only be created on an NTFS file system"), NULL, MB_OK);
goto DONE;
}
// Now check for space
if (!GetDiskFreeSpaceEx(szPath, &FreeBytesAvailable, &TotalBytes,&TotalFreeBytes))
{
// We were unable to retrieve the space info.
// Now what.
goto ERRORS;
}
else
{
if (FreeBytesAvailable.QuadPart < 2000000000)
{
MessageBox(hwnd, _T("There is not enough free space to create a CER tree.\r\nA minimum of 2GB free space is required."), NULL, MB_OK);
goto DONE;
}
}
// Builtin\Administrators
AllocateAndInitializeSid(&pSIANTAuth, 2, SECURITY_BUILTIN_DOMAIN_RID, DOMAIN_ALIAS_RID_ADMINS, 0, 0, 0, 0, 0, 0, &pSidAdministrators);
// Everyone
AllocateAndInitializeSid(&pSIAWorld, 1, SECURITY_WORLD_RID, 0, 0, 0, 0, 0, 0, 0, &pSidEveryone);
// Create the cabs subdirectory
if (StringCbPrintf(szSubDir, sizeof szSubDir, _T("%s\\Cabs"), szPath) != S_OK)
{
goto ERRORS;
}
if ( !CreateDirectory(szSubDir,NULL))
{
ErrorCode = 2;
goto ERRORS;
}
else
{
// Set Everyone = WriteAccess
ZeroMemory (eAccess, sizeof EXPLICIT_ACCESS);
eAccess[0].grfAccessMode = SET_ACCESS;
eAccess[0].grfInheritance = SUB_CONTAINERS_AND_OBJECTS_INHERIT ;
eAccess[0].grfAccessPermissions = FILE_APPEND_DATA | FILE_WRITE_DATA | FILE_WRITE_ATTRIBUTES | FILE_WRITE_EA ;
//eAccess[1].Trustee.TrusteeForm = TRUSTEE_IS_NAME;
eAccess[0].Trustee.TrusteeForm = TRUSTEE_IS_SID;
eAccess[0].Trustee.TrusteeType = TRUSTEE_IS_GROUP;
//eAccess[1].Trustee.ptstrName = _T("Everyone");
eAccess[0].Trustee.ptstrName = (LPTSTR) pSidEveryone;
eAccess[2].grfAccessMode = SET_ACCESS;
eAccess[2].grfInheritance = SUB_CONTAINERS_AND_OBJECTS_INHERIT ;
eAccess[2].grfAccessPermissions = GENERIC_ALL;
eAccess[2].Trustee.TrusteeForm = TRUSTEE_IS_SID;
eAccess[2].Trustee.TrusteeType = TRUSTEE_IS_GROUP;
//eAccess[2].Trustee.ptstrName = _T("Administrators");
eAccess[2].Trustee.ptstrName = (LPTSTR) pSidAdministrators;
eAccess[1].grfAccessMode = SET_ACCESS;
eAccess[1].grfInheritance =CONTAINER_INHERIT_ACE ;
eAccess[1].grfAccessPermissions = FILE_EXECUTE | GENERIC_READ | FILE_LIST_DIRECTORY;
eAccess[1].Trustee.TrusteeForm = TRUSTEE_IS_SID;
eAccess[1].Trustee.TrusteeType = TRUSTEE_IS_GROUP;
eAccess[1].Trustee.ptstrName = (LPTSTR) pSidEveryone;
SetEntriesInAcl( 3, eAccess, NULL, &NewAcl);
ErrorCode = SetNamedSecurityInfo(szSubDir,
SE_FILE_OBJECT,
DACL_SECURITY_INFORMATION |
PROTECTED_DACL_SECURITY_INFORMATION |
PROTECTED_SACL_SECURITY_INFORMATION,
NULL,
NULL,
NewAcl,
NULL) != ERROR_SUCCESS;
if (ErrorCode != ERROR_SUCCESS)
{
LocalFree((HLOCAL) NewAcl);
goto ERRORS;
}
else
LocalFree((HLOCAL) NewAcl);
}
// Create the counts SubDirecotry
if (StringCbPrintf(szSubDir, sizeof szSubDir, _T("%s\\Counts"), szPath) != S_OK)
{
goto ERRORS;
}
if ( !CreateDirectory(szSubDir,NULL))
{
ErrorCode = 2;
goto ERRORS;
}
else
{
// Set Everyone = WriteAccess
ZeroMemory (eAccess, sizeof EXPLICIT_ACCESS);
eAccess[0].grfAccessMode = SET_ACCESS;
eAccess[0].grfInheritance =SUB_CONTAINERS_AND_OBJECTS_INHERIT ;
eAccess[0].grfAccessPermissions = FILE_LIST_DIRECTORY | GENERIC_READ | GENERIC_WRITE | FILE_APPEND_DATA;
eAccess[0].Trustee.TrusteeForm = TRUSTEE_IS_SID;
eAccess[0].Trustee.TrusteeType = TRUSTEE_IS_GROUP;
eAccess[0].Trustee.ptstrName = (LPTSTR) pSidEveryone;
eAccess[1].grfAccessMode = SET_ACCESS;
eAccess[1].grfInheritance = SUB_CONTAINERS_AND_OBJECTS_INHERIT;
eAccess[1].grfAccessPermissions = GENERIC_ALL;
eAccess[1].Trustee.TrusteeForm = TRUSTEE_IS_SID;
eAccess[1].Trustee.TrusteeType = TRUSTEE_IS_GROUP;
eAccess[1].Trustee.ptstrName = (LPTSTR) pSidAdministrators;
eAccess[2].grfAccessMode = SET_ACCESS;
eAccess[2].grfInheritance =CONTAINER_INHERIT_ACE ;
eAccess[2].grfAccessPermissions = FILE_EXECUTE | GENERIC_READ | FILE_LIST_DIRECTORY | FILE_APPEND_DATA;
eAccess[2].Trustee.TrusteeForm = TRUSTEE_IS_SID;
eAccess[2].Trustee.TrusteeType = TRUSTEE_IS_GROUP;
eAccess[2].Trustee.ptstrName =(LPTSTR) pSidEveryone;
SetEntriesInAcl( 3, eAccess, NULL, &NewAcl);
ErrorCode = SetNamedSecurityInfo(szSubDir,
SE_FILE_OBJECT,
DACL_SECURITY_INFORMATION |
PROTECTED_DACL_SECURITY_INFORMATION |
PROTECTED_SACL_SECURITY_INFORMATION,
NULL,
NULL,
NewAcl,
NULL) != ERROR_SUCCESS;
if (ErrorCode != ERROR_SUCCESS)
{
LocalFree((HLOCAL) NewAcl);
goto ERRORS;
}
else
LocalFree((HLOCAL) NewAcl);
}
// Create the Status SubDirecotry
if (StringCbPrintf(szSubDir, sizeof szSubDir, _T("%s\\Status"), szPath) != S_OK)
{
goto ERRORS;
}
if ( !CreateDirectory(szSubDir,NULL))
{
ErrorCode = 2;
goto ERRORS;
}
else
{
ZeroMemory (eAccess, sizeof EXPLICIT_ACCESS);
eAccess[2].grfAccessMode = SET_ACCESS;
eAccess[2].grfInheritance =SUB_CONTAINERS_AND_OBJECTS_INHERIT ;
eAccess[2].grfAccessPermissions = FILE_LIST_DIRECTORY | GENERIC_READ;
eAccess[2].Trustee.TrusteeForm = TRUSTEE_IS_SID;
eAccess[2].Trustee.TrusteeType = TRUSTEE_IS_GROUP;
eAccess[2].Trustee.ptstrName = (LPTSTR) pSidEveryone;
eAccess[1].grfAccessMode = SET_ACCESS;
eAccess[1].grfInheritance = SUB_CONTAINERS_AND_OBJECTS_INHERIT;
eAccess[1].grfAccessPermissions = GENERIC_ALL;
eAccess[1].Trustee.TrusteeForm = TRUSTEE_IS_SID;
eAccess[1].Trustee.TrusteeType = TRUSTEE_IS_GROUP;
eAccess[1].Trustee.ptstrName = (LPTSTR) pSidAdministrators;
eAccess[0].grfAccessMode = SET_ACCESS;
eAccess[0].grfInheritance =CONTAINER_INHERIT_ACE ;
eAccess[0].grfAccessPermissions = FILE_EXECUTE | GENERIC_READ | FILE_LIST_DIRECTORY | FILE_APPEND_DATA;
eAccess[0].Trustee.TrusteeForm = TRUSTEE_IS_SID;
eAccess[0].Trustee.TrusteeType = TRUSTEE_IS_GROUP;
eAccess[0].Trustee.ptstrName =(LPTSTR) pSidEveryone;
SetEntriesInAcl( 3, eAccess, NULL, &NewAcl);
ErrorCode = SetNamedSecurityInfo(szSubDir,
SE_FILE_OBJECT,
DACL_SECURITY_INFORMATION |
PROTECTED_DACL_SECURITY_INFORMATION |
PROTECTED_SACL_SECURITY_INFORMATION,
NULL,
NULL,
NewAcl,
NULL) != ERROR_SUCCESS;
if (ErrorCode != ERROR_SUCCESS)
{
LocalFree((HLOCAL) NewAcl);
goto ERRORS;
}
else
LocalFree((HLOCAL) NewAcl);
}
}
}
MessageBox(hwnd, szPath, _T("Successfully created CER file tree in: "), MB_OK);
}
if(pidl)
{
if (pMalloc)
pMalloc->Free(pidl);
}
if (pMalloc)
pMalloc->Release();
if (pSidAdministrators)
FreeSid(pSidAdministrators);
if (pSidEveryone)
FreeSid(pSidEveryone);
CoUninitialize();
return;
ERRORS:
MessageBox(hwnd, _T("Failed to Create CER File Tree."), NULL,MB_OK);
DONE:
if(pidl)
{
if (pMalloc)
pMalloc->Free(pidl);
}
if (pMalloc)
pMalloc->Release();
if (pSidAdministrators)
FreeSid(pSidAdministrators);
if (pSidEveryone)
FreeSid(pSidEveryone);
CoUninitialize();
return;
}
void PopulateMruList(HWND hwnd, BOOL bUpdate)
{
HMENU hmenu = GetSubMenu(GetMenu(hwnd), 0);
HMENU hNewMenu = NULL;
int iCount = 0;
if (!bUpdate)
{
hNewMenu = CreateMenu();
if (hmenu && hNewMenu)
{
if (_tcscmp(MruList.szMRU1, _T("\0")))
{
AppendMenu(hNewMenu, MF_STRING, ID_FILE_MRU1, MruList.szMRU1);
ModifyMenu(hmenu, ID_FILE_RECENTFILETREES165, MF_BYCOMMAND | MF_POPUP, (UINT_PTR) hNewMenu, _T("Recent File &Trees"));
if (_tcscmp(MruList.szMRU2, _T("\0")))
{
AppendMenu(hNewMenu, MF_STRING, ID_FILE_MRU2, MruList.szMRU2);
}
if (_tcscmp(MruList.szMRU3, _T("\0")))
{
AppendMenu(hNewMenu, MF_STRING, ID_FILE_MRU3, MruList.szMRU3);
}
if (_tcscmp(MruList.szMRU3, _T("\0")))
{
AppendMenu(hNewMenu, MF_STRING, ID_FILE_MRU4, MruList.szMRU4);
}
}
}
}
else
{
hNewMenu= GetSubMenu(hmenu, 6);
if (!hNewMenu)
{
hNewMenu = CreateMenu();
if (_tcscmp(MruList.szMRU1, _T("\0")))
{
ModifyMenu(hmenu, ID_FILE_RECENTFILETREES165, MF_BYCOMMAND | MF_POPUP, (UINT_PTR) hNewMenu, _T("Recent File &Trees"));
AppendMenu(hNewMenu, MF_STRING, ID_FILE_MRU1, MruList.szMRU1);
}
}
else
{
iCount = GetMenuItemCount(hNewMenu);
// Delete the old menu items and rebuild the menu
if (iCount == 4)
{
DeleteMenu(hNewMenu, ID_FILE_MRU4, MF_BYCOMMAND);
DeleteMenu(hNewMenu, ID_FILE_MRU3, MF_BYCOMMAND);
DeleteMenu(hNewMenu, ID_FILE_MRU2, MF_BYCOMMAND);
DeleteMenu(hNewMenu, ID_FILE_MRU1, MF_BYCOMMAND);
AppendMenu(hNewMenu, MF_STRING, ID_FILE_MRU1, MruList.szMRU1);
AppendMenu(hNewMenu, MF_STRING, ID_FILE_MRU2, MruList.szMRU2);
AppendMenu(hNewMenu, MF_STRING, ID_FILE_MRU3, MruList.szMRU3);
AppendMenu(hNewMenu, MF_STRING, ID_FILE_MRU4, MruList.szMRU4);
}
else
{
switch (iCount)
{
case 3:
// display Fourth
if (_tcscmp(MruList.szMRU4, _T("\0")))
{
AppendMenu(hNewMenu, MF_STRING, ID_FILE_MRU4, MruList.szMRU4);
}
break;
case 2:
// display third
if (_tcscmp(MruList.szMRU3, _T("\0")))
{
AppendMenu(hNewMenu, MF_STRING, ID_FILE_MRU3, MruList.szMRU3);
}
break;
case 1:
// Display second
if (_tcscmp(MruList.szMRU2, _T("\0")))
{
AppendMenu(hNewMenu, MF_STRING, ID_FILE_MRU2, MruList.szMRU2);
}
break;
default:
// All the slots were full we need to change the menu item
ModifyMenu(hNewMenu, ID_FILE_MRU4, MF_BYCOMMAND | MF_STRING, ID_FILE_MRU4, MruList.szMRU4);
;
}
}
}
}
}
void OnDlgInit(HWND hwnd)
{
HICON hIcon = LoadIcon(g_hinst, MAKEINTRESOURCE(IDI_MAIN));
if (hIcon)
{
SendMessage(hwnd, WM_SETICON, ICON_BIG, (LPARAM)hIcon);
SendMessage(hwnd, WM_SETICON, ICON_SMALL, (LPARAM)hIcon);
}
if (StringCbCopy(CerRoot, sizeof CerRoot, _T("\0")) != S_OK)
{
goto ERRORS;
}
MakeHelpFileName();
// Disable Menu items that cannot be accessed until a filetree is loaded
HMENU hMenu = GetMenu(hwnd);
if (hMenu)
{
//EnableMenuItem (hMenu, ID_EDIT_COPY115, MF_BYCOMMAND| MF_GRAYED);
EnableMenuItem (hMenu, ID_EDIT_SELECTALL , MF_BYCOMMAND| MF_GRAYED);
EnableMenuItem (hMenu, ID_EDIT_DEFAULTPOLICY , MF_BYCOMMAND| MF_GRAYED);
EnableMenuItem (hMenu, ID_EDIT_SELECTEDBUCKETSPOLICY , MF_BYCOMMAND| MF_GRAYED);
EnableMenuItem (hMenu, ID_REPORT_ALLCRASHES , MF_BYCOMMAND| MF_GRAYED);
EnableMenuItem (hMenu, ID_REPORT_SELECTEDFAULTS , MF_BYCOMMAND| MF_GRAYED);
EnableMenuItem (hMenu, ID_VIEW_BUCKETCABFILEDIRECTORY , MF_BYCOMMAND| MF_GRAYED);
EnableMenuItem (hMenu, ID_VIEW_BUCKETINSTANCEDATA , MF_BYCOMMAND| MF_GRAYED);
EnableMenuItem (hMenu, ID_VIEW_RESPONSESELECTED , MF_BYCOMMAND| MF_GRAYED);
EnableMenuItem (hMenu, ID_VIEW_REFRESH , MF_BYCOMMAND| MF_GRAYED);
//EnableMenuItem (hMenu, ID_FILE_EXPORTBUCKETS , MF_BYCOMMAND| MF_GRAYED);
//EnableMenuItem (hMenu, ID_FILE_EXPORTSELECTEDBUCKETS , MF_BYCOMMAND| MF_GRAYED);
EnableMenuItem (hMenu, ID_FILE_RELOADFILETREE, MF_BYCOMMAND| MF_GRAYED);
EnableMenuItem (hMenu, ID_REPORT_USERMODEFAULTS, MF_BYCOMMAND | MF_GRAYED);
EnableMenuItem (hMenu, ID_REPORT_KERNELMODEFAULTS, MF_BYCOMMAND | MF_GRAYED);
EnableMenuItem (hMenu, ID_VIEW_CRASHLOG, MF_BYCOMMAND | MF_GRAYED);
// EnableMenuItem (hMenu, ID_EXPORT_USERMODEBUCKETDATA, MF_BYCOMMAND | MF_GRAYED);
// EnableMenuItem (hMenu, ID_EXPORT_KERNELMODEFAULTDATA, MF_BYCOMMAND | MF_GRAYED);
}
// Create the User and Kernel mode dialog boxes.
ReportMode = USER_MODE;
DoKernelMode(hwnd);
ReportMode = KERNEL_MODE;
DoUserMode(hwnd);
if (LoadMruList())
{
//Populate the file menu with the mrulist;
PopulateMruList(hwnd, FALSE);
}
ERRORS:
return;
}
int GetTreePBarRange()
{
int Range = 10;
// Count the directories to load - 2 for the . and .. directories.
return Range;
}
DWORD WINAPI tfCollectBucketStart (void *ThreadParam)
{
// Get progress bar range values
// Load the user mode tree
if (!GetAllBuckets (*((HWND *) ThreadParam)))
{
MessageBox(*((HWND *) ThreadParam), _T("Failed to load the CER file tree."), NULL,MB_OK);
}
else
{
PostMessage(hUserMode, WM_FileTreeLoaded, 0,0);
// Load the kernel mode tree
PostMessage(hKrnlMode, WM_FileTreeLoaded, 0,0);
}
PostMessage(*((HWND *) ThreadParam), WmSyncDone, FALSE, 0);
return TRUE;
}
void CenterDialogInParent(HWND hwnd)
{
HWND hParent = GetParent(hwnd);
RECT rcParent;
RECT rcDlg;
RECT newRect;
GetWindowRect(hParent, &rcParent);
GetWindowRect(hwnd, &rcDlg);
newRect.left = rcParent.left + (rcParent.right - rcParent.left) /2 - (rcDlg.right - rcDlg.left) /2;
newRect.top = rcParent.top + (rcParent.bottom - rcParent.top) /2 - (rcDlg.bottom - rcDlg.top) /2;
MoveWindow(hwnd, newRect.left, newRect.top, rcDlg.right - rcDlg.left , rcDlg.bottom - rcDlg.top, TRUE);
}
BOOL CALLBACK LoadDlgProc(HWND hDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
HANDLE hThread = NULL;
// DWORD dw;
LoadThreadParam = hDlg;
switch (uMsg)
{
case WM_INITDIALOG:
CenterDialogInParent(hDlg);
// Set Dialog Title
TCHAR DialogTitle[MAX_PATH];
if (StringCbPrintf(DialogTitle, sizeof DialogTitle, _T("Loading File Tree: %s"), CerRoot)!= S_OK)
{
goto ERRORS;
}
SendMessage(hDlg, WM_SETTEXT, 0, (LPARAM)DialogTitle);
SendDlgItemMessage(hDlg, IDC_PB, PBM_SETSTEP, 1, 0);
hThread = CreateThread(NULL, 0, tfCollectBucketStart, &LoadThreadParam, 0, NULL);
if (hThread)
CloseHandle(hThread);
return TRUE;
case WM_CLOSE:
// Cancel the tree load and clear the Data structures.
EndDialog(hDlg, TRUE);
return TRUE;
case WmSyncDone:
EndDialog(hDlg, 1);
return TRUE;
}
ERRORS:
return FALSE;
}
BOOL DoLoadFileTree(HWND hwnd, TCHAR *Directory, BOOL bGetDir)
{
WCHAR wzPath[MAX_PATH];
BROWSEINFOW bi;
TCHAR szPath[MAX_PATH];
TCHAR szSubDir[MAX_PATH];
ZeroMemory (szSubDir, sizeof szSubDir);
ZeroMemory (szPath, sizeof szPath);
if (bGetDir)
{
bi.hwndOwner = hwnd;
bi.pidlRoot = NULL;
bi.pszDisplayName = wzPath;
bi.lpszTitle = L"Select a CER File Tree to open...";
bi.ulFlags = BIF_USENEWUI | BIF_RETURNONLYFSDIRS | BIF_EDITBOX;
bi.lpfn = NULL;
ITEMIDLIST *pidl = SHBrowseForFolderW(&bi);
if (pidl != NULL)
{
if (!SHGetPathFromIDList(pidl, szPath))
{
goto CANCELED;
}
}
else
{
goto CANCELED;
}
}
else
{
if (!Directory)
{
goto CANCELED;
}
if (StringCbCopy (szPath, sizeof szPath, Directory) != S_OK)
{
goto ERRORS;
}
}
if (PathIsDirectory(szPath))
{
if (StringCbPrintf(szSubDir, sizeof szSubDir, _T("%s\\Cabs"), szPath) != S_OK)
{
goto ERRORS;
}
}
else
{
if (!PathIsDirectory(szSubDir))
{
goto ERRORS;
}
}
if (StringCbPrintf(szSubDir, sizeof szSubDir, _T("%s\\Counts"), szPath) != S_OK)
{
goto ERRORS;
}
else
{
if (!PathIsDirectory(szSubDir))
{
goto ERRORS;
}
}
if (StringCbPrintf(szSubDir, sizeof szSubDir, _T("%s\\Status"), szPath) != S_OK)
{
goto ERRORS;
}
else
{
if (!PathIsDirectory(szSubDir))
{
goto ERRORS;
}
}
if (StringCbCopy (CerRoot,sizeof CerRoot, szPath) != S_OK)
{
goto ERRORS;
}
if (!DialogBox(g_hinst, MAKEINTRESOURCE(IDD_LOAD), hwnd, (DLGPROC) LoadDlgProc))
{
//MessageBox(NULL, "FAILED TO LOAD TREE", NULL, MB_OK);
goto ERRORS;
}
else
{
AddToMruList(hwnd, CerRoot);
SaveMruList();
PopulateMruList(hwnd, TRUE);
}
return TRUE;
ERRORS:
MessageBox(hwnd, _T("Failed to load the CER file tree."), NULL,MB_OK);
CANCELED:
return FALSE;
}
void OnAboutBoxInit(HWND hwnd)
{
TCHAR wz1[MAX_PATH];
TCHAR wz2[MAX_PATH];
HRSRC hRsrc = FindResourceW(GetModuleHandle(NULL), MAKEINTRESOURCEW(1), MAKEINTRESOURCEW(RT_VERSION));
if (!hRsrc)
{
goto ERRORS;
}
void *pver = LoadResource(GetModuleHandle(NULL), hRsrc);
if (!pver)
{
goto ERRORS;
}
VS_FIXEDFILEINFO *pVer = NULL;
UINT dwVer;
VerQueryValue(pver, _T("\\"), (void **)&pVer, &dwVer);
if (!pVer)
{
goto ERRORS;
}
if (StringCbCopy(wz1, sizeof wz1, _T("Version %d.%d.%d.%d")) != S_OK)
{
goto ERRORS;
}
if (StringCbPrintf(wz2,sizeof wz2, wz1,
HIWORD(pVer->dwProductVersionMS), LOWORD(pVer->dwProductVersionMS),
HIWORD(pVer->dwProductVersionLS), LOWORD(pVer->dwProductVersionLS)) != S_OK)
{
goto ERRORS;
}
SetDlgItemText(hwnd, IDC_VERSION_STRING, wz2);
WORD *lpTranslate = NULL;
VerQueryValue(pver,_T("\\VarFileInfo\\Translation"), (void **)&lpTranslate, &dwVer);
if (!lpTranslate)
{
goto ERRORS;
}
if (StringCbPrintf(wz1,sizeof wz1, _T("\\StringFileInfo\\%04x%04x\\LegalCopyright"), lpTranslate[0], lpTranslate[1]) != S_OK)
{
goto ERRORS;
}
TCHAR *wzCopyright = NULL;
VerQueryValue(pver, wz1, (void **)&wzCopyright, &dwVer);
if (!wzCopyright)
{
goto ERRORS;
}
SetDlgItemText(hwnd, IDC_COPYRIGHT_STRING, wzCopyright);
ERRORS:
return;
}
LRESULT CALLBACK AboutDlgProc(HWND hwnd, UINT iMsg, WPARAM wParam, LPARAM lParam)
{
switch (iMsg)
{
case WM_INITDIALOG:
OnAboutBoxInit (hwnd);
return TRUE;
case WM_COMMAND:
{
switch (LOWORD(wParam))
{
case IDOK:
EndDialog(hwnd, TRUE);
return TRUE;
}
}
case WM_DESTROY:
EndDialog(hwnd, TRUE);
return TRUE;
default:
break;
}
return FALSE;
}
void EnableMenuItems(HWND hwnd)
{
// Main menu
HMENU hMenu = GetMenu(hwnd);
//
//EnableMenuItem (hMenu, ID_EDIT_SELECTALL , MF_BYCOMMAND| MF_ENABLED);
if (hMenu)
{
EnableMenuItem (hMenu, ID_EDIT_DEFAULTPOLICY , MF_BYCOMMAND| MF_ENABLED);
EnableMenuItem (hMenu, ID_EDIT_SELECTEDBUCKETSPOLICY , MF_BYCOMMAND| MF_ENABLED);
EnableMenuItem (hMenu, ID_REPORT_ALLCRASHES , MF_BYCOMMAND| MF_ENABLED);
EnableMenuItem (hMenu, ID_REPORT_SELECTEDFAULTS , MF_BYCOMMAND| MF_ENABLED);
EnableMenuItem (hMenu, ID_VIEW_BUCKETCABFILEDIRECTORY , MF_BYCOMMAND| MF_ENABLED);
EnableMenuItem (hMenu, ID_VIEW_BUCKETINSTANCEDATA , MF_BYCOMMAND| MF_ENABLED);
EnableMenuItem (hMenu, ID_VIEW_RESPONSESELECTED , MF_BYCOMMAND| MF_ENABLED);
EnableMenuItem (hMenu, ID_VIEW_REFRESH , MF_BYCOMMAND| MF_ENABLED);
//EnableMenuItem (hMenu, ID_FILE_EXPORTBUCKETS , MF_BYCOMMAND| MF_ENABLED);
//EnableMenuItem (hMenu, ID_FILE_EXPORTSELECTEDBUCKETS , MF_BYCOMMAND| MF_ENABLED);
EnableMenuItem (hMenu, ID_FILE_RELOADFILETREE, MF_BYCOMMAND| MF_ENABLED);
EnableMenuItem (hMenu, ID_REPORT_USERMODEFAULTS, MF_BYCOMMAND | MF_ENABLED);
EnableMenuItem (hMenu, ID_REPORT_KERNELMODEFAULTS, MF_BYCOMMAND | MF_ENABLED);
EnableMenuItem (hMenu, ID_VIEW_CRASHLOG, MF_BYCOMMAND | MF_ENABLED);
//EnableMenuItem (hMenu, ID_EDIT_COPY115, MF_BYCOMMAND| MF_ENABLED);
if ( !g_bAdminAccess)
{
EnableMenuItem (hMenu, ID_REPORT_ALLCRASHES , MF_BYCOMMAND| MF_GRAYED);
//EnableMenuItem (hMenu, ID_REPORT_SELECTEDCRASHES , MF_BYCOMMAND| MF_GRAYED);
EnableMenuItem (hMenu, ID_REPORT_SELECTEDFAULTS , MF_BYCOMMAND| MF_GRAYED);
EnableMenuItem (hMenu, ID_REPORT_USERMODEFAULTS, MF_BYCOMMAND | MF_GRAYED);
EnableMenuItem (hMenu, ID_REPORT_KERNELMODEFAULTS, MF_BYCOMMAND | MF_GRAYED);
EnableMenuItem (hMenu, ID_EDIT_DEFAULTPOLICY , MF_BYCOMMAND| MF_GRAYED);
EnableMenuItem (hMenu, ID_EDIT_SELECTEDBUCKETSPOLICY , MF_BYCOMMAND| MF_GRAYED);
}
}
// EnableMenuItem (hMenu, ID_EXPORT_USERMODEBUCKETDATA, MF_BYCOMMAND | MF_ENABLED);
// EnableMenuItem (hMenu, ID_EXPORT_KERNELMODEFAULTDATA, MF_BYCOMMAND | MF_ENABLED);
}
BOOL SaveMruList()
{
HKEY hHKLM = NULL;
HKEY hMruKey = NULL;
// BYTE Buffer[(MAX_PATH + 1) * sizeof TCHAR] ;
// DWORD Type;
DWORD BufferSize = MAX_PATH +1; // Set for largest value
BOOL Status = TRUE;
if(!RegConnectRegistry(NULL, HKEY_CURRENT_USER, &hHKLM))
{
if(RegOpenKeyEx(hHKLM,_T("Software\\Microsoft\\CER1.5"), 0, KEY_ALL_ACCESS, &hMruKey))
{
// The key probably doesn't exists try to create it.
if (RegCreateKey(hHKLM, _T("Software\\Microsoft\\CER1.5"), &hMruKey))
{
// Ok Now What
goto ERRORS;
}
}
if (hMruKey)
{
// Get the input queue directory path
BufferSize = (_tcslen(MruList.szMRU1) + 1 ) * sizeof TCHAR;
if (BufferSize > sizeof TCHAR)
{
if (RegSetValueEx(hMruKey,_T("CerMRU1"), 0, REG_SZ,(BYTE *) MruList.szMRU1, BufferSize) != ERROR_SUCCESS)
{
//LogEvent(_T("Failed to get InputQueue value from registry."));
Status = FALSE;
}
}
BufferSize = (_tcslen(MruList.szMRU2) + 1 ) * sizeof TCHAR;
if (BufferSize > sizeof TCHAR)
{
if (RegSetValueEx(hMruKey,_T("CerMRU2"), 0, REG_SZ,(BYTE *) MruList.szMRU2, BufferSize) != ERROR_SUCCESS)
{
//LogEvent(_T("Failed to get InputQueue value from registry."));
Status = FALSE;
}
}
BufferSize = (_tcslen(MruList.szMRU3) + 1 ) * sizeof TCHAR;
if (BufferSize > sizeof TCHAR)
{
if (RegSetValueEx(hMruKey,_T("CerMRU3"), 0, REG_SZ, (BYTE *)MruList.szMRU3, BufferSize) != ERROR_SUCCESS)
{
//LogEvent(_T("Failed to get InputQueue value from registry."));
Status = FALSE;
}
}
BufferSize = (_tcslen(MruList.szMRU4) + 1 ) * sizeof TCHAR;
if (BufferSize > sizeof TCHAR)
{
if (RegSetValueEx(hMruKey,_T("CerMRU4"), 0, REG_SZ, (BYTE *)MruList.szMRU4, BufferSize) != ERROR_SUCCESS)
{
Status = FALSE;
}
}
if (hMruKey)
{
RegCloseKey(hMruKey);
hMruKey = NULL;
}
}
if (hHKLM)
{
RegCloseKey(hHKLM);
hHKLM = NULL;
}
return TRUE;
}
ERRORS:
if (hMruKey)
{
RegCloseKey(hMruKey);
}
if (hHKLM)
{
RegCloseKey(hHKLM);
}
return FALSE;
}
// Load the mru's
BOOL LoadMruList()
{
HKEY hHKLM = NULL;
HKEY hMruKey = NULL;
// TCHAR ErrorString[20];
BYTE Buffer[(MAX_PATH + 1) * sizeof TCHAR] ;
DWORD Type;
DWORD BufferSize = MAX_PATH +1; // Set for largest value
BOOL Status = TRUE;
BOOL bNeedSave = FALSE;
if(!RegConnectRegistry(NULL, HKEY_CURRENT_USER, &hHKLM))
{
if(RegOpenKeyEx(hHKLM,_T("Software\\Microsoft\\CER1.5"), 0, KEY_ALL_ACCESS, &hMruKey))
{
hMruKey = NULL;
if (RegOpenKeyEx(hHKLM, _T("Software\\Microsoft\\Office\\10.0\\ResourceKit"), 0, KEY_ALL_ACCESS, &hMruKey))
{
goto ERRORS;
}
else
{
bNeedSave = TRUE;
}
}
if (hMruKey)
{
// Get the input queue directory path
BufferSize = MAX_PATH +1;
ZeroMemory(Buffer, BufferSize);
if (RegQueryValueEx(hMruKey,_T("CerMRU1"), 0, &Type, Buffer, &BufferSize) != ERROR_SUCCESS)
{
//LogEvent(_T("Failed to get InputQueue value from registry."));
Status = FALSE;
}
else
{
if (StringCbCopy(MruList.szMRU1, sizeof MruList.szMRU1, (TCHAR *)Buffer)!= S_OK)
{
goto ERRORS;
}
}
BufferSize = MAX_PATH +1;
ZeroMemory(Buffer, BufferSize);
if (RegQueryValueEx(hMruKey,_T("CerMRU2"), 0, &Type, Buffer, &BufferSize) != ERROR_SUCCESS)
{
//LogEvent(_T("Failed to get InputQueue value from registry."));
Status = FALSE;
}
else
{
if (StringCbCopy(MruList.szMRU2, sizeof MruList.szMRU2, (TCHAR *)Buffer)!= S_OK)
{
goto ERRORS;
}
}
BufferSize = MAX_PATH +1;
ZeroMemory(Buffer, BufferSize);
if (RegQueryValueEx(hMruKey,_T("CerMRU3"), 0, &Type, Buffer, &BufferSize) != ERROR_SUCCESS)
{
//LogEvent(_T("Failed to get InputQueue value from registry."));
Status = FALSE;
}
else
{
if (StringCbCopy(MruList.szMRU3, sizeof MruList.szMRU3, (TCHAR *)Buffer)!= S_OK)
{
goto ERRORS;
}
}
BufferSize = MAX_PATH +1;
ZeroMemory(Buffer, BufferSize);
if (RegQueryValueEx(hMruKey,_T("CerMRU4"), 0, &Type, Buffer, &BufferSize) != ERROR_SUCCESS)
{
//LogEvent(_T("Failed to get InputQueue value from registry."));
Status = FALSE;
}
else
{
if (StringCbCopy(MruList.szMRU4, sizeof MruList.szMRU4, (TCHAR *)Buffer)!= S_OK)
{
goto ERRORS;
}
}
RegCloseKey(hMruKey);
hMruKey = NULL;
}
RegCloseKey(hHKLM);
hHKLM = NULL;
if (bNeedSave)
SaveMruList();
return TRUE;
}
ERRORS:
if (hMruKey)
RegCloseKey (hMruKey);
if (hHKLM)
RegCloseKey (hHKLM);
return FALSE;
}
BOOL AddToMruList(HWND hwnd, TCHAR *NewEntry)
{
// TCHAR Entry[MAX_PATH];
// int EntryNumber = 0;
//if (StringCbPrintf(Entry, sizeof Entry, _T("
// if entry not found in list
if (!_tcscmp (NewEntry, MruList.szMRU1 + 4))
goto DONE;
if (!_tcscmp (NewEntry, MruList.szMRU2 + 4))
goto DONE;
if (!_tcscmp (NewEntry, MruList.szMRU3 + 4))
goto DONE;
if (!_tcscmp (NewEntry, MruList.szMRU4 + 4))
goto DONE;
// Find the first available slot or replace mru4
if (!_tcscmp (_T("\0"), MruList.szMRU1))
{
if (StringCbPrintf(MruList.szMRU1, sizeof MruList.szMRU1, _T("&1. %s"), NewEntry) != S_OK)
goto ERRORS;
}
else
{
if (!_tcscmp (_T("\0"), MruList.szMRU2 ))
{
if (StringCbPrintf(MruList.szMRU2, sizeof MruList.szMRU2, _T("&2. %s"), NewEntry) != S_OK)
goto ERRORS;
}
else
{
if (!_tcscmp (_T("\0"), MruList.szMRU3))
{
if (StringCbPrintf(MruList.szMRU3, sizeof MruList.szMRU3, _T("&3. %s"), NewEntry) != S_OK)
goto ERRORS;
}
else
{
if (!_tcscmp (_T("\0"), MruList.szMRU4))
{
if (StringCbPrintf(MruList.szMRU4, sizeof MruList.szMRU4, _T("&4. %s"), NewEntry) != S_OK)
goto ERRORS;
}
else
{
// all of the slots are full move each of the entries up one
// slot and add the new entry at the bottom of the list.
if (StringCbCopy(MruList.szMRU1 + 4, sizeof MruList.szMRU1, MruList.szMRU2 + 4) != S_OK)
{
goto ERRORS;
}
if (StringCbCopy(MruList.szMRU2 + 4, sizeof MruList.szMRU2, MruList.szMRU3 + 4) != S_OK)
{
goto ERRORS;
}
if (StringCbCopy(MruList.szMRU3 + 4, sizeof MruList.szMRU3, MruList.szMRU4 + 4) != S_OK)
{
goto ERRORS;
}
if (StringCbPrintf(MruList.szMRU4, sizeof MruList.szMRU4, _T("&4. %s"), NewEntry) != S_OK)
goto ERRORS;
}
}
}
}
// Build the new entry String
DONE:
return TRUE;
ERRORS:
return FALSE;
}
void ViewCrashLog()
{
TCHAR szCommandLine[MAX_PATH];
if (_tcscmp(CerRoot, _T("\0")))
{
ZeroMemory (szCommandLine, sizeof szCommandLine);
if (StringCbPrintf(szCommandLine, sizeof szCommandLine, _T("%s\\crash.log"), CerRoot)!= S_OK)
{
goto ERRORS;
}
if (_tcscmp(szCommandLine, _T("\0")))
{
SHELLEXECUTEINFOA sei = {0};
sei.cbSize = sizeof(sei);
sei.lpFile = szCommandLine;
sei.nShow = SW_SHOWDEFAULT;
if (! ShellExecuteEx(&sei) )
{
// What do we display here.
;
}
}
}
ERRORS:
return;
}
BOOL WriteGlobalPolicyFile()
{
TCHAR Buffer[1024];
HANDLE hFile = INVALID_HANDLE_VALUE;
TCHAR szPath[MAX_PATH];
DWORD dwBytesWritten = 0;
if (StringCbPrintf(szPath, sizeof szPath, _T("%s\\Policy.txt"), CerRoot) != S_OK)
{
goto ERRORS;
}
hFile = CreateFile(szPath,
GENERIC_WRITE,
NULL,
NULL,
CREATE_ALWAYS,
FILE_ATTRIBUTE_NORMAL,
NULL);
if (hFile == INVALID_HANDLE_VALUE)
goto ERRORS;
else
{
// write the Global Policy structure to the file
ZeroMemory(Buffer, sizeof Buffer);
if (_tcscmp(GlobalPolicy.EnableCrashTracking,_T("\0")))
{
if (StringCbPrintf(Buffer, sizeof Buffer, _T("%s%s\r\n"), TRACKING_PREFIX, GlobalPolicy.EnableCrashTracking)!= S_OK)
{
goto ERRORS;
}
else
{
if (!WriteFile(hFile, Buffer, _tcslen(Buffer) *sizeof TCHAR, &dwBytesWritten, NULL))
{
goto ERRORS;
}
}
}
ZeroMemory(Buffer, sizeof Buffer);
if (_tcscmp(GlobalPolicy.AllowAdvanced,_T("\0")))
{
if (StringCbPrintf(Buffer, sizeof Buffer, _T("%s%s\r\n"), FILE_COLLECTION_PREFIX, GlobalPolicy.AllowAdvanced )!= S_OK)
{
goto ERRORS;
}
else
{
if (!WriteFile(hFile, Buffer, _tcslen(Buffer) *sizeof TCHAR, &dwBytesWritten, NULL))
{
goto ERRORS;
}
}
}
ZeroMemory(Buffer, sizeof Buffer);
if (_tcscmp(GlobalPolicy.AllowBasic,_T("\0")))
{
if (StringCbPrintf(Buffer, sizeof Buffer, _T("%s%s\r\n"), SECOND_LEVEL_DATA_PREFIX, GlobalPolicy.AllowBasic )!= S_OK)
{
goto ERRORS;
}
else
{
if (!WriteFile(hFile, Buffer, _tcslen(Buffer) *sizeof TCHAR, &dwBytesWritten, NULL))
{
goto ERRORS;
}
}
}
ZeroMemory(Buffer, sizeof Buffer);
if (_tcscmp(GlobalPolicy.CustomURL,_T("\0")))
{
if (StringCbPrintf(Buffer, sizeof Buffer, _T("%s%s\r\n"),URLLAUNCH_PREFIX, GlobalPolicy.CustomURL )!= S_OK)
{
goto ERRORS;
}
else
{
if (!WriteFile(hFile, Buffer, _tcslen(Buffer) *sizeof TCHAR, &dwBytesWritten, NULL))
{
goto ERRORS;
}
}
}
ZeroMemory(Buffer, sizeof Buffer);
if (_tcscmp(GlobalPolicy.AllowMicrosoftResponse,_T("\0")))
{
if (StringCbPrintf(Buffer, sizeof Buffer, _T("%s%s\r\n"), ALLOW_EXTERNAL_PREFIX, GlobalPolicy.AllowMicrosoftResponse )!= S_OK)
{
goto ERRORS;
}
else
{
if (!WriteFile(hFile, Buffer, _tcslen(Buffer) *sizeof TCHAR, &dwBytesWritten, NULL))
{
goto ERRORS;
}
}
}
ZeroMemory(Buffer, sizeof Buffer);
if (_tcscmp(GlobalPolicy.CabsPerBucket,_T("\0")))
{
if (StringCbPrintf(Buffer, sizeof Buffer, _T("%s%s\r\n"), CRASH_PERBUCKET_PREFIX, GlobalPolicy.CabsPerBucket )!= S_OK)
{
goto ERRORS;
}
else
{
if (!WriteFile(hFile, Buffer, _tcslen(Buffer) *sizeof TCHAR, &dwBytesWritten, NULL))
{
goto ERRORS;
}
}
}
ZeroMemory(Buffer, sizeof Buffer);
if (_tcscmp(GlobalPolicy.RedirectFileServer,_T("\0")))
{
if (StringCbPrintf(Buffer, sizeof Buffer, _T("%s%s\r\n"), FILE_TREE_ROOT_PREFIX, GlobalPolicy.RedirectFileServer )!= S_OK)
{
goto ERRORS;
}
else
{
if (!WriteFile(hFile, Buffer, _tcslen(Buffer) *sizeof TCHAR, &dwBytesWritten, NULL))
{
goto ERRORS;
}
}
}
CloseHandle(hFile);
hFile = INVALID_HANDLE_VALUE;
RefreshUserMode(hUserMode);
return TRUE;
}
ERRORS:
if (hFile != INVALID_HANDLE_VALUE)
CloseHandle(hFile);
return FALSE;
}
BOOL InitEditPolicyDlg (HWND hwnd, HWND hMode, BOOL bUserMode, BOOL bGlobal)
{
HWND hList = NULL;
int sel = 0;
PUSER_DATA pUserData = NULL;
LVITEM lvi;
if (bUserMode)
{
hList = GetDlgItem(hUserMode, IDC_USER_LIST);
sel = ListView_GetNextItem(hList,-1, LVNI_SELECTED);
lvi.iItem = sel;
lvi.mask = LVIF_PARAM;
ListView_GetItem(hList, &lvi);
sel = lvi.lParam;
pUserData = cUserData.GetEntry(sel);
if (!pUserData)
{
goto ERRORS;
}
}
if (bGlobal)
{
if (_tcscmp (GlobalPolicy.AllowBasic, _T("YES")))
{
CheckDlgButton(hwnd, IDC_CHECK1,TRUE);
}
if (_tcscmp (GlobalPolicy.AllowAdvanced, _T("YES")))
{
CheckDlgButton(hwnd, IDC_CHECK2,TRUE);
}
if (_tcscmp (GlobalPolicy.AllowMicrosoftResponse, _T("YES")))
{
CheckDlgButton(hwnd, IDC_CHECK3,TRUE);
}
if (!_tcscmp (GlobalPolicy.EnableCrashTracking, _T("YES")))
{
CheckDlgButton(hwnd, IDC_CHECK4,TRUE);
}
SetDlgItemText(hwnd, IDC_EDIT1, GlobalPolicy.CustomURL);
SetDlgItemText(hwnd, IDC_EDIT2, GlobalPolicy.RedirectFileServer);
SetDlgItemText(hwnd, IDC_EDIT3, GlobalPolicy.CabsPerBucket);
}
else
{
// Get the selected bucket.
if (bUserMode)
{
if (pUserData)
{
if (_tcscmp (pUserData->Status.SecondLevelData, _T("YES")))
{
CheckDlgButton(hwnd, IDC_CHECK1,TRUE);
}
if (_tcscmp (pUserData->Status.FileCollection, _T("YES")))
{
CheckDlgButton(hwnd, IDC_CHECK2,TRUE);
}
if (_tcscmp (pUserData->Status.AllowResponse, _T("YES")))
{
CheckDlgButton(hwnd, IDC_CHECK3,TRUE);
}
if (!_tcscmp (pUserData->Status.Tracking, _T("YES")))
{
CheckDlgButton(hwnd, IDC_CHECK4,TRUE);
}
SetDlgItemText(hwnd, IDC_EDIT1, pUserData->Status.UrlToLaunch);
SetDlgItemText(hwnd, IDC_EDIT3, pUserData->Status.CrashPerBucketCount);
}
}
else
{
// we only have one status file location for Bluescreens
if (_tcscmp (CsvContents.KrnlPolicy.SecondLevelData, _T("YES")))
{
CheckDlgButton(hwnd, IDC_CHECK1,TRUE);
}
if (_tcscmp (CsvContents.KrnlPolicy.FileCollection, _T("YES")))
{
CheckDlgButton(hwnd, IDC_CHECK2,TRUE);
}
if (_tcscmp (CsvContents.KrnlPolicy.AllowResponse, _T("YES")))
{
CheckDlgButton(hwnd, IDC_CHECK3,TRUE);
}
if (!_tcscmp (CsvContents.KrnlPolicy.Tracking, _T("YES")))
{
CheckDlgButton(hwnd, IDC_CHECK4,TRUE);
}
SetDlgItemText(hwnd, IDC_EDIT1, CsvContents.KrnlPolicy.UrlToLaunch);
SetDlgItemText(hwnd, IDC_EDIT3, CsvContents.KrnlPolicy.CrashPerBucketCount);
}
}
return TRUE;
ERRORS:
return FALSE;
}
BOOL GetSelPolicyDlgData(BOOL bUserMode, HWND hwnd)
{
PUSER_DATA pUserData = NULL;
HWND hList = NULL;
int sel = 0;
LVITEM lvi;
TCHAR TempBuffer[MAX_PATH];
TCHAR *pTchar = NULL;
ZeroMemory (TempBuffer, sizeof TempBuffer);
if (bUserMode)
{
hList = GetDlgItem(hUserMode, IDC_USER_LIST);
sel = ListView_GetNextItem(hList,-1, LVNI_SELECTED);
lvi.iItem = sel;
lvi.mask = LVIF_PARAM;
ListView_GetItem(hList, &lvi);
sel = lvi.lParam;
pUserData = cUserData.GetEntry(sel);
if (!pUserData)
{
goto ERRORS;
}
}
// read the contents of the dialog box into the global policy structure.
if (IsDlgButtonChecked(hwnd, IDC_CHECK1))
{
if (bUserMode)
{
if (StringCbCopy(pUserData->Status.SecondLevelData, sizeof pUserData->Status.SecondLevelData, _T("NO") )!= S_OK)
{
goto ERRORS;
}
}
else
{
if (StringCbCopy(CsvContents.KrnlPolicy.SecondLevelData, sizeof CsvContents.KrnlPolicy.SecondLevelData, _T("NO") )!= S_OK)
{
goto ERRORS;
}
}
}
else
{
if (bUserMode)
{
if (StringCbCopy(pUserData->Status.SecondLevelData, sizeof pUserData->Status.SecondLevelData, _T("YES") )!= S_OK)
{
goto ERRORS;
}
}
else
{
if (StringCbCopy(CsvContents.KrnlPolicy.SecondLevelData, sizeof CsvContents.KrnlPolicy.SecondLevelData, _T("YES") )!= S_OK)
{
goto ERRORS;
}
}
}
if (IsDlgButtonChecked(hwnd, IDC_CHECK2))
{
if (bUserMode)
{
if (StringCbCopy(pUserData->Status.FileCollection, sizeof pUserData->Status.FileCollection, _T("NO") )!= S_OK)
{
goto ERRORS;
}
}
else
{
if (StringCbCopy(CsvContents.KrnlPolicy.FileCollection, sizeof CsvContents.KrnlPolicy.FileCollection, _T("NO") )!= S_OK)
{
goto ERRORS;
}
}
}
else
{
if (bUserMode)
{
if (StringCbCopy(pUserData->Status.FileCollection, sizeof pUserData->Status.FileCollection, _T("YES") )!= S_OK)
{
goto ERRORS;
}
}
else
{
if (StringCbCopy(CsvContents.KrnlPolicy.FileCollection, sizeof CsvContents.KrnlPolicy.FileCollection, _T("YES") )!= S_OK)
{
goto ERRORS;
}
}
}
if (IsDlgButtonChecked(hwnd, IDC_CHECK3))
{
if (bUserMode)
{
if (StringCbCopy(pUserData->Status.AllowResponse, sizeof pUserData->Status.AllowResponse, _T("NO") )!= S_OK)
{
goto ERRORS;
}
}
else
{
if (StringCbCopy(CsvContents.KrnlPolicy.AllowResponse, sizeof CsvContents.KrnlPolicy.AllowResponse, _T("NO") )!= S_OK)
{
goto ERRORS;
}
}
}
else
{
if (bUserMode)
{
if (StringCbCopy(pUserData->Status.AllowResponse, sizeof pUserData->Status.AllowResponse, _T("YES") )!= S_OK)
{
goto ERRORS;
}
}
else
{
if (StringCbCopy(CsvContents.KrnlPolicy.AllowResponse, sizeof CsvContents.KrnlPolicy.AllowResponse, _T("YES") )!= S_OK)
{
goto ERRORS;
}
}
}
if (IsDlgButtonChecked(hwnd, IDC_CHECK4))
{
if (bUserMode)
{
if (StringCbCopy(pUserData->Status.Tracking, sizeof pUserData->Status.Tracking, _T("YES") )!= S_OK)
{
goto ERRORS;
}
}
else
{
if (StringCbCopy(CsvContents.KrnlPolicy.Tracking, sizeof CsvContents.KrnlPolicy.Tracking, _T("YES") )!= S_OK)
{
goto ERRORS;
}
}
}
else
{
if (bUserMode)
{
if (StringCbCopy(pUserData->Status.Tracking, sizeof pUserData->Status.Tracking, _T("NO") )!= S_OK)
{
goto ERRORS;
}
}
else
{
if (StringCbCopy(CsvContents.KrnlPolicy.Tracking, sizeof CsvContents.KrnlPolicy.Tracking, _T("NO") )!= S_OK)
{
goto ERRORS;
}
}
}
if (bUserMode)
{
// fill temp buffer with string
GetDlgItemText(hwnd, IDC_EDIT1,TempBuffer, sizeof TempBuffer / sizeof TCHAR);
pTchar = TempBuffer;
while ( ( (*pTchar == _T(' ')) || (*pTchar == _T('\t')) ) && (*pTchar != _T('\0')))
{
++pTchar;
}
if (StringCbCopy(pUserData->Status.UrlToLaunch, sizeof pUserData->Status.UrlToLaunch, pTchar) != S_OK)
{
goto ERRORS;
}
GetDlgItemText(hwnd, IDC_EDIT3, pUserData->Status.CrashPerBucketCount, sizeof pUserData->Status.CrashPerBucketCount / sizeof TCHAR);
}
else
{
GetDlgItemText(hwnd, IDC_EDIT1,TempBuffer, sizeof TempBuffer / sizeof TCHAR);
pTchar = TempBuffer;
while ( ( (*pTchar == _T(' ')) || (*pTchar == _T('\t')) ) && (*pTchar != _T('\0')))
{
++pTchar;
}
if (StringCbCopy(CsvContents.KrnlPolicy.UrlToLaunch, sizeof CsvContents.KrnlPolicy.UrlToLaunch, pTchar) != S_OK)
{
goto ERRORS;
}
//GetDlgItemText(hwnd, IDC_EDIT1, CsvContents.KrnlPolicy.UrlToLaunch, sizeof CsvContents.KrnlPolicy.UrlToLaunch / sizeof TCHAR);
GetDlgItemText(hwnd, IDC_EDIT3, CsvContents.KrnlPolicy.CrashPerBucketCount, sizeof CsvContents.KrnlPolicy.CrashPerBucketCount / sizeof TCHAR);
}
return TRUE;
ERRORS:
return FALSE;
}
LRESULT CALLBACK EditSelectedDlgProc(HWND hwnd, UINT iMsg, WPARAM wParam, LPARAM lParam)
{
LVITEM lvi;
switch (iMsg)
{
case WM_INITDIALOG:
if (ReportMode==USER_MODE)
{
InitEditPolicyDlg (hwnd, hUserMode,TRUE,FALSE);
}
else
{
InitEditPolicyDlg (hwnd, hKrnlMode, FALSE,FALSE);
}
return TRUE;
case WM_DESTROY:
EndDialog(hwnd, TRUE);
return TRUE;
case WM_COMMAND:
{
switch (LOWORD(wParam))
{
case IDOK:
if (ReportMode == USER_MODE)
{
GetSelPolicyDlgData(TRUE, hwnd);
PUSER_DATA pUserData;
HWND hList = NULL;
int sel = 0;
hList = GetDlgItem(hUserMode, IDC_USER_LIST);
sel = ListView_GetNextItem(hList,-1, LVNI_SELECTED);
lvi.iItem = sel;
lvi.mask = LVIF_PARAM;
ListView_GetItem(hList, &lvi);
sel = lvi.lParam;
pUserData = cUserData.GetEntry(sel);
if (!pUserData)
{
return TRUE;
}
WriteStatusFile(pUserData);
}
else
{
GetSelPolicyDlgData(FALSE, hwnd);
WriteKernelStatusFile();
}
EndDialog(hwnd, TRUE);
return TRUE;
case IDCANCEL:
EndDialog(hwnd, TRUE);
}
}
break;
default:
break;
}
return FALSE;
}
BOOL GetPolicyDlgData(HWND hwnd)
{
TCHAR *pTchar = NULL;
TCHAR TempBuffer[MAX_PATH];
ZeroMemory(TempBuffer, sizeof TempBuffer);
// read the contents of the dialog box into the global policy structure.
if (IsDlgButtonChecked(hwnd, IDC_CHECK1))
{
if (StringCbCopy(GlobalPolicy.AllowBasic, sizeof GlobalPolicy.AllowBasic, _T("NO") )!= S_OK)
{
goto ERRORS;
}
}
else
{
if (StringCbCopy(GlobalPolicy.AllowBasic, sizeof GlobalPolicy.AllowBasic, _T("YES") )!= S_OK)
{
goto ERRORS;
}
}
if (IsDlgButtonChecked(hwnd, IDC_CHECK2))
{
if (StringCbCopy(GlobalPolicy.AllowAdvanced, sizeof GlobalPolicy.AllowAdvanced, _T("NO") )!= S_OK)
{
goto ERRORS;
}
}
else
{
if (StringCbCopy(GlobalPolicy.AllowAdvanced, sizeof GlobalPolicy.AllowAdvanced, _T("YES") )!= S_OK)
{
goto ERRORS;
}
}
if (IsDlgButtonChecked(hwnd, IDC_CHECK3))
{
if (StringCbCopy(GlobalPolicy.AllowMicrosoftResponse, sizeof GlobalPolicy.AllowMicrosoftResponse, _T("NO") )!= S_OK)
{
goto ERRORS;
}
}
else
{
if (StringCbCopy(GlobalPolicy.AllowMicrosoftResponse, sizeof GlobalPolicy.AllowMicrosoftResponse, _T("YES") )!= S_OK)
{
goto ERRORS;
}
}
if (IsDlgButtonChecked(hwnd, IDC_CHECK4))
{
if (StringCbCopy(GlobalPolicy.EnableCrashTracking, sizeof GlobalPolicy.EnableCrashTracking, _T("YES") )!= S_OK)
{
goto ERRORS;
}
}
else
{
if (StringCbCopy(GlobalPolicy.EnableCrashTracking, sizeof GlobalPolicy.EnableCrashTracking, _T("NO") )!= S_OK)
{
goto ERRORS;
}
}
GetDlgItemText(hwnd, IDC_EDIT1,TempBuffer, sizeof TempBuffer / sizeof TCHAR);
pTchar = TempBuffer;
while ( ( (*pTchar == _T(' ')) || (*pTchar == _T('\t')) ) && (*pTchar != _T('\0')))
{
++pTchar;
}
if (StringCbCopy(GlobalPolicy.CustomURL, sizeof GlobalPolicy.CustomURL, pTchar) != S_OK)
{
goto ERRORS;
}
//GetDlgItemText(hwnd, IDC_EDIT1, GlobalPolicy.CustomURL, sizeof GlobalPolicy.CustomURL / sizeof TCHAR);
GetDlgItemText(hwnd, IDC_EDIT2, GlobalPolicy.RedirectFileServer, sizeof GlobalPolicy.RedirectFileServer / sizeof TCHAR);
GetDlgItemText(hwnd, IDC_EDIT3, GlobalPolicy.CabsPerBucket, sizeof GlobalPolicy.CabsPerBucket / sizeof TCHAR);
return TRUE;
ERRORS:
return FALSE;
}
LRESULT CALLBACK EditDefaultDlgProc(HWND hwnd, UINT iMsg, WPARAM wParam, LPARAM lParam)
{
switch (iMsg)
{
case WM_INITDIALOG:
InitEditPolicyDlg (hwnd,NULL,FALSE, TRUE);
return TRUE;
case WM_DESTROY:
EndDialog(hwnd, TRUE);
return TRUE;
case WM_COMMAND:
{
switch (LOWORD(wParam))
{
case IDOK:
if (GetPolicyDlgData(hwnd))
{
WriteGlobalPolicyFile();
}
else
; // What message do we want to display here.
EndDialog(hwnd, TRUE);
break;
case IDCANCEL:
EndDialog(hwnd, TRUE);
break;
}
}
break;
default:
break;
}
return FALSE;
}
void DoEditSelectedBucketPolicy(HWND hwnd)
{
if (!DialogBox(g_hinst, MAKEINTRESOURCE(IDD_BUCKET_POLICY), hwnd, (DLGPROC) EditSelectedDlgProc))
{
; //MessageBox(NULL, "FAILED TO LOAD TREE", NULL, MB_OK);
}
}
void DoEditDefaultPolicy(HWND hwnd)
{
if (!DialogBox(g_hinst, MAKEINTRESOURCE(IDD_GLOBAL_POLICY), hwnd, (DLGPROC) EditDefaultDlgProc))
{
; //MessageBox(NULL, "FAILED TO LOAD TREE", NULL, MB_OK);
}
}
LRESULT CALLBACK MainDlgProc(HWND hwnd, UINT iMsg, WPARAM wParam, LPARAM lParam)
{
TCHAR szPath[MAX_PATH];
HMENU hMenu = NULL;
switch (iMsg)
{
case WM_INITDIALOG:
OnDlgInit(hwnd);
return TRUE;
case WM_DESTROY:
case WM_CLOSE:
PostQuitMessage(0);
return TRUE;
case WM_COMMAND:
{
switch (LOWORD(wParam))
{
case ID_FILE_MRU1:
case ID_FILE_MRU2:
case ID_FILE_MRU3:
case ID_FILE_MRU4:
g_bFirstBucket = TRUE;
g_bAdminAccess = TRUE;
ZeroMemory(szPath, sizeof szPath);
cUserData.CleanupList();
CsvContents.CleanupList();
hMenu = GetMenu(hwnd);
if (hMenu)
{
GetMenuStringA(hMenu, LOWORD(wParam), szPath, MAX_PATH, MF_BYCOMMAND);
if (_tcslen (szPath) > 4)
{
if (DoLoadFileTree(hwnd, szPath + 4, FALSE))
{
if (ReportMode == USER_MODE)
{
DoUserMode(hwnd);
}
else
{
DoKernelMode(hwnd);
}
EnableMenuItems(hwnd);
}
}
//RefreshUserMode(hUserMode);
//RefreshKrnlView(hKrnlMode);
}
break;
case IDC_KRNLMODE:
DoKernelMode(hwnd);
return TRUE;
case IDC_USERMODE:
DoUserMode(hwnd);
return TRUE;
case ID_REPORT_USERMODEFAULTS:
ReportUserModeFault(hUserMode, FALSE,0);
RefreshUserMode(hUserMode);
return TRUE;
case ID_REPORT_SELECTEDFAULTS:
if (ReportMode==USER_MODE)
{
ReportUserModeFault(hUserMode, TRUE, GetDlgItem(hUserMode,IDC_USER_LIST));
RefreshUserMode(hUserMode);
}
return TRUE;
case ID_REPORT_KERNELMODEFAULTS:
DoSubmitKernelFaults(hKrnlMode);
return TRUE;
case ID_FILE_LOADFILETREE:
g_bFirstBucket = TRUE;
g_bAdminAccess = TRUE;
if (DoLoadFileTree(hwnd, NULL, TRUE))
{
if (ReportMode == USER_MODE)
{
DoUserMode(hwnd);
}
else
{
DoKernelMode(hwnd);
}
EnableMenuItems(hwnd);
}
//RefreshUserMode(hUserMode);
//RefreshKrnlView(hKrnlMode);
return TRUE;
case ID_VIEW_REFRESH:
if (ReportMode == KERNEL_MODE)
RefreshKrnlView(hKrnlMode);
else
RefreshUserMode(hUserMode);
return TRUE;
case ID_EDIT_DEFAULTPOLICY:
DoEditDefaultPolicy(hwnd);
return TRUE;
case ID_EDIT_SELECTEDBUCKETSPOLICY:
if (ReportMode==USER_MODE)
{
if (ListView_GetSelectedCount(GetDlgItem(hUserMode, IDC_USER_LIST)) == 1)
{
DoEditSelectedBucketPolicy(hwnd);
RefreshUserMode(hUserMode);
}
else
{
// what error do we want to display here.
}
}
else
{
DoEditSelectedBucketPolicy(hwnd);
RefreshKrnlView(hKrnlMode);
}
return TRUE;
case ID_VIEW_CRASHLOG:
ViewCrashLog();
return TRUE;
case ID_REPORT_ALLCRASHES:
ReportUserModeFault(hUserMode, FALSE,0);
RefreshUserMode(hUserMode);
DoSubmitKernelFaults(hKrnlMode);
return FALSE;
case ID_FILE_CREATEFILETREE:
DoCreateFileTree(hwnd);
return TRUE;
case ID_HELP_ABOUT:
if (!DialogBox(g_hinst, MAKEINTRESOURCE(IDD_ABOUT), hwnd, (DLGPROC) AboutDlgProc))
{
; //MessageBox(NULL, "FAILED TO LOAD TREE", NULL, MB_OK);
}
break;
case ID_HELP_INDEX120:
OpenHelpIndex(_T(""));
break;
case ID_HELP_CONTENTS:
OpenHelpTopic(0);
break;
/*case ID_EDIT_COPY115:
if (ReportMode==KERNEL_MODE)
{
PostMessage(hKrnlMode, WM_COMMAND, MAKEWPARAM(ID_EDIT_COPY147,0), 0);
}
else
{
PostMessage(hUserMode, WM_COMMAND, MAKEWPARAM(ID_EDIT_COPY144,0),0);
}
break;
*/
case ID_VIEW_BUCKETCABFILEDIRECTORY:
if (ReportMode==KERNEL_MODE)
{
PostMessage(hKrnlMode, WM_COMMAND, MAKEWPARAM(ID_VIEW_BUCKETCABFILEDIRECTORY,0), 0);
}
else
{
PostMessage(hUserMode, WM_COMMAND, MAKEWPARAM(ID_VIEW_BUCKETCABFILEDIRECTORY157,0),0);
}
break;
case ID_VIEW_BUCKETOVERRIDERESPONSE:
if (ReportMode==USER_MODE )
ViewResponse(hUserMode, FALSE);
else
DoLaunchBrowser(hKrnlMode, TRUE);
break;
case ID_VIEW_RESPONSESELECTED:
if (ReportMode==USER_MODE)
ViewResponse(hUserMode, TRUE);
else
DoLaunchBrowser(hKrnlMode, FALSE);
break;
case ID_FILE_RELOADFILETREE:
g_bFirstBucket = TRUE;
g_bAdminAccess = TRUE;
cUserData.CleanupList();
CsvContents.CleanupList();
if (DoLoadFileTree(hwnd, CerRoot, FALSE))
{
EnableMenuItems(hwnd);
}
break;
case ID_FILE_EXIT:
PostQuitMessage(0);
break;
case ID_EXPORT_USERMODEBUCKETDATA:
ExportUserModeData(hwnd);
break;
case ID_EXPORT_KERNELMODEFAULTDATA:
ExportKernelModeData(hwnd);
break;
default:
break;
}
}
return FALSE;
case WM_SIZE:
ResizeKrlMode(hKrnlMode);
ResizeUserMode(hUserMode);
return TRUE;
case WM_ERASEBKGND:
// Don't know why this doesn't happen automatically...
{
HDC hdc = (HDC)wParam;
HPEN hpen = (HPEN)CreatePen(PS_SOLID, 1, GetSysColor(COLOR_BTNFACE));
HPEN hpenOld = (HPEN)SelectObject(hdc, hpen);
SelectObject(hdc, GetSysColorBrush(COLOR_BTNFACE));
RECT rc;
GetClientRect(hwnd, &rc);
Rectangle(hdc, rc.left, rc.top, rc.right, rc.bottom);
SelectObject(hdc, hpenOld);
DeleteObject(hpen);
return TRUE;
}
}
return FALSE;
}
/*----------------------------------------------------------------------------
DragWndProcV
Window Proc for Vertical Drag Control -- UNICODE WndProc
----------------------------------------------------------------- MichMarc --*/
LRESULT CALLBACK DragWndProcV(HWND hwnd, UINT iMsg, WPARAM wParam, LPARAM lParam)
{
switch(iMsg)
{
case WM_CREATE:
return 0;
case WM_PAINT:
{
PAINTSTRUCT ps;
HDC hdc = BeginPaint(hwnd, &ps);
HPEN hpenOld = (HPEN)SelectObject(hdc, CreatePen(PS_SOLID, 1, GetSysColor(COLOR_ACTIVEBORDER)));
MoveToEx(hdc, ps.rcPaint.left, 0, NULL);
LineTo(hdc, ps.rcPaint.right,0);
MoveToEx(hdc, ps.rcPaint.left, 2, NULL);
LineTo(hdc, ps.rcPaint.right,2);
MoveToEx(hdc, ps.rcPaint.left, 3, NULL);
LineTo(hdc, ps.rcPaint.right,3);
DeleteObject(SelectObject(hdc, CreatePen(PS_SOLID, 1, GetSysColor(COLOR_3DHIGHLIGHT))));
MoveToEx(hdc, ps.rcPaint.left, 1, NULL);
LineTo(hdc, ps.rcPaint.right,1);
DeleteObject(SelectObject(hdc, CreatePen(PS_SOLID, 1, GetSysColor(COLOR_3DSHADOW))));
SelectObject(hdc, GetSysColorBrush(COLOR_3DLIGHT));
MoveToEx(hdc, ps.rcPaint.left, 4, NULL);
LineTo(hdc, ps.rcPaint.right,4);
DeleteObject(SelectObject(hdc, CreatePen(PS_SOLID, 1, GetSysColor(COLOR_3DDKSHADOW))));
SelectObject(hdc, GetSysColorBrush(COLOR_3DLIGHT));
MoveToEx(hdc, ps.rcPaint.left, 5, NULL);
LineTo(hdc, ps.rcPaint.right,5);
DeleteObject(SelectObject(hdc, hpenOld));
EndPaint(hwnd, &ps);
return 0;
}
case WM_LBUTTONDOWN:
PostMessage(GetParent(hwnd), WM_COMMAND, GetWindowLong(hwnd, GWL_ID), (LPARAM)hwnd);
return 0;
}
return DefWindowProcW(hwnd, iMsg, wParam, lParam);
};
/*---------------------------------------------------------------------------
InitWindowClasses
Registers custom window classes
--------------------------------------------------------------- MichMarc --*/
/*void InitWindowClasses(void)
{
WNDCLASSEXW wc;
wc.cbSize = sizeof(WNDCLASSEXW);
wc.style = CS_HREDRAW | CS_VREDRAW;
wc.lpfnWndProc = DragWndProcV;
wc.cbClsExtra = 0;
wc.cbWndExtra = 0;
wc.hInstance = g_hinst;
wc.hIcon = NULL;
wc.hCursor = LoadCursor(NULL, IDC_SIZENS);
wc.hbrBackground = (HBRUSH)(COLOR_BTNFACE+1);
wc.lpszMenuName = NULL;
wc.lpszClassName = L"DragControlV";
wc.hIconSm = NULL;
RegisterClassExW(&wc);
wc.cbSize = sizeof(WNDCLASSEXW);
wc.style = CS_HREDRAW | CS_VREDRAW;
wc.lpfnWndProc = DragWndProcH;
wc.cbClsExtra = 0;
wc.cbWndExtra = 0;
wc.hInstance = vhintl;
wc.hIcon = NULL;
wc.hCursor = LoadCursor(NULL, IDC_SIZENS);
wc.hbrBackground = (HBRUSH)(COLOR_BTNFACE+1);
wc.lpszMenuName = NULL;
wc.lpszClassName = L"DragControlH";
wc.hIconSm = NULL;
RegisterClassExW(&wc);
}
*/
int WINAPI WinMain(HINSTANCE hinst, HINSTANCE hinstPrev, LPSTR szCmdLine,
int nShowCmd)
{
MSG msg;
HWND hwnd = NULL;
g_hinst = hinst;
INITCOMMONCONTROLSEX InitCtrls;
InitCommonControlsEx(&InitCtrls);
//InitWindowClasses();
LoadIcon(hinst, MAKEINTRESOURCE(IDI_MAIN));
hwnd = CreateDialog(g_hinst,
MAKEINTRESOURCE(IDD_MAIN),
NULL,
(DLGPROC)MainDlgProc);
g_hWnd = hwnd;
if (hwnd)
{
while(GetMessageW(&msg, NULL, 0, 0))
//if (!TranslateAcceleratorW(hwnd, hAccel, &msg))
if (!IsDialogMessageW(hwnd, &msg))
{
TranslateMessage(&msg);
DispatchMessageW(&msg);
}
}
return 0;
}