ssh.c 19.6 KB
Newer Older
Brad Morgan committed
1 2 3 4 5 6 7
/*
 * Copyright (C) the libgit2 contributors. All rights reserved.
 *
 * This file is part of libgit2, distributed under the GNU GPL v2 with
 * a Linking Exception. For full terms see the included COPYING file.
 */

8 9
#include "ssh.h"

10 11 12 13
#ifdef GIT_SSH
#include <libssh2.h>
#endif

14
#include "global.h"
Brad Morgan committed
15 16 17
#include "git2.h"
#include "buffer.h"
#include "netops.h"
Brad Morgan committed
18
#include "smart.h"
19
#include "cred.h"
20
#include "streams/socket.h"
Brad Morgan committed
21

22 23
#ifdef GIT_SSH

Brad Morgan committed
24 25
#define OWNING_SUBTRANSPORT(s) ((ssh_subtransport *)(s)->parent.subtransport)

26
static const char *ssh_prefixes[] = { "ssh://", "ssh+git://", "git+ssh://" };
27

Brad Morgan committed
28 29 30 31 32
static const char cmd_uploadpack[] = "git-upload-pack";
static const char cmd_receivepack[] = "git-receive-pack";

typedef struct {
	git_smart_subtransport_stream parent;
33
	git_stream *io;
Brad Morgan committed
34 35
	LIBSSH2_SESSION *session;
	LIBSSH2_CHANNEL *channel;
Brad Morgan committed
36 37 38 39 40 41 42
	const char *cmd;
	char *url;
	unsigned sent_command : 1;
} ssh_stream;

typedef struct {
	git_smart_subtransport parent;
Brad Morgan committed
43
	transport_smart *owner;
Brad Morgan committed
44
	ssh_stream *current_stream;
Brad Morgan committed
45
	git_cred *cred;
46 47
	char *cmd_uploadpack;
	char *cmd_receivepack;
Brad Morgan committed
48 49
} ssh_subtransport;

50
static int list_auth_methods(int *out, LIBSSH2_SESSION *session, const char *username);
51

52 53 54 55 56
static void ssh_error(LIBSSH2_SESSION *session, const char *errmsg)
{
	char *ssherr;
	libssh2_session_last_error(session, &ssherr, NULL, 0);

57
	git_error_set(GIT_ERROR_SSH, "%s: %s", errmsg, ssherr);
58 59
}

Brad Morgan committed
60 61 62
/*
 * Create a git protocol request.
 *
Brad Morgan committed
63
 * For example: git-upload-pack '/libgit2/libgit2'
Brad Morgan committed
64
 */
65
static int gen_proto(git_buf *request, const char *cmd, const char *url)
Brad Morgan committed
66
{
67
	const char *repo;
68
	int len;
69
	size_t i;
70

71
	for (i = 0; i < ARRAY_SIZE(ssh_prefixes); ++i) {
72 73 74 75 76 77 78 79 80 81
		const char *p = ssh_prefixes[i];

		if (!git__prefixcmp(url, p)) {
			url = url + strlen(p);
			repo = strchr(url, '/');
			if (repo && repo[1] == '~')
				++repo;

			goto done;
		}
82
	}
83 84
	repo = strchr(url, ':');
	if (repo) repo++;
85

86
done:
87
	if (!repo) {
88
		git_error_set(GIT_ERROR_NET, "malformed git protocol URL");
89 90
		return -1;
	}
91

92
	len = strlen(cmd) + 1 /* Space */ + 1 /* Quote */ + strlen(repo) + 1 /* Quote */ + 1;
93

Brad Morgan committed
94
	git_buf_grow(request, len);
95 96 97 98
	git_buf_puts(request, cmd);
	git_buf_puts(request, " '");
	git_buf_decode_percent(request, repo, strlen(repo));
	git_buf_puts(request, "'");
99

Brad Morgan committed
100 101
	if (git_buf_oom(request))
		return -1;
102

Brad Morgan committed
103 104 105
	return 0;
}

