|
|
// tslsview.cpp : Defines the entry point for the application.
//
#include "stdafx.h"
//#include "resource.h"
extern "C" { BOOL OpenLog(VOID); VOID LogMsg(PWCHAR msgFormat, ...); BOOL LogDiagnosisFile( LPTSTR ); VOID CloseLog(VOID);}
//=---------globals------------
static HINSTANCE g_hinst;static BOOL g_fInitialized;ServerEnumData g_sed;PLIST g_pHead;static HANDLE g_hEvent;static BOOL g_bAutomaticLog;BOOL g_bDiagnosticLog = FALSE;static DWORD g_dwInterval = 1000 * 60 * 5;static BOOL g_bLog = 1;
TCHAR szLsViewKey[] = TEXT( "Software\\Microsoft\\Windows NT\\CurrentVersion\\LsView" );//-----------------------------
//-------------function prototypes ----------------
INT_PTR CALLBACK Dlg_Proc( HWND hwnd , UINT msg , WPARAM wp, LPARAM lp );BOOL OnInitApp( HWND );void OnTimedEvent( HWND hDlg );DWORD DiscoverServers( LPVOID ptr );BOOL ServerEnumCallBack( TLS_HANDLE hHandle, LPCTSTR pszServerName, HANDLE dwUserData );
void OnReSize( HWND hwnd , WPARAM wp , LPARAM lp );BOOL DeleteList( PLIST );BOOL AddItem( LPTSTR , LPTSTR , LPTSTR );void CreateLogFile( HWND );BOOL Tray_Init( HWND hwnd , BOOL );BOOL Tray_ToGreen( HWND hwnd );BOOL Tray_ToYellow( HWND hwnd , LPTSTR szMsg );BOOL Tray_ToRed( HWND hwnd );BOOL Tray_Remove( HWND hwnd );BOOL Tray_ToXXX( HWND hwnd , LPTSTR szTip , UINT resid );BOOL Tray_Notify( HWND hwnd , WPARAM wp , LPARAM lp );UINT_PTR CALLBACK OFNHookProc( HWND hdlg , UINT uiMsg, WPARAM wParam, LPARAM lParam );BOOL RetrieveDataObject( PDATAOBJECT pObj );BOOL StoreDataObject( PDATAOBJECT pObj );BOOL LogFile( LPTSTR szFileName );//-------------------------------------------------
//=---------constants------------
const UINT kTimerId = 23456;const UINT kDefaultElapseTime = 1000 * 60 * 5;const UINT kMaxMinutes = 71582;const UINT kBubbleTimeout = 10 * 1000;#define TN_MESSAGE ( WM_USER + 60 )
//-----------------------------
DWORDGetPageSize( VOID ) {
static DWORD dwPageSize = 0;
if ( !dwPageSize ) {
SYSTEM_INFO sysInfo = { 0 }; GetSystemInfo( &sysInfo ); // cannot fail.
dwPageSize = sysInfo.dwPageSize;
}
return dwPageSize;
}
/*++**************************************************************
NAME: MyVirtualAlloc
as Malloc, but automatically protects the last page of the allocation. This simulates pageheap behavior without requiring it.
MODIFIES: ppvData -- receives memory
TAKES: dwSize -- minimum amount of data to get
RETURNS: TRUE when the function succeeds. FALSE otherwise. LASTERROR: not set Free with MyVirtualFree
**************************************************************--*/
BOOLMyVirtualAlloc( IN DWORD dwSize, OUT PVOID *ppvData ) {
PBYTE pbData; DWORD dwTotalSize; PVOID pvLastPage;
// ensure that we allocate one extra page
dwTotalSize = dwSize / GetPageSize(); if( dwSize % GetPageSize() ) { dwTotalSize ++; }
// this is the guard page
dwTotalSize++; dwTotalSize *= GetPageSize();
// do the alloc
pbData = (PBYTE) VirtualAlloc( NULL, // don't care where
dwTotalSize, MEM_COMMIT | MEM_TOP_DOWN, PAGE_READWRITE ); if ( pbData ) {
pbData += dwTotalSize;
// find the LAST page.
pbData -= GetPageSize();
pvLastPage = pbData;
// now, carve out a chunk for the caller:
pbData -= dwSize;
// last, protect the last page:
if ( VirtualProtect( pvLastPage, 1, // protect the page containing the last byte
PAGE_NOACCESS, &dwSize ) ) {
*ppvData = pbData; return TRUE;
}
VirtualFree( pbData, 0, MEM_RELEASE );
}
return FALSE;
}
VOIDMyVirtualFree( IN PVOID pvData ) {
VirtualFree( pvData, 0, MEM_RELEASE );
}
//-------------------------------------------------------------------------
int APIENTRY WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow){ INITCOMMONCONTROLSEX icc = { sizeof( INITCOMMONCONTROLSEX ) , ICC_LISTVIEW_CLASSES };
HANDLE hMutex = CreateMutex( NULL , FALSE , TEXT("TSLSVIEW2" ) );
if( GetLastError() == ERROR_ALREADY_EXISTS ) { TCHAR szTitle[ 60 ];
DBGMSG( TEXT( "TSLSVIEW: App instance already running\n" ) , 0 );
LoadString( hInstance , IDS_TITLE , szTitle , SIZEOF( szTitle ) );
HWND hWnd = FindWindow( NULL , szTitle );
if( hWnd != NULL ) { SetForegroundWindow( hWnd ); }
return 0; }
if( !InitCommonControlsEx( &icc ) ) { DBGMSG( TEXT("InitCommonControlsEx failed with 0x%x\n") , GetLastError( ) );
return 0; }
g_hinst = hInstance;
g_fInitialized = FALSE;
g_hEvent = CreateEvent( NULL , FALSE , TRUE , NULL );
if(OpenLog() == FALSE) { ODS(TEXT("Failed to open temporary file for discovery diagnostics"));
return 0; }
g_pHead = ( PLIST )new LIST[1];
if( g_pHead == NULL ) { ODS( TEXT( "LSVIEW out of memory\n" ) );
return 0; }
g_pHead->pszMachineName = NULL; g_pHead->pszTimeFormat = NULL; g_pHead->pszType = NULL; g_pHead->pNext = NULL;
DialogBox( hInstance , MAKEINTRESOURCE( IDD_LSVIEW ), NULL, Dlg_Proc);
CloseHandle( hMutex );
CloseHandle( g_hEvent );
DeleteList( g_pHead );
CloseLog();
return 0;}
//-------------------------------------------------------------------------
INT_PTR CALLBACK Dlg_Proc( HWND hwnd , UINT msg , WPARAM wp, LPARAM lp ){ TCHAR szTitle[ 60 ];
switch( msg ) { case WM_COMMAND: switch( LOWORD( wp ) ) { case ID_FILE_EXIT: EndDialog( hwnd , 0 ); break;
case ID_FILE_CREATELOGFILE:
if( WaitForSingleObject( g_hEvent , 0 ) == WAIT_TIMEOUT ) { TCHAR szBuffer[ 255 ];
LoadString( g_hinst , IDS_ERROR_QS , szBuffer , SIZEOF( szBuffer ) );
LoadString( g_hinst , IDS_TITLE , szTitle , SIZEOF( szTitle ) );
MessageBox( hwnd , szBuffer, szTitle , MB_OK | MB_ICONINFORMATION ); } else { SetEvent( g_hEvent );
CreateLogFile( hwnd ); } break;
case ID_HELP_ABOUT:
LoadString( g_hinst , IDS_TITLE , szTitle , SIZEOF( szTitle ) );
ShellAbout( hwnd , szTitle , NULL , LoadIcon( g_hinst , MAKEINTRESOURCE( IDI_TSLSVIEW ) ) ); break;
case IDM_MINIMIZE: ShowWindow( hwnd , SW_MINIMIZE ); break;
case IDM_RESTORE: ShowWindow( hwnd , SW_RESTORE ); break;
case IDM_EXIT: DestroyWindow( hwnd ); break; }
break;
case WM_CLOSE: case WM_DESTROY:
Tray_ToRed( hwnd );
KillTimer( hwnd , kTimerId );
Tray_Remove( hwnd );
EndDialog( hwnd , 0 );
break;
case WM_INITDIALOG: OnInitApp( hwnd ); break;
case WM_TIMER: if( wp == ( WPARAM )kTimerId ) { OnTimedEvent( hwnd ); }
break;
case WM_SIZE: OnReSize( hwnd , wp , lp );
break;
case TN_MESSAGE:
Tray_Notify( hwnd , wp , lp );
break;
} return FALSE;}//-------------------------------------------------------------------------
BOOL InitListView( HWND hwnd ){ int rgIds[] = { IDS_STR_COL1 , IDS_STR_COL2 , IDS_STR_COL3 , -1 };
LV_COLUMN lvc; TCHAR tchBuffer[ 60 ];
HWND hListView = GetDlgItem( hwnd , IDC_LSVIEW_LIST ) ;
lvc.mask = LVCF_FMT | LVCF_WIDTH | LVCF_TEXT | LVCF_SUBITEM; lvc.fmt = LVCFMT_LEFT; for( int idx=0; rgIds[idx] != -1; ++idx ) { LoadString( g_hinst , ( UINT )rgIds[idx], tchBuffer, SIZEOF( tchBuffer ) );
if( idx == 1 ) { lvc.cx = 225; } else { lvc.cx = 75; }
lvc.pszText = tchBuffer; lvc.iSubItem = idx;
ListView_InsertColumn( hListView , idx , &lvc );
}
DWORD dwStyle = ( DWORD )SendMessage( hListView , LVM_GETEXTENDEDLISTVIEWSTYLE , 0 , 0 );
dwStyle |= ( LVS_EX_FULLROWSELECT | LVS_EX_GRIDLINES ) ;
SendMessage( hListView , LVM_SETEXTENDEDLISTVIEWSTYLE , 0 , dwStyle );
return TRUE;}
//-------------------------------------------------------------------------
BOOL OnInitApp( HWND hwnd ){ DATAOBJECT dobj;
// set up listview extended mode
InitListView( hwnd );
LONG_PTR lptrIcon;
lptrIcon = ( LONG_PTR )LoadImage( g_hinst , MAKEINTRESOURCE( IDI_TSLSVIEW ), IMAGE_ICON, 16, 16, 0 );
SetClassLongPtr( hwnd , GCLP_HICONSM , lptrIcon );
lptrIcon = ( LONG_PTR )LoadImage( g_hinst , MAKEINTRESOURCE( IDI_TSLSVIEW ), IMAGE_ICON, 0, 0, 0 );
SetClassLongPtr( hwnd , GCLP_HICON , lptrIcon );
ZeroMemory( &dobj , sizeof( dobj ) );
if( RetrieveDataObject( &dobj ) ) { g_bAutomaticLog = dobj.bIsChecked;
g_bDiagnosticLog = dobj.bIsDiagnosisChecked;
g_dwInterval = dobj.dwTimeInterval * 1000 * 60; }
if( dobj.dwTimeInterval == 0 ) { g_dwInterval = ( DWORD )kDefaultElapseTime; } // setup initial trayicon
Tray_Init( hwnd , dobj.bNotifyOnce );
if( !dobj.bNotifyOnce ) { dobj.bNotifyOnce = TRUE;
StoreDataObject( &dobj ); }
// set cursor to hourglass
OnTimedEvent( hwnd );
SetTimer( hwnd , kTimerId, g_dwInterval, NULL );
return TRUE;}
//-------------------------------------------------------------------------
void OnTimedEvent( HWND hDlg ){ ODS( L"LSVIEW: OnTimedEvent fired " );
g_sed.dwNumServer = 0; g_sed.dwDone = 0; g_sed.hList = GetDlgItem( hDlg , IDC_LSVIEW_LIST ); // remove all listview items
DWORD dwValue;
dwValue = WaitForSingleObject( g_hEvent , 0 );
if( dwValue == WAIT_TIMEOUT ) { ODS( TEXT("still looking for servers\n" ) ); return; }
SetEvent( g_hEvent );
ODS( TEXT("launching thread\n" ) );
HANDLE hThread = CreateThread( NULL , 0 , ( LPTHREAD_START_ROUTINE )DiscoverServers, ( LPVOID )&g_sed , 0, &dwValue );
if( hThread == NULL ) { DBGMSG( TEXT( "Failed to create DiscoverServer thread last error 0x%x\n" ) , GetLastError( ) );
Tray_ToRed( hDlg ); }
CloseHandle( hThread );}
//-------------------------------------------------------------------------
DWORD DiscoverServers( LPVOID ptr ){ WaitForSingleObject( g_hEvent , INFINITE );
ODS( L"LSVIEW -- entering DiscoverServers\n" );
if (!g_fInitialized) { TLSInit(); g_fInitialized = TRUE; }
LPWSTR *ppszEnterpriseServer = NULL; DWORD dwCount; DWORD index; // we could be writing out to a file we should wait
if( g_pHead != NULL ) { DeleteList( g_pHead->pNext );
g_pHead->pNext = NULL; } //
// Look for all license servers in domain
//
ServerEnumData *pEnumData = ( ServerEnumData * )ptr;
TCHAR szBuffer[ 60 ];
LoadString( g_hinst , IDS_YELLOW , szBuffer , SIZEOF( szBuffer ) );
Tray_ToYellow( GetParent( pEnumData->hList ) , szBuffer );
HRESULT hResult = EnumerateTlsServer( ServerEnumCallBack, ptr, 3000, FALSE );
hResult = GetAllEnterpriseServers( &ppszEnterpriseServer, &dwCount );
if( hResult == ERROR_SUCCESS && dwCount != 0 && ppszEnterpriseServer != NULL ) {
TLS_HANDLE TlsHandle = NULL;
//
// Inform dialog
//
for(index = 0; index < dwCount && pEnumData->dwDone == 0; index++) {
if( ppszEnterpriseServer[index] == NULL ) { continue; }
if(ServerEnumCallBack( NULL, (LPTSTR)ppszEnterpriseServer[index], pEnumData ) == TRUE ) { continue; }
TlsHandle = TLSConnectToLsServer( (LPTSTR)ppszEnterpriseServer[index] );
if(TlsHandle == NULL ) { continue; if(g_bLog) LogMsg(L"Can't connect to %s. Maybe it has no License Service running on it.\n", (LPTSTR)ppszEnterpriseServer[index]); }
if(g_bLog) LogMsg(L"!!!Connected to License Service on %s\n",(LPTSTR)ppszEnterpriseServer[index]);
ServerEnumCallBack( TlsHandle, (LPTSTR)ppszEnterpriseServer[index], pEnumData );
TLSDisconnectFromServer(TlsHandle); }
if( ppszEnterpriseServer != NULL ) { for( index = 0; index < dwCount; index ++) { if( ppszEnterpriseServer[ index ] != NULL ) { LocalFree( ppszEnterpriseServer[ index ] ); } }
LocalFree( ppszEnterpriseServer ); }
}
ListView_DeleteAllItems( pEnumData->hList ); PLIST pTemp;
if( g_pHead != NULL ) { pTemp = g_pHead->pNext;
while( pTemp != NULL ) { int nItem = ListView_GetItemCount( pEnumData->hList );
LVITEM lvi;
ZeroMemory( &lvi , sizeof( LVITEM ) );
lvi.mask = LVIF_TEXT; lvi.pszText = pTemp->pszMachineName; lvi.cchTextMax = lstrlen( pTemp->pszMachineName ); lvi.iItem = nItem; lvi.iSubItem = 0;
ListView_InsertItem( pEnumData->hList , &lvi );
// Set item for second column
lvi.pszText = pTemp->pszTimeFormat; lvi.iSubItem = 1; lvi.cchTextMax = sizeof( pTemp->pszTimeFormat );
ListView_SetItem( pEnumData->hList , &lvi );
// Set item for third column
lvi.pszText = pTemp->pszType; lvi.iSubItem = 2; lvi.cchTextMax = sizeof( pTemp->pszType );
ListView_SetItem( pEnumData->hList , &lvi );
pTemp = pTemp->pNext; } }
if( g_bAutomaticLog ) { DATAOBJECT db;
ZeroMemory( &db , sizeof( DATAOBJECT ) );
if( RetrieveDataObject( &db ) ) { if( g_bDiagnosticLog ) LogDiagnosisFile( db.wchFileName ); else LogFile( db.wchFileName ); } }
ODS( L"LSVIEW : DiscoverServers completing\n" );
// motion for green
Tray_ToGreen( GetParent( pEnumData->hList ) );
SetEvent( g_hEvent );
ExitThread( hResult );
return ( DWORD )hResult;}
typedef DWORD (WINAPI* PTLSGETSERVERNAMEFIXED) ( TLS_HANDLE hHandle, LPTSTR *pszMachineName, PDWORD pdwErrCode );
typedef DWORD (WINAPI* PTLSGETSERVERNAMEEX) ( TLS_HANDLE hHandle, LPTSTR pszMachineName, PDWORD pdwSize, PDWORD pdwErrCode );RPC_STATUSTryGetServerName(PCONTEXT_HANDLE hBinding, LPTSTR *pszServer){ RPC_STATUS status; DWORD dwErrCode; HINSTANCE hModule = LoadLibrary(L"mstlsapi.dll");
if (hModule) { PTLSGETSERVERNAMEFIXED pfnGetServerNameFixed = (PTLSGETSERVERNAMEFIXED) GetProcAddress(hModule,"TLSGetServerNameFixed");
if (pfnGetServerNameFixed) { status = pfnGetServerNameFixed(hBinding,pszServer,&dwErrCode); if(status == RPC_S_OK && dwErrCode == ERROR_SUCCESS && pszServer != NULL) { FreeLibrary(hModule); return status; } } LPTSTR lpszMachineName = NULL; try { if ( !MyVirtualAlloc( ( MAX_COMPUTERNAME_LENGTH+1 ) * sizeof( TCHAR ), (PVOID*) &lpszMachineName ) ) { return RPC_S_OUT_OF_MEMORY; }
DWORD uSize = MAX_COMPUTERNAME_LENGTH+1 ;
memset(lpszMachineName, 0, ( MAX_COMPUTERNAME_LENGTH+1 ) * sizeof( TCHAR ));
PTLSGETSERVERNAMEEX pfnGetServerNameEx = (PTLSGETSERVERNAMEEX) GetProcAddress(hModule,"TLSGetServerNameEx"); if (pfnGetServerNameEx != NULL) { status = pfnGetServerNameEx(hBinding,lpszMachineName,&uSize, &dwErrCode); if(status == RPC_S_OK && dwErrCode == ERROR_SUCCESS) { *pszServer = (LPTSTR) MIDL_user_allocate((lstrlen(lpszMachineName)+1)*sizeof(TCHAR));
if (NULL != *pszServer) { lstrcpy(*pszServer,lpszMachineName); } } } } catch (...) { status = ERROR_NOACCESS; } if(lpszMachineName) MyVirtualFree(lpszMachineName); if(hModule) FreeLibrary(hModule); }
return status;}
//-------------------------------------------------------------------------
BOOL ServerEnumCallBack( TLS_HANDLE hHandle, LPCTSTR pszServerName, HANDLE dwUserData ) { int i; ServerEnumData* pEnumData = (ServerEnumData *)dwUserData;
BOOL bCancel;
if( pEnumData == NULL ) { return FALSE; }
bCancel = ( InterlockedExchange( &(pEnumData->dwDone) , pEnumData->dwDone) == 1);
if( bCancel == TRUE ) { return TRUE; }
if( hHandle != NULL ) { DWORD dwStatus; DWORD dwErrCode; DWORD dwVersion;
LPTSTR szServer = NULL; TCHAR szMcType[ MAX_COMPUTERNAME_LENGTH + 1 ]; TCHAR szTimeFormat[ 256 ]; DWORD dwBufSize = SIZEOF( szServer); dwStatus = TryGetServerName( hHandle, &szServer );
if( dwStatus != ERROR_SUCCESS || szServer == NULL ) { return FALSE; }
TLSGetVersion(hHandle, &dwVersion);
if( dwVersion & TLS_VERSION_ENTERPRISE_BIT ) { LoadString( g_hinst , IDS_TYPE_ENT, szMcType, SIZEOF( szMcType ) ); } else { LoadString( g_hinst , IDS_TYPE_DOMAIN, szMcType, SIZEOF( szMcType ) );
}
SYSTEMTIME st;
TCHAR szDate[ 80 ]; TCHAR szTime[ 80 ];
GetLocalTime( &st );
GetDateFormat( LOCALE_USER_DEFAULT , DATE_LONGDATE , &st, NULL, szDate, SIZEOF( szDate ) );
GetTimeFormat( LOCALE_USER_DEFAULT, TIME_NOSECONDS, &st, NULL, szTime, SIZEOF( szTime ) );
wsprintf( szTimeFormat , TEXT( "%s %s") , szDate , szTime );
AddItem( szServer , szTimeFormat , szMcType );
pEnumData->dwNumServer++;
MIDL_user_free(szServer); }
//
// Continue enumeration
//
return InterlockedExchange(&(pEnumData->dwDone), pEnumData->dwDone) == 1;}
//-------------------------------------------------------------------------
void OnReSize( HWND hwnd , WPARAM wp , LPARAM lp ){ HWND hList = GetDlgItem( hwnd , IDC_LSVIEW_LIST ); if( hList != NULL ) { MoveWindow( hList , 0 , 0 , LOWORD( lp ), HIWORD( lp ) , TRUE );
if( wp == SIZE_RESTORED || wp == SIZE_MAXIMIZED ) { ListView_RedrawItems( hList , 0 , ListView_GetItemCount( hList ) ); } }
}
//------------------------------------------------------------------------
// link list methods
//------------------------------------------------------------------------
BOOL AddItem( LPTSTR szMachineName , LPTSTR szTimeFormat , LPTSTR szType ){ ODS( TEXT("LSVIEW : Adding an item\n" ) );
if( g_pHead == NULL ) { return FALSE; }
PLIST pNewItem = ( PLIST )new LIST[1];
if( pNewItem == NULL ) { ODS( TEXT( "LSVIEW AddItem out of memory\n" ) );
return FALSE; }
pNewItem->pNext = NULL;
if( szMachineName != NULL ) { pNewItem->pszMachineName = ( LPTSTR )new TCHAR[ lstrlen( szMachineName ) + 1 ];
if( pNewItem->pszMachineName != NULL ) { lstrcpy( pNewItem->pszMachineName , szMachineName ); } } else { pNewItem->pszMachineName = NULL; }
if( szTimeFormat != NULL ) { pNewItem->pszTimeFormat = ( LPTSTR )new TCHAR[ lstrlen( szTimeFormat ) + 1 ];
if( pNewItem->pszTimeFormat != NULL ) { lstrcpy( pNewItem->pszTimeFormat , szTimeFormat ); } } else { pNewItem->pszTimeFormat = NULL; }
if( szType != NULL ) { pNewItem->pszType = ( LPTSTR )new TCHAR[ lstrlen( szType ) + 1 ];
if( pNewItem->pszType != NULL ) { lstrcpy( pNewItem->pszType , szType ); } } else { pNewItem->pszType = NULL; }
//=--- find the next available entry
PLIST pTemp = g_pHead;
while( pTemp->pNext != NULL ) { pTemp = pTemp->pNext; }
pTemp->pNext = pNewItem;
return TRUE;}
//------------------------------------------------------------------------
BOOL DeleteList( PLIST pStart ){ PLIST pPleaseKillMe;
if( pStart == NULL ) { return TRUE; }
while( pStart != NULL ) { pPleaseKillMe = pStart->pNext;
if( pStart->pszMachineName != NULL ) { delete[] pStart->pszMachineName;
} if( pStart->pszTimeFormat != NULL ) { delete[] pStart->pszTimeFormat; } if( pStart->pszType != NULL ) { delete[] pStart->pszType; }
delete pStart;
pStart = pPleaseKillMe; }
return TRUE;
}
//------------------------------------------------------------------------
void CreateLogFile( HWND hwnd ){ // start the save as dialog
OPENFILENAME ofn; TCHAR szBuffer[ 60 ]; TCHAR szFilter[ 60 ] = { 0 }; TCHAR szFileName[ MAX_PATH ]; TCHAR szExt[ 10 ]; DATAOBJECT dobj;
ZeroMemory( &ofn , sizeof( OPENFILENAME ) );
ZeroMemory( &dobj , sizeof( DATAOBJECT ) );
RetrieveDataObject( &dobj );
if( dobj.bIsChecked ) { lstrcpy( szFileName , dobj.wchFileName ); } else { szFileName[0] = 0; }
LoadString( g_hinst , IDS_FILTER , szFilter , SIZEOF( szFilter ) );
LoadString( g_hinst , IDS_EXTENSION , szExt , SIZEOF( szExt ) );
ofn.lStructSize = sizeof( OPENFILENAME ); ofn.hwndOwner = hwnd; ofn.lpstrFilter = szFilter; ofn.lpstrFile = szFileName; ofn.nMaxFile = MAX_PATH; ofn.Flags = OFN_HIDEREADONLY | OFN_EXPLORER | OFN_ENABLETEMPLATE | OFN_ENABLEHOOK; ofn.lpTemplateName = MAKEINTRESOURCE( IDD_CDEXT ); ofn.lpstrDefExt = szExt; ofn.hInstance = g_hinst; ofn.FlagsEx = OFN_EX_NOPLACESBAR; ofn.lpfnHook = OFNHookProc;
// ok let's make them wait
WaitForSingleObject( g_hEvent , INFINITE );
// motion for yellow
LoadString( g_hinst , IDS_TRAYFILE , szBuffer , SIZEOF( szBuffer ) );
Tray_ToYellow( hwnd , szBuffer );
if( GetSaveFileName( &ofn ) ) { if( g_bDiagnosticLog ) LogDiagnosisFile( szFileName ); else LogFile( szFileName ); } else { DBGMSG( TEXT( "Last error was 0x%x\n" ) , CommDlgExtendedError( ) ); }
// motion for green
Tray_ToGreen( hwnd );
SetEvent( g_hEvent );
}
//-------------------------------------------------------------------------
UINT_PTR CALLBACK OFNHookProc( HWND hdlg, // handle to child dialog box window
UINT uiMsg, // message identifier
WPARAM wParam, // message parameter
LPARAM lParam // message parameter
){ DATAOBJECT dobj; TCHAR szDigits[ 16 ];
switch( uiMsg ) { case WM_INITDIALOG: { ODS( TEXT("OFNHookProc WM_INITDIALOG\n" ) );
ZeroMemory( &dobj , sizeof( DATAOBJECT ) );
SendMessage( GetDlgItem( hdlg , IDC_SPIN1 ) , UDM_SETRANGE32 , ( WPARAM ) 1 , ( LPARAM )71582 ); SendMessage( GetDlgItem( hdlg , IDC_SPIN1 ) , UDM_SETPOS32 , 1 , 0 );
RetrieveDataObject( &dobj ); CheckDlgButton( hdlg , IDC_CHECK1 , dobj.bIsChecked ? BST_CHECKED : BST_UNCHECKED );
CheckDlgButton( hdlg , IDC_CHECK2 , dobj.bIsDiagnosisChecked ? BST_CHECKED : BST_UNCHECKED );
dobj.dwTimeInterval = ( g_dwInterval / 60 ) / 1000 ;
StoreDataObject( &dobj );
if( dobj.dwTimeInterval > 0 ) { wsprintf( szDigits , TEXT("%d" ), dobj.dwTimeInterval ); SetWindowText( GetDlgItem( hdlg , ID_EDT_NUM ) , szDigits ); } }
break;
case WM_NOTIFY: { ODS( TEXT("OFNHookProc WM_NOTIFY\n" ) );
LPOFNOTIFY pOnotify = ( LPOFNOTIFY )lParam;
if( pOnotify != NULL && pOnotify->hdr.code == CDN_FILEOK ) { DBGMSG( TEXT("File name to store in registry %s.\n") , pOnotify->lpOFN->lpstrFile ); ZeroMemory( &dobj , sizeof( DATAOBJECT ) );
lstrcpy( dobj.wchFileName , pOnotify->lpOFN->lpstrFile );
GetWindowText( GetDlgItem( hdlg , ID_EDT_NUM ) , szDigits , SIZEOF( szDigits ) );
if( szDigits[0] == 0 ) { // reset to default elaspe time
dobj.dwTimeInterval = 5; } else { dobj.dwTimeInterval = _wtoi( szDigits ); }
dobj.bIsChecked = IsDlgButtonChecked( hdlg , IDC_CHECK1 ) == BST_CHECKED;
g_bAutomaticLog = dobj.bIsChecked;
dobj.bIsDiagnosisChecked = IsDlgButtonChecked( hdlg , IDC_CHECK2 ) == BST_CHECKED;
g_bDiagnosticLog = dobj.bIsDiagnosisChecked;
if( dobj.dwTimeInterval < 1 ) { dobj.dwTimeInterval = 1; }
if( dobj.dwTimeInterval > kMaxMinutes ) { dobj.dwTimeInterval = kMaxMinutes; }
dobj.bNotifyOnce = TRUE;
g_dwInterval = dobj.dwTimeInterval * 60 * 1000;
KillTimer( GetParent( GetParent( hdlg ) ) , kTimerId );
SetTimer( GetParent( GetParent( hdlg ) ) , kTimerId , g_dwInterval , NULL );
StoreDataObject( &dobj ); } } break; }
return 0;}
//-------------------------------------------------------------------------
BOOL Tray_Init( HWND hwnd , BOOL bNotify ){ NOTIFYICONDATA nid;
TCHAR szTip[ 60 ]; TCHAR szBubble[ 255 ]; TCHAR szTitle[ 60 ];
ZeroMemory( &nid , sizeof( NOTIFYICONDATA ) );
LoadString( g_hinst , IDS_TIP , szTip , SIZEOF( szTip ) );
if( !bNotify ) { LoadString( g_hinst , IDS_BUBBLE , szBubble , SIZEOF( szBubble ) ); LoadString( g_hinst , IDS_TITLE , szTitle , SIZEOF( szTitle ) ); nid.uFlags = NIF_TIP | NIF_INFO; } nid.cbSize = sizeof( NOTIFYICONDATA ); nid.hWnd = hwnd; nid.uID = TN_MESSAGE; nid.uFlags |= NIF_ICON | NIF_MESSAGE; nid.uCallbackMessage = TN_MESSAGE; nid.hIcon = LoadIcon( g_hinst , MAKEINTRESOURCE( IDC_ICON_NONE ) );
lstrcpy( nid.szTip , szTip ); lstrcpy( nid.szInfo , szBubble ); lstrcpy( nid.szInfoTitle , szTitle );
nid.dwInfoFlags = NIIF_INFO; nid.uTimeout = kBubbleTimeout;
return Shell_NotifyIcon( NIM_ADD , &nid );}
//-------------------------------------------------------------------------
BOOL Tray_ToGreen( HWND hwnd ){ TCHAR szBuffer[ 260 ];
LoadString( g_hinst , IDS_TRAYGREEN , szBuffer , SIZEOF( szBuffer) );
return Tray_ToXXX( hwnd , szBuffer , IDC_ICON_GO ); }
//-------------------------------------------------------------------------
BOOL Tray_ToYellow( HWND hwnd , LPTSTR szMsg ){ return Tray_ToXXX( hwnd , szMsg , IDC_ICON_CAUTION ); }
//-------------------------------------------------------------------------
BOOL Tray_ToRed( HWND hwnd ){ TCHAR szBuffer[ 260 ];
LoadString( g_hinst , IDS_TRAYSTOP , szBuffer , SIZEOF( szBuffer) );
return Tray_ToXXX( hwnd , szBuffer , IDC_ICON_STOP );}
//-------------------------------------------------------------------------
BOOL Tray_Remove( HWND hwnd ){ NOTIFYICONDATA nid;
ZeroMemory( &nid , sizeof( NOTIFYICONDATA ) ); nid.cbSize = sizeof( NOTIFYICONDATA ); nid.hWnd = hwnd; nid.uID = TN_MESSAGE; return Shell_NotifyIcon( NIM_DELETE , &nid );}
//-------------------------------------------------------------------------
BOOL Tray_ToXXX( HWND hwnd , LPTSTR szTip , UINT resid ){ NOTIFYICONDATA nid;
ZeroMemory( &nid , sizeof( NOTIFYICONDATA ) ); nid.cbSize = sizeof( NOTIFYICONDATA ); nid.hWnd = hwnd; nid.uID = TN_MESSAGE; nid.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP; nid.uCallbackMessage = TN_MESSAGE; nid.hIcon = LoadIcon( g_hinst , MAKEINTRESOURCE( resid ) ); lstrcpy( nid.szTip , szTip );
return Shell_NotifyIcon( NIM_MODIFY , &nid );}
//-------------------------------------------------------------------------
BOOL Tray_Notify( HWND hwnd , WPARAM wp , LPARAM lp ){ switch( lp ) { case WM_LBUTTONDBLCLK: OpenIcon( hwnd ); SetForegroundWindow( hwnd ); break;
case WM_RBUTTONDOWN: { HMENU hmenuParent = LoadMenu( g_hinst, MAKEINTRESOURCE( IDR_TRAYMENU ) ); if( hmenuParent != NULL ) { HMENU hpopup = GetSubMenu( hmenuParent , 0 );
RemoveMenu( hmenuParent , 0 , MF_BYPOSITION );
DestroyMenu( hmenuParent );
// Display the tray icons context menu at
// the current cursor location
if( hpopup != NULL ) { POINT pt;
GetCursorPos( &pt );
SetForegroundWindow( hwnd );
TrackPopupMenuEx( hpopup, 0, pt.x, pt.y, hwnd, NULL);
DestroyMenu(hpopup); } } } break;
} return FALSE;}
//-------------------------------------------------------------------------
// pObj is a pointer a DATAOBJECT buffer
//-------------------------------------------------------------------------
BOOL StoreDataObject( PDATAOBJECT pObj ){ DWORD dwStatus;
HKEY hKey;
dwStatus = RegCreateKeyEx( HKEY_CURRENT_USER , szLsViewKey, 0, NULL, 0, KEY_WRITE, NULL, &hKey, NULL );
if( dwStatus != ERROR_SUCCESS ) { // format a message and display an error
return FALSE; }
dwStatus = RegSetValueEx( hKey, TEXT( "DataObject" ), 0, REG_BINARY, ( CONST BYTE * )pObj, sizeof( DATAOBJECT ) );
if( dwStatus != ERROR_SUCCESS ) { RegCloseKey( hKey ); return FALSE; }
RegCloseKey( hKey ); return TRUE;
}
//-------------------------------------------------------------------------
// pObj is a pointer to a DATAOBJECT buffer
//-------------------------------------------------------------------------
BOOL RetrieveDataObject( PDATAOBJECT pObj ){ DWORD dwStatus; DWORD dwSizeOfDO = sizeof( DATAOBJECT );
HKEY hKey;
dwStatus = RegOpenKeyEx( HKEY_CURRENT_USER , szLsViewKey, 0, KEY_READ, &hKey ); if( dwStatus != ERROR_SUCCESS ) { // could obtain information which is ok.
return FALSE; }
dwStatus = RegQueryValueEx( hKey, TEXT( "DataObject" ), 0, 0, ( LPBYTE )pObj, &dwSizeOfDO );
if( dwStatus != ERROR_SUCCESS ) { DBGMSG( TEXT( "LSVIEW:RetrieveDataObject RegOpenKey succeed QueryValueEx failed 0x%x\n" ) , dwStatus ); RegCloseKey( hKey ); return FALSE; }
RegCloseKey( hKey );
return TRUE;
}
//-------------------------------------------------------------------------
BOOL LogFile( LPTSTR szFileName ){ FILE *fp = NULL; if( ( fp = _wfopen( szFileName , TEXT( "w" ) ) ) != NULL ) { DBGMSG( TEXT( "File name is %ws\n" ) , szFileName ) ;
// delimiter not specified use tabs,
// loop through list and construct a line
if( g_pHead != NULL ) { PLIST pItem = g_pHead->pNext;
while( pItem != NULL ) { WCHAR wchBuffer[ 260 ]; CHAR chBuffer[ 260 ];
if((sizeof(wchBuffer)/sizeof(wchBuffer[0])) < (wcslen(pItem->pszMachineName) + wcslen(pItem->pszTimeFormat) + wcslen(pItem->pszType) + 1 )) { fclose( fp ); return FALSE; }
wsprintf( wchBuffer , TEXT( "%ws\t%ws\t%ws\n" ) , pItem->pszMachineName , pItem->pszTimeFormat , pItem->pszType );
// DBCS is a hard when streaming; convert this to MBCS
WideCharToMultiByte( CP_ACP , 0, wchBuffer, SIZEOF( wchBuffer ), chBuffer, sizeof( chBuffer ), NULL, NULL );
fprintf( fp , chBuffer );
pItem = pItem->pNext; } }
fclose( fp ); }
return TRUE;}
|
