Subversion Repositories HelenOS

Rev

Rev 3260 | Rev 3263 | Go to most recent revision | Only display areas with differences | Ignore whitespace | Details | Blame | Last modification | View Log | RSS feed

Rev 3260 Rev 3262
1 
#!/usr/bin/env python
 1 
#!/usr/bin/env python
2 
#
 2 
#
3 
# Copyright (c) 2008 Martin Decky
 3 
# Copyright (c) 2008 Martin Decky
4 
# All rights reserved.
 4 
# All rights reserved.
5 
#
 5 
#
6 
# Redistribution and use in source and binary forms, with or without
 6 
# Redistribution and use in source and binary forms, with or without
7 
# modification, are permitted provided that the following conditions
 7 
# modification, are permitted provided that the following conditions
8 
# are met:
 8 
# are met:
9 
#
 9 
#
10 
# - Redistributions of source code must retain the above copyright
 10 
# - Redistributions of source code must retain the above copyright
11 
#   notice, this list of conditions and the following disclaimer.
 11 
#   notice, this list of conditions and the following disclaimer.
12 
# - Redistributions in binary form must reproduce the above copyright
 12 
# - Redistributions in binary form must reproduce the above copyright
13 
#   notice, this list of conditions and the following disclaimer in the
 13 
#   notice, this list of conditions and the following disclaimer in the
14 
#   documentation and/or other materials provided with the distribution.
 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
 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.
 16 
#   derived from this software without specific prior written permission.
17 
#
 17 
#
18 
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 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
 19 
# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
20 
# OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 20 
# OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
21 
# IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 21 
# IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
22 
# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 22 
# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
23 
# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 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
 24 
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25 
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 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
 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.
 27 
# THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28 
#
 28 
#
29 
"""
 29 
"""
30 
TMPFS creator
 30 
TMPFS creator
31 
"""
 31 
"""
32 
 
 32 
 
33 
import sys
 33 
import sys
34 
import os
 34 
import os
35 
import struct
 - 
 
36 
import xstruct
 35 
import xstruct
37 
 
 36 
 
38 
HEADER = xstruct.convert("little: "
 37 
HEADER = """little:
39 
    "char[5]  /* 'TMPFS' */ "
 38 
    char[5] tag  /* 'TMPFS' */
- 
 
 39 
"""
40 
)
 40 
 
- 
 
 41 
DENTRY_NONE = """little:
- 
 
 42 
    uint8_t kind        /* NONE */
- 
 
 43 
    uint32_t fname_len  /* 0 */
- 
 
 44 
"""
41 
 
 45 
 
