diff options
| -rw-r--r-- | c/sha3sum.c | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/c/sha3sum.c b/c/sha3sum.c index 4467a72..edfaddf 100644 --- a/c/sha3sum.c +++ b/c/sha3sum.c @@ -17,7 +17,7 @@ * along with this program. If not, see <http://www.gnu.org/licenses/>. */ #include <stdio.h> -#include <stdlib.h> +#include <sys/stat.h> #include "sha3.h" @@ -566,6 +566,7 @@ int main(int argc, char** argv) char* filename; char* fn; long f, fail, _; + struct stat attr; char* out = binary ? null : (char*)malloc(bn * 2); @@ -592,7 +593,9 @@ int main(int argc, char** argv) if ((filename != null) || (stdin == null)) { initialise(r, c, o); - blksize = 4096; /** XXX os.stat(os.path.realpath(fn)).st_size; **/ + blksize = stat(*(argv + f), &attr) ? 0 : attr.st_blksize; + if (blksize <= 0) + blksize = 4096; chunk = (char*)malloc(blksize); for (;;) { |