Uprating of the passphrase pixie.
[u/mdw/catacomb] / buf.c
diff --git a/buf.c b/buf.c
index 5e0a069..e34d876 100644 (file)
--- a/buf.c
+++ b/buf.c
@@ -1,6 +1,6 @@
 /* -*-c-*-
  *
- * $Id: buf.c,v 1.2 2003/11/10 22:18:30 mdw Exp $
+ * $Id$
  *
  * Buffer handling
  *
  * MA 02111-1307, USA.
  */
 
-/*----- Revision history --------------------------------------------------* 
- *
- * $Log: buf.c,v $
- * Revision 1.2  2003/11/10 22:18:30  mdw
- * Build fixes.
- *
- * Revision 1.1  2003/10/11 21:02:33  mdw
- * Import buf stuff from tripe.
- *
- * Revision 1.4  2001/06/19 22:09:54  mdw
- * Expose interface, for use in the proxy.
- *
- * Revision 1.3  2001/03/03 12:06:48  mdw
- * Use 16-bit lengths on MPs, since there's a packet limit of 64K anyway.
- *
- * Revision 1.2  2001/02/16 21:23:20  mdw
- * Various minor changes.  Check that MPs are in canonical form when
- * loading.
- *
- * Revision 1.1  2001/02/03 20:26:37  mdw
- * Initial checkin.
- *
- */
-
 /*----- Header files ------------------------------------------------------*/
 
 #include <string.h>
 
 #include "mp.h"
+#include "ec.h"
 #include "buf.h"
 
 /*----- Main code ---------------------------------------------------------*/
@@ -268,6 +245,202 @@ int buf_putu32(buf *b, uint32 w)
   return (0);
 }
 
