modules/rostests/apitests/ws2_32/getaddrinfo.c

   1 /*
   2  * PROJECT:         ReactOS api tests
   3  * LICENSE:         GPLv2+ - See COPYING in the top level directory
   4  * PURPOSE:         Test for getaddrinfo
   5  * PROGRAMMER:      Thomas Faber <thomas.faber@reactos.org>
   6  */
   7
   8 #include "ws2_32.h"
   9
  10 #define ok_addrinfo(ai, flags, family, socktype, protocol, addrlen) do  \
  11 {                                                                       \
  12     ok_hex((ai)->ai_flags, flags);                                      \
  13     ok_dec((ai)->ai_family, family);                                    \
  14     ok_dec((ai)->ai_socktype, socktype);                                \
  15     ok_dec((ai)->ai_protocol, protocol);                                \
  16     ok_dec((int)(ai)->ai_addrlen, addrlen);                             \
  17 } while (0)
  18
  19 #define ok_sockaddr_in(sockaddr, family, port, addr) do                 \
  20 {                                                                       \
  21     int _i;                                                             \
  22     ok_dec(((SOCKADDR_IN *)(sockaddr))->sin_family, family);            \
  23     ok_dec(ntohs(((SOCKADDR_IN *)(sockaddr))->sin_port), port);         \
  24     ok_hex(((SOCKADDR_IN *)(sockaddr))->sin_addr.S_un.S_addr,           \
  25            inet_addr(addr));                                            \
  26     for (_i = 0; _i < 7; _i++)                                          \
  27         ok_dec(((SOCKADDR_IN *)(sockaddr))->sin_zero[_i], 0);           \
  28 } while (0)
  29
  30 CHAR LocalAddress[sizeof("255.255.255.255")];
  31
  32 static
  33 VOID
  34 TestNodeName(VOID)
  35 {
  36     int Error;
  37     PADDRINFOA AddrInfo;
  38     ADDRINFOA Hints;
  39     struct
  40     {
  41         PCSTR NodeName;
  42         PCSTR ExpectedAddress;
  43         INT Flags;
  44     } Tests[] =
  45     {
  46         { "",                               LocalAddress },
  47         { " ",                              NULL },
  48         { "doesntexist.example.com",        NULL },
  49         { "localhost",                      "127.0.0.1" },
  50         { "localhost:80",                   NULL },
  51         { "7.8.9.10",                       "7.8.9.10",         AI_NUMERICHOST },
  52         { "0.0.0.0",                        "0.0.0.0",          AI_NUMERICHOST },
  53         { "255.255.255.255",                "255.255.255.255",  AI_NUMERICHOST },
  54         { "0.0.0.0 ",                       "0.0.0.0",    /* no AI_NUMERICHOST */ },
  55         { "0.0.0.0:80",                     NULL },
  56         { "0.0.0.0.0",                      NULL },
  57         { "1.1.1.256",                      NULL },
  58         { "1.2.3",                          NULL },
  59         { "1.2.3.0x4",                      "1.2.3.4",          AI_NUMERICHOST },
  60         { "1.2.3.010",                      "1.2.3.8",          AI_NUMERICHOST },
  61         /* let's just assume this one doesn't change any time soon ;) */
  62         { "google-public-dns-a.google.com", "8.8.8.8" },
  63     };
  64     const INT TestCount = sizeof(Tests) / sizeof(Tests[0]);
  65     INT i;
  66
  67     /* make sure we don't get IPv6 responses */
  68     ZeroMemory(&Hints, sizeof(Hints));
  69     Hints.ai_family = AF_INET;
  70
  71     trace("Nodes\n");
  72     for (i = 0; i < TestCount; i++)
  73     {
  74         trace("%d: '%s'\n", i, Tests[i].NodeName);
  75         StartSeh()
  76             AddrInfo = InvalidPointer;
  77             Error = getaddrinfo(Tests[i].NodeName, NULL, &Hints, &AddrInfo);
  78             if (Tests[i].ExpectedAddress)
  79             {
  80                 ok_dec(Error, 0);
  81                 ok_dec(WSAGetLastError(), 0);
  82                 ok(AddrInfo != NULL && AddrInfo != InvalidPointer,
  83                    "AddrInfo = %p\n", AddrInfo);
  84             }
  85             else
  86             {
  87                 ok_dec(Error, WSAHOST_NOT_FOUND);
  88                 ok_dec(WSAGetLastError(), WSAHOST_NOT_FOUND);
  89                 ok_ptr(AddrInfo, NULL);
  90             }
  91             if (!Error && AddrInfo && AddrInfo != InvalidPointer)
  92             {
  93                 ok_addrinfo(AddrInfo, Tests[i].Flags, AF_INET,
  94                             0, 0, sizeof(SOCKADDR_IN));
  95                 ok_ptr(AddrInfo->ai_canonname, NULL);
  96                 ok_sockaddr_in(AddrInfo->ai_addr, AF_INET,
  97                                0, Tests[i].ExpectedAddress);
  98                 ok_ptr(AddrInfo->ai_next, NULL);
  99                 freeaddrinfo(AddrInfo);
 100             }
 101         EndSeh(STATUS_SUCCESS);
 102     }
 103 }
 104
 105 static
 106 VOID
 107 TestServiceName(VOID)
 108 {
 109     int Error;
 110     PADDRINFOA AddrInfo;
 111     ADDRINFOA Hints;
 112     struct
 113     {
 114         PCSTR ServiceName;
 115         INT ExpectedPort;
 116         INT SockType;
 117     } Tests[] =
 118     {
 119         { "", 0 },
 120         { "0", 0 },
 121         { "1", 1 },
 122         { "a", -1 },
 123         { "010", 10 },
 124         { "0x1a", -1 },
 125         { "http", 80, SOCK_STREAM },
 126         { "smtp", 25, SOCK_STREAM },
 127         { "mail", 25, SOCK_STREAM }, /* alias for smtp */
 128         { "router", 520, SOCK_DGRAM },
 129         { "domain", 53, 0 /* DNS supports both UDP and TCP */ },
 130         { ":0", -1 },
 131         { "123", 123 },
 132         { " 123", 123 },
 133         { "    123", 123 },
 134         { "32767", 32767 },
 135         { "32768", 32768 },
 136         { "65535", 65535 },
 137         { "65536", 0 },
 138         { "65537", 1 },
 139         { "65540", 4 },
 140         { "65536", 0 },
 141         { "4294967295", 65535 },
 142         { "4294967296", 65535 },
 143         { "9999999999", 65535 },
 144         { "999999999999999999999999999999999999", 65535 },
 145         { "+5", 5 },
 146         { "-1", 65535 },
 147         { "-4", 65532 },
 148         { "-65534", 2 },
 149         { "-65535", 1 },
 150         { "-65536", 0 },
 151         { "-65537", 65535 },
 152         { "28a", -1 },
 153         { "28 ", -1 },
 154         { "a28", -1 },
 155     };
 156     const INT TestCount = sizeof(Tests) / sizeof(Tests[0]);
 157     INT i;
 158
 159     /* make sure we don't get IPv6 responses */
 160     ZeroMemory(&Hints, sizeof(Hints));
 161     Hints.ai_family = AF_INET;
 162
 163     trace("Services\n");
 164     for (i = 0; i < TestCount; i++)
 165     {
 166         trace("%d: '%s'\n", i, Tests[i].ServiceName);
 167         StartSeh()
 168             AddrInfo = InvalidPointer;
 169             Error = getaddrinfo(NULL, Tests[i].ServiceName, &Hints, &AddrInfo);
 170             if (Tests[i].ExpectedPort != -1)
 171             {
 172                 ok_dec(Error, 0);
 173                 ok_dec(WSAGetLastError(), 0);
 174                 ok(AddrInfo != NULL && AddrInfo != InvalidPointer,
 175                    "AddrInfo = %p\n", AddrInfo);
 176             }
 177             else
 178             {
 179                 ok_dec(Error, WSATYPE_NOT_FOUND);
 180                 ok_dec(WSAGetLastError(), WSATYPE_NOT_FOUND);
 181                 ok_ptr(AddrInfo, NULL);
 182             }
 183             if (!Error && AddrInfo && AddrInfo != InvalidPointer)
 184             {
 185                 ok_addrinfo(AddrInfo, 0, AF_INET,
 186                             Tests[i].SockType, 0, sizeof(SOCKADDR_IN));
 187                 ok_ptr(AddrInfo->ai_canonname, NULL);
 188                 ok_sockaddr_in(AddrInfo->ai_addr, AF_INET,
 189                                Tests[i].ExpectedPort, "127.0.0.1");
 190                 ok_ptr(AddrInfo->ai_next, NULL);
 191                 freeaddrinfo(AddrInfo);
 192             }
 193         EndSeh(STATUS_SUCCESS);
 194     }
 195 }
 196
 197 START_TEST(getaddrinfo)
 198 {
 199     WSADATA WsaData;
 200     int Error;
 201     PADDRINFOA AddrInfo;
 202     PADDRINFOW AddrInfoW;
 203     ADDRINFOA Hints;
 204     ADDRINFOW HintsW;
 205     CHAR LocalHostName[128];
 206     struct hostent *Hostent;
 207
 208     /* not yet initialized */
 209     StartSeh()
 210         Error = getaddrinfo(NULL, NULL, NULL, NULL);
 211         ok_dec(Error, WSANOTINITIALISED);
 212     EndSeh(STATUS_SUCCESS);
 213     StartSeh()
 214         AddrInfo = InvalidPointer;
 215         Error = getaddrinfo(NULL, NULL, NULL, &AddrInfo);
 216         ok_dec(Error, WSANOTINITIALISED);
 217         ok_ptr(AddrInfo, InvalidPointer);
 218     EndSeh(STATUS_SUCCESS);
 219
 220     Error = getaddrinfo("127.0.0.1", "80", NULL, &AddrInfo);
 221     ok_dec(Error, WSANOTINITIALISED);
 222
 223     Error = GetAddrInfoW(L"127.0.0.1", L"80", NULL, &AddrInfoW);
 224     ok_dec(Error, WSANOTINITIALISED);
 225
 226     Error = WSAStartup(MAKEWORD(2, 2), &WsaData);
 227     ok_dec(Error, 0);
 228
 229     /* initialize LocalAddress for tests */
 230     Error = gethostname(LocalHostName, sizeof(LocalHostName));
 231     ok_dec(Error, 0);
 232     ok_dec(WSAGetLastError(), 0);
 233     trace("Local host name is '%s'\n", LocalHostName);
 234     Hostent = gethostbyname(LocalHostName);
 235     ok(Hostent != NULL, "gethostbyname failed with %d\n", WSAGetLastError());
 236     if (Hostent && Hostent->h_addr_list[0] && Hostent->h_length == sizeof(IN_ADDR))
 237     {
 238         IN_ADDR Address;
 239         memcpy(&Address, Hostent->h_addr_list[0], sizeof(Address));
 240         strcpy(LocalAddress, inet_ntoa(Address));
 241     }
 242     trace("Local address is '%s'\n", LocalAddress);
 243     ok(LocalAddress[0] != '\0',
 244        "Could not determine local address. Following test results may be wrong.\n");
 245
 246     ZeroMemory(&Hints, sizeof(Hints));
 247     /* parameter tests for getaddrinfo */
 248     StartSeh() getaddrinfo(NULL, NULL, NULL, NULL); EndSeh(STATUS_ACCESS_VIOLATION);
 249     StartSeh() getaddrinfo("", "", &Hints, NULL);   EndSeh(STATUS_ACCESS_VIOLATION);
 250     StartSeh()
 251         AddrInfo = InvalidPointer;
 252         Error = getaddrinfo(NULL, NULL, NULL, &AddrInfo);
 253         ok_dec(Error, WSAHOST_NOT_FOUND);
 254         ok_dec(WSAGetLastError(), WSAHOST_NOT_FOUND);
 255         ok_ptr(AddrInfo, NULL);
 256     EndSeh(STATUS_SUCCESS);
 257
 258     /* parameter tests for GetAddrInfoW */
 259     StartSeh() GetAddrInfoW(NULL, NULL, NULL, NULL);  EndSeh(STATUS_ACCESS_VIOLATION);
 260     StartSeh() GetAddrInfoW(L"", L"", &HintsW, NULL); EndSeh(STATUS_ACCESS_VIOLATION);
 261     StartSeh()
 262         AddrInfo = InvalidPointer;
 263         Error = GetAddrInfoW(NULL, NULL, NULL, &AddrInfoW);
 264         ok_dec(Error, WSAHOST_NOT_FOUND);
 265         ok_dec(WSAGetLastError(), WSAHOST_NOT_FOUND);
 266         ok_ptr(AddrInfo, InvalidPointer); /* differs from getaddrinfo */
 267     EndSeh(STATUS_SUCCESS);
 268
 269     TestNodeName();
 270     TestServiceName();
 271     /* TODO: test passing both node name and service name */
 272     /* TODO: test hints */
 273     /* TODO: test IPv6 */
 274
 275     Error = WSACleanup();
 276     ok_dec(Error, 0);
 277
 278     /* not initialized anymore */
 279     Error = getaddrinfo("127.0.0.1", "80", NULL, &AddrInfo);
 280     ok_dec(Error, WSANOTINITIALISED);
 281
 282     Error = GetAddrInfoW(L"127.0.0.1", L"80", NULL, &AddrInfoW);
 283     ok_dec(Error, WSANOTINITIALISED);
 284 }