Rev 3259 | Rev 3262 | Go to most recent revision | Only display areas with differences | Regard whitespace | Details | Blame | Last modification | View Log | RSS feed
Rev 3259 | Rev 3260 | ||
---|---|---|---|
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 | import xstruct |
36 | import xstruct |
37 | 37 | ||
38 | HEADER = xstruct.convert("little: char[5]") |
38 | HEADER = xstruct.convert("little: " |
- | 39 | "char[5] /* 'TMPFS' */ " |
|
- | 40 | ) |
|
- | 41 | ||
39 | DENTRY = xstruct.convert("little: uint8_t uint32_t") |
42 | DENTRY = xstruct.convert("little: " |
- | 43 | "uint8_t /* NONE, FILE or DIRECTORY */ " |
|
- | 44 | "uint32_t /* filename length */ " |
|
- | 45 | ) |
|
- | 46 | ||
40 | SIZE = xstruct.convert("little: uint32_t") |
47 | SIZE = xstruct.convert("little: " |
- | 48 | "uint32_t /* file size */ " |
|
- | 49 | ) |
|
41 | 50 | ||
42 | DENTRY_NONE = 0 |
51 | DENTRY_NONE = 0 |
43 | DENTRY_FILE = 1 |
52 | DENTRY_FILE = 1 |
44 | DENTRY_DIRECTORY = 2 |
53 | DENTRY_DIRECTORY = 2 |
45 | 54 | ||
46 | def usage(prname): |
55 | def usage(prname): |
47 | "Print usage syntax" |
56 | "Print usage syntax" |
48 | print prname + " <PATH> <IMAGE>" |
57 | print prname + " <PATH> <IMAGE>" |
49 | 58 | ||
50 | def recursion(root, outf): |
59 | def recursion(root, outf): |
51 | "Recursive directory walk" |
60 | "Recursive directory walk" |
52 | 61 | ||
53 | for name in os.listdir(root): |
62 | for name in os.listdir(root): |
54 | canon = os.path.join(root, name) |
63 | canon = os.path.join(root, name) |
55 | 64 | ||
56 | if (os.path.isfile(canon)): |
65 | if (os.path.isfile(canon)): |
57 | outf.write(struct.pack(DENTRY, DENTRY_FILE, len(name))) |
66 | outf.write(struct.pack(DENTRY, DENTRY_FILE, len(name))) |
58 | outf.write(xstruct.little_string(name)) |
67 | outf.write(xstruct.little_string(name)) |
59 | size = os.path.getsize(canon) |
68 | size = os.path.getsize(canon) |
60 | rd = 0; |
69 | rd = 0; |
61 | outf.write(struct.pack(SIZE, size)) |
70 | outf.write(struct.pack(SIZE, size)) |
62 | 71 | ||
63 | inf = file(canon, "r") |
72 | inf = file(canon, "r") |
64 | while (rd < size): |
73 | while (rd < size): |
65 | data = inf.read(4096); |
74 | data = inf.read(4096); |
66 | outf.write(data) |
75 | outf.write(data) |
67 | rd += len(data) |
76 | rd += len(data) |
68 | inf.close() |
77 | inf.close() |
69 | 78 | ||
70 | if (os.path.isdir(canon)): |
79 | if (os.path.isdir(canon)): |
71 | outf.write(struct.pack(DENTRY, DENTRY_DIRECTORY, len(name))) |
80 | outf.write(struct.pack(DENTRY, DENTRY_DIRECTORY, len(name))) |
72 | outf.write(xstruct.little_string(name)) |
81 | outf.write(xstruct.little_string(name)) |
73 | recursion(canon, outf) |
82 | recursion(canon, outf) |
74 | outf.write(struct.pack(DENTRY, DENTRY_NONE, 0)) |
83 | outf.write(struct.pack(DENTRY, DENTRY_NONE, 0)) |
75 | 84 | ||
76 | def main(): |
85 | def main(): |
77 | if (len(sys.argv) < 3): |
86 | if (len(sys.argv) < 3): |
78 | usage(sys.argv[0]) |
87 | usage(sys.argv[0]) |
79 | return |
88 | return |
80 | 89 | ||
81 | path = os.path.abspath(sys.argv[1]) |
90 | path = os.path.abspath(sys.argv[1]) |
82 | if (not os.path.isdir(path)): |
91 | if (not os.path.isdir(path)): |
83 | print "<PATH> must be a directory" |
92 | print "<PATH> must be a directory" |
84 | return |
93 | return |
85 | 94 | ||
86 | outf = file(sys.argv[2], "w") |
95 | outf = file(sys.argv[2], "w") |
87 | 96 | ||
88 | outf.write(struct.pack(HEADER, "TMPFS")) |
97 | outf.write(struct.pack(HEADER, "TMPFS")) |
89 | recursion(path, outf) |
98 | recursion(path, outf) |
90 | outf.write(struct.pack(DENTRY, DENTRY_NONE, 0)) |
99 | outf.write(struct.pack(DENTRY, DENTRY_NONE, 0)) |
91 | 100 | ||
92 | outf.close() |
101 | outf.close() |
93 | 102 | ||
94 | if __name__ == '__main__': |
103 | if __name__ == '__main__': |
95 | main() |
104 | main() |
96 | 105 |