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

   1 /* ====================================================================
   2  * The Apache Software License, Version 1.1
   3  *
   4  * Copyright (c) 2000-2002 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 #include "apr_signal.h"
  73
  74 #define APR_WANT_STDIO
  75 #define APR_WANT_STRFUNC
  76 #include "apr_want.h"
  77
  78 #if APR_HAVE_SYS_TYPES_H
  79 #include <sys/types.h>
  80 #endif
  81 #if APR_HAVE_STDLIB_H
  82 #include <stdlib.h>
  83 #endif
  84
  85 #ifdef WIN32
  86 #include <conio.h>
  87 #endif
  88
  89
  90 #if APR_CHARSET_EBCDIC
  91 #define LF '\n'
  92 #define CR '\r'
  93 #else
  94 #define LF 10
  95 #define CR 13
  96 #endif /* APR_CHARSET_EBCDIC */
  97
  98 #define MAX_STRING_LEN 256
  99
 100 apr_file_t *tfp = NULL;
 101 apr_pool_t *cntxt;
 102 #if APR_CHARSET_EBCDIC
 103 apr_xlate_t *to_ascii;
 104 #endif
 105
 106 static void getword(char *word, char *line, char stop)
 107 {
 108     int x = 0, y;
 109
 110     for (x = 0; ((line[x]) && (line[x] != stop)); x++)
 111         word[x] = line[x];
 112
 113     word[x] = '\0';
 114     if (line[x])
 115         ++x;
 116     y = 0;
 117
 118     while ((line[y++] = line[x++]));
 119 }
 120
 121 static int getline(char *s, int n, apr_file_t *f)
 122 {
 123     register int i = 0;
 124     char ch;
 125     apr_status_t rv = APR_EINVAL;
 126
 127     while (i < (n - 1) &&
 128            ((rv = apr_file_getc(&ch, f)) == APR_SUCCESS) && (ch != '\n')) {
 129         s[i++] = ch;
 130     }
 131     if (ch == '\n')
 132         s[i++] = ch;
 133     s[i] = '\0';
 134
 135     if (rv != APR_SUCCESS)
 136         return 1;
 137
 138     return 0;
 139 }
 140
 141 static void putline(apr_file_t *f, char *l)
 142 {
 143     int x;
 144
 145     for (x = 0; l[x]; x++)
 146         apr_file_putc(l[x], f);
 147 }
 148
 149
 150 static void add_password(char *user, char *realm, apr_file_t *f)
 151 {
 152     char *pw;
 153     apr_md5_ctx_t context;
 154     unsigned char digest[16];
 155     char string[MAX_STRING_LEN];
 156     char pwin[MAX_STRING_LEN];
 157     char pwv[MAX_STRING_LEN];
 158     unsigned int i;
 159     size_t len = sizeof(pwin);
 160
 161     if (apr_password_get("New password: ", pwin, &len) != APR_SUCCESS) {
 162         fprintf(stderr, "password too long");
 163         exit(5);
 164     }
 165     len = sizeof(pwin);
 166     apr_password_get("Re-type new password: ", pwv, &len);
 167     if (strcmp(pwin, pwv) != 0) {
 168         fprintf(stderr, "They don't match, sorry.\n");
 169         if (tfp) {
 170             apr_file_close(tfp);
 171         }
 172         exit(1);
 173     }
 174     pw = pwin;
 175     apr_file_printf(f, "%s:%s:", user, realm);
 176
 177     /* Do MD5 stuff */
 178     sprintf(string, "%s:%s:%s", user, realm, pw);
 179
 180     apr_md5_init(&context);
 181 #if APR_CHARSET_EBCDIC
 182     apr_md5_set_xlate(&context, to_ascii);
 183 #endif
 184     apr_md5_update(&context, (unsigned char *) string, strlen(string));
 185     apr_md5_final(digest, &context);
 186
 187     for (i = 0; i < 16; i++)
 188         apr_file_printf(f, "%02x", digest[i]);
 189
 190     apr_file_printf(f, "\n");
 191 }
 192
 193 static void usage(void)
 194 {
 195     fprintf(stderr, "Usage: htdigest [-c] passwordfile realm username\n");
 196     fprintf(stderr, "The -c flag creates a new file.\n");
 197     exit(1);
 198 }
 199
 200 static void interrupted(void)
 201 {
 202     fprintf(stderr, "Interrupted.\n");
 203     if (tfp) {
 204         apr_file_close(tfp);
 205     }
 206     exit(1);
 207 }
 208
 209 static void terminate(void)
 210 {
 211 #ifdef NETWARE
 212     pressanykey();
 213 #endif
 214     apr_terminate();
 215 }
 216
 217 int main(int argc, char *argv[])
 218 {
 219     apr_file_t *f;
 220     apr_status_t rv;
 221     char tn[] = "htdigest.tmp.XXXXXX";
 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(terminate);
 238     apr_pool_create(&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     apr_signal(SIGINT, (void (*)(int)) interrupted);
 250     if (argc == 5) {
 251         if (strcmp(argv[1], "-c"))
 252             usage();
 253         rv = apr_file_open(&f, argv[2], APR_WRITE | APR_CREATE, -1, cntxt);
 254         if (rv != APR_SUCCESS) {
 255             char errmsg[120];
 256
 257             fprintf(stderr, "Could not open passwd file %s for writing: %s\n",
 258                     argv[2],
 259                     apr_strerror(rv, errmsg, sizeof errmsg));
 260             exit(1);
 261         }
 262         printf("Adding password for %s in realm %s.\n", argv[4], argv[3]);
 263         add_password(argv[4], argv[3], f);
 264         apr_file_close(f);
 265         exit(0);
 266     }
 267     else if (argc != 4)
 268         usage();
 269
 270     if (apr_file_mktemp(&tfp, tn, 0, cntxt) != APR_SUCCESS) {
 271         fprintf(stderr, "Could not open temp file.\n");
 272         exit(1);
 273     }
 274
 275     if (apr_file_open(&f, argv[1], APR_READ, -1, cntxt) != APR_SUCCESS) {
 276         fprintf(stderr,
 277                 "Could not open passwd file %s for reading.\n", argv[1]);
 278         fprintf(stderr, "Use -c option to create new one.\n");
 279         exit(1);
 280     }
 281     strcpy(user, argv[3]);
 282     strcpy(realm, argv[2]);
 283
 284     found = 0;
 285     while (!(getline(line, MAX_STRING_LEN, f))) {
 286         if (found || (line[0] == '#') || (!line[0])) {
 287             putline(tfp, line);
 288             continue;
 289         }
 290         strcpy(l, line);
 291         getword(w, l, ':');
 292         getword(x, l, ':');
 293         if (strcmp(user, w) || strcmp(realm, x)) {
 294             putline(tfp, line);
 295             continue;
 296         }
 297         else {
 298             printf("Changing password for user %s in realm %s\n", user, realm);
 299             add_password(user, realm, tfp);
 300             found = 1;
 301         }
 302     }
 303     if (!found) {
 304         printf("Adding user %s in realm %s\n", user, realm);
 305         add_password(user, realm, tfp);
 306     }
 307     apr_file_close(f);
 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_file_close(tfp);
 315     return 0;
 316 }