]>
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
)) {
80 if (error
) zbuf_set_rerror(zb
);
87 #define zbuf_pull(zb, type) ((type *)__zbuf_pull(zb, sizeof(type), 1))
88 #define zbuf_pulln(zb, sz) ((void *)__zbuf_pull(zb, sz, 1))
89 #define zbuf_may_pull(zb, type) ((type *)__zbuf_pull(zb, sizeof(type), 0))
90 #define zbuf_may_pulln(zb, sz) ((void *)__zbuf_pull(zb, sz, 0))
92 void *zbuf_may_pull_until(struct zbuf
*zb
, const char *sep
, struct zbuf
*msg
);
94 static inline void zbuf_get(struct zbuf
*zb
, void *dst
, size_t len
)
96 void *src
= zbuf_pulln(zb
, len
);
97 if (src
) memcpy(dst
, src
, len
);
100 static inline uint8_t zbuf_get8(struct zbuf
*zb
)
102 uint8_t *src
= zbuf_pull(zb
, uint8_t);
103 if (src
) return *src
;
107 static inline uint32_t zbuf_get32(struct zbuf
*zb
)
111 } __attribute__((packed
));
113 struct unaligned32
*v
= zbuf_pull(zb
, struct unaligned32
);
114 if (v
) return v
->value
;
118 static inline uint16_t zbuf_get_be16(struct zbuf
*zb
)
122 } __attribute__((packed
));
124 struct unaligned16
*v
= zbuf_pull(zb
, struct unaligned16
);
125 if (v
) return be16toh(v
->value
);
129 static inline uint32_t zbuf_get_be32(struct zbuf
*zb
)
131 return be32toh(zbuf_get32(zb
));
134 static inline void *__zbuf_push(struct zbuf
*zb
, size_t size
, int error
)
136 void *tail
= zb
->tail
;
137 if (size
> zbuf_tailroom(zb
)) {
138 if (error
) zbuf_set_werror(zb
);
145 #define zbuf_push(zb, type) ((type *)__zbuf_push(zb, sizeof(type), 1))
146 #define zbuf_pushn(zb, sz) ((void *)__zbuf_push(zb, sz, 1))
147 #define zbuf_may_push(zb, type) ((type *)__zbuf_may_push(zb, sizeof(type), 0))
148 #define zbuf_may_pushn(zb, sz) ((void *)__zbuf_push(zb, sz, 0))
150 static inline void zbuf_put(struct zbuf
*zb
, const void *src
, size_t len
)
152 void *dst
= zbuf_pushn(zb
, len
);
153 if (dst
) memcpy(dst
, src
, len
);
156 static inline void zbuf_put8(struct zbuf
*zb
, uint8_t val
)
158 uint8_t *dst
= zbuf_push(zb
, uint8_t);
162 static inline void zbuf_put_be16(struct zbuf
*zb
, uint16_t val
)
166 } __attribute__((packed
));
168 struct unaligned16
*v
= zbuf_push(zb
, struct unaligned16
);
169 if (v
) v
->value
= htobe16(val
);
172 static inline void zbuf_put_be32(struct zbuf
*zb
, uint32_t val
)
176 } __attribute__((packed
));
178 struct unaligned32
*v
= zbuf_push(zb
, struct unaligned32
);
179 if (v
) v
->value
= htobe32(val
);
182 void zbuf_copy(struct zbuf
*zb
, struct zbuf
*src
, size_t len
);
184 void zbufq_init(struct zbuf_queue
*);
185 void zbufq_reset(struct zbuf_queue
*);
186 void zbufq_queue(struct zbuf_queue
*, struct zbuf
*);
187 int zbufq_write(struct zbuf_queue
*, int);