1  /*
  2   * Copyright (c) 2000 - 2001 Kungliga Tekniska Högskolan
  3   * (Royal Institute of Technology, Stockholm, Sweden).
  4   * All rights reserved.
  5   *
  6   * Redistribution and use in source and binary forms, with or without
  7   * modification, are permitted provided that the following conditions
  8   * are met:
  9   *
 10   * 1. Redistributions of source code must retain the above copyright
 11   *    notice, this list of conditions and the following disclaimer.
 12   *
 13   * 2. Redistributions in binary form must reproduce the above copyright
 14   *    notice, this list of conditions and the following disclaimer in the
 15   *    documentation and/or other materials provided with the distribution.
 16   *
 17   * 3. Neither the name of the Institute nor the names of its contributors
 18   *    may be used to endorse or promote products derived from this software
 19   *    without specific prior written permission.
 20   *
 21   * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
 22   * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 23   * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 24   * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
 25   * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 26   * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 27   * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 28   * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 29   * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 30   * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 31   * SUCH DAMAGE.
 32   */
 33  
 34  #include "kadmin_locl.h"
 35  
 36  #define WORDS_FILENAME "/usr/share/dict/words"
 37  
 38  #define NUSERS 1000
 39  
 40  #define WORDBUF_SIZE 65535
 41  
 42  static unsigned
 43  read_words (const char *filename, char ***ret_w)
 44  {
 45      unsigned n, alloc;
 46      FILE *f;
 47      char buf[256];
 48      char **w = NULL;
 49      char *wbuf = NULL, *wptr = NULL, *wend = NULL;
 50  
 51      f = fopen (filename, "r");
 52      if (f == NULL)
 53  	err (1, "cannot open %s", filename);
 54      alloc = n = 0;
 55      while (fgets (buf, sizeof(buf), f) != NULL) {
 56  	size_t len;
 57  
 58  	buf[strcspn(buf, "\r\n")] = '\0';
 59  	if (n >= alloc) {
 60  	    alloc = max(alloc + 16, alloc * 2);
 61  	    w = erealloc (w, alloc * sizeof(char **));
 62  	}
 63  	len = strlen(buf);
 64  	if (wptr + len + 1 >= wend) {
 65  	    wptr = wbuf = emalloc (WORDBUF_SIZE);
 66  	    wend = wbuf + WORDBUF_SIZE;
 67  	}
 68  	memmove (wptr, buf, len + 1);
 69  	w[n++] = wptr;
 70  	wptr += len + 1;
 71      }
 72      if (n == 0)
 73  	errx(1, "%s is an empty file, no words to try", filename);
 74      *ret_w = w;
 75      fclose(f);
 76      return n;
 77  }
 78  
 79  static void
 80  add_user (krb5_context context, void *kadm_handle,
 81  	  unsigned nwords, char **words)
 82  {
 83      kadm5_principal_ent_rec princ;
 84      char name[64];
 85      int r1, r2;
 86      krb5_error_code ret;
 87      int mask;
 88  
 89      r1 = rand();
 90      r2 = rand();
 91  
 92      snprintf (name, sizeof(name), "%s%d", words[r1 % nwords], r2 % 1000);
 93  
 94      mask = KADM5_PRINCIPAL;
 95  
 96      memset(&princ, 0, sizeof(princ));
 97      ret = krb5_parse_name(context, name, &princ.principal);
 98      if (ret)
 99  	krb5_err(context, 1, ret, "krb5_parse_name");
100  
101      ret = kadm5_create_principal (kadm_handle, &princ, mask, name);
102      if (ret)
103  	krb5_err (context, 1, ret, "kadm5_create_principal");
104      kadm5_free_principal_ent(kadm_handle, &princ);
105      printf ("%s\n", name);
106  }
107  
108  static void
109  add_users (const char *filename, unsigned n)
110  {
111      krb5_error_code ret;
112      int i;
113      void *kadm_handle;
114      krb5_context context;
115      unsigned nwords;
116      char **words;
117  
118      ret = krb5_init_context(&context);
119      if (ret)
120  	errx (1, "krb5_init_context failed: %d", ret);
121      ret = kadm5_s_init_with_password_ctx(context,
122  					 KADM5_ADMIN_SERVICE,
123  					 NULL,
124  					 KADM5_ADMIN_SERVICE,
125  					 NULL, 0, 0,
126  					 &kadm_handle);
127      if(ret)
128  	krb5_err(context, 1, ret, "kadm5_init_with_password");
129  
130      nwords = read_words (filename, &words);
131  
132      for (i = 0; i < n; ++i)
133  	add_user (context, kadm_handle, nwords, words);
134      kadm5_destroy(kadm_handle);
135      krb5_free_context(context);
136  }
137  
138  static int version_flag	= 0;
139  static int help_flag	= 0;
140  
141  static struct getargs args[] = {
142      { "version", 	0,   arg_flag, &version_flag },
143      { "help",		0,   arg_flag, &help_flag }
144  };
145  
146  static void
147  usage (int ret)
148  {
149      arg_printusage (args,
150  		    sizeof(args)/sizeof(*args),
151  		    NULL,
152  		    "[filename [n]]");
153      exit (ret);
154  }
155  
156  int
157  main(int argc, char **argv)
158  {
159      int optidx = 0;
160      int n = NUSERS;
161      const char *filename = WORDS_FILENAME;
162  
163      setprogname(argv[0]);
164      if(getarg(args, sizeof(args) / sizeof(args[0]), argc, argv, &optidx))
165  	usage(1);
166      if (help_flag)
167  	usage (0);
168      if (version_flag) {
169  	print_version(NULL);
170  	return 0;
171      }
172      srand (0);
173      argc -= optidx;
174      argv += optidx;
175  
176      if (argc > 0) {
177  	if (argc > 1)
178  	    n = atoi(argv[1]);
179  	filename = argv[0];
180      }
181  
182      add_users (filename, n);
183      return 0;
184  }