add-random-users.c revision 90926
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 3690926SnectarRCSID("$Id: add-random-users.c,v 1.6 2001/09/20 09:17:33 assar Exp $"); 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 6072445Sassar if (buf[strlen (buf) - 1] == '\n') 6172445Sassar buf[strlen (buf) - 1] = '\0'; 6272445Sassar if (n >= alloc) { 6390926Snectar alloc = max(alloc + 16, alloc * 2); 6472445Sassar w = erealloc (w, alloc * sizeof(char **)); 6572445Sassar } 6690926Snectar len = strlen(buf); 6790926Snectar if (wptr + len + 1 >= wend) { 6890926Snectar wptr = wbuf = emalloc (WORDBUF_SIZE); 6990926Snectar wend = wbuf + WORDBUF_SIZE; 7090926Snectar } 7190926Snectar memmove (wptr, buf, len + 1); 7290926Snectar w[n++] = wptr; 7390926Snectar wptr += len + 1; 7472445Sassar } 7572445Sassar *ret_w = w; 7672445Sassar return n; 7772445Sassar} 7872445Sassar 7972445Sassarstatic void 8072445Sassaradd_user (krb5_context context, void *kadm_handle, 8172445Sassar unsigned nwords, char **words) 8272445Sassar{ 8372445Sassar kadm5_principal_ent_rec princ; 8472445Sassar char name[64]; 8572445Sassar int r1, r2; 8672445Sassar krb5_error_code ret; 8772445Sassar int mask; 8872445Sassar 8972445Sassar r1 = rand(); 9072445Sassar r2 = rand(); 9172445Sassar 9272445Sassar snprintf (name, sizeof(name), "%s%d", words[r1 % nwords], r2 % 1000); 9372445Sassar 9472445Sassar mask = KADM5_PRINCIPAL; 9572445Sassar 9672445Sassar memset(&princ, 0, sizeof(princ)); 9772445Sassar ret = krb5_parse_name(context, name, &princ.principal); 9872445Sassar if (ret) 9972445Sassar krb5_err(context, 1, ret, "krb5_parse_name"); 10072445Sassar 10172445Sassar ret = kadm5_create_principal (kadm_handle, &princ, mask, name); 10272445Sassar if (ret) 10372445Sassar krb5_err (context, 1, ret, "kadm5_create_principal"); 10472445Sassar kadm5_free_principal_ent(kadm_handle, &princ); 10572445Sassar printf ("%s\n", name); 10672445Sassar} 10772445Sassar 10872445Sassarstatic void 10990926Snectaradd_users (const char *filename, unsigned n) 11072445Sassar{ 11172445Sassar krb5_error_code ret; 11272445Sassar int i; 11372445Sassar void *kadm_handle; 11472445Sassar krb5_context context; 11572445Sassar unsigned nwords; 11672445Sassar char **words; 11772445Sassar 11872445Sassar ret = krb5_init_context(&context); 11972445Sassar if (ret) 12072445Sassar errx (1, "krb5_init_context failed: %d", ret); 12172445Sassar ret = kadm5_s_init_with_password_ctx(context, 12272445Sassar KADM5_ADMIN_SERVICE, 12372445Sassar NULL, 12472445Sassar KADM5_ADMIN_SERVICE, 12572445Sassar NULL, 0, 0, 12672445Sassar &kadm_handle); 12772445Sassar if(ret) 12872445Sassar krb5_err(context, 1, ret, "kadm5_init_with_password"); 12972445Sassar 13090926Snectar nwords = read_words (filename, &words); 13172445Sassar 13272445Sassar for (i = 0; i < n; ++i) 13372445Sassar add_user (context, kadm_handle, nwords, words); 13472445Sassar kadm5_destroy(kadm_handle); 13572445Sassar krb5_free_context(context); 13672445Sassar} 13772445Sassar 13872445Sassarstatic int version_flag = 0; 13972445Sassarstatic int help_flag = 0; 14072445Sassar 14172445Sassarstatic struct getargs args[] = { 14272445Sassar { "version", 0, arg_flag, &version_flag }, 14372445Sassar { "help", 0, arg_flag, &help_flag } 14472445Sassar}; 14572445Sassar 14672445Sassarstatic void 14772445Sassarusage (int ret) 14872445Sassar{ 14972445Sassar arg_printusage (args, 15072445Sassar sizeof(args)/sizeof(*args), 15172445Sassar NULL, 15290926Snectar "[filename [n]]"); 15372445Sassar exit (ret); 15472445Sassar} 15572445Sassar 15672445Sassarint 15772445Sassarmain(int argc, char **argv) 15872445Sassar{ 15972445Sassar int optind = 0; 16090926Snectar int n = NUSERS; 16190926Snectar const char *filename = WORDS_FILENAME; 16272445Sassar 16378527Sassar setprogname(argv[0]); 16472445Sassar if(getarg(args, sizeof(args) / sizeof(args[0]), argc, argv, &optind)) 16572445Sassar usage(1); 16672445Sassar if (help_flag) 16772445Sassar usage (0); 16890926Snectar if (version_flag) { 16990926Snectar print_version(NULL); 17090926Snectar return 0; 17190926Snectar } 17272445Sassar srand (0); 17390926Snectar argc -= optind; 17490926Snectar argv += optind; 17590926Snectar 17690926Snectar if (argc > 0) { 17790926Snectar if (argc > 1) 17890926Snectar n = atoi(argv[1]); 17990926Snectar filename = argv[0]; 18090926Snectar } 18190926Snectar 18290926Snectar add_users (filename, n); 18372445Sassar return 0; 18472445Sassar} 185