BIRD library: The story continues.

Complete resource manages and IP address handling.

18 files changed, 864 insertions, 21 deletions

diff --git a/lib/Makefile b/lib/Makefile
index 214f76a..6eb14e1 100644
--- a/lib/Makefile
+++ b/lib/Makefile
@@ -1,3 +1,9 @@
-OBJS=lists.o
+OBJS=lists.o bitops.o resource.o xmalloc.o mempool.o slab.o md5.o
+
+ifdef IPV6
+OBJS += ipv6.o
+else
+OBJS += ipv4.o
+endif
 
 include $(TOPDIR)/Rules
diff --git a/lib/birdlib.h b/lib/birdlib.h
index eb95585..5e533b6 100644
--- a/lib/birdlib.h
+++ b/lib/birdlib.h
@@ -13,17 +13,28 @@
 
 #define OFFSETOF(s, i) ((unsigned int)&((s *)0)->i)
 #define SKIP_BACK(s, i, p) ((s *)((char *)p - OFFSETOF(s, i)))
+#define ALIGN(s, a) (((s)+a-1)&~(a-1))
+
+/* Functions which don't return */
+
+#define NORET __attribute__((noreturn))
 
 /* Logging and dying */
 
 void log(char *msg, ...);
-void die(char *msg, ...);
+void die(char *msg, ...) NORET;
 
 #define L_DEBUG "\001"			/* Debugging messages */
 #define L_INFO "\002"			/* Informational messages */
 #define L_WARN "\003"			/* Warnings */
 #define L_ERR "\004"			/* Errors */
 #define L_AUTH "\005"			/* Authorization failed etc. */
+#define L_FATAL "\006"			/* Fatal errors */
+
+void log_init(char *);			/* Initialize logging to given file (NULL=stderr, ""=syslog) */
+void log_init_debug(char *);		/* Initialize debug dump to given file (NULL=stderr, ""=off) */
+
+void debug(char *msg, ...);		/* Printf to debug output */
 
 /* Debugging */
 
diff --git a/lib/bitops.c b/lib/bitops.c
new file mode 100644
index 0000000..10bca04
--- /dev/null
+++ b/lib/bitops.c
@@ -0,0 +1,32 @@
+/*
+ *	BIRD Library -- Generic Bit Operations
+ *
+ *	(c) 1998 Martin Mares <mj@ucw.cz>
+ *
+ *	Can be freely distributed and used under the terms of the GNU GPL.
+ */
+
+#include "nest/bird.h"
+#include "bitops.h"
+
+u32
+u32_mkmask(unsigned n)
+{
+  return n ? ~((1 << (32 - n)) - 1) : 0;
+}
+
+int
+u32_masklen(u32 x)
+{
+  int l = 0;
+  u32 n = ~x;
+
+  if (n & (n+1)) return -1;
+  if (x & 0x0000ffff) { x &= 0x0000ffff; l += 16; }
+  if (x & 0x00ff00ff) { x &= 0x00ff00ff; l += 8; }
+  if (x & 0x0f0f0f0f) { x &= 0x0f0f0f0f; l += 4; }
+  if (x & 0x33333333) { x &= 0x33333333; l += 2; }
+  if (x & 0x55555555) l++;
+  if (x & 0xaaaaaaaa) l++;
+  return l;
+}
diff --git a/lib/bitops.h b/lib/bitops.h
new file mode 100644
index 0000000..6bfc95f
--- /dev/null
+++ b/lib/bitops.h
@@ -0,0 +1,19 @@
+/*
+ *	BIRD Library -- Generic Bit Operations
+ *
+ *	(c) 1998 Martin Mares <mj@ucw.cz>
+ *
+ *	Can be freely distributed and used under the terms of the GNU GPL.
+ */
+
+/*
+ *	Bit mask operations:
+ *
+ *	u32_mkmask	Make bit mask consisting of <n> consecutive ones
+ *			from the left and the rest filled with zeroes.
+ *			E.g., u32_mkmask(5) = 0xf8000000.
+ *	u32_masklen	Inverse operation to u32_mkmask, -1 if not a bitmask.
+ */
+
+u32 u32_mkmask(unsigned);
+int u32_masklen(u32);
diff --git a/lib/ip.h b/lib/ip.h
index dc4f8b8..c368399 100644
--- a/lib/ip.h
+++ b/lib/ip.h
@@ -15,4 +15,24 @@
 #include "ipv6.h"
 #endif
 
