//==========================================================================//
//                                  Includes                                //
//==========================================================================//
#include <string.h>     // strupr
#include <stdio.h>      // for sprintf.


#include "perfmon.h"
#include "init.h"       // external declarations for this file

#include "alert.h"      // for AlertIitializeApplication
#include "command.h"    // for ViewChart
#include "grafdata.h"   // for QuerySaveChart
#include "graph.h"      // for GraphInitializeApplication
#include "legend.h"     // for LegendInitializeApplication
#include "log.h"        // for LogInitializeApplication
#include "intrline.h"   // for ILineInitializeApplication
#include "perfdata.h"   // for PerfDataInitializeInstance
#include "perfmops.h"   // for OpenFileHandler, for now
#include "status.h"     // for StatusInitializeApplication
#include "timeline.h"   // for TLineInitializeApplication
#include "playback.h"   // for PlaybackInitializeInstance
#include "registry.h"   // for Load/SaveMainWindowPlacement
#include "report.h"     // for ReportInitializeApplication
#include "toolbar.h"    // for ToolbarInitializeApplication
#include "utils.h"
#include "fileopen.h"   // for FileOpen
#include "pmemory.h"    // for MemoryFree

extern   TCHAR          DefaultLangId[] ;
extern   TCHAR          EnglishLangId[] ;

static   LPSTR           lpszCommandLine ;

//==========================================================================//
//                                  Constants                               //
//==========================================================================//


#define szPerfmonMainClass TEXT("PerfmonMainClass")

HHOOK   lpMsgFilterProc ;
DWORD FAR PASCAL MessageFilterProc (int nCode, WPARAM wParam,
                                    LPARAM lParam) ;


//==========================================================================//
//                              Local Functions                             //
//==========================================================================//

static
LONG
EnablePrivilege (
                IN  LPTSTR  szPrivName
                )
{
    LUID SePrivNameValue;
    TOKEN_PRIVILEGES tkp;

    HANDLE hToken = NULL;

    /* Retrieve a handle of the access token. */

    if (!OpenProcessToken(GetCurrentProcess(),
                          TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY,
                          &hToken)) {
        goto Exit_Point;
    }

    /*
     * Enable the privilege by name and get the ID
     */

    if (!LookupPrivilegeValue((LPTSTR) NULL,
                              szPrivName,
                              &SePrivNameValue)) {
        goto Exit_Point;
    }

    tkp.PrivilegeCount = 1;
    tkp.Privileges[0].Luid = SePrivNameValue;
    tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;

    AdjustTokenPrivileges(hToken,
                          FALSE,
                          &tkp,
                          sizeof(TOKEN_PRIVILEGES),
                          (PTOKEN_PRIVILEGES) NULL,
                          (PDWORD) NULL);

    /* The return value of AdjustTokenPrivileges be texted. */

    Exit_Point:

    if (hToken != NULL) CloseHandle (hToken);
    return GetLastError();
}

void
GetScalesFonts (void)
{
    LOGFONT      lf ;
    CHARSETINFO  csi;
    DWORD        dw = GetACP();

    if (!TranslateCharsetInfo(UlongToPtr(dw), &csi, TCI_SRCCODEPAGE))
        csi.ciCharset = ANSI_CHARSET;

    memset (&lf, 0, sizeof (lf)) ;

    lstrcpy (lf.lfFaceName, szScalesFontFace) ;
    /*
       if (csi.ciCharset == ANSI_CHARSET)
           lf.lfHeight = iScalesFontHeightAnsi ;
       else
           lf.lfHeight = iScalesFontHeightNonAnsi ;
    */
    // make the text the same size as a menu text
    lf.lfHeight = GetSystemMetrics (SM_CYMENUCHECK);
    lf.lfWeight = FW_REGULAR ;
    lf.lfCharSet = (BYTE)csi.ciCharset;

    hFontScales = CreateFontIndirect (&lf) ;

    if (csi.ciCharset == ANSI_CHARSET)
        lf.lfWeight = FW_BOLD ;
    hFontScalesBold = CreateFontIndirect (&lf) ;
}


