xref: /freebsd/sys/contrib/openzfs/tests/zfs-tests/cmd/mmap_sync.c (revision 924226fba12cc9a228c73b956e1b7fa24c60b055)
1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License (the "License").
6  * You may not use this file except in compliance with the License.
7  *
8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9  * or http://opensource.org/licenses/CDDL-1.0.
10  * See the License for the specific language governing permissions
11  * and limitations under the License.
12  *
13  * When distributing Covered Code, include this CDDL HEADER in each
14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15  * If applicable, add the following below this CDDL HEADER, with the
16  * fields enclosed by brackets "[]" replaced with your own identifying
17  * information: Portions Copyright [yyyy] [name of copyright owner]
18  *
19  * CDDL HEADER END
20  */
21 
22 #include <stdlib.h>
23 #include <stdio.h>
24 #include <string.h>
25 #include <sys/mman.h>
26 #include <sys/stat.h>
27 #include <sys/time.h>
28 #include <fcntl.h>
29 #include <unistd.h>
30 #include <time.h>
31 
32 static void
33 cleanup(char *file)
34 {
35 	remove(file);
36 }
37 
38 int
39 main(int argc, char *argv[])
40 {
41 	char *testdir = getenv("TESTDIR");
42 	if (!testdir) {
43 		fprintf(stderr, "environment variable TESTDIR not set\n");
44 		return (1);
45 	}
46 
47 	struct stat st;
48 	umask(0);
49 	if (stat(testdir, &st) != 0 &&
50 	    mkdir(testdir, 0777) != 0) {
51 		perror("mkdir");
52 		return (1);
53 	}
54 
55 	if (argc > 3) {
56 		fprintf(stderr, "usage: %s "
57 		    "[run time in mins] "
58 		    "[max msync time in ms]\n", argv[0]);
59 		return (1);
60 	}
61 
62 	int run_time_mins = 5;
63 	if (argc >= 2) {
64 		run_time_mins = atoi(argv[1]);
65 	}
66 
67 	int max_msync_time_ms = 1000;
68 	if (argc >= 3) {
69 		max_msync_time_ms = atoi(argv[2]);
70 	}
71 
72 	char filepath[512];
73 	filepath[0] = '\0';
74 	char *file = &filepath[0];
75 
76 	strcat(file, testdir);
77 	strcat(file, "/msync_file");
78 
79 	const int LEN = 8;
80 	cleanup(file);
81 
82 	int fd = open(file, O_CREAT | O_RDWR, S_IRUSR | S_IWUSR |
83 	    S_IRGRP | S_IROTH);
84 
85 	if (fd == -1) {
86 		(void) fprintf(stderr, "%s: %s: ", argv[0], file);
87 		perror("open");
88 		return (1);
89 	}
90 
91 	if (ftruncate(fd, LEN) != 0) {
92 		perror("ftruncate");
93 		cleanup(file);
94 		return (1);
95 	}
96 
97 	void *ptr = mmap(NULL, LEN, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
98 
99 	if (ptr == MAP_FAILED) {
100 		perror("mmap");
101 		cleanup(file);
102 		return (1);
103 	}
104 
105 	struct timeval tstart;
106 	gettimeofday(&tstart, NULL);
107 
108 	long long x = 0LL;
109 
110 	for (;;) {
111 		*((long long *)ptr) = x;
112 		x++;
113 
114 		struct timeval t1, t2;
115 		gettimeofday(&t1, NULL);
116 		if (msync(ptr, LEN, MS_SYNC|MS_INVALIDATE) != 0) {
117 			perror("msync");
118 			cleanup(file);
119 			return (1);
120 		}
121 
122 		gettimeofday(&t2, NULL);
123 
124 		double elapsed = (t2.tv_sec - t1.tv_sec) * 1000.0;
125 		elapsed += ((t2.tv_usec - t1.tv_usec) / 1000.0);
126 		if (elapsed > max_msync_time_ms) {
127 			fprintf(stderr, "slow msync: %f ms\n", elapsed);
128 			munmap(ptr, LEN);
129 			cleanup(file);
130 			return (1);
131 		}
132 
133 		double elapsed_start = (t2.tv_sec - tstart.tv_sec) * 1000.0;
134 		elapsed_start += ((t2.tv_usec - tstart.tv_usec) / 1000.0);
135 		if (elapsed_start > run_time_mins * 60 * 1000) {
136 			break;
137 		}
138 	}
139 
140 	if (munmap(ptr, LEN) != 0) {
141 		perror("munmap");
142 		cleanup(file);
143 		return (1);
144 	}
145 
146 	if (close(fd) != 0) {
147 		perror("close");
148 	}
149 
150 	cleanup(file);
151 	return (0);
152 }
153