fastbufs: added a new function for safe reading of fixed-length array

diff --git a/lib/fastbuf.c b/lib/fastbuf.c
index dcb8d20dc0fe305b1229b2418ee89390bb4ae0ad..26df0c2f9afb17d202ac08d6809948b7bd633973 100644 (file)
--- a/lib/fastbuf.c
+++ b/lib/fastbuf.c
@@ -109,8 +109,11 @@ uns bread_slow(struct fastbuf *f, void *b, uns l, uns check)
       l -= k;
       total += k;
     }
-  if (check && total && l)
-    die("breadb: short read");
+  if (check && l)
+    if (check == 2)
+      die("breada: short read");
+    else if (total)
+      die("breadb: short read");
   return total;
 }
 

diff --git a/lib/fastbuf.h b/lib/fastbuf.h
index 68b2a9db18db938456fcefc309ccfb24a485ebcf..cd739ecbe33484f1a1518a625d806cf39882752e 100644 (file)
--- a/lib/fastbuf.h
+++ b/lib/fastbuf.h
@@ -235,6 +235,17 @@ static inline uns bread(struct fastbuf *f, void *b, uns l)
     return bread_slow(f, b, l, 0);
 }
 
+static inline void breada(struct fastbuf *f, void *b, uns l)
+{
+  if (bavailr(f) >= l)
+    {
+      memcpy(b, f->bptr, l);
+      f->bptr += l;
+    }
+  else
+    bread_slow(f, b, l, 2);
+}
+
 static inline uns breadb(struct fastbuf *f, void *b, uns l)
 {
   if (bavailr(f) >= l)