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

Annotation of src/usr.bin/ssh/authfd.c, Revision 1.114

1.114   ! djm         1: /* $OpenBSD: authfd.c,v 1.113 2018/12/27 23:02:11 djm Exp $ */
1.1       deraadt     2: /*
1.13      deraadt     3:  * Author: Tatu Ylonen <ylo@cs.hut.fi>
                      4:  * Copyright (c) 1995 Tatu Ylonen <ylo@cs.hut.fi>, Espoo, Finland
                      5:  *                    All rights reserved
1.27      deraadt     6:  * Functions for connecting the local authentication agent.
1.18      markus      7:  *
1.27      deraadt     8:  * As far as I am concerned, the code I have written for this software
                      9:  * can be used freely for any purpose.  Any derived versions of this
                     10:  * software must be clearly marked as such, and if the derived work is
                     11:  * incompatible with the protocol description in the RFC file, it must be
                     12:  * called by a name other than "ssh" or "Secure Shell".
1.18      markus     13:  *
1.25      markus     14:  * SSH2 implementation,
1.27      deraadt    15:  * Copyright (c) 2000 Markus Friedl.  All rights reserved.
1.25      markus     16:  *
1.27      deraadt    17:  * Redistribution and use in source and binary forms, with or without
                     18:  * modification, are permitted provided that the following conditions
                     19:  * are met:
                     20:  * 1. Redistributions of source code must retain the above copyright
                     21:  *    notice, this list of conditions and the following disclaimer.
                     22:  * 2. Redistributions in binary form must reproduce the above copyright
                     23:  *    notice, this list of conditions and the following disclaimer in the
                     24:  *    documentation and/or other materials provided with the distribution.
                     25:  *
                     26:  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
                     27:  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
                     28:  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
                     29:  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
                     30:  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
                     31:  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
                     32:  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
                     33:  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
                     34:  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
                     35:  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
1.13      deraadt    36:  */
1.1       deraadt    37:
1.67      stevesk    38:
                     39: #include <sys/types.h>
                     40: #include <sys/un.h>
1.75      stevesk    41: #include <sys/socket.h>
1.33      markus     42:
1.76      stevesk    43: #include <fcntl.h>
1.79      stevesk    44: #include <stdlib.h>
1.80      deraadt    45: #include <signal.h>
1.78      stevesk    46: #include <string.h>
1.77      stevesk    47: #include <unistd.h>
1.94      djm        48: #include <errno.h>
1.1       deraadt    49:
1.80      deraadt    50: #include "xmalloc.h"
1.1       deraadt    51: #include "ssh.h"
1.94      djm        52: #include "sshbuf.h"
                     53: #include "sshkey.h"
1.22      markus     54: #include "authfd.h"
1.33      markus     55: #include "cipher.h"
1.28      markus     56: #include "compat.h"
1.33      markus     57: #include "log.h"
                     58: #include "atomicio.h"
1.74      djm        59: #include "misc.h"
1.94      djm        60: #include "ssherr.h"
1.2       provos     61:
1.94      djm        62: #define MAX_AGENT_IDENTITIES   2048            /* Max keys in agent reply */
                     63: #define MAX_AGENT_REPLY_LEN    (256 * 1024)    /* Max bytes in agent reply */
1.21      markus     64:
1.29      markus     65: /* macro to check for "agent failure" message */
                     66: #define agent_failed(x) \
1.94      djm        67:     ((x == SSH_AGENT_FAILURE) || \
                     68:     (x == SSH_COM_AGENT2_FAILURE) || \
1.55      deraadt    69:     (x == SSH2_AGENT_FAILURE))
1.29      markus     70:
1.94      djm        71: /* Convert success/failure response from agent to a err.h status */
                     72: static int
                     73: decode_reply(u_char type)
1.57      stevesk    74: {
1.94      djm        75:        if (agent_failed(type))
                     76:                return SSH_ERR_AGENT_FAILURE;
                     77:        else if (type == SSH_AGENT_SUCCESS)
1.57      stevesk    78:                return 0;
1.94      djm        79:        else
                     80:                return SSH_ERR_INVALID_FORMAT;
1.57      stevesk    81: }
                     82:
