Subversion Repositories HelenOS

Rev

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

Rev 3042 Rev 3424
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
35
import struct
36
 
36
 
37
def align_up(size, alignment):
37
def align_up(size, alignment):
38
    "Align upwards to a given alignment"
38
    "Align upwards to a given alignment"
39
    return (((size) + ((alignment) - 1)) & ~((alignment) - 1))
39
    return (((size) + ((alignment) - 1)) & ~((alignment) - 1))
40
 
40
 
41
def usage(prname):
41
def usage(prname):
42
    "Print usage syntax"
42
    "Print usage syntax"
43
    print prname + " <ALIGNMENT> <PATH> <IMAGE>"
43
    print prname + " <ALIGNMENT> <PATH> <IMAGE>"
44
 
44
 
-
 
45
def recursion(root, outf):
-
 
46
    "Recursive directory walk"
-
 
47
   
-
 
48
    payload_size = 0
-
 
49
   
-
 
50
    for name in os.listdir(root):
-
 
51
        canon = os.path.join(root, name)
-
 
52
       
-
 
53
        if (os.path.isfile(canon)):
-
 
54
            outf.write(struct.pack("<BL" + ("%d" % len(name)) + "s", 1, len(name), name))
-
 
55
            payload_size += 5 + len(name)
-
 
56
            size = os.path.getsize(canon)
-
 
57
            rd = 0;
-
 
58
            outf.write(struct.pack("<L", size))
-
 
59
            payload_size += 4
-
 
60
           
-
 
61
            inf = file(canon, "r")
-
 
62
            while (rd < size):
-
 
63
                data = inf.read(4096);
-
 
64
                outf.write(data)
-
 
65
                payload_size += len(data)
-
 
66
                rd += len(data)
-
 
67
            inf.close()
-
 
68
       
-
 
69
        if (os.path.isdir(canon)):
-
 
70
            outf.write(struct.pack("<BL" + ("%d" % len(name)) + "s", 2, len(name), name))
-
 
71
            payload_size += 5 + len(name)
-
 
72
            payload_size += recursion(canon, outf)
-
 
73
            outf.write(struct.pack("<BL", 0, 0))
-
 
74
            payload_size += 5
-
 
75
   
-
 
76
    return payload_size
-
 
77
 
45
def main():
78
def main():
46
    if (len(sys.argv) < 4):
79
    if (len(sys.argv) < 4):
47
        usage(sys.argv[0])
80
        usage(sys.argv[0])
48
        return
81
        return
49
   
82
   
50
    if (not sys.argv[1].isdigit()):
83
    if (not sys.argv[1].isdigit()):
51
        print "<ALIGNMENT> must be a number"
84
        print "<ALIGNMENT> must be a number"
52
        return
85
        return
53
   
86
   
54
    align = int(sys.argv[1], 0)
87
    align = int(sys.argv[1], 0)
55
    path = os.path.abspath(sys.argv[2])
88
    path = os.path.abspath(sys.argv[2])
56
    if (not os.path.isdir(path)):
89
    if (not os.path.isdir(path)):
57
        print "<PATH> must be a directory"
90
        print "<PATH> must be a directory"
58
        return
91
        return
59
   
92
   
60
    header_size = align_up(18, align)
93
    header_size = align_up(18, align)
61
    payload_size = 0
-
 
62
    outf = file(sys.argv[3], "w")
94
    outf = file(sys.argv[3], "w")
63
    outf.write(struct.pack("<" + ("%d" % header_size) + "x"))
95
    outf.write(struct.pack("<" + ("%d" % header_size) + "x"))
64
   
96
   
65
    for root, dirs, files in os.walk(path):
-
 
66
        relpath = root[len(path):len(root)]
-
 
67
        for name in files:
-
 
68
            canon = os.path.join(relpath, name)
-
 
69
            outf.write(struct.pack("<BL" + ("%d" % len(canon)) + "s", 1, len(canon), canon))
-
 
70
            payload_size += 5 + len(canon)
-
 
71
           
-
 
72
            fn = os.path.join(root, name)
-
 
73
            size = os.path.getsize(fn)
-
 
74
            rd = 0;
-
 
75
            outf.write(struct.pack("<L", size))
97
    outf.write(struct.pack("<5s", "TMPFS"))
76
            payload_size += 4
98
    payload_size = 5
77
           
99
   
78
            inf = file(fn, "r")
-
 
79
            while (rd < size):
-
 
80
                data = inf.read(4096);
-
 
81
                outf.write(data)
-
 
82
                payload_size += len(data)
100
    payload_size += recursion(path, outf)
83
                rd += len(data)
-
 
84
            inf.close()
-
 
85
       
101
       
86
        for name in dirs:
-
 
87
            canon = os.path.join(relpath, name)
-
 
88
            outf.write(struct.pack("<BL" + ("%d" % len(canon)) + "s", 2, len(canon), canon))
102
    outf.write(struct.pack("<BL", 0, 0))
89
            payload_size += 5 + len(canon)
103
    payload_size += 5
90
   
104
   
91
    aligned_size = align_up(payload_size, align)
105
    aligned_size = align_up(payload_size, align)
92
   
106
   
93
    if (aligned_size - payload_size > 0):
107
    if (aligned_size - payload_size > 0):
94
        outf.write(struct.pack("<" + ("%d" % (aligned_size - payload_size)) + "x"))
108
        outf.write(struct.pack("<" + ("%d" % (aligned_size - payload_size)) + "x"))
95
   
109
       
96
    outf.seek(0)
110
    outf.seek(0)
97
    outf.write(struct.pack("<4sBBLQ", "HORD", 1, 1, header_size, aligned_size))
111
    outf.write(struct.pack("<4sBBLQ", "HORD", 1, 1, header_size, aligned_size))
98
    outf.close()
112
    outf.close()
99
 
113
 
100
if __name__ == '__main__':
114
if __name__ == '__main__':
101
    main()
115
    main()
102
 
116