1 /* $OpenBSD: bufaux.c,v 1.60 2014/04/30 05:29:56 djm Exp $ */
2 /*
3 * Copyright (c) 2012 Damien Miller <djm@mindrot.org>
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 /* Emulation wrappers for legacy OpenSSH buffer API atop sshbuf */
19
20 #include "includes.h"
21
22 #include <sys/types.h>
23
24 #include "buffer.h"
25 #include "log.h"
26 #include "ssherr.h"
27
28 int
buffer_get_short_ret(u_short * v,Buffer * buffer)29 buffer_get_short_ret(u_short *v, Buffer *buffer)
30 {
31 int ret;
32
33 if ((ret = sshbuf_get_u16(buffer, v)) != 0) {
34 ssh_error("%s: %s", __func__, ssh_err(ret));
35 return -1;
36 }
37 return 0;
38 }
39
40 u_short
buffer_get_short(Buffer * buffer)41 buffer_get_short(Buffer *buffer)
42 {
43 u_short ret;
44
45 if (buffer_get_short_ret(&ret, buffer) == -1)
46 fatal("%s: buffer error", __func__);
47
48 return (ret);
49 }
50
51 int
buffer_get_int_ret(u_int * v,Buffer * buffer)52 buffer_get_int_ret(u_int *v, Buffer *buffer)
53 {
54 int ret;
55
56 if ((ret = sshbuf_get_u32(buffer, v)) != 0) {
57 ssh_error("%s: %s", __func__, ssh_err(ret));
58 return -1;
59 }
60 return 0;
61 }
62
63 u_int
buffer_get_int(Buffer * buffer)64 buffer_get_int(Buffer *buffer)
65 {
66 u_int ret;
67
68 if (buffer_get_int_ret(&ret, buffer) == -1)
69 fatal("%s: buffer error", __func__);
70
71 return (ret);
72 }
73
74 int
buffer_get_int64_ret(u_int64_t * v,Buffer * buffer)75 buffer_get_int64_ret(u_int64_t *v, Buffer *buffer)
76 {
77 int ret;
78
79 if ((ret = sshbuf_get_u64(buffer, v)) != 0) {
80 ssh_error("%s: %s", __func__, ssh_err(ret));
81 return -1;
82 }
83 return 0;
84 }
85
86 u_int64_t
buffer_get_int64(Buffer * buffer)87 buffer_get_int64(Buffer *buffer)
88 {
89 u_int64_t ret;
90
91 if (buffer_get_int64_ret(&ret, buffer) == -1)
92 fatal("%s: buffer error", __func__);
93
94 return (ret);
95 }
96
97 void
buffer_put_short(Buffer * buffer,u_short value)98 buffer_put_short(Buffer *buffer, u_short value)
99 {
100 int ret;
101
102 if ((ret = sshbuf_put_u16(buffer, value)) != 0)
103 fatal("%s: %s", __func__, ssh_err(ret));
104 }
105
106 void
buffer_put_int(Buffer * buffer,u_int value)107 buffer_put_int(Buffer *buffer, u_int value)
108 {
109 int ret;
110
111 if ((ret = sshbuf_put_u32(buffer, value)) != 0)
112 fatal("%s: %s", __func__, ssh_err(ret));
113 }
114
115 void
buffer_put_int64(Buffer * buffer,u_int64_t value)116 buffer_put_int64(Buffer *buffer, u_int64_t value)
117 {
118 int ret;
119
120 if ((ret = sshbuf_put_u64(buffer, value)) != 0)
121 fatal("%s: %s", __func__, ssh_err(ret));
122 }
123
124 void *
buffer_get_string_ret(Buffer * buffer,u_int * length_ptr)125 buffer_get_string_ret(Buffer *buffer, u_int *length_ptr)
126 {
127 size_t len;
128 int ret;
129 u_char *value;
130
131 if ((ret = sshbuf_get_string(buffer, &value, &len)) != 0) {
132 ssh_error("%s: %s", __func__, ssh_err(ret));
133 return NULL;
134 }
135 if (length_ptr != NULL)
136 *length_ptr = len; /* Safe: sshbuf never stores len > 2^31 */
137 return value;
138 }
139
140 void *
buffer_get_string(Buffer * buffer,u_int * length_ptr)141 buffer_get_string(Buffer *buffer, u_int *length_ptr)
142 {
143 void *ret;
144
145 if ((ret = buffer_get_string_ret(buffer, length_ptr)) == NULL)
146 fatal("%s: buffer error", __func__);
147 return (ret);
148 }
149
150 char *
buffer_get_cstring_ret(Buffer * buffer,u_int * length_ptr)151 buffer_get_cstring_ret(Buffer *buffer, u_int *length_ptr)
152 {
153 size_t len;
154 int ret;
155 char *value;
156
157 if ((ret = sshbuf_get_cstring(buffer, &value, &len)) != 0) {
158 ssh_error("%s: %s", __func__, ssh_err(ret));
159 return NULL;
160 }
161 if (length_ptr != NULL)
162 *length_ptr = len; /* Safe: sshbuf never stores len > 2^31 */
163 return value;
164 }
165
166 char *
buffer_get_cstring(Buffer * buffer,u_int * length_ptr)167 buffer_get_cstring(Buffer *buffer, u_int *length_ptr)
168 {
169 char *ret;
170
171 if ((ret = buffer_get_cstring_ret(buffer, length_ptr)) == NULL)
172 fatal("%s: buffer error", __func__);
173 return ret;
174 }
175
176 const void *
buffer_get_string_ptr_ret(Buffer * buffer,u_int * length_ptr)177 buffer_get_string_ptr_ret(Buffer *buffer, u_int *length_ptr)
178 {
179 size_t len;
180 int ret;
181 const u_char *value;
182
183 if ((ret = sshbuf_get_string_direct(buffer, &value, &len)) != 0) {
184 ssh_error("%s: %s", __func__, ssh_err(ret));
185 return NULL;
186 }
187 if (length_ptr != NULL)
188 *length_ptr = len; /* Safe: sshbuf never stores len > 2^31 */
189 return value;
190 }
191
192 const void *
buffer_get_string_ptr(Buffer * buffer,u_int * length_ptr)193 buffer_get_string_ptr(Buffer *buffer, u_int *length_ptr)
194 {
195 const void *ret;
196
197 if ((ret = buffer_get_string_ptr_ret(buffer, length_ptr)) == NULL)
198 fatal("%s: buffer error", __func__);
199 return (ret);
200 }
201
202 void
buffer_put_string(Buffer * buffer,const void * buf,u_int len)203 buffer_put_string(Buffer *buffer, const void *buf, u_int len)
204 {
205 int ret;
206
207 if ((ret = sshbuf_put_string(buffer, buf, len)) != 0)
208 fatal("%s: %s", __func__, ssh_err(ret));
209 }
210
211 void
buffer_put_cstring(Buffer * buffer,const char * s)212 buffer_put_cstring(Buffer *buffer, const char *s)
213 {
214 int ret;
215
216 if ((ret = sshbuf_put_cstring(buffer, s)) != 0)
217 fatal("%s: %s", __func__, ssh_err(ret));
218 }
219
220 int
buffer_get_char_ret(char * v,Buffer * buffer)221 buffer_get_char_ret(char *v, Buffer *buffer)
222 {
223 int ret;
224
225 if ((ret = sshbuf_get_u8(buffer, (u_char *)v)) != 0) {
226 ssh_error("%s: %s", __func__, ssh_err(ret));
227 return -1;
228 }
229 return 0;
230 }
231
232 int
buffer_get_char(Buffer * buffer)233 buffer_get_char(Buffer *buffer)
234 {
235 char ch;
236
237 if (buffer_get_char_ret(&ch, buffer) == -1)
238 fatal("%s: buffer error", __func__);
239 return (u_char) ch;
240 }
241
242 void
buffer_put_char(Buffer * buffer,int value)243 buffer_put_char(Buffer *buffer, int value)
244 {
245 int ret;
246
247 if ((ret = sshbuf_put_u8(buffer, value)) != 0)
248 fatal("%s: %s", __func__, ssh_err(ret));
249 }
250
251 void
buffer_put_bignum2_from_string(Buffer * buffer,const u_char * s,u_int l)252 buffer_put_bignum2_from_string(Buffer *buffer, const u_char *s, u_int l)
253 {
254 int ret;
255
256 if ((ret = sshbuf_put_bignum2_bytes(buffer, s, l)) != 0)
257 fatal("%s: %s", __func__, ssh_err(ret));
258 }
259
260