2 * Copyright 2011-2013 André Hentschel
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with this library; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19 #define NONAMELESSUNION
23 #include "wine/unicode.h"
24 #include "wine/debug.h"
26 WINE_DEFAULT_DEBUG_CHANNEL(netstat
);
28 static const WCHAR ipW
[] = {'I', 'P', 0};
29 static const WCHAR ipv6W
[] = {'I', 'P', 'v', '6', 0};
30 static const WCHAR icmpW
[] = {'I', 'C', 'M', 'P', 0};
31 static const WCHAR icmpv6W
[] = {'I', 'C', 'M', 'P', 'v', '6', 0};
32 static const WCHAR tcpW
[] = {'T', 'C', 'P', 0};
33 static const WCHAR tcpv6W
[] = {'T', 'C', 'P', 'v', '6', 0};
34 static const WCHAR udpW
[] = {'U', 'D', 'P', 0};
35 static const WCHAR udpv6W
[] = {'U', 'D', 'P', 'v', '6', 0};
37 static const WCHAR fmtport
[] = {'%', 'd', 0};
38 static const WCHAR fmtip
[] = {'%', 'd', '.', '%', 'd', '.', '%', 'd', '.', '%', 'd', 0};
39 static const WCHAR fmtn
[] = {'\n', 0};
40 static const WCHAR fmtnn
[] = {'\n', '%', 's', '\n', 0};
41 static const WCHAR fmtcolon
[] = {'%', 's', ':', '%', 's', 0};
42 static const WCHAR fmttcpout
[] = {' ', ' ', '%', '-', '6', 's', ' ', '%', '-', '2', '2', 's', ' ', '%', '-', '2', '2', 's', ' ', '%', 's', '\n', 0};
43 static const WCHAR fmtudpout
[] = {' ', ' ', '%', '-', '6', 's', ' ', '%', '-', '2', '2', 's', ' ', '*', ':', '*', '\n', 0};
44 static const WCHAR fmtethout
[] = {'%', '-', '2', '0', 's', ' ', '%', '1', '4', 'l', 'u', ' ', '%', '1', '5', 'l', 'u', '\n', 0};
45 static const WCHAR fmtethoutu
[] = {'%', '-', '2', '0', 's', ' ', '%', '1', '4', 'l', 'u', '\n', '\n', 0};
46 static const WCHAR fmtethheader
[] = {' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
47 ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
48 ' ', '%', '-', '1', '9', 's', ' ', '%', 's', '\n', '\n', 0};
49 static const WCHAR fmttcpstat
[] = {' ', ' ', '%', '-', '3', '5', 's', ' ', '=', ' ', '%', 'l', 'u', '\n', 0};
50 static const WCHAR fmtudpstat
[] = {' ', ' ', '%', '-', '2', '1', 's', ' ', '=', ' ', '%', 'l', 'u', '\n', 0};
52 static const WCHAR tcpstatesW
[][16] = {
54 {'C', 'L', 'O', 'S', 'E', 'D', 0},
55 {'L', 'I', 'S', 'T', 'E', 'N', 'I', 'N', 'G', 0},
56 {'S', 'Y', 'N', '_', 'S', 'E', 'N', 'T', 0},
57 {'S', 'Y', 'N', '_', 'R', 'C', 'V', 'D', 0},
58 {'E', 'S', 'T', 'A', 'B', 'L', 'I', 'S', 'H', 'E', 'D', 0},
59 {'F', 'I', 'N', '_', 'W', 'A', 'I', 'T', '1', 0},
60 {'F', 'I', 'N', '_', 'W', 'A', 'I', 'T', '2', 0},
61 {'C', 'L', 'O', 'S', 'E', '_', 'W', 'A', 'I', 'T', 0},
62 {'C', 'L', 'O', 'S', 'I', 'N', 'G', 0},
63 {'L', 'A', 'S', 'T', '_', 'A', 'C', 'K', 0},
64 {'T', 'I', 'M', 'E', '_', 'W', 'A', 'I', 'T', 0},
65 {'D', 'E', 'L', 'E', 'T', 'E', '_', 'T', 'C', 'B', 0},
68 /* =========================================================================
69 * Output a unicode string. Ideally this will go to the console
70 * and hence required WriteConsoleW to output it, however if file i/o is
71 * redirected, it needs to be WriteFile'd using OEM (not ANSI) format
72 * ========================================================================= */
73 static int __cdecl
NETSTAT_wprintf(const WCHAR
*format
, ...)
75 static WCHAR
*output_bufW
= NULL
;
76 static char *output_bufA
= NULL
;
77 static BOOL toConsole
= TRUE
;
78 static BOOL traceOutput
= FALSE
;
79 #define MAX_WRITECONSOLE_SIZE 65535
87 * Allocate buffer to use when writing to console
88 * Note: Not freed - memory will be allocated once and released when
92 if (!output_bufW
) output_bufW
= HeapAlloc(GetProcessHeap(), 0,
93 MAX_WRITECONSOLE_SIZE
*sizeof(WCHAR
));
95 WINE_FIXME("Out of memory - could not allocate 2 x 64K buffers\n");
99 __ms_va_start(parms
, format
);
100 len
= wvsprintfW(output_bufW
, format
, parms
);
103 /* Try to write as unicode all the time we think its a console */
105 res
= WriteConsoleW(GetStdHandle(STD_OUTPUT_HANDLE
),
106 output_bufW
, len
, &nOut
, NULL
);
109 /* If writing to console has failed (ever) we assume its file
110 i/o so convert to OEM codepage and output */
112 BOOL usedDefaultChar
= FALSE
;
113 DWORD convertedChars
;
118 * Allocate buffer to use when writing to file. Not freed, as above
120 if (!output_bufA
) output_bufA
= HeapAlloc(GetProcessHeap(), 0,
121 MAX_WRITECONSOLE_SIZE
);
123 WINE_FIXME("Out of memory - could not allocate 2 x 64K buffers\n");
127 /* Convert to OEM, then output */
128 convertedChars
= WideCharToMultiByte(GetConsoleOutputCP(), 0, output_bufW
,
129 len
, output_bufA
, MAX_WRITECONSOLE_SIZE
,
130 "?", &usedDefaultChar
);
131 WriteFile(GetStdHandle(STD_OUTPUT_HANDLE
), output_bufA
, convertedChars
,
135 /* Trace whether screen or console */
137 WINE_TRACE("Writing to console? (%d)\n", toConsole
);
143 static WCHAR
*NETSTAT_load_message(UINT id
) {
144 static WCHAR msg
[2048];
145 static const WCHAR failedW
[] = {'F','a','i','l','e','d','!','\0'};
147 if (!LoadStringW(GetModuleHandleW(NULL
), id
, msg
, sizeof(msg
)/sizeof(WCHAR
))) {
148 WINE_FIXME("LoadString failed with %d\n", GetLastError());
149 strcpyW(msg
, failedW
);
154 static WCHAR
*NETSTAT_port_name(UINT port
, WCHAR name
[])
156 /* FIXME: can we get the name? */
157 sprintfW(name
, fmtport
, htons((WORD
)port
));
161 static WCHAR
*NETSTAT_host_name(UINT ip
, WCHAR name
[])
165 /* FIXME: can we get the name? */
167 sprintfW(name
, fmtip
, (nip
>> 24) & 0xFF, (nip
>> 16) & 0xFF, (nip
>> 8) & 0xFF, (nip
) & 0xFF);
171 static void NETSTAT_conn_header(void)
173 WCHAR local
[22], remote
[22], state
[22];
174 NETSTAT_wprintf(fmtnn
, NETSTAT_load_message(IDS_TCP_ACTIVE_CONN
));
175 NETSTAT_wprintf(fmtn
);
176 strcpyW(local
, NETSTAT_load_message(IDS_TCP_LOCAL_ADDR
));
177 strcpyW(remote
, NETSTAT_load_message(IDS_TCP_REMOTE_ADDR
));
178 strcpyW(state
, NETSTAT_load_message(IDS_TCP_STATE
));
179 NETSTAT_wprintf(fmttcpout
, NETSTAT_load_message(IDS_TCP_PROTO
), local
, remote
, state
);
182 static void NETSTAT_eth_stats(void)
186 DWORD octets
[2], ucastpkts
[2], nucastpkts
[2], discards
[2], errors
[2], unknown
;
189 size
= sizeof(MIB_IFTABLE
);
192 table
= HeapAlloc(GetProcessHeap(), 0, size
);
193 err
= GetIfTable(table
, &size
, FALSE
);
194 if (err
!= NO_ERROR
) HeapFree(GetProcessHeap(), 0, table
);
195 } while (err
== ERROR_INSUFFICIENT_BUFFER
);
199 NETSTAT_wprintf(NETSTAT_load_message(IDS_ETH_STAT
));
200 NETSTAT_wprintf(fmtn
);
201 NETSTAT_wprintf(fmtn
);
202 strcpyW(recv
, NETSTAT_load_message(IDS_ETH_RECV
));
203 NETSTAT_wprintf(fmtethheader
, recv
, NETSTAT_load_message(IDS_ETH_SENT
));
205 octets
[0] = octets
[1] = 0;
206 ucastpkts
[0] = ucastpkts
[1] = 0;
207 nucastpkts
[0] = nucastpkts
[1] = 0;
208 discards
[0] = discards
[1] = 0;
209 errors
[0] = errors
[1] = 0;
212 for (i
= 0; i
< table
->dwNumEntries
; i
++)
214 octets
[0] += table
->table
[i
].dwInOctets
;
215 octets
[1] += table
->table
[i
].dwOutOctets
;
216 ucastpkts
[0] += table
->table
[i
].dwInUcastPkts
;
217 ucastpkts
[1] += table
->table
[i
].dwOutUcastPkts
;
218 nucastpkts
[0] += table
->table
[i
].dwInNUcastPkts
;
219 nucastpkts
[1] += table
->table
[i
].dwOutNUcastPkts
;
220 discards
[0] += table
->table
[i
].dwInDiscards
;
221 discards
[1] += table
->table
[i
].dwOutDiscards
;
222 errors
[0] += table
->table
[i
].dwInErrors
;
223 errors
[1] += table
->table
[i
].dwOutErrors
;
224 unknown
+= table
->table
[i
].dwInUnknownProtos
;
227 NETSTAT_wprintf(fmtethout
, NETSTAT_load_message(IDS_ETH_BYTES
), octets
[0], octets
[1]);
228 NETSTAT_wprintf(fmtethout
, NETSTAT_load_message(IDS_ETH_UNICAST
), ucastpkts
[0], ucastpkts
[1]);
229 NETSTAT_wprintf(fmtethout
, NETSTAT_load_message(IDS_ETH_NUNICAST
), nucastpkts
[0], nucastpkts
[1]);
230 NETSTAT_wprintf(fmtethout
, NETSTAT_load_message(IDS_ETH_DISCARDS
), discards
[0], discards
[1]);
231 NETSTAT_wprintf(fmtethout
, NETSTAT_load_message(IDS_ETH_ERRORS
), errors
[0], errors
[1]);
232 NETSTAT_wprintf(fmtethoutu
, NETSTAT_load_message(IDS_ETH_UNKNOWN
), unknown
);
234 HeapFree(GetProcessHeap(), 0, table
);
237 static void NETSTAT_tcp_table(void)
241 WCHAR HostIp
[MAX_HOSTNAME_LEN
], HostPort
[32];
242 WCHAR RemoteIp
[MAX_HOSTNAME_LEN
], RemotePort
[32];
243 WCHAR Host
[MAX_HOSTNAME_LEN
+ 32];
244 WCHAR Remote
[MAX_HOSTNAME_LEN
+ 32];
246 size
= sizeof(MIB_TCPTABLE
);
249 table
= HeapAlloc(GetProcessHeap(), 0, size
);
250 err
= GetTcpTable(table
, &size
, TRUE
);
251 if (err
!= NO_ERROR
) HeapFree(GetProcessHeap(), 0, table
);
252 } while (err
== ERROR_INSUFFICIENT_BUFFER
);
256 for (i
= 0; i
< table
->dwNumEntries
; i
++)
258 if ((table
->table
[i
].u
.dwState
== MIB_TCP_STATE_CLOSE_WAIT
) ||
259 (table
->table
[i
].u
.dwState
== MIB_TCP_STATE_ESTAB
) ||
260 (table
->table
[i
].u
.dwState
== MIB_TCP_STATE_TIME_WAIT
))
262 NETSTAT_host_name(table
->table
[i
].dwLocalAddr
, HostIp
);
263 NETSTAT_port_name(table
->table
[i
].dwLocalPort
, HostPort
);
264 NETSTAT_host_name(table
->table
[i
].dwRemoteAddr
, RemoteIp
);
265 NETSTAT_port_name(table
->table
[i
].dwRemotePort
, RemotePort
);
267 sprintfW(Host
, fmtcolon
, HostIp
, HostPort
);
268 sprintfW(Remote
, fmtcolon
, RemoteIp
, RemotePort
);
269 NETSTAT_wprintf(fmttcpout
, tcpW
, Host
, Remote
, tcpstatesW
[table
->table
[i
].u
.dwState
]);
272 HeapFree(GetProcessHeap(), 0, table
);
275 static void NETSTAT_tcp_stats(void)
279 if (GetTcpStatistics(&stats
) == NO_ERROR
)
281 NETSTAT_wprintf(fmtnn
, NETSTAT_load_message(IDS_TCP_STAT
));
282 NETSTAT_wprintf(fmtn
);
283 NETSTAT_wprintf(fmttcpstat
, NETSTAT_load_message(IDS_TCP_ACTIVE_OPEN
), stats
.dwActiveOpens
);
284 NETSTAT_wprintf(fmttcpstat
, NETSTAT_load_message(IDS_TCP_PASSIV_OPEN
), stats
.dwPassiveOpens
);
285 NETSTAT_wprintf(fmttcpstat
, NETSTAT_load_message(IDS_TCP_FAILED_CONN
), stats
.dwAttemptFails
);
286 NETSTAT_wprintf(fmttcpstat
, NETSTAT_load_message(IDS_TCP_RESET_CONN
), stats
.dwEstabResets
);
287 NETSTAT_wprintf(fmttcpstat
, NETSTAT_load_message(IDS_TCP_CURR_CONN
), stats
.dwCurrEstab
);
288 NETSTAT_wprintf(fmttcpstat
, NETSTAT_load_message(IDS_TCP_SEGM_RECV
), stats
.dwInSegs
);
289 NETSTAT_wprintf(fmttcpstat
, NETSTAT_load_message(IDS_TCP_SEGM_SENT
), stats
.dwOutSegs
);
290 NETSTAT_wprintf(fmttcpstat
, NETSTAT_load_message(IDS_TCP_SEGM_RETRAN
), stats
.dwRetransSegs
);
294 static void NETSTAT_udp_table(void)
298 WCHAR HostIp
[MAX_HOSTNAME_LEN
], HostPort
[32];
299 WCHAR Host
[MAX_HOSTNAME_LEN
+ 32];
301 size
= sizeof(MIB_UDPTABLE
);
304 table
= HeapAlloc(GetProcessHeap(), 0, size
);
305 err
= GetUdpTable(table
, &size
, TRUE
);
306 if (err
!= NO_ERROR
) HeapFree(GetProcessHeap(), 0, table
);
307 } while (err
== ERROR_INSUFFICIENT_BUFFER
);
311 for (i
= 0; i
< table
->dwNumEntries
; i
++)
313 NETSTAT_host_name(table
->table
[i
].dwLocalAddr
, HostIp
);
314 NETSTAT_port_name(table
->table
[i
].dwLocalPort
, HostPort
);
316 sprintfW(Host
, fmtcolon
, HostIp
, HostPort
);
317 NETSTAT_wprintf(fmtudpout
, udpW
, Host
);
319 HeapFree(GetProcessHeap(), 0, table
);
322 static void NETSTAT_udp_stats(void)
326 if (GetUdpStatistics(&stats
) == NO_ERROR
)
328 NETSTAT_wprintf(fmtnn
, NETSTAT_load_message(IDS_UDP_STAT
));
329 NETSTAT_wprintf(fmtn
);
330 NETSTAT_wprintf(fmtudpstat
, NETSTAT_load_message(IDS_UDP_DGRAMS_RECV
), stats
.dwInDatagrams
);
331 NETSTAT_wprintf(fmtudpstat
, NETSTAT_load_message(IDS_UDP_NO_PORTS
), stats
.dwNoPorts
);
332 NETSTAT_wprintf(fmtudpstat
, NETSTAT_load_message(IDS_UDP_RECV_ERRORS
), stats
.dwInErrors
);
333 NETSTAT_wprintf(fmtudpstat
, NETSTAT_load_message(IDS_UDP_DGRAMS_SENT
), stats
.dwOutDatagrams
);
337 static NETSTATPROTOCOLS
NETSTAT_get_protocol(WCHAR name
[])
339 if (!strcmpiW(name
, ipW
)) return PROT_IP
;
340 if (!strcmpiW(name
, ipv6W
)) return PROT_IPV6
;
341 if (!strcmpiW(name
, icmpW
)) return PROT_ICMP
;
342 if (!strcmpiW(name
, icmpv6W
)) return PROT_ICMPV6
;
343 if (!strcmpiW(name
, tcpW
)) return PROT_TCP
;
344 if (!strcmpiW(name
, tcpv6W
)) return PROT_TCPV6
;
345 if (!strcmpiW(name
, udpW
)) return PROT_UDP
;
346 if (!strcmpiW(name
, udpv6W
)) return PROT_UDPV6
;
350 int wmain(int argc
, WCHAR
*argv
[])
353 BOOL output_stats
= FALSE
;
355 if (WSAStartup(MAKEWORD(2, 2), &wsa_data
))
357 WINE_ERR("WSAStartup failed: %d\n", WSAGetLastError());
364 NETSTAT_conn_header();
369 while (argv
[1] && argv
[1][0] == '-')
374 NETSTAT_conn_header();
386 if (argc
== 1) return 1;
387 switch (NETSTAT_get_protocol(argv
[1]))
392 NETSTAT_conn_header();
398 NETSTAT_conn_header();
402 WINE_FIXME("Protocol not yet implemented: %s\n", debugstr_w(argv
[1]));
406 WINE_FIXME("Unknown option: %s\n", debugstr_w(argv
[1]));