src/libsystemd/sd-journal/test-mmap-cache.c

   1 /* SPDX-License-Identifier: LGPL-2.1-or-later */
   2
   3 #include <fcntl.h>
   4 #include <stdlib.h>
   5 #include <sys/mman.h>
   6 #include <unistd.h>
   7
   8 #include "fd-util.h"
   9 #include "macro.h"
  10 #include "mmap-cache.h"
  11 #include "tmpfile-util.h"
  12
  13 int main(int argc, char *argv[]) {
  14         MMapFileDescriptor *fx;
  15         int x, y, z, r;
  16         char px[] = "/tmp/testmmapXXXXXXX", py[] = "/tmp/testmmapYXXXXXX", pz[] = "/tmp/testmmapZXXXXXX";
  17         MMapCache *m;
  18         void *p, *q;
  19
  20         assert_se(m = mmap_cache_new());
  21
  22         x = mkostemp_safe(px);
  23         assert_se(x >= 0);
  24         unlink(px);
  25
  26         assert_se(fx = mmap_cache_add_fd(m, x, PROT_READ));
  27
  28         y = mkostemp_safe(py);
  29         assert_se(y >= 0);
  30         unlink(py);
  31
  32         z = mkostemp_safe(pz);
  33         assert_se(z >= 0);
  34         unlink(pz);
  35
  36         r = mmap_cache_fd_get(fx, 0, false, 1, 2, NULL, &p);
  37         assert_se(r >= 0);
  38
  39         r = mmap_cache_fd_get(fx, 0, false, 2, 2, NULL, &q);
  40         assert_se(r >= 0);
  41
  42         assert_se((uint8_t*) p + 1 == (uint8_t*) q);
  43
  44         r = mmap_cache_fd_get(fx, 1, false, 3, 2, NULL, &q);
  45         assert_se(r >= 0);
  46
  47         assert_se((uint8_t*) p + 2 == (uint8_t*) q);
  48
  49         r = mmap_cache_fd_get(fx, 0, false, 16ULL*1024ULL*1024ULL, 2, NULL, &p);
  50         assert_se(r >= 0);
  51
  52         r = mmap_cache_fd_get(fx, 1, false, 16ULL*1024ULL*1024ULL+1, 2, NULL, &q);
  53         assert_se(r >= 0);
  54
  55         assert_se((uint8_t*) p + 1 == (uint8_t*) q);
  56
  57         mmap_cache_fd_free(fx);
  58         mmap_cache_unref(m);
  59
  60         safe_close(x);
  61         safe_close(y);
  62         safe_close(z);
  63
  64         return 0;
  65 }