Brad Morgan committed
106
static int send_command(ssh_stream *s)
Brad Morgan committed
107 108 109
{
	int error;
	git_buf request = GIT_BUF_INIT;
110

111
	error = gen_proto(&request, s->cmd, s->url);
Brad Morgan committed
112 113
	if (error < 0)
		goto cleanup;
114

115
	error = libssh2_channel_exec(s->channel, request.ptr);
116 117
	if (error < LIBSSH2_ERROR_NONE) {
		ssh_error(s->session, "SSH could not execute request");
Brad Morgan committed
118
		goto cleanup;
119
	}
120

Brad Morgan committed
121
	s->sent_command = 1;
122

Brad Morgan committed
123
cleanup:
124
	git_buf_dispose(&request);
Brad Morgan committed
125 126 127
	return error;
}

Brad Morgan committed
128 129 130 131 132
static int ssh_stream_read(
	git_smart_subtransport_stream *stream,
	char *buffer,
	size_t buf_size,
	size_t *bytes_read)
Brad Morgan committed
133
{
134
	int rc;
Brad Morgan committed
135
	ssh_stream *s = (ssh_stream *)stream;
136

Brad Morgan committed
137
	*bytes_read = 0;
138

Brad Morgan committed
139 140
	if (!s->sent_command && send_command(s) < 0)
		return -1;
141

142
	if ((rc = libssh2_channel_read(s->channel, buffer, buf_size)) < LIBSSH2_ERROR_NONE) {
143
		ssh_error(s->session, "SSH could not read data");
Brad Morgan committed
144
		return -1;
145
	}
146

147 148 149 150 151
	/*
	 * If we can't get anything out of stdout, it's typically a
	 * not-found error, so read from stderr and signal EOF on
	 * stderr.
	 */
152 153
	if (rc == 0) {
		if ((rc = libssh2_channel_read_stderr(s->channel, buffer, buf_size)) > 0) {
154
			git_error_set(GIT_ERROR_SSH, "%*s", rc, buffer);
155 156 157 158 159
			return GIT_EEOF;
		} else if (rc < LIBSSH2_ERROR_NONE) {
			ssh_error(s->session, "SSH could not read stderr");
			return -1;
		}
160 161 162
	}


Brad Morgan committed
163
	*bytes_read = rc;
164

Brad Morgan committed
165 166 167
	return 0;
}

Brad Morgan committed
168 169 170 171
static int ssh_stream_write(
	git_smart_subtransport_stream *stream,
	const char *buffer,
	size_t len)
Brad Morgan committed
172
{
Brad Morgan committed
173
	ssh_stream *s = (ssh_stream *)stream;
174 175
	size_t off = 0;
	ssize_t ret = 0;
176

Brad Morgan committed
177 178
	if (!s->sent_command && send_command(s) < 0)
		return -1;
179

180 181 182 183 184 185 186 187 188 189
	do {
		ret = libssh2_channel_write(s->channel, buffer + off, len - off);
		if (ret < 0)
			break;

		off += ret;

	} while (off < len);

	if (ret < 0) {
190
		ssh_error(s->session, "SSH could not write data");
Brad Morgan committed
191 192
		return -1;
	}
193

194
	return 0;
Brad Morgan committed
195 196
}

Brad Morgan committed
197
static void ssh_stream_free(git_smart_subtransport_stream *stream)
Brad Morgan committed
198
{
Brad Morgan committed
199
	ssh_stream *s = (ssh_stream *)stream;
200
	ssh_subtransport *t;
201

202 203
	if (!stream)
		return;
204

205
	t = OWNING_SUBTRANSPORT(s);
Brad Morgan committed
206
	t->current_stream = NULL;
207

Brad Morgan committed
208 209
	if (s->channel) {
		libssh2_channel_close(s->channel);
210 211
		libssh2_channel_free(s->channel);
		s->channel = NULL;
Brad Morgan committed
212
	}
213

Brad Morgan committed
214
	if (s->session) {
215
		libssh2_session_disconnect(s->session, "closing transport");
216 217
		libssh2_session_free(s->session);
		s->session = NULL;
Brad Morgan committed
218
	}
219

220 221 222 223
	if (s->io) {
		git_stream_close(s->io);
		git_stream_free(s->io);
		s->io = NULL;
Brad Morgan committed
224
	}
225

Brad Morgan committed
226 227 228 229
	git__free(s->url);
	git__free(s);
}

Brad Morgan committed
230 231 232 233 234
static int ssh_stream_alloc(
	ssh_subtransport *t,
	const char *url,
	const char *cmd,
	git_smart_subtransport_stream **stream)
