main.cpp File Reference

#include "shellext.h"
#include <windows.h>
#include <commctrl.h>
#include <strsafe.h>
#include <stddef.h>
#include <stdexcept>
#include "factory.h"
#include "resource.h"

Include dependency graph for main.cpp:

Go to the source code of this file.

Macros
#define	COM_DESCRIPTION_ICON_HANDLER   L"WinBtrfs shell extension (icon handler)"
#define	COM_DESCRIPTION_CONTEXT_MENU   L"WinBtrfs shell extension (context menu)"
#define	COM_DESCRIPTION_PROP_SHEET   L"WinBtrfs shell extension (property sheet)"
#define	COM_DESCRIPTION_VOL_PROP_SHEET   L"WinBtrfs shell extension (volume property sheet)"
#define	ICON_OVERLAY_NAME   L"WinBtrfs"

Typedefs
typedef enum _PROCESS_DPI_AWARENESS	PROCESS_DPI_AWARENESS
typedef ULONG(WINAPI *	_RtlNtStatusToDosError) (NTSTATUS Status)
typedef HRESULT(WINAPI *	_SetProcessDpiAwareness) (PROCESS_DPI_AWARENESS value)

Enumerations
enum	_PROCESS_DPI_AWARENESS { PROCESS_DPI_UNAWARE, PROCESS_SYSTEM_DPI_AWARE, PROCESS_PER_MONITOR_DPI_AWARE }

Functions
void	set_dpi_aware ()
void	format_size (uint64_t size, wstring &s, bool show_bytes)
wstring	format_message (ULONG last_error)
wstring	format_ntstatus (NTSTATUS Status)
bool	load_string (HMODULE module, UINT id, wstring &s)
void	wstring_sprintf (wstring &s, wstring fmt,...)
STDAPI	DllCanUnloadNow (void)
STDAPI	DllGetClassObject (REFCLSID rclsid, REFIID riid, LPVOID *ppv)
static void	write_reg_key (HKEY root, const wstring &keyname, const WCHAR *val, const wstring &data)
static void	register_clsid (const GUID clsid, const WCHAR *description)
static void	reg_delete_tree (HKEY hkey, const wstring &keyname)
static void	unregister_clsid (const GUID clsid)
static void	reg_icon_overlay (const GUID clsid, const wstring &name)
static void	unreg_icon_overlay (const wstring &name)
static void	reg_context_menu_handler (const GUID clsid, const wstring &filetype, const wstring &name)
static void	unreg_context_menu_handler (const wstring &filetype, const wstring &name)
static void	reg_prop_sheet_handler (const GUID clsid, const wstring &filetype, const wstring &name)
static void	unreg_prop_sheet_handler (const wstring &filetype, const wstring &name)
STDAPI	DllRegisterServer (void)
STDAPI	DllUnregisterServer (void)
STDAPI	DllInstall (BOOL bInstall, LPCWSTR pszCmdLine)
BOOL APIENTRY	DllMain (HANDLE hModule, DWORD dwReason, void *lpReserved)
static void	create_subvol (const wstring &fn)
void CALLBACK	CreateSubvolW (HWND hwnd, HINSTANCE hinst, LPWSTR lpszCmdLine, int nCmdShow)
static void	create_snapshot2 (const wstring &source, const wstring &fn)
void CALLBACK	CreateSnapshotW (HWND hwnd, HINSTANCE hinst, LPWSTR lpszCmdLine, int nCmdShow)
void	command_line_to_args (LPWSTR cmdline, vector< wstring > &args)
static string	utf16_to_utf8 (const wstring_view &utf16)
wstring	utf8_to_utf16 (const string_view &utf8)
void	error_message (HWND hwnd, const char *msg)

Variables
static const GUID	CLSID_ShellBtrfsIconHandler = { 0x2690b74f, 0xf353, 0x422d, { 0xbb, 0x12, 0x40, 0x15, 0x81, 0xee, 0xf8, 0xf0 } }
static const GUID	CLSID_ShellBtrfsContextMenu = { 0x2690b74f, 0xf353, 0x422d, { 0xbb, 0x12, 0x40, 0x15, 0x81, 0xee, 0xf8, 0xf1 } }
static const GUID	CLSID_ShellBtrfsPropSheet = { 0x2690b74f, 0xf353, 0x422d, { 0xbb, 0x12, 0x40, 0x15, 0x81, 0xee, 0xf8, 0xf2 } }
static const GUID	CLSID_ShellBtrfsVolPropSheet = { 0x2690b74f, 0xf353, 0x422d, { 0xbb, 0x12, 0x40, 0x15, 0x81, 0xee, 0xf8, 0xf3 } }
HMODULE	module
LONG	objs_loaded = 0

Macro Definition Documentation

COM_DESCRIPTION_CONTEXT_MENU

#define COM_DESCRIPTION_CONTEXT_MENU   L"WinBtrfs shell extension (context menu)"

Definition at line 33 of file main.cpp.

COM_DESCRIPTION_ICON_HANDLER

#define COM_DESCRIPTION_ICON_HANDLER   L"WinBtrfs shell extension (icon handler)"

Definition at line 32 of file main.cpp.

COM_DESCRIPTION_PROP_SHEET

#define COM_DESCRIPTION_PROP_SHEET   L"WinBtrfs shell extension (property sheet)"

Definition at line 34 of file main.cpp.

COM_DESCRIPTION_VOL_PROP_SHEET

