1 /* $OpenBSD: sshbuf-misc.c,v 1.6 2016/05/02 08:49:03 djm Exp $ */ 2 /* 3 * Copyright (c) 2011 Damien Miller 4 * 5 * Permission to use, copy, modify, and distribute this software for any 6 * purpose with or without fee is hereby granted, provided that the above 7 * copyright notice and this permission notice appear in all copies. 8 * 9 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES 10 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF 11 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR 12 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES 13 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN 14 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF 15 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. 16 */ 17 18 #include "includes.h" 19 20 #include <sys/types.h> 21 #include <sys/socket.h> 22 #include <netinet/in.h> 23 #include <errno.h> 24 #include <stdlib.h> 25 #ifdef HAVE_STDINT_H 26 #include <stdint.h> 27 #endif 28 #include <stdio.h> 29 #include <limits.h> 30 #include <string.h> 31 #include <resolv.h> 32 #include <ctype.h> 33 34 #include "ssherr.h" 35 #define SSHBUF_INTERNAL 36 #include "sshbuf.h" 37 38 void 39 sshbuf_dump_data(const void *s, size_t len, FILE *f) 40 { 41 size_t i, j; 42 const u_char *p = (const u_char *)s; 43 44 for (i = 0; i < len; i += 16) { 45 fprintf(f, "%.4zu: ", i); 46 for (j = i; j < i + 16; j++) { 47 if (j < len) 48 fprintf(f, "%02x ", p[j]); 49 else 50 fprintf(f, " "); 51 } 52 fprintf(f, " "); 53 for (j = i; j < i + 16; j++) { 54 if (j < len) { 55 if (isascii(p[j]) && isprint(p[j])) 56 fprintf(f, "%c", p[j]); 57 else 58 fprintf(f, "."); 59 } 60 } 61 fprintf(f, "\n"); 62 } 63 } 64 65 void 66 sshbuf_dump(struct sshbuf *buf, FILE *f) 67 { 68 fprintf(f, "buffer %p len = %zu\n", buf, sshbuf_len(buf)); 69 sshbuf_dump_data(sshbuf_ptr(buf), sshbuf_len(buf), f); 70 } 71 72 char * 73 sshbuf_dtob16(struct sshbuf *buf) 74 { 75 size_t i, j, len = sshbuf_len(buf); 76 const u_char *p = sshbuf_ptr(buf); 77 char *ret; 78 const char hex[] = "0123456789abcdef"; 79 80 if (len == 0) 81 return strdup(""); 82 if (SIZE_MAX / 2 <= len || (ret = malloc(len * 2 + 1)) == NULL) 83 return NULL; 84 for (i = j = 0; i < len; i++) { 85 ret[j++] = hex[(p[i] >> 4) & 0xf]; 86 ret[j++] = hex[p[i] & 0xf]; 87 } 88 ret[j] = '\0'; 89 return ret; 90 } 91 92 char * 93 sshbuf_dtob64(struct sshbuf *buf) 94 { 95 size_t len = sshbuf_len(buf), plen; 96 const u_char *p = sshbuf_ptr(buf); 97 char *ret; 98 int r; 99 100 if (len == 0) 101 return strdup(""); 102 plen = ((len + 2) / 3) * 4 + 1; 103 if (SIZE_MAX / 2 <= len || (ret = malloc(plen)) == NULL) 104 return NULL; 105 if ((r = b64_ntop(p, len, ret, plen)) == -1) { 106 explicit_bzero(ret, plen); 107 free(ret); 108 return NULL; 109 } 110 return ret; 111 } 112 113 int 114 sshbuf_b64tod(struct sshbuf *buf, const char *b64) 115 { 116 size_t plen = strlen(b64); 117 int nlen, r; 118 u_char *p; 119 120 if (plen == 0) 121 return 0; 122 if ((p = malloc(plen)) == NULL) 123 return SSH_ERR_ALLOC_FAIL; 124 if ((nlen = b64_pton(b64, p, plen)) < 0) { 125 explicit_bzero(p, plen); 126 free(p); 127 return SSH_ERR_INVALID_FORMAT; 128 } 129 if ((r = sshbuf_put(buf, p, nlen)) < 0) { 130 explicit_bzero(p, plen); 131 free(p); 132 return r; 133 } 134 explicit_bzero(p, plen); 135 free(p); 136 return 0; 137 } 138 139 char * 140 sshbuf_dup_string(struct sshbuf *buf) 141 { 142 const u_char *p = NULL, *s = sshbuf_ptr(buf); 143 size_t l = sshbuf_len(buf); 144 char *r; 145 146 if (s == NULL || l > SIZE_MAX) 147 return NULL; 148 /* accept a nul only as the last character in the buffer */ 149 if (l > 0 && (p = memchr(s, '\0', l)) != NULL) { 150 if (p != s + l - 1) 151 return NULL; 152 l--; /* the nul is put back below */ 153 } 154 if ((r = malloc(l + 1)) == NULL) 155 return NULL; 156 if (l > 0) 157 memcpy(r, s, l); 158 r[l] = '\0'; 159 return r; 160 } 161 162