172445Sassar/* 2233294Sstas * Copyright (c) 2000 - 2001 Kungliga Tekniska H��gskolan 3233294Sstas * (Royal Institute of Technology, Stockholm, Sweden). 4233294Sstas * All rights reserved. 572445Sassar * 6233294Sstas * Redistribution and use in source and binary forms, with or without 7233294Sstas * modification, are permitted provided that the following conditions 8233294Sstas * are met: 972445Sassar * 10233294Sstas * 1. Redistributions of source code must retain the above copyright 11233294Sstas * notice, this list of conditions and the following disclaimer. 1272445Sassar * 13233294Sstas * 2. Redistributions in binary form must reproduce the above copyright 14233294Sstas * notice, this list of conditions and the following disclaimer in the 15233294Sstas * documentation and/or other materials provided with the distribution. 1672445Sassar * 17233294Sstas * 3. Neither the name of the Institute nor the names of its contributors 18233294Sstas * may be used to endorse or promote products derived from this software 19233294Sstas * without specific prior written permission. 2072445Sassar * 21233294Sstas * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND 22233294Sstas * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 23233294Sstas * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 24233294Sstas * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE 25233294Sstas * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 26233294Sstas * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 27233294Sstas * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 28233294Sstas * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 29233294Sstas * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 30233294Sstas * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 31233294Sstas * SUCH DAMAGE. 3272445Sassar */ 3372445Sassar 3472445Sassar#include "kadmin_locl.h" 3572445Sassar 3672445Sassar#define WORDS_FILENAME "/usr/share/dict/words" 3772445Sassar 3872445Sassar#define NUSERS 1000 3972445Sassar 4090926Snectar#define WORDBUF_SIZE 65535 4190926Snectar 4272445Sassarstatic unsigned 4372445Sassarread_words (const char *filename, char ***ret_w) 4472445Sassar{ 4572445Sassar unsigned n, alloc; 4672445Sassar FILE *f; 4772445Sassar char buf[256]; 4872445Sassar char **w = NULL; 4990926Snectar char *wbuf = NULL, *wptr = NULL, *wend = NULL; 5072445Sassar 5172445Sassar f = fopen (filename, "r"); 5272445Sassar if (f == NULL) 5372445Sassar err (1, "cannot open %s", filename); 5472445Sassar alloc = n = 0; 5572445Sassar while (fgets (buf, sizeof(buf), f) != NULL) { 5690926Snectar size_t len; 5790926Snectar 58178825Sdfr buf[strcspn(buf, "\r\n")] = '\0'; 5972445Sassar if (n >= alloc) { 6090926Snectar alloc = max(alloc + 16, alloc * 2); 6172445Sassar w = erealloc (w, alloc * sizeof(char **)); 6272445Sassar } 6390926Snectar len = strlen(buf); 6490926Snectar if (wptr + len + 1 >= wend) { 6590926Snectar wptr = wbuf = emalloc (WORDBUF_SIZE); 6690926Snectar wend = wbuf + WORDBUF_SIZE; 6790926Snectar } 6890926Snectar memmove (wptr, buf, len + 1); 6990926Snectar w[n++] = wptr; 7090926Snectar wptr += len + 1; 7172445Sassar } 72178825Sdfr if (n == 0) 73178825Sdfr errx(1, "%s is an empty file, no words to try", filename); 7472445Sassar *ret_w = w; 75233294Sstas fclose(f); 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); 121233294Sstas ret = kadm5_s_init_with_password_ctx(context, 12272445Sassar KADM5_ADMIN_SERVICE, 12372445Sassar NULL, 12472445Sassar KADM5_ADMIN_SERVICE, 125233294Sstas 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); 131233294Sstas 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{ 159178825Sdfr int optidx = 0; 16090926Snectar int n = NUSERS; 16190926Snectar const char *filename = WORDS_FILENAME; 16272445Sassar 16378527Sassar setprogname(argv[0]); 164178825Sdfr if(getarg(args, sizeof(args) / sizeof(args[0]), argc, argv, &optidx)) 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); 173178825Sdfr argc -= optidx; 174178825Sdfr argv += optidx; 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