172445Sassar/* 290926Snectar * Copyright (c) 2000 - 2001 Kungliga Tekniska H�gskolan 372445Sassar * (Royal Institute of Technology, Stockholm, Sweden). 472445Sassar * All rights reserved. 572445Sassar * 672445Sassar * Redistribution and use in source and binary forms, with or without 772445Sassar * modification, are permitted provided that the following conditions 872445Sassar * are met: 972445Sassar * 1072445Sassar * 1. Redistributions of source code must retain the above copyright 1172445Sassar * notice, this list of conditions and the following disclaimer. 1272445Sassar * 1372445Sassar * 2. Redistributions in binary form must reproduce the above copyright 1472445Sassar * notice, this list of conditions and the following disclaimer in the 1572445Sassar * documentation and/or other materials provided with the distribution. 1672445Sassar * 1772445Sassar * 3. Neither the name of the Institute nor the names of its contributors 1872445Sassar * may be used to endorse or promote products derived from this software 1972445Sassar * without specific prior written permission. 2072445Sassar * 2172445Sassar * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND 2272445Sassar * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 2372445Sassar * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 2472445Sassar * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE 2572445Sassar * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 2672445Sassar * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 2772445Sassar * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 2872445Sassar * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 2972445Sassar * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 3072445Sassar * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 3172445Sassar * SUCH DAMAGE. 3272445Sassar */ 3372445Sassar 3472445Sassar#include "kadmin_locl.h" 3572445Sassar 36178825SdfrRCSID("$Id: add-random-users.c 19213 2006-12-04 23:36:36Z lha $"); 3772445Sassar 3872445Sassar#define WORDS_FILENAME "/usr/share/dict/words" 3972445Sassar 4072445Sassar#define NUSERS 1000 4172445Sassar 4290926Snectar#define WORDBUF_SIZE 65535 4390926Snectar 4472445Sassarstatic unsigned 4572445Sassarread_words (const char *filename, char ***ret_w) 4672445Sassar{ 4772445Sassar unsigned n, alloc; 4872445Sassar FILE *f; 4972445Sassar char buf[256]; 5072445Sassar char **w = NULL; 5190926Snectar char *wbuf = NULL, *wptr = NULL, *wend = NULL; 5272445Sassar 5372445Sassar f = fopen (filename, "r"); 5472445Sassar if (f == NULL) 5572445Sassar err (1, "cannot open %s", filename); 5672445Sassar alloc = n = 0; 5772445Sassar while (fgets (buf, sizeof(buf), f) != NULL) { 5890926Snectar size_t len; 5990926Snectar 60178825Sdfr buf[strcspn(buf, "\r\n")] = '\0'; 6172445Sassar if (n >= alloc) { 6290926Snectar alloc = max(alloc + 16, alloc * 2); 6372445Sassar w = erealloc (w, alloc * sizeof(char **)); 6472445Sassar } 6590926Snectar len = strlen(buf); 6690926Snectar if (wptr + len + 1 >= wend) { 6790926Snectar wptr = wbuf = emalloc (WORDBUF_SIZE); 6890926Snectar wend = wbuf + WORDBUF_SIZE; 6990926Snectar } 7090926Snectar memmove (wptr, buf, len + 1); 7190926Snectar w[n++] = wptr; 7290926Snectar wptr += len + 1; 7372445Sassar } 74178825Sdfr if (n == 0) 75178825Sdfr errx(1, "%s is an empty file, no words to try", filename); 7672445Sassar *ret_w = w; 7772445Sassar return n; 7872445Sassar} 7972445Sassar 8072445Sassarstatic void 8172445Sassaradd_user (krb5_context context, void *kadm_handle, 8272445Sassar unsigned nwords, char **words) 8372445Sassar{ 8472445Sassar kadm5_principal_ent_rec princ; 8572445Sassar char name[64]; 8672445Sassar int r1, r2; 8772445Sassar krb5_error_code ret; 8872445Sassar int mask; 8972445Sassar 9072445Sassar r1 = rand(); 9172445Sassar r2 = rand(); 9272445Sassar 9372445Sassar snprintf (name, sizeof(name), "%s%d", words[r1 % nwords], r2 % 1000); 9472445Sassar 9572445Sassar mask = KADM5_PRINCIPAL; 9672445Sassar 9772445Sassar memset(&princ, 0, sizeof(princ)); 9872445Sassar ret = krb5_parse_name(context, name, &princ.principal); 9972445Sassar if (ret) 10072445Sassar krb5_err(context, 1, ret, "krb5_parse_name"); 10172445Sassar 10272445Sassar ret = kadm5_create_principal (kadm_handle, &princ, mask, name); 10372445Sassar if (ret) 10472445Sassar krb5_err (context, 1, ret, "kadm5_create_principal"); 10572445Sassar kadm5_free_principal_ent(kadm_handle, &princ); 10672445Sassar printf ("%s\n", name); 10772445Sassar} 10872445Sassar 10972445Sassarstatic void 11090926Snectaradd_users (const char *filename, unsigned n) 11172445Sassar{ 11272445Sassar krb5_error_code ret; 11372445Sassar int i; 11472445Sassar void *kadm_handle; 11572445Sassar krb5_context context; 11672445Sassar unsigned nwords; 11772445Sassar char **words; 11872445Sassar 11972445Sassar ret = krb5_init_context(&context); 12072445Sassar if (ret) 12172445Sassar errx (1, "krb5_init_context failed: %d", ret); 12272445Sassar ret = kadm5_s_init_with_password_ctx(context, 12372445Sassar KADM5_ADMIN_SERVICE, 12472445Sassar NULL, 12572445Sassar KADM5_ADMIN_SERVICE, 12672445Sassar NULL, 0, 0, 12772445Sassar &kadm_handle); 12872445Sassar if(ret) 12972445Sassar krb5_err(context, 1, ret, "kadm5_init_with_password"); 13072445Sassar 13190926Snectar nwords = read_words (filename, &words); 13272445Sassar 13372445Sassar for (i = 0; i < n; ++i) 13472445Sassar add_user (context, kadm_handle, nwords, words); 13572445Sassar kadm5_destroy(kadm_handle); 13672445Sassar krb5_free_context(context); 13772445Sassar} 13872445Sassar 13972445Sassarstatic int version_flag = 0; 14072445Sassarstatic int help_flag = 0; 14172445Sassar 14272445Sassarstatic struct getargs args[] = { 14372445Sassar { "version", 0, arg_flag, &version_flag }, 14472445Sassar { "help", 0, arg_flag, &help_flag } 14572445Sassar}; 14672445Sassar 14772445Sassarstatic void 14872445Sassarusage (int ret) 14972445Sassar{ 15072445Sassar arg_printusage (args, 15172445Sassar sizeof(args)/sizeof(*args), 15272445Sassar NULL, 15390926Snectar "[filename [n]]"); 15472445Sassar exit (ret); 15572445Sassar} 15672445Sassar 15772445Sassarint 15872445Sassarmain(int argc, char **argv) 15972445Sassar{ 160178825Sdfr int optidx = 0; 16190926Snectar int n = NUSERS; 16290926Snectar const char *filename = WORDS_FILENAME; 16372445Sassar 16478527Sassar setprogname(argv[0]); 165178825Sdfr if(getarg(args, sizeof(args) / sizeof(args[0]), argc, argv, &optidx)) 16672445Sassar usage(1); 16772445Sassar if (help_flag) 16872445Sassar usage (0); 16990926Snectar if (version_flag) { 17090926Snectar print_version(NULL); 17190926Snectar return 0; 17290926Snectar } 17372445Sassar srand (0); 174178825Sdfr argc -= optidx; 175178825Sdfr argv += optidx; 17690926Snectar 17790926Snectar if (argc > 0) { 17890926Snectar if (argc > 1) 17990926Snectar n = atoi(argv[1]); 18090926Snectar filename = argv[0]; 18190926Snectar } 18290926Snectar 18390926Snectar add_users (filename, n); 18472445Sassar return 0; 18572445Sassar} 186