[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.82

1.82    ! djm         1: /* $OpenBSD: authfd.c,v 1.81 2009/08/27 17:44:52 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:
                     43: #include <openssl/evp.h>
1.76      stevesk    44:
1.80      deraadt    45: #include <openssl/crypto.h>
1.76      stevesk    46: #include <fcntl.h>
1.79      stevesk    47: #include <stdlib.h>
1.80      deraadt    48: #include <signal.h>
1.78      stevesk    49: #include <string.h>
1.77      stevesk    50: #include <unistd.h>
1.1       deraadt    51:
1.80      deraadt    52: #include "xmalloc.h"
1.1       deraadt    53: #include "ssh.h"
                     54: #include "rsa.h"
                     55: #include "buffer.h"
1.22      markus     56: #include "key.h"
                     57: #include "authfd.h"
1.33      markus     58: #include "cipher.h"
1.22      markus     59: #include "kex.h"
1.28      markus     60: #include "compat.h"
1.33      markus     61: #include "log.h"
                     62: #include "atomicio.h"
1.74      djm        63: #include "misc.h"
1.2       provos     64:
1.57      stevesk    65: static int agent_present = 0;
                     66:
1.21      markus     67: /* helper */
1.24      markus     68: int    decode_reply(int type);
1.21      markus     69:
1.29      markus     70: /* macro to check for "agent failure" message */
                     71: #define agent_failed(x) \
1.44      markus     72:     ((x == SSH_AGENT_FAILURE) || (x == SSH_COM_AGENT2_FAILURE) || \
1.55      deraadt    73:     (x == SSH2_AGENT_FAILURE))
1.29      markus     74:
1.57      stevesk    75: int
                     76: ssh_agent_present(void)
                     77: {
                     78:        int authfd;
                     79:
                     80:        if (agent_present)
                     81:                return 1;
                     82:        if ((authfd = ssh_get_authentication_socket()) == -1)
                     83:                return 0;
                     84:        else {
                     85:                ssh_close_authentication_socket(authfd);
                     86:                return 1;
                     87:        }
                     88: }
                     89:
1.1       deraadt    90: /* Returns the number of the authentication fd, or -1 if there is none. */
                     91:
1.2       provos     92: int
1.32      markus     93: ssh_get_authentication_socket(void)
1.1       deraadt    94: {
1.12      markus     95:        const char *authsocket;
1.45      stevesk    96:        int sock;
1.12      markus     97:        struct sockaddr_un sunaddr;
                     98:
                     99:        authsocket = getenv(SSH_AUTHSOCKET_ENV_NAME);
                    100:        if (!authsocket)
                    101:                return -1;
                    102:
                    103:        sunaddr.sun_family = AF_UNIX;
                    104:        strlcpy(sunaddr.sun_path, authsocket, sizeof(sunaddr.sun_path));
                    105:
                    106:        sock = socket(AF_UNIX, SOCK_STREAM, 0);
                    107:        if (sock < 0)
                    108:                return -1;
                    109:
                    110:        /* close on exec */
                    111:        if (fcntl(sock, F_SETFD, 1) == -1) {
                    112:                close(sock);
                    113:                return -1;
                    114:        }
1.71      deraadt   115:        if (connect(sock, (struct sockaddr *)&sunaddr, sizeof sunaddr) < 0) {
1.12      markus    116:                close(sock);
                    117:                return -1;
                    118:        }
1.57      stevesk   119:        agent_present = 1;
1.12      markus    120:        return sock;
1.1       deraadt   121: }
                    122:
1.41      itojun    123: static int
1.25      markus    124: ssh_request_reply(AuthenticationConnection *auth, Buffer *request, Buffer *reply)
1.24      markus    125: {
1.66      djm       126:        u_int l, len;
1.24      markus    127:        char buf[1024];
                    128:
                    129:        /* Get the length of the message, and format it in the buffer. */
                    130:        len = buffer_len(request);
1.74      djm       131:        put_u32(buf, len);
1.24      markus    132:
                    133:        /* Send the length and then the packet to the agent. */
1.61      deraadt   134:        if (atomicio(vwrite, auth->fd, buf, 4) != 4 ||
                    135:            atomicio(vwrite, auth->fd, buffer_ptr(request),
1.24      markus    136:            buffer_len(request)) != buffer_len(request)) {
                    137:                error("Error writing to authentication socket.");
                    138:                return 0;
                    139:        }
                    140:        /*
                    141:         * Wait for response from the agent.  First read the length of the
                    142:         * response packet.
                    143:         */
1.64      avsm      144:        if (atomicio(read, auth->fd, buf, 4) != 4) {
                    145:            error("Error reading response length from authentication socket.");
                    146:            return 0;
1.24      markus    147:        }
                    148:
                    149:        /* Extract the length, and check it for sanity. */
1.74      djm       150:        len = get_u32(buf);
1.24      markus    151:        if (len > 256 * 1024)
1.62      miod      152:                fatal("Authentication response too long: %u", len);
1.24      markus    153:
                    154:        /* Read the rest of the response in to the buffer. */
                    155:        buffer_clear(reply);
                    156:        while (len > 0) {
                    157:                l = len;
                    158:                if (l > sizeof(buf))
                    159:                        l = sizeof(buf);
1.65      avsm      160:                if (atomicio(read, auth->fd, buf, l) != l) {
1.24      markus    161:                        error("Error reading response from authentication socket.");
                    162:                        return 0;
                    163:                }
1.56      markus    164:                buffer_append(reply, buf, l);
1.24      markus    165:                len -= l;
                    166:        }
                    167:        return 1;
                    168: }
                    169:
1.14      markus    170: /*
                    171:  * Closes the agent socket if it should be closed (depends on how it was
                    172:  * obtained).  The argument must have been returned by
                    173:  * ssh_get_authentication_socket().
                    174:  */
1.1       deraadt   175:
1.18      markus    176: void
1.12      markus    177: ssh_close_authentication_socket(int sock)
1.1       deraadt   178: {
1.12      markus    179:        if (getenv(SSH_AUTHSOCKET_ENV_NAME))
                    180:                close(sock);
1.1       deraadt   181: }
                    182:
1.14      markus    183: /*
                    184:  * Opens and connects a private socket for communication with the
                    185:  * authentication agent.  Returns the file descriptor (which must be
                    186:  * shut down and closed by the caller when no longer needed).
                    187:  * Returns NULL if an error occurred and the connection could not be
                    188:  * opened.
                    189:  */
1.1       deraadt   190:
1.12      markus    191: AuthenticationConnection *
1.32      markus    192: ssh_get_authentication_connection(void)
1.1       deraadt   193: {
1.12      markus    194:        AuthenticationConnection *auth;
                    195:        int sock;
1.1       deraadt   196:
1.12      markus    197:        sock = ssh_get_authentication_socket();
                    198:
1.14      markus    199:        /*
                    200:         * Fail if we couldn't obtain a connection.  This happens if we
                    201:         * exited due to a timeout.
                    202:         */
1.12      markus    203:        if (sock < 0)
                    204:                return NULL;
                    205:
                    206:        auth = xmalloc(sizeof(*auth));
                    207:        auth->fd = sock;
                    208:        buffer_init(&auth->identities);
                    209:        auth->howmany = 0;
                    210:
                    211:        return auth;
1.1       deraadt   212: }
                    213:
1.14      markus    214: /*
                    215:  * Closes the connection to the authentication agent and frees any associated
                    216:  * memory.
                    217:  */
1.1       deraadt   218:
1.18      markus    219: void
1.25      markus    220: ssh_close_authentication_connection(AuthenticationConnection *auth)
1.1       deraadt   221: {
1.25      markus    222:        buffer_free(&auth->identities);
                    223:        close(auth->fd);
                    224:        xfree(auth);
1.50      markus    225: }
                    226:
                    227: /* Lock/unlock agent */
                    228: int
                    229: ssh_lock_agent(AuthenticationConnection *auth, int lock, const char *password)
                    230: {
                    231:        int type;
                    232:        Buffer msg;
                    233:
                    234:        buffer_init(&msg);
                    235:        buffer_put_char(&msg, lock ? SSH_AGENTC_LOCK : SSH_AGENTC_UNLOCK);
                    236:        buffer_put_cstring(&msg, password);
                    237:
                    238:        if (ssh_request_reply(auth, &msg, &msg) == 0) {
                    239:                buffer_free(&msg);
                    240:                return 0;
                    241:        }
                    242:        type = buffer_get_char(&msg);
                    243:        buffer_free(&msg);
                    244:        return decode_reply(type);
1.1       deraadt   245: }
                    246:
1.14      markus    247: /*
                    248:  * Returns the first authentication identity held by the agent.
                    249:  */
1.1       deraadt   250:
1.30      markus    251: int
                    252: ssh_get_num_identities(AuthenticationConnection *auth, int version)
1.1       deraadt   253: {
1.25      markus    254:        int type, code1 = 0, code2 = 0;
1.24      markus    255:        Buffer request;
1.25      markus    256:
1.46      deraadt   257:        switch (version) {
1.25      markus    258:        case 1:
                    259:                code1 = SSH_AGENTC_REQUEST_RSA_IDENTITIES;
                    260:                code2 = SSH_AGENT_RSA_IDENTITIES_ANSWER;
                    261:                break;
                    262:        case 2:
                    263:                code1 = SSH2_AGENTC_REQUEST_IDENTITIES;
                    264:                code2 = SSH2_AGENT_IDENTITIES_ANSWER;
                    265:                break;
                    266:        default:
1.30      markus    267:                return 0;
1.25      markus    268:        }
1.1       deraadt   269:
1.14      markus    270:        /*
                    271:         * Send a message to the agent requesting for a list of the
                    272:         * identities it can represent.
                    273:         */
1.24      markus    274:        buffer_init(&request);
1.25      markus    275:        buffer_put_char(&request, code1);
1.12      markus    276:
                    277:        buffer_clear(&auth->identities);
1.24      markus    278:        if (ssh_request_reply(auth, &request, &auth->identities) == 0) {
                    279:                buffer_free(&request);
1.30      markus    280:                return 0;
1.12      markus    281:        }
1.24      markus    282:        buffer_free(&request);
1.1       deraadt   283:
1.12      markus    284:        /* Get message type, and verify that we got a proper answer. */
1.24      markus    285:        type = buffer_get_char(&auth->identities);
1.29      markus    286:        if (agent_failed(type)) {
1.30      markus    287:                return 0;
1.25      markus    288:        } else if (type != code2) {
1.24      markus    289:                fatal("Bad authentication reply message type: %d", type);
1.25      markus    290:        }
1.12      markus    291:
                    292:        /* Get the number of entries in the response and check it for sanity. */
                    293:        auth->howmany = buffer_get_int(&auth->identities);
1.62      miod      294:        if ((u_int)auth->howmany > 1024)
1.37      millert   295:                fatal("Too many identities in authentication reply: %d",
1.24      markus    296:                    auth->howmany);
1.12      markus    297:
1.30      markus    298:        return auth->howmany;
                    299: }
                    300:
                    301: Key *
                    302: ssh_get_first_identity(AuthenticationConnection *auth, char **comment, int version)
                    303: {
                    304:        /* get number of identities and return the first entry (if any). */
                    305:        if (ssh_get_num_identities(auth, version) > 0)
                    306:                return ssh_get_next_identity(auth, comment, version);
                    307:        return NULL;
1.1       deraadt   308: }
                    309:
1.25      markus    310: Key *
                    311: ssh_get_next_identity(AuthenticationConnection *auth, char **comment, int version)
1.1       deraadt   312: {
1.66      djm       313:        int keybits;
1.31      markus    314:        u_int bits;
                    315:        u_char *blob;
                    316:        u_int blen;
1.25      markus    317:        Key *key = NULL;
1.9       markus    318:
1.12      markus    319:        /* Return failure if no more entries. */
                    320:        if (auth->howmany <= 0)
1.25      markus    321:                return NULL;
1.12      markus    322:
1.14      markus    323:        /*
                    324:         * Get the next entry from the packet.  These will abort with a fatal
                    325:         * error if the packet is too short or contains corrupt data.
                    326:         */
1.46      deraadt   327:        switch (version) {
1.25      markus    328:        case 1:
1.30      markus    329:                key = key_new(KEY_RSA1);
1.25      markus    330:                bits = buffer_get_int(&auth->identities);
                    331:                buffer_get_bignum(&auth->identities, key->rsa->e);
                    332:                buffer_get_bignum(&auth->identities, key->rsa->n);
                    333:                *comment = buffer_get_string(&auth->identities, NULL);
1.66      djm       334:                keybits = BN_num_bits(key->rsa->n);
                    335:                if (keybits < 0 || bits != (u_int)keybits)
1.59      itojun    336:                        logit("Warning: identity keysize mismatch: actual %d, announced %u",
1.25      markus    337:                            BN_num_bits(key->rsa->n), bits);
                    338:                break;
                    339:        case 2:
                    340:                blob = buffer_get_string(&auth->identities, &blen);
                    341:                *comment = buffer_get_string(&auth->identities, NULL);
1.30      markus    342:                key = key_from_blob(blob, blen);
1.25      markus    343:                xfree(blob);
                    344:                break;
                    345:        default:
                    346:                return NULL;
                    347:        }
1.12      markus    348:        /* Decrement the number of remaining entries. */
                    349:        auth->howmany--;
1.25      markus    350:        return key;
1.1       deraadt   351: }
                    352:
1.14      markus    353: /*
                    354:  * Generates a random challenge, sends it to the agent, and waits for
                    355:  * response from the agent.  Returns true (non-zero) if the agent gave the
                    356:  * correct answer, zero otherwise.  Response type selects the style of
                    357:  * response desired, with 0 corresponding to protocol version 1.0 (no longer
                    358:  * supported) and 1 corresponding to protocol version 1.1.
                    359:  */
1.1       deraadt   360:
1.2       provos    361: int
                    362: ssh_decrypt_challenge(AuthenticationConnection *auth,
1.25      markus    363:     Key* key, BIGNUM *challenge,
1.31      markus    364:     u_char session_id[16],
                    365:     u_int response_type,
                    366:     u_char response[16])
1.1       deraadt   367: {
1.12      markus    368:        Buffer buffer;
1.24      markus    369:        int success = 0;
                    370:        int i;
                    371:        int type;
1.12      markus    372:
1.30      markus    373:        if (key->type != KEY_RSA1)
1.25      markus    374:                return 0;
                    375:        if (response_type == 0) {
1.59      itojun    376:                logit("Compatibility with ssh protocol version 1.0 no longer supported.");
1.25      markus    377:                return 0;
                    378:        }
1.12      markus    379:        buffer_init(&buffer);
1.24      markus    380:        buffer_put_char(&buffer, SSH_AGENTC_RSA_CHALLENGE);
1.25      markus    381:        buffer_put_int(&buffer, BN_num_bits(key->rsa->n));
                    382:        buffer_put_bignum(&buffer, key->rsa->e);
                    383:        buffer_put_bignum(&buffer, key->rsa->n);
1.12      markus    384:        buffer_put_bignum(&buffer, challenge);
1.47      stevesk   385:        buffer_append(&buffer, session_id, 16);
1.12      markus    386:        buffer_put_int(&buffer, response_type);
                    387:
1.24      markus    388:        if (ssh_request_reply(auth, &buffer, &buffer) == 0) {
1.12      markus    389:                buffer_free(&buffer);
                    390:                return 0;
                    391:        }
1.24      markus    392:        type = buffer_get_char(&buffer);
1.12      markus    393:
1.29      markus    394:        if (agent_failed(type)) {
1.59      itojun    395:                logit("Agent admitted failure to authenticate using the key.");
1.24      markus    396:        } else if (type != SSH_AGENT_RSA_RESPONSE) {
                    397:                fatal("Bad authentication response: %d", type);
                    398:        } else {
                    399:                success = 1;
                    400:                /*
                    401:                 * Get the response from the packet.  This will abort with a
                    402:                 * fatal error if the packet is corrupt.
                    403:                 */
                    404:                for (i = 0; i < 16; i++)
1.73      deraadt   405:                        response[i] = (u_char)buffer_get_char(&buffer);
1.12      markus    406:        }
                    407:        buffer_free(&buffer);
1.24      markus    408:        return success;
1.12      markus    409: }
1.1       deraadt   410:
1.25      markus    411: /* ask agent to sign data, returns -1 on error, 0 on success */
                    412: int
                    413: ssh_agent_sign(AuthenticationConnection *auth,
                    414:     Key *key,
1.48      markus    415:     u_char **sigp, u_int *lenp,
                    416:     u_char *data, u_int datalen)
1.25      markus    417: {
1.28      markus    418:        extern int datafellows;
1.25      markus    419:        Buffer msg;
1.31      markus    420:        u_char *blob;
                    421:        u_int blen;
1.28      markus    422:        int type, flags = 0;
1.25      markus    423:        int ret = -1;
                    424:
1.30      markus    425:        if (key_to_blob(key, &blob, &blen) == 0)
1.25      markus    426:                return -1;
                    427:
1.28      markus    428:        if (datafellows & SSH_BUG_SIGBLOB)
                    429:                flags = SSH_AGENT_OLD_SIGNATURE;
                    430:
1.25      markus    431:        buffer_init(&msg);
                    432:        buffer_put_char(&msg, SSH2_AGENTC_SIGN_REQUEST);
                    433:        buffer_put_string(&msg, blob, blen);
                    434:        buffer_put_string(&msg, data, datalen);
1.28      markus    435:        buffer_put_int(&msg, flags);
1.25      markus    436:        xfree(blob);
                    437:
                    438:        if (ssh_request_reply(auth, &msg, &msg) == 0) {
                    439:                buffer_free(&msg);
                    440:                return -1;
                    441:        }
                    442:        type = buffer_get_char(&msg);
1.29      markus    443:        if (agent_failed(type)) {
1.59      itojun    444:                logit("Agent admitted failure to sign using the key.");
1.25      markus    445:        } else if (type != SSH2_AGENT_SIGN_RESPONSE) {
                    446:                fatal("Bad authentication response: %d", type);
                    447:        } else {
                    448:                ret = 0;
                    449:                *sigp = buffer_get_string(&msg, lenp);
                    450:        }
                    451:        buffer_free(&msg);
                    452:        return ret;
                    453: }
                    454:
1.22      markus    455: /* Encode key for a message to the agent. */
                    456:
1.41      itojun    457: static void
1.30      markus    458: ssh_encode_identity_rsa1(Buffer *b, RSA *key, const char *comment)
1.22      markus    459: {
                    460:        buffer_put_int(b, BN_num_bits(key->n));
                    461:        buffer_put_bignum(b, key->n);
                    462:        buffer_put_bignum(b, key->e);
                    463:        buffer_put_bignum(b, key->d);
                    464:        /* To keep within the protocol: p < q for ssh. in SSL p > q */
                    465:        buffer_put_bignum(b, key->iqmp);        /* ssh key->u */
                    466:        buffer_put_bignum(b, key->q);   /* ssh key->p, SSL key->q */
                    467:        buffer_put_bignum(b, key->p);   /* ssh key->q, SSL key->p */
1.40      markus    468:        buffer_put_cstring(b, comment);
1.22      markus    469: }
                    470:
1.41      itojun    471: static void
1.30      markus    472: ssh_encode_identity_ssh2(Buffer *b, Key *key, const char *comment)
1.22      markus    473: {
1.30      markus    474:        buffer_put_cstring(b, key_ssh_name(key));
1.46      deraadt   475:        switch (key->type) {
1.30      markus    476:        case KEY_RSA:
                    477:                buffer_put_bignum2(b, key->rsa->n);
                    478:                buffer_put_bignum2(b, key->rsa->e);
                    479:                buffer_put_bignum2(b, key->rsa->d);
                    480:                buffer_put_bignum2(b, key->rsa->iqmp);
                    481:                buffer_put_bignum2(b, key->rsa->p);
                    482:                buffer_put_bignum2(b, key->rsa->q);
                    483:                break;
1.82    ! djm       484:        case KEY_RSA_CERT:
        !           485:                if (key->cert == NULL || buffer_len(&key->cert->certblob) == 0)
        !           486:                        fatal("%s: no cert/certblob", __func__);
        !           487:                buffer_put_string(b, buffer_ptr(&key->cert->certblob),
        !           488:                    buffer_len(&key->cert->certblob));
        !           489:                buffer_put_bignum2(b, key->rsa->d);
        !           490:                buffer_put_bignum2(b, key->rsa->iqmp);
        !           491:                buffer_put_bignum2(b, key->rsa->p);
        !           492:                buffer_put_bignum2(b, key->rsa->q);
        !           493:                break;
1.30      markus    494:        case KEY_DSA:
                    495:                buffer_put_bignum2(b, key->dsa->p);
                    496:                buffer_put_bignum2(b, key->dsa->q);
                    497:                buffer_put_bignum2(b, key->dsa->g);
                    498:                buffer_put_bignum2(b, key->dsa->pub_key);
                    499:                buffer_put_bignum2(b, key->dsa->priv_key);
                    500:                break;
1.82    ! djm       501:        case KEY_DSA_CERT:
        !           502:                if (key->cert == NULL || buffer_len(&key->cert->certblob) == 0)
        !           503:                        fatal("%s: no cert/certblob", __func__);
        !           504:                buffer_put_string(b, buffer_ptr(&key->cert->certblob),
        !           505:                    buffer_len(&key->cert->certblob));
        !           506:                buffer_put_bignum2(b, key->dsa->priv_key);
        !           507:                break;
1.30      markus    508:        }
                    509:        buffer_put_cstring(b, comment);
1.22      markus    510: }
                    511:
1.14      markus    512: /*
                    513:  * Adds an identity to the authentication server.  This call is not meant to
                    514:  * be used by normal applications.
                    515:  */
1.1       deraadt   516:
1.18      markus    517: int
1.54      markus    518: ssh_add_identity_constrained(AuthenticationConnection *auth, Key *key,
1.58      markus    519:     const char *comment, u_int life, u_int confirm)
1.1       deraadt   520: {
1.25      markus    521:        Buffer msg;
1.58      markus    522:        int type, constrained = (life || confirm);
1.12      markus    523:
1.25      markus    524:        buffer_init(&msg);
1.22      markus    525:
                    526:        switch (key->type) {
1.30      markus    527:        case KEY_RSA1:
1.54      markus    528:                type = constrained ?
                    529:                    SSH_AGENTC_ADD_RSA_ID_CONSTRAINED :
                    530:                    SSH_AGENTC_ADD_RSA_IDENTITY;
                    531:                buffer_put_char(&msg, type);
1.30      markus    532:                ssh_encode_identity_rsa1(&msg, key->rsa, comment);
                    533:                break;
1.22      markus    534:        case KEY_RSA:
1.82    ! djm       535:        case KEY_RSA_CERT:
1.22      markus    536:        case KEY_DSA:
1.82    ! djm       537:        case KEY_DSA_CERT:
1.54      markus    538:                type = constrained ?
                    539:                    SSH2_AGENTC_ADD_ID_CONSTRAINED :
                    540:                    SSH2_AGENTC_ADD_IDENTITY;
                    541:                buffer_put_char(&msg, type);
1.30      markus    542:                ssh_encode_identity_ssh2(&msg, key, comment);
1.22      markus    543:                break;
                    544:        default:
1.25      markus    545:                buffer_free(&msg);
1.22      markus    546:                return 0;
                    547:        }
1.54      markus    548:        if (constrained) {
                    549:                if (life != 0) {
                    550:                        buffer_put_char(&msg, SSH_AGENT_CONSTRAIN_LIFETIME);
                    551:                        buffer_put_int(&msg, life);
                    552:                }
1.58      markus    553:                if (confirm != 0)
                    554:                        buffer_put_char(&msg, SSH_AGENT_CONSTRAIN_CONFIRM);
1.54      markus    555:        }
1.25      markus    556:        if (ssh_request_reply(auth, &msg, &msg) == 0) {
                    557:                buffer_free(&msg);
1.12      markus    558:                return 0;
                    559:        }
1.25      markus    560:        type = buffer_get_char(&msg);
                    561:        buffer_free(&msg);
1.24      markus    562:        return decode_reply(type);
1.54      markus    563: }
                    564:
1.14      markus    565: /*
                    566:  * Removes an identity from the authentication server.  This call is not
                    567:  * meant to be used by normal applications.
                    568:  */
1.1       deraadt   569:
1.18      markus    570: int
1.25      markus    571: ssh_remove_identity(AuthenticationConnection *auth, Key *key)
1.1       deraadt   572: {
1.25      markus    573:        Buffer msg;
1.24      markus    574:        int type;
1.31      markus    575:        u_char *blob;
                    576:        u_int blen;
1.12      markus    577:
1.25      markus    578:        buffer_init(&msg);
1.12      markus    579:
1.30      markus    580:        if (key->type == KEY_RSA1) {
1.25      markus    581:                buffer_put_char(&msg, SSH_AGENTC_REMOVE_RSA_IDENTITY);
                    582:                buffer_put_int(&msg, BN_num_bits(key->rsa->n));
                    583:                buffer_put_bignum(&msg, key->rsa->e);
                    584:                buffer_put_bignum(&msg, key->rsa->n);
1.82    ! djm       585:        } else if (key_type_plain(key->type) == KEY_DSA ||
        !           586:            key_type_plain(key->type) == KEY_RSA) {
1.30      markus    587:                key_to_blob(key, &blob, &blen);
1.25      markus    588:                buffer_put_char(&msg, SSH2_AGENTC_REMOVE_IDENTITY);
1.51      markus    589:                buffer_put_string(&msg, blob, blen);
                    590:                xfree(blob);
                    591:        } else {
                    592:                buffer_free(&msg);
                    593:                return 0;
                    594:        }
1.42      markus    595:        if (ssh_request_reply(auth, &msg, &msg) == 0) {
                    596:                buffer_free(&msg);
                    597:                return 0;
                    598:        }
                    599:        type = buffer_get_char(&msg);
                    600:        buffer_free(&msg);
                    601:        return decode_reply(type);
                    602: }
                    603:
                    604: int
1.63      djm       605: ssh_update_card(AuthenticationConnection *auth, int add,
1.60      djm       606:     const char *reader_id, const char *pin, u_int life, u_int confirm)
1.42      markus    607: {
                    608:        Buffer msg;
1.60      djm       609:        int type, constrained = (life || confirm);
                    610:
                    611:        if (add) {
                    612:                type = constrained ?
                    613:                    SSH_AGENTC_ADD_SMARTCARD_KEY_CONSTRAINED :
                    614:                    SSH_AGENTC_ADD_SMARTCARD_KEY;
                    615:        } else
                    616:                type = SSH_AGENTC_REMOVE_SMARTCARD_KEY;
1.42      markus    617:
                    618:        buffer_init(&msg);
1.60      djm       619:        buffer_put_char(&msg, type);
1.43      markus    620:        buffer_put_cstring(&msg, reader_id);
1.49      rees      621:        buffer_put_cstring(&msg, pin);
1.63      djm       622:
1.60      djm       623:        if (constrained) {
                    624:                if (life != 0) {
                    625:                        buffer_put_char(&msg, SSH_AGENT_CONSTRAIN_LIFETIME);
                    626:                        buffer_put_int(&msg, life);
                    627:                }
                    628:                if (confirm != 0)
                    629:                        buffer_put_char(&msg, SSH_AGENT_CONSTRAIN_CONFIRM);
                    630:        }
                    631:
1.25      markus    632:        if (ssh_request_reply(auth, &msg, &msg) == 0) {
                    633:                buffer_free(&msg);
1.12      markus    634:                return 0;
                    635:        }
1.25      markus    636:        type = buffer_get_char(&msg);
                    637:        buffer_free(&msg);
1.24      markus    638:        return decode_reply(type);
1.12      markus    639: }
                    640:
1.14      markus    641: /*
                    642:  * Removes all identities from the agent.  This call is not meant to be used
                    643:  * by normal applications.
                    644:  */
1.1       deraadt   645:
1.18      markus    646: int
1.25      markus    647: ssh_remove_all_identities(AuthenticationConnection *auth, int version)
1.1       deraadt   648: {
1.25      markus    649:        Buffer msg;
1.24      markus    650:        int type;
1.25      markus    651:        int code = (version==1) ?
                    652:                SSH_AGENTC_REMOVE_ALL_RSA_IDENTITIES :
                    653:                SSH2_AGENTC_REMOVE_ALL_IDENTITIES;
1.12      markus    654:
1.25      markus    655:        buffer_init(&msg);
                    656:        buffer_put_char(&msg, code);
1.12      markus    657:
1.25      markus    658:        if (ssh_request_reply(auth, &msg, &msg) == 0) {
                    659:                buffer_free(&msg);
1.12      markus    660:                return 0;
                    661:        }
1.25      markus    662:        type = buffer_get_char(&msg);
                    663:        buffer_free(&msg);
1.24      markus    664:        return decode_reply(type);
1.21      markus    665: }
                    666:
1.35      stevesk   667: int
1.24      markus    668: decode_reply(int type)
1.21      markus    669: {
1.12      markus    670:        switch (type) {
                    671:        case SSH_AGENT_FAILURE:
1.29      markus    672:        case SSH_COM_AGENT2_FAILURE:
1.44      markus    673:        case SSH2_AGENT_FAILURE:
1.59      itojun    674:                logit("SSH_AGENT_FAILURE");
1.12      markus    675:                return 0;
                    676:        case SSH_AGENT_SUCCESS:
                    677:                return 1;
                    678:        default:
1.21      markus    679:                fatal("Bad response from authentication agent: %d", type);
1.12      markus    680:        }
                    681:        /* NOTREACHED */
                    682:        return 0;
                    683: }