• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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