1 files changed, 96 insertions, 0 deletions
diff --git a/mybinview.py b/mybinview.py
new file mode 100644
index 0000000..5064013
--- /dev/null
+++ b/mybinview.py
@@ -0,0 +1,96 @@
+#!/usr/bin/env python
+
+# RaidGuessFS, a FUSE pseudo-filesystem to guess RAID parameters of a damaged device
+# Copyright (C) 2015 Ludovic Pouzenc <ludovic@pouzenc.fr>
+#
+# This file is part of RaidGuessFS.
+#
+# RaidGuessFS is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# RaidGuessFS is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with RaidGuessFS. If not, see <http://www.gnu.org/licenses/>
+
+import struct
+
+class MyBinView():
+    """"Auxiliary class, dumping binary data as image files"""
+
+    def __init__(self, *args, **kwargs):
+        self.bmp_start_offset=0
+        self.bmp_height=4096
+        self.bmp_width=1024
+
+        self.bmp_pixeldata_start=0
+        self.bmp_size=0
+        self.bmp_header=bytearray()
+
+    def get_bmp_start_offset(self):
+        return self.bmp_start_offset
+
+    def get_bmp_height(self):
+        return self.bmp_height
+
+    def get_bmp_width(self):
+        return self.bmp_width
+
+    def set_bmp_start_offset(self, new_bmp_start_offset):
+        """Offset to choose which fiel part you want to render as image"""
+        self.bmp_start_offset = new_bmp_start_offset
+
+    def set_bmp_width(self, new_bmp_width):
+        """Update the BMP (yes bitmap images) width in pixels"""
+        self.bmp_width = new_bmp_width / 4 * 4
+        self.refresh_bmp()
+        self.refresh_disks_dentries()
+
+    def set_bmp_height(self, new_bmp_height):
+        """Update the BMP (yes bitmap images) height in pixels"""
+        self.bmp_height = new_bmp_height
+        self.refresh_bmp()
+        self.refresh_disks_dentries()
+
+    def refresh_bmp(self):
+        """Update the BMP headers accroding to bitmap settings"""
+        bmp_header_fmt = struct.Struct('< 2s I x x x x I I i i H H I I i i I I')
+
+        bitmap_pixel_count = self.bmp_width * self.bmp_height
+        # From https://en.wikipedia.org/wiki/BMP_file_format
+        bitmap_magic='BM'
+        dib_size=40 # BITMAPINFOHEADER (Windows NT, 3.1x or later)
+        bitmap_bpp=8
+        bitmap_compress=0
+        imagehres=2835 # pixels per meter
+        imagevres=2835
+        palette_size=256
+        palette_start = bmp_header_fmt.size
+
+        self.bmp_pixeldata_start = palette_start + palette_size*4
+        self.bmp_size = self.bmp_pixeldata_start + bitmap_pixel_count
+        self.bmp_header = bytearray(self.bmp_pixeldata_start)
+        bmp_header_fmt.pack_into(self.bmp_header,0,
+                'BM', self.bmp_size, self.bmp_pixeldata_start, dib_size,
+                self.bmp_width, self.bmp_height, 1, bitmap_bpp, bitmap_compress, 
+                bitmap_pixel_count, imagehres, imagevres, palette_size, palette_size
+                )
+
+        for i in range(palette_size):
+            struct.pack_into('< B B B B', self.bmp_header, bmp_header_fmt.size + i*4, i,i,i,0) # All shades of gray in RGBA
+
+    def read(self, fd, offset, size):
+        if offset < self.bmp_pixeldata_start:
+            data = str(self.bmp_header[offset:offset+size-1])
+            if offset + size > self.bmp_pixeldata_start:
+                data += self.read(fd, self.bmp_pixeldata_start, size - self.bmp_pixeldata_start)
+            return data
+        else:
+            fd.seek(self.bmp_start_offset + offset - self.bmp_pixeldata_start)
+            return fd.read(size)
+