ipconfig.c 12.9 KB
Newer Older
1 2 3
/*
 * IP configuration utility
 *
4
 * Copyright 2008 Andrew Riedi
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
 * Copyright 2010 Andrew Nguyen
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
 */

22 23
#define NONAMELESSUNION

24
#include <stdio.h>
25
#include <winsock2.h>
26
#include <windows.h>
27
#include <iphlpapi.h>
28

29 30
#include "ipconfig.h"

31
static int ipconfig_vprintfW(const WCHAR *msg, __ms_va_list va_args)
32 33 34 35 36
{
    int wlen;
    DWORD count, ret;
    WCHAR msg_buffer[8192];

37
    wlen = FormatMessageW(FORMAT_MESSAGE_FROM_STRING, msg, 0, 0, msg_buffer,
38
                          ARRAY_SIZE(msg_buffer), &va_args);
39 40 41 42 43 44 45

    ret = WriteConsoleW(GetStdHandle(STD_OUTPUT_HANDLE), msg_buffer, wlen, &count, NULL);
    if (!ret)
    {
        DWORD len;
        char *msgA;

46 47 48 49
        /* On Windows WriteConsoleW() fails if the output is redirected. So fall
         * back to WriteFile(), assuming the console encoding is still the right
         * one in that case.
         */
50 51 52 53 54 55 56 57 58 59 60 61 62 63 64
        len = WideCharToMultiByte(GetConsoleOutputCP(), 0, msg_buffer, wlen,
            NULL, 0, NULL, NULL);
        msgA = HeapAlloc(GetProcessHeap(), 0, len);
        if (!msgA)
            return 0;

        WideCharToMultiByte(GetConsoleOutputCP(), 0, msg_buffer, wlen, msgA, len,
            NULL, NULL);
        WriteFile(GetStdHandle(STD_OUTPUT_HANDLE), msgA, len, &count, FALSE);
        HeapFree(GetProcessHeap(), 0, msgA);
    }

    return count;
}

65
static int WINAPIV ipconfig_printfW(const WCHAR *msg, ...)
66
{
67
    __ms_va_list va_args;
68 69
    int len;

70
    __ms_va_start(va_args, msg);
71
    len = ipconfig_vprintfW(msg, va_args);
72
    __ms_va_end(va_args);
73 74 75 76

    return len;
}

77
static int WINAPIV ipconfig_message_printfW(int msg, ...)
78
{
79
    __ms_va_list va_args;
80 81 82
    WCHAR msg_buffer[8192];
    int len;

83
    LoadStringW(GetModuleHandleW(NULL), msg, msg_buffer, ARRAY_SIZE(msg_buffer));
84

85
    __ms_va_start(va_args, msg);
86
    len = ipconfig_vprintfW(msg_buffer, va_args);
87
    __ms_va_end(va_args);
88 89 90 91

    return len;
}

92 93
static int ipconfig_message(int msg)
{
94
    static const WCHAR formatW[] = {'%','1',0};
95 96
    WCHAR msg_buffer[8192];

97
    LoadStringW(GetModuleHandleW(NULL), msg, msg_buffer, ARRAY_SIZE(msg_buffer));
98

99 100 101
    return ipconfig_printfW(formatW, msg_buffer);
}

102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118
static const WCHAR *iftype_to_string(DWORD type)
{
    static WCHAR msg_buffer[50];

    int msg;

    switch (type)
    {
    case IF_TYPE_ETHERNET_CSMACD:
    /* The loopback adapter appears as an Ethernet device. */
    case IF_TYPE_SOFTWARE_LOOPBACK:
        msg = STRING_ETHERNET;
        break;
    default:
        msg = STRING_UNKNOWN;
    }

119
    LoadStringW(GetModuleHandleW(NULL), msg, msg_buffer, ARRAY_SIZE(msg_buffer));
120 121 122 123 124 125

    return msg_buffer;
}