Brad Morgan committed
235
{
Brad Morgan committed
236
	ssh_stream *s;
237

238
	assert(stream);
239

Brad Morgan committed
240
	s = git__calloc(sizeof(ssh_stream), 1);
241
	GIT_ERROR_CHECK_ALLOC(s);
242

Brad Morgan committed
243
	s->parent.subtransport = &t->parent;
Brad Morgan committed
244 245 246
	s->parent.read = ssh_stream_read;
	s->parent.write = ssh_stream_write;
	s->parent.free = ssh_stream_free;
247

Brad Morgan committed
248
	s->cmd = cmd;
249

250
	s->url = git__strdup(url);
Brad Morgan committed
251 252 253 254
	if (!s->url) {
		git__free(s);
		return -1;
	}
255

Brad Morgan committed
256 257 258 259
	*stream = &s->parent;
	return 0;
}

260
static int git_ssh_extract_url_parts(
Brad Morgan committed
261 262 263 264 265 266
	char **host,
	char **username,
	const char *url)
{
	char *colon, *at;
	const char *start;
267

268
	colon = strchr(url, ':');
269 270


271
	at = strchr(url, '@');
Brad Morgan committed
272
	if (at) {
Etienne Samson committed
273
		start = at + 1;
Brad Morgan committed
274
		*username = git__substrdup(url, at - url);
275
		GIT_ERROR_CHECK_ALLOC(*username);
Brad Morgan committed
276
	} else {
277
		start = url;
278
		*username = NULL;
Brad Morgan committed
279
	}
280

281
	if (colon == NULL || (colon < start)) {
282
		git_error_set(GIT_ERROR_NET, "malformed URL");
283 284 285
		return -1;
	}

Brad Morgan committed
286
	*host = git__substrdup(start, colon - start);
287
	GIT_ERROR_CHECK_ALLOC(*host);
288

Brad Morgan committed
289 290 291
	return 0;
}

292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317
static int ssh_agent_auth(LIBSSH2_SESSION *session, git_cred_ssh_key *c) {
	int rc = LIBSSH2_ERROR_NONE;

	struct libssh2_agent_publickey *curr, *prev = NULL;

	LIBSSH2_AGENT *agent = libssh2_agent_init(session);

	if (agent == NULL)
		return -1;

	rc = libssh2_agent_connect(agent);

	if (rc != LIBSSH2_ERROR_NONE)
		goto shutdown;

	rc = libssh2_agent_list_identities(agent);

	if (rc != LIBSSH2_ERROR_NONE)
		goto shutdown;

	while (1) {
		rc = libssh2_agent_get_identity(agent, &curr, prev);

		if (rc < 0)
			goto shutdown;

318 319 320 321 322 323
		/* rc is set to 1 whenever the ssh agent ran out of keys to check.
		 * Set the error code to authentication failure rather than erroring
		 * out with an untranslatable error code.
		 */
		if (rc == 1) {
			rc = LIBSSH2_ERROR_AUTHENTICATION_FAILED;
324
			goto shutdown;
325
		}
326 327 328 329 330 331 332 333 334 335

		rc = libssh2_agent_userauth(agent, c->username, curr);

		if (rc == 0)
			break;

		prev = curr;
	}

shutdown:
336 337 338 339

	if (rc != LIBSSH2_ERROR_NONE)
		ssh_error(session, "error authenticating");

340 341 342 343 344 345
	libssh2_agent_disconnect(agent);
	libssh2_agent_free(agent);

	return rc;
}

346 347
static int _git_ssh_authenticate_session(
	LIBSSH2_SESSION* session,
348
	git_cred* cred)
