
application can get a pointer back to the internal representation of the host it just added. Useful for example when the app wants to add a host, and then convert that exact same host to a line for storing in a known host file. 'store' can also be set to NULL to simple not care.
839 lines
24 KiB
C
839 lines
24 KiB
C
/*
|
|
* Copyright (c) 2009 by Daniel Stenberg
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms,
|
|
* with or without modification, are permitted provided
|
|
* that the following conditions are met:
|
|
*
|
|
* Redistributions of source code must retain the above
|
|
* copyright notice, this list of conditions and the
|
|
* following disclaimer.
|
|
*
|
|
* Redistributions in binary form must reproduce the above
|
|
* copyright notice, this list of conditions and the following
|
|
* disclaimer in the documentation and/or other materials
|
|
* provided with the distribution.
|
|
*
|
|
* Neither the name of the copyright holder nor the names
|
|
* of any other contributors may be used to endorse or
|
|
* promote products derived from this software without
|
|
* specific prior written permission.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
|
|
* CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
|
|
* INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
|
|
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
|
|
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
|
|
* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
|
|
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
|
|
* WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
|
* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
|
|
* USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY
|
|
* OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#include "libssh2.h"
|
|
#include "libssh2_priv.h"
|
|
#include "misc.h"
|
|
|
|
struct known_host {
|
|
struct list_node node;
|
|
char *name; /* points to the name or the hash (allocated) */
|
|
size_t name_len; /* needed for hashed data */
|
|
int typemask; /* plain, sha1, custom, ... */
|
|
char *salt; /* points to binary salt (allocated) */
|
|
size_t salt_len; /* size of salt */
|
|
char *key; /* the (allocated) associated key. This is kept base64
|
|
encoded in memory. */
|
|
|
|
/* this is the struct we expose externally */
|
|
struct libssh2_knownhost external;
|
|
};
|
|
|
|
struct _LIBSSH2_KNOWNHOSTS
|
|
{
|
|
LIBSSH2_SESSION *session; /* the session this "belongs to" */
|
|
struct list_head head;
|
|
};
|
|
|
|
static void free_host(LIBSSH2_SESSION *session, struct known_host *entry)
|
|
{
|
|
if(entry) {
|
|
if(entry->key)
|
|
LIBSSH2_FREE(session, entry->key);
|
|
if(entry->salt)
|
|
LIBSSH2_FREE(session, entry->salt);
|
|
if(entry->name)
|
|
LIBSSH2_FREE(session, entry->name);
|
|
LIBSSH2_FREE(session, entry);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* libssh2_knownhost_init
|
|
*
|
|
* Init a collection of known hosts. Returns the pointer to a collection.
|
|
*
|
|
*/
|
|
LIBSSH2_API LIBSSH2_KNOWNHOSTS *
|
|
libssh2_knownhost_init(LIBSSH2_SESSION *session)
|
|
{
|
|
LIBSSH2_KNOWNHOSTS *knh =
|
|
LIBSSH2_ALLOC(session, sizeof(struct _LIBSSH2_KNOWNHOSTS));
|
|
|
|
if(!knh)
|
|
return NULL;
|
|
|
|
knh->session = session;
|
|
|
|
_libssh2_list_init(&knh->head);
|
|
|
|
return knh;
|
|
}
|
|
|
|
#define KNOWNHOST_MAGIC 0xdeadcafe
|
|
/*
|
|
* knownhost_to_external()
|
|
*
|
|
* Copies data from the internal to the external representation struct.
|
|
*
|
|
*/
|
|
static struct libssh2_knownhost *knownhost_to_external(struct known_host *node)
|
|
{
|
|
struct libssh2_knownhost *ext = &node->external;
|
|
|
|
ext->magic = KNOWNHOST_MAGIC;
|
|
ext->node = node;
|
|
ext->name = ((node->typemask & LIBSSH2_KNOWNHOST_TYPE_MASK) ==
|
|
LIBSSH2_KNOWNHOST_TYPE_PLAIN)? node->name:NULL;
|
|
ext->key = node->key;
|
|
ext->typemask = node->typemask;
|
|
|
|
return ext;
|
|
}
|
|
|
|
/*
|
|
* libssh2_knownhost_add
|
|
*
|
|
* Add a host and its associated key to the collection of known hosts.
|
|
*
|
|
* The 'type' argument specifies on what format the given host and keys are:
|
|
*
|
|
* plain - ascii "hostname.domain.tld"
|
|
* sha1 - SHA1(<salt> <host>) base64-encoded!
|
|
* custom - another hash
|
|
*
|
|
* If 'sha1' is selected as type, the salt must be provided to the salt
|
|
* argument. This too base64 encoded.
|
|
*
|
|
* The SHA-1 hash is what OpenSSH can be told to use in known_hosts files. If
|
|
* a custom type is used, salt is ignored and you must provide the host
|
|
* pre-hashed when checking for it in the libssh2_knownhost_check() function.
|
|
*
|
|
*/
|
|
|
|
LIBSSH2_API int
|
|
libssh2_knownhost_add(LIBSSH2_KNOWNHOSTS *hosts,
|
|
const char *host, const char *salt,
|
|
const char *key, size_t keylen,
|
|
int typemask, struct libssh2_knownhost **store)
|
|
{
|
|
struct known_host *entry =
|
|
LIBSSH2_ALLOC(hosts->session, sizeof(struct known_host));
|
|
size_t hostlen = strlen(host);
|
|
int rc = LIBSSH2_ERROR_ALLOC;
|
|
char *ptr;
|
|
unsigned int ptrlen;
|
|
|
|
if(!entry)
|
|
return rc;
|
|
|
|
if(!(typemask & LIBSSH2_KNOWNHOST_KEY_MASK))
|
|
/* make sure we have a key type set */
|
|
return LIBSSH2_ERROR_INVAL;
|
|
|
|
memset(entry, 0, sizeof(struct known_host));
|
|
|
|
entry->typemask = typemask;
|
|
|
|
switch(entry->typemask & LIBSSH2_KNOWNHOST_TYPE_MASK) {
|
|
case LIBSSH2_KNOWNHOST_TYPE_PLAIN:
|
|
case LIBSSH2_KNOWNHOST_TYPE_CUSTOM:
|
|
entry->name = LIBSSH2_ALLOC(hosts->session, hostlen+1);
|
|
if(!entry)
|
|
goto error;
|
|
memcpy(entry->name, host, hostlen+1);
|
|
break;
|
|
case LIBSSH2_KNOWNHOST_TYPE_SHA1:
|
|
rc = libssh2_base64_decode(hosts->session, &ptr, &ptrlen,
|
|
host, hostlen);
|
|
if(rc)
|
|
goto error;
|
|
entry->name = ptr;
|
|
entry->name_len = ptrlen;
|
|
|
|
rc = libssh2_base64_decode(hosts->session, &ptr, &ptrlen,
|
|
salt, strlen(salt));
|
|
if(rc)
|
|
goto error;
|
|
entry->salt = ptr;
|
|
entry->salt_len = ptrlen;
|
|
break;
|
|
default:
|
|
rc = LIBSSH2_ERROR_METHOD_NOT_SUPPORTED;
|
|
goto error;
|
|
}
|
|
|
|
if(typemask & LIBSSH2_KNOWNHOST_KEYENC_BASE64) {
|
|
/* the provided key is base64 encoded already */
|
|
if(!keylen)
|
|
keylen = strlen(key);
|
|
entry->key = LIBSSH2_ALLOC(hosts->session, keylen+1);
|
|
if(!entry)
|
|
goto error;
|
|
memcpy(entry->key, key, keylen+1);
|
|
entry->key[keylen]=0; /* force a terminating zero trailer */
|
|
}
|
|
else {
|
|
/* key is raw, we base64 encode it and store it as such */
|
|
size_t nlen = _libssh2_base64_encode(hosts->session, key, keylen,
|
|
&ptr);
|
|
if(!nlen)
|
|
goto error;
|
|
|
|
entry->key = ptr;
|
|
}
|
|
|
|
/* add this new host to the big list of known hosts */
|
|
_libssh2_list_add(&hosts->head, &entry->node);
|
|
|
|
if(store)
|
|
*store = knownhost_to_external(entry);
|
|
|
|
return LIBSSH2_ERROR_NONE;
|
|
error:
|
|
free_host(hosts->session, entry);
|
|
return rc;
|
|
}
|
|
|
|
/*
|
|
* libssh2_knownhost_check
|
|
*
|
|
* Check a host and its associated key against the collection of known hosts.
|
|
*
|
|
* The typemask is the type/format of the given host name and key
|
|
*
|
|
* plain - ascii "hostname.domain.tld"
|
|
* sha1 - NOT SUPPORTED AS INPUT
|
|
* custom - prehashed base64 encoded. Note that this cannot use any salts.
|
|
*
|
|
* Returns:
|
|
*
|
|
* LIBSSH2_KNOWNHOST_CHECK_FAILURE
|
|
* LIBSSH2_KNOWNHOST_CHECK_NOTFOUND
|
|
* LIBSSH2_KNOWNHOST_CHECK_MATCH
|
|
* LIBSSH2_KNOWNHOST_CHECK_MISMATCH
|
|
*/
|
|
LIBSSH2_API int
|
|
libssh2_knownhost_check(LIBSSH2_KNOWNHOSTS *hosts,
|
|
char *host, char *key, size_t keylen,
|
|
int typemask,
|
|
struct libssh2_knownhost **ext)
|
|
{
|
|
struct known_host *node = _libssh2_list_first(&hosts->head);
|
|
struct known_host *badkey = NULL;
|
|
int type = typemask & LIBSSH2_KNOWNHOST_TYPE_MASK;
|
|
char *keyalloc = NULL;
|
|
int rc = LIBSSH2_KNOWNHOST_CHECK_NOTFOUND;
|
|
|
|
if(type == LIBSSH2_KNOWNHOST_TYPE_SHA1)
|
|
/* we can't work with a sha1 as given input */
|
|
return LIBSSH2_KNOWNHOST_CHECK_MISMATCH;
|
|
|
|
if(!(typemask & LIBSSH2_KNOWNHOST_KEYENC_BASE64)) {
|
|
/* we got a raw key input, convert it to base64 for the checks below */
|
|
size_t nlen = _libssh2_base64_encode(hosts->session, key, keylen,
|
|
&keyalloc);
|
|
if(!nlen)
|
|
return LIBSSH2_KNOWNHOST_CHECK_FAILURE;
|
|
|
|
/* make the key point to this */
|
|
key = keyalloc;
|
|
keylen = nlen;
|
|
}
|
|
|
|
while (node) {
|
|
int match = 0;
|
|
switch(node->typemask & LIBSSH2_KNOWNHOST_TYPE_MASK) {
|
|
case LIBSSH2_KNOWNHOST_TYPE_PLAIN:
|
|
if(type == LIBSSH2_KNOWNHOST_TYPE_PLAIN)
|
|
match = !strcmp(host, node->name);
|
|
break;
|
|
case LIBSSH2_KNOWNHOST_TYPE_CUSTOM:
|
|
if(type == LIBSSH2_KNOWNHOST_TYPE_CUSTOM)
|
|
match = !strcmp(host, node->name);
|
|
break;
|
|
case LIBSSH2_KNOWNHOST_TYPE_SHA1:
|
|
if(type == LIBSSH2_KNOWNHOST_TYPE_PLAIN) {
|
|
/* when we have the sha1 version stored, we can use a plain
|
|
input to produce a hash to compare with the stored hash.
|
|
*/
|
|
libssh2_hmac_ctx ctx;
|
|
unsigned char hash[SHA_DIGEST_LENGTH];
|
|
|
|
if(SHA_DIGEST_LENGTH != node->name_len) {
|
|
/* the name hash length must be the sha1 size or
|
|
we can't match it */
|
|
break;
|
|
}
|
|
libssh2_hmac_sha1_init(&ctx, node->salt, node->salt_len);
|
|
libssh2_hmac_update(ctx, (unsigned char *)host, strlen(host));
|
|
libssh2_hmac_final(ctx, hash);
|
|
libssh2_hmac_cleanup(&ctx);
|
|
|
|
if(!memcmp(hash, node->name, SHA_DIGEST_LENGTH))
|
|
/* this is a node we're interested in */
|
|
match = 1;
|
|
}
|
|
break;
|
|
default: /* unsupported type */
|
|
break;
|
|
}
|
|
if(match) {
|
|
/* host name match, now compare the keys */
|
|
if(!strcmp(key, node->key)) {
|
|
/* they match! */
|
|
*ext = knownhost_to_external(node);
|
|
badkey = NULL;
|
|
rc = LIBSSH2_KNOWNHOST_CHECK_MATCH;
|
|
break;
|
|
}
|
|
else {
|
|
/* remember the first node that had a host match but a failed
|
|
key match since we continue our search from here */
|
|
if(!badkey)
|
|
badkey = node;
|
|
}
|
|
}
|
|
node= _libssh2_list_next(&node->node);
|
|
}
|
|
|
|
if(badkey) {
|
|
/* key mismatch */
|
|
*ext = knownhost_to_external(badkey);
|
|
rc = LIBSSH2_KNOWNHOST_CHECK_MISMATCH;
|
|
}
|
|
|
|
if(keyalloc)
|
|
LIBSSH2_FREE(hosts->session, keyalloc);
|
|
|
|
return rc;
|
|
}
|
|
|
|
/*
|
|
* libssh2_knownhost_del
|
|
*
|
|
* Remove a host from the collection of known hosts.
|
|
*
|
|
*/
|
|
LIBSSH2_API int
|
|
libssh2_knownhost_del(LIBSSH2_KNOWNHOSTS *hosts,
|
|
struct libssh2_knownhost *entry)
|
|
{
|
|
struct known_host *node;
|
|
if(!entry || (entry->magic != KNOWNHOST_MAGIC))
|
|
/* check that this was retrieved the right way or get out */
|
|
return LIBSSH2_ERROR_INVAL;
|
|
|
|
/* get the internal node pointer */
|
|
node = entry->node;
|
|
|
|
/* unlink from the list of all hosts */
|
|
_libssh2_list_remove(&node->node);
|
|
|
|
/* free all resources */
|
|
free_host(hosts->session, node);
|
|
|
|
/* clear the struct now since this host entry has been removed! */
|
|
memset(entry, 0, sizeof(struct libssh2_knownhost));
|
|
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* libssh2_knownhost_free
|
|
*
|
|
* Free an entire collection of known hosts.
|
|
*
|
|
*/
|
|
LIBSSH2_API void
|
|
libssh2_knownhost_free(LIBSSH2_KNOWNHOSTS *hosts)
|
|
{
|
|
struct known_host *node;
|
|
struct known_host *next;
|
|
|
|
for(node = _libssh2_list_first(&hosts->head); node; node = next) {
|
|
next = _libssh2_list_next(&node->node);
|
|
free_host(hosts->session, node);
|
|
}
|
|
LIBSSH2_FREE(hosts->session, hosts);
|
|
}
|
|
|
|
/*
|
|
* hostline()
|
|
*
|
|
* Parse a single known_host line pre-split into host and key.
|
|
*
|
|
*/
|
|
static int hostline(LIBSSH2_KNOWNHOSTS *hosts,
|
|
const char *host, size_t hostlen,
|
|
const char *key, size_t keylen)
|
|
{
|
|
const char *p;
|
|
const char *salt = NULL;
|
|
int rc;
|
|
int type = LIBSSH2_KNOWNHOST_TYPE_PLAIN;
|
|
const char *sep = NULL;
|
|
size_t seplen = 0;
|
|
char saltbuf[32];
|
|
char hostbuf[256];
|
|
|
|
/* Figure out host format */
|
|
if((hostlen >2) && memcmp(host, "|1|", 3)) {
|
|
/* old style plain text: [name][,][ip-address]
|
|
|
|
for the sake of simplicity, we add them as two hosts with the same
|
|
key
|
|
*/
|
|
size_t scan = hostlen;
|
|
|
|
while(scan && (*host != ',')) {
|
|
host++;
|
|
scan--;
|
|
}
|
|
|
|
if(scan) {
|
|
sep = host+1;
|
|
seplen = scan-1;
|
|
hostlen -= scan; /* deduct what's left to scan from the first
|
|
host name */
|
|
}
|
|
}
|
|
else {
|
|
/* |1|[salt]|[hash] */
|
|
type = LIBSSH2_KNOWNHOST_TYPE_SHA1;
|
|
|
|
salt = &host[3]; /* skip the magic marker */
|
|
hostlen -= 3; /* deduct the marker */
|
|
|
|
/* this is where the salt starts, find the end of it */
|
|
for(p = salt; *p && (*p != '|'); p++)
|
|
;
|
|
|
|
if(*p=='|') {
|
|
const char *hash = NULL;
|
|
size_t saltlen = p - salt;
|
|
if(saltlen >= (sizeof(saltbuf)-1))
|
|
return LIBSSH2_ERROR_METHOD_NOT_SUPPORTED; /* weird length */
|
|
|
|
memcpy(saltbuf, salt, saltlen);
|
|
saltbuf[saltlen] = 0; /* zero terminate */
|
|
salt = saltbuf; /* point to the stack based buffer */
|
|
|
|
hash = p+1; /* the host hash is after the separator */
|
|
|
|
/* now make the host point to the hash */
|
|
host = hash;
|
|
hostlen -= saltlen+1; /* deduct the salt and separator */
|
|
}
|
|
else
|
|
return 0;
|
|
}
|
|
|
|
/* make some checks that the lenghts seem sensible */
|
|
if((keylen < 20) ||
|
|
(seplen >= sizeof(hostbuf)-1) ||
|
|
(hostlen >= sizeof(hostbuf)-1))
|
|
return LIBSSH2_ERROR_METHOD_NOT_SUPPORTED;
|
|
|
|
switch(key[0]) {
|
|
case '0': case '1': case '2': case '3': case '4':
|
|
case '5': case '6': case '7': case '8': case '9':
|
|
type |= LIBSSH2_KNOWNHOST_KEY_RSA1;
|
|
|
|
/* Note that the old-style keys (RSA1) aren't truly base64, but we
|
|
* claim it is for now since we can get away with strcmp()ing the
|
|
* entire anything anyway! We need to check and fix these to make them
|
|
* work properly.
|
|
*/
|
|
break;
|
|
|
|
case 's': /* ssh-dss or ssh-rsa */
|
|
if(!strncmp(key, "ssh-dss", 7))
|
|
type |= LIBSSH2_KNOWNHOST_KEY_SSHDSS;
|
|
else if(!strncmp(key, "ssh-rsa", 7))
|
|
type |= LIBSSH2_KNOWNHOST_KEY_SSHRSA;
|
|
else
|
|
return LIBSSH2_ERROR_METHOD_NOT_SUPPORTED; /* unknown key type */
|
|
|
|
key += 7;
|
|
keylen -= 7;
|
|
|
|
/* skip whitespaces */
|
|
while((*key ==' ') || (*key == '\t')) {
|
|
key++;
|
|
keylen--;
|
|
}
|
|
break;
|
|
|
|
default: /* unknown key format */
|
|
return LIBSSH2_ERROR_METHOD_NOT_SUPPORTED;
|
|
}
|
|
|
|
if(sep) {
|
|
/* The second host after the comma, add this first. Copy it to the
|
|
temp buffer and zero terminate */
|
|
memcpy(hostbuf, sep, seplen);
|
|
hostbuf[seplen]=0;
|
|
|
|
rc = libssh2_knownhost_add(hosts, hostbuf, salt, key, keylen,
|
|
type | LIBSSH2_KNOWNHOST_KEYENC_BASE64,
|
|
NULL);
|
|
if(rc)
|
|
return rc;
|
|
}
|
|
|
|
memcpy(hostbuf, host, hostlen);
|
|
hostbuf[hostlen]=0;
|
|
|
|
rc = libssh2_knownhost_add(hosts, hostbuf, salt, key, keylen,
|
|
type | LIBSSH2_KNOWNHOST_KEYENC_BASE64,
|
|
NULL);
|
|
return rc;
|
|
}
|
|
|
|
/*
|
|
* libssh2_knownhost_readline()
|
|
*
|
|
* Pass in a line of a file of 'type'.
|
|
*
|
|
* LIBSSH2_KNOWNHOST_FILE_OPENSSH is the only supported type.
|
|
*
|
|
* OpenSSH line format:
|
|
*
|
|
* <host> <key>
|
|
*
|
|
* Where the two parts can be created like:
|
|
*
|
|
* <host> can be either
|
|
* <name> or <hash>
|
|
*
|
|
* <name> consists of
|
|
* [name,address] or just [name] or just [address]
|
|
*
|
|
* <hash> consists of
|
|
* |1|<salt>|hash
|
|
*
|
|
* <key> can be one of:
|
|
* [RSA bits] [e] [n as a decimal number]
|
|
* 'ssh-dss' [base64-encoded-key]
|
|
* 'ssh-rsa' [base64-encoded-key]
|
|
*
|
|
*/
|
|
LIBSSH2_API int
|
|
libssh2_knownhost_readline(LIBSSH2_KNOWNHOSTS *hosts,
|
|
const char *line, size_t len, int type)
|
|
{
|
|
const char *cp;
|
|
const char *hostp;
|
|
const char *keyp;
|
|
size_t hostlen;
|
|
size_t keylen;
|
|
int rc;
|
|
|
|
if(type != LIBSSH2_KNOWNHOST_FILE_OPENSSH)
|
|
return LIBSSH2_ERROR_METHOD_NOT_SUPPORTED;
|
|
|
|
cp = line;
|
|
|
|
/* skip leading whitespaces */
|
|
while(len && ((*cp==' ') || (*cp == '\t'))) {
|
|
cp++;
|
|
len--;
|
|
}
|
|
|
|
if(!len || !*cp || (*cp == '#') || (*cp == '\n'))
|
|
/* comment or empty line */
|
|
return LIBSSH2_ERROR_NONE;
|
|
|
|
/* the host part starts here */
|
|
hostp = cp;
|
|
|
|
/* move over the host to the separator */
|
|
while(len && *cp && (*cp!=' ') && (*cp != '\t')) {
|
|
cp++;
|
|
len--;
|
|
}
|
|
|
|
hostlen = cp - hostp;
|
|
|
|
/* the key starts after the whitespaces */
|
|
while(len && *cp && ((*cp==' ') || (*cp == '\t'))) {
|
|
cp++;
|
|
len--;
|
|
}
|
|
|
|
if(!*cp || !len)
|
|
/* illegal line */
|
|
return LIBSSH2_ERROR_METHOD_NOT_SUPPORTED;
|
|
|
|
keyp = cp; /* the key starts here */
|
|
keylen = len;
|
|
|
|
/* check if the line (key) ends with a newline and if so kill it */
|
|
while(len && *cp && (*cp != '\n')) {
|
|
cp++;
|
|
len--;
|
|
}
|
|
|
|
/* zero terminate where the newline is */
|
|
if(*cp == '\n')
|
|
keylen--; /* don't include this in the count */
|
|
|
|
/* deal with this one host+key line */
|
|
rc = hostline(hosts, hostp, hostlen, keyp, keylen);
|
|
if(rc)
|
|
return rc; /* failed */
|
|
|
|
return LIBSSH2_ERROR_NONE; /* success */
|
|
}
|
|
|
|
/*
|
|
* libssh2_knownhost_readfile
|
|
*
|
|
* Read hosts+key pairs from a given file.
|
|
*
|
|
* Returns a negative value for error or number of successfully added hosts.
|
|
*
|
|
*/
|
|
|
|
LIBSSH2_API int
|
|
libssh2_knownhost_readfile(LIBSSH2_KNOWNHOSTS *hosts,
|
|
const char *filename, int type)
|
|
{
|
|
FILE *file;
|
|
int num = 0;
|
|
char buf[2048];
|
|
|
|
if(type != LIBSSH2_KNOWNHOST_FILE_OPENSSH)
|
|
return LIBSSH2_ERROR_METHOD_NOT_SUPPORTED;
|
|
|
|
file = fopen(filename, "r");
|
|
if(file) {
|
|
while(fgets(buf, sizeof(buf), file)) {
|
|
if(libssh2_knownhost_readline(hosts, buf, strlen(buf), type))
|
|
break;
|
|
num++;
|
|
}
|
|
fclose(file);
|
|
}
|
|
else
|
|
return LIBSSH2_ERROR_FILE;
|
|
return num;
|
|
}
|
|
|
|
/*
|
|
* knownhost_writeline()
|
|
*
|
|
* Ask libssh2 to convert a known host to an output line for storage.
|
|
*
|
|
* Note that this function returns LIBSSH2_ERROR_BUFFER_TOO_SMALL if the given
|
|
* output buffer is too small to hold the desired output. The 'outlen' field
|
|
* will then contain the size libssh2 wanted to store, which then is the
|
|
* smallest sufficient buffer it would require.
|
|
*
|
|
*/
|
|
static int
|
|
knownhost_writeline(LIBSSH2_KNOWNHOSTS *hosts,
|
|
struct known_host *node,
|
|
char *buf, size_t buflen,
|
|
size_t *outlen, int type)
|
|
{
|
|
int rc = LIBSSH2_ERROR_NONE;
|
|
int tindex;
|
|
const char *keytypes[4]={
|
|
"", /* not used */
|
|
"", /* this type has no name in the file */
|
|
" ssh-rsa",
|
|
" ssh-dss"
|
|
};
|
|
const char *keytype;
|
|
size_t nlen;
|
|
|
|
/* we only support this single file type for now, bail out on all other
|
|
attempts */
|
|
if(type != LIBSSH2_KNOWNHOST_FILE_OPENSSH)
|
|
return LIBSSH2_ERROR_METHOD_NOT_SUPPORTED;
|
|
|
|
tindex = (node->typemask & LIBSSH2_KNOWNHOST_KEY_MASK) >>
|
|
LIBSSH2_KNOWNHOST_KEY_SHIFT;
|
|
|
|
/* set the string used in the file */
|
|
keytype = keytypes[tindex];
|
|
|
|
if((node->typemask & LIBSSH2_KNOWNHOST_TYPE_MASK) ==
|
|
LIBSSH2_KNOWNHOST_TYPE_SHA1) {
|
|
char *namealloc;
|
|
char *saltalloc;
|
|
nlen = _libssh2_base64_encode(hosts->session, node->name,
|
|
node->name_len, &namealloc);
|
|
if(!nlen)
|
|
return LIBSSH2_ERROR_ALLOC;
|
|
|
|
nlen = _libssh2_base64_encode(hosts->session,
|
|
node->salt, node->salt_len,
|
|
&saltalloc);
|
|
if(!nlen) {
|
|
free(namealloc);
|
|
return LIBSSH2_ERROR_ALLOC;
|
|
}
|
|
|
|
nlen = strlen(saltalloc) + strlen(namealloc) + strlen(keytype) +
|
|
strlen(node->key) + 7; /* |1| + | + ' ' + \n + \0 = 7 */
|
|
|
|
if(nlen <= buflen)
|
|
sprintf(buf, "|1|%s|%s%s %s\n", saltalloc, namealloc, keytype,
|
|
node->key);
|
|
else
|
|
rc = LIBSSH2_ERROR_BUFFER_TOO_SMALL;
|
|
|
|
free(namealloc);
|
|
free(saltalloc);
|
|
}
|
|
else {
|
|
nlen = strlen(node->name) + strlen(keytype) + strlen(node->key) + 3;
|
|
/* ' ' + '\n' + \0 = 3 */
|
|
if(nlen < buflen)
|
|
/* these types have the plain name */
|
|
sprintf(buf, "%s%s %s\n", node->name, keytype, node->key);
|
|
else
|
|
rc = LIBSSH2_ERROR_BUFFER_TOO_SMALL;
|
|
}
|
|
|
|
*outlen = nlen;
|
|
|
|
return rc;
|
|
}
|
|
|
|
/*
|
|
* libssh2_knownhost_writeline()
|
|
*
|
|
* Ask libssh2 to convert a known host to an output line for storage.
|
|
*
|
|
* Note that this function returns LIBSSH2_ERROR_BUFFER_TOO_SMALL if the given
|
|
* output buffer is too small to hold the desired output.
|
|
*/
|
|
LIBSSH2_API int
|
|
libssh2_knownhost_writeline(LIBSSH2_KNOWNHOSTS *hosts,
|
|
struct libssh2_knownhost *known,
|
|
char *buffer, size_t buflen,
|
|
size_t *outlen, /* the amount of written data */
|
|
int type)
|
|
{
|
|
struct known_host *node;
|
|
|
|
if(known->magic != KNOWNHOST_MAGIC)
|
|
return LIBSSH2_ERROR_INVAL;
|
|
|
|
node = known->node;
|
|
|
|
return knownhost_writeline(hosts, node, buffer, buflen, outlen, type);
|
|
}
|
|
|
|
/*
|
|
* libssh2_knownhost_writefile()
|
|
*
|
|
* Write hosts+key pairs to the given file.
|
|
*/
|
|
LIBSSH2_API int
|
|
libssh2_knownhost_writefile(LIBSSH2_KNOWNHOSTS *hosts,
|
|
const char *filename, int type)
|
|
{
|
|
struct known_host *node;
|
|
FILE *file;
|
|
int rc = LIBSSH2_ERROR_NONE;
|
|
char buffer[2048];
|
|
|
|
/* we only support this single file type for now, bail out on all other
|
|
attempts */
|
|
if(type != LIBSSH2_KNOWNHOST_FILE_OPENSSH)
|
|
return LIBSSH2_ERROR_METHOD_NOT_SUPPORTED;
|
|
|
|
file = fopen(filename, "w");
|
|
if(!file)
|
|
return LIBSSH2_ERROR_FILE;
|
|
|
|
for(node = _libssh2_list_first(&hosts->head);
|
|
node;
|
|
node= _libssh2_list_next(&node->node) ) {
|
|
size_t wrote;
|
|
size_t nwrote;
|
|
rc = knownhost_writeline(hosts, node, buffer, sizeof(buffer), &wrote,
|
|
type);
|
|
if(rc)
|
|
break;
|
|
|
|
nwrote = fwrite(buffer, 1, wrote, file);
|
|
if(nwrote != wrote) {
|
|
/* failed to write the whole thing, bail out */
|
|
rc = LIBSSH2_ERROR_FILE;
|
|
break;
|
|
}
|
|
}
|
|
fclose(file);
|
|
|
|
return rc;
|
|
}
|
|
|
|
|
|
/*
|
|
* libssh2_knownhost_get()
|
|
*
|
|
* Traverse the internal list of known hosts. Pass NULL to 'prev' to get
|
|
* the first one.
|
|
*
|
|
* Returns:
|
|
* 0 if a fine host was stored in 'store'
|
|
* 1 if end of hosts
|
|
* [negative] on errors
|
|
*/
|
|
LIBSSH2_API int
|
|
libssh2_knownhost_get(LIBSSH2_KNOWNHOSTS *hosts,
|
|
struct libssh2_knownhost **ext,
|
|
struct libssh2_knownhost *oprev)
|
|
{
|
|
struct known_host *node;
|
|
if(oprev && oprev->node) {
|
|
/* we have a starting point */
|
|
struct known_host *prev = oprev->node;
|
|
|
|
/* get the next node in the list */
|
|
node = _libssh2_list_next(&prev->node);
|
|
|
|
}
|
|
else
|
|
node = _libssh2_list_first(&hosts->head);
|
|
|
|
if(!node)
|
|
/* no (more) node */
|
|
return 1;
|
|
|
|
*ext = knownhost_to_external(node);
|
|
|
|
return 0;
|
|
}
|