base64: fix for unsigned chars (e.g. ARM).

[ccan] / ccan / mem / mem.c

diff --git a/ccan/mem/mem.c b/ccan/mem/mem.c
index 91c9961a52ecd92d1372dae98b07f54902e25f22..13027a2a7b0fec1b18ae386f9704d84aed41c5f1 100644 (file)
--- a/ccan/mem/mem.c
+++ b/ccan/mem/mem.c
@@ -2,6 +2,7 @@
 
 #include "config.h"
 
 
 #include "config.h"
 

+#include <assert.h>

 #include <string.h>
 #include <ccan/mem/mem.h>
 
 #include <string.h>
 #include <ccan/mem/mem.h>
 


@@ -64,3 +65,64 @@ void *memcchr(void const *data, int c, size_t data_len)
 
        return NULL;
 }
 
        return NULL;
 }

+
+#define MEMSWAP_TMP_SIZE       256
+
+void memswap(void *a, void *b, size_t n)
+{
+       char *ap = a;
+       char *bp = b;
+       char tmp[MEMSWAP_TMP_SIZE];
+
+       assert(!memoverlaps(a, n, b, n));
+
+       while (n) {
+               size_t m = n > MEMSWAP_TMP_SIZE ? MEMSWAP_TMP_SIZE : n;
+
+               memcpy(tmp, bp, m);
+               memcpy(bp, ap, m);
+               memcpy(ap, tmp, m);
+
+               ap += m;
+               bp += m;
+               n -= m;
+       }
+}
+
+bool memeqzero(const void *data, size_t length)
+{
+       const unsigned char *p = data;
+       size_t len;
+
+       /* Check first 16 bytes manually */
+       for (len = 0; len < 16; len++) {
+               if (!length)
+                       return true;
+               if (*p)
+                       return false;
+               p++;
+               length--;
+       }
+
+       /* Now we know that's zero, memcmp with self. */
+       return memcmp(data, p, length) == 0;
+}
+
+void memtaint(void *data, size_t len)
+{
+       /* Using 16 bytes is a bit quicker than 4 */
+       const unsigned tainter[]
+               = { 0xdeadbeef, 0xdeadbeef, 0xdeadbeef, 0xdeadbeef };
+       char *p = data;
+
+       while (len >= sizeof(tainter)) {
+               memcpy(p, tainter, sizeof(tainter));
+               p += sizeof(tainter);
+               len -= sizeof(tainter);
+       }
+       memcpy(p, tainter, len);
+
+#if HAVE_VALGRIND_MEMCHECK_H
+       VALGRIND_MAKE_MEM_UNDEFINED(data, len);
+#endif
+}