lorikeet-heimdal: add wrap_ex_ntlm.diff from abartlet
[metze/heimdal/wip.git] / kadmin / kadmin.c
1 /*
2  * Copyright (c) 1997 - 2004 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 #include "kadmin-commands.h"
36 #include <sl.h>
37
38 RCSID("$Id$");
39
40 static char *config_file;
41 static char *keyfile;
42 int local_flag;
43 static int ad_flag;
44 static int help_flag;
45 static int version_flag;
46 static char *realm;
47 static char *admin_server;
48 static int server_port = 0;
49 static char *client_name;
50 static char *keytab;
51 static char *check_library  = NULL;
52 static char *check_function = NULL;
53 static getarg_strings policy_libraries = { 0, NULL };
54
55 static struct getargs args[] = {
56     {   "principal",    'p',    arg_string,     &client_name,
57         "principal to authenticate as" },
58     {   "keytab",       'K',    arg_string,     &keytab,
59         "keytab for authentication principal" },
60     {
61         "config-file",  'c',    arg_string,     &config_file,
62         "location of config file",      "file"
63     },
64     {
65         "key-file",     'k',    arg_string, &keyfile,
66         "location of master key file", "file"
67     },
68     {   
69         "realm",        'r',    arg_string,   &realm,
70         "realm to use", "realm"
71     },
72     {   
73         "admin-server", 'a',    arg_string,   &admin_server,
74         "server to contact", "host"
75     },
76     {   
77         "server-port",  's',    arg_integer,   &server_port,
78         "port to use", "port number"
79     },
80     {   "ad",           0, arg_flag, &ad_flag, "active directory admin mode" },
81 #ifdef HAVE_DLOPEN
82     { "check-library", 0, arg_string, &check_library,
83       "library to load password check function from", "library" },
84     { "check-function", 0, arg_string, &check_function,
85       "password check function to load", "function" },
86     { "policy-libraries", 0, arg_strings, &policy_libraries,
87       "password check function to load", "function" },
88 #endif
89     {   "local", 'l', arg_flag, &local_flag, "local admin mode" },
90     {   "help",         'h',    arg_flag,   &help_flag },
91     {   "version",      'v',    arg_flag,   &version_flag }
92 };
93
94 static int num_args = sizeof(args) / sizeof(args[0]);
95
96
97 krb5_context context;
98 void *kadm_handle;
99
100 int
101 help(void *opt, int argc, char **argv)
102 {
103     sl_slc_help(commands, argc, argv);
104     return 0;
105 }
106
107 static int exit_seen = 0;
108
109 int
110 exit_kadmin (void *opt, int argc, char **argv)
111 {
112     exit_seen = 1;
113     return 0;
114 }
115
116 static void
117 usage(int ret)
118 {
119     arg_printusage (args, num_args, NULL, "[command]");
120     exit (ret);
121 }
122
123 int
124 get_privs(void *opt, int argc, char **argv)
125 {
126     uint32_t privs;
127     char str[128];
128     kadm5_ret_t ret;
129
130     ret = kadm5_get_privs(kadm_handle, &privs);
131     if(ret)
132         krb5_warn(context, ret, "kadm5_get_privs");
133     else{
134         ret =_kadm5_privs_to_string(privs, str, sizeof(str));
135         if (ret == 0)
136             printf("%s\n", str);
137         else
138             printf("privs: 0x%x\n", (unsigned int)privs);
139     }
140     return 0;
141 }
142
143 int
144 main(int argc, char **argv)
145 {
146     krb5_error_code ret;
147     char **files;
148     kadm5_config_params conf;
149     int optidx = 0;
150     int exit_status = 0;
151
152     setprogname(argv[0]);
153
154     ret = krb5_init_context(&context);
155     if (ret)
156         errx (1, "krb5_init_context failed: %d", ret);
157
158     if(getarg(args, num_args, argc, argv, &optidx))
159         usage(1);
160
161     if (help_flag)
162         usage (0);
163
164     if (version_flag) {
165         print_version(NULL);
166         exit(0);
167     }
168
169     argc -= optidx;
170     argv += optidx;
171
172     if (config_file == NULL) {
173         asprintf(&config_file, "%s/kdc.conf", hdb_db_dir(context));
174         if (config_file == NULL)
175             errx(1, "out of memory");
176     }
177
178     ret = krb5_prepend_config_files_default(config_file, &files);
179     if (ret)
180         krb5_err(context, 1, ret, "getting configuration files");
181
182     ret = krb5_set_config_files(context, files);
183     krb5_free_config_files(files);
184     if(ret)
185         krb5_err(context, 1, ret, "reading configuration files");
186
187     memset(&conf, 0, sizeof(conf));
188     if(realm) {
189         krb5_set_default_realm(context, realm); /* XXX should be fixed
190                                                    some other way */
191         conf.realm = realm;
192         conf.mask |= KADM5_CONFIG_REALM;
193     }
194
195     if (admin_server) {
196         conf.admin_server = admin_server;
197         conf.mask |= KADM5_CONFIG_ADMIN_SERVER;
198     }
199
200     if (server_port) {
201         conf.kadmind_port = htons(server_port);
202         conf.mask |= KADM5_CONFIG_KADMIND_PORT;
203     }
204
205     if (keyfile) {
206         conf.stash_file = keyfile;
207         conf.mask |= KADM5_CONFIG_STASH_FILE;
208     }
209
210     if(local_flag) {
211         int i;
212
213         kadm5_setup_passwd_quality_check (context,
214                                           check_library, check_function);
215         
216         for (i = 0; i < policy_libraries.num_strings; i++) {
217             ret = kadm5_add_passwd_quality_verifier(context,
218                                                     policy_libraries.strings[i]);
219             if (ret)
220                 krb5_err(context, 1, ret, "kadm5_add_passwd_quality_verifier");
221         }
222         ret = kadm5_add_passwd_quality_verifier(context, NULL);
223         if (ret)
224             krb5_err(context, 1, ret, "kadm5_add_passwd_quality_verifier");
225         
226         ret = kadm5_s_init_with_password_ctx(context,
227                                              KADM5_ADMIN_SERVICE,
228                                              NULL,
229                                              KADM5_ADMIN_SERVICE,
230                                              &conf, 0, 0,
231                                              &kadm_handle);
232     } else if (ad_flag) {
233         if (client_name == NULL)
234             krb5_errx(context, 1, "keytab mode require principal name");
235         ret = kadm5_ad_init_with_password_ctx(context,
236                                               client_name,
237                                               NULL,
238                                               KADM5_ADMIN_SERVICE,
239                                               &conf, 0, 0,
240                                               &kadm_handle);
241     } else if (keytab) {
242         if (client_name == NULL)
243             krb5_errx(context, 1, "keytab mode require principal name");
244         ret = kadm5_c_init_with_skey_ctx(context,
245                                          client_name,
246                                          keytab,
247                                          KADM5_ADMIN_SERVICE,
248                                          &conf, 0, 0,
249                                          &kadm_handle);
250     } else
251         ret = kadm5_c_init_with_password_ctx(context,
252                                              client_name,
253                                              NULL,
254                                              KADM5_ADMIN_SERVICE,
255                                              &conf, 0, 0,
256                                              &kadm_handle);
257
258     if(ret)
259         krb5_err(context, 1, ret, "kadm5_init_with_password");
260
261     signal(SIGINT, SIG_IGN); /* ignore signals for now, the sl command
262                                 parser will handle SIGINT its own way;
263                                 we should really take care of this in
264                                 each function, f.i `get' might be
265                                 interruptable, but not `create' */
266     if (argc != 0) {
267         ret = sl_command (commands, argc, argv);
268         if(ret == -1)
269             krb5_warnx (context, "unrecognized command: %s", argv[0]);
270         else if (ret == -2)
271             ret = 0;
272         if(ret != 0)
273             exit_status = 1;
274     } else {
275         while(!exit_seen) {
276             ret = sl_command_loop(commands, "kadmin> ", NULL);
277             if (ret == -2)
278                 exit_seen = 1;
279             else if (ret != 0)
280                 exit_status = 1;
281         }
282     }
283
284     kadm5_destroy(kadm_handle);
285     krb5_free_context(context);
286     return exit_status;
287 }