BOOL
InitializeSystemValues (void)
/*
   Effect:        Read and store in variables the various system values,
                  such as the width and height of the screen and icons,
                  the width of scroll bars, etc.

   Called By:     PerfmonInitialize only.

   Returns:       Whether this function was successful in getting all
                  needed system values.
*/
{
    xScreenWidth =  GetSystemMetrics (SM_CXSCREEN) ;
    yScreenHeight = GetSystemMetrics (SM_CYSCREEN) ;

    xBorderWidth = GetSystemMetrics (SM_CXBORDER) ;
    yBorderHeight = GetSystemMetrics (SM_CYBORDER) ;

    xScrollWidth = GetSystemMetrics (SM_CXVSCROLL) ;
    yScrollHeight = GetSystemMetrics (SM_CYHSCROLL) ;

    xScrollThumbWidth = GetSystemMetrics (SM_CXHTHUMB) ;
    yScrollThumbHeight = GetSystemMetrics (SM_CYVTHUMB) ;

    xDlgBorderWidth = GetSystemMetrics (SM_CXDLGFRAME) ;
    yDlgBorderHeight = GetSystemMetrics (SM_CYDLGFRAME) ;

    MinimumSize = yScrollHeight +
                  GetSystemMetrics (SM_CYMENU) +
                  GetSystemMetrics (SM_CYCAPTION) ;

    //================================================================//
    // create all the brushes and pens for performance improvement    //
    //================================================================//
    CreatePerfmonSystemObjects () ;
    hWhitePen = CreatePen (PS_SOLID, 3, crWhite) ;

    return (TRUE) ;
}


BOOL
InitializeApplication (void)
/*
   Effect:        Perform all initializations required for the FIRST
                  instance of the Perfmon application. In particular,
                  register all of Perfmon's window classes.

   Note:          There is no background brush set for the MainWindow
                  class so that the main window is never erased. The
                  client area of MainWindow is always covered by one
                  of the view windows. If we erase it, it would just
                  flicker needlessly.

   Called By:     PerfmonInitialize only.

   Returns:       Whether this function was successful in initializing.
*/
{
    BOOL           bSuccess ;
    WNDCLASS       wc ;
    TCHAR          LocalHelpFileName [ShortTextLen] ;
    LPTSTR         pFileName ;

    hIcon = LoadIcon (hInstance, idIcon) ;

    //=============================//
    // Register Main window class  //
    //=============================//

    wc.style         = CS_DBLCLKS | CS_BYTEALIGNCLIENT;
    wc.lpfnWndProc   = MainWndProc;
    wc.hInstance     = hInstance;
    wc.cbClsExtra    = 0 ;
    wc.cbWndExtra    = 0;
    wc.hIcon         = hIcon ;
    wc.hCursor       = LoadCursor(NULL, IDI_APPLICATION);
    wc.hbrBackground = NULL ;                             // see note above
    wc.lpszMenuName  = idMenuChart ;
    wc.lpszClassName = szPerfmonMainClass ;

    bSuccess = RegisterClass (&wc) ;

    //=============================//
    // Register Abstract "Systems" //
    //=============================//
    hbLightGray = GetStockObject (LTGRAY_BRUSH) ;

    if (bSuccess)
        bSuccess = StatusInitializeApplication () ;

    if (bSuccess)
        bSuccess = GraphInitializeApplication () ;

#ifdef ADVANCED_PERFMON
    if (bSuccess)
        bSuccess = LogInitializeApplication () ;

    if (bSuccess)
        bSuccess = AlertInitializeApplication () ;

    if (bSuccess)
        bSuccess = ReportInitializeApplication () ;

    if (bSuccess)
        bSuccess = ILineInitializeApplication () ;

    if (bSuccess)
        bSuccess = TLineInitializeApplication () ;
#endif

    // setup messagehook to handle F1 as help
    lpMsgFilterProc = SetWindowsHookEx (WH_MSGFILTER,
                                        (HOOKPROC) MessageFilterProc,
                                        hInstance,
                                        GetCurrentThreadId()) ;

    // get the help file full path name
    LoadString (hInstance, IDS_HELPFILE_NAME,
                (LPTSTR)LocalHelpFileName, ShortTextLen-1);


    if (LocalHelpFileName[0]) {
        DWORD dwLength;
        pszHelpFile = (LPTSTR) MemoryAllocate (FilePathLen * sizeof (TCHAR)) ;
        dwLength = SearchPath (NULL, LocalHelpFileName, NULL,
                               FilePathLen - 1, pszHelpFile, &pFileName) ;
        if (dwLength == 0) {
            // not found in the path so check the help dir
            TCHAR szSearchPath[MAX_PATH];
            ExpandEnvironmentStrings (TEXT("%windir%\\help"),
                                      szSearchPath, ((sizeof (szSearchPath)/sizeof (TCHAR)) -1));
            SearchPath (szSearchPath, LocalHelpFileName, NULL,
                        FilePathLen - 1, pszHelpFile, &pFileName) ;
        }
    } else {
        // no help file
        pszHelpFile = (LPTSTR) MemoryAllocate (sizeof (TCHAR)) ;
        if (pszHelpFile) {
            *pszHelpFile = TEXT('\0') ;
        }
    }

    return (bSuccess) ;
}


