blob: d06cc536c9dd593e611e84d5bf2095d0aebc7994 [file] [log] [blame]
Damien Millerd4a8b7e1999-10-27 13:42:43 +10001/*
Damien Miller95def091999-11-25 00:26:21 +11002 * Author: Tatu Ylonen <ylo@cs.hut.fi>
Damien Miller95def091999-11-25 00:26:21 +11003 * Copyright (c) 1995 Tatu Ylonen <ylo@cs.hut.fi>, Espoo, Finland
4 * All rights reserved
Damien Miller95def091999-11-25 00:26:21 +11005 * Functions for connecting the local authentication agent.
Damien Miller4af51302000-04-16 11:18:38 +10006 *
Damien Millere4340be2000-09-16 13:29:08 +11007 * As far as I am concerned, the code I have written for this software
8 * can be used freely for any purpose. Any derived versions of this
9 * software must be clearly marked as such, and if the derived work is
10 * incompatible with the protocol description in the RFC file, it must be
11 * called by a name other than "ssh" or "Secure Shell".
Damien Millerad833b32000-08-23 10:46:23 +100012 *
Damien Millere4340be2000-09-16 13:29:08 +110013 * SSH2 implementation,
14 * Copyright (c) 2000 Markus Friedl. All rights reserved.
15 *
16 * Redistribution and use in source and binary forms, with or without
17 * modification, are permitted provided that the following conditions
18 * are met:
19 * 1. Redistributions of source code must retain the above copyright
20 * notice, this list of conditions and the following disclaimer.
21 * 2. Redistributions in binary form must reproduce the above copyright
22 * notice, this list of conditions and the following disclaimer in the
23 * documentation and/or other materials provided with the distribution.
24 *
25 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
26 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
27 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
28 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
29 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
30 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
31 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
32 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
33 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
34 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
Damien Miller95def091999-11-25 00:26:21 +110035 */
Damien Millerd4a8b7e1999-10-27 13:42:43 +100036
37#include "includes.h"
Damien Miller874d77b2000-10-14 16:23:11 +110038RCSID("$OpenBSD: authfd.c,v 1.29 2000/10/09 21:51:00 markus Exp $");
Damien Millerd4a8b7e1999-10-27 13:42:43 +100039
40#include "ssh.h"
41#include "rsa.h"
Damien Millerd4a8b7e1999-10-27 13:42:43 +100042#include "buffer.h"
43#include "bufaux.h"
44#include "xmalloc.h"
45#include "getput.h"
46
47#include <openssl/rsa.h>
Damien Miller994cf142000-07-21 10:19:44 +100048#include <openssl/dsa.h>
49#include <openssl/evp.h>
50#include "key.h"
51#include "authfd.h"
52#include "kex.h"
Damien Millerad833b32000-08-23 10:46:23 +100053#include "dsa.h"
Damien Miller62cee002000-09-23 17:15:56 +110054#include "compat.h"
Damien Millerd4a8b7e1999-10-27 13:42:43 +100055
Damien Miller37023962000-07-11 17:31:38 +100056/* helper */
Damien Miller942da032000-08-18 13:59:06 +100057int decode_reply(int type);
Damien Miller37023962000-07-11 17:31:38 +100058
Damien Miller874d77b2000-10-14 16:23:11 +110059/* macro to check for "agent failure" message */
60#define agent_failed(x) \
61 ((x == SSH_AGENT_FAILURE) || (x == SSH_COM_AGENT2_FAILURE))
62
Damien Millerd4a8b7e1999-10-27 13:42:43 +100063/* Returns the number of the authentication fd, or -1 if there is none. */
64
65int
66ssh_get_authentication_socket()
67{
Damien Miller95def091999-11-25 00:26:21 +110068 const char *authsocket;
Damien Miller942da032000-08-18 13:59:06 +100069 int sock, len;
Damien Miller95def091999-11-25 00:26:21 +110070 struct sockaddr_un sunaddr;
Damien Millerd4a8b7e1999-10-27 13:42:43 +100071
Damien Miller95def091999-11-25 00:26:21 +110072 authsocket = getenv(SSH_AUTHSOCKET_ENV_NAME);
73 if (!authsocket)
74 return -1;
Damien Millerd4a8b7e1999-10-27 13:42:43 +100075
Damien Miller95def091999-11-25 00:26:21 +110076 sunaddr.sun_family = AF_UNIX;
77 strlcpy(sunaddr.sun_path, authsocket, sizeof(sunaddr.sun_path));
Damien Miller942da032000-08-18 13:59:06 +100078#ifdef HAVE_SUN_LEN_IN_SOCKADDR_UN
79 sunaddr.sun_len = len = SUN_LEN(&sunaddr)+1;
80#else /* HAVE_SUN_LEN_IN_SOCKADDR_UN */
81 len = SUN_LEN(&sunaddr)+1;
82#endif /* HAVE_SUN_LEN_IN_SOCKADDR_UN */
Damien Miller10f6f6b1999-11-17 17:29:08 +110083
Damien Miller95def091999-11-25 00:26:21 +110084 sock = socket(AF_UNIX, SOCK_STREAM, 0);
85 if (sock < 0)
86 return -1;
Damien Millerd4a8b7e1999-10-27 13:42:43 +100087
Damien Miller95def091999-11-25 00:26:21 +110088 /* close on exec */
89 if (fcntl(sock, F_SETFD, 1) == -1) {
90 close(sock);
91 return -1;
92 }
Damien Miller942da032000-08-18 13:59:06 +100093 if (connect(sock, (struct sockaddr *) & sunaddr, len) < 0) {
Damien Miller95def091999-11-25 00:26:21 +110094 close(sock);
95 return -1;
96 }
97 return sock;
Damien Millerd4a8b7e1999-10-27 13:42:43 +100098}
99
Damien Miller942da032000-08-18 13:59:06 +1000100int
Damien Millerad833b32000-08-23 10:46:23 +1000101ssh_request_reply(AuthenticationConnection *auth, Buffer *request, Buffer *reply)
Damien Miller942da032000-08-18 13:59:06 +1000102{
103 int l, len;
104 char buf[1024];
105
106 /* Get the length of the message, and format it in the buffer. */
107 len = buffer_len(request);
108 PUT_32BIT(buf, len);
109
110 /* Send the length and then the packet to the agent. */
111 if (atomicio(write, auth->fd, buf, 4) != 4 ||
112 atomicio(write, auth->fd, buffer_ptr(request),
113 buffer_len(request)) != buffer_len(request)) {
114 error("Error writing to authentication socket.");
115 return 0;
116 }
117 /*
118 * Wait for response from the agent. First read the length of the
119 * response packet.
120 */
121 len = 4;
122 while (len > 0) {
123 l = read(auth->fd, buf + 4 - len, len);
124 if (l <= 0) {
125 error("Error reading response length from authentication socket.");
126 return 0;
127 }
128 len -= l;
129 }
130
131 /* Extract the length, and check it for sanity. */
132 len = GET_32BIT(buf);
133 if (len > 256 * 1024)
134 fatal("Authentication response too long: %d", len);
135
136 /* Read the rest of the response in to the buffer. */
137 buffer_clear(reply);
138 while (len > 0) {
139 l = len;
140 if (l > sizeof(buf))
141 l = sizeof(buf);
142 l = read(auth->fd, buf, l);
143 if (l <= 0) {
144 error("Error reading response from authentication socket.");
145 return 0;
146 }
147 buffer_append(reply, (char *) buf, l);
148 len -= l;
149 }
150 return 1;
151}
152
Damien Miller5428f641999-11-25 11:54:57 +1100153/*
154 * Closes the agent socket if it should be closed (depends on how it was
155 * obtained). The argument must have been returned by
156 * ssh_get_authentication_socket().
157 */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000158
Damien Miller4af51302000-04-16 11:18:38 +1000159void
Damien Miller95def091999-11-25 00:26:21 +1100160ssh_close_authentication_socket(int sock)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000161{
Damien Miller95def091999-11-25 00:26:21 +1100162 if (getenv(SSH_AUTHSOCKET_ENV_NAME))
163 close(sock);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000164}
165
Damien Miller5428f641999-11-25 11:54:57 +1100166/*
167 * Opens and connects a private socket for communication with the
168 * authentication agent. Returns the file descriptor (which must be
169 * shut down and closed by the caller when no longer needed).
170 * Returns NULL if an error occurred and the connection could not be
171 * opened.
172 */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000173
Damien Miller95def091999-11-25 00:26:21 +1100174AuthenticationConnection *
175ssh_get_authentication_connection()
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000176{
Damien Miller95def091999-11-25 00:26:21 +1100177 AuthenticationConnection *auth;
178 int sock;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000179
Damien Miller95def091999-11-25 00:26:21 +1100180 sock = ssh_get_authentication_socket();
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000181
Damien Miller5428f641999-11-25 11:54:57 +1100182 /*
183 * Fail if we couldn't obtain a connection. This happens if we
184 * exited due to a timeout.
185 */
Damien Miller95def091999-11-25 00:26:21 +1100186 if (sock < 0)
187 return NULL;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000188
Damien Miller95def091999-11-25 00:26:21 +1100189 auth = xmalloc(sizeof(*auth));
190 auth->fd = sock;
Damien Miller95def091999-11-25 00:26:21 +1100191 buffer_init(&auth->identities);
192 auth->howmany = 0;
193
194 return auth;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000195}
196
Damien Miller5428f641999-11-25 11:54:57 +1100197/*
198 * Closes the connection to the authentication agent and frees any associated
199 * memory.
200 */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000201
Damien Miller4af51302000-04-16 11:18:38 +1000202void
Damien Millerad833b32000-08-23 10:46:23 +1000203ssh_close_authentication_connection(AuthenticationConnection *auth)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000204{
Damien Millerad833b32000-08-23 10:46:23 +1000205 buffer_free(&auth->identities);
206 close(auth->fd);
207 xfree(auth);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000208}
209
Damien Miller5428f641999-11-25 11:54:57 +1100210/*
211 * Returns the first authentication identity held by the agent.
Damien Miller5428f641999-11-25 11:54:57 +1100212 */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000213
Damien Millerad833b32000-08-23 10:46:23 +1000214Key *
215ssh_get_first_identity(AuthenticationConnection *auth, char **comment, int version)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000216{
Damien Millerad833b32000-08-23 10:46:23 +1000217 int type, code1 = 0, code2 = 0;
Damien Miller942da032000-08-18 13:59:06 +1000218 Buffer request;
Damien Millerad833b32000-08-23 10:46:23 +1000219
220 switch(version){
221 case 1:
222 code1 = SSH_AGENTC_REQUEST_RSA_IDENTITIES;
223 code2 = SSH_AGENT_RSA_IDENTITIES_ANSWER;
224 break;
225 case 2:
226 code1 = SSH2_AGENTC_REQUEST_IDENTITIES;
227 code2 = SSH2_AGENT_IDENTITIES_ANSWER;
228 break;
229 default:
230 return NULL;
231 }
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000232
Damien Miller5428f641999-11-25 11:54:57 +1100233 /*
234 * Send a message to the agent requesting for a list of the
235 * identities it can represent.
236 */
Damien Miller942da032000-08-18 13:59:06 +1000237 buffer_init(&request);
Damien Millerad833b32000-08-23 10:46:23 +1000238 buffer_put_char(&request, code1);
Damien Miller942da032000-08-18 13:59:06 +1000239
240 buffer_clear(&auth->identities);
241 if (ssh_request_reply(auth, &request, &auth->identities) == 0) {
242 buffer_free(&request);
Damien Millerad833b32000-08-23 10:46:23 +1000243 return NULL;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000244 }
Damien Miller942da032000-08-18 13:59:06 +1000245 buffer_free(&request);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000246
Damien Miller95def091999-11-25 00:26:21 +1100247 /* Get message type, and verify that we got a proper answer. */
Damien Miller942da032000-08-18 13:59:06 +1000248 type = buffer_get_char(&auth->identities);
Damien Miller874d77b2000-10-14 16:23:11 +1100249 if (agent_failed(type)) {
Damien Millerad833b32000-08-23 10:46:23 +1000250 return NULL;
251 } else if (type != code2) {
Damien Miller942da032000-08-18 13:59:06 +1000252 fatal("Bad authentication reply message type: %d", type);
Damien Millerad833b32000-08-23 10:46:23 +1000253 }
Damien Miller95def091999-11-25 00:26:21 +1100254
255 /* Get the number of entries in the response and check it for sanity. */
256 auth->howmany = buffer_get_int(&auth->identities);
257 if (auth->howmany > 1024)
Damien Miller942da032000-08-18 13:59:06 +1000258 fatal("Too many identities in authentication reply: %d\n",
259 auth->howmany);
Damien Miller95def091999-11-25 00:26:21 +1100260
261 /* Return the first entry (if any). */
Damien Millerad833b32000-08-23 10:46:23 +1000262 return ssh_get_next_identity(auth, comment, version);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000263}
264
Damien Millerad833b32000-08-23 10:46:23 +1000265Key *
266ssh_get_next_identity(AuthenticationConnection *auth, char **comment, int version)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000267{
Damien Miller95def091999-11-25 00:26:21 +1100268 unsigned int bits;
Damien Millerad833b32000-08-23 10:46:23 +1000269 unsigned char *blob;
270 unsigned int blen;
271 Key *key = NULL;
Damien Miller7e8e8201999-11-16 13:37:16 +1100272
Damien Miller95def091999-11-25 00:26:21 +1100273 /* Return failure if no more entries. */
274 if (auth->howmany <= 0)
Damien Millerad833b32000-08-23 10:46:23 +1000275 return NULL;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000276
Damien Miller5428f641999-11-25 11:54:57 +1100277 /*
278 * Get the next entry from the packet. These will abort with a fatal
279 * error if the packet is too short or contains corrupt data.
280 */
Damien Millerad833b32000-08-23 10:46:23 +1000281 switch(version){
282 case 1:
283 key = key_new(KEY_RSA);
284 bits = buffer_get_int(&auth->identities);
285 buffer_get_bignum(&auth->identities, key->rsa->e);
286 buffer_get_bignum(&auth->identities, key->rsa->n);
287 *comment = buffer_get_string(&auth->identities, NULL);
288 if (bits != BN_num_bits(key->rsa->n))
289 log("Warning: identity keysize mismatch: actual %d, announced %u",
290 BN_num_bits(key->rsa->n), bits);
291 break;
292 case 2:
293 blob = buffer_get_string(&auth->identities, &blen);
294 *comment = buffer_get_string(&auth->identities, NULL);
295 key = dsa_key_from_blob(blob, blen);
296 xfree(blob);
297 break;
298 default:
299 return NULL;
300 break;
301 }
Damien Miller95def091999-11-25 00:26:21 +1100302 /* Decrement the number of remaining entries. */
303 auth->howmany--;
Damien Millerad833b32000-08-23 10:46:23 +1000304 return key;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000305}
306
Damien Miller5428f641999-11-25 11:54:57 +1100307/*
308 * Generates a random challenge, sends it to the agent, and waits for
309 * response from the agent. Returns true (non-zero) if the agent gave the
310 * correct answer, zero otherwise. Response type selects the style of
311 * response desired, with 0 corresponding to protocol version 1.0 (no longer
312 * supported) and 1 corresponding to protocol version 1.1.
313 */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000314
315int
316ssh_decrypt_challenge(AuthenticationConnection *auth,
Damien Millerad833b32000-08-23 10:46:23 +1000317 Key* key, BIGNUM *challenge,
Damien Miller942da032000-08-18 13:59:06 +1000318 unsigned char session_id[16],
319 unsigned int response_type,
320 unsigned char response[16])
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000321{
Damien Miller95def091999-11-25 00:26:21 +1100322 Buffer buffer;
Damien Miller942da032000-08-18 13:59:06 +1000323 int success = 0;
324 int i;
325 int type;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000326
Damien Millerad833b32000-08-23 10:46:23 +1000327 if (key->type != KEY_RSA)
328 return 0;
329 if (response_type == 0) {
330 log("Compatibility with ssh protocol version 1.0 no longer supported.");
331 return 0;
332 }
Damien Miller95def091999-11-25 00:26:21 +1100333 buffer_init(&buffer);
Damien Miller942da032000-08-18 13:59:06 +1000334 buffer_put_char(&buffer, SSH_AGENTC_RSA_CHALLENGE);
Damien Millerad833b32000-08-23 10:46:23 +1000335 buffer_put_int(&buffer, BN_num_bits(key->rsa->n));
336 buffer_put_bignum(&buffer, key->rsa->e);
337 buffer_put_bignum(&buffer, key->rsa->n);
Damien Miller95def091999-11-25 00:26:21 +1100338 buffer_put_bignum(&buffer, challenge);
339 buffer_append(&buffer, (char *) session_id, 16);
340 buffer_put_int(&buffer, response_type);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000341
Damien Miller942da032000-08-18 13:59:06 +1000342 if (ssh_request_reply(auth, &buffer, &buffer) == 0) {
Damien Miller95def091999-11-25 00:26:21 +1100343 buffer_free(&buffer);
344 return 0;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000345 }
Damien Miller942da032000-08-18 13:59:06 +1000346 type = buffer_get_char(&buffer);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000347
Damien Miller874d77b2000-10-14 16:23:11 +1100348 if (agent_failed(type)) {
Damien Miller95def091999-11-25 00:26:21 +1100349 log("Agent admitted failure to authenticate using the key.");
Damien Miller942da032000-08-18 13:59:06 +1000350 } else if (type != SSH_AGENT_RSA_RESPONSE) {
351 fatal("Bad authentication response: %d", type);
352 } else {
353 success = 1;
354 /*
355 * Get the response from the packet. This will abort with a
356 * fatal error if the packet is corrupt.
357 */
358 for (i = 0; i < 16; i++)
359 response[i] = buffer_get_char(&buffer);
Damien Miller95def091999-11-25 00:26:21 +1100360 }
Damien Miller95def091999-11-25 00:26:21 +1100361 buffer_free(&buffer);
Damien Miller942da032000-08-18 13:59:06 +1000362 return success;
Damien Miller95def091999-11-25 00:26:21 +1100363}
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000364
Damien Millerad833b32000-08-23 10:46:23 +1000365/* ask agent to sign data, returns -1 on error, 0 on success */
366int
367ssh_agent_sign(AuthenticationConnection *auth,
368 Key *key,
369 unsigned char **sigp, int *lenp,
370 unsigned char *data, int datalen)
371{
Damien Miller62cee002000-09-23 17:15:56 +1100372 extern int datafellows;
Damien Millerad833b32000-08-23 10:46:23 +1000373 Buffer msg;
374 unsigned char *blob;
375 unsigned int blen;
Damien Miller62cee002000-09-23 17:15:56 +1100376 int type, flags = 0;
Damien Millerad833b32000-08-23 10:46:23 +1000377 int ret = -1;
378
379 if (dsa_make_key_blob(key, &blob, &blen) == 0)
380 return -1;
381
Damien Miller62cee002000-09-23 17:15:56 +1100382 if (datafellows & SSH_BUG_SIGBLOB)
383 flags = SSH_AGENT_OLD_SIGNATURE;
384
Damien Millerad833b32000-08-23 10:46:23 +1000385 buffer_init(&msg);
386 buffer_put_char(&msg, SSH2_AGENTC_SIGN_REQUEST);
387 buffer_put_string(&msg, blob, blen);
388 buffer_put_string(&msg, data, datalen);
Damien Miller62cee002000-09-23 17:15:56 +1100389 buffer_put_int(&msg, flags);
Damien Millerad833b32000-08-23 10:46:23 +1000390 xfree(blob);
391
392 if (ssh_request_reply(auth, &msg, &msg) == 0) {
393 buffer_free(&msg);
394 return -1;
395 }
396 type = buffer_get_char(&msg);
Damien Miller874d77b2000-10-14 16:23:11 +1100397 if (agent_failed(type)) {
Damien Millerad833b32000-08-23 10:46:23 +1000398 log("Agent admitted failure to sign using the key.");
399 } else if (type != SSH2_AGENT_SIGN_RESPONSE) {
400 fatal("Bad authentication response: %d", type);
401 } else {
402 ret = 0;
403 *sigp = buffer_get_string(&msg, lenp);
404 }
405 buffer_free(&msg);
406 return ret;
407}
408
Damien Miller994cf142000-07-21 10:19:44 +1000409/* Encode key for a message to the agent. */
410
411void
412ssh_encode_identity_rsa(Buffer *b, RSA *key, const char *comment)
413{
414 buffer_clear(b);
415 buffer_put_char(b, SSH_AGENTC_ADD_RSA_IDENTITY);
416 buffer_put_int(b, BN_num_bits(key->n));
417 buffer_put_bignum(b, key->n);
418 buffer_put_bignum(b, key->e);
419 buffer_put_bignum(b, key->d);
420 /* To keep within the protocol: p < q for ssh. in SSL p > q */
421 buffer_put_bignum(b, key->iqmp); /* ssh key->u */
422 buffer_put_bignum(b, key->q); /* ssh key->p, SSL key->q */
423 buffer_put_bignum(b, key->p); /* ssh key->q, SSL key->p */
424 buffer_put_string(b, comment, strlen(comment));
425}
426
427void
428ssh_encode_identity_dsa(Buffer *b, DSA *key, const char *comment)
429{
430 buffer_clear(b);
431 buffer_put_char(b, SSH2_AGENTC_ADD_IDENTITY);
432 buffer_put_cstring(b, KEX_DSS);
433 buffer_put_bignum2(b, key->p);
434 buffer_put_bignum2(b, key->q);
435 buffer_put_bignum2(b, key->g);
436 buffer_put_bignum2(b, key->pub_key);
437 buffer_put_bignum2(b, key->priv_key);
438 buffer_put_string(b, comment, strlen(comment));
439}
440
Damien Miller5428f641999-11-25 11:54:57 +1100441/*
442 * Adds an identity to the authentication server. This call is not meant to
443 * be used by normal applications.
444 */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000445
Damien Miller4af51302000-04-16 11:18:38 +1000446int
Damien Miller994cf142000-07-21 10:19:44 +1000447ssh_add_identity(AuthenticationConnection *auth, Key *key, const char *comment)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000448{
Damien Millerad833b32000-08-23 10:46:23 +1000449 Buffer msg;
Damien Miller942da032000-08-18 13:59:06 +1000450 int type;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000451
Damien Millerad833b32000-08-23 10:46:23 +1000452 buffer_init(&msg);
Damien Miller994cf142000-07-21 10:19:44 +1000453
454 switch (key->type) {
455 case KEY_RSA:
Damien Millerad833b32000-08-23 10:46:23 +1000456 ssh_encode_identity_rsa(&msg, key->rsa, comment);
Damien Miller994cf142000-07-21 10:19:44 +1000457 break;
458 case KEY_DSA:
Damien Millerad833b32000-08-23 10:46:23 +1000459 ssh_encode_identity_dsa(&msg, key->dsa, comment);
Damien Miller994cf142000-07-21 10:19:44 +1000460 break;
461 default:
Damien Millerad833b32000-08-23 10:46:23 +1000462 buffer_free(&msg);
Damien Miller994cf142000-07-21 10:19:44 +1000463 return 0;
464 break;
465 }
Damien Millerad833b32000-08-23 10:46:23 +1000466 if (ssh_request_reply(auth, &msg, &msg) == 0) {
467 buffer_free(&msg);
Damien Miller95def091999-11-25 00:26:21 +1100468 return 0;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000469 }
Damien Millerad833b32000-08-23 10:46:23 +1000470 type = buffer_get_char(&msg);
471 buffer_free(&msg);
Damien Miller942da032000-08-18 13:59:06 +1000472 return decode_reply(type);
Damien Miller95def091999-11-25 00:26:21 +1100473}
474
Damien Miller5428f641999-11-25 11:54:57 +1100475/*
476 * Removes an identity from the authentication server. This call is not
477 * meant to be used by normal applications.
478 */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000479
Damien Miller4af51302000-04-16 11:18:38 +1000480int
Damien Millerad833b32000-08-23 10:46:23 +1000481ssh_remove_identity(AuthenticationConnection *auth, Key *key)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000482{
Damien Millerad833b32000-08-23 10:46:23 +1000483 Buffer msg;
Damien Miller942da032000-08-18 13:59:06 +1000484 int type;
Damien Millerad833b32000-08-23 10:46:23 +1000485 unsigned char *blob;
486 unsigned int blen;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000487
Damien Millerad833b32000-08-23 10:46:23 +1000488 buffer_init(&msg);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000489
Damien Millerad833b32000-08-23 10:46:23 +1000490 if (key->type == KEY_RSA) {
491 buffer_put_char(&msg, SSH_AGENTC_REMOVE_RSA_IDENTITY);
492 buffer_put_int(&msg, BN_num_bits(key->rsa->n));
493 buffer_put_bignum(&msg, key->rsa->e);
494 buffer_put_bignum(&msg, key->rsa->n);
495 } else if (key->type == KEY_DSA) {
496 dsa_make_key_blob(key, &blob, &blen);
497 buffer_put_char(&msg, SSH2_AGENTC_REMOVE_IDENTITY);
498 buffer_put_string(&msg, blob, blen);
499 xfree(blob);
500 } else {
501 buffer_free(&msg);
Damien Miller95def091999-11-25 00:26:21 +1100502 return 0;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000503 }
Damien Millerad833b32000-08-23 10:46:23 +1000504 if (ssh_request_reply(auth, &msg, &msg) == 0) {
505 buffer_free(&msg);
506 return 0;
507 }
508 type = buffer_get_char(&msg);
509 buffer_free(&msg);
Damien Miller942da032000-08-18 13:59:06 +1000510 return decode_reply(type);
Damien Miller95def091999-11-25 00:26:21 +1100511}
512
Damien Miller5428f641999-11-25 11:54:57 +1100513/*
514 * Removes all identities from the agent. This call is not meant to be used
515 * by normal applications.
516 */
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000517
Damien Miller4af51302000-04-16 11:18:38 +1000518int
Damien Millerad833b32000-08-23 10:46:23 +1000519ssh_remove_all_identities(AuthenticationConnection *auth, int version)
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000520{
Damien Millerad833b32000-08-23 10:46:23 +1000521 Buffer msg;
Damien Miller942da032000-08-18 13:59:06 +1000522 int type;
Damien Millerad833b32000-08-23 10:46:23 +1000523 int code = (version==1) ?
524 SSH_AGENTC_REMOVE_ALL_RSA_IDENTITIES :
525 SSH2_AGENTC_REMOVE_ALL_IDENTITIES;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000526
Damien Millerad833b32000-08-23 10:46:23 +1000527 buffer_init(&msg);
528 buffer_put_char(&msg, code);
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000529
Damien Millerad833b32000-08-23 10:46:23 +1000530 if (ssh_request_reply(auth, &msg, &msg) == 0) {
531 buffer_free(&msg);
Damien Miller95def091999-11-25 00:26:21 +1100532 return 0;
Damien Millerd4a8b7e1999-10-27 13:42:43 +1000533 }
Damien Millerad833b32000-08-23 10:46:23 +1000534 type = buffer_get_char(&msg);
535 buffer_free(&msg);
Damien Miller942da032000-08-18 13:59:06 +1000536 return decode_reply(type);
537}
538
539int
540decode_reply(int type)
541{
Damien Miller95def091999-11-25 00:26:21 +1100542 switch (type) {
543 case SSH_AGENT_FAILURE:
Damien Miller874d77b2000-10-14 16:23:11 +1100544 case SSH_COM_AGENT2_FAILURE:
Damien Miller942da032000-08-18 13:59:06 +1000545 log("SSH_AGENT_FAILURE");
Damien Miller95def091999-11-25 00:26:21 +1100546 return 0;
547 case SSH_AGENT_SUCCESS:
Damien Miller95def091999-11-25 00:26:21 +1100548 return 1;
549 default:
Damien Miller37023962000-07-11 17:31:38 +1000550 fatal("Bad response from authentication agent: %d", type);
Damien Miller95def091999-11-25 00:26:21 +1100551 }
552 /* NOTREACHED */
553 return 0;
554}