WebSVN – HelenOS – Diff – /trunk/tools/mktmpfs.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.
#
"""
TMPFS creator
"""
 
import sys
import os
import xstruct
 
HEADER = """little:
    char tag[5]  /* 'TMPFS' */
"""
 
DENTRY_NONE = """little:
    uint8_t kind        /* NONE */
    uint32_t fname_len  /* 0 */
"""
 
DENTRY_FILE = """little:
    uint8_t kind        /* FILE */
    uint32_t fname_len  /* filename length */
    char fname[%d]      /* filename */
    uint32_t flen       /* file length */
"""
 
DENTRY_DIRECTORY = """little:
    uint8_t kind        /* DIRECTORY */
    uint32_t fname_len  /* filename length */
    char fname[%d]      /* filename */
"""
 
TMPFS_NONE = 0
TMPFS_FILE = 1
TMPFS_DIRECTORY = 2
 
def usage(prname):
    "Print usage syntax"
    print prname + " <PATH> <IMAGE>"
 
def recursion(root, outf):
    "Recursive directory walk"
   
    for name in os.listdir(root):
        canon = os.path.join(root, name)
       
        if (os.path.isfile(canon)):
            size = os.path.getsize(canon)
           
            dentry = xstruct.create(DENTRY_FILE % len(name))
            dentry.kind = TMPFS_FILE
            dentry.fname_len = len(name)
            dentry.fname = name
            dentry.flen = size
           
            outf.write(dentry.pack())
           
            inf = file(canon, "r")
            rd = 0;
            while (rd < size):
                data = inf.read(4096);
                outf.write(data)
                rd += len(data)
            inf.close()
       
        if (os.path.isdir(canon)):
            dentry = xstruct.create(DENTRY_DIRECTORY % len(name))
            dentry.kind = TMPFS_DIRECTORY
            dentry.fname_len = len(name)
            dentry.fname = name
           
            outf.write(dentry.pack())
           
            recursion(canon, outf)
           
            dentry = xstruct.create(DENTRY_NONE)
            dentry.kind = TMPFS_NONE
            dentry.fname_len = 0
           
            outf.write(dentry.pack())
 
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
   
    outf = file(sys.argv[2], "w")
   
    header = xstruct.create(HEADER)
    header.tag = "TMPFS"
   
    outf.write(header.pack())
   
    recursion(path, outf)
   
    dentry = xstruct.create(DENTRY_NONE)
    dentry.kind = TMPFS_NONE
    dentry.fname_len = 0
   
    outf.write(dentry.pack())
   
    outf.close()
   
if __name__ == '__main__':
    main()
 

