6 |
|
|
7 |
|
#include <windows.h> |
8 |
|
#include <tchar.h> |
9 |
+ |
#include <signal.h> |
10 |
+ |
#include <wtsapi32.h> |
11 |
|
|
12 |
|
#include <iostream> |
13 |
+ |
#include <string> |
14 |
+ |
#include <sstream> |
15 |
+ |
#include <vector> |
16 |
+ |
|
17 |
+ |
inline std::string Utf8(const std::wstring &wstring) |
18 |
+ |
{ |
19 |
+ |
std::string string(WideCharToMultiByte(CP_UTF8, 0, wstring.data(), int(wstring.size()), NULL, 0, NULL, NULL), '\0'); |
20 |
+ |
|
21 |
+ |
WideCharToMultiByte(CP_UTF8, 0, wstring.data(), int(wstring.size()), const_cast<LPSTR>(string.data()), int(string.size()), NULL, NULL); |
22 |
+ |
|
23 |
+ |
return string; |
24 |
+ |
} |
25 |
+ |
|
26 |
+ |
class Finger |
27 |
+ |
{ |
28 |
+ |
private: |
29 |
+ |
std::ostringstream stream; |
30 |
+ |
|
31 |
+ |
public: |
32 |
+ |
Finger(bool full) |
33 |
+ |
{ |
34 |
+ |
if (!full) |
35 |
+ |
stream << "Login\t\tName\t\tTTY\r\n"; |
36 |
+ |
|
37 |
+ |
PWTS_SESSION_INFO sessions; |
38 |
+ |
DWORD count; |
39 |
+ |
|
40 |
+ |
WTSEnumerateSessions(WTS_CURRENT_SERVER_HANDLE, 0, 1, &sessions, &count); |
41 |
+ |
|
42 |
+ |
for (PWTS_SESSION_INFO session = sessions; session != sessions + count; ++session) |
43 |
+ |
{ |
44 |
+ |
LPTSTR name; |
45 |
+ |
DWORD size; |
46 |
+ |
|
47 |
+ |
WTSQuerySessionInformation(WTS_CURRENT_SERVER_HANDLE, session->SessionId, WTSUserName, &name, &size); |
48 |
+ |
|
49 |
+ |
std::string name_(Utf8(name)); |
50 |
+ |
|
51 |
+ |
if (!name_.empty()) |
52 |
+ |
stream << name_ << "\r\n"; |
53 |
+ |
|
54 |
+ |
WTSFreeMemory(name); |
55 |
+ |
|
56 |
+ |
//PWTS_CLIENT_ADDRESS address; |
57 |
+ |
//WTSQuerySessionInformation(WTS_CURRENT_SERVER_HANDLE, session->SessionId, WTSClientAddress, reinterpret_cast<LPTSTR *>(&address), &size); |
58 |
+ |
//WTSFreeMemory(address); |
59 |
+ |
} |
60 |
+ |
|
61 |
+ |
WTSFreeMemory(sessions); |
62 |
+ |
} |
63 |
+ |
|
64 |
+ |
Finger(const std::string &name) |
65 |
+ |
{ |
66 |
+ |
} |
67 |
+ |
|
68 |
+ |
inline operator std::string() |
69 |
+ |
{ |
70 |
+ |
return stream.str(); |
71 |
+ |
} |
72 |
+ |
}; |
73 |
|
|
74 |
|
LPTSTR name; |
75 |
|
SERVICE_STATUS status; |
76 |
|
SERVICE_STATUS_HANDLE handle; |
77 |
|
HANDLE stop; |
78 |
|
WSADATA data; |
79 |
+ |
std::vector<HANDLE> threads; |
80 |
|
|
81 |
+ |
void FingerDaemon(); |
82 |
+ |
DWORD WINAPI FingerListen(LPVOID server_); |
83 |
+ |
DWORD WINAPI FingerDo(LPVOID client_); |
84 |
+ |
void FingerStop(int); |
85 |
|
void WINAPI FingerMain(DWORD argc, LPTSTR *argv); |
86 |
|
void WINAPI FingerControl(DWORD control); |
87 |
|
|
118 |
|
return 0; |
119 |
|
|
120 |
|
go: |
121 |
< |
std::cout << "Here!" << std::endl; |
121 |
> |
stop = CreateEvent(NULL, TRUE, FALSE, NULL); |
122 |
> |
|
123 |
> |
signal(SIGINT, FingerStop); |
124 |
> |
|
125 |
> |
try |
126 |
> |
{ |
127 |
> |
FingerDaemon(); |
128 |
> |
} |
129 |
> |
catch (...) {} |
130 |
> |
|
131 |
> |
return 0; |
132 |
> |
} |
133 |
> |
|
134 |
> |
void FingerDaemon() |
135 |
> |
{ |
136 |
> |
WSAStartup(MAKEWORD(2, 0), &data); |
137 |
> |
|
138 |
> |
// |
139 |
> |
|
140 |
> |
SOCKET server(socket(PF_INET, SOCK_STREAM, IPPROTO_TCP)); |
141 |
> |
SOCKADDR_IN service; |
142 |
> |
|
143 |
> |
service.sin_family = AF_INET; |
144 |
> |
service.sin_addr.s_addr = inet_addr("0.0.0.0"); |
145 |
> |
service.sin_port = htons(79); |
146 |
> |
|
147 |
> |
bind(server, (SOCKADDR *)(&service), sizeof (service)); |
148 |
> |
listen(server, SOMAXCONN); |
149 |
> |
|
150 |
> |
threads.push_back(CreateThread(NULL, 0, FingerListen, &server, 0, NULL)); |
151 |
> |
|
152 |
> |
while(WaitForSingleObject(stop, 1000) != WAIT_OBJECT_0); |
153 |
> |
|
154 |
> |
// |
155 |
> |
|
156 |
> |
closesocket(server); |
157 |
> |
|
158 |
> |
WSACleanup(); |
159 |
> |
} |
160 |
> |
|
161 |
> |
DWORD WINAPI FingerListen(LPVOID server_) |
162 |
> |
{ |
163 |
> |
SOCKET &server(*reinterpret_cast<SOCKET *>(server_)); |
164 |
> |
|
165 |
> |
while (true) |
166 |
> |
{ |
167 |
> |
SOCKET client(accept(server, NULL, NULL)); |
168 |
> |
|
169 |
> |
threads.push_back(CreateThread(NULL, 0, FingerDo, &client, 0, NULL)); |
170 |
> |
} |
171 |
> |
|
172 |
> |
return 0; |
173 |
> |
} |
174 |
> |
|
175 |
> |
DWORD WINAPI FingerDo(LPVOID client_) |
176 |
> |
{ |
177 |
> |
SOCKET &client(*reinterpret_cast<SOCKET *>(client_)); |
178 |
> |
char buffer[1024]; |
179 |
> |
std::istringstream stream(std::string(buffer, recv(client, buffer, sizeof buffer, 0))); |
180 |
> |
std::string line; |
181 |
> |
|
182 |
> |
std::getline(stream, line); |
183 |
> |
|
184 |
> |
stream.str(line); |
185 |
> |
|
186 |
> |
std::getline(stream, line, '\r'); |
187 |
> |
|
188 |
> |
stream.str(line); |
189 |
> |
|
190 |
> |
std::string name; |
191 |
> |
bool full(false); |
192 |
> |
|
193 |
> |
while (stream >> line) |
194 |
> |
if (line == "/W") |
195 |
> |
full = true; |
196 |
> |
else |
197 |
> |
name = line; |
198 |
> |
|
199 |
> |
std::string finger(name.empty() ? Finger(full) : Finger(name)); |
200 |
> |
|
201 |
> |
send(client, finger.data(), int(finger.size()), 0); |
202 |
> |
closesocket(client); |
203 |
|
|
204 |
|
return 0; |
205 |
|
} |
206 |
|
|
207 |
+ |
void FingerStop(int) |
208 |
+ |
{ |
209 |
+ |
SetEvent(stop); |
210 |
+ |
} |
211 |
+ |
|
212 |
|
void WINAPI FingerMain(DWORD argc, LPTSTR *argv) |
213 |
|
{ |
214 |
|
name = argv[0]; |
217 |
|
|
218 |
|
// |
219 |
|
|
67 |
– |
WSAStartup(MAKEWORD(2, 0), &data); |
68 |
– |
|
69 |
– |
// |
70 |
– |
|
220 |
|
status.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN; |
221 |
|
status.dwServiceType = SERVICE_WIN32_OWN_PROCESS; |
222 |
|
status.dwServiceSpecificExitCode = 0; |
227 |
|
|
228 |
|
SetServiceStatus(handle, &status); |
229 |
|
|
230 |
< |
SOCKET sock(socket(PF_INET, SOCK_STREAM, IPPROTO_TCP)); |
82 |
< |
SOCKADDR_IN service; |
83 |
< |
|
84 |
< |
service.sin_family = AF_INET; |
85 |
< |
service.sin_addr.s_addr = inet_addr("0.0.0.0"); |
86 |
< |
service.sin_port = htons(79); |
87 |
< |
|
88 |
< |
bind(sock, (SOCKADDR *)(&service), sizeof (service)); |
89 |
< |
listen(sock, SOMAXCONN); |
90 |
< |
|
91 |
< |
while(WaitForSingleObject(stop, 1000) != WAIT_OBJECT_0 /*&&*/) |
230 |
> |
try |
231 |
|
{ |
232 |
< |
// |
232 |
> |
FingerDaemon(); |
233 |
|
} |
234 |
< |
|
96 |
< |
// |
97 |
< |
|
98 |
< |
WSACleanup(); |
234 |
> |
catch (...) {} |
235 |
|
|
236 |
|
status.dwCurrentState = SERVICE_STOPPED; |
237 |
|
status.dwWin32ExitCode = NO_ERROR; |