Get page size using 'mmap' module if 'resource' is not available.

The resource module is Unix only, use mmap for Windows.

diff --git a/elftools/elf/elffile.py b/elftools/elf/elffile.py
index 0517b578023d7a75546b0588fcb8639b71596d84..2bc2651bb830d78fdb25702be9a165574ebb6ef1 100644 (file)
--- a/elftools/elf/elffile.py
+++ b/elftools/elf/elffile.py
@@ -7,9 +7,17 @@
 # This code is in the public domain
 #-------------------------------------------------------------------------------
 import io
-import resource
 import struct
 import zlib
+
+try:
+    import resource
+    PAGESIZE = resource.getpagesize()
+except ImportError:
+    # Windows system
+    import mmap
+    PAGESIZE = mmap.PAGESIZE
+
 from ..common.py3compat import BytesIO
 from ..common.exceptions import ELFError
 from ..common.utils import struct_parse, elf_assert
@@ -416,7 +424,7 @@ class ELFFile(object):
         decompressor = zlib.decompressobj()
         uncompressed_stream = BytesIO()
         while True:
-            chunk = section.stream.read(resource.getpagesize())
+            chunk = section.stream.read(PAGESIZE)
             if not chunk:
                 break
             uncompressed_stream.write(decompressor.decompress(chunk))