aboutsummaryrefslogtreecommitdiff
path: root/Exfiltration/mimikatz-1.0/modules/mod_pipe.cpp
blob: bd62e9ec4e64c5306b4d36c14d682e6549d19ac9 (plain)
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
117
118
119
120
121
/*	Benjamin DELPY `gentilkiwi`
	http://blog.gentilkiwi.com
	benjamin@gentilkiwi.com
	Licence : http://creativecommons.org/licenses/by/3.0/fr/
*/
#include "mod_pipe.h"

mod_pipe::mod_pipe(wstring pipeName, wstring serveur) : hPipe(INVALID_HANDLE_VALUE), pipePath(L"\\\\")
{
	pipePath.append(serveur);
	pipePath.append(L"\\pipe\\");
	pipePath.append(pipeName);
}

mod_pipe::~mod_pipe(void)
{
	closePipe();
}

bool mod_pipe::closePipe()
{
	bool reussite = false;
	
	if(hPipe != INVALID_HANDLE_VALUE && hPipe)
	{
		FlushFileBuffers(hPipe); 
		DisconnectNamedPipe(hPipe);
		reussite = CloseHandle(hPipe) == TRUE;
	}
	return reussite;
}

bool mod_pipe::readFromPipe(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 mod_pipe::writeToPipe(const 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 mod_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 mod_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 mod_pipe::isConnected()
{
	return (hPipe && hPipe != INVALID_HANDLE_VALUE);
}