WebSVN – HelenOS – Diff – /branches/tracing/tools/mktmpfs.py

 #!/usr/bin/env python
+#
 # Copyright (c) 2008 Martin Decky
 # All rights reserved.
+#
 # Redistribution and use in source and binary forms, with or without
 # modification, are permitted provided that the following conditions
 # are met:
+#
 # - Redistributions of source code must retain the above copyright
 #   notice, this list of conditions and the following disclaimer.
 # - Redistributions in binary form must reproduce the above copyright
 #   notice, this list of conditions and the following disclaimer in the
 #   documentation and/or other materials provided with the distribution.
 # - The name of the author may not be used to endorse or promote products
 #   derived from this software without specific prior written permission.
+#
 # THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 # IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 # OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 # IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 # INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 # NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 # THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+#
 """
 TMPFS creator
 """
 import sys
 import os
-import struct
+import xstruct
+HEADER = """little:
+    char tag[5]  /* 'TMPFS' */
+"""
-def align_up(size, alignment):
+DENTRY_NONE = """little:
+    uint8_t kind        /* NONE */
+    uint32_t fname_len  /* 0 */
+"""
+DENTRY_FILE = """little:
+    uint8_t kind        /* FILE */
+    uint32_t fname_len  /* filename length */
-    "Align upwards to a given alignment"
+    char fname[%d]      /* filename */
+    uint32_t flen       /* file length */
+"""
+DENTRY_DIRECTORY = """little:
+    uint8_t kind        /* DIRECTORY */
-    return (((size) + ((alignment) - 1)) & ~((alignment) - 1))
+    uint32_t fname_len  /* filename length */
+    char fname[%d]      /* filename */
+"""
+TMPFS_NONE = 0
+TMPFS_FILE = 1
+TMPFS_DIRECTORY = 2
 def usage(prname):
     "Print usage syntax"
-    print prname + " <ALIGNMENT> <PATH> <IMAGE>"
+    print prname + " <PATH> <IMAGE>"
 def recursion(root, outf):
     "Recursive directory walk"
-    payload_size = 0
     for name in os.listdir(root):
         canon = os.path.join(root, name)
         if (os.path.isfile(canon)):
-            outf.write(struct.pack("<BL" + ("%d" % len(name)) + "s", 1, len(name), name))
-            payload_size += 5 + len(name)
             size = os.path.getsize(canon)
-            rd = 0;
-            outf.write(struct.pack("<L", size))
+            dentry = xstruct.create(DENTRY_FILE % len(name))
+            dentry.kind = TMPFS_FILE
+            dentry.fname_len = len(name)
-            payload_size += 4
+            dentry.fname = name
+            dentry.flen = size
+            outf.write(dentry.pack())
             inf = file(canon, "r")
+            rd = 0;
             while (rd < size):
                 data = inf.read(4096);
                 outf.write(data)
-                payload_size += len(data)
                 rd += len(data)
             inf.close()
         if (os.path.isdir(canon)):
-            outf.write(struct.pack("<BL" + ("%d" % len(name)) + "s", 2, len(name), name))
+            dentry = xstruct.create(DENTRY_DIRECTORY % len(name))
+            dentry.kind = TMPFS_DIRECTORY
-            payload_size += 5 + len(name)
+            dentry.fname_len = len(name)
+            dentry.fname = name
+            outf.write(dentry.pack())
-            payload_size += recursion(canon, outf)
+            recursion(canon, outf)
-            outf.write(struct.pack("<BL", 0, 0))
+            dentry = xstruct.create(DENTRY_NONE)
+            dentry.kind = TMPFS_NONE
-            payload_size += 5
+            dentry.fname_len = 0
-    return payload_size
+            outf.write(dentry.pack())
 def main():
-    if (len(sys.argv) < 4):
+    if (len(sys.argv) < 3):
         usage(sys.argv[0])
         return
-    if (not sys.argv[1].isdigit()):
-        print "<ALIGNMENT> must be a number"
-        return
-    align = int(sys.argv[1], 0)
-    path = os.path.abspath(sys.argv[2])
+    path = os.path.abspath(sys.argv[1])
     if (not os.path.isdir(path)):
         print "<PATH> must be a directory"
         return
-    header_size = align_up(18, align)
-    outf = file(sys.argv[3], "w")
+    outf = file(sys.argv[2], "w")
-    outf.write(struct.pack("<" + ("%d" % header_size) + "x"))
-    outf.write(struct.pack("<5s", "TMPFS"))
+    header = xstruct.create(HEADER)
-    payload_size = 5
+    header.tag = "TMPFS"
-    payload_size += recursion(path, outf)
-    outf.write(struct.pack("<BL", 0, 0))
+    outf.write(header.pack())
-    payload_size += 5
+    recursion(path, outf)
-    aligned_size = align_up(payload_size, align)
+    dentry = xstruct.create(DENTRY_NONE)
+    dentry.kind = TMPFS_NONE
+    dentry.fname_len = 0
+    outf.write(dentry.pack())
-    if (aligned_size - payload_size > 0):
-        outf.write(struct.pack("<" + ("%d" % (aligned_size - payload_size)) + "x"))
-    outf.seek(0)
-    outf.write(struct.pack("<4sBBLQ", "HORD", 1, 1, header_size, aligned_size))
     outf.close()
 if __name__ == '__main__':
     main()

Subversion Repositories HelenOS

(root)/branches/tracing/tools/mktmpfs.py – Rev 3424 → 3425