recv.cpp File Reference

#include "shellext.h"
#include <windows.h>
#include <strsafe.h>
#include <stddef.h>
#include <sys/stat.h>
#include <iostream>
#include "recv.h"
#include "resource.h"
#include "crc32c.h"
#include <cpuid.h>

Include dependency graph for recv.cpp:

Go to the source code of this file.

Functions
static __inline uint64_t	unix_time_to_win (BTRFS_TIME *t)
static void	delete_directory (const wstring &dir)
static bool	check_csum (btrfs_send_command *cmd, uint8_t *data)
static DWORD WINAPI	global_recv_thread (LPVOID lpParameter)
static INT_PTR CALLBACK	stub_RecvProgressDlgProc (HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
void CALLBACK	RecvSubvolGUIW (HWND hwnd, HINSTANCE hinst, LPWSTR lpszCmdLine, int nCmdShow)
void CALLBACK	RecvSubvolW (HWND hwnd, HINSTANCE hinst, LPWSTR lpszCmdLine, int nCmdShow)

Variables
const string	EA_NTACL = "security.NTACL"
const string	EA_DOSATTRIB = "user.DOSATTRIB"
const string	EA_REPARSE = "user.reparse"
const string	EA_EA = "user.EA"
const string	XATTR_USER = "user."

Function Documentation

check_csum()

static bool check_csum ( btrfs_send_command *  cmd, uint8_t *  data  )

static

Definition at line 1147 of file recv.cpp.

                                                               {
    uint32_t crc32 = cmd->csum, calc;
 
    cmd->csum = 0;
 
    calc = calc_crc32c(0, (uint8_t*)cmd, sizeof(btrfs_send_command));
 
    if (cmd->length > 0)
        calc = calc_crc32c(calc, data, cmd->length);
 
    return calc == crc32 ? true : false;
}

Referenced by BtrfsRecv::do_recv(), and scrub_extent_dup().

delete_directory()

static void delete_directory ( const wstring &  dir )

static

Definition at line 1117 of file recv.cpp.

                                                 {
    WIN32_FIND_DATAW fff;
 
    fff_handle h = FindFirstFileW((dir + L"*").c_str(), &fff);
 
    if (h == INVALID_HANDLE_VALUE)
        return;
 
    do {
        wstring file;
 
        file = fff.cFileName;
 
        if (file != L"." && file != L"..") {
            if (fff.dwFileAttributes & FILE_ATTRIBUTE_READONLY)
                SetFileAttributesW((dir + file).c_str(), fff.dwFileAttributes & ~FILE_ATTRIBUTE_READONLY);
 
            if (fff.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
                if (!(fff.dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT))
                    delete_directory(dir + file + L"\\");
                else
                    RemoveDirectoryW((dir + file).c_str());
            } else
                DeleteFileW((dir + file).c_str());
        }
    } while (FindNextFileW(h, &fff));
 
    RemoveDirectoryW(dir.c_str());
}

Referenced by delete_directory(), and BtrfsRecv::do_recv().

global_recv_thread()

static DWORD WINAPI global_recv_thread ( LPVOID  lpParameter )

static

Definition at line 1457 of file recv.cpp.

                                                           {
    BtrfsRecv* br = (BtrfsRecv*)lpParameter;
 
    return br->recv_thread();
}

Referenced by BtrfsRecv::RecvProgressDlgProc().

RecvSubvolGUIW()

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

Definition at line 1548 of file recv.cpp.

                                                                                                      {
    try {
        OPENFILENAMEW ofn;
        WCHAR file[MAX_PATH];
        win_handle token;
        TOKEN_PRIVILEGES* tp;
        LUID luid;
        ULONG tplen;
 
        set_dpi_aware();
 
        if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &token))
            throw last_error(GetLastError());
 
        tplen = offsetof(TOKEN_PRIVILEGES, Privileges[0]) + (3 * sizeof(LUID_AND_ATTRIBUTES));
        tp = (TOKEN_PRIVILEGES*)malloc(tplen);
        if (!tp)
            throw string_error(IDS_OUT_OF_MEMORY);
 
        tp->PrivilegeCount = 3;
 
#ifdef __clang__
#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Warray-bounds"
#endif // __clang__
 
        if (!LookupPrivilegeValueW(nullptr, L"SeManageVolumePrivilege", &luid)) {
            free(tp);
            throw last_error(GetLastError());
        }
 
        tp->Privileges[0].Luid = luid;
        tp->Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
 
        if (!LookupPrivilegeValueW(nullptr, L"SeSecurityPrivilege", &luid)) {
            free(tp);
            throw last_error(GetLastError());
        }
 
        tp->Privileges[1].Luid = luid;
        tp->Privileges[1].Attributes = SE_PRIVILEGE_ENABLED;
 
        if (!LookupPrivilegeValueW(nullptr, L"SeRestorePrivilege", &luid)) {
            free(tp);
            throw last_error(GetLastError());
        }
 
        tp->Privileges[2].Luid = luid;
        tp->Privileges[2].Attributes = SE_PRIVILEGE_ENABLED;
 
        if (!AdjustTokenPrivileges(token, false, tp, tplen, nullptr, nullptr)) {
            free(tp);
            throw last_error(GetLastError());
        }
 
#ifdef __clang__
    #pragma clang diagnostic pop
#endif // __clang__
 
        file[0] = 0;
 
        memset(&ofn, 0, sizeof(OPENFILENAMEW));
        ofn.lStructSize = sizeof(OPENFILENAMEW);
        ofn.hwndOwner = hwnd;
        ofn.hInstance = module;
        ofn.lpstrFile = file;
        ofn.nMaxFile = sizeof(file) / sizeof(WCHAR);
        ofn.Flags = OFN_EXPLORER | OFN_FILEMUSTEXIST | OFN_HIDEREADONLY;
 
        if (GetOpenFileNameW(&ofn)) {
            BtrfsRecv recv;
 
            recv.Open(hwnd, file, lpszCmdLine, false);
        }
 
        free(tp);
    } catch (const exception& e) {
        error_message(hwnd, e.what());
    }
}