BOOL
InitializeInstance (
                    int nCmdShow,
                    LPCSTR lpszCmdLine
                    )
/*
   Effect:        Perform all initializations required for EACH instance
                  of the Perfmon application. In particular, create all
                  of Perfmon's initial windows, and perform any other
                  initializations except registering classes (done in
                  InitializeApplication).

   Called By:     PerfmonInitialize only.

   Note:          This function has multiple return points.

   Returns:       Whether this function was successful in initalizing.
*/
{
    DWORD          ComputerNameLength;
    TCHAR          szApplication [WindowCaptionLen] ;

    bCloseLocalMachine = FALSE;

    // enable privileges needed to profile system
    // if this fails, that's ok for now.

    EnablePrivilege (SE_SYSTEM_PROFILE_NAME);    // to access perfdata
    EnablePrivilege (SE_INC_BASE_PRIORITY_NAME); // to raise priority

    //=============================//
    // Set Priority high           //
    //=============================//

    SetPriorityClass(GetCurrentProcess(), HIGH_PRIORITY_CLASS) ;
    SetThreadPriority(GetCurrentThread(), THREAD_PRIORITY_HIGHEST) ;

    //=============================//
    // Load Resources              //
    //=============================//

    GetScalesFonts () ;

    hMenuChart = LoadMenu (hInstance, idMenuChart) ;

#ifdef ADVANCED_PERFMON
    hMenuAlert = LoadMenu (hInstance, idMenuAlert) ;
    hMenuLog = LoadMenu (hInstance, idMenuLog) ;
    hMenuReport = LoadMenu (hInstance, idMenuReport) ;
#endif

    hAccelerators = LoadAccelerators (hInstance, idAccelerators) ;


    //=============================//
    // Initialize Systems          //
    //=============================//

    iLanguage = GetUserDefaultLangID() ;
    iEnglishLanguage = MAKELANGID (LANG_ENGLISH, LANG_NEUTRAL) ;
    //   iEnglishLanguage = MAKELANGID (iLanguage & 0x0ff, LANG_NEUTRAL) ;
    TSPRINTF (DefaultLangId, TEXT("%03x"), iLanguage) ;
    TSPRINTF (EnglishLangId, TEXT("%03x"), iEnglishLanguage) ;

    // GetComputerName returns the name without the "\\" prefix. We add
    // the prefix before even calling the routine. This is so that all our
    // computer names have the prefix and are therefore compatible with
    // I_SetSystemFocus (see perfmops.c).

    ComputerNameLength = MAX_PATH + 1;
    lstrcpy (LocalComputerName, szComputerPrefix) ;
    GetComputerName (LocalComputerName + lstrlen (szComputerPrefix),
                     &ComputerNameLength);

    PlaybackInitializeInstance () ;
    PerfDataInitializeInstance () ;

    //Copy command line into buffer for modification
    lpszCommandLine = (LPSTR)MemoryAllocate (strlen (lpszCmdLine) + sizeof(CHAR)) ;
    if (lpszCommandLine)
        strcpy(lpszCommandLine, lpszCmdLine) ;
    else
        return (FALSE) ;


    StringLoad (IDS_APPNAME, szApplication) ;

    //Insure CmdLineComputerName starts out NULL
    CmdLineComputerName[0] = TEXT('\0') ;

    //Check to see if there is a computer name specified on the command
    //line (indicated by /c, or /C).  If there is, get it and
    //null out the command line string so that it is not mistaken for
    //startup file.  This assumes that specifying a computer and a startup
    //file on the command line are mutually exclusive.  If the /c option
    //is on the command line, anything else on the command line is discarded.
    if (!strempty (lpszCommandLine)) {
        LPSTR               lpszNewCL ;

        //Preserve initial command line pointer position
        lpszNewCL = (LPSTR)lpszCommandLine ;
        while (*lpszNewCL != '\0' && *lpszCommandLine) {
            if (*lpszNewCL  == '/' || *lpszNewCL    == '-') {
                lpszNewCL++ ;
                if (*lpszNewCL == 'c' || *lpszNewCL == 'C') {
                    int i = 0;

                    lpszNewCL++ ;
                    //ignore leading backslashes * spaces
                    while (*lpszNewCL == '\\' || *lpszNewCL == ' ')
                        lpszNewCL++ ;

                    //Terminate at first blank space if there is one.
                    //We don't allow anything else on the command line
                    //if a computer name is specified.
                    while (lpszNewCL[i] != ' ' && lpszNewCL[i] != '\0')
                        i++ ;
                    lpszNewCL[i] = '\0' ;

                    if ((*lpszNewCL != '\0') && (strlen(lpszNewCL) <= MAX_PATH)) {
                        int nSizePrefix ;

                        nSizePrefix = lstrlen (szComputerPrefix) ;
                        lstrcpy( CmdLineComputerName, szComputerPrefix ) ;
                        MultiByteToWideChar (CP_ACP,
                                             MB_PRECOMPOSED,
                                             lpszNewCL,
                                             -1,
                                             CmdLineComputerName + nSizePrefix,
                                             sizeof(CmdLineComputerName) - nSizePrefix) ;
                        //prevent lpszCommandLine from being used as input file & stop while loop
                        *lpszCommandLine = '\0' ;
                    } else {
                        LPTSTR  lpszErrMsg ;
                        TCHAR       lpszFormat[80] ;

                        LoadString(hInstance, ERR_BADCOMPUTERNAME, lpszFormat, sizeof(lpszFormat)/sizeof(TCHAR)) ;
                        lpszErrMsg = (LPTSTR)MemoryAllocate ((strlen (lpszNewCL) + 1)*sizeof(TCHAR) + sizeof(lpszFormat)) ;
                        //If memory allocation failed, don't display error message
                        if (lpszErrMsg) {
                            //lpszFormat uses %S specifier so lpszNewCL can be ansi string
                            wsprintf (lpszErrMsg, lpszFormat, lpszNewCL) ;
                            MessageBox (NULL, lpszErrMsg, szApplication, MB_OK | MB_ICONSTOP | MB_TASKMODAL) ;
                        }

                        //prevent lpszCommandLine from being used as input file & stop while loop
                        *lpszCommandLine = '\0' ;
                    }
                }
            }
            lpszNewCL++ ;
        }
    }

    //=============================//
    // Create Window               //
    //=============================//


    hWndMain = CreateWindow (szPerfmonMainClass,
                             szApplication,
                             WS_OVERLAPPEDWINDOW | WS_BORDER,
                             CW_USEDEFAULT, CW_USEDEFAULT,
                             CW_USEDEFAULT, CW_USEDEFAULT,
                             NULL,
                             NULL,
                             NULL,
                             NULL);

    if (!hWndMain)
        return (FALSE) ;

    ViewChart (hWndMain) ;

    LoadMainWindowPlacement (hWndMain) ;

    //=============================//
    // Setup for event logging     //
    //=============================//
    hEventLog = RegisterEventSource (
                                    (LPTSTR)NULL,            // Use Local Machine
                                    TEXT("PerfMon"));        // event log app name to find in registry

    return (TRUE) ;
}