#define COM_DESCRIPTION_VOL_PROP_SHEET   L"WinBtrfs shell extension (volume property sheet)"

Definition at line 35 of file main.cpp.

ICON_OVERLAY_NAME

#define ICON_OVERLAY_NAME   L"WinBtrfs"

Definition at line 36 of file main.cpp.

Typedef Documentation

_RtlNtStatusToDosError

typedef ULONG(WINAPI * _RtlNtStatusToDosError) (NTSTATUS Status)

Definition at line 44 of file main.cpp.

_SetProcessDpiAwareness

typedef HRESULT(WINAPI * _SetProcessDpiAwareness) (PROCESS_DPI_AWARENESS value)

Definition at line 45 of file main.cpp.

PROCESS_DPI_AWARENESS

typedef enum _PROCESS_DPI_AWARENESS PROCESS_DPI_AWARENESS

Enumeration Type Documentation

_PROCESS_DPI_AWARENESS

enum _PROCESS_DPI_AWARENESS

Enumerator
PROCESS_DPI_UNAWARE
PROCESS_SYSTEM_DPI_AWARE
PROCESS_PER_MONITOR_DPI_AWARE

Definition at line 38 of file main.cpp.

                                    {
    PROCESS_DPI_UNAWARE,
    PROCESS_SYSTEM_DPI_AWARE,
    PROCESS_PER_MONITOR_DPI_AWARE
} PROCESS_DPI_AWARENESS;

Function Documentation

command_line_to_args()

void command_line_to_args	(	LPWSTR	cmdline,
		vector< wstring > &	args
	)

Definition at line 647 of file main.cpp.

                                                                 {
    LPWSTR* l;
    int num_args;

    args.clear();

    l = CommandLineToArgvW(cmdline, &num_args);

    if (!l)
        return;

    try {
        args.reserve(num_args);

        for (unsigned int i = 0; i < (unsigned int)num_args; i++) {
            args.push_back(l[i]);
        }
    } catch (...) {
        LocalFree(l);
        throw;
    }

    LocalFree(l);
}

Referenced by CreateSnapshotW(), CreateSubvolW(), RecvSubvolW(), ReflinkCopyW(), SendSubvolW(), StartScrubW(), and StopScrubW().

create_snapshot2()

static void create_snapshot2 ( const wstring &  source, const wstring &  fn  )

static

Definition at line 601 of file main.cpp.

                                                                       {
    size_t found = fn.rfind(L"\\");
    wstring path, file;
    win_handle h, src;
    btrfs_create_snapshot* bcs;
    IO_STATUS_BLOCK iosb;

    if (found == wstring::npos) {
        path = L"";
        file = fn;
    } else {
        path = fn.substr(0, found);
        file = fn.substr(found + 1);
    }
    path += L"\\";

    src = CreateFileW(source.c_str(), FILE_TRAVERSE, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, nullptr, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, nullptr);
    if (src == INVALID_HANDLE_VALUE)
        return;

    h = CreateFileW(path.c_str(), FILE_ADD_SUBDIRECTORY, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, nullptr, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, nullptr);

    if (h == INVALID_HANDLE_VALUE)
        return;

    size_t bcslen = offsetof(btrfs_create_snapshot, name[0]) + (file.length() * sizeof(WCHAR));
    bcs = (btrfs_create_snapshot*)malloc(bcslen);

    bcs->readonly = false;
    bcs->posix = false;
    bcs->namelen = (uint16_t)(file.length() * sizeof(WCHAR));
    memcpy(bcs->name, file.c_str(), bcs->namelen);
    bcs->subvol = src;

    NtFsControlFile(h, nullptr, nullptr, nullptr, &iosb, FSCTL_BTRFS_CREATE_SNAPSHOT, bcs, (ULONG)bcslen, nullptr, 0);
}

Referenced by CreateSnapshotW().

create_subvol()

static void create_subvol ( const wstring &  fn )

static

Definition at line 560 of file main.cpp.

                                             {
    size_t found = fn.rfind(L"\\");
    wstring path, file;
    win_handle h;
    btrfs_create_subvol* bcs;
    IO_STATUS_BLOCK iosb;

    if (found == wstring::npos) {
        path = L"";
        file = fn;
    } else {
        path = fn.substr(0, found);
        file = fn.substr(found + 1);
    }
    path += L"\\";

    h = CreateFileW(path.c_str(), FILE_ADD_SUBDIRECTORY, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, nullptr, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, nullptr);

    if (h == INVALID_HANDLE_VALUE)
        return;

    size_t bcslen = offsetof(btrfs_create_subvol, name[0]) + (file.length() * sizeof(WCHAR));
    bcs = (btrfs_create_subvol*)malloc(bcslen);

    bcs->readonly = false;
    bcs->posix = false;
    bcs->namelen = (uint16_t)(file.length() * sizeof(WCHAR));
    memcpy(bcs->name, file.c_str(), bcs->namelen);

    NtFsControlFile(h, nullptr, nullptr, nullptr, &iosb, FSCTL_BTRFS_CREATE_SUBVOL, bcs, (ULONG)bcslen, nullptr, 0);
}

Referenced by CreateSubvolW().

CreateSnapshotW()

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

Definition at line 638 of file main.cpp.

                                                                                                       {
    vector<wstring> args;

    command_line_to_args(lpszCmdLine, args);

    if (args.size() >= 2)
        create_snapshot2(args[0], args[1]);
}