RecvSubvolW()

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

Definition at line 1629 of file recv.cpp.

                                                                                                   {
    try {
        vector<wstring> args;
 
        command_line_to_args(lpszCmdLine, args);
 
        if (args.size() >= 2) {
            win_handle token;
            TOKEN_PRIVILEGES* tp;
            ULONG tplen;
            LUID luid;
 
            if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &token))
                return;
 
            tplen = offsetof(TOKEN_PRIVILEGES, Privileges[0]) + (3 * sizeof(LUID_AND_ATTRIBUTES));
            tp = (TOKEN_PRIVILEGES*)malloc(tplen);
            if (!tp)
                return;
 
            tp->PrivilegeCount = 3;
 
#ifdef __clang__
#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Warray-bounds"
#endif // __clang__
 
            if (!LookupPrivilegeValueW(nullptr, L"SeManageVolumePrivilege", &luid)) {
                free(tp);
                return;
            }
 
            tp->Privileges[0].Luid = luid;
            tp->Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
 
            if (!LookupPrivilegeValueW(nullptr, L"SeSecurityPrivilege", &luid)) {
                free(tp);
                return;
            }
 
            tp->Privileges[1].Luid = luid;
            tp->Privileges[1].Attributes = SE_PRIVILEGE_ENABLED;
 
            if (!LookupPrivilegeValueW(nullptr, L"SeRestorePrivilege", &luid)) {
                free(tp);
                return;
            }
 
            tp->Privileges[2].Luid = luid;
            tp->Privileges[2].Attributes = SE_PRIVILEGE_ENABLED;
 
#ifdef __clang__
#pragma clang diagnostic pop
#endif // __clang__
 
            if (!AdjustTokenPrivileges(token, false, tp, tplen, nullptr, nullptr)) {
                free(tp);
                return;
            }
 
            free(tp);
 
            BtrfsRecv br;
            br.Open(nullptr, args[0], args[1], true);
        }
    } catch (const exception& e) {
        cerr << "Error: " << e.what() << endl;
    }
}

stub_RecvProgressDlgProc()

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

static

Definition at line 1515 of file recv.cpp.

                                                                                                        {
    BtrfsRecv* br;
 
    if (uMsg == WM_INITDIALOG) {
        SetWindowLongPtrW(hwndDlg, GWLP_USERDATA, (LONG_PTR)lParam);
        br = (BtrfsRecv*)lParam;
    } else {
        br = (BtrfsRecv*)GetWindowLongPtrW(hwndDlg, GWLP_USERDATA);
    }
 
    if (br)
        return br->RecvProgressDlgProc(hwndDlg, uMsg, wParam, lParam);
    else
        return false;
}

Referenced by BtrfsRecv::Open().

unix_time_to_win()

static __inline uint64_t unix_time_to_win ( BTRFS_TIME *  t )

static

Definition at line 1070 of file recv.cpp.

                                                         {
    return (t->seconds * 10000000) + (t->nanoseconds / 100) + 116444736000000000;
}

Referenced by _Function_class_(), BtrfsRecv::cmd_utimes(), fill_in_file_basic_information(), fill_in_file_network_open_information(), fill_in_file_stat_information(), fill_in_file_stat_lx_information(), and query_dir_item().

Variable Documentation

EA_DOSATTRIB

const string EA_DOSATTRIB = "user.DOSATTRIB"

Definition at line 40 of file recv.cpp.

EA_EA

const string EA_EA = "user.EA"

Definition at line 42 of file recv.cpp.

EA_NTACL

const string EA_NTACL = "security.NTACL"

Definition at line 39 of file recv.cpp.

EA_REPARSE

const string EA_REPARSE = "user.reparse"

Definition at line 41 of file recv.cpp.

XATTR_USER

const string XATTR_USER = "user."

Definition at line 43 of file recv.cpp.

Referenced by BtrfsRecv::cmd_removexattr(), and BtrfsRecv::cmd_setxattr().