#include "IsapiStress.h"

// Globals
HANDLE hLogFile = INVALID_HANDLE_VALUE;


void Usage()
{
	_tprintf(_T("Usage:\n\r"));
	_tprintf(_T("IsapiStress /s:<IISServer> /v:<VirtualDirectory> /d:<Directory> /f:<FileName> /l:<LogFileName> /? <Usage>\r\n"));
	_tprintf(_T("Where:\r\n"));
	_tprintf(_T("\t<IISServer>   - Website to upload file to (ie. ocadeviis) \r\n"));
	_tprintf(_T("\t<VirualDirectory> - Location on Web server to upload the file(s) to.\r\n"));
	_tprintf(_T("\t<Directory>   - Path to a directory of .cab files to be uploaded\r\n"));
	_tprintf(_T("\t<FileName>    - Qualified path of a file to be uploaded.\r\n"));
	_tprintf(_T("\t<LogFileName> - Name of file to log OCA response url or errors.\r\n\r\n"));
	_tprintf(_T("\tNote: Default iisserver is ocatest. \r\n\t      Default VirtualDirectory is OCA"));

}


void 
LogMessage(TCHAR *pFormat,...)
{
	// Routine to Log Fatal Errors to NT Event Log
    TCHAR    chMsg[256];
	DWORD    dwBytesWritten;
    va_list  pArg;

    va_start(pArg, pFormat);
    StringCbVPrintf(chMsg, sizeof chMsg,pFormat, pArg);
    va_end(pArg);
	// add a cr\lf combination for file fomrating.
    StringCbCat(chMsg, sizeof chMsg,_T("\r\n"));
    if (hLogFile != INVALID_HANDLE_VALUE)
    {
        /* Write to event log. */
        WriteFile(hLogFile,
				  chMsg,
				  _tcslen(chMsg) * (DWORD)sizeof TCHAR,
				  &dwBytesWritten,
				  NULL);
     }
	else // write it to the console.
		_tprintf(chMsg);

}