349 350
{
	int rc;
351

352
	do {
353
		git_error_clear();
354
		switch (cred->credtype) {
355 356
		case GIT_CREDTYPE_USERPASS_PLAINTEXT: {
			git_cred_userpass_plaintext *c = (git_cred_userpass_plaintext *)cred;
357
			rc = libssh2_userauth_password(session, c->username, c->password);
358 359
			break;
		}
360 361
		case GIT_CREDTYPE_SSH_KEY: {
			git_cred_ssh_key *c = (git_cred_ssh_key *)cred;
362 363 364 365 366 367 368 369

			if (c->privatekey)
				rc = libssh2_userauth_publickey_fromfile(
					session, c->username, c->publickey,
					c->privatekey, c->passphrase);
			else
				rc = ssh_agent_auth(session, c);

370 371
			break;
		}
372 373
		case GIT_CREDTYPE_SSH_CUSTOM: {
			git_cred_ssh_custom *c = (git_cred_ssh_custom *)cred;
374

375
			rc = libssh2_userauth_publickey(
376
				session, c->username, (const unsigned char *)c->publickey,
377
				c->publickey_len, c->sign_callback, &c->payload);
378 379
			break;
		}
380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400
		case GIT_CREDTYPE_SSH_INTERACTIVE: {
			void **abstract = libssh2_session_abstract(session);
			git_cred_ssh_interactive *c = (git_cred_ssh_interactive *)cred;

			/* ideally, we should be able to set this by calling
			 * libssh2_session_init_ex() instead of libssh2_session_init().
			 * libssh2's API is inconsistent here i.e. libssh2_userauth_publickey()
			 * allows you to pass the `abstract` as part of the call, whereas
			 * libssh2_userauth_keyboard_interactive() does not!
			 *
			 * The only way to set the `abstract` pointer is by calling
			 * libssh2_session_abstract(), which will replace the existing
			 * pointer as is done below. This is safe for now (at time of writing),
			 * but may not be valid in future.
			 */
			*abstract = c->payload;

			rc = libssh2_userauth_keyboard_interactive(
				session, c->username, c->prompt_callback);
			break;
		}
401 402 403 404
#ifdef GIT_SSH_MEMORY_CREDENTIALS
		case GIT_CREDTYPE_SSH_MEMORY: {
			git_cred_ssh_key *c = (git_cred_ssh_key *)cred;

405 406 407
			assert(c->username);
			assert(c->privatekey);

408 409 410 411 412
			rc = libssh2_userauth_publickey_frommemory(
				session,
				c->username,
				strlen(c->username),
				c->publickey,
413
				c->publickey ? strlen(c->publickey) : 0,
414 415 416 417 418 419
				c->privatekey,
				strlen(c->privatekey),
				c->passphrase);
			break;
		}
#endif
420 421
		default:
			rc = LIBSSH2_ERROR_AUTHENTICATION_FAILED;
422
		}
Etienne Samson committed
423
	} while (LIBSSH2_ERROR_EAGAIN == rc || LIBSSH2_ERROR_TIMEOUT == rc);
424

425 426 427 428
	if (rc == LIBSSH2_ERROR_PASSWORD_EXPIRED ||
		rc == LIBSSH2_ERROR_AUTHENTICATION_FAILED ||
		rc == LIBSSH2_ERROR_PUBLICKEY_UNVERIFIED)
			return GIT_EAUTH;
429

430
	if (rc != LIBSSH2_ERROR_NONE) {
431
		if (!git_error_last())
432
			ssh_error(session, "Failed to authenticate SSH session");
433 434 435 436
		return -1;
	}

	return 0;
437 438
}

439 440 441 442 443 444 445 446 447 448 449
static int request_creds(git_cred **out, ssh_subtransport *t, const char *user, int auth_methods)
{
	int error, no_callback = 0;
	git_cred *cred = NULL;

	if (!t->owner->cred_acquire_cb) {
		no_callback = 1;
	} else {
		error = t->owner->cred_acquire_cb(&cred, t->owner->url, user, auth_methods,
						  t->owner->cred_acquire_payload);

450
		if (error == GIT_PASSTHROUGH) {
451
			no_callback = 1;
452
		} else if (error < 0) {
453
			return error;
454
		} else if (!cred) {
455
			git_error_set(GIT_ERROR_SSH, "callback failed to initialize SSH credentials");
456 457 458 459 460
			return -1;
		}
	}

	if (no_callback) {
461
		git_error_set(GIT_ERROR_SSH, "authentication required but no callback set");
462 463 464
		return -1;
	}

465 466
	if (!(cred->credtype & auth_methods)) {
		cred->free(cred);
467
		git_error_set(GIT_ERROR_SSH, "callback returned unsupported credentials type");
468 469 470
		return -1;
	}

471 472 473 474 475
	*out = cred;

	return 0;
}

