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 RCSID("$Id: add-random-users.c,v 1.6 2001/09/20 09:17:33 assar Exp $"); 37 38 #define WORDS_FILENAME "/usr/share/dict/words" 39 40 #define NUSERS 1000 41 42 #define WORDBUF_SIZE 65535 43 44 static unsigned 45 read_words (const char *filename, char ***ret_w) 46 { 47 unsigned n, alloc; 48 FILE *f; 49 char buf[256]; 50 char **w = NULL; 51 char *wbuf = NULL, *wptr = NULL, *wend = NULL; 52 53 f = fopen (filename, "r"); 54 if (f == NULL) 55 err (1, "cannot open %s", filename); 56 alloc = n = 0; 57 while (fgets (buf, sizeof(buf), f) != NULL) { 58 size_t len; 59 60 if (buf[strlen (buf) - 1] == '\n') 61 buf[strlen (buf) - 1] = '\0'; 62 if (n >= alloc) { 63 alloc = max(alloc + 16, alloc * 2); 64 w = erealloc (w, alloc * sizeof(char **)); 65 } 66 len = strlen(buf); 67 if (wptr + len + 1 >= wend) { 68 wptr = wbuf = emalloc (WORDBUF_SIZE); 69 wend = wbuf + WORDBUF_SIZE; 70 } 71 memmove (wptr, buf, len + 1); 72 w[n++] = wptr; 73 wptr += len + 1; 74 } 75 *ret_w = w; 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 optind = 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, &optind)) 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 -= optind; 174 argv += optind; 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 } 185