/* 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/>. */

#include "shellext.h"
#include "send.h"
#include "resource.h"
#include <stddef.h>
#include <shlobj.h>
#ifdef __REACTOS__
#undef DeleteFile
#endif

#define SEND_BUFFER_LEN 1048576

void BtrfsSend::ShowSendError(UINT msg, ...) {
    WCHAR s[1024], t[1024];
    va_list ap;

    if (!LoadStringW(module, msg, s, sizeof(s) / sizeof(WCHAR))) {
        ShowError(hwnd, GetLastError());
        return;
    }

    va_start(ap, msg);
#ifndef __REACTOS__
    vswprintf(t, sizeof(t) / sizeof(WCHAR), s, ap);
#else
    vsnwprintf(t, sizeof(t) / sizeof(WCHAR), s, ap);
#endif

    SetDlgItemTextW(hwnd, IDC_SEND_STATUS, t);

    va_end(ap);
}

DWORD BtrfsSend::Thread() {
    NTSTATUS Status;
    IO_STATUS_BLOCK iosb;
    btrfs_send_subvol* bss;
    btrfs_send_header header;
    btrfs_send_command end;
    BOOL success = FALSE;
    ULONG bss_size, i;

    buf = (char*)malloc(SEND_BUFFER_LEN);

    dirh = CreateFileW(subvol.c_str(), FILE_READ_ATTRIBUTES, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, NULL, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL);
    if (dirh == INVALID_HANDLE_VALUE) {
        ShowSendError(IDS_SEND_CANT_OPEN_DIR, subvol.c_str(), GetLastError(), format_message(GetLastError()).c_str());
        goto end3;
    }

    bss_size = offsetof(btrfs_send_subvol, clones[0]) + (clones.size() * sizeof(HANDLE));
    bss = (btrfs_send_subvol*)malloc(bss_size);
    memset(bss, 0, bss_size);

    if (incremental) {
        WCHAR parent[MAX_PATH];
        HANDLE parenth;

        parent[0] = 0;

        GetDlgItemTextW(hwnd, IDC_PARENT_SUBVOL, parent, sizeof(parent) / sizeof(WCHAR));

        parenth = CreateFileW(parent, FILE_READ_ATTRIBUTES, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, NULL, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL);
        if (parenth == INVALID_HANDLE_VALUE) {
            ShowSendError(IDS_SEND_CANT_OPEN_DIR, parent, GetLastError(), format_message(GetLastError()).c_str());
            goto end2;
        }

        bss->parent = parenth;
    } else
        bss->parent = NULL;

    bss->num_clones = clones.size();

    for (i = 0; i < bss->num_clones; i++) {
        HANDLE h;

        h = CreateFileW(clones[i].c_str(), FILE_READ_ATTRIBUTES, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, NULL, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL);
        if (h == INVALID_HANDLE_VALUE) {
            ULONG j;

            ShowSendError(IDS_SEND_CANT_OPEN_DIR, clones[i].c_str(), GetLastError(), format_message(GetLastError()).c_str());

            for (j = 0; j < i; j++) {
                CloseHandle(bss->clones[j]);
            }

            if (bss->parent) CloseHandle(bss->parent);
            goto end2;
        }

        bss->clones[i] = h;
    }

    Status = NtFsControlFile(dirh, NULL, NULL, NULL, &iosb, FSCTL_BTRFS_SEND_SUBVOL, bss, bss_size, NULL, 0);

    for (i = 0; i < bss->num_clones; i++) {
        CloseHandle(bss->clones[i]);
    }

    if (!NT_SUCCESS(Status)) {
        if (Status == (NTSTATUS)STATUS_INVALID_PARAMETER) {
            BY_HANDLE_FILE_INFORMATION fileinfo;
            if (!GetFileInformationByHandle(dirh, &fileinfo)) {
                ShowSendError(IDS_SEND_GET_FILE_INFO_FAILED, GetLastError(), format_message(GetLastError()).c_str());
                if (bss->parent) CloseHandle(bss->parent);
                goto end2;
            }

            if (!(fileinfo.dwFileAttributes & FILE_ATTRIBUTE_READONLY)) {
                ShowSendError(IDS_SEND_NOT_READONLY);
                if (bss->parent) CloseHandle(bss->parent);
                goto end2;
            }

            if (bss->parent) {
                if (!GetFileInformationByHandle(bss->parent, &fileinfo)) {
                    ShowSendError(IDS_SEND_GET_FILE_INFO_FAILED, GetLastError(), format_message(GetLastError()).c_str());
                    CloseHandle(bss->parent);
                    goto end2;
                }

                if (!(fileinfo.dwFileAttributes & FILE_ATTRIBUTE_READONLY)) {
                    ShowSendError(IDS_SEND_PARENT_NOT_READONLY);
                    CloseHandle(bss->parent);
                    goto end2;
                }
            }
        }

        ShowSendError(IDS_SEND_FSCTL_BTRFS_SEND_SUBVOL_FAILED, Status, format_ntstatus(Status).c_str());
        if (bss->parent) CloseHandle(bss->parent);
        goto end2;
    }

    if (bss->parent) CloseHandle(bss->parent);

    stream = CreateFileW(file, FILE_WRITE_DATA | DELETE, 0, NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
    if (stream == INVALID_HANDLE_VALUE) {
        ShowSendError(IDS_SEND_CANT_OPEN_FILE, file, GetLastError(), format_message(GetLastError()).c_str());
        goto end2;
    }

    memcpy(header.magic, BTRFS_SEND_MAGIC, sizeof(header.magic));
    header.version = 1;

    if (!WriteFile(stream, &header, sizeof(header), NULL, NULL)) {
        ShowSendError(IDS_SEND_WRITEFILE_FAILED, GetLastError(), format_message(GetLastError()).c_str());
        goto end;
    }

    do {
        Status = NtFsControlFile(dirh, NULL, NULL, NULL, &iosb, FSCTL_BTRFS_READ_SEND_BUFFER, NULL, 0, buf, SEND_BUFFER_LEN);

        if (NT_SUCCESS(Status)) {
            if (!WriteFile(stream, buf, iosb.Information, NULL, NULL))
                ShowSendError(IDS_SEND_WRITEFILE_FAILED, GetLastError(), format_message(GetLastError()).c_str());
        }
    } while (NT_SUCCESS(Status));

    if (Status != STATUS_END_OF_FILE) {
        ShowSendError(IDS_SEND_FSCTL_BTRFS_READ_SEND_BUFFER_FAILED, Status, format_ntstatus(Status).c_str());
        goto end;
    }

    end.length = 0;
    end.cmd = BTRFS_SEND_CMD_END;
    end.csum = 0x9dc96c50;

    if (!WriteFile(stream, &end, sizeof(end), NULL, NULL)) {
        ShowSendError(IDS_SEND_WRITEFILE_FAILED, GetLastError(), format_message(GetLastError()).c_str());
        goto end;
    }

    SetEndOfFile(stream);

    ShowSendError(IDS_SEND_SUCCESS);
    success = TRUE;

end:
    if (!success) {
        FILE_DISPOSITION_INFO fdi;

        fdi.DeleteFile = TRUE;

        SetFileInformationByHandle(stream, FileDispositionInfo, &fdi, sizeof(FILE_DISPOSITION_INFO));
    }

    CloseHandle(stream);
    stream = INVALID_HANDLE_VALUE;

end2:
    CloseHandle(dirh);
    dirh = INVALID_HANDLE_VALUE;

end3:
    free(buf);
    buf = NULL;

    started = FALSE;

    SetDlgItemTextW(hwnd, IDCANCEL, closetext);
    EnableWindow(GetDlgItem(hwnd, IDOK), TRUE);
    EnableWindow(GetDlgItem(hwnd, IDC_STREAM_DEST), TRUE);
    EnableWindow(GetDlgItem(hwnd, IDC_BROWSE), TRUE);

    return 0;
}

static DWORD WINAPI send_thread(LPVOID lpParameter) {
    BtrfsSend* bs = (BtrfsSend*)lpParameter;

    return bs->Thread();
}

void BtrfsSend::StartSend(HWND hwnd) {
    WCHAR s[255];
    HWND cl;
    ULONG num_clones;

    if (started)
        return;

    GetDlgItemTextW(hwnd, IDC_STREAM_DEST, file, sizeof(file) / sizeof(WCHAR));

    if (file[0] == 0)
        return;

    if (incremental) {
        WCHAR parent[MAX_PATH];

        GetDlgItemTextW(hwnd, IDC_PARENT_SUBVOL, parent, sizeof(parent) / sizeof(WCHAR));

        if (parent[0] == 0)
            return;
    }

    started = TRUE;
    ShowSendError(IDS_SEND_WRITING);

    LoadStringW(module, IDS_SEND_CANCEL, s, sizeof(s) / sizeof(WCHAR));
    SetDlgItemTextW(hwnd, IDCANCEL, s);

    EnableWindow(GetDlgItem(hwnd, IDOK), FALSE);
    EnableWindow(GetDlgItem(hwnd, IDC_STREAM_DEST), FALSE);
    EnableWindow(GetDlgItem(hwnd, IDC_BROWSE), FALSE);

    clones.clear();

    cl = GetDlgItem(hwnd, IDC_CLONE_LIST);
    num_clones = SendMessageW(cl, LB_GETCOUNT, 0, 0);

    if ((LRESULT)num_clones != LB_ERR) {
        ULONG i;

        for (i = 0; i < num_clones; i++) {
            WCHAR* s;
            ULONG len;

            len = SendMessageW(cl, LB_GETTEXTLEN, i, 0);
            s = (WCHAR*)malloc((len + 1) * sizeof(WCHAR));

            SendMessageW(cl, LB_GETTEXT, i, (LPARAM)s);

            clones.push_back(s);

            free(s);
        }
    }

    thread = CreateThread(NULL, 0, send_thread, this, 0, NULL);

    if (!thread)
        ShowError(NULL, GetLastError());
}

void BtrfsSend::Browse(HWND hwnd) {
    OPENFILENAMEW ofn;

    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);

    if (!GetSaveFileNameW(&ofn))
        return;

    SetDlgItemTextW(hwnd, IDC_STREAM_DEST, file);
}

