/snf-image-helper/disklabel.py - Diff - snf-image - Greek Research and Technology Network's projects

Revision 25cdca3e snf-image-helper/disklabel.py

                         d.seek(BLOCKSIZE * self.mbr.part[i].first_sector)
                         self.part_num = i
                         if ptype == 0xa5:  # FreeBSD
                             self.disklabel = BSDDisklabel(d)
                             self.disklabel = FreeBSDDisklabel(d)
                         elif ptype == 0xa6:  # OpenBSD
                             self.disklabel = OpenBSDDisklabel(d)
                         else:  # NetBSD
                             self.disklabel = BSDDisklabel(d)
                             self.disklabel = NetBSDDisklabel(d)
                         break
             assert self.disklabel is not None, "No *BSD partition found"
-...
             return (cylinder, header, sector)
         @abc.abstractmethod
         def enlarge(self, new_size):
             """Enlarge the disk and return the last usable sector"""
             pass
         @abc.abstractmethod
         def enlarge_last_partition(self):
             """Enlarge the last partition to consume all the usable space"""
             pass
         @abc.abstractmethod
         def get_last_partition_id(self):
             """Get the ID of the last partition"""
             pass
-...
             """Print the Disklabel"""
             pass
         def enlarge(self, new_size):
             """Enlarge the disk and return the last usable sector"""
             raise NotImplementedError
         def enlarge_last_partition(self):
             """Enlarge the last partition to consume all the usable space"""
             raise NotImplementedError
     class PartitionTableBase(object):
         """Base Class for disklabel partition tables"""
-...
                     'cpg'       # 14      2      File system cylinders per group
+                    ]
             def setsize(self, i, size):
                 """Set size for partition i"""
                 tmp = self.part[i]
                 self.part[i] = self.Partition(size, tmp.offset, tmp.fsize,
                                               tmp.fstype, tmp.frag, tmp.cpg)
         @property
         def fmt(self):
             return "<IHH16s16sIIIIIIHHIHHHHIII20s20sIHHII364s"
-...
             self.field = d_
         def enlarge(self, new_size):
             """Enlarge the disk and return the last usable sector"""
             raise NotImplementedError
         def enlarge_last_partition(self):
             raise NotImplementedError
         def get_last_partition_id(self):
             raise NotImplementedError
         def lba2chs(self, lba, hpc=None, spt=None):
             """Return the CHS address for a given LBA address"""
             # NetBSD uses LBA-Assisted translation. The sectors per track (spt)
             # value is always 63 and the heads per cylinder (hpc) value depends on
             # the size of the disk
             disk_size = self.field['secperunit']
             # Maximum disk size that can be addressed is 1024 * HPC* SPT
             spt = 63
             for hpc in 16, 32, 64, 128, 255:
                 if disk_size <= 1024 * hpc * spt:
                     break
             """Returns the id of the last partition"""
             end = 0
             chs = super(BSDDisklabel, self).lba2chs(lba, hpc, spt)
             # Exclude partition 'c' which describes the whole disk
             for i in [n for n in range(len(self.ptable.part)) if n != 2]:
                 current_end = self.ptable.part[i].size + self.ptable.part[i].offset
                 if end < current_end:
                     end = current_end
                     part = i
             if chs[0] > 1023:  # Cylinders overflowed
                 assert hpc == 255  # Cylinders may overflow only for large disks
                 return (1023, chs[1], chs[2])
             assert end > 0, "No partition found"
             return chs
             return part
         def __str__(self):
             """Print the Disklabel"""
-...
                 "%s" % self.ptable
     class FreeBSDDisklabel(BSDDisklabel):
         """Represents a FreeBSD Disklabel"""
         pass
     class NetBSDDisklabel(BSDDisklabel):
         """Represents a NetBSD Disklabel"""
         def lba2chs(self, lba, hpc=None, spt=None):
             """Return the CHS address for a given LBA address"""
             # NetBSD uses LBA-Assisted translation. The sectors per track (spt)
             # value is always 63 and the heads per cylinder (hpc) value depends on
             # the size of the disk
             disk_size = self.field['secperunit']
             # Maximum disk size that can be addressed is 1024 * HPC* SPT
             spt = 63
             for hpc in 16, 32, 64, 128, 255:
                 if disk_size <= 1024 * hpc * spt:
                     break
             chs = super(NetBSDDisklabel, self).lba2chs(lba, hpc, spt)
             if chs[0] > 1023:  # Cylinders overflowed
                 assert hpc == 255  # Cylinders may overflow only for large disks
                 return (1023, chs[1], chs[2])
             return chs
         def enlarge(self, new_size):
             """Enlarge the disk and return the last usable sector"""
             assert new_size >= self.field['secperunit'], \
                 "New size cannot be smaller than %d" % self.field['secperunit']
             self.field['secperunit'] = new_size
             # According to the ATA specifications, "If the content of words (61:60)
             # is greater than or equal to 16,514,064 then the content of word 1
             # [the number of logical cylinders] shall be equal to 16,383."
             # NetBSD respects this.
             self.field['ncylinders'] = 16383 if new_size >= 16514064 else \
                 new_size // self.field['secpercyl']
             # Partition 'c' describes the NetBSD MBR partition
             self.ptable.setsize(2, new_size - self.ptable.part[2].offset)
             # Partition 'd' describes the entire disk
             self.ptable.setsize(3, new_size)
             # Update the checksum
             self.field['checksum'] = self.compute_checksum()
             return new_size - 1
         def enlarge_last_partition(self):
             """Enlarge the last partition to cover up all the free space"""
             last = self.get_last_partition_id()
             size = self.field['secperunit'] - self.ptable.part[last].offset
             self.ptable.setsize(last, size)
             self.field['checksum'] = self.compute_checksum()
         def get_last_partition_id(self):
             """Returns the id of the last partition"""
             end = 0
             # Exclude partition 'c' which describes the NetBSD MBR partition and
             # partition 'd' which describes the whole disk
             for i in [n for n in range(len(self.ptable.part)) if n not in (2, 3)]:
                 current_end = self.ptable.part[i].size + self.ptable.part[i].offset
                 if end < current_end:
                     end = current_end
                     part = i
             assert end > 0, "No partition found"
             return part
     class OpenBSDDisklabel(DisklabelBase):
         """Represents an OpenBSD Disklabel"""
-...
             """Enlarge the disk and return the last usable sector"""
             assert new_size >= self.dsize, \
                 "New size cannot be smaller that %s" % self.dsize
                 "New size cannot be smaller than %d" % self.dsize
             # Fix the disklabel
             self.dsize = new_size
-...
             self.bend = (self.field['ncylinders'] * self.field['nsectors'] *
                          self.field['ntracks'])
             # Partition 'c' descriptes the entire disk
             # Partition 'c' describes the entire disk
             self.ptable.setpsize(2, new_size)
             # Update the checksum
-...
         def get_last_partition_id(self):
             """Returns the id of the last partition"""
             part = 0
             end = 0
             # Don't check partition 'c' which is the whole disk
             for i in filter(lambda x: x != 2, range(len(self.ptable.part))):
             for i in [n for n in range(len(self.ptable.part)) if n != 2]:
                 curr_end = self.ptable.getpsize(i) + self.ptable.getpoffset(i)
                 if end < curr_end:
                     end = curr_end

Also available in: Unified diff

Synnefo » snf-image

Revision 25cdca3e snf-image-helper/disklabel.py