/*

 * Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).

 * All rights reserved.

 * This component and the accompanying materials are made available

 * under the terms of the License "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:

 *

 */

 //

 // T_BIGFILE_WRITER.CPP

 //

 #include <windows.h>

 #pragma warning (disable:4201) // warning C4201: nonstandard extension used : nameless struct/union

 #include <winioctl.h>

 #include <stdio.h>

 #define __WRITE_FILE_CONTENTS__

 const DWORD K1Kb = 1 << 10;

 const DWORD K1Mb = 1 << 20;

 const DWORD K1Gb = 1 << 30;

 const DWORD K2Gb = 0x80000000;

 const DWORD K3Gb = 0xC0000000;

 const DWORD K4GbMinusOne = 0xFFFFFFFF;

 const DWORD KBufSize = 256 * K1Kb;

 BYTE gBuffer[KBufSize];

 template <class T>

 T Min(T aLeft,T aRight)

 	{return(aLeft<aRight ? aLeft : aRight);}

 BOOL WriteFile(char* aFileName, DWORD aSize)

 	{

 	BOOL success = TRUE;

 	HANDLE deviceHandle = CreateFileA(

 		aFileName,

         GENERIC_WRITE, FILE_SHARE_WRITE, 

         NULL, CREATE_ALWAYS, 0, NULL); 

     if (deviceHandle == INVALID_HANDLE_VALUE) 

 		{

 		printf("Open file error %d", GetLastError());

 		return FALSE;

 		}

 #ifdef __WRITE_FILE_CONTENTS__

 //printf("size of DWORD is %d", sizeof(DWORD));

 	DWORD nNumberOfBytesToWrite;

 	for (DWORD pos = 0; pos < aSize; pos+=nNumberOfBytesToWrite)

 		{

 		for (DWORD n=0; n<KBufSize; n+=4)

 			{

 			*((DWORD*) &gBuffer[n]) = pos + n;

 			}

 		nNumberOfBytesToWrite = Min(KBufSize, aSize - pos);

 		DWORD nNumberOfBytesWritten;

 		success = WriteFile (deviceHandle, gBuffer, nNumberOfBytesToWrite, &nNumberOfBytesWritten, NULL);

 		if (!success)

 			{

 			printf("Write file error %d", GetLastError());

 			break;

 			}

 		printf("\rWriting %s %lu%% done...", aFileName, (LONG64(pos) + LONG64(nNumberOfBytesToWrite)) * LONG64(100) / LONG64(aSize));

 		}

 	printf("\n");

 #else

 	printf("Setting file size for %s to %u\n", aFileName, aSize);

 	LONG dwSeekLo = (LONG) aSize;

 	LONG dwSeekHi = 0;

 	DWORD dwPos = SetFilePointer(deviceHandle, dwSeekLo, &dwSeekHi, FILE_BEGIN);

 	if (dwPos  != aSize)

 		{

 		printf("SetFilePointer() error %d", GetLastError());

 		success = FALSE;

 		}

 	else

 		{

 		success = SetEndOfFile(deviceHandle);

 		if (!success)

 			{

 			printf("SetFilePointer() error %d", GetLastError());

 			}

 		}

 #endif

 	CloseHandle(deviceHandle);

 	return success;

 	}

 int main(int argc,char *argv[])

 	{

 	printf("BigFileWriter...\n");

 	if (argc != 2)

 		{

 		printf("Creates big (between 2 & 4GB) files on a Windows drive for use by T_BIGFILE.EXE\n");

 		printf("Syntax : BigFileWriter <drive letter>\n");

 		exit(0);

 		}

 	char filePath[] = "\\\\.\\?:\\F32-TST";

 	char driveLetter = argv[1][0];

 	filePath[4] = driveLetter;

 	printf("Creating big files on %s\n", filePath);

 	BOOL success;

 	success =  CreateDirectory(filePath, NULL);

 	if (!success && GetLastError() != ERROR_ALREADY_EXISTS)

 		{

 		printf("Unable to create directory %d", GetLastError());

 		exit(4);

 		}

 	success =  SetCurrentDirectory(filePath);

 	if (!success)

 		{

 		printf("Unable to change to directory %d", GetLastError());

 		exit(4);

 		}

 	success = WriteFile("File2GBMinusOne.txt", K2Gb-1);

 	if (!success) exit(4);

 	success = WriteFile("File2GB.txt", K2Gb);

 	if (!success) exit(4);

 	success = WriteFile("File3GB.txt", K3Gb);

 	if (!success) exit(4);

 	// NB This won't fit on an 8GB drive

 	success = WriteFile("File4GBMinusOne.txt", K4GbMinusOne);

 	if (!success) 

 		{

 		DeleteFile("File4GBMinusOne.txt");

 		exit(4);

 		}

 	return 0;

 	}