void BtrfsSend::BrowseParent(HWND hwnd) {
    BROWSEINFOW bi;
    PIDLIST_ABSOLUTE root, pidl;
    HRESULT hr;
    WCHAR parent[MAX_PATH], volpathw[MAX_PATH];
    HANDLE h;
    NTSTATUS Status;
    IO_STATUS_BLOCK iosb;
    btrfs_get_file_ids bgfi;

    if (!GetVolumePathNameW(subvol.c_str(), volpathw, (sizeof(volpathw) / sizeof(WCHAR)) - 1)) {
        ShowStringError(hwnd, IDS_RECV_GETVOLUMEPATHNAME_FAILED, GetLastError(), format_message(GetLastError()).c_str());
        return;
    }

    hr = SHParseDisplayName(volpathw, 0, &root, 0, 0);
    if (FAILED(hr)) {
        ShowStringError(hwnd, IDS_SHPARSEDISPLAYNAME_FAILED);
        return;
    }

    memset(&bi, 0, sizeof(BROWSEINFOW));

    bi.hwndOwner = hwnd;
    bi.pidlRoot = root;
    bi.ulFlags = BIF_RETURNONLYFSDIRS | BIF_USENEWUI | BIF_NONEWFOLDERBUTTON;

    pidl = SHBrowseForFolderW(&bi);

    if (!pidl)
        return;

    if (!SHGetPathFromIDListW(pidl, parent)) {
        ShowStringError(hwnd, IDS_SHGETPATHFROMIDLIST_FAILED);
        return;
    }

    h = CreateFileW(parent, FILE_TRAVERSE, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, NULL, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL);
    if (h == INVALID_HANDLE_VALUE) {
        ShowStringError(hwnd, IDS_SEND_CANT_OPEN_DIR, parent, GetLastError(), format_message(GetLastError()).c_str());
        return;
    }

    Status = NtFsControlFile(h, NULL, NULL, NULL, &iosb, FSCTL_BTRFS_GET_FILE_IDS, NULL, 0, &bgfi, sizeof(btrfs_get_file_ids));
    if (!NT_SUCCESS(Status)) {
        ShowStringError(hwnd, IDS_GET_FILE_IDS_FAILED, Status, format_ntstatus(Status).c_str());
        CloseHandle(h);
        return;
    }

    CloseHandle(h);

    if (bgfi.inode != 0x100 || bgfi.top) {
        ShowStringError(hwnd, IDS_NOT_SUBVOL);
        return;
    }

    SetDlgItemTextW(hwnd, IDC_PARENT_SUBVOL, parent);
}

