WebSVN – HelenOS – Diff – /branches/dd/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
-def align_up(size, alignment):
+exclude_names = set(['.svn'])
+HEADER = """little:
+    char tag[5]  /* 'TMPFS' */
+"""
+DENTRY_NONE = """little:
+    uint8_t kind        /* NONE */
+    uint32_t fname_len  /* 0 */
+"""
+DENTRY_FILE = """little:
+    uint8_t kind        /* FILE */
-    return (((size) + ((alignment) - 1)) & ~((alignment) - 1))
+    uint32_t fname_len  /* filename length */
+    char fname[%d]      /* filename */
+    uint32_t flen       /* file length */
+"""
+DENTRY_DIRECTORY = """little:
+    uint8_t kind        /* DIRECTORY */
+    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 main():
-    if (len(sys.argv) < 4):
+def recursion(root, outf):
-        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])
-    if (not os.path.isdir(path)):
-        print "<PATH> must be a directory"
+    "Recursive directory walk"
-        return
-    header_size = align_up(18, align)
+    for name in os.listdir(root):
-    payload_size = 0
-    outf = file(sys.argv[3], "w")
+        canon = os.path.join(root, name)
-    outf.write(struct.pack("<" + ("%d" % header_size) + "x"))
-    for root, dirs, files in os.walk(path):
+        if (os.path.isfile(canon) and (not name in exclude_names)):
-        relpath = root[len(path):len(root)]
-        for name in files:
-            canon = os.path.join(relpath, name)
+            size = os.path.getsize(canon)
-            outf.write(struct.pack("<BL" + ("%d" % len(canon)) + "s", 1, len(canon), canon))
-            payload_size += 5 + len(canon)
-            fn = os.path.join(root, name)
+            dentry = xstruct.create(DENTRY_FILE % len(name))
-            size = os.path.getsize(fn)
+            dentry.kind = TMPFS_FILE
-            rd = 0;
+            dentry.fname_len = len(name)
-            outf.write(struct.pack("<L", size))
+            dentry.fname = name
-            payload_size += 4
+            dentry.flen = size
+            outf.write(dentry.pack())
-            inf = file(fn, "r")
+            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) and (not name in exclude_names)):
+            dentry = xstruct.create(DENTRY_DIRECTORY % len(name))
+            dentry.kind = TMPFS_DIRECTORY
+            dentry.fname_len = len(name)
-        for name in dirs:
+            dentry.fname = name
+            outf.write(dentry.pack())
+            recursion(canon, outf)
+            dentry = xstruct.create(DENTRY_NONE)
+            dentry.kind = TMPFS_NONE
+            dentry.fname_len = 0
+            outf.write(dentry.pack())
+def main():
+    if (len(sys.argv) < 3):
+        usage(sys.argv[0])
+        return
+    path = os.path.abspath(sys.argv[1])
-            canon = os.path.join(relpath, name)
+    if (not os.path.isdir(path)):
-            outf.write(struct.pack("<BL" + ("%d" % len(canon)) + "s", 2, len(canon), canon))
+        print "<PATH> must be a directory"
+        return
+    outf = file(sys.argv[2], "w")
-            payload_size += 5 + len(canon)
+    header = xstruct.create(HEADER)
+    header.tag = "TMPFS"
-    aligned_size = align_up(payload_size, align)
+    outf.write(header.pack())
+    recursion(path, outf)
-    if (aligned_size - payload_size > 0):
+    dentry = xstruct.create(DENTRY_NONE)
+    dentry.kind = TMPFS_NONE
+    dentry.fname_len = 0
-        outf.write(struct.pack("<" + ("%d" % (aligned_size - payload_size)) + "x"))
+    outf.write(dentry.pack())
-    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/dd/tools/mktmpfs.py – Rev 3022 → 4055