]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - usr.bin/cmp/cmp.c
cmp: accept SI suffixes for skip1 and skip2
[FreeBSD/FreeBSD.git] / usr.bin / cmp / cmp.c
1 /*
2  * SPDX-License-Identifier: BSD-3-Clause
3  *
4  * Copyright (c) 1987, 1990, 1993, 1994
5  *      The Regents of the University of California.  All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  * 3. Neither the name of the University nor the names of its contributors
16  *    may be used to endorse or promote products derived from this software
17  *    without specific prior written permission.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29  * SUCH DAMAGE.
30  */
31
32 #ifndef lint
33 static const char copyright[] =
34 "@(#) Copyright (c) 1987, 1990, 1993, 1994\n\
35         The Regents of the University of California.  All rights reserved.\n";
36 #endif
37
38 #if 0
39 #ifndef lint
40 static char sccsid[] = "@(#)cmp.c       8.3 (Berkeley) 4/2/94";
41 #endif
42 #endif
43
44 #include <sys/cdefs.h>
45 __FBSDID("$FreeBSD$");
46
47 #include <sys/types.h>
48 #include <sys/stat.h>
49
50 #include <capsicum_helpers.h>
51 #include <err.h>
52 #include <errno.h>
53 #include <fcntl.h>
54 #include <getopt.h>
55 #include <nl_types.h>
56 #include <stdio.h>
57 #include <stdlib.h>
58 #include <string.h>
59 #include <unistd.h>
60
61 #include <libutil.h>
62
63 #include "extern.h"
64
65 bool    lflag, sflag, xflag, zflag;
66
67 static const struct option long_opts[] =
68 {
69         {"verbose",     no_argument,            NULL, 'l'},
70         {"silent",      no_argument,            NULL, 's'},
71         {"quiet",       no_argument,            NULL, 's'},
72         {NULL,          no_argument,            NULL, 0}
73 };
74
75 static void usage(void);
76
77 int
78 main(int argc, char *argv[])
79 {
80         struct stat sb1, sb2;
81         off_t skip1, skip2;
82         int ch, fd1, fd2, oflag;
83         bool special;
84         const char *file1, *file2;
85
86         skip1 = skip2 = 0;
87         oflag = O_RDONLY;
88         while ((ch = getopt_long(argc, argv, "+hlsxz", long_opts, NULL)) != -1)
89                 switch (ch) {
90                 case 'h':               /* Don't follow symlinks */
91                         oflag |= O_NOFOLLOW;
92                         break;
93                 case 'l':               /* print all differences */
94                         lflag = true;
95                         break;
96                 case 's':               /* silent run */
97                         sflag = true;
98                         break;
99                 case 'x':               /* hex output */
100                         lflag = true;
101                         xflag = true;
102                         break;
103                 case 'z':               /* compare size first */
104                         zflag = true;
105                         break;
106                 case '?':
107                 default:
108                         usage();
109                 }
110         argv += optind;
111         argc -= optind;
112
113         if (lflag && sflag)
114                 errx(ERR_EXIT, "specifying -s with -l or -x is not permitted");
115
116         if (argc < 2 || argc > 4)
117                 usage();
118
119         /* Don't limit rights on stdin since it may be one of the inputs. */
120         if (caph_limit_stream(STDOUT_FILENO, CAPH_WRITE | CAPH_IGNORE_EBADF))
121                 err(ERR_EXIT, "unable to limit rights on stdout");
122         if (caph_limit_stream(STDERR_FILENO, CAPH_WRITE | CAPH_IGNORE_EBADF))
123                 err(ERR_EXIT, "unable to limit rights on stderr");
124
125         /* Backward compatibility -- handle "-" meaning stdin. */
126         special = false;
127         if (strcmp(file1 = argv[0], "-") == 0) {
128                 special = true;
129                 fd1 = STDIN_FILENO;
130                 file1 = "stdin";
131         } else if ((fd1 = open(file1, oflag, 0)) < 0 && errno != EMLINK) {
132                 if (!sflag)
133                         err(ERR_EXIT, "%s", file1);
134                 else
135                         exit(ERR_EXIT);
136         }
137         if (strcmp(file2 = argv[1], "-") == 0) {
138                 if (special)
139                         errx(ERR_EXIT,
140                                 "standard input may only be specified once");
141                 special = true;
142                 fd2 = STDIN_FILENO;
143                 file2 = "stdin";
144         } else if ((fd2 = open(file2, oflag, 0)) < 0 && errno != EMLINK) {
145                 if (!sflag)
146                         err(ERR_EXIT, "%s", file2);
147                 else
148                         exit(ERR_EXIT);
149         }
150
151         if (argc > 2 && expand_number(argv[2], &skip1) < 0) {
152                 fprintf(stderr, "Invalid skip1: %s\n", argv[2]);
153                 usage();
154         }
155
156         if (argc == 4 && expand_number(argv[3], &skip2) < 0) {
157                 fprintf(stderr, "Invalid skip2: %s\n", argv[3]);
158                 usage();
159         }
160
161         if (sflag && skip1 == 0 && skip2 == 0)
162                 zflag = true;
163
164         if (fd1 == -1) {
165                 if (fd2 == -1) {
166                         c_link(file1, skip1, file2, skip2);
167                         exit(0);
168                 } else if (!sflag)
169                         errx(ERR_EXIT, "%s: Not a symbolic link", file2);
170                 else
171                         exit(ERR_EXIT);
172         } else if (fd2 == -1) {
173                 if (!sflag)
174                         errx(ERR_EXIT, "%s: Not a symbolic link", file1);
175                 else
176                         exit(ERR_EXIT);
177         }
178
179         /* FD rights are limited in c_special() and c_regular(). */
180         caph_cache_catpages();
181
182         if (!special) {
183                 if (fstat(fd1, &sb1)) {
184                         if (!sflag)
185                                 err(ERR_EXIT, "%s", file1);
186                         else
187                                 exit(ERR_EXIT);
188                 }
189                 if (!S_ISREG(sb1.st_mode))
190                         special = true;
191                 else {
192                         if (fstat(fd2, &sb2)) {
193                                 if (!sflag)
194                                         err(ERR_EXIT, "%s", file2);
195                                 else
196                                         exit(ERR_EXIT);
197                         }
198                         if (!S_ISREG(sb2.st_mode))
199                                 special = true;
200                 }
201         }
202
203         if (special)
204                 c_special(fd1, file1, skip1, fd2, file2, skip2);
205         else {
206                 if (zflag && sb1.st_size != sb2.st_size) {
207                         if (!sflag)
208                                 (void) printf("%s %s differ: size\n",
209                                     file1, file2);
210                         exit(DIFF_EXIT);
211                 }
212                 c_regular(fd1, file1, skip1, sb1.st_size,
213                     fd2, file2, skip2, sb2.st_size);
214         }
215         exit(0);
216 }
217
218 static void
219 usage(void)
220 {
221
222         (void)fprintf(stderr,
223             "usage: cmp [-l | -s | -x] [-hz] file1 file2 [skip1 [skip2]]\n");
224         exit(ERR_EXIT);
225 }