void BtrfsSend::AddClone(HWND hwnd) {
    BROWSEINFOW bi;
    PIDLIST_ABSOLUTE root, pidl;
    HRESULT hr;
    WCHAR path[MAX_PATH], volpathw[MAX_PATH];
    HANDLE h;
    NTSTATUS Status;
    IO_STATUS_BLOCK iosb;
    btrfs_get_file_ids bgfi;

    if (!GetVolumePathNameW(subvol.c_str(), volpathw, (sizeof(volpathw) / sizeof(WCHAR)) - 1)) {
        ShowStringError(hwnd, IDS_RECV_GETVOLUMEPATHNAME_FAILED, GetLastError(), format_message(GetLastError()).c_str());
        return;
    }

    hr = SHParseDisplayName(volpathw, 0, &root, 0, 0);
    if (FAILED(hr)) {
        ShowStringError(hwnd, IDS_SHPARSEDISPLAYNAME_FAILED);
        return;
    }

    memset(&bi, 0, sizeof(BROWSEINFOW));

    bi.hwndOwner = hwnd;
    bi.pidlRoot = root;
    bi.ulFlags = BIF_RETURNONLYFSDIRS | BIF_USENEWUI | BIF_NONEWFOLDERBUTTON;

    pidl = SHBrowseForFolderW(&bi);

    if (!pidl)
        return;

    if (!SHGetPathFromIDListW(pidl, path)) {
        ShowStringError(hwnd, IDS_SHGETPATHFROMIDLIST_FAILED);
        return;
    }

    h = CreateFileW(path, FILE_TRAVERSE, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, NULL, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL);
    if (h == INVALID_HANDLE_VALUE) {
        ShowStringError(hwnd, IDS_SEND_CANT_OPEN_DIR, path, GetLastError(), format_message(GetLastError()).c_str());
        return;
    }

    Status = NtFsControlFile(h, NULL, NULL, NULL, &iosb, FSCTL_BTRFS_GET_FILE_IDS, NULL, 0, &bgfi, sizeof(btrfs_get_file_ids));
    if (!NT_SUCCESS(Status)) {
        ShowStringError(hwnd, IDS_GET_FILE_IDS_FAILED, Status, format_ntstatus(Status).c_str());
        CloseHandle(h);
        return;
    }

    CloseHandle(h);

    if (bgfi.inode != 0x100 || bgfi.top) {
        ShowStringError(hwnd, IDS_NOT_SUBVOL);
        return;
    }

    SendMessageW(GetDlgItem(hwnd, IDC_CLONE_LIST), LB_ADDSTRING, 0, (LPARAM)path);
}

