Annotation of src/usr.bin/ssh/ssh-pkcs11-client.c, Revision 1.2
1.2 ! djm 1: /* $OpenBSD$ */
1.1 markus 2: /*
3: * Copyright (c) 2010 Markus Friedl. All rights reserved.
4: *
5: * Permission to use, copy, modify, and distribute this software for any
6: * purpose with or without fee is hereby granted, provided that the above
7: * copyright notice and this permission notice appear in all copies.
8: *
9: * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
10: * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11: * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
12: * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13: * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14: * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
15: * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
16: */
17:
18: #include <sys/types.h>
19: #include <sys/time.h>
20: #include <sys/socket.h>
21:
22: #include <stdarg.h>
23: #include <string.h>
24: #include <unistd.h>
25: #include <errno.h>
26:
27: #include "pathnames.h"
28: #include "xmalloc.h"
29: #include "buffer.h"
30: #include "log.h"
31: #include "misc.h"
32: #include "key.h"
33: #include "authfd.h"
34: #include "atomicio.h"
35: #include "ssh-pkcs11.h"
36:
37: /* borrows code from sftp-server and ssh-agent */
38:
39: int fd = -1;
40: pid_t pid = -1;
41:
42: static void
43: send_msg(Buffer *m)
44: {
45: u_char buf[4];
46: int mlen = buffer_len(m);
47:
48: put_u32(buf, mlen);
49: if (atomicio(vwrite, fd, buf, 4) != 4 ||
50: atomicio(vwrite, fd, buffer_ptr(m),
51: buffer_len(m)) != buffer_len(m))
52: error("write to helper failed");
53: buffer_consume(m, mlen);
54: }
55:
56: static int
57: recv_msg(Buffer *m)
58: {
59: u_int l, len;
60: u_char buf[1024];
61:
62: if ((len = atomicio(read, fd, buf, 4)) != 4) {
63: error("read from helper failed: %u", len);
64: return (0); /* XXX */
65: }
66: len = get_u32(buf);
67: if (len > 256 * 1024)
68: fatal("response too long: %u", len);
69: /* read len bytes into m */
70: buffer_clear(m);
71: while (len > 0) {
72: l = len;
73: if (l > sizeof(buf))
74: l = sizeof(buf);
75: if (atomicio(read, fd, buf, l) != l) {
76: error("response from helper failed.");
77: return (0); /* XXX */
78: }
79: buffer_append(m, buf, l);
80: len -= l;
81: }
82: return (buffer_get_char(m));
83: }
84:
85: int
86: pkcs11_init(int interactive)
87: {
88: return (0);
89: }
90:
91: void
92: pkcs11_terminate(void)
93: {
94: close(fd);
95: }
96:
97: static int
98: pkcs11_rsa_private_encrypt(int flen, const u_char *from, u_char *to, RSA *rsa,
99: int padding)
100: {
101: Key key;
102: u_char *blob, *signature = NULL;
103: u_int blen, slen = 0;
104: int ret = -1;
105: Buffer msg;
106:
107: if (padding != RSA_PKCS1_PADDING)
108: return (-1);
109: key.type = KEY_RSA;
110: key.rsa = rsa;
111: if (key_to_blob(&key, &blob, &blen) == 0)
112: return -1;
113: buffer_init(&msg);
114: buffer_put_char(&msg, SSH2_AGENTC_SIGN_REQUEST);
115: buffer_put_string(&msg, blob, blen);
116: buffer_put_string(&msg, from, flen);
117: buffer_put_int(&msg, 0);
118: xfree(blob);
119: send_msg(&msg);
120:
121: if (recv_msg(&msg) == SSH2_AGENT_SIGN_RESPONSE) {
122: signature = buffer_get_string(&msg, &slen);
123: if (slen <= (u_int)RSA_size(rsa)) {
124: memcpy(to, signature, slen);
125: ret = slen;
126: }
127: xfree(signature);
128: }
129: return (ret);
130: }
131:
132: /* redirect the private key encrypt operation to the ssh-pkcs11-helper */
133: static int
134: wrap_key(RSA *rsa)
135: {
136: static RSA_METHOD helper_rsa;
137:
138: memcpy(&helper_rsa, RSA_get_default_method(), sizeof(helper_rsa));
139: helper_rsa.name = "ssh-pkcs11-helper";
140: helper_rsa.rsa_priv_enc = pkcs11_rsa_private_encrypt;
141: RSA_set_method(rsa, &helper_rsa);
142: return (0);
143: }
144:
145: static int
146: pkcs11_start_helper(void)
147: {
148: int pair[2];
149:
150: if (socketpair(AF_UNIX, SOCK_STREAM, 0, pair) == -1) {
151: error("socketpair: %s", strerror(errno));
152: return (-1);
153: }
154: if ((pid = fork()) == -1) {
155: error("fork: %s", strerror(errno));
156: return (-1);
157: } else if (pid == 0) {
158: if ((dup2(pair[1], STDIN_FILENO) == -1) ||
159: (dup2(pair[1], STDOUT_FILENO) == -1)) {
160: fprintf(stderr, "dup2: %s\n", strerror(errno));
161: _exit(1);
162: }
163: close(pair[0]);
164: close(pair[1]);
165: execlp(_PATH_SSH_PKCS11_HELPER, _PATH_SSH_PKCS11_HELPER,
166: (char *) 0);
167: fprintf(stderr, "exec: %s: %s\n", _PATH_SSH_PKCS11_HELPER,
168: strerror(errno));
169: _exit(1);
170: }
171: close(pair[1]);
172: fd = pair[0];
173: return (0);
174: }
175:
176: int
177: pkcs11_add_provider(char *name, char *pin, Key ***keysp)
178: {
179: Key *k;
180: int i, nkeys;
181: u_char *blob;
182: u_int blen;
183: Buffer msg;
184:
185: if (fd < 0 && pkcs11_start_helper() < 0)
186: return (-1);
187:
188: buffer_init(&msg);
189: buffer_put_char(&msg, SSH_AGENTC_ADD_SMARTCARD_KEY);
190: buffer_put_cstring(&msg, name);
191: buffer_put_cstring(&msg, pin);
192: send_msg(&msg);
193: buffer_clear(&msg);
194:
195: if (recv_msg(&msg) == SSH2_AGENT_IDENTITIES_ANSWER) {
196: nkeys = buffer_get_int(&msg);
197: *keysp = xcalloc(nkeys, sizeof(Key *));
198: for (i = 0; i < nkeys; i++) {
199: blob = buffer_get_string(&msg, &blen);
200: xfree(buffer_get_string(&msg, NULL));
201: k = key_from_blob(blob, blen);
202: wrap_key(k->rsa);
203: (*keysp)[i] = k;
204: xfree(blob);
205: }
206: } else {
207: nkeys = -1;
208: }
209: buffer_free(&msg);
210: return (nkeys);
211: }
212:
213: int
214: pkcs11_del_provider(char *name)
215: {
216: int ret = -1;
217: Buffer msg;
218:
219: buffer_init(&msg);
220: buffer_put_char(&msg, SSH_AGENTC_REMOVE_SMARTCARD_KEY);
221: buffer_put_cstring(&msg, name);
222: buffer_put_cstring(&msg, "");
223: send_msg(&msg);
224: buffer_clear(&msg);
225:
226: if (recv_msg(&msg) == SSH_AGENT_SUCCESS)
227: ret = 0;
228: buffer_free(&msg);
229: return (ret);
230: }