//==========================================================================//
//                             Exported Functions                           //
//==========================================================================//


BOOL
PerfmonInitialize (
                   HINSTANCE hCurrentInstance,
                   HINSTANCE hPrevInstance,
                   LPCSTR lpszCmdLine,
                   int nCmdShow
                   )
/*
   Effect:        Performa all initializations required when Perfmon is
                  started. In particular, initialize all "systems", register
                  all window classes, create needed windows, read in and
                  process font and Perfmon lists.

   Called By:     WinMain only, at the start of the application.

   Assert:        There are no other instances of Perfmon currently
                  executing.

   Returns:       Whether initialization was successful. If this function
                  returns FALSE, Perfmon should exit immediately.

   Internals:     The bSuccess variable is used to conditionalize each
                  successive initialization step.
*/
{
    BOOL           bSuccess ;
    TCHAR          szFilePath [FilePathLen + 1] ;
    LPTSTR         pFileNameStart ;
    HANDLE         hFindFile ;
    WIN32_FIND_DATA FindFileInfo ;
    CHAR           QuoteChar ;
    LPSTR          pCmdLine ;
    LPSTR          lpszCmdLineStart = NULL ;
    SIZE_T         NameOffset ;
    int            StringLen ;

    hInstance = hCurrentInstance ;
    bSuccess = InitializeSystemValues () ;

    if (bSuccess && !hPrevInstance)
        bSuccess = InitializeApplication () ;

    if (bSuccess)
        bSuccess = InitializeInstance (nCmdShow, lpszCmdLine) ;

    GetDateTimeFormats() ;

    lpszCmdLineStart = lpszCommandLine ;


    if (bSuccess) {

        if (strempty (lpszCommandLine))
            StringLoad (IDS_DEFAULTPATH, szFilePath) ;
        else {
            // check for single or double quote
            QuoteChar = *lpszCommandLine ;
            if (QuoteChar == '\'' || QuoteChar == '\"') {
                lpszCommandLine++ ;

                // remove the matching QuoteChar if found
                pCmdLine = (LPSTR) lpszCommandLine ;
                while (*pCmdLine != '\0') {
                    if (*pCmdLine == QuoteChar) {
                        *pCmdLine = '\0' ;
                        break ;
                    } else {
                        pCmdLine++ ;
                    }
                }
            }

            // convert the LPSTR to LPTSTR

            StringLen = strlen(lpszCommandLine) + 1 ;
            if (StringLen > sizeof(szFilePath)) {
                StringLen = sizeof (szFilePath) - sizeof (TEXT('\0')) ;
            }
            szFilePath[FilePathLen] = TEXT('\0') ;

            mbstowcs (szFilePath, lpszCommandLine, StringLen) ;

            if (!(QuoteChar == '\'' || QuoteChar == '\"')) {
                // if there is no quote, then looking for trailing space
                LPTSTR  pThisChar = szFilePath ;
                LPTSTR  pLastChar = NULL;
                LPTSTR  pSpaceChar = NULL;
                BOOL    bWord = FALSE;

                while (*pThisChar != TEXT('\0')) {
                    if (*pThisChar > TEXT(' ')) {
                        // this is a printable char
                        bWord = TRUE;
                        pLastChar = pThisChar;
                    } else if (bWord) {
                        bWord = FALSE;
                        pSpaceChar = pThisChar;
                    }
                    pThisChar++ ;
                }
                // terminate string after last printable character.
                if (pSpaceChar > pLastChar) *pSpaceChar = TEXT('\0');
            }

            pFileNameStart = ExtractFileName (szFilePath) ;
            NameOffset = pFileNameStart - szFilePath ;

            // convert short filename to long NTFS filename if necessary
            hFindFile = FindFirstFile (szFilePath, &FindFileInfo) ;
            if (hFindFile && hFindFile != INVALID_HANDLE_VALUE) {
                // append the file name back to the path name
                lstrcpy (&szFilePath[NameOffset], FindFileInfo.cFileName) ;
                FindClose (hFindFile) ;
            }
        }

        //      OpenFileHandler (hWndMain, szFilePath) ;
        FileOpen (hWndMain, (int)0, (LPTSTR)szFilePath) ;
        PrepareMenu (GetMenu (hWndMain));
    }

    if (lpszCmdLineStart) {
        MemoryFree (lpszCmdLineStart) ;
        lpszCommandLine = NULL ;
    }

    return (bSuccess) ;
}