Etienne Samson committed
476
static int _git_ssh_session_create(
Brad Morgan committed
477
	LIBSSH2_SESSION** session,
478
	git_stream *io)
Brad Morgan committed
479
{
480 481
	int rc = 0;
	LIBSSH2_SESSION* s;
482
	git_socket_stream *socket = (git_socket_stream *) io;
483

484 485 486 487
	assert(session);

	s = libssh2_session_init();
	if (!s) {
488
		git_error_set(GIT_ERROR_NET, "failed to initialize SSH session");
Etienne Samson committed
489
		return -1;
490
	}
491

Etienne Samson committed
492
	do {
493
		rc = libssh2_session_handshake(s, socket->s);
Etienne Samson committed
494
	} while (LIBSSH2_ERROR_EAGAIN == rc || LIBSSH2_ERROR_TIMEOUT == rc);
495

496
	if (rc != LIBSSH2_ERROR_NONE) {
497
		ssh_error(s, "failed to start SSH session");
498 499
		libssh2_session_free(s);
		return -1;
500
	}
501

Brad Morgan committed
502
	libssh2_session_set_blocking(s, 1);
503

Brad Morgan committed
504
	*session = s;
505

Brad Morgan committed
506 507 508
	return 0;
}

Brad Morgan committed
509
static int _git_ssh_setup_conn(
Brad Morgan committed
510 511
	ssh_subtransport *t,
	const char *url,
Brad Morgan committed
512
	const char *cmd,
513
	git_smart_subtransport_stream **stream)