static void print_field(int msg, const WCHAR *value)
{
126
    static const WCHAR formatW[] = {' ',' ',' ',' ','%','1',':',' ','%','2','\n',0};
127 128 129

    WCHAR field[] = {'.',' ','.',' ','.',' ','.',' ','.',' ','.',' ','.',' ','.',' ','.',
                     ' ','.',' ','.',' ','.',' ','.',' ','.',' ','.',' ','.',' ','.',' ',0};
130
    WCHAR name_buffer[ARRAY_SIZE(field)];
131

132
    LoadStringW(GetModuleHandleW(NULL), msg, name_buffer, ARRAY_SIZE(name_buffer));
133
    memcpy(field, name_buffer, sizeof(WCHAR) * min(lstrlenW(name_buffer), ARRAY_SIZE(field) - 1));
134 135 136 137

    ipconfig_printfW(formatW, field, value);
}

138 139 140 141 142 143
static void print_value(const WCHAR *value)
{
    static const WCHAR formatW[] = {' ',' ',' ',' ',' ',' ',' ',' ',' ',' ',
                                    ' ',' ',' ',' ',' ',' ',' ',' ',' ',' ',
                                    ' ',' ',' ',' ',' ',' ',' ',' ',' ',' ',
                                    ' ',' ',' ',' ',' ',' ',' ',' ',' ',' ',
144
                                    '%','1','\n',0};
145 146 147 148

    ipconfig_printfW(formatW, value);
}

149 150 151 152 153 154 155
static BOOL socket_address_to_string(WCHAR *buf, DWORD len, SOCKET_ADDRESS *addr)
{
    return WSAAddressToStringW(addr->lpSockaddr,
                               addr->iSockaddrLength, NULL,
                               buf, &len) == 0;
}

156 157 158 159 160
static void print_basic_information(void)
{
    IP_ADAPTER_ADDRESSES *adapters;
    ULONG out = 0;

161 162
    if (GetAdaptersAddresses(AF_UNSPEC, GAA_FLAG_INCLUDE_ALL_GATEWAYS,
                             NULL, NULL, &out) == ERROR_BUFFER_OVERFLOW)
163 164 165 166 167
    {
        adapters = HeapAlloc(GetProcessHeap(), 0, out);
        if (!adapters)
            exit(1);

168 169
        if (GetAdaptersAddresses(AF_UNSPEC, GAA_FLAG_INCLUDE_ALL_GATEWAYS,
                                 NULL, adapters, &out) == ERROR_SUCCESS)
170 171 172 173 174 175
        {
            IP_ADAPTER_ADDRESSES *p;

            for (p = adapters; p; p = p->Next)
            {
                static const WCHAR newlineW[] = {'\n',0};
176
                static const WCHAR emptyW[] = {0};
177 178

                IP_ADAPTER_UNICAST_ADDRESS *addr;
179
                IP_ADAPTER_GATEWAY_ADDRESS_LH *gateway;
180
                WCHAR addr_buf[54];
181 182 183 184 185 186 187

                ipconfig_message_printfW(STRING_ADAPTER_FRIENDLY, iftype_to_string(p->IfType), p->FriendlyName);
                ipconfig_printfW(newlineW);
                print_field(STRING_CONN_DNS_SUFFIX, p->DnsSuffix);

                for (addr = p->FirstUnicastAddress; addr; addr = addr->Next)
                {
188
                    if (addr->Address.lpSockaddr->sa_family == AF_INET &&
189
                        socket_address_to_string(addr_buf, ARRAY_SIZE(addr_buf), &addr->Address))
190
                        print_field(STRING_IP_ADDRESS, addr_buf);
191
                    else if (addr->Address.lpSockaddr->sa_family == AF_INET6 &&
192
                             socket_address_to_string(addr_buf, ARRAY_SIZE(addr_buf), &addr->Address))
193
                        print_field(STRING_IP6_ADDRESS, addr_buf);
194 195 196
                    /* FIXME: Output corresponding subnet mask. */
                }

197 198
                if (p->FirstGatewayAddress)
                {
199
                    if (socket_address_to_string(addr_buf, ARRAY_SIZE(addr_buf), &p->FirstGatewayAddress->Address))
200 201 202 203
                        print_field(STRING_DEFAULT_GATEWAY, addr_buf);

                    for (gateway = p->FirstGatewayAddress->Next; gateway; gateway = gateway->Next)
                    {
204
                        if (socket_address_to_string(addr_buf, ARRAY_SIZE(addr_buf), &gateway->Address))
205 206 207 208 209 210
                            print_value(addr_buf);
                    }
                }
                else
                    print_field(STRING_DEFAULT_GATEWAY, emptyW);

211 212 213 214 215 216 217 218
                ipconfig_printfW(newlineW);
            }
        }

        HeapFree(GetProcessHeap(), 0, adapters);
    }
}

