wpabuf.h

Go to the documentation of this file.

 
#ifndef WPABUF_H
#define WPABUF_H
 
#include "common.h"
#include "utils/allocs.h"
#include "utils/os.h"
 
/* wpabuf::buf is a pointer to external data */
#define WPABUF_FLAG_EXT_DATA BIT(0)
 
/*
 * Internal data structure for wpabuf. Please do not touch this directly from
 * elsewhere. This is only defined in header file to allow inline functions
 * from this file to access data.
 */
struct wpabuf {
  size_t size; /* total size of the allocated buffer */
  size_t used; /* length of data in the buffer */
  u8 *buf;     /* pointer to the head of the buffer */
  unsigned int flags;
  /* optionally followed by the allocated buffer */
};
 
int wpabuf_resize(struct wpabuf **buf, size_t add_len);
struct wpabuf *wpabuf_alloc(size_t len);
struct wpabuf *wpabuf_alloc_ext_data(u8 *data, size_t len);
struct wpabuf *wpabuf_alloc_copy(const void *data, size_t len);
struct wpabuf *wpabuf_dup(const struct wpabuf *src);
void wpabuf_free(struct wpabuf *buf);
void wpabuf_clear_free(struct wpabuf *buf);
void *wpabuf_put(struct wpabuf *buf, size_t len);
struct wpabuf *wpabuf_concat(struct wpabuf *a, struct wpabuf *b);
struct wpabuf *wpabuf_zeropad(struct wpabuf *buf, size_t len);
void wpabuf_printf(struct wpabuf *buf, char *fmt, ...) PRINTF_FORMAT(2, 3);
struct wpabuf *wpabuf_parse_bin(const char *buf);
 
static inline size_t wpabuf_size(const struct wpabuf *buf) { return buf->size; }
 
static inline size_t wpabuf_len(const struct wpabuf *buf) { return buf->used; }
 
static inline size_t wpabuf_tailroom(const struct wpabuf *buf) {
  return buf->size - buf->used;
}
 
static inline int wpabuf_cmp(const struct wpabuf *a, const struct wpabuf *b) {
  if (!a && !b)
    return 0;
  if (a && b && wpabuf_size(a) == wpabuf_size(b))
    return os_memcmp(a->buf, b->buf, wpabuf_size(a));
  return -1;
}
 
static inline const void *wpabuf_head(const struct wpabuf *buf) {
  return buf->buf;
}
 
static inline const u8 *wpabuf_head_u8(const struct wpabuf *buf) {
  return (const u8 *)wpabuf_head(buf);
}
 
static inline void *wpabuf_mhead(struct wpabuf *buf) { return buf->buf; }
 
static inline u8 *wpabuf_mhead_u8(struct wpabuf *buf) {
  return (u8 *)wpabuf_mhead(buf);
}
 
static inline void wpabuf_put_u8(struct wpabuf *buf, u8 data) {
  u8 *pos = (u8 *)wpabuf_put(buf, 1);
  *pos = data;
}
 
static inline void wpabuf_put_le16(struct wpabuf *buf, u16 data) {
  u8 *pos = (u8 *)wpabuf_put(buf, 2);
  WPA_PUT_LE16(pos, data);
}
 
static inline void wpabuf_put_le32(struct wpabuf *buf, u32 data) {
  u8 *pos = (u8 *)wpabuf_put(buf, 4);
  WPA_PUT_LE32(pos, data);
}
 
static inline void wpabuf_put_le64(struct wpabuf *buf, u64 data) {
  u8 *pos = (u8 *)wpabuf_put(buf, 8);
  WPA_PUT_LE64(pos, data);
}
 
static inline void wpabuf_put_be16(struct wpabuf *buf, u16 data) {
  u8 *pos = (u8 *)wpabuf_put(buf, 2);
  WPA_PUT_BE16(pos, data);
}
 
static inline void wpabuf_put_be24(struct wpabuf *buf, u32 data) {
  u8 *pos = (u8 *)wpabuf_put(buf, 3);
  WPA_PUT_BE24(pos, data);
}
 
static inline void wpabuf_put_be32(struct wpabuf *buf, u32 data) {
  u8 *pos = (u8 *)wpabuf_put(buf, 4);
  WPA_PUT_BE32(pos, data);
}
 
static inline void wpabuf_put_be64(struct wpabuf *buf, u64 data) {
  u8 *pos = (u8 *)wpabuf_put(buf, 8);
  WPA_PUT_BE64(pos, data);
}
 
static inline void wpabuf_put_data(struct wpabuf *buf, const void *data,
                                   size_t len) {
  if (data)
    os_memcpy(wpabuf_put(buf, len), data, len);
}
 
static inline void wpabuf_put_buf(struct wpabuf *dst,
                                  const struct wpabuf *src) {
  wpabuf_put_data(dst, wpabuf_head(src), wpabuf_len(src));
}
 
static inline void wpabuf_set(struct wpabuf *buf, const void *data,
                              size_t len) {
  buf->buf = (u8 *)data;
  buf->flags = WPABUF_FLAG_EXT_DATA;
  buf->size = buf->used = len;
}
 
static inline void wpabuf_put_str(struct wpabuf *dst, const char *str) {
  wpabuf_put_data(dst, str, os_strlen(str));
}
 
#endif /* WPABUF_H */