Brad Morgan committed
514
{
Ben Straub committed
515
	char *host=NULL, *port=NULL, *path=NULL, *user=NULL, *pass=NULL;
Brad Morgan committed
516
	const char *default_port="22";
517
	int auth_methods, error = 0;
518
	size_t i;
Brad Morgan committed
519
	ssh_stream *s;
520
	git_cred *cred = NULL;
Brad Morgan committed
521 522
	LIBSSH2_SESSION* session=NULL;
	LIBSSH2_CHANNEL* channel=NULL;
523

524 525
	t->current_stream = NULL;

Brad Morgan committed
526
	*stream = NULL;
Brad Morgan committed
527
	if (ssh_stream_alloc(t, url, cmd, stream) < 0)
Brad Morgan committed
528
		return -1;
529

Brad Morgan committed
530
	s = (ssh_stream *)*stream;
531 532
	s->session = NULL;
	s->channel = NULL;
533

534
	for (i = 0; i < ARRAY_SIZE(ssh_prefixes); ++i) {
535 536 537 538 539 540 541 542
		const char *p = ssh_prefixes[i];

		if (!git__prefixcmp(url, p)) {
			if ((error = gitno_extract_url_parts(&host, &port, &path, &user, &pass, url, default_port)) < 0)
				goto done;

			goto post_extract;
		}
543
	}
544 545 546
	if ((error = git_ssh_extract_url_parts(&host, &user, url)) < 0)
		goto done;
	port = git__strdup(default_port);
547
	GIT_ERROR_CHECK_ALLOC(port);
548

549
post_extract:
550 551
	if ((error = git_socket_stream_new(&s->io, host, port)) < 0 ||
	    (error = git_stream_connect(s->io)) < 0)
552
		goto done;
553

554
	if ((error = _git_ssh_session_create(&session, s->io)) < 0)
555
		goto done;
556 557

	if (t->owner->certificate_check_cb != NULL) {
558
		git_cert_hostkey cert = {{ 0 }}, *cert_ptr;
559 560
		const char *key;

561
		cert.parent.cert_type = GIT_CERT_HOSTKEY_LIBSSH2;
562

563 564
		key = libssh2_hostkey_hash(session, LIBSSH2_HOSTKEY_HASH_SHA1);
		if (key != NULL) {
565 566
			cert.type |= GIT_CERT_SSH_SHA1;
			memcpy(&cert.hash_sha1, key, 20);
567 568
		}

569 570 571 572 573 574 575
		key = libssh2_hostkey_hash(session, LIBSSH2_HOSTKEY_HASH_MD5);
		if (key != NULL) {
			cert.type |= GIT_CERT_SSH_MD5;
			memcpy(&cert.hash_md5, key, 16);
		}

		if (cert.type == 0) {
576
			git_error_set(GIT_ERROR_SSH, "unable to get the host key");
577 578
			error = -1;
			goto done;
579 580 581
		}

		/* We don't currently trust any hostkeys */
582
		git_error_clear();
583 584 585 586

		cert_ptr = &cert;

		error = t->owner->certificate_check_cb((git_cert *) cert_ptr, 0, host, t->owner->message_cb_payload);
587 588

		if (error < 0 && error != GIT_PASSTHROUGH) {
589 590
			if (!git_error_last())
				git_error_set(GIT_ERROR_NET, "user cancelled hostkey check");
591

592
			goto done;
593
		}
594
	}
595

596 597 598
	/* we need the username to ask for auth methods */
	if (!user) {
		if ((error = request_creds(&cred, t, NULL, GIT_CREDTYPE_USERNAME)) < 0)
599
			goto done;
600

601 602 603 604
		user = git__strdup(((git_cred_username *) cred)->username);
		cred->free(cred);
		cred = NULL;
		if (!user)
605
			goto done;
606
	} else if (user && pass) {
607
		if ((error = git_cred_userpass_plaintext_new(&cred, user, pass)) < 0)
608
			goto done;
609 610
	}

611
	if ((error = list_auth_methods(&auth_methods, session, user)) < 0)
612
		goto done;
613

614 615
	error = GIT_EAUTH;
	/* if we already have something to try */
616
	if (cred && auth_methods & cred->credtype)
617
		error = _git_ssh_authenticate_session(session, cred);
618

619 620 621 622 623 624 625
	while (error == GIT_EAUTH) {
		if (cred) {
			cred->free(cred);
			cred = NULL;
		}

		if ((error = request_creds(&cred, t, user, auth_methods)) < 0)
626
			goto done;
627

628
		if (strcmp(user, git_cred__username(cred))) {
629
			git_error_set(GIT_ERROR_SSH, "username does not match previous request");
630
			error = -1;
631
			goto done;
632 633
		}

634 635
		error = _git_ssh_authenticate_session(session, cred);
	}
636

637
	if (error < 0)
638
		goto done;
639

Brad Morgan committed
640
	channel = libssh2_channel_open_session(session);
641
	if (!channel) {
642
		error = -1;
643
		ssh_error(session, "Failed to open SSH channel");
644
		goto done;
645
	}
646

Brad Morgan committed
647
	libssh2_channel_set_blocking(channel, 1);
648

Brad Morgan committed
649 650
	s->session = session;
	s->channel = channel;
651

Brad Morgan committed
652
	t->current_stream = s;
653

654 655
done:
	if (error < 0) {
656
		ssh_stream_free(*stream);
657

658 659 660
		if (session)
			libssh2_session_free(session);
	}
661

662 663 664
	if (cred)
		cred->free(cred);

Brad Morgan committed
665
	git__free(host);
Brad Morgan committed
666
	git__free(port);
667
	git__free(path);
Brad Morgan committed
668 669 670
	git__free(user);
	git__free(pass);

671
	return error;
Brad Morgan committed
672 673
}

Brad Morgan committed
674
static int ssh_uploadpack_ls(
Brad Morgan committed
675 676 677 678
	ssh_subtransport *t,
	const char *url,
	git_smart_subtransport_stream **stream)
{
679 680
	const char *cmd = t->cmd_uploadpack ? t->cmd_uploadpack : cmd_uploadpack;

681
	return _git_ssh_setup_conn(t, url, cmd, stream);
Brad Morgan committed
682 683
}

Brad Morgan committed
684
static int ssh_uploadpack(
Brad Morgan committed
685 686 687 688 689
	ssh_subtransport *t,
	const char *url,
	git_smart_subtransport_stream **stream)
{
	GIT_UNUSED(url);
690

Brad Morgan committed
691 692 693 694
	if (t->current_stream) {
		*stream = &t->current_stream->parent;
		return 0;
	}
695

696
	git_error_set(GIT_ERROR_NET, "must call UPLOADPACK_LS before UPLOADPACK");
Brad Morgan committed
697 698 699
	return -1;
}