CreateSubvolW()

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

Definition at line 592 of file main.cpp.

                                                                                                     {
    vector<wstring> args;

    command_line_to_args(lpszCmdLine, args);

    if (args.size() >= 1)
        create_subvol(args[0]);
}

DllCanUnloadNow()

STDAPI DllCanUnloadNow

(

void

)

Definition at line 246 of file main.cpp.

                                        {
    return objs_loaded == 0 ? S_OK : S_FALSE;
}

DllGetClassObject()

STDAPI DllGetClassObject	(	REFCLSID	rclsid,
		REFIID	riid,
		LPVOID *	ppv
	)

Definition at line 250 of file main.cpp.

                                                                               {
    if (rclsid == CLSID_ShellBtrfsIconHandler) {
        Factory* fact = new Factory;
        if (!fact)
            return E_OUTOFMEMORY;
        else {
            fact->type = FactoryIconHandler;

            return fact->QueryInterface(riid, ppv);
        }
    } else if (rclsid == CLSID_ShellBtrfsContextMenu) {
        Factory* fact = new Factory;
        if (!fact)
            return E_OUTOFMEMORY;
        else {
            fact->type = FactoryContextMenu;

            return fact->QueryInterface(riid, ppv);
        }
    } else if (rclsid == CLSID_ShellBtrfsPropSheet) {
        Factory* fact = new Factory;
        if (!fact)
            return E_OUTOFMEMORY;
        else {
            fact->type = FactoryPropSheet;

            return fact->QueryInterface(riid, ppv);
        }
    } else if (rclsid == CLSID_ShellBtrfsVolPropSheet) {
        Factory* fact = new Factory;
        if (!fact)
            return E_OUTOFMEMORY;
        else {
            fact->type = FactoryVolPropSheet;

            return fact->QueryInterface(riid, ppv);
        }
    }

    return CLASS_E_CLASSNOTAVAILABLE;
}

DllInstall()

STDAPI DllInstall	(	BOOL	bInstall,
		LPCWSTR	pszCmdLine
	)

Definition at line 546 of file main.cpp.

                                                                {
    if (bInstall)
        return DllRegisterServer();
    else
        return DllUnregisterServer();
}

DllMain()

BOOL APIENTRY DllMain	(	HANDLE	hModule,
		DWORD	dwReason,
		void *	lpReserved
	)

Definition at line 553 of file main.cpp.

                                                                                   {
    if (dwReason == DLL_PROCESS_ATTACH)
        module = (HMODULE)hModule;

    return true;
}

DllRegisterServer()

STDAPI DllRegisterServer

(

void

)

Definition at line 502 of file main.cpp.

                                          {
    try {
        register_clsid(CLSID_ShellBtrfsIconHandler, COM_DESCRIPTION_ICON_HANDLER);
        register_clsid(CLSID_ShellBtrfsContextMenu, COM_DESCRIPTION_CONTEXT_MENU);
        register_clsid(CLSID_ShellBtrfsPropSheet, COM_DESCRIPTION_PROP_SHEET);
        register_clsid(CLSID_ShellBtrfsVolPropSheet, COM_DESCRIPTION_VOL_PROP_SHEET);

        reg_icon_overlay(CLSID_ShellBtrfsIconHandler, ICON_OVERLAY_NAME);

        reg_context_menu_handler(CLSID_ShellBtrfsContextMenu, L"Directory\\Background", ICON_OVERLAY_NAME);
        reg_context_menu_handler(CLSID_ShellBtrfsContextMenu, L"Folder", ICON_OVERLAY_NAME);

        reg_prop_sheet_handler(CLSID_ShellBtrfsPropSheet, L"Folder", ICON_OVERLAY_NAME);
        reg_prop_sheet_handler(CLSID_ShellBtrfsPropSheet, L"*", ICON_OVERLAY_NAME);
        reg_prop_sheet_handler(CLSID_ShellBtrfsVolPropSheet, L"Drive", ICON_OVERLAY_NAME);
    } catch (const exception& e) {
        error_message(nullptr, e.what());
        return E_FAIL;
    }

    return S_OK;
}

Referenced by DllInstall().

DllUnregisterServer()

STDAPI DllUnregisterServer

(

void

)

Definition at line 525 of file main.cpp.

                                            {
    try {
        unreg_prop_sheet_handler(L"Folder", ICON_OVERLAY_NAME);
        unreg_prop_sheet_handler(L"*", ICON_OVERLAY_NAME);
        unreg_prop_sheet_handler(L"Drive", ICON_OVERLAY_NAME);
        unreg_context_menu_handler(L"Folder", ICON_OVERLAY_NAME);
        unreg_context_menu_handler(L"Directory\\Background", ICON_OVERLAY_NAME);
        unreg_icon_overlay(ICON_OVERLAY_NAME);

        unregister_clsid(CLSID_ShellBtrfsVolPropSheet);
        unregister_clsid(CLSID_ShellBtrfsPropSheet);
        unregister_clsid(CLSID_ShellBtrfsContextMenu);
        unregister_clsid(CLSID_ShellBtrfsIconHandler);
    } catch (const exception& e) {
        error_message(nullptr, e.what());
        return E_FAIL;
    }

    return S_OK;
}

Referenced by DllInstall().

error_message()

void error_message	(	HWND	hwnd,
		const char *	msg
	)

