1 /* sum -- checksum and count the blocks in a file
2 Copyright (C) 1986-2019 Free Software Foundation, Inc.
4 This program is free software: you can redistribute it and/or modify
5 it under the terms of the GNU General Public License as published by
6 the Free Software Foundation, either version 3 of the License, or
7 (at your option) any later version.
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
14 You should have received a copy of the GNU General Public License
15 along with this program. If not, see <https://www.gnu.org/licenses/>. */
17 /* Like BSD sum or SysV sum -r, except like SysV sum if -s option is given. */
19 /* Written by Kayvan Aghaiepour and David MacKenzie. */
24 #include <sys/types.h>
31 #include "safe-read.h"
32 #include "xbinary-io.h"
34 /* The official name of this program (e.g., no 'g' prefix). */
35 #define PROGRAM_NAME "sum"
38 proper_name ("Kayvan Aghaiepour"), \
39 proper_name ("David MacKenzie")
41 /* True if any of the files read were the standard input. */
42 static bool have_read_stdin
;
44 static struct option
const longopts
[] =
46 {"sysv", no_argument
, NULL
, 's'},
47 {GETOPT_HELP_OPTION_DECL
},
48 {GETOPT_VERSION_OPTION_DECL
},
55 if (status
!= EXIT_SUCCESS
)
60 Usage: %s [OPTION]... [FILE]...\n\
64 Print checksum and block counts for each FILE.\n\
71 -r use BSD sum algorithm, use 1K blocks\n\
72 -s, --sysv use System V sum algorithm, use 512 bytes blocks\n\
74 fputs (HELP_OPTION_DESCRIPTION
, stdout
);
75 fputs (VERSION_OPTION_DESCRIPTION
, stdout
);
76 emit_ancillary_info (PROGRAM_NAME
);
81 /* Calculate and print the rotated checksum and the size in 1K blocks
82 of file FILE, or of the standard input if FILE is "-".
83 If PRINT_NAME is >1, print FILE next to the checksum and size.
84 The checksum varies depending on sizeof (int).
85 Return true if successful. */
88 bsd_sum_file (const char *file
, int print_name
)
91 int checksum
= 0; /* The checksum mod 2^16. */
92 uintmax_t total_bytes
= 0; /* The number of bytes. */
93 int ch
; /* Each character read. */
94 char hbuf
[LONGEST_HUMAN_READABLE
+ 1];
95 bool is_stdin
= STREQ (file
, "-");
100 have_read_stdin
= true;
101 xset_binary_mode (STDIN_FILENO
, O_BINARY
);
105 fp
= fopen (file
, (O_BINARY
? "rb" : "r"));
108 error (0, errno
, "%s", quotef (file
));
113 fadvise (fp
, FADVISE_SEQUENTIAL
);
115 while ((ch
= getc (fp
)) != EOF
)
118 checksum
= (checksum
>> 1) + ((checksum
& 1) << 15);
120 checksum
&= 0xffff; /* Keep it within bounds. */
125 error (0, errno
, "%s", quotef (file
));
131 if (!is_stdin
&& fclose (fp
) != 0)
133 error (0, errno
, "%s", quotef (file
));
137 printf ("%05d %5s", checksum
,
138 human_readable (total_bytes
, hbuf
, human_ceiling
, 1, 1024));
140 printf (" %s", file
);
146 /* Calculate and print the checksum and the size in 512-byte blocks
147 of file FILE, or of the standard input if FILE is "-".
148 If PRINT_NAME is >0, print FILE next to the checksum and size.
149 Return true if successful. */
152 sysv_sum_file (const char *file
, int print_name
)
155 unsigned char buf
[8192];
156 uintmax_t total_bytes
= 0;
157 char hbuf
[LONGEST_HUMAN_READABLE
+ 1];
161 /* The sum of all the input bytes, modulo (UINT_MAX + 1). */
164 bool is_stdin
= STREQ (file
, "-");
169 have_read_stdin
= true;
170 xset_binary_mode (STDIN_FILENO
, O_BINARY
);
174 fd
= open (file
, O_RDONLY
| O_BINARY
);
177 error (0, errno
, "%s", quotef (file
));
184 size_t bytes_read
= safe_read (fd
, buf
, sizeof buf
);
189 if (bytes_read
== SAFE_READ_ERROR
)
191 error (0, errno
, "%s", quotef (file
));
197 for (size_t i
= 0; i
< bytes_read
; i
++)
199 total_bytes
+= bytes_read
;
202 if (!is_stdin
&& close (fd
) != 0)
204 error (0, errno
, "%s", quotef (file
));
208 r
= (s
& 0xffff) + ((s
& 0xffffffff) >> 16);
209 checksum
= (r
& 0xffff) + (r
>> 16);
211 printf ("%d %s", checksum
,
212 human_readable (total_bytes
, hbuf
, human_ceiling
, 1, 512));
214 printf (" %s", file
);
221 main (int argc
, char **argv
)
226 bool (*sum_func
) (const char *, int) = bsd_sum_file
;
228 initialize_main (&argc
, &argv
);
229 set_program_name (argv
[0]);
230 setlocale (LC_ALL
, "");
231 bindtextdomain (PACKAGE
, LOCALEDIR
);
232 textdomain (PACKAGE
);
234 atexit (close_stdout
);
236 /* Line buffer stdout to ensure lines are written atomically and immediately
237 so that processes running in parallel do not intersperse their output. */
238 setvbuf (stdout
, NULL
, _IOLBF
, 0);
240 have_read_stdin
= false;
242 while ((optc
= getopt_long (argc
, argv
, "rs", longopts
, NULL
)) != -1)
246 case 'r': /* For SysV compatibility. */
247 sum_func
= bsd_sum_file
;
251 sum_func
= sysv_sum_file
;
254 case_GETOPT_HELP_CHAR
;
256 case_GETOPT_VERSION_CHAR (PROGRAM_NAME
, AUTHORS
);
259 usage (EXIT_FAILURE
);
263 files_given
= argc
- optind
;
264 if (files_given
<= 0)
265 ok
= sum_func ("-", files_given
);
267 for (ok
= true; optind
< argc
; optind
++)
268 ok
&= sum_func (argv
[optind
], files_given
);
270 if (have_read_stdin
&& fclose (stdin
) == EOF
)
271 die (EXIT_FAILURE
, errno
, "%s", quotef ("-"));
272 return ok
? EXIT_SUCCESS
: EXIT_FAILURE
;