Subversion Repositories HelenOS

Rev

Rev 3343 | 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. TMPFS creator
  31. """
  32.  
  33. import sys
  34. import os
  35. import xstruct
  36.  
  37. exclude_names = set(['.svn'])
  38.  
  39. HEADER = """little:
  40.     char tag[5]  /* 'TMPFS' */
  41. """
  42.  
  43. DENTRY_NONE = """little:
  44.     uint8_t kind        /* NONE */
  45.     uint32_t fname_len  /* 0 */
  46. """
  47.  
  48. DENTRY_FILE = """little:
  49.     uint8_t kind        /* FILE */
  50.     uint32_t fname_len  /* filename length */
  51.     char fname[%d]      /* filename */
  52.     uint32_t flen       /* file length */
  53. """
  54.  
  55. DENTRY_DIRECTORY = """little:
  56.     uint8_t kind        /* DIRECTORY */
  57.     uint32_t fname_len  /* filename length */
  58.     char fname[%d]      /* filename */
  59. """
  60.  
  61. TMPFS_NONE = 0
  62. TMPFS_FILE = 1
  63. TMPFS_DIRECTORY = 2
  64.  
  65. def usage(prname):
  66.     "Print usage syntax"
  67.     print prname + " <PATH> <IMAGE>"
  68.  
  69. def recursion(root, outf):
  70.     "Recursive directory walk"
  71.    
  72.     for name in os.listdir(root):
  73.         canon = os.path.join(root, name)
  74.        
  75.         if (os.path.isfile(canon) and (not name in exclude_names)):
  76.             size = os.path.getsize(canon)
  77.            
  78.             dentry = xstruct.create(DENTRY_FILE % len(name))
  79.             dentry.kind = TMPFS_FILE
  80.             dentry.fname_len = len(name)
  81.             dentry.fname = name
  82.             dentry.flen = size
  83.            
  84.             outf.write(dentry.pack())
  85.            
  86.             inf = file(canon, "r")
  87.             rd = 0;
  88.             while (rd < size):
  89.                 data = inf.read(4096);
  90.                 outf.write(data)
  91.                 rd += len(data)
  92.             inf.close()
  93.        
  94.         if (os.path.isdir(canon) and (not name in exclude_names)):
  95.             dentry = xstruct.create(DENTRY_DIRECTORY % len(name))
  96.             dentry.kind = TMPFS_DIRECTORY
  97.             dentry.fname_len = len(name)
  98.             dentry.fname = name
  99.            
  100.             outf.write(dentry.pack())
  101.            
  102.             recursion(canon, outf)
  103.            
  104.             dentry = xstruct.create(DENTRY_NONE)
  105.             dentry.kind = TMPFS_NONE
  106.             dentry.fname_len = 0
  107.            
  108.             outf.write(dentry.pack())
  109.  
  110. def main():
  111.     if (len(sys.argv) < 3):
  112.         usage(sys.argv[0])
  113.         return
  114.    
  115.     path = os.path.abspath(sys.argv[1])
  116.     if (not os.path.isdir(path)):
  117.         print "<PATH> must be a directory"
  118.         return
  119.    
  120.     outf = file(sys.argv[2], "w")
  121.    
  122.     header = xstruct.create(HEADER)
  123.     header.tag = "TMPFS"
  124.    
  125.     outf.write(header.pack())
  126.    
  127.     recursion(path, outf)
  128.    
  129.     dentry = xstruct.create(DENTRY_NONE)
  130.     dentry.kind = TMPFS_NONE
  131.     dentry.fname_len = 0
  132.    
  133.     outf.write(dentry.pack())
  134.    
  135.     outf.close()
  136.    
  137. if __name__ == '__main__':
  138.     main()
  139.