DWORD  Upload(TCHAR *SourceFileName, TCHAR *VirtualDir, TCHAR *HostName, TCHAR *RemoteFileName)
{
	static		const TCHAR *pszAccept[]			= {_T("*.*"), 0};
	//TCHAR       RemoteFileName[MAX_PATH]; // Host/Virtualdirectory/filename
	TCHAR		*pUploadUrl				= NULL;
	BOOL		bRet				= FALSE;
	BOOL		UploadSuccess		= FALSE;
	DWORD		dwBytesRead;
	DWORD		dwBytesWritten;
	DWORD		ResponseCode		= 0;
	DWORD		ResLength = 255;
	DWORD		index = 0;
	DWORD		ErrorCode			= 0;
	HINTERNET   hSession			= NULL;
	HINTERNET	hConnect			= NULL;
	HINTERNET	hRequest			= NULL;
	INTERNET_BUFFERS	BufferIn	= {0};
	INTERNET_BUFFERS   BufferOut	= {0};
	HANDLE      hFile = INVALID_HANDLE_VALUE;
	BYTE		*pBuffer;
	BOOL		bOnce				= FALSE;
	GUID		guidNewGuid;
	TCHAR		*szGuidRaw			= NULL;
	HRESULT		hResult = S_OK;
	wchar_t		*wszGuidRaw = NULL;

	CoInitialize(NULL);
	hResult = CoCreateGuid(&guidNewGuid);
	if (FAILED(hResult))
	{
		//-------------What do we send here....
		//goto ERRORS;
		;
	}
	else
	{
		if (UuidToStringW(&guidNewGuid, &wszGuidRaw) == RPC_S_OK)
		{
			if ( (szGuidRaw = (TCHAR *) malloc ( wcslen(wszGuidRaw)*2 )) != NULL)
			{
				// clear the memory
				ZeroMemory(szGuidRaw, wcslen(wszGuidRaw) * 2);
				wcstombs( szGuidRaw, wszGuidRaw, wcslen(wszGuidRaw));
			}
			else
			{
				LogMessage(_T("Memory allocation failed: ErrorCode:%d"),GetLastError());
				if (wszGuidRaw)
				{
					RpcStringFreeW(&wszGuidRaw);
				}
					CoUninitialize();
				return GetLastError();
			}
		}
	}
	if (wszGuidRaw)
	{
		RpcStringFreeW(&wszGuidRaw);
	}

	
	StringCbPrintf(RemoteFileName, MAX_PATH * sizeof TCHAR, _T("\\%s\\%s%s"),VirtualDir,szGuidRaw + 19, PathFindFileName(SourceFileName));
	if (szGuidRaw)
			free (szGuidRaw);
	hSession = InternetOpen(	_T("IsapiStress"),
								INTERNET_OPEN_TYPE_PRECONFIG,
								NULL,
								NULL,
								0);
	if (!hSession)
	{
		LogMessage(_T("Failed to create an internet session."));
			CoUninitialize();
		return GetLastError();
	}

	hConnect = InternetConnect(hSession,
								HostName,
								INTERNET_DEFAULT_HTTP_PORT,
								NULL,
								NULL,
								INTERNET_SERVICE_HTTP,
								0,
								NULL);

	if (hConnect == INVALID_HANDLE_VALUE)
	{
		InternetCloseHandle(hSession);	
		LogMessage(_T("Failed to create an internet connection."));
		CoUninitialize();
		return GetLastError();
	}
	LogMessage(_T("Connecting to: %s"),HostName);
	LogMessage(_T("Uploading file: %s"),RemoteFileName);

	hRequest = HttpOpenRequest(	hConnect,
								_T("PUT"),
								RemoteFileName, 
								NULL,
								NULL,
								pszAccept,
								INTERNET_FLAG_NEED_FILE|INTERNET_FLAG_NO_CACHE_WRITE ,
								0);
	if (hRequest != INVALID_HANDLE_VALUE)
	{
		hFile = CreateFile( SourceFileName,
							GENERIC_READ,
							FILE_SHARE_READ,
							NULL,
							OPEN_EXISTING,
							FILE_ATTRIBUTE_NORMAL,
							NULL);
						
		if (hFile != INVALID_HANDLE_VALUE)
		{
		

			// Clear the buffer
			if ( (pBuffer = (BYTE *)malloc (70000)) != NULL)
			{
				BufferIn.dwStructSize = sizeof( INTERNET_BUFFERS );
				BufferIn.Next = NULL; 
				BufferIn.lpcszHeader = NULL;
				BufferIn.dwHeadersLength = 0;
				BufferIn.dwHeadersTotal = 0;
				BufferIn.lpvBuffer = NULL;                
				BufferIn.dwBufferLength = 0;
				BufferIn.dwOffsetLow = 0;
				BufferIn.dwOffsetHigh = 0;
				BufferIn.dwBufferTotal = GetFileSize (hFile, NULL);
				FillMemory(pBuffer, 70000,'/0'); // Fill buffer with data
				DWORD dwFlags;
				DWORD dwBuffLen = sizeof(dwFlags); 

				if(!HttpSendRequestEx(	hRequest,
										&BufferIn,
										NULL,
										HSR_INITIATE,
										0))
				{
					LogMessage(_T("HttpSendRequestEx Failed."));
				}
				else
				{
					do
					{
						dwBytesRead = 0;
						bRet = ReadFile(hFile,
										pBuffer, 
										70000,
										&dwBytesRead,
										NULL);
						if (bRet != 0)
						{
							bRet = InternetWriteFile(hRequest,
													pBuffer,
													dwBytesRead,
													&dwBytesWritten);

							if ( (!bRet) || (dwBytesWritten==0) )
							{
								LogMessage(_T("Error Writting File: %d"),ErrorCode = GetLastError());

							}

						
						}
					} while (dwBytesRead == sizeof (pBuffer));

					CloseHandle(hFile);
					hFile = INVALID_HANDLE_VALUE;
					bRet = HttpEndRequest(	hRequest,
											NULL, 
											0, 
											0);
					if (bRet)
					{
						ResponseCode = 0;
						HttpQueryInfo(hRequest,
									HTTP_QUERY_STATUS_CODE |HTTP_QUERY_FLAG_NUMBER,
									&ResponseCode,
									&ResLength,
									&index);
					
						if ( (ResponseCode != 200) && (ResponseCode != 201))
						{
							ErrorCode=ResponseCode;
							LogMessage(_T("IIS Response Code = %d"),ResponseCode);
							// Cleanup for retry
						}						
						else
						{
							ErrorCode = 0;
							LogMessage(_T("IIS Response Code = %d"),ResponseCode);
							UploadSuccess = TRUE;

						}
					}
					else
					{
						LogMessage(_T("HttpEndrequest Returned an Error"));
						ErrorCode = GetLastError();
					}
				}
				

			}
			else
			{
				LogMessage(_T("Failed to allocate buffer memory"));
				ErrorCode = GetLastError();
			}
		}
		else
		{
			LogMessage(_T("Failed to Open Source File"));
			ErrorCode = GetLastError();
		}
		
	}
	else
	{
		LogMessage(_T("Failed to Create Put Request"));
		ErrorCode = GetLastError();
	}

	// Clean up
	if (hFile!= INVALID_HANDLE_VALUE)
		CloseHandle (hFile);
	if (hRequest)
		if (!InternetCloseHandle(hRequest))
		{
			LogMessage(_T("Failed to close the hRequest handle: error: %d"), GetLastError);
		}
	if (hConnect)
		if (!InternetCloseHandle(hConnect))
		{
			LogMessage(_T("Failed to close the hRequest handle: error: %d"), GetLastError);
		}
		

	if (hSession)
		if (!InternetCloseHandle(hSession))
		{
			LogMessage(_T("Failed to close the hRequest handle: error: %d"), GetLastError);
		}
		
	if (pUploadUrl)
		free (pUploadUrl);
	CoUninitialize();
	return ErrorCode;
	
}

