BUG: work round int overflow in size calculation

Omar pointed out the integer overflow in calculating the image size of
an image of size > 4GB, on a 32-bit platform.

Author: matthew-brett

nibabel/tests/test_utils.py

@@ -1227,3 +1227,16 @@ def assert_rt(data,
                       slope = slope,
                       post_clips = post_clips,
                       nan_fill = nan_fill)
+
+
+def test_array_from_file_overflow():
+    # Test for int overflow in size calculation in array_from_file
+    shape = (1500,) * 6
+    try:
+        array_from_file(shape, np.int8, BytesIO())
+    except IOError as err:
+        pass
+    assert_equal(err.message,
+                 'Expected {0} bytes, got {1} bytes from {2}\n'
+                 ' - could the file be damaged?'.format(
+                     11390625000000000000, 0, 'object'))

nibabel/volumeutils.py

@@ -14,6 +14,9 @@
 import gzip
 import bz2
 from os.path import exists, splitext
+from operator import mul
+
+from .externals.six.moves import reduce
 
 import numpy as np
 
@@ -504,7 +507,8 @@ def array_from_file(shape, in_dtype, infile, offset=0, order='F', mmap=True):
             pass
     if len(shape) == 0:
         return np.array([])
-    n_bytes = int(np.prod(shape) * in_dtype.itemsize)
+    # Use reduce and mul to work round integer overflow on 32-bit platforms
+    n_bytes = reduce(mul, shape) * in_dtype.itemsize
     if n_bytes == 0:
         return np.array([])
     # Read data from file