Apply fixes suggested by jjk

Signed-off-by: Laurent Bercot <ska@appnovation.com>

3 files changed, 9 insertions, 7 deletions

diff --git a/src/libstddjb/buffer_get.c b/src/libstddjb/buffer_get.c
index fbc0de3..6fe394d 100644
--- a/src/libstddjb/buffer_get.c
+++ b/src/libstddjb/buffer_get.c
@@ -1,12 +1,13 @@
 /* ISC license. */
 
-#include <errno.h>
+#include <stddef.h>
+
+#include <skalibs/allreadwrite.h>
 #include <skalibs/buffer.h>
 
 ssize_t buffer_get (buffer *b, char *s, size_t len)
 {
   size_t w = 0 ;
-  int r = buffer_getall(b, s, len, &w) ;
-  return r == -1 ? errno == EPIPE ? (errno = 0, w) : -1 :
-         !r ? (errno = EWOULDBLOCK, -1) : w ;
+  ssize_t r = unsanitize_read(buffer_getall(b, s, len, &w)) ;
+  return r < 0 ? r : w ;
 }
diff --git a/src/libstddjb/buffer_getv.c b/src/libstddjb/buffer_getv.c
index 280a3b4..ab637a3 100644
--- a/src/libstddjb/buffer_getv.c
+++ b/src/libstddjb/buffer_getv.c
@@ -1,6 +1,7 @@
 /* ISC license. */
 
-#include <sys/uio.h>
+#include <stddef.h>
+
 #include <skalibs/allreadwrite.h>
 #include <skalibs/buffer.h>
 
@@ -8,5 +9,5 @@ ssize_t buffer_getv (buffer *b, struct iovec const *v, unsigned int n)
 {
   size_t w = 0 ;
   ssize_t r = unsanitize_read(buffer_getvall(b, v, n, &w)) ;
-  return r <= 0 ? r : w ;
+  return r < 0 ? r : w ;
 }
diff --git a/src/libstddjb/slurpn.c b/src/libstddjb/slurpn.c
index c9c5c11..ee5157f 100644
--- a/src/libstddjb/slurpn.c
+++ b/src/libstddjb/slurpn.c
@@ -16,7 +16,7 @@ int slurpn (int fd, stralloc *sa, size_t max)
   for (;;)
   {
     ssize_t r ;
-    size_t n = max && sa->len + N > max ? max - sa->len : N ;
+    size_t n = max && sa->len - sabase + N > max ? max - (sa->len - sabase) : N ;
     if (!n) { errno = ENOBUFS ; goto err ; }
     if (!stralloc_readyplus(sa, n)) break ;
     r = fd_read(fd, sa->s + sa->len, n) ;