1.1       deraadt    83: /* Returns the number of the authentication fd, or -1 if there is none. */
1.2       provos     84: int
1.94      djm        85: ssh_get_authentication_socket(int *fdp)
1.1       deraadt    86: {
1.12      markus     87:        const char *authsocket;
1.94      djm        88:        int sock, oerrno;
1.12      markus     89:        struct sockaddr_un sunaddr;
                     90:
1.94      djm        91:        if (fdp != NULL)
                     92:                *fdp = -1;
                     93:
1.12      markus     94:        authsocket = getenv(SSH_AUTHSOCKET_ENV_NAME);
1.112     djm        95:        if (authsocket == NULL || *authsocket == '\0')
1.94      djm        96:                return SSH_ERR_AGENT_NOT_PRESENT;
1.12      markus     97:
1.92      tedu       98:        memset(&sunaddr, 0, sizeof(sunaddr));
1.12      markus     99:        sunaddr.sun_family = AF_UNIX;
                    100:        strlcpy(sunaddr.sun_path, authsocket, sizeof(sunaddr.sun_path));
                    101:
1.94      djm       102:        if ((sock = socket(AF_UNIX, SOCK_STREAM, 0)) < 0)
                    103:                return SSH_ERR_SYSTEM_ERROR;
1.12      markus    104:
                    105:        /* close on exec */
1.94      djm       106:        if (fcntl(sock, F_SETFD, FD_CLOEXEC) == -1 ||
                    107:            connect(sock, (struct sockaddr *)&sunaddr, sizeof(sunaddr)) < 0) {
                    108:                oerrno = errno;
1.12      markus    109:                close(sock);
1.94      djm       110:                errno = oerrno;
                    111:                return SSH_ERR_SYSTEM_ERROR;
1.12      markus    112:        }
1.94      djm       113:        if (fdp != NULL)
                    114:                *fdp = sock;
                    115:        else
1.12      markus    116:                close(sock);
1.94      djm       117:        return 0;
1.1       deraadt   118: }
                    119:
1.94      djm       120: /* Communicate with agent: send request and read reply */
1.41      itojun    121: static int
1.94      djm       122: ssh_request_reply(int sock, struct sshbuf *request, struct sshbuf *reply)
1.24      markus    123: {
1.94      djm       124:        int r;
                    125:        size_t l, len;
1.24      markus    126:        char buf[1024];
                    127:
                    128:        /* Get the length of the message, and format it in the buffer. */
1.94      djm       129:        len = sshbuf_len(request);
1.108     markus    130:        POKE_U32(buf, len);
1.24      markus    131:
                    132:        /* Send the length and then the packet to the agent. */
1.94      djm       133:        if (atomicio(vwrite, sock, buf, 4) != 4 ||
1.111     markus    134:            atomicio(vwrite, sock, sshbuf_mutable_ptr(request),
1.94      djm       135:            sshbuf_len(request)) != sshbuf_len(request))
                    136:                return SSH_ERR_AGENT_COMMUNICATION;
1.24      markus    137:        /*
                    138:         * Wait for response from the agent.  First read the length of the
                    139:         * response packet.
                    140:         */
1.94      djm       141:        if (atomicio(read, sock, buf, 4) != 4)
                    142:            return SSH_ERR_AGENT_COMMUNICATION;
1.24      markus    143:
                    144:        /* Extract the length, and check it for sanity. */
1.108     markus    145:        len = PEEK_U32(buf);
1.94      djm       146:        if (len > MAX_AGENT_REPLY_LEN)
                    147:                return SSH_ERR_INVALID_FORMAT;
1.24      markus    148:
                    149:        /* Read the rest of the response in to the buffer. */
1.94      djm       150:        sshbuf_reset(reply);
1.24      markus    151:        while (len > 0) {
                    152:                l = len;
                    153:                if (l > sizeof(buf))
                    154:                        l = sizeof(buf);
1.94      djm       155:                if (atomicio(read, sock, buf, l) != l)
                    156:                        return SSH_ERR_AGENT_COMMUNICATION;
                    157:                if ((r = sshbuf_put(reply, buf, l)) != 0)
                    158:                        return r;
1.24      markus    159:                len -= l;
                    160:        }
1.94      djm       161:        return 0;
1.24      markus    162: }
                    163:
1.14      markus    164: /*
                    165:  * Closes the agent socket if it should be closed (depends on how it was
                    166:  * obtained).  The argument must have been returned by
                    167:  * ssh_get_authentication_socket().
                    168:  */
1.18      markus    169: void
1.12      markus    170: ssh_close_authentication_socket(int sock)
1.1       deraadt   171: {
1.12      markus    172:        if (getenv(SSH_AUTHSOCKET_ENV_NAME))
                    173:                close(sock);
1.1       deraadt   174: }
                    175:
1.94      djm       176: /* Lock/unlock agent */
                    177: int
                    178: ssh_lock_agent(int sock, int lock, const char *password)
1.1       deraadt   179: {
1.94      djm       180:        int r;
                    181:        u_char type = lock ? SSH_AGENTC_LOCK : SSH_AGENTC_UNLOCK;
                    182:        struct sshbuf *msg;
                    183:
                    184:        if ((msg = sshbuf_new()) == NULL)
                    185:                return SSH_ERR_ALLOC_FAIL;
                    186:        if ((r = sshbuf_put_u8(msg, type)) != 0 ||
                    187:            (r = sshbuf_put_cstring(msg, password)) != 0)
                    188:                goto out;
                    189:        if ((r = ssh_request_reply(sock, msg, msg)) != 0)
                    190:                goto out;
                    191:        if ((r = sshbuf_get_u8(msg, &type)) != 0)
                    192:                goto out;
                    193:        r = decode_reply(type);
                    194:  out:
                    195:        sshbuf_free(msg);
                    196:        return r;
1.1       deraadt   197: }
                    198:
1.50      markus    199:
1.94      djm       200: static int
                    201: deserialise_identity2(struct sshbuf *ids, struct sshkey **keyp, char **commentp)
1.50      markus    202: {
1.94      djm       203:        int r;
                    204:        char *comment = NULL;
                    205:        const u_char *blob;
                    206:        size_t blen;
                    207:
                    208:        if ((r = sshbuf_get_string_direct(ids, &blob, &blen)) != 0 ||
                    209:            (r = sshbuf_get_cstring(ids, &comment, NULL)) != 0)
                    210:                goto out;
                    211:        if ((r = sshkey_from_blob(blob, blen, keyp)) != 0)
                    212:                goto out;
                    213:        if (commentp != NULL) {
                    214:                *commentp = comment;
                    215:                comment = NULL;
                    216:        }
                    217:        r = 0;
                    218:  out:
                    219:        free(comment);
                    220:        return r;
1.1       deraadt   221: }
                    222:
1.14      markus    223: /*
1.94      djm       224:  * Fetch list of identities held by the agent.
1.14      markus    225:  */
1.30      markus    226: int
1.103     naddy     227: ssh_fetch_identitylist(int sock, struct ssh_identitylist **idlp)
1.1       deraadt   228: {
1.103     naddy     229:        u_char type;
1.94      djm       230:        u_int32_t num, i;
                    231:        struct sshbuf *msg;
                    232:        struct ssh_identitylist *idl = NULL;
                    233:        int r;
1.25      markus    234:
1.14      markus    235:        /*
                    236:         * Send a message to the agent requesting for a list of the
                    237:         * identities it can represent.
                    238:         */
1.94      djm       239:        if ((msg = sshbuf_new()) == NULL)
                    240:                return SSH_ERR_ALLOC_FAIL;
1.103     naddy     241:        if ((r = sshbuf_put_u8(msg, SSH2_AGENTC_REQUEST_IDENTITIES)) != 0)
1.94      djm       242:                goto out;
1.12      markus    243:
1.94      djm       244:        if ((r = ssh_request_reply(sock, msg, msg)) != 0)
                    245:                goto out;
1.1       deraadt   246:
1.12      markus    247:        /* Get message type, and verify that we got a proper answer. */
1.94      djm       248:        if ((r = sshbuf_get_u8(msg, &type)) != 0)
                    249:                goto out;
1.29      markus    250:        if (agent_failed(type)) {
1.94      djm       251:                r = SSH_ERR_AGENT_FAILURE;
                    252:                goto out;
1.103     naddy     253:        } else if (type != SSH2_AGENT_IDENTITIES_ANSWER) {
1.94      djm       254:                r = SSH_ERR_INVALID_FORMAT;
                    255:                goto out;
1.25      markus    256:        }
1.12      markus    257:
                    258:        /* Get the number of entries in the response and check it for sanity. */
1.94      djm       259:        if ((r = sshbuf_get_u32(msg, &num)) != 0)
                    260:                goto out;
                    261:        if (num > MAX_AGENT_IDENTITIES) {
                    262:                r = SSH_ERR_INVALID_FORMAT;
                    263:                goto out;
                    264:        }
                    265:        if (num == 0) {
                    266:                r = SSH_ERR_AGENT_NO_IDENTITIES;
                    267:                goto out;
                    268:        }
                    269:
                    270:        /* Deserialise the response into a list of keys/comments */
                    271:        if ((idl = calloc(1, sizeof(*idl))) == NULL ||
                    272:            (idl->keys = calloc(num, sizeof(*idl->keys))) == NULL ||
                    273:            (idl->comments = calloc(num, sizeof(*idl->comments))) == NULL) {
                    274:                r = SSH_ERR_ALLOC_FAIL;
                    275:                goto out;
                    276:        }
                    277:        for (i = 0; i < num;) {
1.103     naddy     278:                if ((r = deserialise_identity2(msg, &(idl->keys[i]),
                    279:                    &(idl->comments[i]))) != 0) {
                    280:                        if (r == SSH_ERR_KEY_TYPE_UNKNOWN) {
                    281:                                /* Gracefully skip unknown key types */
                    282:                                num--;
                    283:                                continue;
                    284:                        } else
                    285:                                goto out;
1.94      djm       286:                }
                    287:                i++;
                    288:        }
                    289:        idl->nkeys = num;
                    290:        *idlp = idl;
                    291:        idl = NULL;
                    292:        r = 0;
                    293:  out:
                    294:        sshbuf_free(msg);
                    295:        if (idl != NULL)
                    296:                ssh_free_identitylist(idl);
                    297:        return r;
1.30      markus    298: }
                    299:
1.94      djm       300: void
                    301: ssh_free_identitylist(struct ssh_identitylist *idl)
1.30      markus    302: {
1.94      djm       303:        size_t i;
1.1       deraadt   304:
1.94      djm       305:        if (idl == NULL)
                    306:                return;
                    307:        for (i = 0; i < idl->nkeys; i++) {
                    308:                if (idl->keys != NULL)
                    309:                        sshkey_free(idl->keys[i]);
                    310:                if (idl->comments != NULL)
                    311:                        free(idl->comments[i]);
1.25      markus    312:        }
1.94      djm       313:        free(idl);
1.1       deraadt   314: }
                    315:
1.14      markus    316: /*
1.94      djm       317:  * Sends a challenge (typically from a server via ssh(1)) to the agent,
                    318:  * and waits for a response from the agent.
                    319:  * Returns true (non-zero) if the agent gave the correct answer, zero
                    320:  * otherwise.
1.14      markus    321:  */
1.1       deraadt   322:
                    323:
1.109     djm       324: /* encode signature algorithm in flag bits, so we can keep the msg format */
1.100     markus    325: static u_int
1.104     djm       326: agent_encode_alg(const struct sshkey *key, const char *alg)
1.100     markus    327: {
1.113     djm       328:        if (alg != NULL && sshkey_type_plain(key->type) == KEY_RSA) {
                    329:                if (strcmp(alg, "rsa-sha2-256") == 0 ||
                    330:                    strcmp(alg, "rsa-sha2-256-cert-v01@openssh.com") == 0)
1.100     markus    331:                        return SSH_AGENT_RSA_SHA2_256;
1.113     djm       332:                if (strcmp(alg, "rsa-sha2-512") == 0 ||
                    333:                    strcmp(alg, "rsa-sha2-512-cert-v01@openssh.com") == 0)
1.100     markus    334:                        return SSH_AGENT_RSA_SHA2_512;
                    335:        }
                    336:        return 0;
                    337: }
                    338:
1.94      djm       339: /* ask agent to sign data, returns err.h code on error, 0 on success */
1.25      markus    340: int
1.104     djm       341: ssh_agent_sign(int sock, const struct sshkey *key,
1.94      djm       342:     u_char **sigp, size_t *lenp,
1.100     markus    343:     const u_char *data, size_t datalen, const char *alg, u_int compat)
1.94      djm       344: {
                    345:        struct sshbuf *msg;
1.110     djm       346:        u_char *sig = NULL, type = 0;
                    347:        size_t len = 0;
1.94      djm       348:        u_int flags = 0;
                    349:        int r = SSH_ERR_INTERNAL_ERROR;
                    350:
1.97      markus    351:        *sigp = NULL;
                    352:        *lenp = 0;
1.94      djm       353:
                    354:        if (datalen > SSH_KEY_MAX_SIGN_DATA_SIZE)
                    355:                return SSH_ERR_INVALID_ARGUMENT;
                    356:        if ((msg = sshbuf_new()) == NULL)
                    357:                return SSH_ERR_ALLOC_FAIL;
1.100     markus    358:        flags |= agent_encode_alg(key, alg);
1.94      djm       359:        if ((r = sshbuf_put_u8(msg, SSH2_AGENTC_SIGN_REQUEST)) != 0 ||
1.110     djm       360:            (r = sshkey_puts(key, msg)) != 0 ||
1.94      djm       361:            (r = sshbuf_put_string(msg, data, datalen)) != 0 ||
                    362:            (r = sshbuf_put_u32(msg, flags)) != 0)
                    363:                goto out;
1.99      jsg       364:        if ((r = ssh_request_reply(sock, msg, msg)) != 0)
1.94      djm       365:                goto out;
                    366:        if ((r = sshbuf_get_u8(msg, &type)) != 0)
                    367:                goto out;
1.29      markus    368:        if (agent_failed(type)) {
1.94      djm       369:                r = SSH_ERR_AGENT_FAILURE;
                    370:                goto out;
1.25      markus    371:        } else if (type != SSH2_AGENT_SIGN_RESPONSE) {
1.94      djm       372:                r = SSH_ERR_INVALID_FORMAT;
                    373:                goto out;
                    374:        }
1.110     djm       375:        if ((r = sshbuf_get_string(msg, &sig, &len)) != 0)
                    376:                goto out;
                    377:        /* Check what we actually got back from the agent. */
                    378:        if ((r = sshkey_check_sigtype(sig, len, alg)) != 0)
1.94      djm       379:                goto out;
1.110     djm       380:        /* success */
                    381:        *sigp = sig;
1.97      markus    382:        *lenp = len;
1.110     djm       383:        sig = NULL;
                    384:        len = 0;
1.94      djm       385:        r = 0;
                    386:  out:
1.110     djm       387:        freezero(sig, len);
1.94      djm       388:        sshbuf_free(msg);
                    389:        return r;
1.25      markus    390: }
                    391:
1.22      markus    392: /* Encode key for a message to the agent. */
                    393:
                    394:
1.94      djm       395: static int
1.108     markus    396: encode_constraints(struct sshbuf *m, u_int life, u_int confirm, u_int maxsign)
1.94      djm       397: {
                    398:        int r;
                    399:
                    400:        if (life != 0) {
                    401:                if ((r = sshbuf_put_u8(m, SSH_AGENT_CONSTRAIN_LIFETIME)) != 0 ||
                    402:                    (r = sshbuf_put_u32(m, life)) != 0)
                    403:                        goto out;
                    404:        }
                    405:        if (confirm != 0) {
                    406:                if ((r = sshbuf_put_u8(m, SSH_AGENT_CONSTRAIN_CONFIRM)) != 0)
                    407:                        goto out;
                    408:        }
1.108     markus    409:        if (maxsign != 0) {
                    410:                if ((r = sshbuf_put_u8(m, SSH_AGENT_CONSTRAIN_MAXSIGN)) != 0 ||
                    411:                    (r = sshbuf_put_u32(m, maxsign)) != 0)
                    412:                        goto out;
                    413:        }
1.94      djm       414:        r = 0;
                    415:  out:
                    416:        return r;
1.22      markus    417: }
                    418:
1.14      markus    419: /*
1.94      djm       420:  * Adds an identity to the authentication server.
                    421:  * This call is intended only for use by ssh-add(1) and like applications.
1.14      markus    422:  */
1.18      markus    423: int
1.114   ! djm       424: ssh_add_identity_constrained(int sock, struct sshkey *key,
1.108     markus    425:     const char *comment, u_int life, u_int confirm, u_int maxsign)
1.1       deraadt   426: {
1.94      djm       427:        struct sshbuf *msg;
1.108     markus    428:        int r, constrained = (life || confirm || maxsign);
1.94      djm       429:        u_char type;
1.12      markus    430:
1.94      djm       431:        if ((msg = sshbuf_new()) == NULL)
                    432:                return SSH_ERR_ALLOC_FAIL;
1.22      markus    433:
                    434:        switch (key->type) {
1.93      markus    435: #ifdef WITH_OPENSSL
1.22      markus    436:        case KEY_RSA:
1.82      djm       437:        case KEY_RSA_CERT:
1.22      markus    438:        case KEY_DSA:
1.82      djm       439:        case KEY_DSA_CERT:
1.84      djm       440:        case KEY_ECDSA:
                    441:        case KEY_ECDSA_CERT:
1.93      markus    442: #endif
1.90      markus    443:        case KEY_ED25519:
                    444:        case KEY_ED25519_CERT:
1.108     markus    445:        case KEY_XMSS:
                    446:        case KEY_XMSS_CERT:
1.54      markus    447:                type = constrained ?
                    448:                    SSH2_AGENTC_ADD_ID_CONSTRAINED :
                    449:                    SSH2_AGENTC_ADD_IDENTITY;
1.94      djm       450:                if ((r = sshbuf_put_u8(msg, type)) != 0 ||
1.108     markus    451:                    (r = sshkey_private_serialize_maxsign(key, msg, maxsign,
                    452:                    NULL)) != 0 ||
                    453:                    (r = sshbuf_put_cstring(msg, comment)) != 0)
1.94      djm       454:                        goto out;
1.22      markus    455:                break;
                    456:        default:
1.94      djm       457:                r = SSH_ERR_INVALID_ARGUMENT;
                    458:                goto out;
1.12      markus    459:        }
1.94      djm       460:        if (constrained &&
1.108     markus    461:            (r = encode_constraints(msg, life, confirm, maxsign)) != 0)
1.94      djm       462:                goto out;
                    463:        if ((r = ssh_request_reply(sock, msg, msg)) != 0)
                    464:                goto out;
                    465:        if ((r = sshbuf_get_u8(msg, &type)) != 0)
                    466:                goto out;
                    467:        r = decode_reply(type);
                    468:  out:
                    469:        sshbuf_free(msg);
                    470:        return r;
1.54      markus    471: }
                    472:
1.14      markus    473: /*
1.94      djm       474:  * Removes an identity from the authentication server.
                    475:  * This call is intended only for use by ssh-add(1) and like applications.
1.14      markus    476:  */
1.18      markus    477: int
1.94      djm       478: ssh_remove_identity(int sock, struct sshkey *key)
1.1       deraadt   479: {
1.94      djm       480:        struct sshbuf *msg;
                    481:        int r;
                    482:        u_char type, *blob = NULL;
                    483:        size_t blen;
1.12      markus    484:
1.94      djm       485:        if ((msg = sshbuf_new()) == NULL)
                    486:                return SSH_ERR_ALLOC_FAIL;
1.12      markus    487:
1.93      markus    488:        if (key->type != KEY_UNSPEC) {
1.94      djm       489:                if ((r = sshkey_to_blob(key, &blob, &blen)) != 0)
                    490:                        goto out;
                    491:                if ((r = sshbuf_put_u8(msg,
                    492:                    SSH2_AGENTC_REMOVE_IDENTITY)) != 0 ||
                    493:                    (r = sshbuf_put_string(msg, blob, blen)) != 0)
                    494:                        goto out;
1.51      markus    495:        } else {
1.94      djm       496:                r = SSH_ERR_INVALID_ARGUMENT;
                    497:                goto out;
1.51      markus    498:        }
1.94      djm       499:        if ((r = ssh_request_reply(sock, msg, msg)) != 0)
                    500:                goto out;
                    501:        if ((r = sshbuf_get_u8(msg, &type)) != 0)
                    502:                goto out;
                    503:        r = decode_reply(type);
                    504:  out:
                    505:        if (blob != NULL) {
                    506:                explicit_bzero(blob, blen);
                    507:                free(blob);
1.42      markus    508:        }
1.94      djm       509:        sshbuf_free(msg);
                    510:        return r;
1.42      markus    511: }
                    512:
1.94      djm       513: /*
                    514:  * Add/remove an token-based identity from the authentication server.
                    515:  * This call is intended only for use by ssh-add(1) and like applications.
                    516:  */
1.42      markus    517: int
1.94      djm       518: ssh_update_card(int sock, int add, const char *reader_id, const char *pin,
                    519:     u_int life, u_int confirm)
1.42      markus    520: {
1.94      djm       521:        struct sshbuf *msg;
                    522:        int r, constrained = (life || confirm);
                    523:        u_char type;
1.60      djm       524:
                    525:        if (add) {
                    526:                type = constrained ?
                    527:                    SSH_AGENTC_ADD_SMARTCARD_KEY_CONSTRAINED :
                    528:                    SSH_AGENTC_ADD_SMARTCARD_KEY;
                    529:        } else
                    530:                type = SSH_AGENTC_REMOVE_SMARTCARD_KEY;
1.42      markus    531:
1.94      djm       532:        if ((msg = sshbuf_new()) == NULL)
                    533:                return SSH_ERR_ALLOC_FAIL;
                    534:        if ((r = sshbuf_put_u8(msg, type)) != 0 ||
                    535:            (r = sshbuf_put_cstring(msg, reader_id)) != 0 ||
                    536:            (r = sshbuf_put_cstring(msg, pin)) != 0)
                    537:                goto out;
                    538:        if (constrained &&
1.108     markus    539:            (r = encode_constraints(msg, life, confirm, 0)) != 0)
1.94      djm       540:                goto out;
                    541:        if ((r = ssh_request_reply(sock, msg, msg)) != 0)
                    542:                goto out;
                    543:        if ((r = sshbuf_get_u8(msg, &type)) != 0)
                    544:                goto out;
                    545:        r = decode_reply(type);
                    546:  out:
                    547:        sshbuf_free(msg);
                    548:        return r;
1.12      markus    549: }
                    550:
1.14      markus    551: /*
1.94      djm       552:  * Removes all identities from the agent.
                    553:  * This call is intended only for use by ssh-add(1) and like applications.
1.102     djm       554:  *
                    555:  * This supports the SSH protocol 1 message to because, when clearing all
                    556:  * keys from an agent, we generally want to clear both protocol v1 and v2
                    557:  * keys.
1.14      markus    558:  */
1.18      markus    559: int
1.94      djm       560: ssh_remove_all_identities(int sock, int version)
1.1       deraadt   561: {
1.94      djm       562:        struct sshbuf *msg;
                    563:        u_char type = (version == 1) ?
                    564:            SSH_AGENTC_REMOVE_ALL_RSA_IDENTITIES :
                    565:            SSH2_AGENTC_REMOVE_ALL_IDENTITIES;
                    566:        int r;
                    567:
                    568:        if ((msg = sshbuf_new()) == NULL)
                    569:                return SSH_ERR_ALLOC_FAIL;
                    570:        if ((r = sshbuf_put_u8(msg, type)) != 0)
                    571:                goto out;
                    572:        if ((r = ssh_request_reply(sock, msg, msg)) != 0)
                    573:                goto out;
                    574:        if ((r = sshbuf_get_u8(msg, &type)) != 0)
                    575:                goto out;
                    576:        r = decode_reply(type);
                    577:  out:
                    578:        sshbuf_free(msg);
                    579:        return r;
1.12      markus    580: }