+/*
+ *	ip_classify() returns either a negative number for invalid addresses
+ *	or scope OR'ed together with address type.
+ */
+
+#define IADDR_INVALID		-1
+#define IADDR_SCOPE_MASK       	0xfff
+#define IADDR_HOST		0x1000
+#define IADDR_BROADCAST		0x2000
+#define IADDR_MULTICAST		0x4000
+
+/*
+ *	Address scope
+ */
+
+#define SCOPE_HOST 0
+#define SCOPE_LINK 1
+#define SCOPE_SITE 2
+#define SCOPE_UNIVERSE 3
+
 #endif
diff --git a/lib/ipv4.c b/lib/ipv4.c
new file mode 100644
index 0000000..52f5b0b
--- /dev/null
+++ b/lib/ipv4.c
@@ -0,0 +1,29 @@
+/*
+ *	BIRD Library -- IPv4 Address Manipulation Functions
+ *
+ *	(c) 1998 Martin Mares <mj@ucw.cz>
+ *
+ *	Can be freely distributed and used under the terms of the GNU GPL.
+ */
+
+#include "nest/bird.h"
+#include "lib/ip.h"
+
+int
+ipv4_classify(u32 a)
+{
+  u32 b = a >> 24U;
+
+  if (b && b <= 0xdf)
+    {
+      if (b == 0x7f)
+	return IADDR_HOST | SCOPE_HOST;
+      else
+	return IADDR_HOST | SCOPE_UNIVERSE;
+    }
+  if (b >= 0xe0 && b <= 0xef)
+    return IADDR_MULTICAST | SCOPE_UNIVERSE;
+  if (a == 0xffffffff)
+    return IADDR_BROADCAST | SCOPE_LINK;
+  return IADDR_INVALID;
+}
diff --git a/lib/ipv4.h b/lib/ipv4.h
index 4b8c44c..fa8a27b 100644
--- a/lib/ipv4.h
+++ b/lib/ipv4.h
@@ -11,6 +11,15 @@
 
 #include <netinet/in.h>
 
+#include "lib/bitops.h"
+
+#ifdef DEBUG
+
+/*
+ *	Use the structural representation when you want to make sure
+ *	nobody unauthorized attempts to handle ip_addr as number.
+ */
+
 typedef struct ipv4_addr {
   u32 addr;
 } ip_addr;