void BtrfsSend::RemoveClone(HWND hwnd) {
    LRESULT sel;
    HWND cl = GetDlgItem(hwnd, IDC_CLONE_LIST);

    sel = SendMessageW(cl, LB_GETCURSEL, 0, 0);

    if (sel == LB_ERR)
        return;

    SendMessageW(cl, LB_DELETESTRING, sel, 0);

    if (SendMessageW(cl, LB_GETCURSEL, 0, 0) == LB_ERR)
        EnableWindow(GetDlgItem(hwnd, IDC_CLONE_REMOVE), FALSE);
}

INT_PTR BtrfsSend::SendDlgProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam) {
    switch (uMsg) {
        case WM_INITDIALOG:
            this->hwnd = hwndDlg;

            GetDlgItemTextW(hwndDlg, IDCANCEL, closetext, sizeof(closetext) / sizeof(WCHAR));
        break;

        case WM_COMMAND:
            switch (HIWORD(wParam)) {
                case BN_CLICKED:
                    switch (LOWORD(wParam)) {
                        case IDOK:
                            StartSend(hwndDlg);
                        return TRUE;

                        case IDCANCEL:
                            if (started) {
                                TerminateThread(thread, 0);

                                if (stream != INVALID_HANDLE_VALUE) {
                                    FILE_DISPOSITION_INFO fdi;

                                    fdi.DeleteFile = TRUE;

                                    SetFileInformationByHandle(stream, FileDispositionInfo, &fdi, sizeof(FILE_DISPOSITION_INFO));
                                    CloseHandle(stream);
                                }

                                if (dirh != INVALID_HANDLE_VALUE)
                                    CloseHandle(dirh);

                                started = FALSE;

                                SetDlgItemTextW(hwndDlg, IDCANCEL, closetext);

                                EnableWindow(GetDlgItem(hwnd, IDOK), TRUE);
                                EnableWindow(GetDlgItem(hwnd, IDC_STREAM_DEST), TRUE);
                                EnableWindow(GetDlgItem(hwnd, IDC_BROWSE), TRUE);
                            } else
                                EndDialog(hwndDlg, 1);
                        return TRUE;

                        case IDC_BROWSE:
                            Browse(hwndDlg);
                        return TRUE;

                        case IDC_INCREMENTAL:
                            incremental = IsDlgButtonChecked(hwndDlg, LOWORD(wParam));

                            EnableWindow(GetDlgItem(hwnd, IDC_PARENT_SUBVOL), incremental);
                            EnableWindow(GetDlgItem(hwnd, IDC_PARENT_BROWSE), incremental);
                        return TRUE;

                        case IDC_PARENT_BROWSE:
                            BrowseParent(hwndDlg);
                        return TRUE;

                        case IDC_CLONE_ADD:
                            AddClone(hwndDlg);
                        return TRUE;

                        case IDC_CLONE_REMOVE:
                            RemoveClone(hwndDlg);
                        return TRUE;
                    }
                break;

                case LBN_SELCHANGE:
                    switch (LOWORD(wParam)) {
                        case IDC_CLONE_LIST:
                            EnableWindow(GetDlgItem(hwnd, IDC_CLONE_REMOVE), TRUE);
                        return TRUE;
                    }
                break;
            }
        break;
    }

    return FALSE;
}

