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
sshbuf_dump_data(const void * s,size_t len,FILE * f)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
sshbuf_dump(struct sshbuf * buf,FILE * f)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 *
sshbuf_dtob16(struct sshbuf * buf)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 *
sshbuf_dtob64(struct sshbuf * buf)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
sshbuf_b64tod(struct sshbuf * buf,const char * b64)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 *
sshbuf_dup_string(struct sshbuf * buf)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