Annotation of src/usr.bin/ssh/ssh-pkcs11-client.c, Revision 1.4
1.4 ! djm 1: /* $OpenBSD: ssh-pkcs11-client.c,v 1.3 2012/01/16 20:34:09 miod Exp $ */
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);
1.4 ! djm 118: free(blob);
1.1 markus 119: send_msg(&msg);
1.3 miod 120: buffer_clear(&msg);
1.1 markus 121:
122: if (recv_msg(&msg) == SSH2_AGENT_SIGN_RESPONSE) {
123: signature = buffer_get_string(&msg, &slen);
124: if (slen <= (u_int)RSA_size(rsa)) {
125: memcpy(to, signature, slen);
126: ret = slen;
127: }
1.4 ! djm 128: free(signature);
1.1 markus 129: }
1.3 miod 130: buffer_free(&msg);
1.1 markus 131: return (ret);
132: }
133:
134: /* redirect the private key encrypt operation to the ssh-pkcs11-helper */
135: static int
136: wrap_key(RSA *rsa)
137: {
138: static RSA_METHOD helper_rsa;
139:
140: memcpy(&helper_rsa, RSA_get_default_method(), sizeof(helper_rsa));
141: helper_rsa.name = "ssh-pkcs11-helper";
142: helper_rsa.rsa_priv_enc = pkcs11_rsa_private_encrypt;
143: RSA_set_method(rsa, &helper_rsa);
144: return (0);
145: }
146:
147: static int
148: pkcs11_start_helper(void)
149: {
150: int pair[2];
151:
152: if (socketpair(AF_UNIX, SOCK_STREAM, 0, pair) == -1) {
153: error("socketpair: %s", strerror(errno));
154: return (-1);
155: }
156: if ((pid = fork()) == -1) {
157: error("fork: %s", strerror(errno));
158: return (-1);
159: } else if (pid == 0) {
160: if ((dup2(pair[1], STDIN_FILENO) == -1) ||
161: (dup2(pair[1], STDOUT_FILENO) == -1)) {
162: fprintf(stderr, "dup2: %s\n", strerror(errno));
163: _exit(1);
164: }
165: close(pair[0]);
166: close(pair[1]);
167: execlp(_PATH_SSH_PKCS11_HELPER, _PATH_SSH_PKCS11_HELPER,
168: (char *) 0);
169: fprintf(stderr, "exec: %s: %s\n", _PATH_SSH_PKCS11_HELPER,
170: strerror(errno));
171: _exit(1);
172: }
173: close(pair[1]);
174: fd = pair[0];
175: return (0);
176: }
177:
178: int
179: pkcs11_add_provider(char *name, char *pin, Key ***keysp)
180: {
181: Key *k;
182: int i, nkeys;
183: u_char *blob;
184: u_int blen;
185: Buffer msg;
186:
187: if (fd < 0 && pkcs11_start_helper() < 0)
188: return (-1);
189:
190: buffer_init(&msg);
191: buffer_put_char(&msg, SSH_AGENTC_ADD_SMARTCARD_KEY);
192: buffer_put_cstring(&msg, name);
193: buffer_put_cstring(&msg, pin);
194: send_msg(&msg);
195: buffer_clear(&msg);
196:
197: if (recv_msg(&msg) == SSH2_AGENT_IDENTITIES_ANSWER) {
198: nkeys = buffer_get_int(&msg);
199: *keysp = xcalloc(nkeys, sizeof(Key *));
200: for (i = 0; i < nkeys; i++) {
201: blob = buffer_get_string(&msg, &blen);
1.4 ! djm 202: free(buffer_get_string(&msg, NULL));
1.1 markus 203: k = key_from_blob(blob, blen);
204: wrap_key(k->rsa);
205: (*keysp)[i] = k;
1.4 ! djm 206: free(blob);
1.1 markus 207: }
208: } else {
209: nkeys = -1;
210: }
211: buffer_free(&msg);
212: return (nkeys);
213: }
214:
215: int
216: pkcs11_del_provider(char *name)
217: {
218: int ret = -1;
219: Buffer msg;
220:
221: buffer_init(&msg);
222: buffer_put_char(&msg, SSH_AGENTC_REMOVE_SMARTCARD_KEY);
223: buffer_put_cstring(&msg, name);
224: buffer_put_cstring(&msg, "");
225: send_msg(&msg);
226: buffer_clear(&msg);
227:
228: if (recv_msg(&msg) == SSH_AGENT_SUCCESS)
229: ret = 0;
230: buffer_free(&msg);
231: return (ret);
232: }