static INT_PTR CALLBACK stub_SendDlgProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam) {
    BtrfsSend* bs;

    if (uMsg == WM_INITDIALOG) {
        SetWindowLongPtr(hwndDlg, GWLP_USERDATA, (LONG_PTR)lParam);
        bs = (BtrfsSend*)lParam;
    } else
        bs = (BtrfsSend*)GetWindowLongPtr(hwndDlg, GWLP_USERDATA);

    if (bs)
        return bs->SendDlgProc(hwndDlg, uMsg, wParam, lParam);
    else
        return FALSE;
}

void BtrfsSend::Open(HWND hwnd, LPWSTR path) {
    subvol = path;

    if (DialogBoxParamW(module, MAKEINTRESOURCEW(IDD_SEND_SUBVOL), hwnd, stub_SendDlgProc, (LPARAM)this) <= 0)
        ShowError(hwnd, GetLastError());
}

void CALLBACK SendSubvolGUIW(HWND hwnd, HINSTANCE hinst, LPWSTR lpszCmdLine, int nCmdShow) {
    HANDLE token;
    TOKEN_PRIVILEGES tp;
    LUID luid;
    BtrfsSend* bs;

    set_dpi_aware();

    if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &token)) {
        ShowError(hwnd, GetLastError());
        return;
    }

    if (!LookupPrivilegeValueW(NULL, L"SeManageVolumePrivilege", &luid)) {
        ShowError(hwnd, GetLastError());
        CloseHandle(token);
        return;
    }

    tp.PrivilegeCount = 1;
    tp.Privileges[0].Luid = luid;
    tp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;

    if (!AdjustTokenPrivileges(token, FALSE, &tp, sizeof(TOKEN_PRIVILEGES), NULL, NULL)) {
        ShowError(hwnd, GetLastError());
        CloseHandle(token);
        return;
    }

    bs = new BtrfsSend;

    bs->Open(hwnd, lpszCmdLine);

    delete bs;

    CloseHandle(token);
}

