Click here to Skip to main content
15,886,071 members
Articles / Programming Languages / C

How to develop a virtual disk for Windows

,
Rate me:
Please Sign up or sign in to vote.
4.99/5 (98 votes)
15 Feb 2010CPOL10 min read 195.4K   8.2K   335  
This article is oriented to Windows-developers and considers how to create virtual disk in Windows system.
#include "Mountmanager.h"

#include "new"
#include "exception"
#include "memory"
#include "IrpHandler.h"

MountManager::MountManager(PDRIVER_OBJECT driverObject):
        driverObject_(driverObject), 
        diskCount_(0)
{
    NTSTATUS          status;
    UNICODE_STRING    device_dir_name;
    OBJECT_ATTRIBUTES object_attributes;

    RtlInitUnicodeString(&device_dir_name, ROOT_DIR_NAME);

    InitializeObjectAttributes(
        &object_attributes,
        &device_dir_name,
        OBJ_OPENIF,
        NULL,
        NULL
        );

    HANDLE dir_handle;
    status = ZwCreateDirectoryObject(
        &dir_handle,
        DIRECTORY_ALL_ACCESS,
        &object_attributes
        );

    if (!NT_SUCCESS(status)) 
        ASSERT(false);
}
void MountManager::Unmount(int devId)
{
    AutoMutex guard(diskMapLock_);
    diskMap_.erase(devId);
}
NTSTATUS MountManager::DispatchIrp(int devId, PIRP irp)
{
    boost::shared_ptr<MountedDisk> disk;
    {
        AutoMutex guard(diskMapLock_);
        MountedDiskMapIter i = diskMap_.find(devId);
        if(i == diskMap_.end())
        {
            irp->IoStatus.Status = STATUS_DEVICE_NOT_READY;
            IoCompleteRequest( irp, IO_NO_INCREMENT);
            return STATUS_DEVICE_NOT_READY;
        }
        disk = i->second;
    }
    return disk->DispatchIrp(irp);
}
int MountManager::Mount(UINT64 totalLength)
{
    // generate id
    int devId = 0;
    { 
        AutoMutex guard(diskMapLock_);
        do
        {
            devId = diskCount_++;
        }
        while(diskMap_.find(devId) != diskMap_.end());
    }
    boost::shared_ptr<MountedDisk> disk(
        new MountedDisk(driverObject_, this, devId, totalLength));
    AutoMutex guard(diskMapLock_);
    MountedDiskMapPairIB pairIb = 
        diskMap_.insert(std::make_pair(devId, disk));
    if(!pairIb.second)
        throw std::exception(__FUNCTION__" - device ID already exist.");
    return devId;
}
void MountManager::RequestExchange(UINT32 devId, 
                                   UINT32 lastType, 
                                   UINT32 lastStatus, 
                                   UINT32 lastSize, 
                                   char* buf, 
                                   UINT32 bufSize,
                                   UINT32 * type, 
                                   UINT32 * length, 
                                   UINT64 * offset)
{
    boost::shared_ptr<MountedDisk> disk;
    {
        AutoMutex guard(diskMapLock_);
        MountedDiskMapIter i = diskMap_.find(devId);
        if(i == diskMap_.end())
            throw std::exception("MountManager::RequestExchange - device NOT found.");
        disk = i->second;
    }
    disk->RequestExchange(lastType, lastStatus, lastSize, buf, bufSize,
                          type, length, offset);
}

By viewing downloads associated with this article you agree to the Terms of Service and the article's licence.

If a file you wish to view isn't highlighted, and is a text file (not binary), please let us know and we'll add colourisation support for it.

License

This article, along with any associated source code and files, is licensed under The Code Project Open License (CPOL)


Written By
Chief Technology Officer Apriorit Inc.
United States United States
ApriorIT is a software research and development company specializing in cybersecurity and data management technology engineering. We work for a broad range of clients from Fortune 500 technology leaders to small innovative startups building unique solutions.

As Apriorit offers integrated research&development services for the software projects in such areas as endpoint security, network security, data security, embedded Systems, and virtualization, we have strong kernel and driver development skills, huge system programming expertise, and are reals fans of research projects.

Our specialty is reverse engineering, we apply it for security testing and security-related projects.

A separate department of Apriorit works on large-scale business SaaS solutions, handling tasks from business analysis, data architecture design, and web development to performance optimization and DevOps.

Official site: https://www.apriorit.com
Clutch profile: https://clutch.co/profile/apriorit
This is a Organisation

33 members

Written By
Team Leader ApriorIT
Ukraine Ukraine
This member has not yet provided a Biography. Assume it's interesting and varied, and probably something to do with programming.

Comments and Discussions