Definition at line 785 of file main.cpp.

                                               {
    wstring title;

    load_string(module, IDS_ERROR, title);

    auto wmsg = utf8_to_utf16(msg);

    MessageBoxW(hwnd, wmsg.c_str(), title.c_str(), MB_ICONERROR);
}

Referenced by AddDeviceW(), BtrfsVolPropSheet::AddPages(), BtrfsPropSheet::AddPages(), BtrfsBalance::BalanceDlgProc(), BtrfsBalance::BalanceOptsDlgProc(), BtrfsDeviceAdd::DeviceAddDlgProc(), BtrfsVolPropSheet::DeviceDlgProc(), BtrfsDeviceResize::DeviceResizeDlgProc(), BtrfsChangeDriveLetter::DlgProc(), DllRegisterServer(), DllUnregisterServer(), BtrfsChangeDriveLetter::do_change(), BtrfsPropSheet::Initialize(), BtrfsContextMenu::InvokeCommand(), PauseBalanceW(), PropSheetDlgProc(), RecvSubvolGUIW(), RemoveDeviceW(), ResetStatsW(), ResizeDeviceW(), BtrfsScrub::ScrubDlgProc(), BtrfsSend::SendDlgProc(), SendSubvolGUIW(), ShowPropSheetW(), ShowScrubW(), SizeDetailsDlgProc(), StartBalanceW(), BtrfsVolPropSheet::StatsDlgProc(), StopBalanceW(), and BtrfsVolPropSheet::UsageDlgProc().

format_message()

wstring format_message

(

ULONG

last_error

)

Definition at line 164 of file main.cpp.

                                         {
    WCHAR* buf;
    wstring s;

    if (FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS, nullptr,
        last_error, 0, (WCHAR*)&buf, 0, nullptr) == 0) {
        return L"(error retrieving message)";
    }

    s = buf;

    LocalFree(buf);

    // remove trailing newline
    while (s.length() > 0 && (s.substr(s.length() - 1, 1) == L"\r" || s.substr(s.length() - 1, 1) == L"\n"))
        s = s.substr(0, s.length() - 1);

    return s;
}

Referenced by format_ntstatus().

format_ntstatus()

wstring format_ntstatus

(

NTSTATUS

Status

)

Definition at line 184 of file main.cpp.

                                         {
    _RtlNtStatusToDosError RtlNtStatusToDosError;
    wstring s;
    HMODULE ntdll = LoadLibraryW(L"ntdll.dll");

    if (!ntdll)
        return L"(error loading ntdll.dll)";

    RtlNtStatusToDosError = (_RtlNtStatusToDosError)GetProcAddress(ntdll, "RtlNtStatusToDosError");

    if (!RtlNtStatusToDosError) {
        FreeLibrary(ntdll);
        return L"(error loading RtlNtStatusToDosError)";
    }

    s = format_message(RtlNtStatusToDosError(Status));

    FreeLibrary(ntdll);

    return s;
}

Referenced by BtrfsSend::AddClone(), BtrfsSend::BrowseParent(), BtrfsRecv::cmd_chmod(), BtrfsRecv::cmd_chown(), BtrfsRecv::cmd_clone(), BtrfsRecv::cmd_mkfile(), BtrfsRecv::cmd_removexattr(), BtrfsRecv::cmd_setxattr(), BtrfsRecv::cmd_snapshot(), BtrfsRecv::cmd_subvol(), BtrfsRecv::do_recv(), BtrfsBalance::RefreshBalanceDlg(), and BtrfsSend::Thread().

format_size()

void format_size	(	uint64_t	size,
		wstring &	s,
		bool	show_bytes
	)

Definition at line 65 of file main.cpp.

                                                             {
    wstring t, bytes, kb, nb;
    WCHAR nb2[255];
    ULONG sr;
    float f;
    NUMBERFMTW fmt;
    WCHAR dec[2], thou[4], grouping[64], *c;
#ifdef __REACTOS__
    WCHAR buffer[64];
#endif

#ifndef __REACTOS__
    nb = to_wstring(size);
#else
    swprintf(buffer, L"%I64d", size);
    nb = wstring(buffer);
#endif

    GetLocaleInfoW(LOCALE_USER_DEFAULT, LOCALE_STHOUSAND, thou, sizeof(thou) / sizeof(WCHAR));

    dec[0] = '.'; dec[1] = 0; // not used, but silences gcc warning

    fmt.NumDigits = 0;
    fmt.LeadingZero = 1;
    fmt.lpDecimalSep = dec;
    fmt.lpThousandSep = thou;
    fmt.NegativeOrder = 0;

    // Grouping code copied from dlls/shlwapi/string.c in Wine - thank you

    fmt.Grouping = 0;
    GetLocaleInfoW(LOCALE_USER_DEFAULT, LOCALE_SGROUPING, grouping, sizeof(grouping) / sizeof(WCHAR));

    c = grouping;
    while (*c) {
        if (*c >= '0' && *c < '9') {
            fmt.Grouping *= 10;
            fmt.Grouping += *c - '0';
        }

        c++;
    }

    if (fmt.Grouping % 10 == 0)
        fmt.Grouping /= 10;
    else
        fmt.Grouping *= 10;

    GetNumberFormatW(LOCALE_USER_DEFAULT, 0, nb.c_str(), &fmt, nb2, sizeof(nb2) / sizeof(WCHAR));

    if (size < 1024) {
        if (!load_string(module, size == 1 ? IDS_SIZE_BYTE : IDS_SIZE_BYTES, t))
            throw last_error(GetLastError());

        wstring_sprintf(s, t, nb2);
        return;
    }

    if (show_bytes) {
        if (!load_string(module, IDS_SIZE_BYTES, t))
            throw last_error(GetLastError());

        wstring_sprintf(bytes, t, nb2);
    }

    if (size >= 1152921504606846976) {
        sr = IDS_SIZE_EB;
        f = (float)size / 1152921504606846976.0f;
    } else if (size >= 1125899906842624) {
        sr = IDS_SIZE_PB;
        f = (float)size / 1125899906842624.0f;
    } else if (size >= 1099511627776) {
        sr = IDS_SIZE_TB;
        f = (float)size / 1099511627776.0f;
    } else if (size >= 1073741824) {
        sr = IDS_SIZE_GB;
        f = (float)size / 1073741824.0f;
    } else if (size >= 1048576) {
        sr = IDS_SIZE_MB;
        f = (float)size / 1048576.0f;
    } else {
        sr = IDS_SIZE_KB;
        f = (float)size / 1024.0f;
    }

    if (!load_string(module, sr, t))
        throw last_error(GetLastError());

    if (show_bytes) {
        wstring_sprintf(kb, t, f);

        if (!load_string(module, IDS_SIZE_LARGE, t))
            throw last_error(GetLastError());

        wstring_sprintf(s, t, kb.c_str(), bytes.c_str());
    } else
        wstring_sprintf(s, t, f);
}