219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242
static const WCHAR *nodetype_to_string(DWORD type)
{
    static WCHAR msg_buffer[50];

    int msg;

    switch (type)
    {
    case BROADCAST_NODETYPE:
        msg = STRING_BROADCAST;
        break;
    case PEER_TO_PEER_NODETYPE:
        msg = STRING_PEER_TO_PEER;
        break;
    case MIXED_NODETYPE:
        msg = STRING_MIXED;
        break;
    case HYBRID_NODETYPE:
        msg = STRING_HYBRID;
        break;
    default:
        msg = STRING_UNKNOWN;
    }

243
    LoadStringW(GetModuleHandleW(NULL), msg, msg_buffer, ARRAY_SIZE(msg_buffer));
244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261

    return msg_buffer;
}

static WCHAR *physaddr_to_string(WCHAR *buf, BYTE *addr, DWORD len)
{
    static const WCHAR fmtW[] = {'%','0','2','X','-',0};
    static const WCHAR fmt2W[] = {'%','0','2','X',0};

    if (!len)
        *buf = '\0';
    else
    {
        WCHAR *p = buf;
        DWORD i;

        for (i = 0; i < len - 1; i++)
        {
262
            swprintf(p, 4, fmtW, addr[i]);
263 264
            p += 3;
        }
265
        swprintf(p, 3, fmt2W, addr[i]);
266 267 268 269 270
    }

    return buf;
}

271 272 273 274 275
static const WCHAR *boolean_to_string(int value)
{
    static WCHAR msg_buffer[15];

    LoadStringW(GetModuleHandleW(NULL), value ? STRING_YES : STRING_NO,
276
        msg_buffer, ARRAY_SIZE(msg_buffer));
277 278 279 280

    return msg_buffer;
}