DWORD GetResponseUrl(TCHAR * HostName, TCHAR *RemoteFileName,TCHAR *ResponseURL)
{
	TCHAR		IsapiUrl[255];
	TCHAR		*pUploadUrl			= NULL;
	TCHAR       *temp				= NULL;
	BOOL		bRet				= FALSE;
	DWORD		dwUrlLength			= 0;
	DWORD		ErrorCode			= 0;
	DWORD		dwLastError			= 0;
	TCHAR 		NewState;
	HINTERNET	hSession;
	HINTERNET	hRedirUrl;
	BOOL		bOnce				= FALSE;
	DWORD		dwBuffLen			= 0;
	DWORD		dwFlags				= 0;
	TCHAR       LocalUrl[255];
	ZeroMemory (IsapiUrl, sizeof IsapiUrl);
	ZeroMemory (LocalUrl, sizeof LocalUrl);	
	HINTERNET hUrlFile = NULL;


	
	StringCbPrintf (IsapiUrl,sizeof IsapiUrl,  _T("http://%s/isapi/oca_extension.dll?id=%s&Type=5"),HostName, RemoteFileName);
	LogMessage(_T("Connecting to url: %s"),IsapiUrl);
	//StringCbPrintf (IsapiUrl,sizeof IsapiUrl,  _T("http://www.microsoft.com"));

	hSession = InternetOpen(_T("Isapi Stress"),
							 INTERNET_OPEN_TYPE_PRECONFIG,
                             NULL,
							 NULL,
							 0);
	if (hSession)
	{
		// Open the url we want to connect to.
		hUrlFile = InternetOpenUrl(hSession,
								   IsapiUrl, 
								   NULL,
								   0,
								   0,
								   0);

		// Read the page returned by the isapi dll.
		TCHAR buffer[255] ;
		ZeroMemory (buffer, sizeof buffer);
		DWORD dwBytesRead = 0;
		BOOL bRead = InternetReadFile(hUrlFile,
									  buffer,
									  sizeof(buffer),
									  &dwBytesRead);

		//buffer[sizeof (buffer) -1] = _T('\0');
		StringCbCopy (ResponseURL, sizeof buffer, buffer);
	}
	InternetCloseHandle(hUrlFile);
	InternetCloseHandle(hSession);

	return 0;
}



