2 * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
4 * Copyright (c) 1997 Robert Nordier
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
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
14 * the documentation and/or other materials provided with the
17 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS
18 * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
19 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY
21 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
23 * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
24 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
25 * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
26 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
27 * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29 * $FreeBSD: head/usr.sbin/ckdist/ckdist.c 326276 2017-11-27 15:37:16Z pfg $
32 #include <sys/types.h>
46 extern int crc(int fd
, uint32_t *cval
, off_t
*clen
);
48 #define DISTMD5 1 /* MD5 format */
49 #define DISTINF 2 /* .inf format */
50 #define DISTTYPES 2 /* types supported */
52 #define E_UNKNOWN 1 /* Unknown format */
53 #define E_BADMD5 2 /* Invalid MD5 format */
54 #define E_BADINF 3 /* Invalid .inf format */
55 #define E_NAME 4 /* Can't derive component name */
56 #define E_LENGTH 5 /* Length mismatch */
57 #define E_CHKSUM 6 /* Checksum mismatch */
58 #define E_ERRNO 7 /* sys_errlist[errno] */
60 #define isfatal(err) ((err) && (err) <= E_NAME)
62 #define NAMESIZE 256 /* filename buffer size */
63 #define MDSUMLEN 32 /* length of MD5 message digest */
65 #define isstdin(path) ((path)[0] == '-' && !(path)[1])
67 static const char *opt_dir
; /* where to look for components */
68 static const char *opt_name
; /* name for accessing components */
69 static int opt_all
; /* report on all components */
70 static int opt_ignore
; /* ignore missing components */
71 static int opt_recurse
; /* search directories recursively */
72 static int opt_silent
; /* silent about inaccessible files */
73 static int opt_type
; /* dist type: md5 or inf */
74 static int opt_exist
; /* just verify existence */
76 static int ckdist(const char *path
, int type
);
77 static int chkmd5(FILE * fp
, const char *path
);
78 static int chkinf(FILE * fp
, const char *path
);
79 static int report(const char *path
, const char *name
, int error
);
80 static const char *distname(const char *path
, const char *name
,
82 static const char *stripath(const char *path
);
83 static int distfile(const char *path
);
84 static int disttype(const char *name
);
85 static int fail(const char *path
, const char *msg
);
86 static void usage(void);
89 main(int argc
, char *argv
[])
97 while ((c
= getopt(argc
, argv
, "ad:in:rst:x")) != -1)
118 if ((opt_type
= disttype(optarg
)) == 0) {
119 warnx("illegal argument to -t option");
134 if (stat(opt_dir
, &sb
))
135 err(2, "%s", opt_dir
);
136 if (!S_ISDIR(sb
.st_mode
))
137 errx(2, "%s: not a directory", opt_dir
);
142 rval
|= ckdist(*argv
, opt_type
);
143 else if (stat(*argv
, &sb
))
144 rval
|= fail(*argv
, NULL
);
145 else if (S_ISREG(sb
.st_mode
))
146 rval
|= ckdist(*argv
, opt_type
);
149 if ((ftsp
= fts_open(arg
, FTS_LOGICAL
, NULL
)) == NULL
)
151 while ((f
= fts_read(ftsp
)) != NULL
)
152 switch (f
->fts_info
) {
154 rval
= fail(f
->fts_path
, "Directory causes a cycle");
159 rval
= fail(f
->fts_path
, sys_errlist
[f
->fts_errno
]);
162 if (!opt_recurse
&& f
->fts_level
> FTS_ROOTLEVEL
&&
163 fts_set(ftsp
, f
, FTS_SKIP
))
167 if ((type
= distfile(f
->fts_name
)) != 0 &&
168 (!opt_type
|| type
== opt_type
))
169 rval
|= ckdist(f
->fts_path
, type
);
183 ckdist(const char *path
, int type
)
191 } else if ((fp
= fopen(path
, "r")) == NULL
)
192 return fail(path
, NULL
);
195 type
= distfile(path
);
197 if ((c
= fgetc(fp
)) != EOF
) {
198 type
= c
== 'M' ? DISTMD5
: c
== 'P' ? DISTINF
: 0;
204 rval
= chkmd5(fp
, path
);
207 rval
= chkinf(fp
, path
);
210 rval
= report(path
, NULL
, E_UNKNOWN
);
214 if (fp
!= stdin
&& fclose(fp
))
220 chkmd5(FILE * fp
, const char *path
)
222 char buf
[298]; /* "MD5 (NAMESIZE = MDSUMLEN" */
223 char name
[NAMESIZE
+ 1];
224 char sum
[MDSUMLEN
+ 1], chk
[MDSUMLEN
+ 1];
227 int rval
, error
, c
, fd
;
231 while (fgets(buf
, sizeof(buf
), fp
)) {
234 if (((c
= sscanf(buf
, "MD5 (%256s = %32s%c", name
, sum
,
235 &ch
)) != 3 && (!feof(fp
) || c
!= 2)) ||
236 (c
== 3 && ch
!= '\n') ||
237 (s
= strrchr(name
, ')')) == NULL
||
238 strlen(sum
) != MDSUMLEN
)
242 if ((dname
= distname(path
, name
, NULL
)) == NULL
)
244 else if (opt_exist
) {
245 if ((fd
= open(dname
, O_RDONLY
)) == -1)
249 } else if (!MD5File(dname
, chk
))
251 else if (strcmp(chk
, sum
))
254 if (opt_ignore
&& error
== E_ERRNO
&& errno
== ENOENT
)
256 if (error
|| opt_all
)
257 rval
|= report(path
, dname
, error
);
265 chkinf(FILE * fp
, const char *path
)
267 char buf
[30]; /* "cksum.2 = 10 6" */
275 int rval
, error
, c
, pieces
, cnt
, fd
;
279 for (cnt
= -1; fgets(buf
, sizeof(buf
), fp
); cnt
++) {
284 if ((c
= sscanf(buf
, "Pieces = %d%c", &pieces
, &ch
)) != 2 ||
285 ch
!= '\n' || pieces
< 1)
287 } else if (((c
= sscanf(buf
, "cksum.%2s = %lu %jd%c", ext
, &sum
,
288 &sumlen
, &ch
)) != 4 &&
289 (!feof(fp
) || c
!= 3)) || (c
== 4 && ch
!= '\n') ||
290 ext
[0] != 'a' + cnt
/ 26 || ext
[1] != 'a' + cnt
% 26)
292 else if ((dname
= distname(fp
== stdin
? NULL
: path
, NULL
,
295 else if ((fd
= open(dname
, O_RDONLY
)) == -1)
297 else if (fstat(fd
, &sb
))
299 else if (sb
.st_size
!= (off_t
)sumlen
)
301 else if (!opt_exist
) {
302 if (crc(fd
, &chk
, &len
))
307 if (fd
!= -1 && close(fd
))
309 if (opt_ignore
&& error
== E_ERRNO
&& errno
== ENOENT
)
311 if (error
|| (opt_all
&& cnt
>= 0))
312 rval
|= report(path
, dname
, error
);
320 report(const char *path
, const char *name
, int error
)
323 name
= stripath(name
);
326 printf("%s: Unknown format\n", path
);
329 printf("%s: Invalid MD5 format\n", path
);
332 printf("%s: Invalid .inf format\n", path
);
335 printf("%s: Can't derive component name\n", path
);
338 printf("%s: %s: Size mismatch\n", path
, name
);
341 printf("%s: %s: Checksum mismatch\n", path
, name
);
344 printf("%s: %s: %s\n", path
, name
, sys_errlist
[errno
]);
347 printf("%s: %s: OK\n", path
, name
);
353 distname(const char *path
, const char *name
, const char *ext
)
355 static char buf
[NAMESIZE
];
364 name
= stripath(path
);
367 if (ext
&& nlen
> 4 && name
[nlen
- 4] == '.' &&
368 disttype(name
+ nlen
- 3) == DISTINF
)
374 plen
= path
&& (s
= strrchr(path
, '/')) != NULL
?
375 (size_t)(s
- path
) : 0;
376 if (plen
+ (plen
> 0) + nlen
+ (ext
? 3 : 0) >= sizeof(buf
))
380 memcpy(s
, path
, plen
);
384 memcpy(s
, name
, nlen
);
396 stripath(const char *path
)
400 return ((s
= strrchr(path
, '/')) != NULL
&& s
[1] ?
405 distfile(const char *path
)
410 if ((type
= disttype(path
)) == DISTMD5
||
411 ((s
= strrchr(path
, '.')) != NULL
&& s
> path
&&
412 (type
= disttype(s
+ 1)) != 0))
418 disttype(const char *name
)
420 static const char dname
[DISTTYPES
][4] = {"md5", "inf"};
423 for (i
= 0; i
< DISTTYPES
; i
++)
424 if (!strcmp(dname
[i
], name
))
430 fail(const char *path
, const char *msg
)
434 warnx("%s: %s", path
, msg
? msg
: sys_errlist
[errno
]);
442 "usage: ckdist [-airsx] [-d dir] [-n name] [-t type] file ...\n");