mirror of
https://github.com/uroni/urbackup_backend.git
synced 2025-10-26 11:36:50 +00:00
101 lines
2.7 KiB
C++
101 lines
2.7 KiB
C++
#include "ntfs_win.h"
|
|
#include <Windows.h>
|
|
#include "../../Interface/Server.h"
|
|
#include "../../stringtools.h"
|
|
|
|
FSNTFSWIN::FSNTFSWIN(const std::wstring &pDev)
|
|
: Filesystem(pDev), bitmap(NULL)
|
|
{
|
|
HANDLE hDev=CreateFileW( pDev.c_str(), GENERIC_READ, FILE_SHARE_READ|FILE_SHARE_WRITE, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL );
|
|
if(hDev==INVALID_HANDLE_VALUE)
|
|
{
|
|
Server->Log("Error opening device -2", LL_ERROR);
|
|
has_error=true;
|
|
return;
|
|
}
|
|
|
|
DWORD sectors_per_cluster;
|
|
DWORD bytes_per_sector;
|
|
DWORD NumberOfFreeClusters;
|
|
DWORD TotalNumberOfClusters;
|
|
BOOL b=GetDiskFreeSpaceW((pDev+L"\\").c_str(), §ors_per_cluster, &bytes_per_sector, &NumberOfFreeClusters, &TotalNumberOfClusters);
|
|
if(!b)
|
|
{
|
|
Server->Log("Error in GetDiskFreeSpaceW", LL_ERROR);
|
|
has_error=true;
|
|
CloseHandle(hDev);
|
|
return;
|
|
}
|
|
|
|
DWORD r_bytes;
|
|
NTFS_VOLUME_DATA_BUFFER vdb;
|
|
b=DeviceIoControl(hDev, FSCTL_GET_NTFS_VOLUME_DATA, NULL, 0, &vdb, sizeof(NTFS_VOLUME_DATA_BUFFER), &r_bytes, NULL);
|
|
if(!b)
|
|
{
|
|
Server->Log("Error in DeviceIoControl(FSCTL_GET_NTFS_VOLUME_DATA)", LL_ERROR);
|
|
has_error=true;
|
|
CloseHandle(hDev);
|
|
return;
|
|
}
|
|
|
|
sectorsize=bytes_per_sector;
|
|
clustersize=sectorsize*sectors_per_cluster;
|
|
drivesize=vdb.NumberSectors.QuadPart*sectorsize;
|
|
|
|
uint64 numberOfClusters=drivesize/clustersize;
|
|
|
|
STARTING_LCN_INPUT_BUFFER start;
|
|
LARGE_INTEGER li;
|
|
li.QuadPart=0;
|
|
start.StartingLcn=li;
|
|
|
|
size_t new_size=(size_t)(numberOfClusters/8);
|
|
if(numberOfClusters%8!=0)
|
|
++new_size;
|
|
|
|
size_t n_clusters=new_size;
|
|
|
|
new_size+=sizeof(VOLUME_BITMAP_BUFFER);
|
|
|
|
VOLUME_BITMAP_BUFFER *vbb=(VOLUME_BITMAP_BUFFER*)(new char[new_size]);
|
|
|
|
|
|
b=DeviceIoControl(hDev, FSCTL_GET_VOLUME_BITMAP, &start, sizeof(STARTING_LCN_INPUT_BUFFER), vbb, (DWORD)new_size, &r_bytes, NULL);
|
|
if(!b)
|
|
{
|
|
Server->Log("DeviceIoControl(FSCTL_GET_VOLUME_BITMAP) failed", LL_ERROR);
|
|
has_error=true;
|
|
CloseHandle(hDev);
|
|
return;
|
|
}
|
|
|
|
Server->Log("TotalNumberOfClusters="+nconvert((size_t)TotalNumberOfClusters)+" numberOfClusters="+nconvert(numberOfClusters)+" n_clusters="+nconvert(n_clusters)+" StartingLcn="+nconvert(vbb->StartingLcn.QuadPart)+" BitmapSize="+nconvert(vbb->BitmapSize.QuadPart)+" r_bytes="+nconvert((size_t)r_bytes), LL_DEBUG);
|
|
|
|
bitmap=new unsigned char[(unsigned int)(n_clusters)];
|
|
memset(bitmap, 0xFF, n_clusters);
|
|
memcpy(bitmap+(unsigned int)(vbb->StartingLcn.QuadPart/8), vbb->Buffer, (size_t)(vbb->BitmapSize.QuadPart/8));
|
|
|
|
delete []vbb;
|
|
|
|
CloseHandle(hDev);
|
|
}
|
|
|
|
FSNTFSWIN::~FSNTFSWIN(void)
|
|
{
|
|
delete [] bitmap;
|
|
}
|
|
|
|
int64 FSNTFSWIN::getBlocksize(void)
|
|
{
|
|
return clustersize;
|
|
}
|
|
|
|
int64 FSNTFSWIN::getSize(void)
|
|
{
|
|
return drivesize;
|
|
}
|
|
|
|
const unsigned char * FSNTFSWIN::getBitmap(void)
|
|
{
|
|
return bitmap;
|
|
} |