mdw@git.distorted.org.uk Git - secnet/blob - util.h

   1 #ifndef util_h
   2 #define util_h
   3
   4 #include "secnet.h"
   5 #include <gmp.h>
   6
   7 #include "hackypar.h"
   8
   9 #define BUF_ASSERT_FREE(buf) do { buffer_assert_free((buf), \
  10                                                      __FILE__,__LINE__); } \
  11 while(0)
  12 #define BUF_ASSERT_USED(buf) do { buffer_assert_used((buf), \
  13                                                      __FILE__,__LINE__); } \
  14 while(0)
  15 #define BUF_ALLOC(buf,own) do { buffer_assert_free((buf),__FILE__,__LINE__); \
  16          (buf)->free=False; (buf)->owner=(own); (buf)->start=(buf)->base; \
  17          (buf)->size=0; } while(0)
  18 #define BUF_FREE(buf) do { (buf)->free=True; } while(0)
  19
  20 extern void buffer_assert_free(struct buffer_if *buffer, cstring_t file,
  21                                int line);
  22 extern void buffer_assert_used(struct buffer_if *buffer, cstring_t file,
  23                                int line);
  24 extern void buffer_new(struct buffer_if *buffer, int32_t len);
  25 extern void buffer_init(struct buffer_if *buffer, int32_t max_start_pad);
  26 extern void buffer_copy(struct buffer_if *dst, const struct buffer_if *src);
  27 extern void *buf_append(struct buffer_if *buf, int32_t amount);
  28 extern void *buf_prepend(struct buffer_if *buf, int32_t amount);
  29 extern void *buf_unappend(struct buffer_if *buf, int32_t amount);
  30 extern void *buf_unprepend(struct buffer_if *buf, int32_t amount);
  31
  32 /*
  33  * void BUF_ADD_BYTES(append,    struct buffer_if*, const void*, int32_t size);
  34  * void BUF_ADD_BYTES(prepend,   struct buffer_if*, const void*, int32_t size);
  35  * void BUF_GET_BYTES(unappend,  struct buffer_if*,       void*, int32_t size);
  36  * void BUF_GET_BYTES(unprepend, struct buffer_if*,       void*, int32_t size);
  37  *     // all of these evaluate size twice
  38  *
  39  * void BUF_ADD_OBJ(append,    struct_buffer_if*, const OBJECT& something);
  40  * void BUF_ADD_OBJ(prepend,   struct_buffer_if*, const OBJECT& something);
  41  * void BUF_GET_OBJ(unappend,  struct_buffer_if*,       OBJECT& something);
  42  * void BUF_GET_OBJ(unprepend, struct_buffer_if*,       OBJECT& something);
  43  */
  44 #define BUF_ADD_BYTES(appendprepend, bufp, datap, size)                 \
  45     (buf_un##appendprepend /* ensures we have correct direction */,     \
  46      memcpy(buf_##appendprepend((bufp),(size)),(datap),(size)))
  47 #define BUF_ADD_OBJ(appendprepend, bufp, obj) \
  48     BUF_ADD_BYTES(appendprepend,(bufp),&(obj),sizeof((obj)))
  49 #define BUF_GET_BYTES(unappendunprepend, bufp, datap, size)             \
  50     (BUF_GET__DOESNOTEXIST__buf_un##unappendunprepend,                  \
  51      memcpy((datap),buf_##unappendunprepend((bufp),(size)),(size)))
  52 #define BUF_GET_OBJ(unappendunprepend, bufp, obj) \
  53     BUF_ADD_BYTES(unappendunprepend,&(obj),(bufp),sizeof((obj)))
  54 #define BUF_GET__DOESNOTEXIST__buf_ununappend  0
  55 #define BUF_GET__DOESNOTEXIST__buf_ununprepend 0
  56
  57 static inline int32_t buf_remaining_space(const struct buffer_if *buf)
  58 {
  59     return (buf->base + buf->alloclen) - (buf->start + buf->size);
  60 }
  61
  62 extern void buffer_readonly_view(struct buffer_if *n, const void*, int32_t len);
  63 extern void buffer_readonly_clone(struct buffer_if *n, const struct buffer_if*);
  64   /* Caller must only use unappend, unprepend et al. on n.
  65    * New buffer state (in n) before this can be undefined.  After use,
  66    * it must NOT be freed. */
  67
  68 extern void buf_append_string(struct buffer_if *buf, cstring_t s);
  69
  70 extern void read_mpbin(MP_INT *a, uint8_t *bin, int binsize);
  71
  72 extern char *write_mpstring(MP_INT *a);
  73
  74 extern int32_t write_mpbin(MP_INT *a, uint8_t *buffer, int32_t buflen);
  75
  76 extern struct log_if *init_log(list_t *loglist);
  77
  78 extern void send_nak(const struct comm_addr *dest, uint32_t our_index,
  79                      uint32_t their_index, uint32_t msgtype,
  80                      struct buffer_if *buf, const char *logwhy);
  81
  82 extern int consttime_memeq(const void *s1, const void *s2, size_t n);
  83
  84 const char *iaddr_to_string(const union iaddr *ia);
  85 int iaddr_socklen(const union iaddr *ia);
  86
  87 void string_item_to_iaddr(const item_t *item, uint16_t port, union iaddr *ia,
  88                           const char *desc);
  89
  90 #define MINMAX(ae,be,op) ({                     \
  91         typeof((ae)) a=(ae);                    \
  92         typeof((be)) b=(be);                    \
  93         a op b ? a : b;                         \
  94     })
  95 #define MAX(a,b) MINMAX((a),(b),>)
  96 #define MIN(a,b) MINMAX((a),(b),<)
  97
  98 #endif /* util_h */