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
read_words(const char * filename,char *** ret_w)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
add_user(krb5_context ctx,void * hndl,unsigned nwords,char ** words)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
add_users(const char * filename,unsigned n)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
usage(int ret)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
main(int argc,char ** argv)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