forked from wyrover/book-code
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathpipe.cpp
More file actions
108 lines (85 loc) · 2.66 KB
/
Copy pathpipe.cpp
File metadata and controls
108 lines (85 loc) · 2.66 KB
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
#include "pipe.h"
namespace cactus
{
pipe::pipe(std::wstring pipeName, std::wstring serveur) : hPipe(INVALID_HANDLE_VALUE), pipePath(L"\\\\")
{
pipePath.append(serveur);
pipePath.append(L"\\pipe\\");
pipePath.append(pipeName);
}
pipe::~pipe(void)
{
closePipe();
}
bool pipe::closePipe()
{
bool reussite = false;
if (hPipe != INVALID_HANDLE_VALUE && hPipe) {
FlushFileBuffers(hPipe);
DisconnectNamedPipe(hPipe);
reussite = CloseHandle(hPipe) == TRUE;
}
return reussite;
}
bool pipe::readFromPipe(std::wstring &laReponse)
{
bool reussite = false;
wchar_t monBuffer[128];
bool fSuccess;
DWORD longueurReponse;
laReponse.clear();
do {
fSuccess = ReadFile(hPipe, monBuffer, sizeof(monBuffer), &longueurReponse, NULL) ? true : false;
if (reussite = (fSuccess || GetLastError() == ERROR_MORE_DATA)/* && longueurReponse != 0 */) {
laReponse.append(monBuffer, longueurReponse / sizeof(wchar_t));
} else {
break;
}
} while (!fSuccess);
return reussite;
}
bool pipe::writeToPipe(const std::wstring &leMessage)
{
bool reussite = false;
DWORD longueurMessage;
DWORD longueurOctetsEcris;
longueurMessage = (static_cast<DWORD>(leMessage.size())) * sizeof(wchar_t);
if (WriteFile(hPipe, leMessage.c_str(), longueurMessage, &longueurOctetsEcris, NULL) && longueurMessage == longueurOctetsEcris) {
reussite = FlushFileBuffers(hPipe) != 0;
}
return reussite;
}
bool pipe::createServer()
{
bool reussite = false;
if (!hPipe || hPipe == INVALID_HANDLE_VALUE) {
hPipe = CreateNamedPipe(pipePath.c_str(), PIPE_ACCESS_DUPLEX, PIPE_TYPE_MESSAGE | PIPE_READMODE_MESSAGE | PIPE_WAIT, 1, 0, 0, 30000, NULL);
if (hPipe && hPipe != INVALID_HANDLE_VALUE) {
reussite = ConnectNamedPipe(hPipe, NULL) ? TRUE : (GetLastError() == ERROR_PIPE_CONNECTED);
} else {
closePipe();
}
}
return reussite;
}
bool pipe::createClient()
{
bool reussite = false;
if (!hPipe || hPipe == INVALID_HANDLE_VALUE) {
if (WaitNamedPipe(pipePath.c_str(), NMPWAIT_USE_DEFAULT_WAIT)) {
hPipe = CreateFile(pipePath.c_str(), GENERIC_READ | GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, NULL);
if (hPipe != INVALID_HANDLE_VALUE) {
DWORD dwMode = PIPE_READMODE_MESSAGE | PIPE_WAIT;
if (!(reussite = SetNamedPipeHandleState(hPipe, &dwMode, NULL, NULL) != 0)) {
closePipe();
}
}
}
}
return reussite;
}
bool pipe::isConnected()
{
return (hPipe && hPipe != INVALID_HANDLE_VALUE);
}
}