src/journal/test-mmap-cache.c

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