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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
// license:BSD-3-Clause
// copyright-holders:Aaron Giles, Vas Crabb
#include "winfile.h"
#include "strconv.h"
#include "winutil.h"
#include <cassert>
#include <windows.h>
#include <stdlib.h>
namespace {
char const *const winfile_ptty_identifier = "\\\\.\\pipe\\";
class win_osd_ptty : public osd_file
{
public:
win_osd_ptty(win_osd_ptty const &) = delete;
win_osd_ptty(win_osd_ptty &&) = delete;
win_osd_ptty& operator=(win_osd_ptty const &) = delete;
win_osd_ptty& operator=(win_osd_ptty &&) = delete;
win_osd_ptty(HANDLE handle) : m_handle(handle)
{
assert(m_handle);
assert(INVALID_HANDLE_VALUE != m_handle);
}
~win_osd_ptty()
{
FlushFileBuffers(m_handle);
DisconnectNamedPipe(m_handle);
CloseHandle(m_handle);
}
virtual error read(void *buffer, std::uint64_t offset, std::uint32_t count, std::uint32_t &actual) override
{
DWORD bytes_read;
if (!ReadFile(m_handle, buffer, count, &bytes_read, nullptr))
return win_error_to_file_error(GetLastError());
actual = bytes_read;
return error::NONE;
}
virtual error write(void const *buffer, std::uint64_t offset, std::uint32_t count, std::uint32_t &actual) override
{
DWORD bytes_written;
if (!WriteFile(m_handle, buffer, count, &bytes_written, nullptr))
return win_error_to_file_error(GetLastError());
actual = bytes_written;
return error::NONE;
}
virtual error truncate(std::uint64_t offset) override
{
// doesn't make sense for a PTTY
return error::INVALID_ACCESS;
}
virtual error flush() override
{
// don't want to wait for client to read all data as implied by FlushFileBuffers
return error::NONE;
}
private:
HANDLE m_handle;
};
} // anonymous namespace
bool win_check_ptty_path(std::string const &path)
{
if (strncmp(path.c_str(), winfile_ptty_identifier, strlen(winfile_ptty_identifier)) == 0) return true;
return false;
}
osd_file::error win_open_ptty(std::string const &path, std::uint32_t openflags, osd_file::ptr &file, std::uint64_t &filesize)
{
osd::text::tstring t_name = osd::text::to_tstring(path);
HANDLE pipe = CreateFileW(t_name.c_str(), GENERIC_READ | GENERIC_WRITE, 0, nullptr, OPEN_EXISTING, 0, nullptr);
if (INVALID_HANDLE_VALUE == pipe)
{
pipe = CreateNamedPipe(t_name.c_str(), PIPE_ACCESS_DUPLEX, PIPE_TYPE_BYTE | PIPE_READMODE_BYTE | PIPE_NOWAIT, 1, 32, 32, 0, nullptr);
if (INVALID_HANDLE_VALUE == pipe)
return osd_file::error::ACCESS_DENIED;
}
else
{
DWORD state = PIPE_NOWAIT;
SetNamedPipeHandleState(pipe, &state, NULL, NULL);
}
try
{
file = std::make_unique<win_osd_ptty>(pipe);
filesize = 0;
return osd_file::error::NONE;
}
catch (...)
{
CloseHandle(pipe);
return osd_file::error::OUT_OF_MEMORY;
}
}
|