From 8918ebf3c6e1ef4b801f9690b439f50f83fe9920 Mon Sep 17 00:00:00 2001
From: Rob McMullen <feedback@playermissile.com>
Date: Wed, 2 Mar 2016 14:25:12 -0800
Subject: [PATCH] Added boot segment parsing for raw boot disks (no assumptions
 about valid DOS flags in the boot sector)

---
 atrcopy/diskimages.py | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/atrcopy/diskimages.py b/atrcopy/diskimages.py
index 4ea3802..597bb6e 100755
--- a/atrcopy/diskimages.py
+++ b/atrcopy/diskimages.py
@@ -202,8 +202,27 @@ class DiskImageBase(object):
         self.segments.extend(self.get_directory_segments())
         self.segments.extend(self.get_file_segments())
     
+    boot_record_type = np.dtype([
+        ('BFLAG', 'u1'),
+        ('BRCNT', 'u1'),
+        ('BLDADR', '<u2'),
+        ('BWTARR', '<u2'),
+        ])
+    
     def get_boot_segments(self):
-        return []
+        data, style = self.get_sectors(1)
+        values = data[0:6].view(dtype=self.boot_record_type)[0]  
+        flag = int(values[0])
+        segments = []
+        if flag == 0:
+            num = int(values[1])
+            addr = int(values[2])
+            bytes, style = self.get_sectors(1, num)
+            header = ObjSegment(bytes[0:6], style[0:6], 0, 0, addr, addr + 6, name="Boot Header")
+            sectors = ObjSegment(bytes, style, 0, 0, addr, addr + len(bytes), name="Boot Sectors")
+            code = ObjSegment(bytes[6:], style[6:], 0, 0, addr + 6, addr + len(bytes), name="Boot Code")
+            segments = [sectors, header, code]
+        return segments
     
     def get_vtoc_segments(self):
         return []