Referenced by BtrfsDeviceResize::DeviceResizeDlgProc(), BtrfsDeviceResize::do_resize(), BtrfsVolPropSheet::FormatUsage(), midi_IDirectMusicPort_GetFormat(), BtrfsDeviceAdd::populate_device_tree(), BtrfsVolPropSheet::RefreshDevList(), BtrfsPropSheet::set_size_on_disk(), BtrfsPropSheet::update_size_details_dialog(), and BtrfsScrub::UpdateTextBox().

load_string()

bool load_string	(	HMODULE	module,
		UINT	id,
		wstring &	s
	)

Definition at line 206 of file main.cpp.

                                                      {
    int len;
    LPWSTR retstr = nullptr;

    len = LoadStringW(module, id, (LPWSTR)&retstr, 0);

    if (len == 0)
        return false;

    s = wstring(retstr, len);

    return true;
}

Referenced by error_message(), format_size(), and string_error::string_error().

reg_context_menu_handler()

static void reg_context_menu_handler ( const GUID  clsid, const wstring &  filetype, const wstring &  name  )

static

Definition at line 446 of file main.cpp.

                                                                                                     {
    WCHAR* clsidstring;

    StringFromCLSID(clsid, &clsidstring);

    try {
#ifndef __REACTOS__
        wstring path = filetype + L"\\ShellEx\\ContextMenuHandlers\\"s + name;
#else
        wstring path = filetype + wstring(L"\\ShellEx\\ContextMenuHandlers\\") + name;
#endif

        write_reg_key(HKEY_CLASSES_ROOT, path, nullptr, clsidstring);
    } catch (...) {
        CoTaskMemFree(clsidstring);
        throw;
    }
}

Referenced by DllRegisterServer().

reg_delete_tree()

static void reg_delete_tree ( HKEY  hkey, const wstring &  keyname  )

static

Definition at line 344 of file main.cpp.

                                                               {
    HKEY k;
    LSTATUS ret;

    ret = RegOpenKeyExW(hkey, keyname.c_str(), 0, KEY_READ, &k);

    if (ret != ERROR_SUCCESS)
        throw last_error(ret);

    try {
        WCHAR* buf;
        ULONG bufsize;

        ret = RegQueryInfoKeyW(k, nullptr, nullptr, nullptr, nullptr, &bufsize, nullptr,
                               nullptr, nullptr, nullptr, nullptr, nullptr);
        if (ret != ERROR_SUCCESS)
            throw last_error(ret);

        bufsize++;
        buf = new WCHAR[bufsize];

        try {
            do {
                ULONG size = bufsize;

                ret = RegEnumKeyExW(k, 0, buf, &size, nullptr, nullptr, nullptr, nullptr);

                if (ret == ERROR_NO_MORE_ITEMS)
                    break;
                else if (ret != ERROR_SUCCESS)
                    throw last_error(ret);

                reg_delete_tree(k, buf);
            } while (true);

            ret = RegDeleteKeyW(hkey, keyname.c_str());
            if (ret != ERROR_SUCCESS)
                throw last_error(ret);
        } catch (...) {
            delete[] buf;
            throw;
        }

        delete[] buf;
    } catch (...) {
        RegCloseKey(k);
        throw;
    }

    RegCloseKey(k);
}

Referenced by unreg_context_menu_handler(), unreg_icon_overlay(), unreg_prop_sheet_handler(), and unregister_clsid().

reg_icon_overlay()

static void reg_icon_overlay ( const GUID  clsid, const wstring &  name  )

static

Definition at line 416 of file main.cpp.

                                                                    {
    WCHAR* clsidstring;

    StringFromCLSID(clsid, &clsidstring);

    try {
#ifndef __REACTOS__
        wstring path = L"Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\ShellIconOverlayIdentifiers\\"s + name;
#else
        wstring path = wstring(L"Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\ShellIconOverlayIdentifiers\\") + name;
#endif

        write_reg_key(HKEY_LOCAL_MACHINE, path, nullptr, clsidstring);
    } catch (...) {
        CoTaskMemFree(clsidstring);
        throw;
    }

    CoTaskMemFree(clsidstring);
}

