Skip to main content

Locking a physical disk on windows....

In this, I've tried to lock a physical disk using a small Windows C++ program. The steps to follow to achieve this are:


1. Get the Physical Drive and volume mapping. Say, the computer is attached with three physical drives, and we're interested to lock the Physical Drive 1 ("\\\\.\\PhysicalDrive1"). Then we need to figure out how many volumes are there on that physical disk.

2. Then Lock that volume one by one using control code FSCTL_LOCK_VOLUME.

3. Do the stuff we'd like to perform on the disk and then unlock each volume using control code  FSCTL_UNLOCK_VOLUME.

4. Close disk and volume handle(s).

Few things to remember here. As per Microsoft documentation,

a. The NTFS file system treats a locked volume as a dismounted volume.
b. Lock volume call will fail with Access Code 5 (Access Denied) if the volume is in use. If we're not sure who's using the volume, just unmount it once.
c. The FSCTL_DISMOUNT_VOLUME control code functions similarly but does not check for open files before dismounting.
d. The operating system tries to mount an unmounted disk as soon as an attempt is made to access it. So, a call to GetLogicalDrives API will trigger the operating system to mount the disk.

Note: Since this deals with disk locks, please use it wisely.

Here is the code snippet, I tried to lock the drive here which has only one volume (J:).

void LockDisk()
{
    HANDLE    hDisk_Device = NULL;
    HANDLE    hDisk_Drive = NULL;
    DWORD    returnedLength;

    // Open the drive(s) associated with this disk and lock.
    hDisk_Drive = CreateFile(
            L"\\\\.\\J:",                                                     // device interface name
            GENERIC_READ | GENERIC_WRITE,            // dwDesiredAccess
            FILE_SHARE_READ | FILE_SHARE_WRITE, // dwShareMode
            NULL,                                                          // lpSecurityAttributes
            OPEN_EXISTING,                                        // dwCreationDistribution
            0,                                                                // dwFlagsAndAttributes
            NULL                                                         // hTemplateFile
            );

    if(hDisk_Drive != INVALID_HANDLE_VALUE)
    {
        BOOL status;
        status = DeviceIoControl( hDisk_Drive, FSCTL_LOCK_VOLUME, NULL, 0, NULL, 0, &returnedLength, NULL );

        if ( !status )
        {
            wprintf(L"IOCTL failed with error code: %d.\n\n", GetLastError() );
        }
        else
        {
            wprintf(L"IOCTL success, volume locked\n\n");
           
            status = DeviceIoControl( hDisk_Drive, FSCTL_UNLOCK_VOLUME, NULL, 0, NULL, 0, &returnedLength, NULL );
            if( !status)
                wprintf(L"IOCTL failed with error code: %d.\n\n", GetLastError() );

            CloseHandle(hDisk_Drive);
        }
    }
}

We can test this piece of code by using debug breakpoint to see if it really works. I've used Windows 7 (x64).

On successful lock, if we try to access the drive from Windows Explorer, we'll get the following error:



Labels:

Comments

Post a Comment

Popular posts from this blog

Reversing char array without splitting the array to tokens

 I was reading about strdup, a C++ function and suddenly an idea came to my mind if this can be leveraged to aid in reversing a character array without splitting the array into words and reconstructing it again by placing spaces and removing trailing spaces. Again, I wanted an array to be passed as a function argument and an array size to be passed implicitly with the array to the function. Assumed, a well-formed char array has been passed into the function. No malformed array checking is done inside the function. So, the function signature and definition are like below: Below is the call from the client code to reverse the array without splitting tokens and reconstructing it. Finally, copy the reversed array to the destination.  For GNU C++, we should use strdup instead _strdup . On run, we get the following output: Demo code
Post a Comment
Read more

XOR (Exclusive OR) for branchless coding

The following example shows the array reversing using the  XOR operator . No need to take any additional variable to reverse the array.   int main(int argc, _TCHAR* argv[]) { char str[] = "I AM STUDENT"; int length = strlen(str); for(int i = 0; i < ((length/2)); i++) { str[i] ^= str[length - (1+i)]; str[length - (1+i)] ^= str[i]; str[i] ^= str[length - (1+i)]; } cout << str << endl; return 0; } The above example is one of the uses of XOR but XOR comes in handy when we can do branchless coding  methods like butterfly switch etc. Sometimes this is very effective in speeding up the execution.  Let's see one of the uses of XOR in branchless coding. I am taking a simple example of Y = | X |.  Yes, I am generating abs of a supplied number. So, my function signature/definition in C++ looks like below: int absoluteBranch( int x) {     if (x < 0 ) {         return -x;     }     else {         retur
1 comment
Read more

Power of Two

  I n this post will be discussing how to calculate if a number is a power of two or not. As an example, 8 is a power of two but the number 10 is not. There are many ways we can solve this. First , we will take an approach which is simple and iterative. In this case, we will calculate the power of two one by one and check with the supplied number. The below code illustrates it. bool isPowerofTwo(unsigned num) { auto y = 1; while (0 != y) { if (num == y) return true; if (num < y) return false; y <<= 1; } return false; } Second , assuming, the number is a 32-bit number, this is also an iterative solution. In this scenario, iterating all bits and counting the set bits. Any number which is a power of 2 will have only one bit set and the rest will be zeros. As an example, 8 in binary representation is 1000. Using this observation, we can implement an iterative solution. bool isPowerofTwo(unsigned num) { auto one_count = 0; for (auto index = 0; index < 32;
Post a Comment
Read more