281 282 283
static void print_full_information(void)
{
    static const WCHAR newlineW[] = {'\n',0};
284
    static const WCHAR emptyW[] = {0};
285 286 287 288 289 290 291 292 293 294 295 296 297 298 299

    FIXED_INFO *info;
    IP_ADAPTER_ADDRESSES *adapters;
    ULONG out = 0;

    if (GetNetworkParams(NULL, &out) == ERROR_BUFFER_OVERFLOW)
    {
        info = HeapAlloc(GetProcessHeap(), 0, out);
        if (!info)
            exit(1);

        if (GetNetworkParams(info, &out) == ERROR_SUCCESS)
        {
            WCHAR hostnameW[MAX_HOSTNAME_LEN + 4];

300
            MultiByteToWideChar(CP_ACP, 0, info->HostName, -1, hostnameW, ARRAY_SIZE(hostnameW));
301 302 303 304 305
            print_field(STRING_HOSTNAME, hostnameW);

            /* FIXME: Output primary DNS suffix. */

            print_field(STRING_NODE_TYPE, nodetype_to_string(info->NodeType));
306
            print_field(STRING_IP_ROUTING, boolean_to_string(info->EnableRouting));
307 308 309 310 311 312 313 314 315

            /* FIXME: Output WINS proxy status and DNS suffix search list. */

            ipconfig_printfW(newlineW);
        }

        HeapFree(GetProcessHeap(), 0, info);
    }

316 317
    if (GetAdaptersAddresses(AF_UNSPEC, GAA_FLAG_INCLUDE_ALL_GATEWAYS,
                             NULL, NULL, &out) == ERROR_BUFFER_OVERFLOW)
318 319 320 321 322
    {
        adapters = HeapAlloc(GetProcessHeap(), 0, out);
        if (!adapters)
            exit(1);

323 324
        if (GetAdaptersAddresses(AF_UNSPEC, GAA_FLAG_INCLUDE_ALL_GATEWAYS,
                                 NULL, adapters, &out) == ERROR_SUCCESS)
325 326 327 328 329 330 331
        {
            IP_ADAPTER_ADDRESSES *p;

            for (p = adapters; p; p = p->Next)
            {
                IP_ADAPTER_UNICAST_ADDRESS *addr;
                WCHAR physaddr_buf[3 * MAX_ADAPTER_ADDRESS_LENGTH];
332
                IP_ADAPTER_GATEWAY_ADDRESS_LH *gateway;
333
                WCHAR addr_buf[54];
334 335 336 337 338 339

                ipconfig_message_printfW(STRING_ADAPTER_FRIENDLY, iftype_to_string(p->IfType), p->FriendlyName);
                ipconfig_printfW(newlineW);
                print_field(STRING_CONN_DNS_SUFFIX, p->DnsSuffix);
                print_field(STRING_DESCRIPTION, p->Description);
                print_field(STRING_PHYS_ADDR, physaddr_to_string(physaddr_buf, p->PhysicalAddress, p->PhysicalAddressLength));
340
                print_field(STRING_DHCP_ENABLED, boolean_to_string(p->u1.Flags & IP_ADAPTER_DHCP_ENABLED));
341 342 343 344 345

                /* FIXME: Output autoconfiguration status. */

                for (addr = p->FirstUnicastAddress; addr; addr = addr->Next)
                {
346
                    if (addr->Address.lpSockaddr->sa_family == AF_INET &&
347
                        socket_address_to_string(addr_buf, ARRAY_SIZE(addr_buf), &addr->Address))
348
                        print_field(STRING_IP_ADDRESS, addr_buf);
349
                    else if (addr->Address.lpSockaddr->sa_family == AF_INET6 &&
350
                             socket_address_to_string(addr_buf, ARRAY_SIZE(addr_buf), &addr->Address))
351
                        print_field(STRING_IP6_ADDRESS, addr_buf);
352 353 354
                    /* FIXME: Output corresponding subnet mask. */
                }

355 356
                if (p->FirstGatewayAddress)
                {
357
                    if (socket_address_to_string(addr_buf, ARRAY_SIZE(addr_buf), &p->FirstGatewayAddress->Address))
358 359 360 361
                        print_field(STRING_DEFAULT_GATEWAY, addr_buf);

                    for (gateway = p->FirstGatewayAddress->Next; gateway; gateway = gateway->Next)
                    {
362
                        if (socket_address_to_string(addr_buf, ARRAY_SIZE(addr_buf), &gateway->Address))
363 364 365 366 367
                            print_value(addr_buf);
                    }
                }
                else
                    print_field(STRING_DEFAULT_GATEWAY, emptyW);
368 369 370 371 372 373 374 375 376

                ipconfig_printfW(newlineW);
            }
        }

        HeapFree(GetProcessHeap(), 0, adapters);
    }
}

377
int __cdecl wmain(int argc, WCHAR *argv[])
378
{
379 380 381
    static const WCHAR slashHelp[] = {'/','?',0};
    static const WCHAR slashAll[] = {'/','a','l','l',0};

382 383 384 385 386
    WSADATA data;

    if (WSAStartup(MAKEWORD(2, 2), &data))
        return 1;

387 388
    if (argc > 1)
    {
389
        if (!lstrcmpW(slashHelp, argv[1]))
390 391
        {
            ipconfig_message(STRING_USAGE);
392
            WSACleanup();
393 394
            return 1;
        }
395
        else if (!wcsicmp(slashAll, argv[1]))
396 397 398 399 400
        {
            if (argv[2])
            {
                ipconfig_message(STRING_INVALID_CMDLINE);
                ipconfig_message(STRING_USAGE);
401
                WSACleanup();
402 403 404
                return 1;
            }

405
            print_full_information();
406 407 408 409 410
        }
        else
        {
            ipconfig_message(STRING_INVALID_CMDLINE);
            ipconfig_message(STRING_USAGE);
411
            WSACleanup();
412 413 414
            return 1;
        }
    }
415 416
    else
        print_basic_information();
417

418
    WSACleanup();
419 420
    return 0;
}