Referenced by DllRegisterServer().

reg_prop_sheet_handler()

static void reg_prop_sheet_handler ( const GUID  clsid, const wstring &  filetype, const wstring &  name  )

static

Definition at line 474 of file main.cpp.

                                                                                                   {
    WCHAR* clsidstring;

    StringFromCLSID(clsid, &clsidstring);

    try {
#ifndef __REACTOS__
        wstring path = filetype + L"\\ShellEx\\PropertySheetHandlers\\"s + name;
#else
        wstring path = filetype + wstring(L"\\ShellEx\\PropertySheetHandlers\\") + name;
#endif

        write_reg_key(HKEY_CLASSES_ROOT, path, nullptr, clsidstring);
    } catch (...) {
        CoTaskMemFree(clsidstring);
        throw;
    }
}

Referenced by DllRegisterServer().

register_clsid()

static void register_clsid ( const GUID  clsid, const WCHAR *  description  )

static

Definition at line 310 of file main.cpp.

                                                                       {
    WCHAR* clsidstring;
    wstring inproc, progid, clsidkeyname;
    WCHAR dllpath[MAX_PATH];

    StringFromCLSID(clsid, &clsidstring);

    try {
#ifndef __REACTOS__
        inproc = L"CLSID\\"s + clsidstring + L"\\InprocServer32"s;
        progid = L"CLSID\\"s + clsidstring + L"\\ProgId"s;
        clsidkeyname = L"CLSID\\"s + clsidstring;
#else
        inproc = wstring(L"CLSID\\") + clsidstring + wstring(L"\\InprocServer32");
        progid = wstring(L"CLSID\\") + clsidstring + wstring(L"\\ProgId");
        clsidkeyname = wstring(L"CLSID\\") + clsidstring;
#endif

        write_reg_key(HKEY_CLASSES_ROOT, clsidkeyname, nullptr, description);

        GetModuleFileNameW(module, dllpath, sizeof(dllpath) / sizeof(WCHAR));

        write_reg_key(HKEY_CLASSES_ROOT, inproc, nullptr, dllpath);

        write_reg_key(HKEY_CLASSES_ROOT, inproc, L"ThreadingModel", L"Apartment");
    } catch (...) {
        CoTaskMemFree(clsidstring);
        throw;
    }

    CoTaskMemFree(clsidstring);
}

Referenced by DllRegisterServer().

set_dpi_aware()

void set_dpi_aware

(

)

Definition at line 50 of file main.cpp.

                     {
    _SetProcessDpiAwareness SetProcessDpiAwareness;
    HMODULE shcore = LoadLibraryW(L"shcore.dll");

    if (!shcore)
        return;

    SetProcessDpiAwareness = (_SetProcessDpiAwareness)GetProcAddress(shcore, "SetProcessDpiAwareness");

    if (!SetProcessDpiAwareness)
        return;

    SetProcessDpiAwareness(PROCESS_PER_MONITOR_DPI_AWARE);
}

Referenced by AddDeviceW(), RecvSubvolGUIW(), RemoveDeviceW(), ResetStatsW(), ResizeDeviceW(), SendSubvolGUIW(), ShowPropSheetW(), and ShowScrubW().

unreg_context_menu_handler()

static void unreg_context_menu_handler ( const wstring &  filetype, const wstring &  name  )

static

Definition at line 465 of file main.cpp.

                                                                                     {
#ifndef __REACTOS__
    reg_delete_tree(HKEY_CLASSES_ROOT, filetype + L"\\ShellEx\\ContextMenuHandlers\\"s + name);
#else
    wstring path = filetype + wstring(L"\\ShellEx\\ContextMenuHandlers\\") + name;
    reg_delete_tree(HKEY_CLASSES_ROOT, path);
#endif
}

Referenced by DllUnregisterServer().

unreg_icon_overlay()

static void unreg_icon_overlay ( const wstring &  name )

static

Definition at line 437 of file main.cpp.

                                                    {
#ifndef __REACTOS__
    reg_delete_tree(HKEY_LOCAL_MACHINE, L"Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\ShellIconOverlayIdentifiers\\"s + name);
#else
    wstring path = wstring(L"Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\ShellIconOverlayIdentifiers\\") + name;
    reg_delete_tree(HKEY_LOCAL_MACHINE, path);
#endif
}

Referenced by DllUnregisterServer().

unreg_prop_sheet_handler()

static void unreg_prop_sheet_handler ( const wstring &  filetype, const wstring &  name  )

static

Definition at line 493 of file main.cpp.

                                                                                   {
#ifndef __REACTOS__
    reg_delete_tree(HKEY_CLASSES_ROOT, filetype + L"\\ShellEx\\PropertySheetHandlers\\"s + name);
#else
    wstring path = filetype + wstring(L"\\ShellEx\\PropertySheetHandlers\\") + name;
    reg_delete_tree(HKEY_CLASSES_ROOT, path);
#endif
}

Referenced by DllUnregisterServer().

unregister_clsid()

static void unregister_clsid ( const GUID  clsid )

static