Brad Morgan committed
700
static int ssh_receivepack_ls(
Brad Morgan committed
701 702 703 704
	ssh_subtransport *t,
	const char *url,
	git_smart_subtransport_stream **stream)
{
705 706
	const char *cmd = t->cmd_receivepack ? t->cmd_receivepack : cmd_receivepack;

707

708
	return _git_ssh_setup_conn(t, url, cmd, stream);
Brad Morgan committed
709 710
}

Brad Morgan committed
711
static int ssh_receivepack(
Brad Morgan committed
712 713 714
	ssh_subtransport *t,
	const char *url,
	git_smart_subtransport_stream **stream)
Brad Morgan committed
715 716
{
	GIT_UNUSED(url);
717

Brad Morgan committed
718 719 720 721
	if (t->current_stream) {
		*stream = &t->current_stream->parent;
		return 0;
	}
722

723
	git_error_set(GIT_ERROR_NET, "must call RECEIVEPACK_LS before RECEIVEPACK");
Brad Morgan committed
724 725 726
	return -1;
}

Brad Morgan committed
727
static int _ssh_action(
Brad Morgan committed
728 729 730 731
	git_smart_subtransport_stream **stream,
	git_smart_subtransport *subtransport,
	const char *url,
	git_smart_service_t action)
Brad Morgan committed
732 733
{
	ssh_subtransport *t = (ssh_subtransport *) subtransport;
734

Brad Morgan committed
735 736
	switch (action) {
		case GIT_SERVICE_UPLOADPACK_LS:
Brad Morgan committed
737
			return ssh_uploadpack_ls(t, url, stream);
738

Brad Morgan committed
739
		case GIT_SERVICE_UPLOADPACK:
Brad Morgan committed
740
			return ssh_uploadpack(t, url, stream);
741

Brad Morgan committed
742
		case GIT_SERVICE_RECEIVEPACK_LS:
Brad Morgan committed
743
			return ssh_receivepack_ls(t, url, stream);
744

Brad Morgan committed
745
		case GIT_SERVICE_RECEIVEPACK:
Brad Morgan committed
746
			return ssh_receivepack(t, url, stream);
Brad Morgan committed
747
	}
748

Brad Morgan committed
749 750 751 752
	*stream = NULL;
	return -1;
}

Brad Morgan committed
753
static int _ssh_close(git_smart_subtransport *subtransport)
Brad Morgan committed
754 755
{
	ssh_subtransport *t = (ssh_subtransport *) subtransport;
756

Brad Morgan committed
757
	assert(!t->current_stream);
758

Brad Morgan committed
759
	GIT_UNUSED(t);
760

Brad Morgan committed
761 762 763
	return 0;
}

Brad Morgan committed
764
static void _ssh_free(git_smart_subtransport *subtransport)
Brad Morgan committed
765 766
{
	ssh_subtransport *t = (ssh_subtransport *) subtransport;
767

Brad Morgan committed
768
	assert(!t->current_stream);
769

770 771
	git__free(t->cmd_uploadpack);
	git__free(t->cmd_receivepack);
Brad Morgan committed
772 773
	git__free(t);
}
774 775 776 777 778

#define SSH_AUTH_PUBLICKEY "publickey"
#define SSH_AUTH_PASSWORD "password"
#define SSH_AUTH_KEYBOARD_INTERACTIVE "keyboard-interactive"

779
static int list_auth_methods(int *out, LIBSSH2_SESSION *session, const char *username)
780 781 782 783 784
{
	const char *list, *ptr;

	*out = 0;

785
	list = libssh2_userauth_list(session, username, strlen(username));
786 787

	/* either error, or the remote accepts NONE auth, which is bizarre, let's punt */
788 789
	if (list == NULL && !libssh2_userauth_authenticated(session)) {
		ssh_error(session, "Failed to retrieve list of SSH authentication methods");
790
		return -1;
791
	}
792 793 794 795 796 797 798 799 800

	ptr = list;
	while (ptr) {
		if (*ptr == ',')
			ptr++;

		if (!git__prefixcmp(ptr, SSH_AUTH_PUBLICKEY)) {
			*out |= GIT_CREDTYPE_SSH_KEY;
			*out |= GIT_CREDTYPE_SSH_CUSTOM;
801 802 803
#ifdef GIT_SSH_MEMORY_CREDENTIALS
			*out |= GIT_CREDTYPE_SSH_MEMORY;
#endif
804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825
			ptr += strlen(SSH_AUTH_PUBLICKEY);
			continue;
		}

		if (!git__prefixcmp(ptr, SSH_AUTH_PASSWORD)) {
			*out |= GIT_CREDTYPE_USERPASS_PLAINTEXT;
			ptr += strlen(SSH_AUTH_PASSWORD);
			continue;
		}

		if (!git__prefixcmp(ptr, SSH_AUTH_KEYBOARD_INTERACTIVE)) {
			*out |= GIT_CREDTYPE_SSH_INTERACTIVE;
			ptr += strlen(SSH_AUTH_KEYBOARD_INTERACTIVE);
			continue;
		}

		/* Skipt it if we don't know it */
		ptr = strchr(ptr, ',');
	}

	return 0;
}
826
#endif
Brad Morgan committed
827

