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