roken: Test time add/sub overflow prot.
[heimdal.git] / kadmin / stash.c
blobc33623038ce4452f14404782abad8deb730f4919
1 /*
2 * Copyright (c) 2004 Kungliga Tekniska Högskolan
3 * (Royal Institute of Technology, Stockholm, Sweden).
4 * All rights reserved.
6 * Portions Copyright (c) 2009 Apple Inc. All rights reserved.
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
10 * are met:
12 * 1. Redistributions of source code must retain the above copyright
13 * notice, this list of conditions and the following disclaimer.
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.
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.
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.
36 #include "kadmin_locl.h"
37 #include "kadmin-commands.h"
39 extern int local_flag;
41 int
42 stash(struct stash_options *opt, int argc, char **argv)
44 char buf[1024+1];
45 krb5_error_code ret;
46 krb5_enctype enctype;
47 hdb_master_key mkey;
48 int aret;
50 if(!local_flag) {
51 krb5_warnx(context, "stash is only available in local (-l) mode");
52 return 0;
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;
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");
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;
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;
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;
108 ret = krb5_string_to_key_salt(context, enctype, buf, salt, &key);
109 if (ret == 0)
110 ret = hdb_add_master_key(context, &key, &mkey);
111 if (ret)
112 krb5_warn(context, errno, "setting master key");
113 krb5_free_keyblock_contents(context, &key);
117 char *new = NULL, *old = NULL;
119 aret = asprintf(&old, "%s.old", opt->key_file_string);
120 if (aret == -1) {
121 ret = ENOMEM;
122 goto out;
124 aret = asprintf(&new, "%s.new", opt->key_file_string);
125 if (aret == -1) {
126 ret = ENOMEM;
127 goto out;
130 if(unlink(new) < 0 && errno != ENOENT) {
131 ret = errno;
132 goto out;
134 krb5_warnx(context, "writing key to \"%s\"", opt->key_file_string);
135 ret = hdb_write_master_key(context, new, mkey);
136 if(ret)
137 unlink(new);
138 else {
139 unlink(old);
140 #ifndef NO_POSIX_LINKS
141 if(link(opt->key_file_string, old) < 0 && errno != ENOENT) {
142 ret = errno;
143 unlink(new);
144 } else {
145 #endif
146 if(rename(new, opt->key_file_string) < 0) {
147 ret = errno;
149 #ifndef NO_POSIX_LINKS
151 #endif
153 out:
154 free(old);
155 free(new);
156 if(ret)
157 krb5_warn(context, errno, "writing master key file");
160 hdb_free_master_key(context, mkey);
161 return 0;