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 | HORD encapsulator |
30 | HORD encapsulator |
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[4] /* 'HORD' */ " |
38 | char[4] tag /* 'HORD' */ |
40 | "uint8_t /* version */ " |
39 | uint8_t version /* version */ |
41 | "uint8_t /* encoding */ " |
40 | uint8_t encoding /* encoding */ |
42 | "uint32_t /* header size */ " |
41 | uint32_t header_size /* header size */ |
43 | "uint64_t /* payload size */ " |
42 | uint64_t payload_size /* payload size */ |
44 | ) |
43 | """ |
45 | 44 | ||
46 | HORD_VERSION = 1 |
- | |
47 | HORD_LSB = 1 |
45 | HORD_LSB = 1 |
48 | 46 | ||
49 | def align_up(size, alignment): |
47 | def align_up(size, alignment): |
50 | "Align upwards to a given alignment" |
48 | "Align upwards to a given alignment" |
51 | return (((size) + ((alignment) - 1)) & ~((alignment) - 1)) |
49 | return (((size) + ((alignment) - 1)) & ~((alignment) - 1)) |
52 | 50 | ||
53 | def usage(prname): |
51 | def usage(prname): |
54 | "Print usage syntax" |
52 | "Print usage syntax" |
55 | print prname + " <ALIGNMENT> <FS_IMAGE> <HORD_IMAGE>" |
53 | print prname + " <ALIGNMENT> <FS_IMAGE> <HORD_IMAGE>" |
56 | 54 | ||
57 | def main(): |
55 | def main(): |
58 | if (len(sys.argv) < 4): |
56 | if (len(sys.argv) < 4): |
59 | usage(sys.argv[0]) |
57 | usage(sys.argv[0]) |
60 | return |
58 | return |
61 | 59 | ||
62 | if (not sys.argv[1].isdigit()): |
60 | if (not sys.argv[1].isdigit()): |
63 | print "<ALIGNMENT> must be a number" |
61 | print "<ALIGNMENT> must be a number" |
64 | return |
62 | return |
65 | 63 | ||
66 | align = int(sys.argv[1], 0) |
64 | align = int(sys.argv[1], 0) |
67 | if (align <= 0): |
65 | if (align <= 0): |
68 | print "<ALIGNMENT> must be positive" |
66 | print "<ALIGNMENT> must be positive" |
69 | return |
67 | return |
70 | 68 | ||
71 | fs_image = os.path.abspath(sys.argv[2]) |
69 | fs_image = os.path.abspath(sys.argv[2]) |
72 | if (not os.path.isfile(fs_image)): |
70 | if (not os.path.isfile(fs_image)): |
73 | print "<FS_IMAGE> must be a file" |
71 | print "<FS_IMAGE> must be a file" |
74 | return |
72 | return |
75 | 73 | ||
76 | inf = file(fs_image, "rb") |
74 | inf = file(fs_image, "rb") |
77 | outf = file(sys.argv[3], "wb") |
75 | outf = file(sys.argv[3], "wb") |
78 | 76 | ||
79 | header_size = struct.calcsize(HEADER) |
77 | header = xstruct.create(HEADER) |
- | 78 | ||
- | 79 | header_size = header.size() |
|
80 | payload_size = os.path.getsize(fs_image) |
80 | payload_size = os.path.getsize(fs_image) |
81 | 81 | ||
82 | header_size_aligned = align_up(header_size, align) |
82 | header_size_aligned = align_up(header_size, align) |
83 | payload_size_aligned = align_up(payload_size, align) |
83 | payload_size_aligned = align_up(payload_size, align) |
84 | 84 | ||
- | 85 | header.tag = "HORD" |
|
- | 86 | header.version = 1 |
|
- | 87 | header.encoding = HORD_LSB |
|
- | 88 | header.header_size = header_size_aligned |
|
85 | outf.write(struct.pack(HEADER, "HORD", HORD_VERSION, HORD_LSB, header_size_aligned, payload_size_aligned)) |
89 | header.payload_size = payload_size_aligned |
- | 90 | ||
- | 91 | outf.write(header.pack()) |
|
86 | outf.write(xstruct.little_padding(header_size_aligned - header_size)) |
92 | outf.write(xstruct.create("little: padding[%d]" % (header_size_aligned - header_size)).pack()) |
87 | 93 | ||
88 | outf.write(inf.read()) |
94 | outf.write(inf.read()) |
89 | 95 | ||
90 | padding = payload_size_aligned - payload_size |
96 | padding = payload_size_aligned - payload_size |
91 | if (padding > 0): |
97 | if (padding > 0): |
92 | outf.write(xstruct.little_padding(padding)) |
98 | outf.write(xstruct.create("little: padding[%d]" % padding).pack()) |
93 | 99 | ||
94 | inf.close() |
100 | inf.close() |
95 | outf.close() |
101 | outf.close() |
96 | 102 | ||
97 | if __name__ == '__main__': |
103 | if __name__ == '__main__': |
98 | main() |
104 | main() |
99 | 105 |