// Copyright (c) 2001-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:
// e32test\usb\t_usb_win\src\eject.cpp
// eject functions
//

#include "stdafx.h"
#include <windows.h>
#include <winioctl.h>
#include <tchar.h>
#include <stdio.h>
#include "global.h"

// Prototypes
extern void PrintOut(BOOL screenFlag, BOOL logFlag, BOOL timeFlag, const char *format, ...);

HANDLE OpenVolume(char driveLetter)
{
    HANDLE hVolume;
    UINT uDriveType;
    TCHAR szVolumeName[8];
    TCHAR szRootName[5];
 	TCHAR cDriveLetter = driveLetter;
	
	if (driveLetter == ' ')
	{
		return INVALID_HANDLE_VALUE;
	}

    wsprintf(szRootName, "%c:\\", cDriveLetter);

    uDriveType = GetDriveType(szRootName);
    if (uDriveType != DRIVE_REMOVABLE)
	{
        return INVALID_HANDLE_VALUE;
    }

    wsprintf(szVolumeName, "\\\\.\\%c:", cDriveLetter);

    hVolume = CreateFile(   szVolumeName,
                            GENERIC_READ | GENERIC_WRITE,
                            FILE_SHARE_READ | FILE_SHARE_WRITE,
                            NULL,
                            OPEN_EXISTING,
                            0,
                            NULL );
    
    return hVolume;
}

#define LOCK_RETRY_DELAY	500       // half a second
#define LOCK_RETRY_MAX		20

BOOL LockAndDismout(HANDLE hVolume)
{
   DWORD dwBytesReturned;

   // Retry a number of times
   for (int i = 0; i < LOCK_RETRY_MAX; i++)
   {
       if (DeviceIoControl(hVolume, FSCTL_LOCK_VOLUME, NULL, 0, NULL, 0, &dwBytesReturned, NULL))
			{
			return DeviceIoControl( hVolume, FSCTL_DISMOUNT_VOLUME, NULL, 0, NULL, 0, &dwBytesReturned, NULL);
			}
       Sleep (LOCK_RETRY_MAX);
   }

   return FALSE;
}

BOOL AllowRemovalAndEject (HANDLE hVolume)
{
    DWORD dwBytesReturned;
    PREVENT_MEDIA_REMOVAL PMRBuffer;

    PMRBuffer.PreventMediaRemoval = FALSE;

    if (DeviceIoControl( hVolume, IOCTL_STORAGE_MEDIA_REMOVAL, &PMRBuffer, sizeof(PREVENT_MEDIA_REMOVAL),
                            NULL, 0, &dwBytesReturned, NULL))
	    return DeviceIoControl( hVolume, IOCTL_STORAGE_EJECT_MEDIA, NULL, 0, NULL, 0, &dwBytesReturned, NULL);

	return FALSE;   
}


BOOL EjectVolume(char driveLetter)
{
    HANDLE hVolume = INVALID_HANDLE_VALUE;


	hVolume = OpenVolume(driveLetter);

	if (hVolume == INVALID_HANDLE_VALUE)
		return FALSE;

    // Lock and dismount the volume.
    if (LockAndDismout(hVolume))
		{
        // Set prevent removal to false and eject the volume.
        AllowRemovalAndEject(hVolume);
		}

    // Close the volume so other processes can use the drive.
    if (!CloseHandle(hVolume))
        return FALSE;

    return TRUE;
}

#define WAIT_RETRIES 500
#define WAIT_TIME 250
#define WAIT_DELAY 1500

BOOL WaitDriveReady (char driveLetter)
{
    UINT uDriveType = DRIVE_UNKNOWN ;
    TCHAR szRootName[5];
	UINT waitTime = 0;

    wsprintf(szRootName, "%c:\\", driveLetter);

	for (int i = 0; i < WAIT_RETRIES && (uDriveType != DRIVE_REMOVABLE); i++)
		{
		uDriveType = GetDriveType(szRootName);
		if (uDriveType != DRIVE_REMOVABLE)
			{
			Sleep (WAIT_TIME);
			waitTime += WAIT_TIME;
			}
		else
			{
			Sleep (WAIT_DELAY);
			waitTime += WAIT_DELAY;
			}
		}
	

    PRINT_ALWAYS "Drive %c ready after %d milliseconds.\n",driveLetter,waitTime);
    return uDriveType == DRIVE_REMOVABLE;
}