tests/lib/libc/hash/h_hash.c

   1 /*      $NetBSD: h_hash.c,v 1.1 2011/01/02 22:03:25 pgoyette Exp $      */
   2
   3 /*-
   4  * Copyright (c) 2000 The NetBSD Foundation, Inc.
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
  17  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
  18  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  19  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
  20  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  21  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  22  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  23  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  24  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  25  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  26  * POSSIBILITY OF SUCH DAMAGE.
  27  */
  28
  29 /*
  30  * Combined MD5/SHA1 time and regression test.
  31  */
  32
  33 #include <stdio.h>
  34 #include <stdlib.h>
  35 #include <unistd.h>
  36 #include <string.h>
  37 #include <md5.h>
  38 #include <sha1.h>
  39
  40
  41 int mflag, rflag, sflag, tflag;
  42
  43 static void
  44 usage(void)
  45 {
  46         (void)fprintf(stderr,
  47             "Usage:\t%s -r[ms] < test-file\n"
  48             "\t%s -t[ms]\n",
  49             getprogname(), getprogname());
  50         exit(1);
  51         /* NOTREACHED */
  52 }
  53
  54 static void
  55 hexdump (unsigned char *buf, int len)
  56 {
  57         int i;
  58         for (i=0; i<len; i++) {
  59                 printf("%02x", buf[i]);
  60         }
  61         printf("\n");
  62 }
  63
  64
  65 static void
  66 timetest(void)
  67 {
  68         printf("sorry, not yet\n");
  69 }
  70
  71 #define CHOMP(buf, len, last)                           \
  72         if ((len > 0) &&                                \
  73             (buf[len-1] == '\n')) {                     \
  74                 buf[len-1] = '\0';                      \
  75                 len--;                                  \
  76                 last = 1;                               \
  77         }
  78
  79 static void
  80 regress(void)
  81 {
  82         unsigned char buf[1024];
  83         unsigned char out[20];
  84         int len, outlen, last;
  85
  86         while (fgets((char *)buf, sizeof(buf), stdin) != NULL) {
  87                 last = 0;
  88
  89                 len = strlen((char *)buf);
  90                 CHOMP(buf, len, last);
  91                 if (mflag) {
  92                         MD5_CTX ctx;
  93
  94                         MD5Init(&ctx);
  95                         MD5Update(&ctx, buf, len);
  96                         while (!last &&
  97                             fgets((char *)buf, sizeof(buf), stdin) != NULL) {
  98                                 len = strlen((char *)buf);
  99                                 CHOMP(buf, len, last);
 100                                 MD5Update(&ctx, buf, len);
 101                         }
 102                         MD5Final(out, &ctx);
 103                         outlen = 16;
 104                 } else {
 105                         SHA1_CTX ctx;
 106
 107                         SHA1Init(&ctx);
 108                         SHA1Update(&ctx, buf, len);
 109                         while (!last &&
 110                             fgets((char *)buf, sizeof(buf), stdin) != NULL) {
 111                                 len = strlen((char *)buf);
 112                                 CHOMP(buf, len, last);
 113                                 SHA1Update(&ctx, buf, len);
 114                         }
 115                         SHA1Final(out, &ctx);
 116                         outlen = 20;
 117                 }
 118                 hexdump(out, outlen);
 119         }
 120 }
 121
 122 int
 123 main(int argc, char **argv)
 124 {
 125         int ch;
 126
 127         while ((ch = getopt(argc, argv, "mrst")) != -1)
 128                 switch (ch) {
 129                 case 'm':
 130                         mflag = 1;
 131                         break;
 132                 case 'r':
 133                         rflag = 1;
 134                         break;
 135                 case 's':
 136                         sflag = 1;
 137                         break;
 138                 case 't':
 139                         tflag = 1;
 140                         break;
 141                 case '?':
 142                 default:
 143                         usage();
 144                 }
 145         argc -= optind;
 146         argv += optind;
 147         if (argc > 0)
 148                 usage();
 149
 150         if (!(mflag || sflag))
 151                 mflag = 1;
 152
 153         if ((mflag ^ sflag) != 1)
 154                 usage();
 155
 156         if ((tflag ^ rflag) != 1)
 157                 usage();
 158
 159         if (tflag)
 160                 timetest();
 161
 162         if (rflag)
 163                 regress();
 164
 165         exit(0);
 166
 167 }