vmd5sum.c
上传用户:sy_wanhua
上传日期:2013-07-25
资源大小:3048k
文件大小:9k
- /* ====================================================================
- * The Vovida Software License, Version 1.0
- *
- * Copyright (c) 2000 Vovida Networks, Inc. All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions
- * are met:
- *
- * 1. Redistributions of source code must retain the above copyright
- * notice, this list of conditions and the following disclaimer.
- *
- * 2. Redistributions in binary form must reproduce the above copyright
- * notice, this list of conditions and the following disclaimer in
- * the documentation and/or other materials provided with the
- * distribution.
- *
- * 3. The names "VOCAL", "Vovida Open Communication Application Library",
- * and "Vovida Open Communication Application Library (VOCAL)" must
- * not be used to endorse or promote products derived from this
- * software without prior written permission. For written
- * permission, please contact vocal@vovida.org.
- *
- * 4. Products derived from this software may not be called "VOCAL", nor
- * may "VOCAL" appear in their name, without prior written
- * permission of Vovida Networks, Inc.
- *
- * THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESSED OR IMPLIED
- * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
- * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, TITLE AND
- * NON-INFRINGEMENT ARE DISCLAIMED. IN NO EVENT SHALL VOVIDA
- * NETWORKS, INC. OR ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT DAMAGES
- * IN EXCESS OF $1,000, NOR FOR ANY INDIRECT, INCIDENTAL, SPECIAL,
- * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
- * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
- * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
- * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
- * USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
- * DAMAGE.
- *
- * ====================================================================
- *
- * This software consists of voluntary contributions made by Vovida
- * Networks, Inc. and many individuals on behalf of Vovida Networks,
- * Inc. For more information on Vovida Networks, Inc., please see
- * <http://www.vovida.org/>.
- *
- */
- static const char* const vmd5sum_c_Version =
- "$Id: vmd5sum.c,v 1.1 2001/03/26 10:35:31 icahoon Exp $";
- /*
- * md5sum.c - Generate/check MD5 Message Digests
- *
- * Compile and link with md5.c. If you don't have getopt() in your library
- * also include getopt.c. For MSDOS you can also link with the wildcard
- * initialization function (wildargs.obj for Turbo C and setargv.obj for MSC)
- * so that you can use wildcards on the commandline.
- *
- * Written March 1993 by Branko Lankester
- * Modified June 1993 by Colin Plumb for altered md5.c.
- * Modified Feburary 1995 by Ian Jackson for use with Colin Plumb's md5.c.
- * Hacked (modified is too nice a word) January 1997 by Galen Hazelwood
- * to support GNU gettext.
- * This file is in the public domain.
- */
- #include <stdlib.h>
- #include <stdio.h>
- #include <string.h>
- #ifdef __linux__
- #include <getopt.h>
- #endif
- /* #include "config.h" */
- #include "vmd5.h"
- /* Take care of NLS matters. */
- #if HAVE_LOCALE_H
- # include < locale.h >
- #endif
- #if !HAVE_SETLOCALE
- # define setlocale(Category, Locale) /* empty */
- #endif
- #if ENABLE_NLS
- # include < libintl.h >
- # define _(Text) gettext (Text)
- #else
- # undef bindtextdomain
- # define bindtextdomain(Domain, Directory) /* empty */
- # undef textdomain
- # define textdomain(Domain) /* empty */
- # define _(Text) Text
- #endif
- #ifdef UNIX
- #define FOPRTXT "r"
- #define FOPRBIN "r"
- #else
- #ifdef VMS
- #define FOPRTXT "r","ctx=stm"
- #define FOPRBIN "rb","ctx=stm"
- #else
- #define FOPRTXT "r"
- #define FOPRBIN "rb"
- #endif
- #endif
- extern char *optarg;
- extern int optind;
- void usage(void);
- void print_digest(unsigned char *p);
- int mdfile(FILE *fp, unsigned char *digest);
- int do_check(FILE *chkf);
- int hex_digit(int c);
- int get_md5_line(FILE *fp, unsigned char *digest, char *file);
- char *progname;
- int verbose = 0;
- int bin_mode = 0;
- void
- main(int argc, char **argv)
- {
- int opt, rc = 0;
- int check = 0;
- FILE *fp = NULL;
- unsigned char digest[16];
- setlocale(LC_ALL, "");
- bindtextdomain(PACKAGE, LOCALEDIR);
- textdomain(PACKAGE);
- progname = *argv;
- while ((opt = getopt(argc, argv, "cbvp:h")) != EOF)
- {
- switch (opt)
- {
- case 'c':
- check = 1;
- break;
- case 'v':
- verbose = 1;
- break;
- case 'b':
- bin_mode = 1;
- break;
- default:
- usage();
- }
- }
- argc -= optind;
- argv += optind;
- if (check)
- {
- switch (argc)
- {
- case 0:
- fp = stdin;
- break;
- case 1:
- if ((fp = fopen(*argv, FOPRTXT)) == NULL)
- {
- perror(*argv);
- exit(2);
- }
- break;
- default:
- usage();
- }
- exit(do_check(fp));
- }
- if (argc == 0)
- {
- if (mdfile(stdin, digest))
- {
- fprintf(stderr, _("%s: read error on stdinn"), progname);
- exit(2);
- }
- print_digest(digest);
- printf("n");
- exit(0);
- }
- for ( ; argc > 0; --argc, ++argv)
- {
- if (bin_mode)
- fp = fopen(*argv, FOPRBIN);
- else
- fp = fopen(*argv, FOPRTXT);
- if (fp == NULL)
- {
- perror(*argv);
- rc = 2;
- continue;
- }
- if (mdfile(fp, digest))
- {
- fprintf(stderr, _("%s: error reading %sn"), progname, *argv);
- rc = 2;
- }
- else
- {
- print_digest(digest);
- printf(" %c%sn", bin_mode ? '*' : ' ', *argv);
- }
- fclose(fp);
- }
- exit(rc);
- }
- void
- usage()
- {
- fputs(_("usage: md5sum [-bv] [-c [file]] | [file...]n
- Generates or checks MD5 Message Digestsn
- -c check message digests (default is generate)n
- -v verbose, print file names when checkingn
- -b read files in binary moden
- The input for -c should be the list of message digests and file namesn
- that is printed on stdout by this program when it generates digests.n"), stderr);
- exit(2);
- }
- int
- mdfile(FILE *fp, unsigned char *digest)
- {
- unsigned char buf[1024];
- struct MD5Context ctx;
- int n;
- MD5Init(&ctx);
- while ((n = fread(buf, 1, sizeof(buf), fp)) > 0)
- MD5Update(&ctx, buf, n);
- MD5Final(digest, &ctx);
- if (ferror(fp))
- return -1;
- return 0;
- }
- void
- print_digest(unsigned char *p)
- {
- int i;
- for (i = 0; i < 16; ++i)
- printf("%02x", *p++);
- }
- int
- hex_digit(int c)
- {
- if (c >= '0' && c <= '9')
- return c - '0';
- if (c >= 'a' && c <= 'f')
- return c - 'a' + 10;
- return -1;
- }
- int
- get_md5_line(FILE *fp, unsigned char *digest, char *file)
- {
- char buf[1024];
- int i, d1, d2, rc;
- char *p = buf;
- if (fgets(buf, sizeof(buf), fp) == NULL)
- return -1;
- for (i = 0; i < 16; ++i)
- {
- if ((d1 = hex_digit(*p++)) == -1)
- return 0;
- if ((d2 = hex_digit(*p++)) == -1)
- return 0;
- *digest++ = d1 * 16 + d2;
- }
- if (*p++ != ' ')
- return 0;
- /*
- * next char is an attribute char, space means text file
- * if it's a '*' the file should be checked in binary mode.
- */
- if (*p == ' ')
- rc = 1;
- else if (*p == '*')
- rc = 2;
- else
- {
- fprintf(stderr, _("%s: unrecognized line: %s"), progname, buf);
- return 0;
- }
- ++p;
- i = strlen(p);
- if (i < 2 || i > 255)
- return 0;
- p[i - 1] = ' ';
- strcpy(file, p);
- return rc;
- }
- int
- do_check(FILE *chkf)
- {
- int rc, ex = 0, failed = 0, checked = 0;
- unsigned char chk_digest[16], file_digest[16];
- char filename[256];
- FILE *fp;
- int flen = 14;
- while ((rc = get_md5_line(chkf, chk_digest, filename)) >= 0)
- {
- if (rc == 0) /* not an md5 line */
- continue;
- if (verbose)
- {
- if (strlen(filename) > flen)
- flen = strlen(filename);
- fprintf(stderr, "%-*s ", flen, filename);
- }
- if (bin_mode || rc == 2)
- fp = fopen(filename, FOPRBIN);
- else
- fp = fopen(filename, FOPRTXT);
- if (fp == NULL)
- {
- fprintf(stderr, _("%s: can't open %sn"), progname, filename);
- ex = 2;
- continue;
- }
- if (mdfile(fp, file_digest))
- {
- fprintf(stderr, _("%s: error reading %sn"), progname, filename);
- ex = 2;
- fclose(fp);
- continue;
- }
- fclose(fp);
- if (memcmp(chk_digest, file_digest, 16) != 0)
- {
- if (verbose)
- fprintf(stderr, _("FAILEDn"));
- else
- fprintf(stderr, _("%s: MD5 check failed for '%s'n"), progname, filename);
- ++failed;
- }
- else if (verbose)
- fprintf(stderr, _("OKn"));
- ++checked;
- }
- if (verbose && failed)
- fprintf(stderr, _("%s: %d of %d file(s) failed MD5 checkn"), progname, failed, checked);
- if (!checked)
- {
- fprintf(stderr, _("%s: no files checkedn"), progname);
- return 3;
- }
- if (!ex && failed)
- ex = 1;
- return ex;
- }