--- Makefile | 3 ++- crypt.h | 10 ++++++++++ md5.h | 6 +++--- md5sum.c | 52 +++++++++++++++++----------------------------------- sha1.h | 6 +++--- sha1sum.c | 52 +++++++++++++++++----------------------------------- util/crypt.c | 33 +++++++++++++++++++++++++++++++++ util/md5.c | 9 ++++++--- util/sha1.c | 10 +++++++--- 9 files changed, 98 insertions(+), 83 deletions(-) create mode 100644 crypt.h create mode 100644 util/crypt.c diff --git a/Makefile b/Makefile index f825ec4..909f1ce 100644 --- a/Makefile +++ b/Makefile _AT_@ -3,13 +3,14 @@ include config.mk .POSIX: .SUFFIXES: .c .o -HDR = fs.h text.h util.h arg.h +HDR = crypt.h fs.h text.h md5.h sha1.h util.h arg.h LIB = \ util/afgets.o \ util/agetcwd.o \ util/apathmax.o \ util/concat.o \ util/cp.o \ + util/crypt.o \ util/enmasse.o \ util/eprintf.o \ util/estrtol.o \ diff --git a/crypt.h b/crypt.h new file mode 100644 index 0000000..9eedabf --- /dev/null +++ b/crypt.h _AT_@ -0,0 +1,10 @@ +struct crypt_ops { + void (*init)(void *); + void (*update)(void *, const void *, unsigned long); + void (*sum)(void *, uint8_t *); + void *s; +}; + +int cryptsum(struct crypt_ops *ops, FILE *fp, const char *f, + uint8_t *md); +void mdprint(const uint8_t *md, const char *f, size_t len); diff --git a/md5.h b/md5.h index 0895d3e..0b5005e 100644 --- a/md5.h +++ b/md5.h _AT_@ -9,10 +9,10 @@ struct md5 { enum { MD5_DIGEST_LENGTH = 16 }; /* reset state */ -void md5_init(struct md5 *s); +void md5_init(void *ctx); /* process message */ -void md5_update(struct md5 *s, const void *m, unsigned long len); +void md5_update(void *ctx, const void *m, unsigned long len); /* get message digest */ /* state is ruined after sum, keep a copy if multiple sum is needed */ /* part of the message might be left in s, zero it if secrecy is needed */ -void md5_sum(struct md5 *s, uint8_t md[MD5_DIGEST_LENGTH]); +void md5_sum(void *ctx, uint8_t md[MD5_DIGEST_LENGTH]); diff --git a/md5sum.c b/md5sum.c index 722416f..4fd449f 100644 --- a/md5sum.c +++ b/md5sum.c _AT_@ -1,15 +1,18 @@ /* See LICENSE file for copyright and license details. */ -#include <sys/types.h> -#include <sys/stat.h> -#include <fcntl.h> -#include <unistd.h> #include <stdio.h> #include <stdlib.h> #include <stdint.h> #include "util.h" +#include "crypt.h" #include "md5.h" -static void md5sum(int fd, const char *f); +struct md5 s; +struct crypt_ops md5_ops = { + md5_init, + md5_update, + md5_sum, + &s, +}; static void usage(void) _AT_@ -20,7 +23,8 @@ usage(void) int main(int argc, char *argv[]) { - int fd; + FILE *fp; + uint8_t md[MD5_DIGEST_LENGTH]; ARGBEGIN { default: _AT_@ -28,40 +32,18 @@ main(int argc, char *argv[]) } ARGEND; if (argc == 0) { - md5sum(STDIN_FILENO, "<stdin>"); + cryptsum(&md5_ops, stdin, "<stdin>", md); + mdprint(md, "<stdin>", sizeof(md)); } else { for (; argc > 0; argc--) { - if ((fd = open(*argv, O_RDONLY)) < 0) - eprintf("open %s:", *argv); - md5sum(fd, *argv); - close(fd); + if ((fp = fopen(*argv, "r")) == NULL) + eprintf("fopen %s:", *argv); + cryptsum(&md5_ops, fp, *argv, md); + mdprint(md, *argv, sizeof(md)); + fclose(fp); argv++; } } return 0; } - -static void -md5sum(int fd, const char *f) -{ - unsigned char buf[BUFSIZ]; - unsigned char digest[MD5_DIGEST_LENGTH]; - struct md5 s; - ssize_t n; - int i; - - md5_init(&s); - while ((n = read(fd, buf, sizeof buf)) > 0) - md5_update(&s, buf, n); - if (n < 0) { - eprintf("%s: read error:", f); - return; - } - - md5_sum(&s, digest); - - for (i = 0; i < sizeof(digest); i++) - printf("%02x", digest[i]); - printf(" %s\n", f); -} diff --git a/sha1.h b/sha1.h index e11f49e..8631777 100644 --- a/sha1.h +++ b/sha1.h _AT_@ -9,10 +9,10 @@ struct sha1 { enum { SHA1_DIGEST_LENGTH = 20 }; /* reset state */ -void sha1_init(struct sha1 *s); +void sha1_init(void *ctx); /* process message */ -void sha1_update(struct sha1 *s, const void *m, unsigned long len); +void sha1_update(void *ctx, const void *m, unsigned long len); /* get message digest */ /* state is ruined after sum, keep a copy if multiple sum is needed */ /* part of the message might be left in s, zero it if secrecy is needed */ -void sha1_sum(struct sha1 *s, uint8_t md[SHA1_DIGEST_LENGTH]); +void sha1_sum(void *ctx, uint8_t md[SHA1_DIGEST_LENGTH]); diff --git a/sha1sum.c b/sha1sum.c index e38346a..11b9111 100644 --- a/sha1sum.c +++ b/sha1sum.c _AT_@ -1,15 +1,18 @@ /* See LICENSE file for copyright and license details. */ -#include <sys/types.h> -#include <sys/stat.h> -#include <fcntl.h> -#include <unistd.h> #include <stdio.h> #include <stdlib.h> #include <stdint.h> #include "util.h" +#include "crypt.h" #include "sha1.h" -static void sha1sum(int fd, const char *f); +struct sha1 s; +struct crypt_ops sha1_ops = { + sha1_init, + sha1_update, + sha1_sum, + &s, +}; static void usage(void) _AT_@ -20,7 +23,8 @@ usage(void) int main(int argc, char *argv[]) { - int fd; + FILE *fp; + uint8_t md[SHA1_DIGEST_LENGTH]; ARGBEGIN { default: _AT_@ -28,40 +32,18 @@ main(int argc, char *argv[]) } ARGEND; if (argc == 0) { - sha1sum(STDIN_FILENO, "<stdin>"); + cryptsum(&sha1_ops, stdin, "<stdin>", md); + mdprint(md, "<stdin>", sizeof(md)); } else { for (; argc > 0; argc--) { - if ((fd = open(*argv, O_RDONLY)) < 0) - eprintf("open %s:", *argv); - sha1sum(fd, *argv); - close(fd); + if ((fp = fopen(*argv, "r")) == NULL) + eprintf("fopen %s:", *argv); + cryptsum(&sha1_ops, fp, *argv, md); + mdprint(md, *argv, sizeof(md)); + fclose(fp); argv++; } } return 0; } - -static void -sha1sum(int fd, const char *f) -{ - unsigned char buf[BUFSIZ]; - unsigned char digest[SHA1_DIGEST_LENGTH]; - struct sha1 s; - ssize_t n; - int i; - - sha1_init(&s); - while ((n = read(fd, buf, sizeof buf)) > 0) - sha1_update(&s, buf, n); - if (n < 0) { - eprintf("%s: read error:", f); - return; - } - - sha1_sum(&s, digest); - - for (i = 0; i < sizeof(digest); i++) - printf("%02x", digest[i]); - printf(" %s\n", f); -} diff --git a/util/crypt.c b/util/crypt.c new file mode 100644 index 0000000..2f05275 --- /dev/null +++ b/util/crypt.c _AT_@ -0,0 +1,33 @@ +#include <stdio.h> +#include <stdlib.h> +#include <stdint.h> +#include "../util.h" +#include "../crypt.h" + +int +cryptsum(struct crypt_ops *ops, FILE *fp, const char *f, + uint8_t *md) +{ + unsigned char buf[BUFSIZ]; + size_t n; + + ops->init(ops->s); + while ((n = fread(buf, 1, sizeof(buf), fp)) > 0) + ops->update(ops->s, buf, n); + if (ferror(fp)) { + eprintf("%s: read error:", f); + return 1; + } + ops->sum(ops->s, md); + return 0; +} + +void +mdprint(const uint8_t *md, const char *f, size_t len) +{ + size_t i; + + for (i = 0; i < len; i++) + printf("%02x", md[i]); + printf(" %s\n", f); +} diff --git a/util/md5.c b/util/md5.c index 0d8353d..b68f729 100644 --- a/util/md5.c +++ b/util/md5.c _AT_@ -95,8 +95,9 @@ static void pad(struct md5 *s) processblock(s, s->buf); } -void md5_init(struct md5 *s) +void md5_init(void *ctx) { + struct md5 *s = ctx; s->len = 0; s->h[0] = 0x67452301; s->h[1] = 0xefcdab89; _AT_@ -104,8 +105,9 @@ void md5_init(struct md5 *s) s->h[3] = 0x10325476; } -void md5_sum(struct md5 *s, uint8_t md[16]) +void md5_sum(void *ctx, uint8_t md[16]) { + struct md5 *s = ctx; int i; pad(s); _AT_@ -117,8 +119,9 @@ void md5_sum(struct md5 *s, uint8_t md[16]) } } -void md5_update(struct md5 *s, const void *m, unsigned long len) +void md5_update(void *ctx, const void *m, unsigned long len) { + struct md5 *s = ctx; const uint8_t *p = m; unsigned r = s->len % 64; diff --git a/util/sha1.c b/util/sha1.c index 9910d78..a298f9f 100644 --- a/util/sha1.c +++ b/util/sha1.c _AT_@ -89,8 +89,10 @@ static void pad(struct sha1 *s) processblock(s, s->buf); } -void sha1_init(struct sha1 *s) +void sha1_init(void *ctx) { + struct sha1 *s = ctx; + s->len = 0; s->h[0] = 0x67452301; s->h[1] = 0xEFCDAB89; _AT_@ -99,8 +101,9 @@ void sha1_init(struct sha1 *s) s->h[4] = 0xC3D2E1F0; } -void sha1_sum(struct sha1 *s, uint8_t md[20]) +void sha1_sum(void *ctx, uint8_t md[20]) { + struct sha1 *s = ctx; int i; pad(s); _AT_@ -112,8 +115,9 @@ void sha1_sum(struct sha1 *s, uint8_t md[20]) } } -void sha1_update(struct sha1 *s, const void *m, unsigned long len) +void sha1_update(void *ctx, const void *m, unsigned long len) { + struct sha1 *s = ctx; const uint8_t *p = m; unsigned r = s->len % 64; -- 1.8.3.2 --rwEMma7ioTxnRzrJ--Received on Mon Sep 17 2001 - 00:00:00 CEST
This archive was generated by hypermail 2.3.0 : Sun Jul 14 2013 - 02:24:02 CEST