Rev 3264 | Rev 3511 | Go to most recent revision | Only display areas with differences | Ignore whitespace | Details | Blame | Last modification | View Log | RSS feed
Rev 3264 | Rev 3509 | ||
---|---|---|---|
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 | FAT creator |
30 | FAT creator |
31 | """ |
31 | """ |
32 | 32 | ||
33 | import sys |
33 | import sys |
34 | import os |
34 | import os |
35 | import random |
35 | import random |
36 | import xstruct |
36 | import xstruct |
37 | 37 | ||
- | 38 | def align_up(size, alignment): |
|
- | 39 | "Return size aligned up to alignment" |
|
- | 40 | ||
- | 41 | if (size % alignment == 0): |
|
- | 42 | return size |
|
- | 43 | ||
- | 44 | return (((size / alignment) + 1) * alignment) |
|
- | 45 | ||
- | 46 | def subtree_size(root, cluster_size): |
|
- | 47 | "Recursive directory walk and calculate size" |
|
- | 48 | ||
- | 49 | size = 0 |
|
- | 50 | files = 0 |
|
- | 51 | ||
- | 52 | for name in os.listdir(root): |
|
- | 53 | canon = os.path.join(root, name) |
|
- | 54 | ||
- | 55 | if (os.path.isfile(canon)): |
|
- | 56 | size += align_up(os.path.getsize(canon), cluster_size) |
|
- | 57 | files += 1 |
|
- | 58 | ||
- | 59 | if (os.path.isdir(canon)): |
|
- | 60 | size += subtree_size(canon, cluster_size) |
|
- | 61 | files += 1 |
|
- | 62 | ||
- | 63 | return size + align_up(files * 32, cluster_size) |
|
- | 64 | ||
- | 65 | def root_entries(root): |
|
- | 66 | "Return number of root directory entries" |
|
- | 67 | ||
- | 68 | return len(os.listdir(root)) |
|
- | 69 | ||
38 | BOOT_SECTOR = """little: |
70 | BOOT_SECTOR = """little: |
39 | uint8_t jmp[3] /* jump instruction */ |
71 | uint8_t jmp[3] /* jump instruction */ |
40 | char oem[8] /* OEM string */ |
72 | char oem[8] /* OEM string */ |
41 | uint16_t sector /* bytes per sector */ |
73 | uint16_t sector /* bytes per sector */ |
42 | uint8_t cluster /* sectors per cluster */ |
74 | uint8_t cluster /* sectors per cluster */ |
43 | uint16_t reserved /* reserved sectors */ |
75 | uint16_t reserved /* reserved sectors */ |
44 | uint8_t fats /* number of FATs */ |
76 | uint8_t fats /* number of FATs */ |
45 | uint16_t rootdir /* root directory entries */ |
77 | uint16_t rootdir /* root directory entries */ |
46 | uint16_t sectors /* total number of sectors */ |
78 | uint16_t sectors /* total number of sectors */ |
47 | uint8_t descriptor /* media descriptor */ |
79 | uint8_t descriptor /* media descriptor */ |
48 | uint16_t fat_sectors /* sectors per single FAT */ |
80 | uint16_t fat_sectors /* sectors per single FAT */ |
49 | uint16_t track_sectors /* sectors per track */ |
81 | uint16_t track_sectors /* sectors per track */ |
50 | uint16_t heads /* number of heads */ |
82 | uint16_t heads /* number of heads */ |
51 | uint32_t hidden /* hidden sectors */ |
83 | uint32_t hidden /* hidden sectors */ |
52 | uint32_t sectors_big /* total number of sectors (if sectors == 0) */ |
84 | uint32_t sectors_big /* total number of sectors (if sectors == 0) */ |
53 | |
85 | |
54 | /* Extended BIOS Parameter Block */ |
86 | /* Extended BIOS Parameter Block */ |
55 | uint8_t drive /* physical drive number */ |
87 | uint8_t drive /* physical drive number */ |
56 | padding[1] /* reserved (current head) */ |
88 | padding[1] /* reserved (current head) */ |
57 | uint8_t extboot_signature /* extended boot signature */ |
89 | uint8_t extboot_signature /* extended boot signature */ |
58 | uint32_t serial /* serial number */ |
90 | uint32_t serial /* serial number */ |
59 | char label[11] /* volume label */ |
91 | char label[11] /* volume label */ |
60 | char fstype[8] /* filesystem type */ |
92 | char fstype[8] /* filesystem type */ |
61 | padding[448] /* boot code */ |
93 | padding[448] /* boot code */ |
62 | uint8_t boot_signature[2] /* boot signature */ |
94 | uint8_t boot_signature[2] /* boot signature */ |
63 | """ |
95 | """ |
64 | 96 | ||
65 | def usage(prname): |
97 | def usage(prname): |
66 | "Print usage syntax" |
98 | "Print usage syntax" |
67 | print prname + " <PATH> <IMAGE>" |
99 | print prname + " <PATH> <IMAGE>" |
68 | 100 | ||
69 | def main(): |
101 | def main(): |
70 | if (len(sys.argv) < 3): |
102 | if (len(sys.argv) < 3): |
71 | usage(sys.argv[0]) |
103 | usage(sys.argv[0]) |
72 | return |
104 | return |
73 | 105 | ||
74 | path = os.path.abspath(sys.argv[1]) |
106 | path = os.path.abspath(sys.argv[1]) |
75 | if (not os.path.isdir(path)): |
107 | if (not os.path.isdir(path)): |
76 | print "<PATH> must be a directory" |
108 | print "<PATH> must be a directory" |
77 | return |
109 | return |
78 | 110 | ||
- | 111 | sector_size = 512 |
|
- | 112 | cluster_size = 4096 |
|
- | 113 | ||
- | 114 | root_size = align_up(root_entries(sys.argv[1]) * 32, sector_size) |
|
- | 115 | size = subtree_size(sys.argv[1], cluster_size) |
|
- | 116 | fat_size = align_up(size / cluster_size * 2, sector_size) |
|
- | 117 | ||
- | 118 | sectors = (cluster_size + 2 * fat_size + root_size + size) / sector_size |
|
- | 119 | ||
79 | outf = file(sys.argv[2], "w") |
120 | outf = file(sys.argv[2], "w") |
80 | 121 | ||
81 | boot_sector = xstruct.create(BOOT_SECTOR) |
122 | boot_sector = xstruct.create(BOOT_SECTOR) |
82 | boot_sector.jmp = [0xEB, 0x3C, 0x90] |
123 | boot_sector.jmp = [0xEB, 0x3C, 0x90] |
83 | boot_sector.oem = "MSDOS5.0" |
124 | boot_sector.oem = "MSDOS5.0" |
84 | boot_sector.sector = 512 |
125 | boot_sector.sector = sector_size |
85 | boot_sector.cluster = 8 # 4096 bytes per cluster |
126 | boot_sector.cluster = cluster_size / sector_size |
86 | boot_sector.reserved = 1 |
127 | boot_sector.reserved = cluster_size / sector_size |
87 | boot_sector.fats = 2 |
128 | boot_sector.fats = 2 |
88 | boot_sector.rootdir = 224 # FIXME: root directory should be sector aligned |
129 | boot_sector.rootdir = root_size / 32 |
89 | boot_sector.sectors = 0 # FIXME |
130 | boot_sector.sectors = (sectors if (sectors <= 65535) else 0) |
90 | boot_sector.descriptor = 0xF8 |
131 | boot_sector.descriptor = 0xF8 |
91 | boot_sector.fat_sectors = 0 # FIXME |
132 | boot_sector.fat_sectors = fat_size / sector_size |
92 | boot_sector.track_sectors = 0 # FIXME |
133 | boot_sector.track_sectors = 63 |
93 | boot_sector.heads = 0 # FIXME |
134 | boot_sector.heads = 6 |
94 | boot_sector.hidden = 0 |
135 | boot_sector.hidden = 0 |
95 | boot_sector.sectors_big = 0 # FIXME |
136 | boot_sector.sectors_big = (sectors if (sectors > 65535) else 0) |
96 | 137 | ||
97 | boot_sector.drive = 0 |
138 | boot_sector.drive = 0x80 |
98 | boot_sector.extboot_signature = 0x29 |
139 | boot_sector.extboot_signature = 0x29 |
99 | boot_sector.serial = random.randint(0, 0xFFFFFFFF) |
140 | boot_sector.serial = random.randint(0, 0xFFFFFFFF) |
100 | boot_sector.label = "HELENOS" |
141 | boot_sector.label = "HELENOS" |
101 | boot_sector.fstype = "FAT16 " |
142 | boot_sector.fstype = "FAT16 " |
102 | boot_sector.boot_signature = [0x55, 0xAA] |
143 | boot_sector.boot_signature = [0x55, 0xAA] |
103 | 144 | ||
104 | outf.write(boot_sector.pack()) |
145 | outf.write(boot_sector.pack()) |
105 | 146 | ||
106 | outf.close() |
147 | outf.close() |
107 | 148 | ||
108 | if __name__ == '__main__': |
149 | if __name__ == '__main__': |
109 | main() |
150 | main() |
110 | 151 |