1 /* $OpenBSD: buffer.c,v 1.36 2014/04/30 05:29:56 djm Exp $ */
2
3 /*
4 * Copyright (c) 2012 Damien Miller <djm@mindrot.org>
5 *
6 * Permission to use, copy, modify, and distribute this software for any
7 * purpose with or without fee is hereby granted, provided that the above
8 * copyright notice and this permission notice appear in all copies.
9 *
10 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17 */
18
19 /* Emulation wrappers for legacy OpenSSH buffer API atop sshbuf */
20
21 #include "includes.h"
22
23 #include <sys/types.h>
24
25 #include "buffer.h"
26 #include "log.h"
27 #include "ssherr.h"
28
29 void
buffer_append(Buffer * buffer,const void * data,u_int len)30 buffer_append(Buffer *buffer, const void *data, u_int len)
31 {
32 int ret;
33
34 if ((ret = sshbuf_put(buffer, data, len)) != 0)
35 fatal("%s: %s", __func__, ssh_err(ret));
36 }
37
38 void *
buffer_append_space(Buffer * buffer,u_int len)39 buffer_append_space(Buffer *buffer, u_int len)
40 {
41 int ret;
42 u_char *p;
43
44 if ((ret = sshbuf_reserve(buffer, len, &p)) != 0)
45 fatal("%s: %s", __func__, ssh_err(ret));
46 return p;
47 }
48
49 int
buffer_check_alloc(Buffer * buffer,u_int len)50 buffer_check_alloc(Buffer *buffer, u_int len)
51 {
52 int ret = sshbuf_check_reserve(buffer, len);
53
54 if (ret == 0)
55 return 1;
56 if (ret == SSH_ERR_NO_BUFFER_SPACE)
57 return 0;
58 fatal("%s: %s", __func__, ssh_err(ret));
59 }
60
61 int
buffer_get_ret(Buffer * buffer,void * buf,u_int len)62 buffer_get_ret(Buffer *buffer, void *buf, u_int len)
63 {
64 int ret;
65
66 if ((ret = sshbuf_get(buffer, buf, len)) != 0) {
67 error("%s: %s", __func__, ssh_err(ret));
68 return -1;
69 }
70 return 0;
71 }
72
73 void
buffer_get(Buffer * buffer,void * buf,u_int len)74 buffer_get(Buffer *buffer, void *buf, u_int len)
75 {
76 if (buffer_get_ret(buffer, buf, len) == -1)
77 fatal("%s: buffer error", __func__);
78 }
79
80 int
buffer_consume_ret(Buffer * buffer,u_int bytes)81 buffer_consume_ret(Buffer *buffer, u_int bytes)
82 {
83 int ret = sshbuf_consume(buffer, bytes);
84
85 if (ret == 0)
86 return 0;
87 if (ret == SSH_ERR_MESSAGE_INCOMPLETE)
88 return -1;
89 fatal("%s: %s", __func__, ssh_err(ret));
90 }
91
92 void
buffer_consume(Buffer * buffer,u_int bytes)93 buffer_consume(Buffer *buffer, u_int bytes)
94 {
95 if (buffer_consume_ret(buffer, bytes) == -1)
96 fatal("%s: buffer error", __func__);
97 }
98
99 int
buffer_consume_end_ret(Buffer * buffer,u_int bytes)100 buffer_consume_end_ret(Buffer *buffer, u_int bytes)
101 {
102 int ret = sshbuf_consume_end(buffer, bytes);
103
104 if (ret == 0)
105 return 0;
106 if (ret == SSH_ERR_MESSAGE_INCOMPLETE)
107 return -1;
108 fatal("%s: %s", __func__, ssh_err(ret));
109 }
110
111 void
buffer_consume_end(Buffer * buffer,u_int bytes)112 buffer_consume_end(Buffer *buffer, u_int bytes)
113 {
114 if (buffer_consume_end_ret(buffer, bytes) == -1)
115 fatal("%s: buffer error", __func__);
116 }
117
118
119