reactos/dll/shellext/shellbtrfs/recv.h
2017-11-26 14:34:53 +01:00

91 lines
3.2 KiB
C++

/* Copyright (c) Mark Harmstone 2017
*
* This file is part of WinBtrfs.
*
* WinBtrfs is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public Licence as published by
* the Free Software Foundation, either version 3 of the Licence, or
* (at your option) any later version.
*
* WinBtrfs is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public Licence for more details.
*
* You should have received a copy of the GNU Lesser General Public Licence
* along with WinBtrfs. If not, see <http://www.gnu.org/licenses/>. */
#pragma once
#include <shlobj.h>
#include <string>
#include <vector>
#ifndef __REACTOS__
#include "../btrfs.h"
#else
#include "btrfs.h"
#endif
extern LONG objs_loaded;
typedef struct {
BTRFS_UUID uuid;
UINT64 transid;
std::wstring path;
} subvol_cache;
class BtrfsRecv {
public:
BtrfsRecv() {
thread = NULL;
master = INVALID_HANDLE_VALUE;
dir = INVALID_HANDLE_VALUE;
running = FALSE;
cancelling = FALSE;
stransid = 0;
num_received = 0;
hwnd = NULL;
cache.clear();
}
virtual ~BtrfsRecv() {
cache.clear();
}
void Open(HWND hwnd, WCHAR* file, WCHAR* path, BOOL quiet);
DWORD recv_thread();
INT_PTR CALLBACK RecvProgressDlgProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam);
private:
BOOL cmd_subvol(HWND hwnd, btrfs_send_command* cmd, UINT8* data);
BOOL cmd_snapshot(HWND hwnd, btrfs_send_command* cmd, UINT8* data);
BOOL cmd_mkfile(HWND hwnd, btrfs_send_command* cmd, UINT8* data);
BOOL cmd_rename(HWND hwnd, btrfs_send_command* cmd, UINT8* data);
BOOL cmd_link(HWND hwnd, btrfs_send_command* cmd, UINT8* data);
BOOL cmd_unlink(HWND hwnd, btrfs_send_command* cmd, UINT8* data);
BOOL cmd_rmdir(HWND hwnd, btrfs_send_command* cmd, UINT8* data);
BOOL cmd_setxattr(HWND hwnd, btrfs_send_command* cmd, UINT8* data);
BOOL cmd_removexattr(HWND hwnd, btrfs_send_command* cmd, UINT8* data);
BOOL cmd_write(HWND hwnd, btrfs_send_command* cmd, UINT8* data);
BOOL cmd_clone(HWND hwnd, btrfs_send_command* cmd, UINT8* data);
BOOL cmd_truncate(HWND hwnd, btrfs_send_command* cmd, UINT8* data);
BOOL cmd_chmod(HWND hwnd, btrfs_send_command* cmd, UINT8* data);
BOOL cmd_chown(HWND hwnd, btrfs_send_command* cmd, UINT8* data);
BOOL cmd_utimes(HWND hwnd, btrfs_send_command* cmd, UINT8* data);
void add_cache_entry(BTRFS_UUID* uuid, UINT64 transid, std::wstring path);
BOOL utf8_to_utf16(HWND hwnd, char* utf8, ULONG utf8len, std::wstring* utf16);
void ShowRecvError(int resid, ...);
BOOL find_tlv(UINT8* data, ULONG datalen, UINT16 type, void** value, ULONG* len);
BOOL do_recv(HANDLE f, UINT64* pos, UINT64 size);
HANDLE dir, parent, master, thread, lastwritefile;
HWND hwnd;
std::wstring streamfile, dirpath, subvolpath, lastwritepath;
DWORD lastwriteatt;
ULONG num_received;
UINT64 stransid;
BTRFS_UUID subvol_uuid;
BOOL running, cancelling;
std::vector<subvol_cache> cache;
};