buffer.c revision 295367 
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
29void
30buffer_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
38void *
39buffer_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
49int
50buffer_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
61int
62buffer_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
73void
74buffer_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
80int
81buffer_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
92void
93buffer_consume(Buffer *buffer, u_int bytes)
94{
95	if (buffer_consume_ret(buffer, bytes) == -1)
96		fatal("%s: buffer error", __func__);
97}
98
99int
100buffer_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
111void
112buffer_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