1*96c8483aSYuri Pankov /*
2*96c8483aSYuri Pankov * This file and its contents are supplied under the terms of the
3*96c8483aSYuri Pankov * Common Development and Distribution License ("CDDL"), version 1.0.
4*96c8483aSYuri Pankov * You may only use this file in accordance with the terms of version
5*96c8483aSYuri Pankov * 1.0 of the CDDL.
6*96c8483aSYuri Pankov *
7*96c8483aSYuri Pankov * A full copy of the text of the CDDL should have accompanied this
8*96c8483aSYuri Pankov * source. A copy of the CDDL is also available via the Internet at
9*96c8483aSYuri Pankov * http://www.illumos.org/license/CDDL.
10*96c8483aSYuri Pankov */
11*96c8483aSYuri Pankov
12*96c8483aSYuri Pankov /*
13*96c8483aSYuri Pankov * Copyright 2012 Jilin Xpd <jilinxpd@gmail.com>
14*96c8483aSYuri Pankov * Copyright 2018 Nexenta Systems, Inc.
15*96c8483aSYuri Pankov */
16*96c8483aSYuri Pankov
17*96c8483aSYuri Pankov /*
18*96c8483aSYuri Pankov * use mmap to copy data from src file to des file,
19*96c8483aSYuri Pankov * with given flags and modes.
20*96c8483aSYuri Pankov * the src & des file should exist and have the same size.
21*96c8483aSYuri Pankov */
22*96c8483aSYuri Pankov
23*96c8483aSYuri Pankov #include <sys/mman.h>
24*96c8483aSYuri Pankov #include <sys/types.h>
25*96c8483aSYuri Pankov #include <sys/stat.h>
26*96c8483aSYuri Pankov #include <fcntl.h>
27*96c8483aSYuri Pankov #include <stdio.h>
28*96c8483aSYuri Pankov #include <stdlib.h>
29*96c8483aSYuri Pankov #include <unistd.h>
30*96c8483aSYuri Pankov #include <string.h>
31*96c8483aSYuri Pankov #include <errno.h>
32*96c8483aSYuri Pankov
33*96c8483aSYuri Pankov void
usage(void)34*96c8483aSYuri Pankov usage(void)
35*96c8483aSYuri Pankov {
36*96c8483aSYuri Pankov fprintf(stderr,
37*96c8483aSYuri Pankov "usage: "
38*96c8483aSYuri Pankov "prot_mmap -o <r|w> <r|w>"
39*96c8483aSYuri Pankov " -m <r|w|s|p> <r|w|s|p>"
40*96c8483aSYuri Pankov " -f <srcfile> <desfile>\n");
41*96c8483aSYuri Pankov exit(1);
42*96c8483aSYuri Pankov }
43*96c8483aSYuri Pankov
44*96c8483aSYuri Pankov int
main(int argc,char ** argv)45*96c8483aSYuri Pankov main(int argc, char **argv)
46*96c8483aSYuri Pankov {
47*96c8483aSYuri Pankov struct stat sb;
48*96c8483aSYuri Pankov char *src_addr, *des_addr;
49*96c8483aSYuri Pankov char *src_file = NULL, *des_file = NULL;
50*96c8483aSYuri Pankov off_t offset;
51*96c8483aSYuri Pankov size_t filesize;
52*96c8483aSYuri Pankov size_t blksize;
53*96c8483aSYuri Pankov size_t numblks;
54*96c8483aSYuri Pankov size_t i, j;
55*96c8483aSYuri Pankov int src_fid, des_fid;
56*96c8483aSYuri Pankov int mret = 0;
57*96c8483aSYuri Pankov int flags0 = 0, mflags0 = 0, prot0 = 0; /* flags for src file */
58*96c8483aSYuri Pankov int flags1 = 0, mflags1 = 0, prot1 = 0; /* flags for des file */
59*96c8483aSYuri Pankov
60*96c8483aSYuri Pankov /*
61*96c8483aSYuri Pankov * parse arguments
62*96c8483aSYuri Pankov * Not getopt because -o -m -f all have 2 optargs each.
63*96c8483aSYuri Pankov */
64*96c8483aSYuri Pankov if (argc != 10) {
65*96c8483aSYuri Pankov usage();
66*96c8483aSYuri Pankov }
67*96c8483aSYuri Pankov for (i = 1; i < argc; ) {
68*96c8483aSYuri Pankov switch (argv[i][1]) {
69*96c8483aSYuri Pankov case 'o': /* options for open() */
70*96c8483aSYuri Pankov i++;
71*96c8483aSYuri Pankov for (j = 0; argv[i][j]; j++) {
72*96c8483aSYuri Pankov if (argv[i][j] == 'r')
73*96c8483aSYuri Pankov flags0 |= O_RDONLY;
74*96c8483aSYuri Pankov else if (argv[i][j] == 'w')
75*96c8483aSYuri Pankov flags0 |= O_WRONLY;
76*96c8483aSYuri Pankov }
77*96c8483aSYuri Pankov if ((flags0 & (O_RDONLY | O_WRONLY)) ==
78*96c8483aSYuri Pankov (O_RDONLY | O_WRONLY))
79*96c8483aSYuri Pankov flags0 = O_RDWR;
80*96c8483aSYuri Pankov i++;
81*96c8483aSYuri Pankov for (j = 0; argv[i][j]; j++) {
82*96c8483aSYuri Pankov if (argv[i][j] == 'r')
83*96c8483aSYuri Pankov flags1 |= O_RDONLY;
84*96c8483aSYuri Pankov else if (argv[i][j] == 'w')
85*96c8483aSYuri Pankov flags1 |= O_WRONLY;
86*96c8483aSYuri Pankov }
87*96c8483aSYuri Pankov if ((flags1 & (O_RDONLY | O_WRONLY)) ==
88*96c8483aSYuri Pankov (O_RDONLY | O_WRONLY))
89*96c8483aSYuri Pankov flags1 = O_RDWR;
90*96c8483aSYuri Pankov i++;
91*96c8483aSYuri Pankov break;
92*96c8483aSYuri Pankov case 'm': /* options for mmap() */
93*96c8483aSYuri Pankov i++;
94*96c8483aSYuri Pankov for (j = 0; argv[i][j]; j++) {
95*96c8483aSYuri Pankov if (argv[i][j] == 'r')
96*96c8483aSYuri Pankov prot0 |= PROT_READ;
97*96c8483aSYuri Pankov else if (argv[i][j] == 'w')
98*96c8483aSYuri Pankov prot0 |= PROT_WRITE;
99*96c8483aSYuri Pankov else if (argv[i][j] == 's')
100*96c8483aSYuri Pankov mflags0 |= MAP_SHARED;
101*96c8483aSYuri Pankov else if (argv[i][j] == 'p')
102*96c8483aSYuri Pankov mflags0 |= MAP_PRIVATE;
103*96c8483aSYuri Pankov }
104*96c8483aSYuri Pankov i++;
105*96c8483aSYuri Pankov for (j = 0; argv[i][j]; j++) {
106*96c8483aSYuri Pankov if (argv[i][j] == 'r')
107*96c8483aSYuri Pankov prot1 |= PROT_READ;
108*96c8483aSYuri Pankov else if (argv[i][j] == 'w')
109*96c8483aSYuri Pankov prot1 |= PROT_WRITE;
110*96c8483aSYuri Pankov else if (argv[i][j] == 's')
111*96c8483aSYuri Pankov mflags1 |= MAP_SHARED;
112*96c8483aSYuri Pankov else if (argv[i][j] == 'p')
113*96c8483aSYuri Pankov mflags1 |= MAP_PRIVATE;
114*96c8483aSYuri Pankov }
115*96c8483aSYuri Pankov i++;
116*96c8483aSYuri Pankov break;
117*96c8483aSYuri Pankov case 'f': /* src file and des file */
118*96c8483aSYuri Pankov i++;
119*96c8483aSYuri Pankov src_file = argv[i];
120*96c8483aSYuri Pankov i++;
121*96c8483aSYuri Pankov des_file = argv[i];
122*96c8483aSYuri Pankov i++;
123*96c8483aSYuri Pankov }
124*96c8483aSYuri Pankov }
125*96c8483aSYuri Pankov
126*96c8483aSYuri Pankov /* source file */
127*96c8483aSYuri Pankov src_fid = open(src_file, flags0);
128*96c8483aSYuri Pankov if (src_fid == -1) {
129*96c8483aSYuri Pankov fprintf(stderr, "open %s error=%d\n", src_file, errno);
130*96c8483aSYuri Pankov return (1);
131*96c8483aSYuri Pankov }
132*96c8483aSYuri Pankov /* destination file */
133*96c8483aSYuri Pankov des_fid = open(des_file, flags1);
134*96c8483aSYuri Pankov if (des_fid == -1) {
135*96c8483aSYuri Pankov fprintf(stderr, "open %s error=%d\n", des_file, errno);
136*96c8483aSYuri Pankov mret = 1;
137*96c8483aSYuri Pankov goto exit3;
138*96c8483aSYuri Pankov }
139*96c8483aSYuri Pankov
140*96c8483aSYuri Pankov /* get file size */
141*96c8483aSYuri Pankov if (fstat(src_fid, &sb) == -1) {
142*96c8483aSYuri Pankov fprintf(stderr, "fstat %s error=%d\n", src_file, errno);
143*96c8483aSYuri Pankov mret = 1;
144*96c8483aSYuri Pankov goto exit2;
145*96c8483aSYuri Pankov }
146*96c8483aSYuri Pankov filesize = sb.st_size;
147*96c8483aSYuri Pankov if (filesize < 4096) {
148*96c8483aSYuri Pankov fprintf(stderr, "file too small\n");
149*96c8483aSYuri Pankov mret = 1;
150*96c8483aSYuri Pankov goto exit2;
151*96c8483aSYuri Pankov }
152*96c8483aSYuri Pankov
153*96c8483aSYuri Pankov if (fstat(des_fid, &sb) == -1) {
154*96c8483aSYuri Pankov fprintf(stderr, "fstat %s error=%d\n", des_file, errno);
155*96c8483aSYuri Pankov mret = 1;
156*96c8483aSYuri Pankov goto exit2;
157*96c8483aSYuri Pankov }
158*96c8483aSYuri Pankov if (filesize != sb.st_size) {
159*96c8483aSYuri Pankov fprintf(stderr, "file sizes differ\n");
160*96c8483aSYuri Pankov mret = 1;
161*96c8483aSYuri Pankov goto exit2;
162*96c8483aSYuri Pankov }
163*96c8483aSYuri Pankov
164*96c8483aSYuri Pankov /* copy data */
165*96c8483aSYuri Pankov blksize = 64 * 1024 * 1024;
166*96c8483aSYuri Pankov numblks = (filesize + blksize - 1) / blksize;
167*96c8483aSYuri Pankov for (i = 0; i < numblks && mret == 0; i++) {
168*96c8483aSYuri Pankov
169*96c8483aSYuri Pankov offset = (i % numblks) * blksize;
170*96c8483aSYuri Pankov if (offset + blksize > filesize)
171*96c8483aSYuri Pankov blksize = filesize - offset;
172*96c8483aSYuri Pankov
173*96c8483aSYuri Pankov /* map file */
174*96c8483aSYuri Pankov src_addr = mmap(NULL, blksize, prot0, mflags0, src_fid, offset);
175*96c8483aSYuri Pankov if (src_addr == MAP_FAILED) {
176*96c8483aSYuri Pankov fprintf(stderr, "mmap %s error=%d\n", src_file, errno);
177*96c8483aSYuri Pankov mret = 1;
178*96c8483aSYuri Pankov break;
179*96c8483aSYuri Pankov }
180*96c8483aSYuri Pankov des_addr = mmap(NULL, blksize, prot1, mflags1, des_fid, offset);
181*96c8483aSYuri Pankov if (des_addr == MAP_FAILED) {
182*96c8483aSYuri Pankov fprintf(stderr, "mmap %s error=%d\n", des_file, errno);
183*96c8483aSYuri Pankov mret = 1;
184*96c8483aSYuri Pankov goto exit1;
185*96c8483aSYuri Pankov }
186*96c8483aSYuri Pankov
187*96c8483aSYuri Pankov /* cp data from src addr to des addr */
188*96c8483aSYuri Pankov memcpy(des_addr, src_addr, blksize);
189*96c8483aSYuri Pankov /* sync mapped pages to file */
190*96c8483aSYuri Pankov if (msync(des_addr, blksize, MS_SYNC) == -1) {
191*96c8483aSYuri Pankov fprintf(stderr, "msync %s error=%d\n", des_file, errno);
192*96c8483aSYuri Pankov mret = 1;
193*96c8483aSYuri Pankov }
194*96c8483aSYuri Pankov
195*96c8483aSYuri Pankov /* unmap file */
196*96c8483aSYuri Pankov if (munmap(des_addr, blksize) == -1) {
197*96c8483aSYuri Pankov fprintf(stderr, "munmap %s error=%d\n",
198*96c8483aSYuri Pankov des_file, errno);
199*96c8483aSYuri Pankov mret = 1;
200*96c8483aSYuri Pankov }
201*96c8483aSYuri Pankov exit1:
202*96c8483aSYuri Pankov if (munmap(src_addr, blksize) == -1) {
203*96c8483aSYuri Pankov fprintf(stderr, "munmap %s error=%d\n",
204*96c8483aSYuri Pankov src_file, errno);
205*96c8483aSYuri Pankov mret = 1;
206*96c8483aSYuri Pankov }
207*96c8483aSYuri Pankov }
208*96c8483aSYuri Pankov
209*96c8483aSYuri Pankov /* close file */
210*96c8483aSYuri Pankov exit2:
211*96c8483aSYuri Pankov close(des_fid);
212*96c8483aSYuri Pankov exit3:
213*96c8483aSYuri Pankov close(src_fid);
214*96c8483aSYuri Pankov
215*96c8483aSYuri Pankov return (mret);
216*96c8483aSYuri Pankov }
217