828
int git_smart_subtransport_ssh(
829
	git_smart_subtransport **out, git_transport *owner, void *param)
Brad Morgan committed
830
{
831
#ifdef GIT_SSH
Brad Morgan committed
832
	ssh_subtransport *t;
833 834 835

	assert(out);

836 837
	GIT_UNUSED(param);

Brad Morgan committed
838
	t = git__calloc(sizeof(ssh_subtransport), 1);
839
	GIT_ERROR_CHECK_ALLOC(t);
840

Brad Morgan committed
841
	t->owner = (transport_smart *)owner;
Brad Morgan committed
842 843 844
	t->parent.action = _ssh_action;
	t->parent.close = _ssh_close;
	t->parent.free = _ssh_free;
845

Brad Morgan committed
846 847
	*out = (git_smart_subtransport *) t;
	return 0;
848 849
#else
	GIT_UNUSED(owner);
850
	GIT_UNUSED(param);
851 852 853

	assert(out);
	*out = NULL;
854

855
	git_error_set(GIT_ERROR_INVALID, "cannot create SSH transport. Library was built without SSH support");
856
	return -1;
857
#endif
858
}
859 860 861 862 863 864 865 866 867 868 869 870

int git_transport_ssh_with_paths(git_transport **out, git_remote *owner, void *payload)
{
#ifdef GIT_SSH
	git_strarray *paths = (git_strarray *) payload;
	git_transport *transport;
	transport_smart *smart;
	ssh_subtransport *t;
	int error;
	git_smart_subtransport_definition ssh_definition = {
		git_smart_subtransport_ssh,
		0, /* no RPC */
871
		NULL,
872 873 874
	};

	if (paths->count != 2) {
875
		git_error_set(GIT_ERROR_SSH, "invalid ssh paths, must be two strings");
876 877 878 879 880 881 882 883 884 885
		return GIT_EINVALIDSPEC;
	}

	if ((error = git_transport_smart(&transport, owner, &ssh_definition)) < 0)
		return error;

	smart = (transport_smart *) transport;
	t = (ssh_subtransport *) smart->wrapped;

	t->cmd_uploadpack = git__strdup(paths->strings[0]);
886
	GIT_ERROR_CHECK_ALLOC(t->cmd_uploadpack);
887
	t->cmd_receivepack = git__strdup(paths->strings[1]);
888
	GIT_ERROR_CHECK_ALLOC(t->cmd_receivepack);
889 890 891 892 893

	*out = transport;
	return 0;
#else
	GIT_UNUSED(owner);
Vicent Marti committed
894
	GIT_UNUSED(payload);
895 896 897 898

	assert(out);
	*out = NULL;

899
	git_error_set(GIT_ERROR_INVALID, "cannot create SSH transport. Library was built without SSH support");
900 901 902
	return -1;
#endif
}
903

904 905 906 907 908 909 910
#ifdef GIT_SSH
static void shutdown_ssh(void)
{
    libssh2_exit();
}
#endif

911 912 913
int git_transport_ssh_global_init(void)
{
#ifdef GIT_SSH
914
	if (libssh2_init(0) < 0) {
915
		git_error_set(GIT_ERROR_SSH, "unable to initialize libssh2");
916 917
		return -1;
	}
918

919
	git__on_shutdown(shutdown_ssh);
920 921 922 923 924 925 926 927 928
	return 0;

#else

	/* Nothing to initialize */
	return 0;

#endif
}