WebSVN – HelenOS – Diff – /trunk/tools/mkfat.py


#!/usr/bin/env python
#
# Copyright (c) 2008 Martin Decky
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
#
# - Redistributions of source code must retain the above copyright
#   notice, this list of conditions and the following disclaimer.
# - Redistributions in binary form must reproduce the above copyright
#   notice, this list of conditions and the following disclaimer in the
#   documentation and/or other materials provided with the distribution.
# - The name of the author may not be used to endorse or promote products
#   derived from this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
# OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
# IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
# THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#
"""
FAT creator
"""
 
import sys
import os
import random
import xstruct
 
def align_up(size, alignment):
    "Return size aligned up to alignment"
   
    if (size % alignment == 0):
        return size
   
    return (((size / alignment) + 1) * alignment)
 
def subtree_size(root, cluster_size):
    "Recursive directory walk and calculate size"
   
    size = 0
    files = 0
   
    for name in os.listdir(root):
        canon = os.path.join(root, name)
       
        if (os.path.isfile(canon)):
            size += align_up(os.path.getsize(canon), cluster_size)
            files += 1
       
        if (os.path.isdir(canon)):
            size += subtree_size(canon, cluster_size)
            files += 1
   
    return size + align_up(files * 32, cluster_size)
 
def root_entries(root):
    "Return number of root directory entries"
   
    return len(os.listdir(root))
 
BOOT_SECTOR = """little:
    uint8_t jmp[3]             /* jump instruction */
    char oem[8]                /* OEM string */
    uint16_t sector            /* bytes per sector */
    uint8_t cluster            /* sectors per cluster */
    uint16_t reserved          /* reserved sectors */
    uint8_t fats               /* number of FATs */
    uint16_t rootdir           /* root directory entries */
    uint16_t sectors           /* total number of sectors */
    uint8_t descriptor         /* media descriptor */
    uint16_t fat_sectors       /* sectors per single FAT */
    uint16_t track_sectors     /* sectors per track */
    uint16_t heads             /* number of heads */
    uint32_t hidden            /* hidden sectors */
    uint32_t sectors_big       /* total number of sectors (if sectors == 0) */
   
    /* Extended BIOS Parameter Block */
    uint8_t drive              /* physical drive number */
    padding[1]                 /* reserved (current head) */
    uint8_t extboot_signature  /* extended boot signature */
    uint32_t serial            /* serial number */
    char label[11]             /* volume label */
    char fstype[8]             /* filesystem type */
    padding[448]               /* boot code */
    uint8_t boot_signature[2]  /* boot signature */
"""
 
EMPTY_SECTOR = """little:
    padding[512]
"""
 
def usage(prname):
    "Print usage syntax"
    print prname + " <PATH> <IMAGE>"
 
def main():
    if (len(sys.argv) < 3):
        usage(sys.argv[0])
        return
   
    path = os.path.abspath(sys.argv[1])
    if (not os.path.isdir(path)):
        print "<PATH> must be a directory"
        return
   
    sector_size = 512
    cluster_size = 4096
   
    root_size = align_up(root_entries(sys.argv[1]) * 32, sector_size)
    size = subtree_size(sys.argv[1], cluster_size)
    fat_size = align_up(size / cluster_size * 2, sector_size)
   
    sectors = (cluster_size + 2 * fat_size + root_size + size) / sector_size
   
    outf = file(sys.argv[2], "w")
   
    boot_sector = xstruct.create(BOOT_SECTOR)
    boot_sector.jmp = [0xEB, 0x3C, 0x90]
    boot_sector.oem = "MSDOS5.0"
    boot_sector.sector = sector_size
    boot_sector.cluster = cluster_size / sector_size
    boot_sector.reserved = cluster_size / sector_size
    boot_sector.fats = 2
    boot_sector.rootdir = root_size / 32
    boot_sector.sectors = (sectors if (sectors <= 65535) else 0)
    boot_sector.descriptor = 0xF8
    boot_sector.fat_sectors = fat_size / sector_size
    boot_sector.track_sectors = 63
    boot_sector.heads = 6
    boot_sector.hidden = 0
    boot_sector.sectors_big = (sectors if (sectors > 65535) else 0)
   
    boot_sector.drive = 0x80
    boot_sector.extboot_signature = 0x29
    boot_sector.serial = random.randint(0, 0xFFFFFFFF)
    boot_sector.label = "HELENOS"
    boot_sector.fstype = "FAT16   "
    boot_sector.boot_signature = [0x55, 0xAA]
   
    outf.write(boot_sector.pack())
   
    empty_sector = xstruct.create(EMPTY_SECTOR)
   
    # Reserved sectors (boot_sector.reserved - boot_sector)
    for i in range(1, boot_sector.reserved):
        outf.write(empty_sector.pack())
   
    # FAT tables
    for i in range(0, boot_sector.fats):
        for j in range(0, boot_sector.fat_sectors):
            outf.write(empty_sector.pack())
   
    # Root directory
    for i in range(0, root_size / sector_size):
        outf.write(empty_sector.pack())
   
    # Data
    for i in range(0, size / sector_size):
        outf.write(empty_sector.pack())
   
    outf.close()
   
if __name__ == '__main__':
    main()
 

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

Subversion Repositories HelenOS

(root)/trunk/tools/mkfat.py @ 3520 – Rev 3509 → 3511