]>
git.proxmox.com Git - mirror_frr.git/blob - nhrpd/zbuf.h
1 /* Stream/packet buffer API
2 * Copyright (c) 2014-2015 Timo Teräs
4 * This file is free software: you may copy, redistribute and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation, either version 2 of the License, or
7 * (at your option) any later version.
16 #include <sys/types.h>
22 struct list_head queue_list
;
23 unsigned allocated
: 1;
30 struct list_head queue_head
;
33 struct zbuf
*zbuf_alloc(size_t size
);
34 void zbuf_init(struct zbuf
*zb
, void *buf
, size_t len
, size_t datalen
);
35 void zbuf_free(struct zbuf
*zb
);
37 static inline size_t zbuf_size(struct zbuf
*zb
)
39 return zb
->end
- zb
->buf
;
42 static inline size_t zbuf_used(struct zbuf
*zb
)
44 return zb
->tail
- zb
->head
;
47 static inline size_t zbuf_tailroom(struct zbuf
*zb
)
49 return zb
->end
- zb
->tail
;
52 static inline size_t zbuf_headroom(struct zbuf
*zb
)
54 return zb
->head
- zb
->buf
;
57 void zbuf_reset(struct zbuf
*zb
);
58 void zbuf_reset_head(struct zbuf
*zb
, void *ptr
);
59 ssize_t
zbuf_read(struct zbuf
*zb
, int fd
, size_t maxlen
);
60 ssize_t
zbuf_write(struct zbuf
*zb
, int fd
);
61 ssize_t
zbuf_recv(struct zbuf
*zb
, int fd
);
62 ssize_t
zbuf_send(struct zbuf
*zb
, int fd
);
64 static inline void zbuf_set_rerror(struct zbuf
*zb
)
70 static inline void zbuf_set_werror(struct zbuf
*zb
)
76 static inline void *__zbuf_pull(struct zbuf
*zb
, size_t size
, int error
)
78 void *head
= zb
->head
;
79 if (size
> zbuf_used(zb
)) {
88 #define zbuf_pull(zb, type) ((type *)__zbuf_pull(zb, sizeof(type), 1))
89 #define zbuf_pulln(zb, sz) ((void *)__zbuf_pull(zb, sz, 1))
90 #define zbuf_may_pull(zb, type) ((type *)__zbuf_pull(zb, sizeof(type), 0))
91 #define zbuf_may_pulln(zb, sz) ((void *)__zbuf_pull(zb, sz, 0))
93 void *zbuf_may_pull_until(struct zbuf
*zb
, const char *sep
, struct zbuf
*msg
);
95 static inline void zbuf_get(struct zbuf
*zb
, void *dst
, size_t len
)
97 void *src
= zbuf_pulln(zb
, len
);
99 memcpy(dst
, src
, len
);
102 static inline uint8_t zbuf_get8(struct zbuf
*zb
)
104 uint8_t *src
= zbuf_pull(zb
, uint8_t);
110 static inline uint32_t zbuf_get32(struct zbuf
*zb
)
114 } __attribute__((packed
));
116 struct unaligned32
*v
= zbuf_pull(zb
, struct unaligned32
);
122 static inline uint16_t zbuf_get_be16(struct zbuf
*zb
)
126 } __attribute__((packed
));
128 struct unaligned16
*v
= zbuf_pull(zb
, struct unaligned16
);
130 return be16toh(v
->value
);
134 static inline uint32_t zbuf_get_be32(struct zbuf
*zb
)
136 return be32toh(zbuf_get32(zb
));
139 static inline void *__zbuf_push(struct zbuf
*zb
, size_t size
, int error
)
141 void *tail
= zb
->tail
;
142 if (size
> zbuf_tailroom(zb
)) {
151 #define zbuf_push(zb, type) ((type *)__zbuf_push(zb, sizeof(type), 1))
152 #define zbuf_pushn(zb, sz) ((void *)__zbuf_push(zb, sz, 1))
153 #define zbuf_may_push(zb, type) ((type *)__zbuf_may_push(zb, sizeof(type), 0))
154 #define zbuf_may_pushn(zb, sz) ((void *)__zbuf_push(zb, sz, 0))
156 static inline void zbuf_put(struct zbuf
*zb
, const void *src
, size_t len
)
158 void *dst
= zbuf_pushn(zb
, len
);
160 memcpy(dst
, src
, len
);
163 static inline void zbuf_put8(struct zbuf
*zb
, uint8_t val
)
165 uint8_t *dst
= zbuf_push(zb
, uint8_t);
170 static inline void zbuf_put_be16(struct zbuf
*zb
, uint16_t val
)
174 } __attribute__((packed
));
176 struct unaligned16
*v
= zbuf_push(zb
, struct unaligned16
);
178 v
->value
= htobe16(val
);
181 static inline void zbuf_put_be32(struct zbuf
*zb
, uint32_t val
)
185 } __attribute__((packed
));
187 struct unaligned32
*v
= zbuf_push(zb
, struct unaligned32
);
189 v
->value
= htobe32(val
);
192 void zbuf_copy(struct zbuf
*zb
, struct zbuf
*src
, size_t len
);
194 void zbufq_init(struct zbuf_queue
*);
195 void zbufq_reset(struct zbuf_queue
*);
196 void zbufq_queue(struct zbuf_queue
*, struct zbuf
*);
197 int zbufq_write(struct zbuf_queue
*, int);