Mesa (master): nir/spirv: improve lseek() error handling

[Jason Ekstrand]

Module: Mesa
Branch: master
Commit: 65c8cbe89df3c0e23e83f32111cce6609f86c85d
URL:    http://cgit.freedesktop.org/mesa/mesa/commit/?id=65c8cbe89df3c0e23e83f32111cce6609f86c85d

Author: Eric Engestrom <eric at engestrom.ch>
Date:   Sun Sep 25 16:49:51 2016 +0100

nir/spirv: improve lseek() error handling

Signed-off-by: Eric Engestrom <eric at engestrom.ch>
Reviewed-by: Jason Ekstrand <jason at jlekstrand.net>

---

 src/compiler/spirv/spirv2nir.c | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/compiler/spirv/spirv2nir.c b/src/compiler/spirv/spirv2nir.c
index d151c2c..a024028 100644
--- a/src/compiler/spirv/spirv2nir.c
+++ b/src/compiler/spirv/spirv2nir.c
@@ -39,6 +39,8 @@
 #include <unistd.h>
 #include <stdio.h>
 
+#define WORD_SIZE 4
+
 int main(int argc, char **argv)
 {
    int fd = open(argv[1], O_RDONLY);
@@ -49,9 +51,15 @@ int main(int argc, char **argv)
    }
 
    off_t len = lseek(fd, 0, SEEK_END);
+   if (len % WORD_SIZE != 0)
+   {
+      fprintf(stderr, "File length isn't a multiple of the word size\n");
+      fprintf(stderr, "Are you sure this is a valid SPIR-V shader?\n");
+      close(fd);
+      return 1;
+   }
 
-   assert(len % 4 == 0);
-   size_t word_count = len / 4;
+   size_t word_count = len / WORD_SIZE;
 
    const void *map = mmap(NULL, len, PROT_READ, MAP_PRIVATE, fd, 0);
    assert(map != NULL);