source: MondoRescue/branches/3.3/mindi-busybox/coreutils/md5_sha1_sum.c@ 3621

Last change on this file since 3621 was 3621, checked in by Bruno Cornec, 7 years ago

New 3?3 banch for incorporation of latest busybox 1.25. Changing minor version to handle potential incompatibilities.
File size: 7.5 KB
Line 
1/* vi: set sw=4 ts=4: */
2/*
3 * Copyright (C) 2003 Glenn L. McGrath
4 * Copyright (C) 2003-2004 Erik Andersen
5 *
6 * Licensed under GPLv2 or later, see file LICENSE in this source tree.
7 */
8
9//usage:#define md5sum_trivial_usage
10//usage: IF_FEATURE_MD5_SHA1_SUM_CHECK("[-c[sw]] ")"[FILE]..."
11//usage:#define md5sum_full_usage "\n\n"
12//usage: "Print" IF_FEATURE_MD5_SHA1_SUM_CHECK(" or check") " MD5 checksums"
13//usage: IF_FEATURE_MD5_SHA1_SUM_CHECK( "\n"
14//usage: "\n -c Check sums against list in FILEs"
15//usage: "\n -s Don't output anything, status code shows success"
16//usage: "\n -w Warn about improperly formatted checksum lines"
17//usage: )
18//usage:
19//usage:#define md5sum_example_usage
20//usage: "$ md5sum < busybox\n"
21//usage: "6fd11e98b98a58f64ff3398d7b324003\n"
22//usage: "$ md5sum busybox\n"
23//usage: "6fd11e98b98a58f64ff3398d7b324003 busybox\n"
24//usage: "$ md5sum -c -\n"
25//usage: "6fd11e98b98a58f64ff3398d7b324003 busybox\n"
26//usage: "busybox: OK\n"
27//usage: "^D\n"
28//usage:
29//usage:#define sha1sum_trivial_usage
30//usage: IF_FEATURE_MD5_SHA1_SUM_CHECK("[-c[sw]] ")"[FILE]..."
31//usage:#define sha1sum_full_usage "\n\n"
32//usage: "Print" IF_FEATURE_MD5_SHA1_SUM_CHECK(" or check") " SHA1 checksums"
33//usage: IF_FEATURE_MD5_SHA1_SUM_CHECK( "\n"
34//usage: "\n -c Check sums against list in FILEs"
35//usage: "\n -s Don't output anything, status code shows success"
36//usage: "\n -w Warn about improperly formatted checksum lines"
37//usage: )
38//usage:
39//usage:#define sha256sum_trivial_usage
40//usage: IF_FEATURE_MD5_SHA1_SUM_CHECK("[-c[sw]] ")"[FILE]..."
41//usage:#define sha256sum_full_usage "\n\n"
42//usage: "Print" IF_FEATURE_MD5_SHA1_SUM_CHECK(" or check") " SHA256 checksums"
43//usage: IF_FEATURE_MD5_SHA1_SUM_CHECK( "\n"
44//usage: "\n -c Check sums against list in FILEs"
45//usage: "\n -s Don't output anything, status code shows success"
46//usage: "\n -w Warn about improperly formatted checksum lines"
47//usage: )
48//usage:
49//usage:#define sha512sum_trivial_usage
50//usage: IF_FEATURE_MD5_SHA1_SUM_CHECK("[-c[sw]] ")"[FILE]..."
51//usage:#define sha512sum_full_usage "\n\n"
52//usage: "Print" IF_FEATURE_MD5_SHA1_SUM_CHECK(" or check") " SHA512 checksums"
53//usage: IF_FEATURE_MD5_SHA1_SUM_CHECK( "\n"
54//usage: "\n -c Check sums against list in FILEs"
55//usage: "\n -s Don't output anything, status code shows success"
56//usage: "\n -w Warn about improperly formatted checksum lines"
57//usage: )
58//usage:
59//usage:#define sha3sum_trivial_usage
60//usage: IF_FEATURE_MD5_SHA1_SUM_CHECK("[-c[sw]] ")"[FILE]..."
61//usage:#define sha3sum_full_usage "\n\n"
62//usage: "Print" IF_FEATURE_MD5_SHA1_SUM_CHECK(" or check") " SHA3-512 checksums"
63//usage: IF_FEATURE_MD5_SHA1_SUM_CHECK( "\n"
64//usage: "\n -c Check sums against list in FILEs"
65//usage: "\n -s Don't output anything, status code shows success"
66//usage: "\n -w Warn about improperly formatted checksum lines"
67//usage: )
68
69#include "libbb.h"
70
71/* This is a NOEXEC applet. Be very careful! */
72
73enum {
74 /* 4th letter of applet_name is... */
75 HASH_MD5 = 's', /* "md5>s<um" */
76 HASH_SHA1 = '1',
77 HASH_SHA256 = '2',
78 HASH_SHA3 = '3',
79 HASH_SHA512 = '5',
80};
81
82#define FLAG_SILENT 1
83#define FLAG_CHECK 2
84#define FLAG_WARN 4
85
86/* This might be useful elsewhere */
87static unsigned char *hash_bin_to_hex(unsigned char *hash_value,
88 unsigned hash_length)
89{
90 /* xzalloc zero-terminates */
91 char *hex_value = xzalloc((hash_length * 2) + 1);
92 bin2hex(hex_value, (char*)hash_value, hash_length);
93 return (unsigned char *)hex_value;
94}
95
96static uint8_t *hash_file(const char *filename)
97{
98 int src_fd, hash_len, count;
99 union _ctx_ {
100 sha3_ctx_t sha3;
101 sha512_ctx_t sha512;
102 sha256_ctx_t sha256;
103 sha1_ctx_t sha1;
104 md5_ctx_t md5;
105 } context;
106 uint8_t *hash_value;
107 void FAST_FUNC (*update)(void*, const void*, size_t);
108 void FAST_FUNC (*final)(void*, void*);
109 char hash_algo;
110
111 src_fd = open_or_warn_stdin(filename);
112 if (src_fd < 0) {
113 return NULL;
114 }
115
116 hash_algo = applet_name[3];
117
118 /* figure specific hash algorithms */
119 if (ENABLE_MD5SUM && hash_algo == HASH_MD5) {
120 md5_begin(&context.md5);
121 update = (void*)md5_hash;
122 final = (void*)md5_end;
123 hash_len = 16;
124 } else if (ENABLE_SHA1SUM && hash_algo == HASH_SHA1) {
125 sha1_begin(&context.sha1);
126 update = (void*)sha1_hash;
127 final = (void*)sha1_end;
128 hash_len = 20;
129 } else if (ENABLE_SHA256SUM && hash_algo == HASH_SHA256) {
130 sha256_begin(&context.sha256);
131 update = (void*)sha256_hash;
132 final = (void*)sha256_end;
133 hash_len = 32;
134 } else if (ENABLE_SHA512SUM && hash_algo == HASH_SHA512) {
135 sha512_begin(&context.sha512);
136 update = (void*)sha512_hash;
137 final = (void*)sha512_end;
138 hash_len = 64;
139 } else if (ENABLE_SHA3SUM && hash_algo == HASH_SHA3) {
140 sha3_begin(&context.sha3);
141 update = (void*)sha3_hash;
142 final = (void*)sha3_end;
143 hash_len = 64;
144 } else {
145 xfunc_die(); /* can't reach this */
146 }
147
148 {
149 RESERVE_CONFIG_UBUFFER(in_buf, 4096);
150 while ((count = safe_read(src_fd, in_buf, 4096)) > 0) {
151 update(&context, in_buf, count);
152 }
153 hash_value = NULL;
154 if (count < 0)
155 bb_perror_msg("can't read '%s'", filename);
156 else /* count == 0 */ {
157 final(&context, in_buf);
158 hash_value = hash_bin_to_hex(in_buf, hash_len);
159 }
160 RELEASE_CONFIG_BUFFER(in_buf);
161 }
162
163 if (src_fd != STDIN_FILENO) {
164 close(src_fd);
165 }
166
167 return hash_value;
168}
169
170int md5_sha1_sum_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
171int md5_sha1_sum_main(int argc UNUSED_PARAM, char **argv)
172{
173 int return_value = EXIT_SUCCESS;
174 unsigned flags;
175
176 if (ENABLE_FEATURE_MD5_SHA1_SUM_CHECK) {
177 /* -b "binary", -t "text" are ignored (shaNNNsum compat) */
178 flags = getopt32(argv, "scwbt");
179 argv += optind;
180 //argc -= optind;
181 } else {
182 argv += 1;
183 //argc -= 1;
184 }
185 if (!*argv)
186 *--argv = (char*)"-";
187
188 if (ENABLE_FEATURE_MD5_SHA1_SUM_CHECK && !(flags & FLAG_CHECK)) {
189 if (flags & FLAG_SILENT) {
190 bb_error_msg_and_die("-%c is meaningful only with -c", 's');
191 }
192 if (flags & FLAG_WARN) {
193 bb_error_msg_and_die("-%c is meaningful only with -c", 'w');
194 }
195 }
196
197 do {
198 if (ENABLE_FEATURE_MD5_SHA1_SUM_CHECK && (flags & FLAG_CHECK)) {
199 FILE *pre_computed_stream;
200 char *line;
201 int count_total = 0;
202 int count_failed = 0;
203
204 pre_computed_stream = xfopen_stdin(*argv);
205
206 while ((line = xmalloc_fgetline(pre_computed_stream)) != NULL) {
207 uint8_t *hash_value;
208 char *filename_ptr;
209
210 count_total++;
211 filename_ptr = strstr(line, " ");
212 /* handle format for binary checksums */
213 if (filename_ptr == NULL) {
214 filename_ptr = strstr(line, " *");
215 }
216 if (filename_ptr == NULL) {
217 if (flags & FLAG_WARN) {
218 bb_error_msg("invalid format");
219 }
220 count_failed++;
221 return_value = EXIT_FAILURE;
222 free(line);
223 continue;
224 }
225 *filename_ptr = '\0';
226 filename_ptr += 2;
227
228 hash_value = hash_file(filename_ptr);
229
230 if (hash_value && (strcmp((char*)hash_value, line) == 0)) {
231 if (!(flags & FLAG_SILENT))
232 printf("%s: OK\n", filename_ptr);
233 } else {
234 if (!(flags & FLAG_SILENT))
235 printf("%s: FAILED\n", filename_ptr);
236 count_failed++;
237 return_value = EXIT_FAILURE;
238 }
239 /* possible free(NULL) */
240 free(hash_value);
241 free(line);
242 }
243 if (count_failed && !(flags & FLAG_SILENT)) {
244 bb_error_msg("WARNING: %d of %d computed checksums did NOT match",
245 count_failed, count_total);
246 }
247 fclose_if_not_stdin(pre_computed_stream);
248 } else {
249 uint8_t *hash_value = hash_file(*argv);
250 if (hash_value == NULL) {
251 return_value = EXIT_FAILURE;
252 } else {
253 printf("%s %s\n", hash_value, *argv);
254 free(hash_value);
255 }
256 }
257 } while (*++argv);
258
259 return return_value;
260}
Note: See TracBrowser for help on using the repository browser.