Definition at line 396 of file main.cpp.

                                               {
    WCHAR* clsidstring;

    StringFromCLSID(clsid, &clsidstring);

    try {
#ifndef __REACTOS__
        reg_delete_tree(HKEY_CLASSES_ROOT, L"CLSID\\"s + clsidstring);
#else
        wstring path = wstring(L"CLSID\\") + clsidstring;
        reg_delete_tree(HKEY_CLASSES_ROOT, path);
#endif
    } catch (...) {
        CoTaskMemFree(clsidstring);
        throw;
    }

    CoTaskMemFree(clsidstring);
}

Referenced by DllUnregisterServer().

utf16_to_utf8()

static string utf16_to_utf8 ( const wstring_view &  utf16 )

static

Definition at line 672 of file main.cpp.

                                                       {
    string utf8;
    char* buf;

    if (utf16.empty())
        return "";

    auto utf8len = WideCharToMultiByte(CP_UTF8, 0, utf16.data(), static_cast<int>(utf16.length()), nullptr, 0, nullptr, nullptr);

    if (utf8len == 0)
        throw last_error(GetLastError());

    buf = (char*)malloc(utf8len + sizeof(char));

    if (!buf)
        throw string_error(IDS_OUT_OF_MEMORY);

    if (WideCharToMultiByte(CP_UTF8, 0, utf16.data(), static_cast<int>(utf16.length()), buf, utf8len, nullptr, nullptr) == 0) {
        auto le = GetLastError();
        free(buf);
        throw last_error(le);
    }

    buf[utf8len] = 0;

    utf8 = buf;

    free(buf);

    return utf8;
}

Referenced by create_snapshot(), create_stream(), create_subvol(), file_create2(), last_error::last_error(), mknod(), ntstatus_error::ntstatus_error(), rename_file_to_stream(), rename_stream(), set_link_information(), set_rename_information(), set_symlink(), and string_error::string_error().

utf8_to_utf16()

wstring utf8_to_utf16

(

const string_view &

utf8

)

Definition at line 736 of file main.cpp.

                                               {
    wstring ret;
    WCHAR* buf;

    if (utf8.empty())
        return L"";

    auto utf16len = MultiByteToWideChar(CP_UTF8, 0, utf8.data(), (int)utf8.length(), nullptr, 0);

    if (utf16len == 0)
        throw last_error(GetLastError());

    buf = (WCHAR*)malloc((utf16len + 1) * sizeof(WCHAR));

    if (!buf)
        throw string_error(IDS_OUT_OF_MEMORY);

    if (MultiByteToWideChar(CP_UTF8, 0, utf8.data(), (int)utf8.length(), buf, utf16len) == 0) {
        auto le = GetLastError();
        free(buf);
        throw last_error(le);
    }

    buf[utf16len] = 0;

    ret = buf;

    free(buf);

    return ret;
}

Referenced by BtrfsRecv::cmd_chmod(), BtrfsRecv::cmd_chown(), BtrfsRecv::cmd_clone(), BtrfsRecv::cmd_link(), BtrfsRecv::cmd_mkfile(), BtrfsRecv::cmd_removexattr(), BtrfsRecv::cmd_rename(), BtrfsRecv::cmd_rmdir(), BtrfsRecv::cmd_setxattr(), BtrfsRecv::cmd_snapshot(), BtrfsRecv::cmd_subvol(), BtrfsRecv::cmd_truncate(), BtrfsRecv::cmd_unlink(), BtrfsRecv::cmd_utimes(), BtrfsRecv::cmd_write(), error_message(), get_reparse_block(), get_reparse_point(), load_dir_children(), log_file_checksum_error(), open_fcb(), open_fileref_by_inode(), BtrfsRecv::recv_thread(), BtrfsRecv::RecvProgressDlgProc(), and BtrfsSend::Thread().

write_reg_key()

static void write_reg_key ( HKEY  root, const wstring &  keyname, const WCHAR *  val, const wstring &  data  )

static

Definition at line 292 of file main.cpp.

                                                                                                    {
    LONG l;
    HKEY hk;
    DWORD dispos;

    l = RegCreateKeyExW(root, keyname.c_str(), 0, nullptr, 0, KEY_ALL_ACCESS, nullptr, &hk, &dispos);
    if (l != ERROR_SUCCESS)
        throw string_error(IDS_REGCREATEKEY_FAILED, l);

    l = RegSetValueExW(hk, val, 0, REG_SZ, (const BYTE*)data.c_str(), (DWORD)((data.length() + 1) * sizeof(WCHAR)));
    if (l != ERROR_SUCCESS)
        throw string_error(IDS_REGSETVALUEEX_FAILED, l);

    l = RegCloseKey(hk);
    if (l != ERROR_SUCCESS)
        throw string_error(IDS_REGCLOSEKEY_FAILED, l);
}

Referenced by reg_context_menu_handler(), reg_icon_overlay(), reg_prop_sheet_handler(), and register_clsid().

wstring_sprintf()

void wstring_sprintf	(	wstring &	s,
		wstring	fmt,
			...
	)

Definition at line 225 of file main.cpp.

                                                   {
    int len;
    va_list args;

    va_start(args, fmt);
    len = _vsnwprintf(nullptr, 0, fmt.c_str(), args);

    if (len == 0)
        s = L"";
    else {
        s.resize(len);
        _vsnwprintf((wchar_t*)s.c_str(), len, fmt.c_str(), args);
    }

    va_end(args);
}