int __cdecl _tmain( int argc, TCHAR *argv[])
{
	
	TCHAR SourceFileName[MAX_PATH];
	TCHAR SourcePath[MAX_PATH];
	TCHAR HostName[MAX_PATH];
	TCHAR VirtualDir[MAX_PATH];
	TCHAR LogPath[MAX_PATH];
	TCHAR temp;
	TCHAR RemoteFileName[MAX_PATH];
	DWORD ErrorCode = 0;
	TCHAR		ResponseUrl[255];

	int i = 1;
	
	ZeroMemory (SourceFileName, sizeof SourceFileName);
	ZeroMemory (SourcePath,		sizeof SourcePath);
	ZeroMemory (HostName,		sizeof HostName);
	ZeroMemory (VirtualDir,		sizeof VirtualDir);
	ZeroMemory (LogPath,		sizeof LogPath);
	ZeroMemory (ResponseUrl,sizeof ResponseUrl);

	StringCbCopy (HostName,	  sizeof HostName,		_T("Ocatest"));
	StringCbCopy (VirtualDir, sizeof VirtualDir,	_T("OCA"));
	
	if ( (argc < 2) || (argc > 6))
	{
		Usage();
		return (1);
	}

	for (i = 1; i < argc; i++)
	{
		switch (*argv[i])
		{
		case _T('/'):
		case _T('-'): 
			{   
				temp = _toupper( *(argv[i]+1));
				switch (temp)
				{
				case _T('S'):					// IIS Sever- Override Default
					if(*(argv[i]+2) == _T(':'))
					{	
						StringCbCopy(HostName, sizeof HostName, argv[i]+3);
					}
					break;
				case _T('V'):					// Virtual Directory override default
					if(*(argv[i]+2) == _T(':'))
					{	
						StringCbCopy(VirtualDir, sizeof VirtualDir, argv[i]+3);
					}
					break;
				case _T('?'):
					Usage();
					break;
				case _T('D'):
					if(*(argv[i]+2) == _T(':'))
					{			
						StringCbCopy(SourcePath, sizeof SourcePath, argv[i]+3);
					}
                    break;

				case _T('F'):						// User specified an offset in days...
					if(*(argv[i]+2) == _T(':'))
					{
						StringCbCopy(SourceFileName, sizeof SourceFileName, argv[i]+3);
					}
					break;
				case _T('L'):
					if(*(argv[i]+2) == _T(':'))
					{
						StringCbCopy(LogPath, sizeof LogPath, argv[i]+3);
					}
					break;
				default:
					LogMessage(_T("Unknown option: %s"),argv[i]);
					Usage();
					return (1);
				} // end switch
			} // end case
			break;
		default:
			LogMessage(_T("Unknown option: %s\n"),argv[i]);
			Usage();
			return (1);
			break;
		}// end switch
	}// end for


	// Ok now the fun part. 
	// Create the log file if one was specified.
	if (LogPath[0] != _T('\0'))
	{
		hLogFile = CreateFile(LogPath,
							  GENERIC_WRITE,
							  FILE_SHARE_READ,
							  NULL,
							  OPEN_ALWAYS,
							  FILE_ATTRIBUTE_NORMAL,
	    					  NULL);
		if (hLogFile == INVALID_HANDLE_VALUE)
		{
			LogMessage(_T("Failed to open log file: %s, Logging disabled."), LogPath);
		}
	}




	// if we have a file name we just want to upload it.

	if (SourceFileName[0] != _T('\0'))
	{
		LogMessage(_T("Uploading file: %s"), SourceFileName);
		ErrorCode = Upload(SourceFileName, VirtualDir, HostName,RemoteFileName);
		if(ErrorCode != 0)
		{
			LogMessage(_T("Upload Failed: ErrorCode = %d"),ErrorCode);
		}
		else
		{
			LogMessage(_T("Upload Succeeded. "));
		}
		
		ErrorCode = GetResponseUrl(HostName,PathFindFileName(RemoteFileName),ResponseUrl);
		if (ErrorCode != 0)
		{
			LogMessage(_T("GetResponseUrl Failed ErrorCode=%d"),ErrorCode);
		}
		LogMessage(_T("ResponseUrl = %s"),ResponseUrl);
		
	}
	else
	{
		if (SourcePath[0] != _T('\0'))
		{
			// if we have a directory we want to walk the directory and submit all the cabs.
			;
		}
		else
		{
			// If we don't have either we just exit with usage
			Usage();
			if (hLogFile != INVALID_HANDLE_VALUE)
				CloseHandle (hLogFile);
			return (1);
		}
	}
	if (hLogFile != INVALID_HANDLE_VALUE)
		CloseHandle (hLogFile);


}