utils/libnt_path_converter/src/device.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

/**
 * \file
 * \author Egor Tensin <Egor.Tensin@gmail.com>
 * \date 2015
 * \copyright This file is licensed under the terms of the MIT License.
 *            See LICENSE.txt for details.
 */

#include "libnt_path_converter/device.hpp"

#include "libservice/all.hpp"

#include <Windows.h>

#include <string>
#include <vector>

namespace libnt_path_converter
{
    namespace
    {
        const char* const device_path = "\\\\.\\nt_path_converter";
        const auto control_code = CTL_CODE(0x8000, 0x800, METHOD_BUFFERED, FILE_ANY_ACCESS);
    }

    Device::Device()
        : libservice::Device(libservice::Device::open(device_path))
    { }

    std::wstring Device::convert_nt_path(const std::wstring& src)
    {
        const auto in_buf = src.c_str();
        const auto in_buf_size = (src.size() + 1) * sizeof(wchar_t);

        const auto nbreq = get_required_output_size(
            control_code,
            in_buf,
            in_buf_size);

        std::vector<unsigned char> output(nbreq);

        send_control_code(
            control_code,
            in_buf,
            in_buf_size,
            output.data(),
            nbreq);

        return reinterpret_cast<wchar_t*>(output.data());
    }
}