Referenced by BtrfsDeviceAdd::AddDevice(), BtrfsBalance::BalanceOptsDlgProc(), create_snapshot(), BtrfsVolPropSheet::DeviceDlgProc(), BtrfsDeviceResize::DeviceResizeDlgProc(), BtrfsDeviceResize::do_resize(), format_size(), BtrfsVolPropSheet::FormatUsage(), BtrfsPropSheet::init_propsheet(), BtrfsDeviceAdd::populate_device_tree(), PropSheetDlgProc(), BtrfsRecv::recv_thread(), BtrfsBalance::RefreshBalanceDlg(), BtrfsVolPropSheet::RefreshDevList(), BtrfsScrub::RefreshScrubDlg(), BtrfsPropSheet::set_size_on_disk(), BtrfsVolPropSheet::StatsDlgProc(), and BtrfsScrub::UpdateTextBox().

Variable Documentation

CLSID_ShellBtrfsContextMenu

const GUID CLSID_ShellBtrfsContextMenu = { 0x2690b74f, 0xf353, 0x422d, { 0xbb, 0x12, 0x40, 0x15, 0x81, 0xee, 0xf8, 0xf1 } }

static

Definition at line 28 of file main.cpp.

Referenced by DllGetClassObject(), DllRegisterServer(), and DllUnregisterServer().

CLSID_ShellBtrfsIconHandler

const GUID CLSID_ShellBtrfsIconHandler = { 0x2690b74f, 0xf353, 0x422d, { 0xbb, 0x12, 0x40, 0x15, 0x81, 0xee, 0xf8, 0xf0 } }

static

Definition at line 27 of file main.cpp.

Referenced by DllGetClassObject(), DllRegisterServer(), and DllUnregisterServer().

CLSID_ShellBtrfsPropSheet

const GUID CLSID_ShellBtrfsPropSheet = { 0x2690b74f, 0xf353, 0x422d, { 0xbb, 0x12, 0x40, 0x15, 0x81, 0xee, 0xf8, 0xf2 } }

static

Definition at line 29 of file main.cpp.

Referenced by DllGetClassObject(), DllRegisterServer(), and DllUnregisterServer().

CLSID_ShellBtrfsVolPropSheet

const GUID CLSID_ShellBtrfsVolPropSheet = { 0x2690b74f, 0xf353, 0x422d, { 0xbb, 0x12, 0x40, 0x15, 0x81, 0xee, 0xf8, 0xf3 } }

static

Definition at line 30 of file main.cpp.

Referenced by DllGetClassObject(), DllRegisterServer(), and DllUnregisterServer().

module

HMODULE module

Definition at line 47 of file main.cpp.

Referenced by BtrfsVolPropSheet::AddPages(), BtrfsPropSheet::AddPages(), af_face_globals_new(), BtrfsSend::Browse(), cache_connection(), cff_size_get_globals_funcs(), cff_slot_init(), check_hook_thread(), cid_size_get_globals_funcs(), cid_slot_init(), DECLARE_INTERFACE_(), do_register_dll(), dsound_enum(), dwarf2_parse(), dwarf2_parse_compilation_unit(), elf_load_cb(), elf_search_and_load_file(), elf_synchronize_module_list(), FT_Add_Module(), ft_add_renderer(), FT_Get_Module_Interface(), FxpGetImageBase(), get_process_name_from_pid(), http_release_netconn(), InternalIsOS2OrOldWin(), ITERATE_RegisterTypeLibraries(), ITypeInfo_fnAddressOfMember(), load_d3dcompiler(), load_d3dcompiler_43(), load_d3dcompiler_47(), load_libglu(), LoadShimDLL2(), macho_load_debug_info(), macho_search_and_load_file(), macho_synchronize_module_list(), pe_load_builtin_module(), pe_load_debug_directory(), pe_load_native_module(), RecvSubvolGUIW(), ResProtocolInfo_ParseUrl(), RtlCreateActivationContext(), setup_dll(), ShowPropSheetW(), source_new(), stabs_parse_typedef(), START_TEST(), SymFromName(), T1_GlyphSlot_Init(), T1_Size_Get_Globals_Funcs(), test_comctl32_class(), test_FakeDLL(), test_HashLinks(), test_md5_ctx(), test_queryvirtualmemory(), test_thread_start_address(), TIFFError(), TIFFErrorExt(), TIFFWarning(), TIFFWarningExt(), x86_64_fetch_minidump_module(), xsltGetExtData(), xsltInitCtxtExt(), xsltRegisterExtModuleFull(), xsltRegisterExtPrefix(), xsltShutdownCtxtExt(), xsltShutdownExt(), and xsltStyleInitializeStylesheetModule().

objs_loaded

LONG objs_loaded = 0

Definition at line 48 of file main.cpp.

Referenced by BtrfsVolPropSheet::AddPages(), BtrfsPropSheet::AddPages(), BtrfsContextMenu::BtrfsContextMenu(), BtrfsIconOverlay::BtrfsIconOverlay(), BtrfsPropSheet::BtrfsPropSheet(), BtrfsVolPropSheet::BtrfsVolPropSheet(), DllCanUnloadNow(), Factory::Factory(), BtrfsContextMenu::~BtrfsContextMenu(), BtrfsIconOverlay::~BtrfsIconOverlay(), BtrfsPropSheet::~BtrfsPropSheet(), BtrfsVolPropSheet::~BtrfsVolPropSheet(), and Factory::~Factory().