Rev 3262	Rev 3263
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 xstruct	35	import xstruct
36		36
37	HEADER = """little:	37	HEADER = """little:
38	char[5] tag /* 'TMPFS' */	38	char tag[5] /* 'TMPFS' */
39	"""	39	"""
40		40
41	DENTRY_NONE = """little:	41	DENTRY_NONE = """little:
42	uint8_t kind /* NONE */	42	uint8_t kind /* NONE */
43	uint32_t fname_len /* 0 */	43	uint32_t fname_len /* 0 */
44	"""	44	"""
45		45
46	DENTRY_FILE = """little:	46	DENTRY_FILE = """little:
47	uint8_t kind /* FILE */	47	uint8_t kind /* FILE */
48	uint32_t fname_len /* filename length */	48	uint32_t fname_len /* filename length */
49	char[%d] fname /* filename */	49	char fname[%d] /* filename */
50	uint32_t flen /* file length */	50	uint32_t flen /* file length */
51	"""	51	"""
52		52
53	DENTRY_DIRECTORY = """little:	53	DENTRY_DIRECTORY = """little:
54	uint8_t kind /* DIRECTORY */	54	uint8_t kind /* DIRECTORY */
55	uint32_t fname_len /* filename length */	55	uint32_t fname_len /* filename length */
56	char[%d] fname /* filename */	56	char fname[%d] /* filename */
57	"""	57	"""
58		58
59	TMPFS_NONE = 0	59	TMPFS_NONE = 0
60	TMPFS_FILE = 1	60	TMPFS_FILE = 1
61	TMPFS_DIRECTORY = 2	61	TMPFS_DIRECTORY = 2
62		62
63	def usage(prname):	63	def usage(prname):
64	"Print usage syntax"	64	"Print usage syntax"
65	print prname + " <PATH> <IMAGE>"	65	print prname + " <PATH> <IMAGE>"
66		66
67	def recursion(root, outf):	67	def recursion(root, outf):
68	"Recursive directory walk"	68	"Recursive directory walk"
69		69
70	for name in os.listdir(root):	70	for name in os.listdir(root):
71	canon = os.path.join(root, name)	71	canon = os.path.join(root, name)
72		72
73	if (os.path.isfile(canon)):	73	if (os.path.isfile(canon)):
74	size = os.path.getsize(canon)	74	size = os.path.getsize(canon)
75		75
76	dentry = xstruct.create(DENTRY_FILE % len(name))	76	dentry = xstruct.create(DENTRY_FILE % len(name))
77	dentry.kind = TMPFS_FILE	77	dentry.kind = TMPFS_FILE
78	dentry.fname_len = len(name)	78	dentry.fname_len = len(name)
79	dentry.fname = name	79	dentry.fname = name
80	dentry.flen = size	80	dentry.flen = size
81		81
82	outf.write(dentry.pack())	82	outf.write(dentry.pack())
83		83
84	inf = file(canon, "r")	84	inf = file(canon, "r")
85	rd = 0;	85	rd = 0;
86	while (rd < size):	86	while (rd < size):
87	data = inf.read(4096);	87	data = inf.read(4096);
88	outf.write(data)	88	outf.write(data)
89	rd += len(data)	89	rd += len(data)
90	inf.close()	90	inf.close()
91		91
92	if (os.path.isdir(canon)):	92	if (os.path.isdir(canon)):
93	dentry = xstruct.create(DENTRY_DIRECTORY % len(name))	93	dentry = xstruct.create(DENTRY_DIRECTORY % len(name))
94	dentry.kind = TMPFS_DIRECTORY	94	dentry.kind = TMPFS_DIRECTORY
95	dentry.fname_len = len(name)	95	dentry.fname_len = len(name)
96	dentry.fname = name	96	dentry.fname = name
97		97
98	outf.write(dentry.pack())	98	outf.write(dentry.pack())
99		99
100	recursion(canon, outf)	100	recursion(canon, outf)
101		101
102	dentry = xstruct.create(DENTRY_NONE)	102	dentry = xstruct.create(DENTRY_NONE)
103	dentry.kind = TMPFS_NONE	103	dentry.kind = TMPFS_NONE
104	dentry.fname_len = 0	104	dentry.fname_len = 0
105		105
106	outf.write(dentry.pack())	106	outf.write(dentry.pack())
107		107
108	def main():	108	def main():
109	if (len(sys.argv) < 3):	109	if (len(sys.argv) < 3):
110	usage(sys.argv[0])	110	usage(sys.argv[0])
111	return	111	return
112		112
113	path = os.path.abspath(sys.argv[1])	113	path = os.path.abspath(sys.argv[1])
114	if (not os.path.isdir(path)):	114	if (not os.path.isdir(path)):
115	print "<PATH> must be a directory"	115	print "<PATH> must be a directory"
116	return	116	return
117		117
118	outf = file(sys.argv[2], "w")	118	outf = file(sys.argv[2], "w")
119		119
120	header = xstruct.create(HEADER)	120	header = xstruct.create(HEADER)
121	header.tag = "TMPFS"	121	header.tag = "TMPFS"
122		122
123	outf.write(header.pack())	123	outf.write(header.pack())
124		124
125	recursion(path, outf)	125	recursion(path, outf)
126		126
127	dentry = xstruct.create(DENTRY_NONE)	127	dentry = xstruct.create(DENTRY_NONE)
128	dentry.kind = TMPFS_NONE	128	dentry.kind = TMPFS_NONE
129	dentry.fname_len = 0	129	dentry.fname_len = 0
130		130
131	outf.write(dentry.pack())	131	outf.write(dentry.pack())
132		132
133	outf.close()	133	outf.close()
134		134
135	if __name__ == '__main__':	135	if __name__ == '__main__':
136	main()	136	main()
137		137

Subversion Repositories HelenOS

(root)/trunk/tools/mktmpfs.py – Rev 3262 → 3263