From: Vicent Marti <tan...@gmail.com>

The caller may hand us an unaligned buffer (e.g., because it
is an mmap of a file with many ewah bitmaps). On some
platforms (like SPARC) this can cause a bus error. We can
fix it with a combination of force-align macros and moving
the data into an aligned buffer (which we would do anyway,
but we can move it before fixing the endianness).

Signed-off-by: Vicent Marti <tan...@gmail.com>
Signed-off-by: Jeff King <p...@peff.net>
---
 ewah/ewah_io.c | 33 ++++++++++++++++++++++++---------
 1 file changed, 24 insertions(+), 9 deletions(-)

diff --git a/ewah/ewah_io.c b/ewah/ewah_io.c
index aed0da6..1948ba5 100644
--- a/ewah/ewah_io.c
+++ b/ewah/ewah_io.c
@@ -112,23 +112,38 @@ int ewah_serialize(struct ewah_bitmap *self, int fd)
 
 int ewah_read_mmap(struct ewah_bitmap *self, void *map, size_t len)
 {
-       uint32_t *read32 = map;
-       eword_t *read64;
-       size_t i;
+       uint8_t *ptr = map;
+
+       self->bit_size = align_ntohl(*(uint32_t *)ptr);
+       ptr += sizeof(uint32_t);
+
+       self->buffer_size = self->alloc_size = align_ntohl(*(uint32_t *)ptr);
+       ptr += sizeof(uint32_t);
 
-       self->bit_size = ntohl(*read32++);
-       self->buffer_size = self->alloc_size = ntohl(*read32++);
        self->buffer = ewah_realloc(self->buffer,
                self->alloc_size * sizeof(eword_t));
 
        if (!self->buffer)
                return -1;
 
-       for (i = 0, read64 = (void *)read32; i < self->buffer_size; ++i)
-               self->buffer[i] = ntohll(*read64++);
+       /*
+        * Copy the raw data for the bitmap as a whole chunk;
+        * if we're in a little-endian platform, we'll perform
+        * the endianness conversion in a separate pass to ensure
+        * we're loading 8-byte aligned words.
+        */
+       memcpy(self->buffer, ptr, self->buffer_size * sizeof(uint64_t));
+       ptr += self->buffer_size * sizeof(uint64_t);
+
+#if __BYTE_ORDER != __BIG_ENDIAN
+       {
+               size_t i;
+               for (i = 0; i < self->buffer_size; ++i)
+                       self->buffer[i] = ntohll(self->buffer[i]);
+       }
+#endif
 
-       read32 = (void *)read64;
-       self->rlw = self->buffer + ntohl(*read32++);
+       self->rlw = self->buffer + align_ntohl(*(uint32_t *)ptr);
 
        return (3 * 4) + (self->buffer_size * 8);
 }
-- 
1.8.5.2.500.g8060133
--
To unsubscribe from this list: send the line "unsubscribe git" in
the body of a message to majord...@vger.kernel.org
More majordomo info at  http://vger.kernel.org/majordomo-info.html

Reply via email to