| 1 | /* $Id: tstRTNetIPv4.cpp 65579 2017-02-01 21:32:38Z vboxsync $ */
|
|---|
| 2 | /** @file
|
|---|
| 3 | * IPRT Testcase - IPv4.
|
|---|
| 4 | */
|
|---|
| 5 |
|
|---|
| 6 | /*
|
|---|
| 7 | * Copyright (C) 2008-2016 Oracle Corporation
|
|---|
| 8 | *
|
|---|
| 9 | * This file is part of VirtualBox Open Source Edition (OSE), as
|
|---|
| 10 | * available from http://www.virtualbox.org. This file is free software;
|
|---|
| 11 | * you can redistribute it and/or modify it under the terms of the GNU
|
|---|
| 12 | * General Public License (GPL) as published by the Free Software
|
|---|
| 13 | * Foundation, in version 2 as it comes in the "COPYING" file of the
|
|---|
| 14 | * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
|
|---|
| 15 | * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
|
|---|
| 16 | *
|
|---|
| 17 | * The contents of this file may alternatively be used under the terms
|
|---|
| 18 | * of the Common Development and Distribution License Version 1.0
|
|---|
| 19 | * (CDDL) only, as it comes in the "COPYING.CDDL" file of the
|
|---|
| 20 | * VirtualBox OSE distribution, in which case the provisions of the
|
|---|
| 21 | * CDDL are applicable instead of those of the GPL.
|
|---|
| 22 | *
|
|---|
| 23 | * You may elect to license modified versions of this file under the
|
|---|
| 24 | * terms and conditions of either the GPL or the CDDL or both.
|
|---|
| 25 | */
|
|---|
| 26 |
|
|---|
| 27 |
|
|---|
| 28 | /*********************************************************************************************************************************
|
|---|
| 29 | * Header Files *
|
|---|
| 30 | *********************************************************************************************************************************/
|
|---|
| 31 | #include <iprt/net.h>
|
|---|
| 32 |
|
|---|
| 33 | #include <iprt/err.h>
|
|---|
| 34 | #include <iprt/initterm.h>
|
|---|
| 35 | #include <iprt/test.h>
|
|---|
| 36 |
|
|---|
| 37 |
|
|---|
| 38 | /*********************************************************************************************************************************
|
|---|
| 39 | * Defined Constants And Macros *
|
|---|
| 40 | *********************************************************************************************************************************/
|
|---|
| 41 | #define CHECKADDR(String, rcExpected, ExpectedAddr) \
|
|---|
| 42 | do { \
|
|---|
| 43 | RTNETADDRIPV4 Addr; \
|
|---|
| 44 | int rc2 = RTNetStrToIPv4Addr(String, &Addr); \
|
|---|
| 45 | if ((rcExpected) && !rc2) \
|
|---|
| 46 | { \
|
|---|
| 47 | RTTestIFailed("at line %d: '%s': expected %Rrc got %Rrc\n", \
|
|---|
| 48 | __LINE__, String, (rcExpected), rc2); \
|
|---|
| 49 | } \
|
|---|
| 50 | else if ( (rcExpected) != rc2 \
|
|---|
| 51 | || ( rc2 == VINF_SUCCESS \
|
|---|
| 52 | && RT_H2N_U32_C(ExpectedAddr) != Addr.u)) \
|
|---|
| 53 | { \
|
|---|
| 54 | RTTestIFailed("at line %d: '%s': expected %Rrc got %Rrc," \
|
|---|
| 55 | " expected address %RTnaipv4 got %RTnaipv4\n", \
|
|---|
| 56 | __LINE__, String, rcExpected, rc2, \
|
|---|
| 57 | RT_H2N_U32_C(ExpectedAddr), Addr.u); \
|
|---|
| 58 | } \
|
|---|
| 59 | } while (0)
|
|---|
| 60 |
|
|---|
| 61 | #define GOODADDR(String, ExpectedAddr) \
|
|---|
| 62 | CHECKADDR(String, VINF_SUCCESS, ExpectedAddr)
|
|---|
| 63 |
|
|---|
| 64 | #define BADADDR(String) \
|
|---|
| 65 | CHECKADDR(String, VERR_INVALID_PARAMETER, 0)
|
|---|
| 66 |
|
|---|
| 67 |
|
|---|
| 68 | #define CHECKADDREX(String, Trailer, rcExpected, ExpectedAddr) \
|
|---|
| 69 | do { \
|
|---|
| 70 | RTNETADDRIPV4 Addr; \
|
|---|
| 71 | const char *strAll = String /* concat */ Trailer; \
|
|---|
| 72 | const char *pTrailer = strAll + sizeof(String) - 1; \
|
|---|
| 73 | char *pNext = NULL; \
|
|---|
| 74 | int rc2 = RTNetStrToIPv4AddrEx(strAll, &Addr, &pNext); \
|
|---|
| 75 | if ((rcExpected) && !rc2) \
|
|---|
| 76 | { \
|
|---|
| 77 | RTTestIFailed("at line %d: '%s': expected %Rrc got %Rrc\n", \
|
|---|
| 78 | __LINE__, String, (rcExpected), rc2); \
|
|---|
| 79 | } \
|
|---|
| 80 | else if ((rcExpected) != rc2 \
|
|---|
| 81 | || (rc2 == VINF_SUCCESS \
|
|---|
| 82 | && (RT_H2N_U32_C(ExpectedAddr) != Addr.u \
|
|---|
| 83 | || pTrailer != pNext))) \
|
|---|
| 84 | { \
|
|---|
| 85 | RTTestIFailed("at line %d: '%s': expected %Rrc got %Rrc," \
|
|---|
| 86 | " expected address %RTnaipv4 got %RTnaipv4" \
|
|---|
| 87 | " expected trailer \"%s\" got %s%s%s" \
|
|---|
| 88 | "\n", \
|
|---|
| 89 | __LINE__, String, rcExpected, rc2, \
|
|---|
| 90 | RT_H2N_U32_C(ExpectedAddr), Addr.u, \
|
|---|
| 91 | pTrailer, \
|
|---|
| 92 | pNext ? "\"" : "", \
|
|---|
| 93 | pNext ? pNext : "(null)", \
|
|---|
| 94 | pNext ? "\"" : ""); \
|
|---|
| 95 | } \
|
|---|
| 96 | } while (0)
|
|---|
| 97 |
|
|---|
| 98 |
|
|---|
| 99 | #define CHECKANY(String, fExpected) \
|
|---|
| 100 | do { \
|
|---|
| 101 | bool fRc = RTNetStrIsIPv4AddrAny(String); \
|
|---|
| 102 | if (fRc != fExpected) \
|
|---|
| 103 | { \
|
|---|
| 104 | RTTestIFailed("at line %d: '%s':" \
|
|---|
| 105 | " expected %RTbool got %RTbool\n", \
|
|---|
| 106 | __LINE__, (String), fExpected, fRc); \
|
|---|
| 107 | } \
|
|---|
| 108 | } while (0)
|
|---|
| 109 |
|
|---|
| 110 | #define IS_ANY(String) CHECKANY((String), true)
|
|---|
| 111 | #define NOT_ANY(String) CHECKANY((String), false)
|
|---|
| 112 |
|
|---|
| 113 |
|
|---|
| 114 | #define CHECKMASK(_mask, _rcExpected, _iExpectedPrefix) \
|
|---|
| 115 | do { \
|
|---|
| 116 | /* const */ RTNETADDRIPV4 Mask; \
|
|---|
| 117 | int iExpectedPrefix = (_iExpectedPrefix); \
|
|---|
| 118 | int iPrefix; \
|
|---|
| 119 | const int rcExpected = (_rcExpected); \
|
|---|
| 120 | int rc2; \
|
|---|
| 121 | \
|
|---|
| 122 | Mask.u = RT_H2N_U32_C(UINT32_C(_mask)); \
|
|---|
| 123 | rc2 = RTNetMaskToPrefixIPv4(&Mask, &iPrefix); \
|
|---|
| 124 | \
|
|---|
| 125 | if (rcExpected == VINF_SUCCESS) \
|
|---|
| 126 | { \
|
|---|
| 127 | if (rc2 != rcExpected) \
|
|---|
| 128 | { \
|
|---|
| 129 | /* unexpected failure */ \
|
|---|
| 130 | RTTestIFailed("at line %d: mask %RTnaipv4:" \
|
|---|
| 131 | " expected prefix length %d got %Rrc\n", \
|
|---|
| 132 | __LINE__, Mask.u, \
|
|---|
| 133 | iExpectedPrefix, rc2); \
|
|---|
| 134 | } \
|
|---|
| 135 | else if (iPrefix != iExpectedPrefix) \
|
|---|
| 136 | { \
|
|---|
| 137 | /* unexpected result */ \
|
|---|
| 138 | RTTestIFailed("at line %d: mask %RTnaipv4:" \
|
|---|
| 139 | " expected prefix length %d got %d\n", \
|
|---|
| 140 | __LINE__, Mask.u, \
|
|---|
| 141 | iExpectedPrefix, iPrefix); \
|
|---|
| 142 | } \
|
|---|
| 143 | } \
|
|---|
| 144 | else /* expect failure */ \
|
|---|
| 145 | { \
|
|---|
| 146 | if (rc2 == VINF_SUCCESS) \
|
|---|
| 147 | { \
|
|---|
| 148 | /* unexpected success */ \
|
|---|
| 149 | RTTestIFailed("at line %d: mask %RTnaipv4:" \
|
|---|
| 150 | " expected %Rrc got prefix length %d\n", \
|
|---|
| 151 | __LINE__, Mask.u, \
|
|---|
| 152 | rcExpected, iPrefix); \
|
|---|
| 153 | } \
|
|---|
| 154 | else if (rc2 != rcExpected) \
|
|---|
| 155 | { \
|
|---|
| 156 | /* unexpected error */ \
|
|---|
| 157 | RTTestIFailed("at line %d: mask %RTnaipv4:" \
|
|---|
| 158 | " expected %Rrc got %Rrc\n", \
|
|---|
| 159 | __LINE__, Mask.u, \
|
|---|
| 160 | rcExpected, rc2); \
|
|---|
| 161 | } \
|
|---|
| 162 | } \
|
|---|
| 163 | } while (0)
|
|---|
| 164 |
|
|---|
| 165 | #define CHECKPREFIX(_prefix, _rcExpected, _mask) \
|
|---|
| 166 | do { \
|
|---|
| 167 | const int iPrefix = (_prefix); \
|
|---|
| 168 | RTNETADDRIPV4 ExpectedMask, Mask; \
|
|---|
| 169 | const int rcExpected = (_rcExpected); \
|
|---|
| 170 | int rc2; \
|
|---|
| 171 | \
|
|---|
| 172 | ExpectedMask.u = RT_H2N_U32_C(UINT32_C(_mask)); \
|
|---|
| 173 | rc2 = RTNetPrefixToMaskIPv4(iPrefix, &Mask); \
|
|---|
| 174 | \
|
|---|
| 175 | if (rcExpected == VINF_SUCCESS) \
|
|---|
| 176 | { \
|
|---|
| 177 | if (rc2 != rcExpected) \
|
|---|
| 178 | { \
|
|---|
| 179 | /* unexpected failure */ \
|
|---|
| 180 | RTTestIFailed("at line %d: prefix %d:" \
|
|---|
| 181 | " expected mask %RTnaipv4 got %Rrc\n", \
|
|---|
| 182 | __LINE__, iPrefix, \
|
|---|
| 183 | ExpectedMask.u, rc2); \
|
|---|
| 184 | } \
|
|---|
| 185 | else if (Mask.u != ExpectedMask.u) \
|
|---|
| 186 | { \
|
|---|
| 187 | /* unexpected result */ \
|
|---|
| 188 | RTTestIFailed("at line %d: prefix %d:" \
|
|---|
| 189 | " expected mask %RTnaipv4 got %RTnaipv4\n", \
|
|---|
| 190 | __LINE__, iPrefix, \
|
|---|
| 191 | ExpectedMask.u, Mask.u); \
|
|---|
| 192 | } \
|
|---|
| 193 | } \
|
|---|
| 194 | else /* expect failure */ \
|
|---|
| 195 | { \
|
|---|
| 196 | if (rc2 == VINF_SUCCESS) \
|
|---|
| 197 | { \
|
|---|
| 198 | /* unexpected success */ \
|
|---|
| 199 | RTTestIFailed("at line %d: prefix %d:" \
|
|---|
| 200 | " expected %Rrc got mask %RTnaipv4\n", \
|
|---|
| 201 | __LINE__, iPrefix, \
|
|---|
| 202 | rcExpected, Mask.u); \
|
|---|
| 203 | } \
|
|---|
| 204 | else if (rc2 != rcExpected) \
|
|---|
| 205 | { \
|
|---|
| 206 | /* unexpected error */ \
|
|---|
| 207 | RTTestIFailed("at line %d: prefix %d:" \
|
|---|
| 208 | " expected %Rrc got %Rrc\n", \
|
|---|
| 209 | __LINE__, iPrefix, \
|
|---|
| 210 | rcExpected, rc2); \
|
|---|
| 211 | } \
|
|---|
| 212 | } \
|
|---|
| 213 | } while (0)
|
|---|
| 214 |
|
|---|
| 215 | #define GOODMASK(_mask, _prefix) \
|
|---|
| 216 | do { \
|
|---|
| 217 | CHECKMASK(_mask, VINF_SUCCESS, _prefix); \
|
|---|
| 218 | CHECKPREFIX(_prefix, VINF_SUCCESS, _mask); \
|
|---|
| 219 | } while (0)
|
|---|
| 220 |
|
|---|
| 221 | #define BADMASK(_mask) \
|
|---|
| 222 | CHECKMASK(_mask, VERR_INVALID_PARAMETER, -1)
|
|---|
| 223 |
|
|---|
| 224 | #define BADPREFIX(_prefix) \
|
|---|
| 225 | CHECKPREFIX(_prefix, VERR_INVALID_PARAMETER, 0)
|
|---|
| 226 |
|
|---|
| 227 |
|
|---|
| 228 | int main()
|
|---|
| 229 | {
|
|---|
| 230 | RTTEST hTest;
|
|---|
| 231 | int rc = RTTestInitAndCreate("tstRTNetIPv4", &hTest);
|
|---|
| 232 | if (rc)
|
|---|
| 233 | return rc;
|
|---|
| 234 | RTTestBanner(hTest);
|
|---|
| 235 |
|
|---|
| 236 | GOODADDR("1.2.3.4", 0x01020304);
|
|---|
| 237 | GOODADDR("0.0.0.0", 0x00000000);
|
|---|
| 238 | GOODADDR("255.255.255.255", 0xFFFFFFFF);
|
|---|
| 239 |
|
|---|
| 240 | /* leading and trailing whitespace is allowed */
|
|---|
| 241 | GOODADDR(" 1.2.3.4 ", 0x01020304);
|
|---|
| 242 | GOODADDR("\t1.2.3.4\t", 0x01020304);
|
|---|
| 243 |
|
|---|
| 244 | BADADDR("1.2.3.4x");
|
|---|
| 245 | BADADDR("1.2.3.4.");
|
|---|
| 246 | BADADDR("1.2.3");
|
|---|
| 247 | BADADDR("0x1.2.3.4");
|
|---|
| 248 | BADADDR("666.2.3.4");
|
|---|
| 249 | BADADDR("1.666.3.4");
|
|---|
| 250 | BADADDR("1.2.666.4");
|
|---|
| 251 | BADADDR("1.2.3.666");
|
|---|
| 252 |
|
|---|
| 253 | /*
|
|---|
| 254 | * Parsing itself is covered by the tests above, here we only
|
|---|
| 255 | * check trailers
|
|---|
| 256 | */
|
|---|
| 257 | CHECKADDREX("1.2.3.4", "", VINF_SUCCESS, 0x01020304);
|
|---|
| 258 | CHECKADDREX("1.2.3.4", " ", VINF_SUCCESS, 0x01020304);
|
|---|
| 259 | CHECKADDREX("1.2.3.4", "x", VINF_SUCCESS, 0x01020304);
|
|---|
| 260 | CHECKADDREX("1.2.3.444", "", VERR_INVALID_PARAMETER, 0);
|
|---|
| 261 |
|
|---|
| 262 |
|
|---|
| 263 | IS_ANY("0.0.0.0");
|
|---|
| 264 | IS_ANY("\t 0.0.0.0 \t");
|
|---|
| 265 |
|
|---|
| 266 | NOT_ANY("1.1.1.1"); /* good address, but not INADDR_ANY */
|
|---|
| 267 | NOT_ANY("0.0.0.0x"); /* bad address */
|
|---|
| 268 |
|
|---|
| 269 |
|
|---|
| 270 | /*
|
|---|
| 271 | * The mask <-> prefix table is small so we can test it all.
|
|---|
| 272 | */
|
|---|
| 273 | GOODMASK(0x00000000, 0); /* 0.0.0.0 */
|
|---|
| 274 | GOODMASK(0x80000000, 1); /* 128.0.0.0 */
|
|---|
| 275 | GOODMASK(0xc0000000, 2); /* 192.0.0.0 */
|
|---|
| 276 | GOODMASK(0xe0000000, 3); /* 224.0.0.0 */
|
|---|
| 277 | GOODMASK(0xf0000000, 4); /* 240.0.0.0 */
|
|---|
| 278 | GOODMASK(0xf8000000, 5); /* 248.0.0.0 */
|
|---|
| 279 | GOODMASK(0xfc000000, 6); /* 252.0.0.0 */
|
|---|
| 280 | GOODMASK(0xfe000000, 7); /* 254.0.0.0 */
|
|---|
| 281 | GOODMASK(0xff000000, 8); /* 255.0.0.0 */
|
|---|
| 282 | GOODMASK(0xff800000, 9); /* 255.128.0.0 */
|
|---|
| 283 | GOODMASK(0xffc00000, 10); /* 255.192.0.0 */
|
|---|
| 284 | GOODMASK(0xffe00000, 11); /* 255.224.0.0 */
|
|---|
| 285 | GOODMASK(0xfff00000, 12); /* 255.240.0.0 */
|
|---|
| 286 | GOODMASK(0xfff80000, 13); /* 255.248.0.0 */
|
|---|
| 287 | GOODMASK(0xfffc0000, 14); /* 255.252.0.0 */
|
|---|
| 288 | GOODMASK(0xfffe0000, 15); /* 255.254.0.0 */
|
|---|
| 289 | GOODMASK(0xffff0000, 16); /* 255.255.0.0 */
|
|---|
| 290 | GOODMASK(0xffff8000, 17); /* 255.255.128.0 */
|
|---|
| 291 | GOODMASK(0xffffc000, 18); /* 255.255.192.0 */
|
|---|
| 292 | GOODMASK(0xffffe000, 19); /* 255.255.224.0 */
|
|---|
| 293 | GOODMASK(0xfffff000, 20); /* 255.255.240.0 */
|
|---|
| 294 | GOODMASK(0xfffff800, 21); /* 255.255.248.0 */
|
|---|
| 295 | GOODMASK(0xfffffc00, 22); /* 255.255.252.0 */
|
|---|
| 296 | GOODMASK(0xfffffe00, 23); /* 255.255.254.0 */
|
|---|
| 297 | GOODMASK(0xffffff00, 24); /* 255.255.255.0 */
|
|---|
| 298 | GOODMASK(0xffffff80, 25); /* 255.255.255.128 */
|
|---|
| 299 | GOODMASK(0xffffffc0, 26); /* 255.255.255.192 */
|
|---|
| 300 | GOODMASK(0xffffffe0, 27); /* 255.255.255.224 */
|
|---|
| 301 | GOODMASK(0xfffffff0, 28); /* 255.255.255.240 */
|
|---|
| 302 | GOODMASK(0xfffffff8, 29); /* 255.255.255.248 */
|
|---|
| 303 | GOODMASK(0xfffffffc, 30); /* 255.255.255.252 */
|
|---|
| 304 | GOODMASK(0xfffffffe, 31); /* 255.255.255.254 */
|
|---|
| 305 | GOODMASK(0xffffffff, 32); /* 255.255.255.255 */
|
|---|
| 306 |
|
|---|
| 307 | BADMASK(0x01020304);
|
|---|
| 308 |
|
|---|
| 309 | BADPREFIX(-1);
|
|---|
| 310 | BADPREFIX(33);
|
|---|
| 311 |
|
|---|
| 312 | return RTTestSummaryAndDestroy(hTest);
|
|---|
| 313 | }
|
|---|
