2 * Copyright (c) 1989, 1993
3 * The Regents of the University of California. All rights reserved.
5 * This code is derived from software contributed to Berkeley by
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
11 * 1. Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 * notice, this list of conditions and the following disclaimer in the
15 * documentation and/or other materials provided with the distribution.
16 * 4. Neither the name of the University nor the names of its contributors
17 * may be used to endorse or promote products derived from this software
18 * without specific prior written permission.
20 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
21 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
24 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34 static const char copyright
[] =
35 "@(#) Copyright (c) 1989, 1993\n\
36 The Regents of the University of California. All rights reserved.\n";
41 static char sccsid
[] = "@(#)uniq.c 8.3 (Berkeley) 5/4/95";
43 static const char rcsid
[] =
47 #include <sys/capsicum.h>
65 static int cflag
, dflag
, uflag
, iflag
;
66 static int numchars
, numfields
, repeats
;
68 static FILE *file(const char *, const char *);
69 static wchar_t *convert(const char *);
70 static int inlcmp(const char *, const char *);
71 static void show(FILE *, const char *);
72 static wchar_t *skip(wchar_t *);
73 static void obsolete(char *[]);
74 static void usage(void);
81 * Cache NLS data before entering capability mode.
82 * XXXPJD: There should be strerror_init() and strsignal_init() in libc.
84 (void)catopen("libc", NL_CAT_LOCALE
);
88 main (int argc
, char *argv
[])
90 wchar_t *tprev
, *tthis
;
93 size_t prevbuflen
, thisbuflen
, b1
;
94 char *prevline
, *thisline
, *p
;
98 (void) setlocale(LC_ALL
, "");
101 while ((ch
= getopt(argc
, argv
, "cdif:s:u")) != -1)
113 numfields
= strtol(optarg
, &p
, 10);
114 if (numfields
< 0 || *p
)
115 errx(1, "illegal field skip value: %s", optarg
);
118 numchars
= strtol(optarg
, &p
, 10);
119 if (numchars
< 0 || *p
)
120 errx(1, "illegal character skip value: %s", optarg
);
133 /* If no flags are set, default is -d -u. */
137 } else if (!dflag
&& !uflag
)
146 if (argc
> 0 && strcmp(argv
[0], "-") != 0)
147 ifp
= file(ifn
= argv
[0], "r");
148 cap_rights_init(&rights
, CAP_FSTAT
, CAP_READ
);
149 if (cap_rights_limit(fileno(ifp
), &rights
) < 0 && errno
!= ENOSYS
)
150 err(1, "unable to limit rights for %s", ifn
);
151 cap_rights_init(&rights
, CAP_FSTAT
, CAP_WRITE
);
153 ofp
= file(argv
[1], "w");
155 cap_rights_set(&rights
, CAP_IOCTL
);
156 if (cap_rights_limit(fileno(ofp
), &rights
) < 0 && errno
!= ENOSYS
) {
157 err(1, "unable to limit rights for %s",
158 argc
> 1 ? argv
[1] : "stdout");
160 if (cap_rights_is_set(&rights
, CAP_IOCTL
)) {
163 cmd
= TIOCGETA
; /* required by isatty(3) in printf(3) */
165 if (cap_ioctls_limit(fileno(ofp
), &cmd
, 1) < 0 &&
167 err(1, "unable to limit ioctls for %s",
168 argc
> 1 ? argv
[1] : "stdout");
173 if (cap_enter() < 0 && errno
!= ENOSYS
)
174 err(1, "unable to enter capability mode");
176 prevbuflen
= thisbuflen
= 0;
177 prevline
= thisline
= NULL
;
179 if (getline(&prevline
, &prevbuflen
, ifp
) < 0) {
184 tprev
= convert(prevline
);
186 if (!cflag
&& uflag
&& dflag
)
190 while (getline(&thisline
, &thisbuflen
, ifp
) >= 0) {
193 tthis
= convert(thisline
);
195 if (tthis
== NULL
&& tprev
== NULL
)
196 comp
= inlcmp(thisline
, prevline
);
197 else if (tthis
== NULL
|| tprev
== NULL
)
200 comp
= wcscoll(tthis
, tprev
);
203 /* If different, print; set previous to new value. */
204 if (cflag
|| !dflag
|| !uflag
)
209 prevbuflen
= thisbuflen
;
213 if (!cflag
&& uflag
&& dflag
)
224 if (cflag
|| !dflag
|| !uflag
)
230 convert(const char *str
)
233 wchar_t *buf
, *ret
, *p
;
235 if ((n
= mbstowcs(NULL
, str
, 0)) == (size_t)-1)
237 if (SIZE_MAX
/ sizeof(*buf
) < n
+ 1)
238 errx(1, "conversion buffer length overflow");
239 if ((buf
= malloc((n
+ 1) * sizeof(*buf
))) == NULL
)
241 if (mbstowcs(buf
, str
, n
+ 1) != n
)
242 errx(1, "internal mbstowcs() error");
243 /* The last line may not end with \n. */
244 if (n
> 0 && buf
[n
- 1] == L
'\n')
247 /* If requested get the chosen fields + character offsets. */
248 if (numfields
|| numchars
) {
249 if ((ret
= wcsdup(skip(buf
))) == NULL
)
256 for (p
= ret
; *p
!= L
'\0'; p
++)
264 inlcmp(const char *s1
, const char *s2
)
271 c1
= (unsigned char)*s1
;
272 c2
= (unsigned char)*(s2
- 1);
273 /* The last line may not end with \n. */
283 * Output a line depending on the flags and number of repetitions
287 show(FILE *ofp
, const char *str
)
291 (void)fprintf(ofp
, "%4d %s", repeats
+ 1, str
);
292 if ((dflag
&& repeats
) || (uflag
&& !repeats
))
293 (void)fprintf(ofp
, "%s", str
);
301 for (nfields
= 0; *str
!= L
'\0' && nfields
++ != numfields
; ) {
302 while (iswblank(*str
))
304 while (*str
!= L
'\0' && !iswblank(*str
))
307 for (nchars
= numchars
; nchars
-- && *str
!= L
'\0'; ++str
)
313 file(const char *name
, const char *mode
)
317 if ((fp
= fopen(name
, mode
)) == NULL
)
323 obsolete(char *argv
[])
326 char *ap
, *p
, *start
;
328 while ((ap
= *++argv
)) {
329 /* Return if "--" or not an option of any form. */
333 } else if (ap
[1] == '-')
335 if (!isdigit((unsigned char)ap
[1]))
338 * Digit signifies an old-style option. Malloc space for dash,
339 * new option and argument.
342 if ((start
= p
= malloc(len
+ 3)) == NULL
)
345 *p
++ = ap
[0] == '+' ? 's' : 'f';
346 (void)strcpy(p
, ap
+ 1);
354 (void)fprintf(stderr
,
355 "usage: uniq [-c | -d | -u] [-i] [-f fields] [-s chars] [input [output]]\n");