granicus.if.org Git - apache/blob - support/htdigest.c

   1 /* ====================================================================
   2  * The Apache Software License, Version 1.1
   3  *
   4  * Copyright (c) 2000 The Apache Software Foundation.  All rights
   5  * 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  *
  11  * 1. Redistributions of source code must retain the above copyright
  12  *    notice, this list of conditions and the following disclaimer.
  13  *
  14  * 2. Redistributions in binary form must reproduce the above copyright
  15  *    notice, this list of conditions and the following disclaimer in
  16  *    the documentation and/or other materials provided with the
  17  *    distribution.
  18  *
  19  * 3. The end-user documentation included with the redistribution,
  20  *    if any, must include the following acknowledgment:
  21  *       "This product includes software developed by the
  22  *        Apache Software Foundation (http://www.apache.org/)."
  23  *    Alternately, this acknowledgment may appear in the software itself,
  24  *    if and wherever such third-party acknowledgments normally appear.
  25  *
  26  * 4. The names "Apache" and "Apache Software Foundation" must
  27  *    not be used to endorse or promote products derived from this
  28  *    software without prior written permission. For written
  29  *    permission, please contact apache@apache.org.
  30  *
  31  * 5. Products derived from this software may not be called "Apache",
  32  *    nor may "Apache" appear in their name, without prior written
  33  *    permission of the Apache Software Foundation.
  34  *
  35  * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
  36  * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  37  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  38  * DISCLAIMED.  IN NO EVENT SHALL THE APACHE SOFTWARE FOUNDATION OR
  39  * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  40  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  41  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
  42  * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  43  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  44  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
  45  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  46  * SUCH DAMAGE.
  47  * ====================================================================
  48  *
  49  * This software consists of voluntary contributions made by many
  50  * individuals on behalf of the Apache Software Foundation.  For more
  51  * information on the Apache Software Foundation, please see
  52  * <http://www.apache.org/>.
  53  */
  54
  55 /******************************************************************************
  56  ******************************************************************************
  57  * NOTE! This program is not safe as a setuid executable!  Do not make it
  58  * setuid!
  59  ******************************************************************************
  60  *****************************************************************************/
  61 /*
  62  * htdigest.c: simple program for manipulating digest passwd file for Apache
  63  *
  64  * by Alexei Kosut, based on htpasswd.c, by Rob McCool
  65  */
  66
  67 #include "apr.h"
  68 #include "apr_file_io.h"
  69 #include "apr_md5.h"
  70 #include "apr_lib.h"            /* for apr_getpass() */
  71 #include "apr_general.h"
  72
  73 #if APR_HAVE_STDIO_H
  74 #include <stdio.h>
  75 #endif
  76
  77 #ifdef HAVE_SYS_TYPES_H
  78 #include <sys/types.h>
  79 #endif
  80
  81 #ifdef HAVE_SYS_SIGNAL_H
  82 #include <sys/signal.h>
  83 #endif
  84
  85 #ifdef HAVE_SIGNAL_H
  86 #include <signal.h>
  87 #endif
  88
  89 #include <stdlib.h>
  90
  91 #ifdef WIN32
  92 #include <conio.h>
  93 #endif
  94
  95 #if APR_CHARSET_EBCDIC
  96 #define LF '\n'
  97 #define CR '\r'
  98 #else
  99 #define LF 10
 100 #define CR 13
 101 #endif /* APR_CHARSET_EBCDIC */
 102
 103 #define MAX_STRING_LEN 256
 104
 105 char *tn;
 106 apr_pool_t *cntxt;
 107 #if APR_CHARSET_EBCDIC
 108 apr_xlate_t *to_ascii;
 109 #endif
 110
 111 static void getword(char *word, char *line, char stop)
 112 {
 113     int x = 0, y;
 114
 115     for (x = 0; ((line[x]) && (line[x] != stop)); x++)
 116         word[x] = line[x];
 117
 118     word[x] = '\0';
 119     if (line[x])
 120         ++x;
 121     y = 0;
 122
 123     while ((line[y++] = line[x++]));
 124 }
 125
 126 static int getline(char *s, int n, apr_file_t *f)
 127 {
 128     register int i = 0;
 129     char ch;
 130
 131     while (1) {
 132         apr_getc(&ch, f);
 133             s[i] = ch;
 134
 135         if (s[i] == CR)
 136             apr_getc(&ch, f);
 137             s[i] = ch;
 138
 139         if ((s[i] == 0x4) || (s[i] == LF) || (i == (n - 1))) {
 140             s[i] = '\0';
 141             if (apr_eof(f) == APR_EOF) {
 142                 return 1;
 143             }
 144             return 0;
 145         }
 146         ++i;
 147     }
 148 }
 149
 150 static void putline(apr_file_t *f, char *l)
 151 {
 152     int x;
 153
 154     for (x = 0; l[x]; x++)
 155         apr_putc(l[x], f);
 156     apr_putc('\n', f);
 157 }
 158
 159
 160 static void add_password(char *user, char *realm, apr_file_t *f)
 161 {
 162     char *pw;
 163     apr_md5_ctx_t context;
 164     unsigned char digest[16];
 165     char string[MAX_STRING_LEN];
 166     char pwin[MAX_STRING_LEN];
 167     char pwv[MAX_STRING_LEN];
 168     unsigned int i;
 169     size_t len = sizeof(pwin);
 170
 171     if (apr_getpass("New password: ", pwin, &len) != APR_SUCCESS) {
 172         fprintf(stderr, "password too long");
 173         exit(5);
 174     }
 175     len = sizeof(pwin);
 176     apr_getpass("Re-type new password: ", pwv, &len);
 177     if (strcmp(pwin, pwv) != 0) {
 178         fprintf(stderr, "They don't match, sorry.\n");
 179         if (tn) {
 180             apr_remove_file(tn, cntxt);
 181         }
 182         exit(1);
 183     }
 184     pw = pwin;
 185     apr_fprintf(f, "%s:%s:", user, realm);
 186
 187     /* Do MD5 stuff */
 188     sprintf(string, "%s:%s:%s", user, realm, pw);
 189
 190     apr_MD5Init(&context);
 191 #if APR_CHARSET_EBCDIC
 192     apr_MD5SetXlate(&context, to_ascii);
 193 #endif
 194     apr_MD5Update(&context, (unsigned char *) string, strlen(string));
 195     apr_MD5Final(digest, &context);
 196
 197     for (i = 0; i < 16; i++)
 198         apr_fprintf(f, "%02x", digest[i]);
 199
 200     apr_fprintf(f, "\n");
 201 }
 202
 203 static void usage(void)
 204 {
 205     fprintf(stderr, "Usage: htdigest [-c] passwordfile realm username\n");
 206     fprintf(stderr, "The -c flag creates a new file.\n");
 207     exit(1);
 208 }
 209
 210 static void interrupted(void)
 211 {
 212     fprintf(stderr, "Interrupted.\n");
 213     if (tn)
 214         apr_remove_file(tn, cntxt);
 215     exit(1);
 216 }
 217
 218 int main(int argc, char *argv[])
 219 {
 220     apr_file_t *tfp = NULL, *f;
 221     apr_status_t rv;
 222     char user[MAX_STRING_LEN];
 223     char realm[MAX_STRING_LEN];
 224     char line[MAX_STRING_LEN];
 225     char l[MAX_STRING_LEN];
 226     char w[MAX_STRING_LEN];
 227     char x[MAX_STRING_LEN];
 228     char command[MAX_STRING_LEN];
 229     int found;
 230
 231     rv = apr_initialize();
 232     if (rv) {
 233         fprintf(stderr, "apr_initialize(): %s (%d)\n",
 234                 apr_strerror(rv, line, sizeof(line)), rv);
 235         exit(1);
 236     }
 237     atexit(apr_terminate);
 238     apr_create_pool(&cntxt, NULL);
 239
 240 #if APR_CHARSET_EBCDIC
 241     rv = apr_xlate_open(&to_ascii, "ISO8859-1", APR_DEFAULT_CHARSET, cntxt);
 242     if (rv) {
 243         fprintf(stderr, "apr_xlate_open(): %s (%d)\n",
 244                 apr_strerror(rv, line, sizeof(line)), rv);
 245         exit(1);
 246     }
 247 #endif
 248
 249     tn = NULL;
 250     apr_signal(SIGINT, (void (*)(int)) interrupted);
 251     if (argc == 5) {
 252         if (strcmp(argv[1], "-c"))
 253             usage();
 254         if (apr_open(&tfp, argv[2], APR_WRITE | APR_CREATE, -1, cntxt) != APR_SUCCESS) {
 255             fprintf(stderr, "Could not open passwd file %s for writing.\n",
 256                     argv[2]);
 257             perror("apr_open");
 258             exit(1);
 259         }
 260         printf("Adding password for %s in realm %s.\n", argv[4], argv[3]);
 261         add_password(argv[4], argv[3], tfp);
 262         apr_close(tfp);
 263         exit(0);
 264     }
 265     else if (argc != 4)
 266         usage();
 267
 268     tn = tmpnam(NULL);
 269     if (apr_open(&tfp, tn, APR_WRITE | APR_CREATE, -1, cntxt)!= APR_SUCCESS) {
 270         fprintf(stderr, "Could not open temp file.\n");
 271         exit(1);
 272     }
 273
 274     if (apr_open(&f, argv[1], APR_READ, -1, cntxt) != APR_SUCCESS) {
 275         fprintf(stderr,
 276                 "Could not open passwd file %s for reading.\n", argv[1]);
 277         fprintf(stderr, "Use -c option to create new one.\n");
 278         exit(1);
 279     }
 280     strcpy(user, argv[3]);
 281     strcpy(realm, argv[2]);
 282
 283     found = 0;
 284     while (!(getline(line, MAX_STRING_LEN, f))) {
 285         if (found || (line[0] == '#') || (!line[0])) {
 286             putline(tfp, line);
 287             continue;
 288         }
 289         strcpy(l, line);
 290         getword(w, l, ':');
 291         getword(x, l, ':');
 292         if (strcmp(user, w) || strcmp(realm, x)) {
 293             putline(tfp, line);
 294             continue;
 295         }
 296         else {
 297             printf("Changing password for user %s in realm %s\n", user, realm);
 298             add_password(user, realm, tfp);
 299             found = 1;
 300         }
 301     }
 302     if (!found) {
 303         printf("Adding user %s in realm %s\n", user, realm);
 304         add_password(user, realm, tfp);
 305     }
 306     apr_close(f);
 307     apr_close(tfp);
 308 #if defined(OS2) || defined(WIN32)
 309     sprintf(command, "copy \"%s\" \"%s\"", tn, argv[1]);
 310 #else
 311     sprintf(command, "cp %s %s", tn, argv[1]);
 312 #endif
 313     system(command);
 314     apr_remove_file(tn, cntxt);
 315     return 0;
 316 }