static void send_subvol(std::wstring subvol, std::wstring file, std::wstring parent, std::vector<std::wstring> clones) {
    char* buf;
    HANDLE dirh, stream;
    ULONG bss_size, i;
    btrfs_send_subvol* bss;
    IO_STATUS_BLOCK iosb;
    NTSTATUS Status;
    btrfs_send_header header;
    btrfs_send_command end;
    BOOL success = FALSE;

    buf = (char*)malloc(SEND_BUFFER_LEN);

    dirh = CreateFileW(subvol.c_str(), FILE_READ_ATTRIBUTES, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, NULL, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL);
    if (dirh == INVALID_HANDLE_VALUE)
        goto end3;

    stream = CreateFileW(file.c_str(), FILE_WRITE_DATA | DELETE, 0, NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
    if (stream == INVALID_HANDLE_VALUE) {
        CloseHandle(dirh);
        goto end3;
    }

    bss_size = offsetof(btrfs_send_subvol, clones[0]) + (clones.size() * sizeof(HANDLE));
    bss = (btrfs_send_subvol*)malloc(bss_size);
    memset(bss, 0, bss_size);

    if (parent != L"") {
        HANDLE parenth;

        parenth = CreateFileW(parent.c_str(), FILE_READ_ATTRIBUTES, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, NULL, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL);
        if (parenth == INVALID_HANDLE_VALUE)
            goto end2;

        bss->parent = parenth;
    } else
        bss->parent = NULL;

    bss->num_clones = clones.size();

    for (i = 0; i < bss->num_clones; i++) {
        HANDLE h;

        h = CreateFileW(clones[i].c_str(), FILE_READ_ATTRIBUTES, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, NULL, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL);
        if (h == INVALID_HANDLE_VALUE) {
            ULONG j;

            for (j = 0; j < i; j++) {
                CloseHandle(bss->clones[j]);
            }

            if (bss->parent) CloseHandle(bss->parent);
            goto end2;
        }

        bss->clones[i] = h;
    }

    Status = NtFsControlFile(dirh, NULL, NULL, NULL, &iosb, FSCTL_BTRFS_SEND_SUBVOL, bss, bss_size, NULL, 0);

    for (i = 0; i < bss->num_clones; i++) {
        CloseHandle(bss->clones[i]);
    }

    if (bss->parent) CloseHandle(bss->parent);

    if (!NT_SUCCESS(Status))
        goto end2;

    memcpy(header.magic, BTRFS_SEND_MAGIC, sizeof(header.magic));
    header.version = 1;

    if (!WriteFile(stream, &header, sizeof(header), NULL, NULL))
        goto end2;

    do {
        Status = NtFsControlFile(dirh, NULL, NULL, NULL, &iosb, FSCTL_BTRFS_READ_SEND_BUFFER, NULL, 0, buf, SEND_BUFFER_LEN);

        if (NT_SUCCESS(Status))
            WriteFile(stream, buf, iosb.Information, NULL, NULL);
    } while (NT_SUCCESS(Status));

    if (Status != STATUS_END_OF_FILE)
        goto end2;

    end.length = 0;
    end.cmd = BTRFS_SEND_CMD_END;
    end.csum = 0x9dc96c50;

    if (!WriteFile(stream, &end, sizeof(end), NULL, NULL))
        goto end2;

    SetEndOfFile(stream);

    success = TRUE;

end2:
    if (!success) {
        FILE_DISPOSITION_INFO fdi;

        fdi.DeleteFile = TRUE;

        SetFileInformationByHandle(stream, FileDispositionInfo, &fdi, sizeof(FILE_DISPOSITION_INFO));
    }

    CloseHandle(dirh);
    CloseHandle(stream);

end3:
    free(buf);
}

void CALLBACK SendSubvolW(HWND hwnd, HINSTANCE hinst, LPWSTR lpszCmdLine, int nCmdShow) {
    LPWSTR* args;
    int num_args;
    std::wstring subvol = L"", parent = L"", file = L"";
    std::vector<std::wstring> clones;

    args = CommandLineToArgvW(lpszCmdLine, &num_args);

    if (!args)
        return;

    if (num_args >= 2) {
        HANDLE token;
        TOKEN_PRIVILEGES tp;
        LUID luid;
        int i;

        if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &token))
            goto end;

        if (!LookupPrivilegeValueW(NULL, L"SeManageVolumePrivilege", &luid))
            goto end;

        tp.PrivilegeCount = 1;
        tp.Privileges[0].Luid = luid;
        tp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;

        if (!AdjustTokenPrivileges(token, FALSE, &tp, sizeof(TOKEN_PRIVILEGES), NULL, NULL))
            goto end;

        CloseHandle(token);

        for (i = 0; i < num_args; i++) {
            if (args[i][0] == '-') {
                if (args[i][2] == 0 && i < num_args - 1) {
                    if (args[i][1] == 'p') {
                        parent = args[i+1];
                        i++;
                    } else if (args[i][1] == 'c') {
                        clones.push_back(args[i+1]);
                        i++;
                    }
                }
            } else {
                if (subvol == L"")
                    subvol = args[i];
                else if (file == L"")
                    file = args[i];
            }
        }

        if (subvol != L"" && file != L"")
            send_subvol(subvol, file, parent, clones);
    }

end:
    LocalFree(args);
}