Subversion Repositories HelenOS

Rev

Rev 3263 | Blame | Compare with Previous | Last modification | View Log | Download | RSS feed

  1. #!/usr/bin/env python
  2. #
  3. # Copyright (c) 2008 Martin Decky
  4. # All rights reserved.
  5. #
  6. # Redistribution and use in source and binary forms, with or without
  7. # modification, are permitted provided that the following conditions
  8. # are met:
  9. #
  10. # - Redistributions of source code must retain the above copyright
  11. #   notice, this list of conditions and the following disclaimer.
  12. # - Redistributions in binary form must reproduce the above copyright
  13. #   notice, this list of conditions and the following disclaimer in the
  14. #   documentation and/or other materials provided with the distribution.
  15. # - The name of the author may not be used to endorse or promote products
  16. #   derived from this software without specific prior written permission.
  17. #
  18. # THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  19. # IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  20. # OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  21. # IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  22. # INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  23. # NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  24. # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  25. # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  26. # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  27. # THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28. #
  29. """
  30. HORD encapsulator
  31. """
  32.  
  33. import sys
  34. import os
  35. import xstruct
  36.  
  37. HEADER = """little:
  38.     char tag[4]            /* 'HORD' */
  39.     uint8_t version        /* version */
  40.     uint8_t encoding       /* encoding */
  41.     uint32_t header_size   /* header size */
  42.     uint64_t payload_size  /* payload size */
  43. """
  44.  
  45. HORD_LSB = 1
  46.  
  47. def align_up(size, alignment):
  48.     "Align upwards to a given alignment"
  49.     return (((size) + ((alignment) - 1)) & ~((alignment) - 1))
  50.  
  51. def usage(prname):
  52.     "Print usage syntax"
  53.     print prname + " <ALIGNMENT> <FS_IMAGE> <HORD_IMAGE>"
  54.  
  55. def main():
  56.     if (len(sys.argv) < 4):
  57.         usage(sys.argv[0])
  58.         return
  59.    
  60.     if (not sys.argv[1].isdigit()):
  61.         print "<ALIGNMENT> must be a number"
  62.         return
  63.    
  64.     align = int(sys.argv[1], 0)
  65.     if (align <= 0):
  66.         print "<ALIGNMENT> must be positive"
  67.         return
  68.    
  69.     fs_image = os.path.abspath(sys.argv[2])
  70.     if (not os.path.isfile(fs_image)):
  71.         print "<FS_IMAGE> must be a file"
  72.         return
  73.    
  74.     inf = file(fs_image, "rb")
  75.     outf = file(sys.argv[3], "wb")
  76.    
  77.     header = xstruct.create(HEADER)
  78.    
  79.     header_size = header.size()
  80.     payload_size = os.path.getsize(fs_image)
  81.    
  82.     header_size_aligned = align_up(header_size, align)
  83.     payload_size_aligned = align_up(payload_size, align)
  84.    
  85.     header.tag = "HORD"
  86.     header.version = 1
  87.     header.encoding = HORD_LSB
  88.     header.header_size = header_size_aligned
  89.     header.payload_size = payload_size_aligned
  90.    
  91.     outf.write(header.pack())
  92.     outf.write(xstruct.create("little: padding[%d]" % (header_size_aligned - header_size)).pack())
  93.    
  94.     outf.write(inf.read())
  95.    
  96.     padding = payload_size_aligned - payload_size
  97.     if (padding > 0):
  98.         outf.write(xstruct.create("little: padding[%d]" % padding).pack())
  99.    
  100.     inf.close()
  101.     outf.close()
  102.  
  103. if __name__ == '__main__':
  104.     main()
  105.