[BACK]Return to ssh-pkcs11-client.c CVS log [TXT][DIR] Up to [local] / src / usr.bin / ssh

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: }