+/* --- @findz@ --- *
+ *
+ * Arguments:  @buf *b@ = pointer to a buffer block
+ *             @size_t *nn@ = where to put the length
+ *
+ * Returns:    Zero if OK, nonzero if there wasn't a null byte to be found.
+ *
+ * Use:                Finds a terminating null byte.
+ */
+
+static int findz(buf *b, size_t *nn)
+{
+  octet *p;
+
+  if ((p = memchr(BCUR(b), 0, BLEN(b))) == 0) {
+    buf_break(b);
+    return (-1);
+  }
+  *nn = p - BCUR(b);
+  return (0);
+}
+
+#define DOSUFFIXES(DO) DO(8) DO(16) DO(32) DO(z)
+
+/* --- @buf_getmem{8,16,32,z} --- *
+ *
+ * Arguments:  @buf *b@ = pointer to a buffer block
+ *             @size_t *nn@ = where to put the length
+ *
+ * Returns:    Pointer to the buffer data, or null.
+ *
+ * Use:                Gets a chunk of memory from a buffer.  The number, @16@ or
+ *             @32@, is the width of the length; @z@ means null-terminated.
+ */
+
+void *buf_getmem8(buf *b, size_t *nn)
+{
+  int n;
+
+  if ((n = buf_getbyte(b)) < 0) return (0);
+  *nn = n;
+  return (buf_get(b, n));
+}
+
+void *buf_getmem16(buf *b, size_t *nn)
+{
+  uint16 n;
+
+  if (buf_getu16(b, &n)) return (0);
+  *nn = n;
+  return (buf_get(b, n));
+}
+
+void *buf_getmem32(buf *b, size_t *nn)
+{
+  uint32 n;
+
+  if (buf_getu32(b, &n)) return (0);
+  *nn = n;
+  return (buf_get(b, n));
+}
+  
+void *buf_getmemz(buf *b, size_t *nn)
+{
+  if (findz(b, nn)) return (0);
+  return (buf_get(b, *nn));
+}
+
+/* --- @buf_putmem{8,16,32,z} --- *
+ *
+ * Arguments:  @buf *b@ = pointer to a buffer block
+ *             @const void *p@ = pointer to data to write
+ *             @size_t n@ = length to write
+ *
+ * Returns:    Zero if OK, nonzero if there wasn't enough space.
+ *
+ * Use:                Writes a chunk of data to a buffer.  The number, @16@ or
+ *             @32@, is the width of the length; @z@ means null-terminated.
+ */
+
+int buf_putmem8(buf *b, const void *p, size_t n)
+{
+  assert(n <= 0xfful);
+  if (buf_putbyte(b, n) || buf_put(b, p, n))
+    return (-1);
+  return (0);
+}
+
+int buf_putmem16(buf *b, const void *p, size_t n)
+{
+  assert(n <= 0xfffful);
+  if (buf_putu16(b, n) || buf_put(b, p, n))
+    return (-1);
+  return (0);
+}
+
+int buf_putmem32(buf *b, const void *p, size_t n)
+{
+  assert(n <= 0xfffffffful);
+  if (buf_putu32(b, n) || buf_put(b, p, n))
+    return (-1);
+  return (0);
+}
+
+int buf_putmemz(buf *b, const void *p, size_t n)
+{
+  octet *q;
+
+  assert(!memchr(p, 0, n));
+  if ((q = buf_get(b, n + 1)) == 0)
+    return (-1);
+  memcpy(q, p, n);
+  q[n] = 0;
+  return (0);
+}
+
+/* --- @buf_getbuf{8,16,32,z}@ --- *
+ *
+ * Arguments:  @buf *b@ = pointer to a buffer block
+ *             @buf *bb@ = where to put the result
+ *
+ * Returns:    Zero if it worked, nonzero if there wasn't enough space.
+ *
+ * Use:                Gets a block of data from a buffer, and writes its bounds to
+ *             another buffer.  The number, @16@ or @32@, is the width of
+ *             the length; @z@ means null-terminated.
+ */
+
+#define GETBUF(suff) int buf_getbuf##suff(buf *b, buf *bb)             \
+{                                                                      \
+  void *p;                                                             \
+  size_t n;                                                            \
+                                                                       \
+  if ((p = buf_getmem##suff(b, &n)) == 0)                              \
+    return (-1);                                                       \
+  buf_init(bb, p, n);                                                  \
+  return (0);                                                          \
+}
+DOSUFFIXES(GETBUF)
+
+/* --- @buf_putbuf{8,16,32,z}@ --- *
+ *
+ * Arguments:  @buf *b@ = pointer to a buffer block
+ *             @buf *bb@ = buffer to write
+ *
+ * Returns:    Zero if it worked, nonzero if there wasn't enough space.
+ *
+ * Use:                Puts the contents of a buffer to a buffer.  The number, @16@
+ *             or @32@, is the width of the length; @z@ means null-
+ *             terminated.
+ */
+
+#define PUTBUF(suff) int buf_putbuf##suff(buf *b, buf *bb)             \
+  { return (buf_putmem##suff(b, BBASE(bb), BLEN(bb))); }
+DOSUFFIXES(PUTBUF)  
+
+/* --- @buf_getstr{8,16,32,z}@ --- *
+ *
+ * Arguments:  @buf *b@ = pointer to a buffer block
+ *             @dstr *d@ = where to put the result
+ *
+ * Returns:    Zero if it worked, nonzero if there wasn't enough space.
+ *
+ * Use:                Gets a block of data from a buffer, and writes its contents
+ *             to a string.  The number, @16@ or @32@, is the width of the
+ *             length; @z@ means null-terminated.
+ */
+
+#define GETSTR(suff) int buf_getstr##suff(buf *b, dstr *d)             \
+{                                                                      \
+  void *p;                                                             \
+  size_t n;                                                            \
+                                                                       \
+  if ((p = buf_getmem##suff(b, &n)) == 0)                              \
+    return (-1);                                                       \
+  DPUTM(d, p, n);                                                      \
+  return (0);                                                          \
+}
+DOSUFFIXES(GETSTR)
+
+/* --- @buf_putstr{8,16,32,z}@ --- *
+ *
+ * Arguments:  @buf *b@ = pointer to a buffer block
+ *             @dstr *d@ = string to write
+ *
+ * Returns:    Zero if it worked, nonzero if there wasn't enough space.
+ *
+ * Use:                Puts a string to a buffer, and writes its bounds to
+ *             another buffer.  The number, @16@ or @32@, is the width of
+ *             the length; @z@ means null-terminated.
+ */
+
+#define PUTSTR(suff) int buf_putstr##suff(buf *b, dstr *d)             \
+  { return (buf_putmem##suff(b, d->buf, d->len)); }
+DOSUFFIXES(PUTSTR)
+
 /* --- @buf_getmp@ --- *
  *
  * Arguments:  @buf *b@ = pointer to a buffer block
@@ -280,11 +453,13 @@ int buf_putu32(buf *b, uint32 w)
 mp *buf_getmp(buf *b)
 {
   uint16 sz;
+  size_t n;
   mp *m;
   if (buf_getu16(b, &sz) || buf_ensure(b, sz))
     return (0);
   m = mp_loadb(MP_NEW, BCUR(b), sz);
-  if (mp_octets(m) != sz) {
+  n = mp_octets(m);
+  if (n != sz && n != 0 && sz != 1) {
     mp_drop(m);
     return (0);
   }
@@ -306,6 +481,7 @@ int buf_putmp(buf *b, mp *m)
 {
   size_t sz = mp_octets(m);
   assert(sz < MASK16);
+  if (!sz) sz = 1;
   if (buf_putu16(b, sz) || buf_ensure(b, sz))
     return (-1);
   mp_storeb(m, BCUR(b), sz);
@@ -313,4 +489,45 @@ int buf_putmp(buf *b, mp *m)
   return (0);
 }
 
+/* --- @buf_getec@ --- *
+ *
+ * Arguments:  @buf *b@ = pointer to a buffer block
+ *             @ec *p@ = where to put the point
+ *
+ * Returns:    Zero if it worked, nonzero if it failed.
+ *
+ * Use:                Gets a multiprecision integer from a buffer.  The point must
+ *             be initialized.
+ */
+
+int buf_getec(buf *b, ec *p)
+{
+  mp *x = 0, *y = 0;
+  uint16 n;
+  if (buf_ensure(b, 2)) return (-1);
+  n = LOAD16(BCUR(b)); if (!n) { BSTEP(b, 2); EC_SETINF(p); return (0); }
+  if ((x = buf_getmp(b)) == 0 || (y = buf_getmp(b)) == 0) {
+    mp_drop(x); mp_drop(y); return (-1);
+  }
+  EC_DESTROY(p); p->x = x; p->y = y; p->z = 0;
+  return (0);
+}
+
+/* --- @buf_putec@ --- *
+ *
+ * Arguments:  @buf *b@ = pointer to a buffer block
+ *             @ec *p@ = an elliptic curve point
+ *
+ * Returns:    Zero if it worked, nonzero if there wasn't enough space.
+ *
+ * Use:                Puts an elliptic curve point to a buffer.
+ */
+
+int buf_putec(buf *b, ec *p)
+{
+  if (EC_ATINF(p)) return (buf_putu16(b, 0));
+  if (buf_putmp(b, p->x) || buf_putmp(b, p->y)) return (-1);
+  return (0);
+}
+
 /*----- That's all, folks -------------------------------------------------*/