/*
* Copyright (c) 2005-2009 Nokia Corporation and/or its subsidiary(-ies).
* All rights reserved.
* This component and the accompanying materials are made available
* under the terms of "Eclipse Public License v1.0"
* which accompanies this distribution, and is available
* at the URL "http://www.eclipse.org/legal/epl-v10.html".
*
* Initial Contributors:
* Nokia Corporation - initial contribution.
*
* Contributors:
*
* Description: 
*
*/


#include <windows.h>
#include <stdio.h>
#include <setupapi.h>
#include <cfgmgr32.h>
#include <malloc.h>
#include <newdev.h>

#ifdef _UNICODE
#define UPDATEDRIVERFORPLUGANDPLAYDEVICES "UpdateDriverForPlugAndPlayDevicesW"
#else
#define UPDATEDRIVERFORPLUGANDPLAYDEVICES "UpdateDriverForPlugAndPlayDevicesA"
#endif

typedef BOOL (WINAPI *PUPDATEDRIVERFORPLUGANDPLAYDEVICES)(HWND hwndParent,
                                                         LPCTSTR HardwareId,
                                                         LPCTSTR FullInfPath,
                                                         DWORD InstallFlags,
                                                         PBOOL bRebootRequired OPTIONAL
                                                         );

int UpdateDevice(LPCTSTR inf, LPCTSTR hwid)
{
    HMODULE hMod = NULL;
    int nRet = -1;
    PUPDATEDRIVERFORPLUGANDPLAYDEVICES pUpdate;
    BOOL bReboot = FALSE;
    DWORD dwFlags = 0;
    DWORD res;
    TCHAR InfPath[MAX_PATH];

    res = GetFullPathName(inf,MAX_PATH,InfPath,NULL);
    if((res < MAX_PATH) && (res != 0)) {
		if(GetFileAttributes(InfPath) != (DWORD)(-1)) {
			inf = InfPath;
			dwFlags |= INSTALLFLAG_FORCE;

			hMod = LoadLibrary(TEXT("newdev.dll"));
			if(hMod) {
				pUpdate = (PUPDATEDRIVERFORPLUGANDPLAYDEVICES)GetProcAddress(hMod,UPDATEDRIVERFORPLUGANDPLAYDEVICES);
				if(pUpdate)
				{
					if(pUpdate(NULL,hwid,inf,dwFlags,&bReboot)) {
						nRet = 0;
					}
				}
			}
		}
	}

    if(hMod) {
        FreeLibrary(hMod);
    }

    return -1;
}



int install(LPCTSTR inf, LPCTSTR hwid)
{
	
    HDEVINFO hDeviceInfoSet = INVALID_HANDLE_VALUE;
    SP_DEVINFO_DATA DeviceInfoData;
    GUID ClassGUID;
    TCHAR ClassName[MAX_CLASS_NAME_LEN];
    TCHAR hwIdList[LINE_LEN+4];
    TCHAR InfPath[MAX_PATH];
    int nRet = -1;
    DWORD flags = 0;

    if(GetFullPathName(inf,MAX_PATH,InfPath,NULL) < MAX_PATH) {
		
		ZeroMemory(hwIdList,sizeof(hwIdList));
		lstrcpyn(hwIdList,hwid,LINE_LEN);

		if (SetupDiGetINFClass(InfPath,&ClassGUID,ClassName,sizeof(ClassName)/sizeof(ClassName[0]),0))
		{

			hDeviceInfoSet = SetupDiCreateDeviceInfoList(&ClassGUID,0);
			if(hDeviceInfoSet != INVALID_HANDLE_VALUE)
			{

				DeviceInfoData.cbSize = sizeof(SP_DEVINFO_DATA);
				if (SetupDiCreateDeviceInfo(hDeviceInfoSet,
					ClassName,
					&ClassGUID,
					NULL,
					0,
					DICD_GENERATE_ID,
					&DeviceInfoData))
				{

					if(SetupDiSetDeviceRegistryProperty(hDeviceInfoSet,
						&DeviceInfoData,
						SPDRP_HARDWAREID,
						(LPBYTE)hwIdList,
						(lstrlen(hwIdList)+1+1)*sizeof(TCHAR)))
					{

						if (SetupDiCallClassInstaller(DIF_REGISTERDEVICE,
							hDeviceInfoSet,
							&DeviceInfoData))
						{
							// update the driver for the device we just created
							//
							nRet = UpdateDevice(inf,hwid);
						}
					}
				}

				if (hDeviceInfoSet != INVALID_HANDLE_VALUE) {
					SetupDiDestroyDeviceInfoList(hDeviceInfoSet);
				}
			}
		}
	}
	return nRet;
}



BOOL IsWow64()
{
	typedef BOOL (WINAPI *LPFN_ISWOW64PROCESS) (HANDLE hProcess, BOOL *Wow64Process);
	BOOL bIsWow64 = FALSE;
	LPFN_ISWOW64PROCESS fnIsWow64Process = (LPFN_ISWOW64PROCESS)GetProcAddress( GetModuleHandle("kernel32"),"IsWow64Process");
	if( NULL != fnIsWow64Process)
	{
		if( !fnIsWow64Process( GetCurrentProcess(), &bIsWow64))
		{
		// handle error
		}
	}
	return bIsWow64;
}

void CreateChildProcessAndWaitUntilDone(const LPSTR strCmdLine) 
{
	PROCESS_INFORMATION piProcInfo; 
	STARTUPINFO siStartInfo;

	// Set up members of STARTUPINFO structure.
	siStartInfo.cb = sizeof(STARTUPINFO); 
	siStartInfo.lpReserved = NULL;
	siStartInfo.lpReserved2 = NULL; 
	siStartInfo.cbReserved2 = 0;
	siStartInfo.lpDesktop = NULL; 
	siStartInfo.dwFlags = 0;


	// Create the child process.
	CreateProcess(
		NULL,
		strCmdLine,
		NULL, // process security attributes
		NULL, // primary thread security attributes
		0, // handles are inherited
		0, // creation flags
		NULL, // use parent's environment
		NULL, // use parent's current directory
		&siStartInfo, // STARTUPINFO pointer
		&piProcInfo); // receives PROCESS_INFORMATION

	// Wait for the processs to finish
	DWORD rc = WaitForSingleObject(
	piProcInfo.hProcess, // process handle
	INFINITE); 
}
void main(int argc, LPTSTR argv[])
{
	int a=0;

	if(IsWow64())
	{
		SetCurrentDirectory("x64driver");
		CreateChildProcessAndWaitUntilDone("tapinstaller.x64.exe");
	}
	else 
	{
		OSVERSIONINFO ver;
		ver.dwOSVersionInfoSize = sizeof(ver);
		GetVersionEx(&ver);
		if(ver.dwMajorVersion == 6)
		{
			LPTSTR inf="tap0901.inf";
			LPTSTR hwid = "tap0901";
			install(inf,hwid);
		}
		else 
		{
			LPTSTR inf="OemWin2k.inf";
			LPTSTR hwid = "tap0801";
			install(inf,hwid);
		}
	}
	return;
}