42 
DENTRY = xstruct.convert("little: "
 46 
DENTRY_FILE = """little:
43 
    "uint8_t   /* NONE, FILE or DIRECTORY */ "
 47 
    uint8_t kind        /* FILE */
44 
    "uint32_t  /* filename length */ "
 48 
    uint32_t fname_len  /* filename length */
- 
 
 49 
    char[%d] fname      /* filename */
- 
 
 50 
    uint32_t flen       /* file length */
45 
)
 51 
"""
46 
 
 52 
 
47 
SIZE = xstruct.convert("little: "
 53 
DENTRY_DIRECTORY = """little:
- 
 
 54 
    uint8_t kind        /* DIRECTORY */
48 
    "uint32_t  /* file size */ "
 55 
    uint32_t fname_len  /* filename length */
- 
 
 56 
    char[%d] fname      /* filename */
49 
)
 57 
"""
50 
 
 58 
 
51 
DENTRY_NONE = 0
 59 
TMPFS_NONE = 0
52 
DENTRY_FILE = 1
 60 
TMPFS_FILE = 1
53 
DENTRY_DIRECTORY = 2
 61 
TMPFS_DIRECTORY = 2
54 
 
 62 
 
55 
def usage(prname):
 63 
def usage(prname):
56 
    "Print usage syntax"
 64 
    "Print usage syntax"
57 
    print prname + " <PATH> <IMAGE>"
 65 
    print prname + " <PATH> <IMAGE>"
58 
 
 66 
 
59 
def recursion(root, outf):
 67 
def recursion(root, outf):
60 
    "Recursive directory walk"
 68 
    "Recursive directory walk"
61 
   
 69 
   
62 
    for name in os.listdir(root):
 70 
    for name in os.listdir(root):
63 
        canon = os.path.join(root, name)
 71 
        canon = os.path.join(root, name)
64 
       
 72 
       
65 
        if (os.path.isfile(canon)):
 73 
        if (os.path.isfile(canon)):
66 
            outf.write(struct.pack(DENTRY, DENTRY_FILE, len(name)))
 - 
 
67 
            outf.write(xstruct.little_string(name))
 - 
 
68 
            size = os.path.getsize(canon)
 74 
            size = os.path.getsize(canon)
- 
 
 75 
           
- 
 
 76 
            dentry = xstruct.create(DENTRY_FILE % len(name))
- 
 
 77 
            dentry.kind = TMPFS_FILE
- 
 
 78 
            dentry.fname_len = len(name)
- 
 
 79 
            dentry.fname = name
69 
            rd = 0;
 80 
            dentry.flen = size
- 
 
 81 
           
70 
            outf.write(struct.pack(SIZE, size))
 82 
            outf.write(dentry.pack())
71 
           
 83 
           
72 
            inf = file(canon, "r")
 84 
            inf = file(canon, "r")
- 
 
 85 
            rd = 0;
73 
            while (rd < size):
 86 
            while (rd < size):
74 
                data = inf.read(4096);
 87 
                data = inf.read(4096);
75 
                outf.write(data)
 88 
                outf.write(data)
76 
                rd += len(data)
 89 
                rd += len(data)
77 
            inf.close()
 90 
            inf.close()
78 
       
 91 
       
79 
        if (os.path.isdir(canon)):
 92 
        if (os.path.isdir(canon)):
80 
            outf.write(struct.pack(DENTRY, DENTRY_DIRECTORY, len(name)))
 93 
            dentry = xstruct.create(DENTRY_DIRECTORY % len(name))
- 
 
 94 
            dentry.kind = TMPFS_DIRECTORY
- 
 
 95 
            dentry.fname_len = len(name)
- 
 
 96 
            dentry.fname = name
- 
 
 97 
           
81 
            outf.write(xstruct.little_string(name))
 98 
            outf.write(dentry.pack())
- 
 
 99 
           
82 
            recursion(canon, outf)
 100 
            recursion(canon, outf)
- 
 
 101 
           
83 
            outf.write(struct.pack(DENTRY, DENTRY_NONE, 0))
 102 
            dentry = xstruct.create(DENTRY_NONE)
- 
 
 103 
            dentry.kind = TMPFS_NONE
- 
 
 104 
            dentry.fname_len = 0
- 
 
 105 
           
- 
 
 106 
            outf.write(dentry.pack())
84 
 
 107 
 
85 
def main():
 108 
def main():
86 
    if (len(sys.argv) < 3):
 109 
    if (len(sys.argv) < 3):
87 
        usage(sys.argv[0])
 110 
        usage(sys.argv[0])
88 
        return
 111 
        return
89 
   
 112 
   
90 
    path = os.path.abspath(sys.argv[1])
 113 
    path = os.path.abspath(sys.argv[1])
91 
    if (not os.path.isdir(path)):
 114 
    if (not os.path.isdir(path)):
92 
        print "<PATH> must be a directory"
 115 
        print "<PATH> must be a directory"
93 
        return
 116 
        return
94 
   
 117 
   
95 
    outf = file(sys.argv[2], "w")
 118 
    outf = file(sys.argv[2], "w")
96 
   
 119 
   
- 
 
 120 
    header = xstruct.create(HEADER)
- 
 
 121 
    header.tag = "TMPFS"
- 
 
 122 
   
97 
    outf.write(struct.pack(HEADER, "TMPFS"))
 123 
    outf.write(header.pack())
- 
 
 124 
   
98 
    recursion(path, outf)
 125 
    recursion(path, outf)
- 
 
 126 
   
99 
    outf.write(struct.pack(DENTRY, DENTRY_NONE, 0))
 127 
    dentry = xstruct.create(DENTRY_NONE)
- 
 
 128 
    dentry.kind = TMPFS_NONE
- 
 
 129 
    dentry.fname_len = 0
100 
 
 130 
   
- 
 
 131 
    outf.write(dentry.pack())
- 
 
 132 
   
101 
    outf.close()
 133 
    outf.close()
102 
       
 134 
   
103 
if __name__ == '__main__':
 135 
if __name__ == '__main__':
104 
    main()
 136 
    main()
105 
 
 137