s4:torture: Adapt KDC canon test to Heimdal upstream changes
[samba.git] / third_party / heimdal / kadmin / stash.c
1 /*
2  * Copyright (c) 2004 Kungliga Tekniska Högskolan
3  * (Royal Institute of Technology, Stockholm, Sweden).
4  * All rights reserved.
5  *
6  * Portions Copyright (c) 2009 Apple Inc. 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 #include "kadmin-commands.h"
38
39 extern int local_flag;
40
41 int
42 stash(struct stash_options *opt, int argc, char **argv)
43 {
44     char buf[1024+1];
45     krb5_error_code ret;
46     krb5_enctype enctype;
47     hdb_master_key mkey;
48     int aret;
49
50     if(!local_flag) {
51         krb5_warnx(context, "stash is only available in local (-l) mode");
52         return 0;
53     }
54
55     ret = krb5_string_to_enctype(context, opt->enctype_string, &enctype);
56     if(ret) {
57         krb5_warn(context, ret, "%s", opt->enctype_string);
58         return 0;
59     }
60
61     if(opt->key_file_string == NULL) {
62         aret = asprintf(&opt->key_file_string, "%s/m-key", hdb_db_dir(context));
63         if (aret == -1)
64             errx(1, "out of memory");
65     }
66
67     ret = hdb_read_master_key(context, opt->key_file_string, &mkey);
68     if(ret && ret != ENOENT) {
69         krb5_warn(context, ret, "reading master key from %s",
70                   opt->key_file_string);
71         return 0;
72     }
73
74     if (opt->convert_file_flag) {
75         if (ret)
76             krb5_warn(context, ret, "reading master key from %s",
77                       opt->key_file_string);
78         hdb_free_master_key(context, mkey);
79         return 0;
80     } else {
81         krb5_keyblock key;
82         krb5_salt salt;
83         salt.salttype = KRB5_PW_SALT;
84         /* XXX better value? */
85         salt.saltvalue.data = NULL;
86         salt.saltvalue.length = 0;
87         if(opt->master_key_fd_integer != -1) {
88             ssize_t n;
89             n = read(opt->master_key_fd_integer, buf, sizeof(buf)-1);
90             if(n == 0)
91                 krb5_warnx(context, "end of file reading passphrase");
92             else if(n < 0) {
93                 krb5_warn(context, errno, "reading passphrase");
94                 n = 0;
95             }
96             buf[n] = '\0';
97             buf[strcspn(buf, "\r\n")] = '\0';
98         } else if (opt->random_password_flag) {
99             random_password (buf, sizeof(buf));
100             printf("Using random master stash password: %s\n", buf);
101         } else {
102             if(UI_UTIL_read_pw_string(buf, sizeof(buf), "Master key: ",
103                                       UI_UTIL_FLAG_VERIFY)) {
104                 hdb_free_master_key(context, mkey);
105                 return 0;
106             }
107         }
108         ret = krb5_string_to_key_salt(context, enctype, buf, salt, &key);
109         ret = hdb_add_master_key(context, &key, &mkey);
110         krb5_free_keyblock_contents(context, &key);
111     }
112
113     {
114         char *new = NULL, *old = NULL;
115
116         aret = asprintf(&old, "%s.old", opt->key_file_string);
117         if (aret == -1) {
118             ret = ENOMEM;
119             goto out;
120         }
121         aret = asprintf(&new, "%s.new", opt->key_file_string);
122         if (aret == -1) {
123             ret = ENOMEM;
124             goto out;
125         }
126
127         if(unlink(new) < 0 && errno != ENOENT) {
128             ret = errno;
129             goto out;
130         }
131         krb5_warnx(context, "writing key to \"%s\"", opt->key_file_string);
132         ret = hdb_write_master_key(context, new, mkey);
133         if(ret)
134             unlink(new);
135         else {
136             unlink(old);
137 #ifndef NO_POSIX_LINKS
138             if(link(opt->key_file_string, old) < 0 && errno != ENOENT) {
139                 ret = errno;
140                 unlink(new);
141             } else {
142 #endif
143                 if(rename(new, opt->key_file_string) < 0) {
144                     ret = errno;
145                 }
146 #ifndef NO_POSIX_LINKS
147             }
148 #endif
149         }
150     out:
151         free(old);
152         free(new);
153         if(ret)
154             krb5_warn(context, errno, "writing master key file");
155     }
156
157     hdb_free_master_key(context, mkey);
158     return 0;
159 }