/
niceSOFT
/
libssh2
Обзор
Документация
Войти
/
niceSOFT
/
libssh2
Код
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
libssh2-1.2.2
src/knownhost.c
843 строки
24 KB
Daniel Stenberg
Fix knownhost checking of non-hashed hosts
12 июл 2009, 09:05
12 июл 2009, 09:05
2361f50
Код
Авторство
О чём код?
/* * 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_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, const char *host, const 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 *orig = host; 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 host = orig; } 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; } if (!salt) host = orig; 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; } /* we report the full length of the data with the trailing zero excluded */ *outlen = nlen-1; 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; }