void
PerfmonClose (
              HWND hWndMain
              )
{
    DWORD          dwException;
    // close the log file for now.
    // need to query the user later..!!
#if 0
    if (LogCollecting (hWndLog)) {
        PLOG pLog = LogData (hWndLog) ;

        if (pLog) {
            CloseLog (hWndLog, pLog) ;
        }
    }
#endif


    // reset all views - will free all systems as well
    ResetGraphView (hWndGraph) ;
    ResetAlertView (hWndAlert) ;
    ResetLogView (hWndLog) ;
    ResetReportView (hWndReport) ;

    try {
        // if the key has already been closed, this will
        // generate an exception so we'll catch that here
        // and continue...
        //
        // close the local machine
        if (bCloseLocalMachine) {
            RegCloseKey (HKEY_PERFORMANCE_DATA) ;
        }
    }except (EXCEPTION_EXECUTE_HANDLER) {
        // get the exception for the debugger user, that's all
        dwException = GetExceptionCode();
    }


    // free all the filenames
    if (pChartFullFileName) {
        MemoryFree (pChartFullFileName) ;
        pChartFullFileName = NULL ;
    }
    if (pAlertFullFileName) {
        MemoryFree (pAlertFullFileName) ;
        pAlertFullFileName = NULL ;
    }
    if (pLogFullFileName) {
        MemoryFree (pLogFullFileName) ;
        pLogFullFileName = NULL ;
    }
    if (pReportFullFileName) {
        MemoryFree (pReportFullFileName) ;
        pReportFullFileName = NULL ;
    }

    // free all the GDI resources
    DeletePen (hWhitePen) ;
    DeletePerfmonSystemObjects () ;

    // close event log
    DeregisterEventSource (hEventLog);

    SaveMainWindowPlacement (hWndMain) ;
    DestroyWindow (hWndMain) ;
}