//*********************************************************************
//*                  Microsoft Windows                               **
//*            Copyright(c) Microsoft Corp., 1994                    **
//*********************************************************************

//
//	INIT.C - WinMain and initialization code for Internet setup/signup wizard
//

//	HISTORY:
//	
//	11/20/94	jeremys	Created.
//  96/03/26  markdu  Put #ifdef __cplusplus around extern "C"
//

#include "wizard.h"

#ifdef WIN32
#include "..\inc\semaphor.h"
#endif

#define IEAK_RESTRICTION_REGKEY        TEXT("Software\\Policies\\Microsoft\\Internet Explorer\\Control Panel")
#define IEAK_RESTRICTION_REGKEY_VALUE  TEXT("Connwiz Admin Lock")

// superceded by definition in semaphor.h
//#define SEMAPHORE_NAME "Internet Connection Wizard INETWIZ.EXE"

HINSTANCE ghInstance=NULL;

LPTSTR LoadSz(UINT idString,LPTSTR lpszBuf,UINT cbBuf);

#ifdef __cplusplus
extern "C"
{
#endif // __cplusplus

	VOID WINAPI LaunchSignupWizard(LPTSTR lpCmdLine,int nCmdShow, PBOOL pReboot);

#ifdef __cplusplus
}
#endif // __cplusplus

BOOL DoesUserHaveAdminPrivleges(HINSTANCE hInstance)
{
    HKEY hKey = NULL;
    BOOL bRet = FALSE;

    if (!IsNT())
        return TRUE;

    // BUGBUG: We should allow NT5 to run in all user groups
    // except normal users.
    if (IsNT5())
        return TRUE;

    //
    // Ensure caller is an administrator on this machine.
    //
    if (ERROR_SUCCESS == RegOpenKeyEx(HKEY_LOCAL_MACHINE,
                                      (TCHAR*)TEXT("SYSTEM\\CurrentControlSet"),
                                      0,
                                      KEY_ALL_ACCESS,
                                      &hKey))
    {
        bRet = TRUE;
        CloseHandle(hKey);
    }
    else
    {
        TCHAR szAdminDenied      [MAX_PATH] = TEXT("\0");
        TCHAR szAdminDeniedTitle [MAX_PATH] = TEXT("\0");
        LoadString(hInstance, IDS_ADMIN_ACCESS_DENIED, szAdminDenied, MAX_PATH);
        LoadString(hInstance, IDS_ADMIN_ACCESS_DENIED_TITLE, szAdminDeniedTitle, MAX_PATH);
        MessageBox(NULL, szAdminDenied, szAdminDeniedTitle, MB_OK | MB_ICONSTOP);
    }

    return bRet;
}

BOOL CheckForIEAKRestriction(HINSTANCE hInstance)
{
    HKEY hkey = NULL;
    BOOL bRC = FALSE;
    DWORD dwType = 0;
    DWORD dwSize = 0;
    DWORD dwData = 0;

    if (ERROR_SUCCESS == RegOpenKey(HKEY_CURRENT_USER,
        IEAK_RESTRICTION_REGKEY,&hkey))
    {
        dwSize = sizeof(dwData);
        if (ERROR_SUCCESS == RegQueryValueEx(hkey,IEAK_RESTRICTION_REGKEY_VALUE,0,&dwType,
            (LPBYTE)&dwData,&dwSize))
        {
            if (dwData)
            {   
                TCHAR szIEAKDenied[MAX_PATH];
                TCHAR szIEAKDeniedTitle[MAX_PATH];
                LoadString(hInstance, IDS_IEAK_ACCESS_DENIED, szIEAKDenied, MAX_PATH);
                LoadString(hInstance, IDS_IEAK_ACCESS_DENIED_TITLE, szIEAKDeniedTitle, MAX_PATH);
                MessageBox(NULL, szIEAKDenied, szIEAKDeniedTitle, MB_OK | MB_ICONSTOP);
                bRC = TRUE;
            }
        }
   }

   if (hkey)
        RegCloseKey(hkey);

    return bRC;
}

