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
36RCSID("$Id: add-random-users.c 19213 2006-12-04 23:36:36Z lha $");
37
38#define WORDS_FILENAME "/usr/share/dict/words"
39
40#define NUSERS 1000
41
42#define WORDBUF_SIZE 65535
43
44static unsigned
45read_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    return n;
78}
79
80static void
81add_user (krb5_context context, void *kadm_handle,
82	  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(context, name, &princ.principal);
99    if (ret)
100	krb5_err(context, 1, ret, "krb5_parse_name");
101
102    ret = kadm5_create_principal (kadm_handle, &princ, mask, name);
103    if (ret)
104	krb5_err (context, 1, ret, "kadm5_create_principal");
105    kadm5_free_principal_ent(kadm_handle, &princ);
106    printf ("%s\n", name);
107}
108
109static void
110add_users (const char *filename, unsigned n)
111{
112    krb5_error_code ret;
113    int i;
114    void *kadm_handle;
115    krb5_context context;
116    unsigned nwords;
117    char **words;
118
119    ret = krb5_init_context(&context);
120    if (ret)
121	errx (1, "krb5_init_context failed: %d", ret);
122    ret = kadm5_s_init_with_password_ctx(context,
123					 KADM5_ADMIN_SERVICE,
124					 NULL,
125					 KADM5_ADMIN_SERVICE,
126					 NULL, 0, 0,
127					 &kadm_handle);
128    if(ret)
129	krb5_err(context, 1, ret, "kadm5_init_with_password");
130
131    nwords = read_words (filename, &words);
132
133    for (i = 0; i < n; ++i)
134	add_user (context, kadm_handle, nwords, words);
135    kadm5_destroy(kadm_handle);
136    krb5_free_context(context);
137}
138
139static int version_flag	= 0;
140static int help_flag	= 0;
141
142static struct getargs args[] = {
143    { "version", 	0,   arg_flag, &version_flag },
144    { "help",		0,   arg_flag, &help_flag }
145};
146
147static void
148usage (int ret)
149{
150    arg_printusage (args,
151		    sizeof(args)/sizeof(*args),
152		    NULL,
153		    "[filename [n]]");
154    exit (ret);
155}
156
157int
158main(int argc, char **argv)
159{
160    int optidx = 0;
161    int n = NUSERS;
162    const char *filename = WORDS_FILENAME;
163
164    setprogname(argv[0]);
165    if(getarg(args, sizeof(args) / sizeof(args[0]), argc, argv, &optidx))
166	usage(1);
167    if (help_flag)
168	usage (0);
169    if (version_flag) {
170	print_version(NULL);
171	return 0;
172    }
173    srand (0);
174    argc -= optidx;
175    argv += optidx;
176
177    if (argc > 0) {
178	if (argc > 1)
179	    n = atoi(argv[1]);
180	filename = argv[0];
181    }
182
183    add_users (filename, n);
184    return 0;
185}
186