@@ -18,18 +27,37 @@ typedef struct ipv4_addr {
 #define _I(x) (x).addr
 #define _MI(x) ((struct ip_addr) { x })
 
+#else
+
+typedef u32 ip_addr;
+
+#define _I(x) (x)
+#define _MI(x) (x)
+
+#endif
+
 #define IPA_NONE (_MI(0))
 
 #define ipa_equal(x,y) (_I(x) == _I(y))
+#define ipa_nonzero(x) _I(x)
 #define ipa_and(x,y) _MI(_I(x) & _I(y))
 #define ipa_or(x,y) _MI(_I(x) | _I(y))
 #define ipa_not(x) _MI(~_I(x))
-#define ipa_mkmask(x) _MI(ipv4_mkmask(x))
-#define ipa_mklen(x) ipv4_mklen(_I(x))
+#define ipa_mkmask(x) _MI(u32_mkmask(x))
+#define ipa_mklen(x) u32_masklen(_I(x))
+#define ipa_hash(x) ipv4_hash(_I(x))
+#define ipa_hton(x) x = _MI(htonl(_I(x)))
+#define ipa_ntoh(x) x = _MI(ntohl(_I(x)))
+#define ipa_classify(x) ipv4_classify(_I(x))
 
 unsigned ipv4_mklen(u32);
 u32 ipv4_mkmask(unsigned);
+int ipv4_classify(u32);
 
-/* ??? htonl and ntohl ??? */
+/* FIXME: Is this hash function uniformly distributed over standard routing tables? */
+static inline unsigned ipv4_hash(u32 a)
+{
+  return a ^ (a >> 16) ^ (a >> 24);
+}
 
 #endif
diff --git a/lib/ipv6.c b/lib/ipv6.c
new file mode 100644
index 0000000..e612af7
--- /dev/null
+++ b/lib/ipv6.c
@@ -0,0 +1,12 @@
+/*
+ *	BIRD Library -- IPv6 Address Manipulation Functions
+ *
+ *	(c) 1998 Martin Mares <mj@ucw.cz>
+ *
+ *	Can be freely distributed and used under the terms of the GNU GPL.
+ */
+
+#include "nest/bird.h"
+#include "lib/ip.h"
+
+#error "Ought to implement these."
diff --git a/lib/ipv6.h b/lib/ipv6.h
index 98d7811..845955f 100644
--- a/lib/ipv6.h
+++ b/lib/ipv6.h
@@ -25,6 +25,7 @@ typedef struct ipv4_addr {
 #define IPA_NONE _MI(0,0,0,0)
 
 #define ipa_equal(x,y) (!memcmp(&(x),&(y),sizeof(ip_addr)))
+#define ipa_nonzero(x) (_I0(a) || _I1(a) || _I2(a) || _I3(a))
 #define ipa_and(a,b) _MI(_I0(a) & _I0(b), \
 			 _I1(a) & _I1(b), \
 			 _I2(a) & _I2(b), \
@@ -34,11 +35,24 @@ typedef struct ipv4_addr {
 			_I2(a) | _I2(b), \
 			_I3(a) | _I3(b))
 #define ipa_not(a) _MI(~_I0(a),~_I1(a),~_I2(a),~_I3(a))
-
 #define ipa_mkmask(x) ipv6_mkmask(x)
-#define ipa_mklen(x) ipv6_mklen(x)
+#define ipa_mklen(x) ipv6_mklen(&(x))
+#define ipa_hash(x) ipv6_hash(&(x))
+#define ipa_hton(x) ipv6_hton(&(x))
+#define ipa_ntoh(x) ipv6_ntoh(&(x))
+#define ipa_classify(x) ipv6_classify(&(x))
 
 ip_addr ipv6_mkmask(unsigned);
-unsigned ipv6_mklen(ip_addr);
+unsigned ipv6_mklen(ip_addr *);
+int ipv6_classify(ip_addr *);
+void ipv6_hton(ip_addr *);
+void ipv6_ntoh(ip_addr *);
+
+/* FIXME: Is this hash function uniformly distributed over standard routing tables? */
+static inline unsigned ipv6_hash(ip_addr *a)
+{
+  u32 x = _I0(*a) ^ _I1(*a) ^ _I2(*a) ^ _I3(*a);
+  return x ^ (x >> 16) ^ (x >> 8);
+}
 
 #endif
diff --git a/lib/lists.c b/lib/lists.c
index 55c2b0f..c14eb4c 100644
--- a/lib/lists.c
+++ b/lib/lists.c
@@ -16,7 +16,7 @@ add_tail(list *l, node *n)
 {
   node *z = l->tail;
 
-  n->next = (node *) &l->tail;
+  n->next = (node *) &l->null;
   n->prev = z;
   z->next = n;
   l->tail = n;
diff --git a/lib/lists.h b/lib/lists.h
index acc054d..3230724 100644
--- a/lib/lists.h
+++ b/lib/lists.h
@@ -33,7 +33,7 @@ void insert_node(node *, node *);
 
 #ifndef _BIRD_LISTS_C_
 #define LIST_INLINE extern inline
-#include <lib/lists.c>
+#include "lib/lists.c"
 #undef LIST_INLINE
 #else
 #define LIST_INLINE
diff --git a/lib/md5.c b/lib/md5.c
new file mode 100644
index 0000000..9fec594
--- /dev/null
+++ b/lib/md5.c
@@ -0,0 +1,252 @@
+/*
+ * This code implements the MD5 message-digest algorithm.
+ * The algorithm is due to Ron Rivest.  This code was
+ * written by Colin Plumb in 1993, no copyright is claimed.
+ * This code is in the public domain; do with it what you wish.
+ *
+ * Equivalent code is available from RSA Data Security, Inc.
+ * This code has been tested against that, and is equivalent,
+ * except that you don't need to include two pages of legalese
+ * with every copy.
+ *
+ * To compute the message digest of a chunk of bytes, declare an
+ * MD5Context structure, pass it to MD5Init, call MD5Update as
+ * needed on buffers full of bytes, and then call MD5Final, which
+ * will fill a supplied 16-byte array with the digest.
+ */
+
+/*
+ * Adapted for BIRD by Martin Mares <mj@atrey.karlin.mff.cuni.cz>
+ */
+
+#include <string.h>		/* for memcpy() */
+#include "nest/bird.h"
+#include "md5.h"
+
+#ifdef CPU_LITTLE_ENDIAN
+#define byteReverse(buf, len)	/* Nothing */
+#else
+void byteReverse(unsigned char *buf, unsigned longs);
+
+/*
+ * Note: this code is harmless on little-endian machines.
+ */
+void byteReverse(unsigned char *buf, unsigned longs)
+{
+    u32 t;
+    do {
+	t = (u32) ((unsigned) buf[3] << 8 | buf[2]) << 16 |
+	    ((unsigned) buf[1] << 8 | buf[0]);
+	*(u32 *) buf = t;
+	buf += 4;
+    } while (--longs);
+}
+#endif
+
+/*
+ * Start MD5 accumulation.  Set bit count to 0 and buffer to mysterious
+ * initialization constants.
+ */
+void MD5Init(struct MD5Context *ctx)
+{
+    ctx->buf[0] = 0x67452301;
+    ctx->buf[1] = 0xefcdab89;
+    ctx->buf[2] = 0x98badcfe;
+    ctx->buf[3] = 0x10325476;
+
+    ctx->bits[0] = 0;
+    ctx->bits[1] = 0;
+}
+
+/*
+ * Update context to reflect the concatenation of another buffer full
+ * of bytes.
+ */
+void MD5Update(struct MD5Context *ctx, unsigned char const *buf, unsigned len)
+{
+    u32 t;
+
+    /* Update bitcount */
+
+    t = ctx->bits[0];
+    if ((ctx->bits[0] = t + ((u32) len << 3)) < t)
+	ctx->bits[1]++;		/* Carry from low to high */
+    ctx->bits[1] += len >> 29;
+
+    t = (t >> 3) & 0x3f;	/* Bytes already in shsInfo->data */
+
+    /* Handle any leading odd-sized chunks */
+
+    if (t) {
+	unsigned char *p = (unsigned char *) ctx->in + t;
+
+	t = 64 - t;
+	if (len < t) {
+	    memcpy(p, buf, len);
+	    return;
+	}
+	memcpy(p, buf, t);
+	byteReverse(ctx->in, 16);
+	MD5Transform(ctx->buf, (u32 *) ctx->in);
+	buf += t;
+	len -= t;
+    }
+    /* Process data in 64-byte chunks */
+
+    while (len >= 64) {
+	memcpy(ctx->in, buf, 64);
+	byteReverse(ctx->in, 16);
+	MD5Transform(ctx->buf, (u32 *) ctx->in);
+	buf += 64;
+	len -= 64;
+    }
+
+    /* Handle any remaining bytes of data. */
+
+    memcpy(ctx->in, buf, len);
+}
+
+/*
+ * Final wrapup - pad to 64-byte boundary with the bit pattern 
+ * 1 0* (64-bit count of bits processed, MSB-first)
+ */
+void MD5Final(unsigned char digest[16], struct MD5Context *ctx)
+{
+    unsigned count;
+    unsigned char *p;
+
+    /* Compute number of bytes mod 64 */
+    count = (ctx->bits[0] >> 3) & 0x3F;
+
+    /* Set the first char of padding to 0x80.  This is safe since there is
+       always at least one byte free */
+    p = ctx->in + count;
+    *p++ = 0x80;
+
+    /* Bytes of padding needed to make 64 bytes */
+    count = 64 - 1 - count;
+
+    /* Pad out to 56 mod 64 */
+    if (count < 8) {
+	/* Two lots of padding:  Pad the first block to 64 bytes */
+	memset(p, 0, count);
+	byteReverse(ctx->in, 16);
+	MD5Transform(ctx->buf, (u32 *) ctx->in);
+
+	/* Now fill the next block with 56 bytes */
+	memset(ctx->in, 0, 56);
+    } else {
+	/* Pad block to 56 bytes */
+	memset(p, 0, count - 8);
+    }
+    byteReverse(ctx->in, 14);
+
+    /* Append length in bits and transform */
+    ((u32 *) ctx->in)[14] = ctx->bits[0];
+    ((u32 *) ctx->in)[15] = ctx->bits[1];
+
+    MD5Transform(ctx->buf, (u32 *) ctx->in);
+    byteReverse((unsigned char *) ctx->buf, 4);
+    memcpy(digest, ctx->buf, 16);
+    memset((char *) ctx, 0, sizeof(ctx));	/* In case it's sensitive */
+}
+
+/* The four core functions - F1 is optimized somewhat */
+
+/* #define F1(x, y, z) (x & y | ~x & z) */
+#define F1(x, y, z) (z ^ (x & (y ^ z)))
+#define F2(x, y, z) F1(z, x, y)
+#define F3(x, y, z) (x ^ y ^ z)
+#define F4(x, y, z) (y ^ (x | ~z))
+
+/* This is the central step in the MD5 algorithm. */
+#define MD5STEP(f, w, x, y, z, data, s) \
+	( w += f(x, y, z) + data,  w = w<<s | w>>(32-s),  w += x )
+
+/*
+ * The core of the MD5 algorithm, this alters an existing MD5 hash to
+ * reflect the addition of 16 longwords of new data.  MD5Update blocks
+ * the data and converts bytes into longwords for this routine.
+ */
+void MD5Transform(u32 buf[4], u32 const in[16])
+{
+    register u32 a, b, c, d;
+
+    a = buf[0];
+    b = buf[1];
+    c = buf[2];
+    d = buf[3];
+
+    MD5STEP(F1, a, b, c, d, in[0] + 0xd76aa478, 7);
+    MD5STEP(F1, d, a, b, c, in[1] + 0xe8c7b756, 12);
+    MD5STEP(F1, c, d, a, b, in[2] + 0x242070db, 17);
+    MD5STEP(F1, b, c, d, a, in[3] + 0xc1bdceee, 22);
+    MD5STEP(F1, a, b, c, d, in[4] + 0xf57c0faf, 7);
+    MD5STEP(F1, d, a, b, c, in[5] + 0x4787c62a, 12);
+    MD5STEP(F1, c, d, a, b, in[6] + 0xa8304613, 17);
+    MD5STEP(F1, b, c, d, a, in[7] + 0xfd469501, 22);
+    MD5STEP(F1, a, b, c, d, in[8] + 0x698098d8, 7);
+    MD5STEP(F1, d, a, b, c, in[9] + 0x8b44f7af, 12);
+    MD5STEP(F1, c, d, a, b, in[10] + 0xffff5bb1, 17);
+    MD5STEP(F1, b, c, d, a, in[11] + 0x895cd7be, 22);
+    MD5STEP(F1, a, b, c, d, in[12] + 0x6b901122, 7);
+    MD5STEP(F1, d, a, b, c, in[13] + 0xfd987193, 12);
+    MD5STEP(F1, c, d, a, b, in[14] + 0xa679438e, 17);
+    MD5STEP(F1, b, c, d, a, in[15] + 0x49b40821, 22);
+
+    MD5STEP(F2, a, b, c, d, in[1] + 0xf61e2562, 5);
+    MD5STEP(F2, d, a, b, c, in[6] + 0xc040b340, 9);
+    MD5STEP(F2, c, d, a, b, in[11] + 0x265e5a51, 14);
+    MD5STEP(F2, b, c, d, a, in[0] + 0xe9b6c7aa, 20);
+    MD5STEP(F2, a, b, c, d, in[5] + 0xd62f105d, 5);
+    MD5STEP(F2, d, a, b, c, in[10] + 0x02441453, 9);
+    MD5STEP(F2, c, d, a, b, in[15] + 0xd8a1e681, 14);
+    MD5STEP(F2, b, c, d, a, in[4] + 0xe7d3fbc8, 20);
+    MD5STEP(F2, a, b, c, d, in[9] + 0x21e1cde6, 5);
+    MD5STEP(F2, d, a, b, c, in[14] + 0xc33707d6, 9);
+    MD5STEP(F2, c, d, a, b, in[3] + 0xf4d50d87, 14);
+    MD5STEP(F2, b, c, d, a, in[8] + 0x455a14ed, 20);
+    MD5STEP(F2, a, b, c, d, in[13] + 0xa9e3e905, 5);
+    MD5STEP(F2, d, a, b, c, in[2] + 0xfcefa3f8, 9);
+    MD5STEP(F2, c, d, a, b, in[7] + 0x676f02d9, 14);
+    MD5STEP(F2, b, c, d, a, in[12] + 0x8d2a4c8a, 20);
+
+    MD5STEP(F3, a, b, c, d, in[5] + 0xfffa3942, 4);
+    MD5STEP(F3, d, a, b, c, in[8] + 0x8771f681, 11);
+    MD5STEP(F3, c, d, a, b, in[11] + 0x6d9d6122, 16);
+    MD5STEP(F3, b, c, d, a, in[14] + 0xfde5380c, 23);
+    MD5STEP(F3, a, b, c, d, in[1] + 0xa4beea44, 4);
+    MD5STEP(F3, d, a, b, c, in[4] + 0x4bdecfa9, 11);
+    MD5STEP(F3, c, d, a, b, in[7] + 0xf6bb4b60, 16);
+    MD5STEP(F3, b, c, d, a, in[10] + 0xbebfbc70, 23);
+    MD5STEP(F3, a, b, c, d, in[13] + 0x289b7ec6, 4);
+    MD5STEP(F3, d, a, b, c, in[0] + 0xeaa127fa, 11);
+    MD5STEP(F3, c, d, a, b, in[3] + 0xd4ef3085, 16);
+    MD5STEP(F3, b, c, d, a, in[6] + 0x04881d05, 23);
+    MD5STEP(F3, a, b, c, d, in[9] + 0xd9d4d039, 4);
+    MD5STEP(F3, d, a, b, c, in[12] + 0xe6db99e5, 11);
+    MD5STEP(F3, c, d, a, b, in[15] + 0x1fa27cf8, 16);
+    MD5STEP(F3, b, c, d, a, in[2] + 0xc4ac5665, 23);
+
+    MD5STEP(F4, a, b, c, d, in[0] + 0xf4292244, 6);
+    MD5STEP(F4, d, a, b, c, in[7] + 0x432aff97, 10);
+    MD5STEP(F4, c, d, a, b, in[14] + 0xab9423a7, 15);
+    MD5STEP(F4, b, c, d, a, in[5] + 0xfc93a039, 21);
+    MD5STEP(F4, a, b, c, d, in[12] + 0x655b59c3, 6);
+    MD5STEP(F4, d, a, b, c, in[3] + 0x8f0ccc92, 10);
+    MD5STEP(F4, c, d, a, b, in[10] + 0xffeff47d, 15);
+    MD5STEP(F4, b, c, d, a, in[1] + 0x85845dd1, 21);
+    MD5STEP(F4, a, b, c, d, in[8] + 0x6fa87e4f, 6);
+    MD5STEP(F4, d, a, b, c, in[15] + 0xfe2ce6e0, 10);
+    MD5STEP(F4, c, d, a, b, in[6] + 0xa3014314, 15);
+    MD5STEP(F4, b, c, d, a, in[13] + 0x4e0811a1, 21);
+    MD5STEP(F4, a, b, c, d, in[4] + 0xf7537e82, 6);
+    MD5STEP(F4, d, a, b, c, in[11] + 0xbd3af235, 10);
+    MD5STEP(F4, c, d, a, b, in[2] + 0x2ad7d2bb, 15);
+    MD5STEP(F4, b, c, d, a, in[9] + 0xeb86d391, 21);
+
+    buf[0] += a;
+    buf[1] += b;
+    buf[2] += c;
+    buf[3] += d;
+}
diff --git a/lib/md5.h b/lib/md5.h
new file mode 100644
index 0000000..1258635
--- /dev/null
+++ b/lib/md5.h
@@ -0,0 +1,16 @@
+#ifndef MD5_H
+#define MD5_H
+
+struct MD5Context {
+	u32 buf[4];
+	u32 bits[2];
+	unsigned char in[64];
+};
+
+void MD5Init(struct MD5Context *context);
+void MD5Update(struct MD5Context *context, unsigned char const *buf,
+	       unsigned len);
+void MD5Final(unsigned char digest[16], struct MD5Context *context);
+void MD5Transform(u32 buf[4], u32 const in[16]);
+
+#endif /* !MD5_H */
diff --git a/lib/mempool.c b/lib/mempool.c
new file mode 100644
index 0000000..d2f2c09
--- /dev/null
+++ b/lib/mempool.c
@@ -0,0 +1,133 @@
+/*
+ *	BIRD Resource Manager -- Memory Pools
+ *
+ *	(c) 1998 Martin Mares <mj@ucw.cz>
+ *
+ *	Can be freely distributed and used under the terms of the GNU GPL.
+ */
+
+#include <stdlib.h>
+#include <string.h>
+
+#include "nest/bird.h"
+#include "lib/resource.h"
+
+struct mp_chunk {
+  struct mp_chunk *next;
+  byte data[0];
+};
+
+struct mempool {
+  resource r;
+  byte *ptr, *end;
+  struct mp_chunk *first, **plast;
+  unsigned chunk_size, threshold, total;
+};
+
+void mp_free(resource *);
+void mp_dump(resource *);
+
+struct resclass mp_class = {
+  "MemPool",
+  sizeof(struct mempool),
+  mp_free,
+  mp_dump
+};
+
+mempool
+*mp_new(pool *p, unsigned blk)
+{
+  mempool *m = ralloc(p, &mp_class);
+  m->ptr = m->end = NULL;
+  m->first = NULL;
+  m->plast = &m->first;
+  m->chunk_size = blk;
+  m->threshold = 3*blk/4;
+  m->total = 0;
+  return m;
+}
+
+void *
+mp_alloc(mempool *m, unsigned size)
+{
+  byte *a = (byte *) ALIGN((unsigned long) m->ptr, CPU_STRUCT_ALIGN);
+  byte *e = a + size;
+
+  if (e <= m->end)
+    {
+      m->ptr = e;
+      return a;
+    }
+  else
+    {
+      struct mp_chunk *c;
+      if (size >= m->threshold)
+	{
+	  c = xmalloc(sizeof(struct mp_chunk) + size);
+	  m->total += size;
+	}
+      else
+	{
+	  c = xmalloc(sizeof(struct mp_chunk) + m->chunk_size);
+	  m->ptr = c->data + size;
+	  m->end = c->data + m->chunk_size;
+	  m->total += m->chunk_size;
+	}
+      *m->plast = c;
+      m->plast = &c->next;
+      c->next = NULL;
+      return c->data;
+    }
+}
+
+void *
+mp_allocu(mempool *m, unsigned size)
+{
+  byte *a = m->ptr;
+  byte *e = a + size;
+
+  if (e <= m->end)
+    {
+      m->ptr = e;
+      return a;
+    }
+  return mp_alloc(m, size);
+}
+
+void *
+mp_allocz(mempool *m, unsigned size)
+{
+  void *z = mp_alloc(m, size);
+
+  bzero(z, size);
+  return z;
+}
+
+void
+mp_free(resource *r)
+{
+  mempool *m = (mempool *) r;
+  struct mp_chunk *c, *d;
+
+  for(d=m->first; d; d = c)
+    {
+      c = d->next;
+      xfree(d);
+    }
+}
+
+void
+mp_dump(resource *r)
+{
+  mempool *m = (mempool *) r;
+  struct mp_chunk *c;
+  int cnt;
+
+  for(cnt=0, c=m->first; c; c=c->next, cnt++)
+    ;
+  debug("(chunk=%d threshold=%d count=%d total=%d)\n",
+	m->chunk_size,
+	m->threshold,
+	cnt,
+	m->total);
+}
diff --git a/lib/resource.c b/lib/resource.c
new file mode 100644
index 0000000..2806a90
--- /dev/null
+++ b/lib/resource.c
@@ -0,0 +1,162 @@
+/*
+ *	BIRD Resource Manager
+ *
+ *	(c) 1998 Martin Mares <mj@ucw.cz>
+ *
+ *	Can be freely distributed and used under the terms of the GNU GPL.
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+
+#include "nest/bird.h"
+#include "lib/resource.h"
+
+struct pool {
+  resource r;
+  list inside;
+};
+
+void pool_dump(resource *);
+void pool_free(resource *);
+
+static struct resclass pool_class = {
+  "Pool",
+  sizeof(pool),
+  pool_free,
+  pool_dump
+};
+
+pool root_pool;
+
+static int indent;
+
+pool *
+rp_new(pool *p)
+{
+  pool *z = ralloc(p, &pool_class);
+  init_list(&z->inside);
+  return z;
+}
+
+void
+pool_free(resource *P)
+{
+  pool *p = (pool *) P;
+  resource *r, *rr;
+
+  r = HEAD(p->inside);
+  while (rr = (resource *) r->n.next)
+    {
+      r->class->free(r);
+      xfree(r);
+      r = rr;
+    }
+}
+
+void
+pool_dump(resource *P)
+{
+  pool *p = (pool *) P;
+  resource *r;
+
+  debug("\n");
+  indent += 3;
+  WALK_LIST(r, p->inside)
+    rdump(r);
+  indent -= 3;
+}
+
+void
+rfree(void *res)
+{
+  resource *r = res;
+
+  if (r)
+    {
+      if (r->n.next)
+	rem_node(&r->n);
+      r->class->free(r);
+      xfree(r);
+    }
+}
+
+void
+rdump(void *res)
+{
+  char x[16];
+  resource *r = res;
+
+  sprintf(x, "%%%ds%%08x ", indent);
+  debug(x, "", (int) r);
+  if (r)
+    {
+      debug("%-6s", r->class->name);
+      r->class->dump(r);
+    }
+  else
+    debug("NULL\n");
+}
+
+void *
+ralloc(pool *p, struct resclass *c)
+{
+  resource *r = xmalloc(c->size);
+
+  r->class = c;
+  add_tail(&p->inside, &r->n);
+  return r;
+}
+
+void
+resource_init(void)
+{
+  root_pool.r.class = &pool_class;
+  init_list(&root_pool.inside);
+}
+
+/*
+ *	Memory blocks.
+ */
+
+struct mblock {
+  resource r;
+  unsigned size;
+  byte data[0];
+};
+
+void mbl_free(resource *r)
+{
+}
+
+void mbl_debug(resource *r)
+{
+  struct mblock *m = (struct mblock *) r;
+
+  debug("(size=%d)\n", m->size);
+}
+
+struct resclass mb_class = {
+  "Memory",
+  0,
+  mbl_free,
+  mbl_debug,
+};
+
+void *
+mb_alloc(pool *p, unsigned size)
+{
+  struct mblock *b = xmalloc(sizeof(struct mblock) + size);
+
+  b->r.class = &mb_class;
+  add_tail(&p->inside, &b->r.n);
+  b->size = size;
+  return b->data;
+}
+
+void
+mb_free(void *m)
+{
+  struct mblock *b = SKIP_BACK(struct mblock, data, m);
+  rfree(b);
+}
diff --git a/lib/resource.h b/lib/resource.h
index eabc963..03c083a 100644
--- a/lib/resource.h
+++ b/lib/resource.h
@@ -14,42 +14,43 @@
 /* Resource */
 
 typedef struct resource {
-	node n;				/* Inside resource pool */
-	struct resclass *class;		/* Resource class */
+  node n;				/* Inside resource pool */
+  struct resclass *class;		/* Resource class */
 } resource;
 
 /* Resource class */
 
 struct resclass {
-	char *name;			/* Resource class name */
-	unsigned size;			/* Standard size of single resource */
-	void (*free)(resource *);	/* Freeing function */
-	void (*dump)(resource *);	/* Dump to debug output */
+  char *name;				/* Resource class name */
+  unsigned size;			/* Standard size of single resource */
+  void (*free)(resource *);		/* Freeing function */
+  void (*dump)(resource *);		/* Dump to debug output */
 };
 
 /* Generic resource manipulation */
 
 typedef struct pool pool;
 
+void resource_init(void);
 pool *rp_new(pool *);			/* Create new pool */
-void rp_init(pool *);			/* Initialize static pool */
-void rp_empty(pool *);			/* Free everything in the pool */
+void rp_free(pool *);			/* Free everything in the pool */
 void rfree(void *);			/* Free single resource */
 void rdump(void *);			/* Dump to debug output */
 
-void ralloc(pool *, struct resclass *);
+void *ralloc(pool *, struct resclass *);
+
+extern pool root_pool;
 
 /* Normal memory blocks */
 
 void *mb_alloc(pool *, unsigned size);
-void *mb_free(void *);
+void mb_free(void *);
 
 /* Memory pools with linear allocation */
 
 typedef struct mempool mempool;
 
 mempool *mp_new(pool *, unsigned blk);
-void mp_trim(pool *);				/* Free unused memory */
 void *mp_alloc(mempool *, unsigned size);	/* Aligned */
 void *mp_allocu(mempool *, unsigned size);	/* Unaligned */
 void *mp_allocz(mempool *, unsigned size);	/* With clear */
diff --git a/lib/slab.c b/lib/slab.c
new file mode 100644
index 0000000..e71c7de
--- /dev/null
+++ b/lib/slab.c
@@ -0,0 +1,87 @@
+/*
+ *	BIRD Resource Manager -- SLABs
+ *
+ *	(c) 1998 Martin Mares <mj@ucw.cz>
+ *
+ *	Can be freely distributed and used under the terms of the GNU GPL.
+ */
+
+#include <stdlib.h>
+#include <string.h>
+
+#include "nest/bird.h"
+#include "lib/resource.h"
+
+/*
+ *	These are only fake, soon to change...
+ */
+
+struct sl_obj {
+  node n;
+  byte data[0];
+};
+
+struct slab {
+  resource r;
+  unsigned size;
+  list objs;
+};
+
+void slab_free(resource *r);
+void slab_dump(resource *r);
+
+struct resclass sl_class = {
+  "Slab",
+  sizeof(struct slab),
+  slab_free,
+  slab_dump
+};
+
+slab *
+sl_new(pool *p, unsigned size)
+{
+  slab *s = ralloc(p, &sl_class);
+  s->size = size;
+  init_list(&s->objs);
+  return s;
+}
+
+void *
+sl_alloc(slab *s)
+{
+  struct sl_obj *o = xmalloc(sizeof(struct sl_obj) + s->size);
+
+  add_tail(&s->objs, &o->n);
+  return o->data;
+}
+
+void
+sl_free(slab *s, void *oo)
+{
+  struct sl_obj *o = SKIP_BACK(struct sl_obj, data, oo);
+
+  rem_node(&o->n);
+  xfree(o);
+}
+
+void
+slab_free(resource *r)
+{
+  slab *s = (slab *) r;
+  struct sl_obj *o, *p;
+
+  for(o = HEAD(s->objs); p = (struct sl_obj *) o->n.next; o = p)
+    xfree(o);
+}
+
+void
+slab_dump(resource *r)
+{
+  slab *s = (slab *) r;
+  int cnt = 0;
+  struct sl_obj *o;
+
+  WALK_LIST(o, s->objs)
+    cnt++;
+  debug("(%d objects per %d bytes)\n", cnt, s->size);
+}
diff --git a/lib/xmalloc.c b/lib/xmalloc.c
new file mode 100644
index 0000000..c1ce7ea
--- /dev/null
+++ b/lib/xmalloc.c
@@ -0,0 +1,21 @@
+/*
+ *	BIRD Library -- malloc() With Checking
+ *
+ *	(c) 1998 Martin Mares <mj@ucw.cz>
+ *
+ *	Can be freely distributed and used under the terms of the GNU GPL.
+ */
+
+#include <stdlib.h>
+
+#include "nest/bird.h"
+#include "lib/resource.h"
+
+void *
+xmalloc(unsigned size)
+{
+  void *p = malloc(size);
+  if (p)
+    return p;
+  die("Unable to allocate %d bytes of memory", size);
+}