1 // SPDX-License-Identifier: GPL-2.0-only 2 /* 3 * Copyright 2016, Anton Blanchard, Michael Ellerman, IBM Corp. 4 */ 5 6 #include <stdio.h> 7 #include <stdlib.h> 8 #include <sys/mman.h> 9 #include <time.h> 10 #include <getopt.h> 11 12 #include "utils.h" 13 14 #define ITERATIONS 5000000 15 16 #define MEMSIZE (1UL << 27) 17 #define PAGE_SIZE (1UL << 16) 18 #define CHUNK_COUNT (MEMSIZE/PAGE_SIZE) 19 20 static int pg_fault; 21 static int iterations = ITERATIONS; 22 23 static struct option options[] = { 24 { "pgfault", no_argument, &pg_fault, 1 }, 25 { "iterations", required_argument, 0, 'i' }, 26 { 0, }, 27 }; 28 usage(void)29 static void usage(void) 30 { 31 printf("mmap_bench <--pgfault> <--iterations count>\n"); 32 } 33 test_mmap(void)34 int test_mmap(void) 35 { 36 struct timespec ts_start, ts_end; 37 unsigned long i = iterations; 38 39 clock_gettime(CLOCK_MONOTONIC, &ts_start); 40 41 while (i--) { 42 char *c = mmap(NULL, MEMSIZE, PROT_READ|PROT_WRITE, 43 MAP_PRIVATE|MAP_ANONYMOUS, -1, 0); 44 FAIL_IF(c == MAP_FAILED); 45 if (pg_fault) { 46 int count; 47 for (count = 0; count < CHUNK_COUNT; count++) 48 c[count << 16] = 'c'; 49 } 50 munmap(c, MEMSIZE); 51 } 52 53 clock_gettime(CLOCK_MONOTONIC, &ts_end); 54 55 printf("time = %.6f\n", ts_end.tv_sec - ts_start.tv_sec + (ts_end.tv_nsec - ts_start.tv_nsec) / 1e9); 56 57 return 0; 58 } 59 main(int argc,char * argv[])60 int main(int argc, char *argv[]) 61 { 62 signed char c; 63 while (1) { 64 int option_index = 0; 65 66 c = getopt_long(argc, argv, "", options, &option_index); 67 68 if (c == -1) 69 break; 70 71 switch (c) { 72 case 0: 73 if (options[option_index].flag != 0) 74 break; 75 76 usage(); 77 exit(1); 78 break; 79 case 'i': 80 iterations = atoi(optarg); 81 break; 82 default: 83 usage(); 84 exit(1); 85 } 86 } 87 88 test_harness_set_timeout(300); 89 return test_harness(test_mmap, "mmap_bench"); 90 } 91