/*******************************************************************

	NAME:		WinMain

	SYNOPSIS:	App entry point

********************************************************************/
int APIENTRY WinMain(HINSTANCE hInstance,HINSTANCE hPrevInstance,
	LPSTR lpCmdLine,int nCmdShow)
{
    BOOL bReboot = FALSE;
    
#ifdef UNICODE
    // Initialize the C runtime locale to the system locale.
    setlocale(LC_ALL, "");
#endif

	// only allow one instance of wizard
	// note: hPrevInstance is always NULL for Win32 apps, so need to look
	// for existing window

    HANDLE hSemaphore = NULL;
	hSemaphore = CreateSemaphore(NULL, 1, 1, ICW_ELSE_SEMAPHORE);
	DWORD dwErr = GetLastError();
	if ( ERROR_ALREADY_EXISTS == dwErr )
	{
		IsAnotherComponentRunning32( FALSE );
		// every instance should close its own semaphore handle
		goto WinMainExit;
	}
	else
	{
		if( IsAnotherComponentRunning32( TRUE ) )
			goto WinMainExit;
	}

    //
    // remove the batch files if we are running of NT and
    // started of 1
    //
    if (IsNT ())
        DeleteStartUpCommand();

    if (!DoesUserHaveAdminPrivleges(hInstance))
        //no, bail.
        goto WinMainExit;

    if (CheckForIEAKRestriction(hInstance))
        //yes, bail.
        goto WinMainExit;

    //
	// call our DLL to run the wizard
    //
#ifdef UNICODE
        TCHAR szCmdLine[MAX_PATH+1];
        mbstowcs(szCmdLine, lpCmdLine, MAX_PATH+1);
	LaunchSignupWizard(szCmdLine, nCmdShow, &bReboot);
#else
	LaunchSignupWizard(lpCmdLine, nCmdShow, &bReboot);
#endif

    //
    // we should reboot, if the flag is set
    //
    if (TRUE == bReboot)
    {
        //
        // as the user if we would like to reboot at this time
        //
        TCHAR szMessage[256];
        TCHAR szTitle[256];
        LoadSz (IDS_WANTTOREBOOT, szMessage, sizeof (szMessage));
        LoadSz (IDS_WIZ_WINDOW_NAME, szTitle, sizeof (szTitle));

	    if (IDYES == MessageBox(NULL, szMessage, szTitle, MB_ICONQUESTION | MB_YESNO | MB_DEFBUTTON2))

		{
            SetRunOnce ();
		    if (!FGetSystemShutdownPrivledge() ||
					!ExitWindowsEx(EWX_REBOOT,0))
	        {
                
                TCHAR szFailMessage[256];
                LoadSz (IDS_EXITFAILED, szFailMessage, sizeof (szFailMessage));
	            MessageBox(NULL, szFailMessage, szTitle, MB_ICONERROR | MB_OK);

		    }

			//
			// ChrisK Olympus 4212, Allow Inetwiz to exit on restart
			//
            //else
            //{
            //    //      
            //    //  We will wait for the System to release all
            //    //  resources, 5 minutes should be more than 
            //    //  enough for this 
            //    //  - MKarki (4/22/97)  Fix for Bug #3109
            //    //
            //    Sleep (300000);
            //}
		}
    }


WinMainExit:
	if( hSemaphore ) 
		CloseHandle(hSemaphore);

	return 0;

}

/*******************************************************************

	NAME:		LoadSz

	SYNOPSIS:	Loads specified string resource into buffer

	EXIT:		returns a pointer to the passed-in buffer

	NOTES:		If this function fails (most likely due to low
				memory), the returned buffer will have a leading NULL
				so it is generally safe to use this without checking for
				failure.

********************************************************************/
LPTSTR LoadSz(UINT idString,LPTSTR lpszBuf,UINT cbBuf)
{
	ASSERT(lpszBuf);

	// Clear the buffer and load the string
    if ( lpszBuf )
    {
        *lpszBuf = '\0';
        LoadString( ghInstance, idString, lpszBuf, cbBuf );
    }
    return lpszBuf;
}



#ifdef WIN32
//+---------------------------------------------------------------------------
//
//  Function:   IsAnotherComponentRunning32()
//
//  Synopsis:   Checks if there's another ICW component already
//				running.  If so, it will set focus to that component's window.
//
//				This functionality is needed by all of our .exe's.  However,
//				the actual components to check for differ between .exe's.
//				The comment COMPONENT SPECIFIC designates lines of code
//				that vary between components' source code.
//
//				For ICWCONN1, we return FALSE if the following are already running:
//				  another instance of ICWCONN1, any other component
//
//  Arguments:  bCreatedSemaphore --
//					--	if TRUE, then this component has successfully created
//						it's semaphore, and we want to check whether other
//						components are running
//					--	if FALSE, then this component could not create
//						it's semaphore, and we want to find the already
//						running instance of this component
//
//	Returns:	TRUE if another component is already running
//				FALSE otherwise
//
//  History:    12/3/96	jmazner		Created, with help from IsAnotherInstanceRunning
//									in icwconn1\connmain.cpp
//
//----------------------------------------------------------------------------
BOOL IsAnotherComponentRunning32(BOOL bCreatedSemaphore)
{

	HWND hWnd = NULL;
	HANDLE hSemaphore = NULL;
	DWORD dwErr = 0;

	TCHAR szWindowName[SMALL_BUF_LEN+1];
	
	if( !bCreatedSemaphore )
	{
		// something other than conn1 is running,
		// let's find it!
		// COMPONENT SPECIFIC
		LoadSz(IDS_WIZ_WINDOW_NAME,szWindowName,sizeof(szWindowName));
		hWnd = FindWindow(DIALOG_CLASS_NAME, szWindowName);

		if( hWnd )
		{
			SetFocus(hWnd);
			SetForegroundWindow(hWnd);
		}

		// regardless of whether we found the window, return TRUE
		// since bCreatedSemaphore tells us this component was already running.
		return TRUE;
	}
	else
	{
		// check whether CONN1 is running
	    // The particular semaphores we look for are
		// COMPONENT SPECIFIC

		// is conn1 running?
		hSemaphore = CreateSemaphore(NULL, 1, 1, ICWCONN1_SEMAPHORE);
		dwErr = GetLastError();

		// close the semaphore right away since we have no use for it.
		if( hSemaphore )
			CloseHandle(hSemaphore);
		
		if( ERROR_ALREADY_EXISTS == dwErr )
		{
			// conn1 is running.
			// Bring the running instance's window to the foreground
			LoadSz(IDS_WIZ_WINDOW_NAME,szWindowName,sizeof(szWindowName));
			hWnd = FindWindow(DIALOG_CLASS_NAME, szWindowName);
			
			if( hWnd )
			{
				SetFocus(hWnd);
				SetForegroundWindow(hWnd);
			}

			return( TRUE );
		}

		// didn't find any other running components!
		return( FALSE );
	}
}
#endif