volpropsheet.cpp File Reference

#include "shellext.h"
#include <windows.h>
#include <strsafe.h>
#include <winternl.h>
#include <shlwapi.h>
#include <uxtheme.h>
#include "volpropsheet.h"
#include "resource.h"
#include "mountmgr.h"

Include dependency graph for volpropsheet.cpp:

Go to the source code of this file.

Classes
struct	dev

Macros
#define	ISOLATION_AWARE_ENABLED   1
#define	STRSAFE_NO_DEPRECATE
#define	NO_SHLWAPI_STRFCNS

Functions
static INT_PTR CALLBACK	stub_UsageDlgProc (HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
static void	add_lv_column (HWND list, int string, int cx)
static int CALLBACK	lv_sort (LPARAM lParam1, LPARAM lParam2, LPARAM lParamSort)
static uint64_t	find_dev_alloc (uint64_t dev_id, btrfs_usage *usage)
static INT_PTR CALLBACK	stub_StatsDlgProc (HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
static INT_PTR CALLBACK	stub_DeviceDlgProc (HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
static INT_PTR CALLBACK	PropSheetDlgProc (HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
static INT_PTR __stdcall	dlg_proc (HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
void CALLBACK	ResetStatsW (HWND hwnd, HINSTANCE hinst, LPWSTR lpszCmdLine, int nCmdShow)
void CALLBACK	ShowChangeDriveLetterW (HWND hwnd, HINSTANCE hinst, LPWSTR lpszCmdLine, int nCmdShow)

Variables
static const NTSTATUS	STATUS_OBJECT_NAME_NOT_FOUND = 0xC0000034

Macro Definition Documentation

ISOLATION_AWARE_ENABLED

#define ISOLATION_AWARE_ENABLED   1

Definition at line 18 of file volpropsheet.cpp.

NO_SHLWAPI_STRFCNS

#define NO_SHLWAPI_STRFCNS

Definition at line 35 of file volpropsheet.cpp.

STRSAFE_NO_DEPRECATE

#define STRSAFE_NO_DEPRECATE

Definition at line 19 of file volpropsheet.cpp.

Function Documentation

add_lv_column()

static void add_lv_column ( HWND  list, int  string, int  cx  )

static

Definition at line 586 of file volpropsheet.cpp.

                                                         {
    LVCOLUMNW lvc;
    wstring s;
 
    if (!load_string(module, string, s))
        throw last_error(GetLastError());
 
    lvc.mask = LVCF_TEXT|LVCF_WIDTH;
    lvc.pszText = (WCHAR*)s.c_str();
    lvc.cx = cx;
    SendMessageW(list, LVM_INSERTCOLUMNW, 0, (LPARAM)&lvc);
}

Referenced by BtrfsVolPropSheet::DeviceDlgProc().

dlg_proc()

static INT_PTR __stdcall dlg_proc ( HWND  hwndDlg, UINT  uMsg, WPARAM  wParam, LPARAM  lParam  )

static

Definition at line 1593 of file volpropsheet.cpp.

                                                                                         {
    BtrfsChangeDriveLetter* bcdl;
 
    if (uMsg == WM_INITDIALOG) {
        SetWindowLongPtrW(hwndDlg, GWLP_USERDATA, (LONG_PTR)lParam);
        bcdl = (BtrfsChangeDriveLetter*)lParam;
    } else
        bcdl = (BtrfsChangeDriveLetter*)GetWindowLongPtrW(hwndDlg, GWLP_USERDATA);
 
    return bcdl->DlgProc(hwndDlg, uMsg, wParam, lParam);
}

Referenced by BtrfsChangeDriveLetter::show().

find_dev_alloc()

static uint64_t find_dev_alloc ( uint64_t  dev_id, btrfs_usage *  usage  )

static

Definition at line 608 of file volpropsheet.cpp.

                                                                    {
    btrfs_usage* bue;
    uint64_t alloc;
 
    alloc = 0;
 
    bue = usage;
    while (true) {
        uint64_t k;
 
        for (k = 0; k < bue->num_devices; k++) {
            if (bue->devices[k].dev_id == dev_id)
                alloc += bue->devices[k].alloc;
        }
 
        if (bue->next_entry > 0)
            bue = (btrfs_usage*)((uint8_t*)bue + bue->next_entry);
        else
            break;
    }
 
    return alloc;
}

Referenced by BtrfsVolPropSheet::RefreshDevList().

lv_sort()

static int CALLBACK lv_sort ( LPARAM  lParam1, LPARAM  lParam2, LPARAM  lParamSort  )

static

Definition at line 599 of file volpropsheet.cpp.

                                                                               {
    if (lParam1 < lParam2)
        return -1;
    else if (lParam1 > lParam2)
        return 1;
    else
        return 0;
}

Referenced by BtrfsVolPropSheet::RefreshDevList().

PropSheetDlgProc()

static INT_PTR CALLBACK PropSheetDlgProc ( HWND  hwndDlg, UINT  uMsg, WPARAM  wParam, LPARAM  lParam  )

static

Definition at line 1329 of file volpropsheet.cpp.

                                                                                                {
    try {
        switch (uMsg) {
            case WM_INITDIALOG:
            {
                PROPSHEETPAGEW* psp = (PROPSHEETPAGEW*)lParam;
                BtrfsVolPropSheet* bps = (BtrfsVolPropSheet*)psp->lParam;
                btrfs_device* bd;
 
                EnableThemeDialogTexture(hwndDlg, ETDT_ENABLETAB);
 
                SetWindowLongPtrW(hwndDlg, GWLP_USERDATA, (LONG_PTR)bps);
 
                bps->readonly = true;
                bd = bps->devices;
 
                while (true) {
                    if (!bd->readonly) {
                        bps->readonly = false;
                        break;
                    }
 
                    if (bd->next_entry > 0)
                        bd = (btrfs_device*)((uint8_t*)bd + bd->next_entry);
                    else
                        break;
                }
 
                if (bps->uuid_set) {
                    WCHAR s[255];
                    wstring t;
 
                    GetDlgItemTextW(hwndDlg, IDC_UUID, s, sizeof(s) / sizeof(WCHAR));
 
                    wstring_sprintf(t, s, bps->uuid.uuid[0], bps->uuid.uuid[1], bps->uuid.uuid[2], bps->uuid.uuid[3], bps->uuid.uuid[4], bps->uuid.uuid[5],
                                    bps->uuid.uuid[6], bps->uuid.uuid[7], bps->uuid.uuid[8], bps->uuid.uuid[9], bps->uuid.uuid[10], bps->uuid.uuid[11],
                                    bps->uuid.uuid[12], bps->uuid.uuid[13], bps->uuid.uuid[14], bps->uuid.uuid[15]);
 
                    SetDlgItemTextW(hwndDlg, IDC_UUID, t.c_str());
                } else
                    SetDlgItemTextW(hwndDlg, IDC_UUID, L"");
 
                SendMessageW(GetDlgItem(hwndDlg, IDC_VOL_SCRUB), BCM_SETSHIELD, 0, true);
                SendMessageW(GetDlgItem(hwndDlg, IDC_VOL_CHANGE_DRIVE_LETTER), BCM_SETSHIELD, 0, true);
 
                return false;
            }
 
            case WM_NOTIFY:
            {
                switch (((LPNMHDR)lParam)->code) {
                    case PSN_KILLACTIVE:
                        SetWindowLongPtrW(hwndDlg, DWLP_MSGRESULT, false);
                    break;
                }
                break;
            }
 
            case WM_COMMAND:
            {
                BtrfsVolPropSheet* bps = (BtrfsVolPropSheet*)GetWindowLongPtrW(hwndDlg, GWLP_USERDATA);
 
                if (bps) {
                    switch (HIWORD(wParam)) {
                        case BN_CLICKED: {
                            switch (LOWORD(wParam)) {
                                case IDC_VOL_SHOW_USAGE:
                                    bps->ShowUsage(hwndDlg);
                                break;
 
                                case IDC_VOL_BALANCE:
                                    bps->balance->ShowBalance(hwndDlg);
                                break;
 
                                case IDC_VOL_DEVICES:
                                    bps->ShowDevices(hwndDlg);
                                break;
 
                                case IDC_VOL_SCRUB:
                                    bps->ShowScrub(hwndDlg);
                                break;
 
                                case IDC_VOL_CHANGE_DRIVE_LETTER:
                                    bps->ShowChangeDriveLetter(hwndDlg);
                                break;
                            }
                        }
                    }
                }
 
                break;
            }
        }
    } catch (const exception& e) {
        error_message(hwndDlg, e.what());
    }
 
    return false;
}

ResetStatsW()

void CALLBACK ResetStatsW	(	HWND	hwnd,
		HINSTANCE	hinst,
		LPWSTR	lpszCmdLine,
		int	nCmdShow
	)

Definition at line 1613 of file volpropsheet.cpp.

                                                                                        {
    try {
        win_handle token;
        NTSTATUS Status;
        TOKEN_PRIVILEGES tp;
        LUID luid;
        uint64_t devid;
        wstring cmdline, vol, dev;
        size_t pipe;
        IO_STATUS_BLOCK iosb;
 
        set_dpi_aware();
 
        cmdline = lpszCmdLine;
 
        if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &token))
            throw last_error(GetLastError());
 
        if (!LookupPrivilegeValueW(nullptr, L"SeManageVolumePrivilege", &luid))
            throw last_error(GetLastError());
 
        tp.PrivilegeCount = 1;
        tp.Privileges[0].Luid = luid;
        tp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
 
        if (!AdjustTokenPrivileges(token, false, &tp, sizeof(TOKEN_PRIVILEGES), nullptr, nullptr))
            throw last_error(GetLastError());
 
        pipe = cmdline.find(L"|");
 
        if (pipe == string::npos)
            return;
 
        vol = cmdline.substr(0, pipe);
        dev = cmdline.substr(pipe + 1);
 
        devid = _wtoi(dev.c_str());
        if (devid == 0)
            return;
 
        win_handle h = CreateFileW(vol.c_str(), FILE_TRAVERSE | FILE_READ_ATTRIBUTES, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, nullptr,
                                OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS | FILE_FLAG_OPEN_REPARSE_POINT, nullptr);
 
        if (h == INVALID_HANDLE_VALUE)
            throw last_error(GetLastError());
 
        Status = NtFsControlFile(h, nullptr, nullptr, nullptr, &iosb, FSCTL_BTRFS_RESET_STATS, &devid, sizeof(uint64_t), nullptr, 0);
        if (!NT_SUCCESS(Status))
            throw ntstatus_error(Status);
    } catch (const exception& e) {
        error_message(hwnd, e.what());
    }
}

ShowChangeDriveLetterW()

void CALLBACK ShowChangeDriveLetterW	(	HWND	hwnd,
		HINSTANCE	hinst,
		LPWSTR	lpszCmdLine,
		int	nCmdShow
	)

Definition at line 1667 of file volpropsheet.cpp.

                                                                                                   {
    BtrfsChangeDriveLetter bcdl(hwnd, lpszCmdLine);
 
    bcdl.show();
}

stub_DeviceDlgProc()

static INT_PTR CALLBACK stub_DeviceDlgProc ( HWND  hwndDlg, UINT  uMsg, WPARAM  wParam, LPARAM  lParam  )

static

Definition at line 1249 of file volpropsheet.cpp.

                                                                                                  {
    BtrfsVolPropSheet* bvps;
 
    if (uMsg == WM_INITDIALOG) {
        SetWindowLongPtrW(hwndDlg, GWLP_USERDATA, (LONG_PTR)lParam);
        bvps = (BtrfsVolPropSheet*)lParam;
    } else {
        bvps = (BtrfsVolPropSheet*)GetWindowLongPtrW(hwndDlg, GWLP_USERDATA);
    }
 
    if (bvps)
        return bvps->DeviceDlgProc(hwndDlg, uMsg, wParam, lParam);
    else
        return false;
}

Referenced by BtrfsVolPropSheet::ShowDevices().

stub_StatsDlgProc()

static INT_PTR CALLBACK stub_StatsDlgProc ( HWND  hwndDlg, UINT  uMsg, WPARAM  wParam, LPARAM  lParam  )

static

Definition at line 942 of file volpropsheet.cpp.

                                                                                                 {
    BtrfsVolPropSheet* bvps;
 
    if (uMsg == WM_INITDIALOG) {
        SetWindowLongPtrW(hwndDlg, GWLP_USERDATA, (LONG_PTR)lParam);
        bvps = (BtrfsVolPropSheet*)lParam;
    } else {
        bvps = (BtrfsVolPropSheet*)GetWindowLongPtrW(hwndDlg, GWLP_USERDATA);
    }
 
    if (bvps)
        return bvps->StatsDlgProc(hwndDlg, uMsg, wParam, lParam);
    else
        return false;
}

Referenced by BtrfsVolPropSheet::ShowStats().

stub_UsageDlgProc()

static INT_PTR CALLBACK stub_UsageDlgProc ( HWND  hwndDlg, UINT  uMsg, WPARAM  wParam, LPARAM  lParam  )

static

Definition at line 566 of file volpropsheet.cpp.

                                                                                                 {
    BtrfsVolPropSheet* bvps;
 
    if (uMsg == WM_INITDIALOG) {
        SetWindowLongPtrW(hwndDlg, GWLP_USERDATA, (LONG_PTR)lParam);
        bvps = (BtrfsVolPropSheet*)lParam;
    } else {
        bvps = (BtrfsVolPropSheet*)GetWindowLongPtrW(hwndDlg, GWLP_USERDATA);
    }
 
    if (bvps)
        return bvps->UsageDlgProc(hwndDlg, uMsg, wParam, lParam);
    else
        return false;
}

Referenced by BtrfsVolPropSheet::ShowUsage().

Variable Documentation

STATUS_OBJECT_NAME_NOT_FOUND

const NTSTATUS STATUS_OBJECT_NAME_NOT_FOUND = 0xC0000034

static

Definition at line 48 of file volpropsheet.cpp.