test/test/test_memzone.c

   1 /* SPDX-License-Identifier: BSD-3-Clause
   2  * Copyright(c) 2010-2014 Intel Corporation
   3  */
   4
   5 #include <stdio.h>
   6 #include <stdint.h>
   7 #include <string.h>
   8 #include <inttypes.h>
   9 #include <sys/queue.h>
  10
  11 #include <rte_random.h>
  12 #include <rte_cycles.h>
  13 #include <rte_memory.h>
  14 #include <rte_memzone.h>
  15 #include <rte_eal.h>
  16 #include <rte_eal_memconfig.h>
  17 #include <rte_common.h>
  18 #include <rte_string_fns.h>
  19 #include <rte_errno.h>
  20 #include <rte_malloc.h>
  21 #include "../../lib/librte_eal/common/malloc_elem.h"
  22
  23 #include "test.h"
  24
  25 /*
  26  * Memzone
  27  * =======
  28  *
  29  * - Search for three reserved zones or reserve them if they do not exist:
  30  *
  31  *   - One is on any socket id.
  32  *   - The second is on socket 0.
  33  *   - The last one is on socket 1 (if socket 1 exists).
  34  *
  35  * - Check that the zones exist.
  36  *
  37  * - Check that the zones are cache-aligned.
  38  *
  39  * - Check that zones do not overlap.
  40  *
  41  * - Check that the zones are on the correct socket id.
  42  *
  43  * - Check that a lookup of the first zone returns the same pointer.
  44  *
  45  * - Check that it is not possible to create another zone with the
  46  *   same name as an existing zone.
  47  *
  48  * - Check flags for specific huge page size reservation
  49  */
  50
  51 #define TEST_MEMZONE_NAME(suffix) "MZ_TEST_" suffix
  52
  53 /* Test if memory overlaps: return 1 if true, or 0 if false. */
  54 static int
  55 is_memory_overlap(rte_iova_t ptr1, size_t len1, rte_iova_t ptr2, size_t len2)
  56 {
  57         if (ptr2 >= ptr1 && (ptr2 - ptr1) < len1)
  58                 return 1;
  59         else if (ptr2 < ptr1 && (ptr1 - ptr2) < len2)
  60                 return 1;
  61         return 0;
  62 }
  63
  64 static int
  65 test_memzone_invalid_alignment(void)
  66 {
  67         const struct rte_memzone * mz;
  68
  69         mz = rte_memzone_lookup(TEST_MEMZONE_NAME("invalid_alignment"));
  70         if (mz != NULL) {
  71                 printf("Zone with invalid alignment has been reserved\n");
  72                 return -1;
  73         }
  74
  75         mz = rte_memzone_reserve_aligned(TEST_MEMZONE_NAME("invalid_alignment"),
  76                                          100, SOCKET_ID_ANY, 0, 100);
  77         if (mz != NULL) {
  78                 printf("Zone with invalid alignment has been reserved\n");
  79                 return -1;
  80         }
  81         return 0;
  82 }
  83
  84 static int
  85 test_memzone_reserving_zone_size_bigger_than_the_maximum(void)
  86 {
  87         const struct rte_memzone * mz;
  88
  89         mz = rte_memzone_lookup(
  90                         TEST_MEMZONE_NAME("zone_size_bigger_than_the_maximum"));
  91         if (mz != NULL) {
  92                 printf("zone_size_bigger_than_the_maximum has been reserved\n");
  93                 return -1;
  94         }
  95
  96         mz = rte_memzone_reserve(
  97                         TEST_MEMZONE_NAME("zone_size_bigger_than_the_maximum"),
  98                         (size_t)-1, SOCKET_ID_ANY, 0);
  99         if (mz != NULL) {
 100                 printf("It is impossible to reserve such big a memzone\n");
 101                 return -1;
 102         }
 103
 104         return 0;
 105 }
 106
 107 struct walk_arg {
 108         int hugepage_2MB_avail;
 109         int hugepage_1GB_avail;
 110         int hugepage_16MB_avail;
 111         int hugepage_16GB_avail;
 112 };
 113 static int
 114 find_available_pagesz(const struct rte_memseg *ms, void *arg)
 115 {
 116         struct walk_arg *wa = arg;
 117
 118         if (ms->hugepage_sz == RTE_PGSIZE_2M)
 119                 wa->hugepage_2MB_avail = 1;
 120         if (ms->hugepage_sz == RTE_PGSIZE_1G)
 121                 wa->hugepage_1GB_avail = 1;
 122         if (ms->hugepage_sz == RTE_PGSIZE_16M)
 123                 wa->hugepage_16MB_avail = 1;
 124         if (ms->hugepage_sz == RTE_PGSIZE_16G)
 125                 wa->hugepage_16GB_avail = 1;
 126
 127         return 0;
 128 }
 129
 130 static int
 131 test_memzone_reserve_flags(void)
 132 {
 133         const struct rte_memzone *mz;
 134         struct walk_arg wa;
 135         int hugepage_2MB_avail, hugepage_1GB_avail;
 136         int hugepage_16MB_avail, hugepage_16GB_avail;
 137         const size_t size = 100;
 138
 139         memset(&wa, 0, sizeof(wa));
 140
 141         rte_memseg_walk(find_available_pagesz, &wa);
 142
 143         hugepage_2MB_avail = wa.hugepage_2MB_avail;
 144         hugepage_1GB_avail = wa.hugepage_1GB_avail;
 145         hugepage_16MB_avail = wa.hugepage_16MB_avail;
 146         hugepage_16GB_avail = wa.hugepage_16GB_avail;
 147
 148         /* Display the availability of 2MB ,1GB, 16MB, 16GB pages */
 149         if (hugepage_2MB_avail)
 150                 printf("2MB Huge pages available\n");
 151         if (hugepage_1GB_avail)
 152                 printf("1GB Huge pages available\n");
 153         if (hugepage_16MB_avail)
 154                 printf("16MB Huge pages available\n");
 155         if (hugepage_16GB_avail)
 156                 printf("16GB Huge pages available\n");
 157         /*
 158          * If 2MB pages available, check that a small memzone is correctly
 159          * reserved from 2MB huge pages when requested by the RTE_MEMZONE_2MB flag.
 160          * Also check that RTE_MEMZONE_SIZE_HINT_ONLY flag only defaults to an
 161          * available page size (i.e 1GB ) when 2MB pages are unavailable.
 162          */
 163         if (hugepage_2MB_avail) {
 164                 mz = rte_memzone_reserve(TEST_MEMZONE_NAME("flag_zone_2M"),
 165                                 size, SOCKET_ID_ANY, RTE_MEMZONE_2MB);
 166                 if (mz == NULL) {
 167                         printf("MEMZONE FLAG 2MB\n");
 168                         return -1;
 169                 }
 170                 if (mz->hugepage_sz != RTE_PGSIZE_2M) {
 171                         printf("hugepage_sz not equal 2M\n");
 172                         return -1;
 173                 }
 174                 if (rte_memzone_free(mz)) {
 175                         printf("Fail memzone free\n");
 176                         return -1;
 177                 }
 178
 179                 mz = rte_memzone_reserve(TEST_MEMZONE_NAME("flag_zone_2M_HINT"),
 180                                 size, SOCKET_ID_ANY,
 181                                 RTE_MEMZONE_2MB|RTE_MEMZONE_SIZE_HINT_ONLY);
 182                 if (mz == NULL) {
 183                         printf("MEMZONE FLAG 2MB\n");
 184                         return -1;
 185                 }
 186                 if (mz->hugepage_sz != RTE_PGSIZE_2M) {
 187                         printf("hugepage_sz not equal 2M\n");
 188                         return -1;
 189                 }
 190                 if (rte_memzone_free(mz)) {
 191                         printf("Fail memzone free\n");
 192                         return -1;
 193                 }
 194
 195                 /* Check if 1GB huge pages are unavailable, that function fails unless
 196                  * HINT flag is indicated
 197                  */
 198                 if (!hugepage_1GB_avail) {
 199                         mz = rte_memzone_reserve(
 200                                         TEST_MEMZONE_NAME("flag_zone_1G_HINT"),
 201                                         size, SOCKET_ID_ANY,
 202                                         RTE_MEMZONE_1GB|RTE_MEMZONE_SIZE_HINT_ONLY);
 203                         if (mz == NULL) {
 204                                 printf("MEMZONE FLAG 1GB & HINT\n");
 205                                 return -1;
 206                         }
 207                         if (mz->hugepage_sz != RTE_PGSIZE_2M) {
 208                                 printf("hugepage_sz not equal 2M\n");
 209                                 return -1;
 210                         }
 211                         if (rte_memzone_free(mz)) {
 212                                 printf("Fail memzone free\n");
 213                                 return -1;
 214                         }
 215
 216                         mz = rte_memzone_reserve(
 217                                         TEST_MEMZONE_NAME("flag_zone_1G"), size,
 218                                         SOCKET_ID_ANY, RTE_MEMZONE_1GB);
 219                         if (mz != NULL) {
 220                                 printf("MEMZONE FLAG 1GB\n");
 221                                 return -1;
 222                         }
 223                 }
 224         }
 225
 226         /*As with 2MB tests above for 1GB huge page requests*/
 227         if (hugepage_1GB_avail) {
 228                 mz = rte_memzone_reserve(TEST_MEMZONE_NAME("flag_zone_1G"),
 229                                 size, SOCKET_ID_ANY, RTE_MEMZONE_1GB);
 230                 if (mz == NULL) {
 231                         printf("MEMZONE FLAG 1GB\n");
 232                         return -1;
 233                 }
 234                 if (mz->hugepage_sz != RTE_PGSIZE_1G) {
 235                         printf("hugepage_sz not equal 1G\n");
 236                         return -1;
 237                 }
 238                 if (rte_memzone_free(mz)) {
 239                         printf("Fail memzone free\n");
 240                         return -1;
 241                 }
 242
 243                 mz = rte_memzone_reserve(TEST_MEMZONE_NAME("flag_zone_1G_HINT"),
 244                                 size, SOCKET_ID_ANY,
 245                                 RTE_MEMZONE_1GB|RTE_MEMZONE_SIZE_HINT_ONLY);
 246                 if (mz == NULL) {
 247                         printf("MEMZONE FLAG 1GB\n");
 248                         return -1;
 249                 }
 250                 if (mz->hugepage_sz != RTE_PGSIZE_1G) {
 251                         printf("hugepage_sz not equal 1G\n");
 252                         return -1;
 253                 }
 254                 if (rte_memzone_free(mz)) {
 255                         printf("Fail memzone free\n");
 256                         return -1;
 257                 }
 258
 259                 /* Check if 1GB huge pages are unavailable, that function fails unless
 260                  * HINT flag is indicated
 261                  */
 262                 if (!hugepage_2MB_avail) {
 263                         mz = rte_memzone_reserve(
 264                                         TEST_MEMZONE_NAME("flag_zone_2M_HINT"),
 265                                         size, SOCKET_ID_ANY,
 266                                         RTE_MEMZONE_2MB|RTE_MEMZONE_SIZE_HINT_ONLY);
 267                         if (mz == NULL){
 268                                 printf("MEMZONE FLAG 2MB & HINT\n");
 269                                 return -1;
 270                         }
 271                         if (mz->hugepage_sz != RTE_PGSIZE_1G) {
 272                                 printf("hugepage_sz not equal 1G\n");
 273                                 return -1;
 274                         }
 275                         if (rte_memzone_free(mz)) {
 276                                 printf("Fail memzone free\n");
 277                                 return -1;
 278                         }
 279                         mz = rte_memzone_reserve(
 280                                         TEST_MEMZONE_NAME("flag_zone_2M"), size,
 281                                         SOCKET_ID_ANY, RTE_MEMZONE_2MB);
 282                         if (mz != NULL) {
 283                                 printf("MEMZONE FLAG 2MB\n");
 284                                 return -1;
 285                         }
 286                 }
 287
 288                 if (hugepage_2MB_avail && hugepage_1GB_avail) {
 289                         mz = rte_memzone_reserve(
 290                                         TEST_MEMZONE_NAME("flag_zone_2M_HINT"),
 291                                         size, SOCKET_ID_ANY,
 292                                         RTE_MEMZONE_2MB|RTE_MEMZONE_1GB);
 293                         if (mz == NULL) {
 294                                 printf("BOTH SIZES SET\n");
 295                                 return -1;
 296                         }
 297                         if (mz->hugepage_sz != RTE_PGSIZE_1G &&
 298                                         mz->hugepage_sz != RTE_PGSIZE_2M) {
 299                                 printf("Wrong size when both sizes set\n");
 300                                 return -1;
 301                         }
 302                         if (rte_memzone_free(mz)) {
 303                                 printf("Fail memzone free\n");
 304                                 return -1;
 305                         }
 306                 }
 307         }
 308         /*
 309          * This option is for IBM Power. If 16MB pages available, check
 310          * that a small memzone is correctly reserved from 16MB huge pages
 311          * when requested by the RTE_MEMZONE_16MB flag. Also check that
 312          * RTE_MEMZONE_SIZE_HINT_ONLY flag only defaults to an available
 313          * page size (i.e 16GB ) when 16MB pages are unavailable.
 314          */
 315         if (hugepage_16MB_avail) {
 316                 mz = rte_memzone_reserve(TEST_MEMZONE_NAME("flag_zone_16M"),
 317                                 size, SOCKET_ID_ANY, RTE_MEMZONE_16MB);
 318                 if (mz == NULL) {
 319                         printf("MEMZONE FLAG 16MB\n");
 320                         return -1;
 321                 }
 322                 if (mz->hugepage_sz != RTE_PGSIZE_16M) {
 323                         printf("hugepage_sz not equal 16M\n");
 324                         return -1;
 325                 }
 326                 if (rte_memzone_free(mz)) {
 327                         printf("Fail memzone free\n");
 328                         return -1;
 329                 }
 330
 331                 mz = rte_memzone_reserve(
 332                                 TEST_MEMZONE_NAME("flag_zone_16M_HINT"), size,
 333                                 SOCKET_ID_ANY,
 334                                 RTE_MEMZONE_16MB|RTE_MEMZONE_SIZE_HINT_ONLY);
 335                 if (mz == NULL) {
 336                         printf("MEMZONE FLAG 16MB\n");
 337                         return -1;
 338                 }
 339                 if (mz->hugepage_sz != RTE_PGSIZE_16M) {
 340                         printf("hugepage_sz not equal 16M\n");
 341                         return -1;
 342                 }
 343                 if (rte_memzone_free(mz)) {
 344                         printf("Fail memzone free\n");
 345                         return -1;
 346                 }
 347
 348                 /* Check if 1GB huge pages are unavailable, that function fails
 349                  * unless HINT flag is indicated
 350                  */
 351                 if (!hugepage_16GB_avail) {
 352                         mz = rte_memzone_reserve(
 353                                         TEST_MEMZONE_NAME("flag_zone_16G_HINT"),
 354                                         size, SOCKET_ID_ANY,
 355                                         RTE_MEMZONE_16GB |
 356                                         RTE_MEMZONE_SIZE_HINT_ONLY);
 357                         if (mz == NULL) {
 358                                 printf("MEMZONE FLAG 16GB & HINT\n");
 359                                 return -1;
 360                         }
 361                         if (mz->hugepage_sz != RTE_PGSIZE_16M) {
 362                                 printf("hugepage_sz not equal 16M\n");
 363                                 return -1;
 364                         }
 365                         if (rte_memzone_free(mz)) {
 366                                 printf("Fail memzone free\n");
 367                                 return -1;
 368                         }
 369
 370                         mz = rte_memzone_reserve(
 371                                         TEST_MEMZONE_NAME("flag_zone_16G"),
 372                                         size,
 373                                         SOCKET_ID_ANY, RTE_MEMZONE_16GB);
 374                         if (mz != NULL) {
 375                                 printf("MEMZONE FLAG 16GB\n");
 376                                 return -1;
 377                         }
 378                 }
 379         }
 380         /*As with 16MB tests above for 16GB huge page requests*/
 381         if (hugepage_16GB_avail) {
 382                 mz = rte_memzone_reserve(TEST_MEMZONE_NAME("flag_zone_16G"),
 383                                 size, SOCKET_ID_ANY, RTE_MEMZONE_16GB);
 384                 if (mz == NULL) {
 385                         printf("MEMZONE FLAG 16GB\n");
 386                         return -1;
 387                 }
 388                 if (mz->hugepage_sz != RTE_PGSIZE_16G) {
 389                         printf("hugepage_sz not equal 16G\n");
 390                         return -1;
 391                 }
 392                 if (rte_memzone_free(mz)) {
 393                         printf("Fail memzone free\n");
 394                         return -1;
 395                 }
 396
 397                 mz = rte_memzone_reserve(
 398                                 TEST_MEMZONE_NAME("flag_zone_16G_HINT"), size,
 399                                 SOCKET_ID_ANY,
 400                                 RTE_MEMZONE_16GB|RTE_MEMZONE_SIZE_HINT_ONLY);
 401                 if (mz == NULL) {
 402                         printf("MEMZONE FLAG 16GB\n");
 403                         return -1;
 404                 }
 405                 if (mz->hugepage_sz != RTE_PGSIZE_16G) {
 406                         printf("hugepage_sz not equal 16G\n");
 407                         return -1;
 408                 }
 409                 if (rte_memzone_free(mz)) {
 410                         printf("Fail memzone free\n");
 411                         return -1;
 412                 }
 413
 414                 /* Check if 1GB huge pages are unavailable, that function fails
 415                  * unless HINT flag is indicated
 416                  */
 417                 if (!hugepage_16MB_avail) {
 418                         mz = rte_memzone_reserve(
 419                                         TEST_MEMZONE_NAME("flag_zone_16M_HINT"),
 420                                         size, SOCKET_ID_ANY,
 421                                         RTE_MEMZONE_16MB |
 422                                         RTE_MEMZONE_SIZE_HINT_ONLY);
 423                         if (mz == NULL) {
 424                                 printf("MEMZONE FLAG 16MB & HINT\n");
 425                                 return -1;
 426                         }
 427                         if (mz->hugepage_sz != RTE_PGSIZE_16G) {
 428                                 printf("hugepage_sz not equal 16G\n");
 429                                 return -1;
 430                         }
 431                         if (rte_memzone_free(mz)) {
 432                                 printf("Fail memzone free\n");
 433                                 return -1;
 434                         }
 435                         mz = rte_memzone_reserve(
 436                                         TEST_MEMZONE_NAME("flag_zone_16M"),
 437                                         size, SOCKET_ID_ANY, RTE_MEMZONE_16MB);
 438                         if (mz != NULL) {
 439                                 printf("MEMZONE FLAG 16MB\n");
 440                                 return -1;
 441                         }
 442                 }
 443
 444                 if (hugepage_16MB_avail && hugepage_16GB_avail) {
 445                         mz = rte_memzone_reserve(
 446                                         TEST_MEMZONE_NAME("flag_zone_16M_HINT"),
 447                                         size, SOCKET_ID_ANY,
 448                                         RTE_MEMZONE_16MB|RTE_MEMZONE_16GB);
 449                         if (mz == NULL) {
 450                                 printf("BOTH SIZES SET\n");
 451                                 return -1;
 452                         }
 453                         if (mz->hugepage_sz != RTE_PGSIZE_16G &&
 454                                         mz->hugepage_sz != RTE_PGSIZE_16M) {
 455                                 printf("Wrong size when both sizes set\n");
 456                                 return -1;
 457                         }
 458                         if (rte_memzone_free(mz)) {
 459                                 printf("Fail memzone free\n");
 460                                 return -1;
 461                         }
 462                 }
 463         }
 464         return 0;
 465 }
 466
 467
 468 /* Find the heap with the greatest free block size */
 469 static size_t
 470 find_max_block_free_size(const unsigned _align)
 471 {
 472         struct rte_malloc_socket_stats stats;
 473         unsigned i, align = _align;
 474         size_t len = 0;
 475
 476         for (i = 0; i < RTE_MAX_NUMA_NODES; i++) {
 477                 rte_malloc_get_socket_stats(i, &stats);
 478                 if (stats.greatest_free_size > len)
 479                         len = stats.greatest_free_size;
 480         }
 481
 482         if (align < RTE_CACHE_LINE_SIZE)
 483                 align = RTE_CACHE_LINE_ROUNDUP(align+1);
 484
 485         if (len <= MALLOC_ELEM_OVERHEAD + align)
 486                 return 0;
 487
 488         return len - MALLOC_ELEM_OVERHEAD - align;
 489 }
 490
 491 static int
 492 test_memzone_reserve_max(void)
 493 {
 494         const struct rte_memzone *mz;
 495         size_t maxlen;
 496
 497         maxlen = find_max_block_free_size(0);
 498
 499         if (maxlen == 0) {
 500                 printf("There is no space left!\n");
 501                 return 0;
 502         }
 503
 504         mz = rte_memzone_reserve(TEST_MEMZONE_NAME("max_zone"), 0,
 505                         SOCKET_ID_ANY, 0);
 506         if (mz == NULL){
 507                 printf("Failed to reserve a big chunk of memory - %s\n",
 508                                 rte_strerror(rte_errno));
 509                 rte_dump_physmem_layout(stdout);
 510                 rte_memzone_dump(stdout);
 511                 return -1;
 512         }
 513
 514         if (mz->len != maxlen) {
 515                 printf("Memzone reserve with 0 size did not return bigest block\n");
 516                 printf("Expected size = %zu, actual size = %zu\n", maxlen, mz->len);
 517                 rte_dump_physmem_layout(stdout);
 518                 rte_memzone_dump(stdout);
 519                 return -1;
 520         }
 521
 522         if (rte_memzone_free(mz)) {
 523                 printf("Fail memzone free\n");
 524                 return -1;
 525         }
 526
 527         return 0;
 528 }
 529
 530 static int
 531 test_memzone_reserve_max_aligned(void)
 532 {
 533         const struct rte_memzone *mz;
 534         size_t maxlen = 0;
 535
 536         /* random alignment */
 537         rte_srand((unsigned)rte_rdtsc());
 538         const unsigned align = 1 << ((rte_rand() % 8) + 5); /* from 128 up to 4k alignment */
 539
 540         maxlen = find_max_block_free_size(align);
 541
 542         if (maxlen == 0) {
 543                 printf("There is no space left for biggest %u-aligned memzone!\n", align);
 544                 return 0;
 545         }
 546
 547         mz = rte_memzone_reserve_aligned(TEST_MEMZONE_NAME("max_zone_aligned"),
 548                         0, SOCKET_ID_ANY, 0, align);
 549         if (mz == NULL){
 550                 printf("Failed to reserve a big chunk of memory - %s\n",
 551                                 rte_strerror(rte_errno));
 552                 rte_dump_physmem_layout(stdout);
 553                 rte_memzone_dump(stdout);
 554                 return -1;
 555         }
 556
 557         if (mz->len != maxlen) {
 558                 printf("Memzone reserve with 0 size and alignment %u did not return"
 559                                 " bigest block\n", align);
 560                 printf("Expected size = %zu, actual size = %zu\n",
 561                                 maxlen, mz->len);
 562                 rte_dump_physmem_layout(stdout);
 563                 rte_memzone_dump(stdout);
 564                 return -1;
 565         }
 566
 567         if (rte_memzone_free(mz)) {
 568                 printf("Fail memzone free\n");
 569                 return -1;
 570         }
 571
 572         return 0;
 573 }
 574
 575 static int
 576 test_memzone_aligned(void)
 577 {
 578         const struct rte_memzone *memzone_aligned_32;
 579         const struct rte_memzone *memzone_aligned_128;
 580         const struct rte_memzone *memzone_aligned_256;
 581         const struct rte_memzone *memzone_aligned_512;
 582         const struct rte_memzone *memzone_aligned_1024;
 583
 584         /* memzone that should automatically be adjusted to align on 64 bytes */
 585         memzone_aligned_32 = rte_memzone_reserve_aligned(
 586                         TEST_MEMZONE_NAME("aligned_32"), 100, SOCKET_ID_ANY, 0,
 587                         32);
 588
 589         /* memzone that is supposed to be aligned on a 128 byte boundary */
 590         memzone_aligned_128 = rte_memzone_reserve_aligned(
 591                         TEST_MEMZONE_NAME("aligned_128"), 100, SOCKET_ID_ANY, 0,
 592                         128);
 593
 594         /* memzone that is supposed to be aligned on a 256 byte boundary */
 595         memzone_aligned_256 = rte_memzone_reserve_aligned(
 596                         TEST_MEMZONE_NAME("aligned_256"), 100, SOCKET_ID_ANY, 0,
 597                         256);
 598
 599         /* memzone that is supposed to be aligned on a 512 byte boundary */
 600         memzone_aligned_512 = rte_memzone_reserve_aligned(
 601                         TEST_MEMZONE_NAME("aligned_512"), 100, SOCKET_ID_ANY, 0,
 602                         512);
 603
 604         /* memzone that is supposed to be aligned on a 1024 byte boundary */
 605         memzone_aligned_1024 = rte_memzone_reserve_aligned(
 606                         TEST_MEMZONE_NAME("aligned_1024"), 100, SOCKET_ID_ANY,
 607                         0, 1024);
 608
 609         printf("check alignments and lengths\n");
 610         if (memzone_aligned_32 == NULL) {
 611                 printf("Unable to reserve 64-byte aligned memzone!\n");
 612                 return -1;
 613         }
 614         if ((memzone_aligned_32->iova & RTE_CACHE_LINE_MASK) != 0)
 615                 return -1;
 616         if (((uintptr_t) memzone_aligned_32->addr & RTE_CACHE_LINE_MASK) != 0)
 617                 return -1;
 618         if ((memzone_aligned_32->len & RTE_CACHE_LINE_MASK) != 0)
 619                 return -1;
 620
 621         if (memzone_aligned_128 == NULL) {
 622                 printf("Unable to reserve 128-byte aligned memzone!\n");
 623                 return -1;
 624         }
 625         if ((memzone_aligned_128->iova & 127) != 0)
 626                 return -1;
 627         if (((uintptr_t) memzone_aligned_128->addr & 127) != 0)
 628                 return -1;
 629         if ((memzone_aligned_128->len & RTE_CACHE_LINE_MASK) != 0)
 630                 return -1;
 631
 632         if (memzone_aligned_256 == NULL) {
 633                 printf("Unable to reserve 256-byte aligned memzone!\n");
 634                 return -1;
 635         }
 636         if ((memzone_aligned_256->iova & 255) != 0)
 637                 return -1;
 638         if (((uintptr_t) memzone_aligned_256->addr & 255) != 0)
 639                 return -1;
 640         if ((memzone_aligned_256->len & RTE_CACHE_LINE_MASK) != 0)
 641                 return -1;
 642
 643         if (memzone_aligned_512 == NULL) {
 644                 printf("Unable to reserve 512-byte aligned memzone!\n");
 645                 return -1;
 646         }
 647         if ((memzone_aligned_512->iova & 511) != 0)
 648                 return -1;
 649         if (((uintptr_t) memzone_aligned_512->addr & 511) != 0)
 650                 return -1;
 651         if ((memzone_aligned_512->len & RTE_CACHE_LINE_MASK) != 0)
 652                 return -1;
 653
 654         if (memzone_aligned_1024 == NULL) {
 655                 printf("Unable to reserve 1024-byte aligned memzone!\n");
 656                 return -1;
 657         }
 658         if ((memzone_aligned_1024->iova & 1023) != 0)
 659                 return -1;
 660         if (((uintptr_t) memzone_aligned_1024->addr & 1023) != 0)
 661                 return -1;
 662         if ((memzone_aligned_1024->len & RTE_CACHE_LINE_MASK) != 0)
 663                 return -1;
 664
 665         /* check that zones don't overlap */
 666         printf("check overlapping\n");
 667         if (is_memory_overlap(memzone_aligned_32->iova, memzone_aligned_32->len,
 668                                         memzone_aligned_128->iova, memzone_aligned_128->len))
 669                 return -1;
 670         if (is_memory_overlap(memzone_aligned_32->iova, memzone_aligned_32->len,
 671                                         memzone_aligned_256->iova, memzone_aligned_256->len))
 672                 return -1;
 673         if (is_memory_overlap(memzone_aligned_32->iova, memzone_aligned_32->len,
 674                                         memzone_aligned_512->iova, memzone_aligned_512->len))
 675                 return -1;
 676         if (is_memory_overlap(memzone_aligned_32->iova, memzone_aligned_32->len,
 677                                         memzone_aligned_1024->iova, memzone_aligned_1024->len))
 678                 return -1;
 679         if (is_memory_overlap(memzone_aligned_128->iova, memzone_aligned_128->len,
 680                                         memzone_aligned_256->iova, memzone_aligned_256->len))
 681                 return -1;
 682         if (is_memory_overlap(memzone_aligned_128->iova, memzone_aligned_128->len,
 683                                         memzone_aligned_512->iova, memzone_aligned_512->len))
 684                 return -1;
 685         if (is_memory_overlap(memzone_aligned_128->iova, memzone_aligned_128->len,
 686                                         memzone_aligned_1024->iova, memzone_aligned_1024->len))
 687                 return -1;
 688         if (is_memory_overlap(memzone_aligned_256->iova, memzone_aligned_256->len,
 689                                         memzone_aligned_512->iova, memzone_aligned_512->len))
 690                 return -1;
 691         if (is_memory_overlap(memzone_aligned_256->iova, memzone_aligned_256->len,
 692                                         memzone_aligned_1024->iova, memzone_aligned_1024->len))
 693                 return -1;
 694         if (is_memory_overlap(memzone_aligned_512->iova, memzone_aligned_512->len,
 695                                         memzone_aligned_1024->iova, memzone_aligned_1024->len))
 696                 return -1;
 697
 698         /* free all used zones */
 699         if (rte_memzone_free(memzone_aligned_32)) {
 700                 printf("Fail memzone free\n");
 701                 return -1;
 702         }
 703         if (rte_memzone_free(memzone_aligned_128)) {
 704                 printf("Fail memzone free\n");
 705                 return -1;
 706         }
 707         if (rte_memzone_free(memzone_aligned_256)) {
 708                 printf("Fail memzone free\n");
 709                 return -1;
 710         }
 711         if (rte_memzone_free(memzone_aligned_512)) {
 712                 printf("Fail memzone free\n");
 713                 return -1;
 714         }
 715         if (rte_memzone_free(memzone_aligned_1024)) {
 716                 printf("Fail memzone free\n");
 717                 return -1;
 718         }
 719         return 0;
 720 }
 721
 722 static int
 723 check_memzone_bounded(const char *name, uint32_t len,  uint32_t align,
 724         uint32_t bound)
 725 {
 726         const struct rte_memzone *mz;
 727         rte_iova_t bmask;
 728
 729         bmask = ~((rte_iova_t)bound - 1);
 730
 731         if ((mz = rte_memzone_reserve_bounded(name, len, SOCKET_ID_ANY, 0,
 732                         align, bound)) == NULL) {
 733                 printf("%s(%s): memzone creation failed\n",
 734                         __func__, name);
 735                 return -1;
 736         }
 737
 738         if ((mz->iova & ((rte_iova_t)align - 1)) != 0) {
 739                 printf("%s(%s): invalid phys addr alignment\n",
 740                         __func__, mz->name);
 741                 return -1;
 742         }
 743
 744         if (((uintptr_t) mz->addr & ((uintptr_t)align - 1)) != 0) {
 745                 printf("%s(%s): invalid virtual addr alignment\n",
 746                         __func__, mz->name);
 747                 return -1;
 748         }
 749
 750         if ((mz->len & RTE_CACHE_LINE_MASK) != 0 || mz->len < len ||
 751                         mz->len < RTE_CACHE_LINE_SIZE) {
 752                 printf("%s(%s): invalid length\n",
 753                         __func__, mz->name);
 754                 return -1;
 755         }
 756
 757         if ((mz->iova & bmask) !=
 758                         ((mz->iova + mz->len - 1) & bmask)) {
 759                 printf("%s(%s): invalid memzone boundary %u crossed\n",
 760                         __func__, mz->name, bound);
 761                 return -1;
 762         }
 763
 764         if (rte_memzone_free(mz)) {
 765                 printf("Fail memzone free\n");
 766                 return -1;
 767         }
 768
 769         return 0;
 770 }
 771
 772 static int
 773 test_memzone_bounded(void)
 774 {
 775         const struct rte_memzone *memzone_err;
 776         int rc;
 777
 778         /* should fail as boundary is not power of two */
 779         memzone_err = rte_memzone_reserve_bounded(
 780                         TEST_MEMZONE_NAME("bounded_error_31"), 100,
 781                         SOCKET_ID_ANY, 0, 32, UINT32_MAX);
 782         if (memzone_err != NULL) {
 783                 printf("%s(%s)created a memzone with invalid boundary "
 784                         "conditions\n", __func__, memzone_err->name);
 785                 return -1;
 786         }
 787
 788         /* should fail as len is greater then boundary */
 789         memzone_err = rte_memzone_reserve_bounded(
 790                         TEST_MEMZONE_NAME("bounded_error_32"), 100,
 791                         SOCKET_ID_ANY, 0, 32, 32);
 792         if (memzone_err != NULL) {
 793                 printf("%s(%s)created a memzone with invalid boundary "
 794                         "conditions\n", __func__, memzone_err->name);
 795                 return -1;
 796         }
 797
 798         rc = check_memzone_bounded(TEST_MEMZONE_NAME("bounded_128"), 100, 128,
 799                         128);
 800         if (rc != 0)
 801                 return rc;
 802
 803         rc = check_memzone_bounded(TEST_MEMZONE_NAME("bounded_256"), 100, 256,
 804                         128);
 805         if (rc != 0)
 806                 return rc;
 807
 808         rc = check_memzone_bounded(TEST_MEMZONE_NAME("bounded_1K"), 100, 64,
 809                         1024);
 810         if (rc != 0)
 811                 return rc;
 812
 813         rc = check_memzone_bounded(TEST_MEMZONE_NAME("bounded_1K_MAX"), 0, 64,
 814                         1024);
 815         if (rc != 0)
 816                 return rc;
 817
 818         return 0;
 819 }
 820
 821 static int
 822 test_memzone_free(void)
 823 {
 824         const struct rte_memzone *mz[RTE_MAX_MEMZONE + 1];
 825         int i;
 826         char name[20];
 827
 828         mz[0] = rte_memzone_reserve(TEST_MEMZONE_NAME("tempzone0"), 2000,
 829                         SOCKET_ID_ANY, 0);
 830         mz[1] = rte_memzone_reserve(TEST_MEMZONE_NAME("tempzone1"), 4000,
 831                         SOCKET_ID_ANY, 0);
 832
 833         if (mz[0] > mz[1])
 834                 return -1;
 835         if (!rte_memzone_lookup(TEST_MEMZONE_NAME("tempzone0")))
 836                 return -1;
 837         if (!rte_memzone_lookup(TEST_MEMZONE_NAME("tempzone1")))
 838                 return -1;
 839
 840         if (rte_memzone_free(mz[0])) {
 841                 printf("Fail memzone free - tempzone0\n");
 842                 return -1;
 843         }
 844         if (rte_memzone_lookup(TEST_MEMZONE_NAME("tempzone0"))) {
 845                 printf("Found previously free memzone - tempzone0\n");
 846                 return -1;
 847         }
 848         mz[2] = rte_memzone_reserve(TEST_MEMZONE_NAME("tempzone2"), 2000,
 849                         SOCKET_ID_ANY, 0);
 850
 851         if (mz[2] > mz[1]) {
 852                 printf("tempzone2 should have gotten the free entry from tempzone0\n");
 853                 return -1;
 854         }
 855         if (rte_memzone_free(mz[2])) {
 856                 printf("Fail memzone free - tempzone2\n");
 857                 return -1;
 858         }
 859         if (rte_memzone_lookup(TEST_MEMZONE_NAME("tempzone2"))) {
 860                 printf("Found previously free memzone - tempzone2\n");
 861                 return -1;
 862         }
 863         if (rte_memzone_free(mz[1])) {
 864                 printf("Fail memzone free - tempzone1\n");
 865                 return -1;
 866         }
 867         if (rte_memzone_lookup(TEST_MEMZONE_NAME("tempzone1"))) {
 868                 printf("Found previously free memzone - tempzone1\n");
 869                 return -1;
 870         }
 871
 872         i = 0;
 873         do {
 874                 snprintf(name, sizeof(name), TEST_MEMZONE_NAME("tempzone%u"),
 875                                 i);
 876                 mz[i] = rte_memzone_reserve(name, 1, SOCKET_ID_ANY, 0);
 877         } while (mz[i++] != NULL);
 878
 879         if (rte_memzone_free(mz[0])) {
 880                 printf("Fail memzone free - tempzone0\n");
 881                 return -1;
 882         }
 883         mz[0] = rte_memzone_reserve(TEST_MEMZONE_NAME("tempzone0new"), 0,
 884                         SOCKET_ID_ANY, 0);
 885
 886         if (mz[0] == NULL) {
 887                 printf("Fail to create memzone - tempzone0new - when MAX memzones were "
 888                                 "created and one was free\n");
 889                 return -1;
 890         }
 891
 892         for (i = i - 2; i >= 0; i--) {
 893                 if (rte_memzone_free(mz[i])) {
 894                         printf("Fail memzone free - tempzone%d\n", i);
 895                         return -1;
 896                 }
 897         }
 898
 899         return 0;
 900 }
 901
 902 static int
 903 test_memzone_basic(void)
 904 {
 905         const struct rte_memzone *memzone1;
 906         const struct rte_memzone *memzone2;
 907         const struct rte_memzone *memzone3;
 908         const struct rte_memzone *memzone4;
 909         const struct rte_memzone *mz;
 910         int memzone_cnt_after, memzone_cnt_expected;
 911         int memzone_cnt_before =
 912                         rte_eal_get_configuration()->mem_config->memzone_cnt;
 913
 914         memzone1 = rte_memzone_reserve(TEST_MEMZONE_NAME("testzone1"), 100,
 915                                 SOCKET_ID_ANY, 0);
 916
 917         memzone2 = rte_memzone_reserve(TEST_MEMZONE_NAME("testzone2"), 1000,
 918                                 0, 0);
 919
 920         memzone3 = rte_memzone_reserve(TEST_MEMZONE_NAME("testzone3"), 1000,
 921                                 1, 0);
 922
 923         memzone4 = rte_memzone_reserve(TEST_MEMZONE_NAME("testzone4"), 1024,
 924                                 SOCKET_ID_ANY, 0);
 925
 926         /* memzone3 may be NULL if we don't have NUMA */
 927         if (memzone1 == NULL || memzone2 == NULL || memzone4 == NULL)
 928                 return -1;
 929
 930         /* check how many memzones we are expecting */
 931         memzone_cnt_expected = memzone_cnt_before +
 932                         (memzone1 != NULL) + (memzone2 != NULL) +
 933                         (memzone3 != NULL) + (memzone4 != NULL);
 934
 935         memzone_cnt_after =
 936                         rte_eal_get_configuration()->mem_config->memzone_cnt;
 937
 938         if (memzone_cnt_after != memzone_cnt_expected)
 939                 return -1;
 940
 941
 942         rte_memzone_dump(stdout);
 943
 944         /* check cache-line alignments */
 945         printf("check alignments and lengths\n");
 946
 947         if ((memzone1->iova & RTE_CACHE_LINE_MASK) != 0)
 948                 return -1;
 949         if ((memzone2->iova & RTE_CACHE_LINE_MASK) != 0)
 950                 return -1;
 951         if (memzone3 != NULL && (memzone3->iova & RTE_CACHE_LINE_MASK) != 0)
 952                 return -1;
 953         if ((memzone1->len & RTE_CACHE_LINE_MASK) != 0 || memzone1->len == 0)
 954                 return -1;
 955         if ((memzone2->len & RTE_CACHE_LINE_MASK) != 0 || memzone2->len == 0)
 956                 return -1;
 957         if (memzone3 != NULL && ((memzone3->len & RTE_CACHE_LINE_MASK) != 0 ||
 958                         memzone3->len == 0))
 959                 return -1;
 960         if (memzone4->len != 1024)
 961                 return -1;
 962
 963         /* check that zones don't overlap */
 964         printf("check overlapping\n");
 965
 966         if (is_memory_overlap(memzone1->iova, memzone1->len,
 967                         memzone2->iova, memzone2->len))
 968                 return -1;
 969         if (memzone3 != NULL &&
 970                         is_memory_overlap(memzone1->iova, memzone1->len,
 971                                         memzone3->iova, memzone3->len))
 972                 return -1;
 973         if (memzone3 != NULL &&
 974                         is_memory_overlap(memzone2->iova, memzone2->len,
 975                                         memzone3->iova, memzone3->len))
 976                 return -1;
 977
 978         printf("check socket ID\n");
 979
 980         /* memzone2 must be on socket id 0 and memzone3 on socket 1 */
 981         if (memzone2->socket_id != 0)
 982                 return -1;
 983         if (memzone3 != NULL && memzone3->socket_id != 1)
 984                 return -1;
 985
 986         printf("test zone lookup\n");
 987         mz = rte_memzone_lookup(TEST_MEMZONE_NAME("testzone1"));
 988         if (mz != memzone1)
 989                 return -1;
 990
 991         printf("test duplcate zone name\n");
 992         mz = rte_memzone_reserve(TEST_MEMZONE_NAME("testzone1"), 100,
 993                         SOCKET_ID_ANY, 0);
 994         if (mz != NULL)
 995                 return -1;
 996
 997         if (rte_memzone_free(memzone1)) {
 998                 printf("Fail memzone free - memzone1\n");
 999                 return -1;
1000         }
1001         if (rte_memzone_free(memzone2)) {
1002                 printf("Fail memzone free - memzone2\n");
1003                 return -1;
1004         }
1005         if (memzone3 && rte_memzone_free(memzone3)) {
1006                 printf("Fail memzone free - memzone3\n");
1007                 return -1;
1008         }
1009         if (rte_memzone_free(memzone4)) {
1010                 printf("Fail memzone free - memzone4\n");
1011                 return -1;
1012         }
1013
1014         memzone_cnt_after =
1015                         rte_eal_get_configuration()->mem_config->memzone_cnt;
1016         if (memzone_cnt_after != memzone_cnt_before)
1017                 return -1;
1018
1019         return 0;
1020 }
1021
1022 static int test_memzones_left;
1023 static int memzone_walk_cnt;
1024 static void memzone_walk_clb(const struct rte_memzone *mz,
1025                              void *arg __rte_unused)
1026 {
1027         memzone_walk_cnt++;
1028         if (!strncmp(TEST_MEMZONE_NAME(""), mz->name, RTE_MEMZONE_NAMESIZE))
1029                 test_memzones_left++;
1030 }
1031
1032 static int
1033 test_memzone(void)
1034 {
1035         /* take note of how many memzones were allocated before running */
1036         int memzone_cnt = rte_eal_get_configuration()->mem_config->memzone_cnt;
1037
1038         printf("test basic memzone API\n");
1039         if (test_memzone_basic() < 0)
1040                 return -1;
1041
1042         printf("test free memzone\n");
1043         if (test_memzone_free() < 0)
1044                 return -1;
1045
1046         printf("test reserving memzone with bigger size than the maximum\n");
1047         if (test_memzone_reserving_zone_size_bigger_than_the_maximum() < 0)
1048                 return -1;
1049
1050         printf("test memzone_reserve flags\n");
1051         if (test_memzone_reserve_flags() < 0)
1052                 return -1;
1053
1054         printf("test alignment for memzone_reserve\n");
1055         if (test_memzone_aligned() < 0)
1056                 return -1;
1057
1058         printf("test boundary alignment for memzone_reserve\n");
1059         if (test_memzone_bounded() < 0)
1060                 return -1;
1061
1062         printf("test invalid alignment for memzone_reserve\n");
1063         if (test_memzone_invalid_alignment() < 0)
1064                 return -1;
1065
1066         printf("test reserving the largest size memzone possible\n");
1067         if (test_memzone_reserve_max() < 0)
1068                 return -1;
1069
1070         printf("test reserving the largest size aligned memzone possible\n");
1071         if (test_memzone_reserve_max_aligned() < 0)
1072                 return -1;
1073
1074         printf("check memzone cleanup\n");
1075         memzone_walk_cnt = 0;
1076         test_memzones_left = 0;
1077         rte_memzone_walk(memzone_walk_clb, NULL);
1078         if (memzone_walk_cnt != memzone_cnt || test_memzones_left > 0) {
1079                 printf("there are some memzones left after test\n");
1080                 rte_memzone_dump(stdout);
1081                 return -1;
1082         }
1083
1084         return 0;
1085 }
1086
1087 REGISTER_TEST_COMMAND(memzone_autotest, test_memzone);