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 struct
import xstruct
 
HEADER = xstruct.convert("little: "
    "char[5]  /* 'TMPFS' */ "
)
 
DENTRY = xstruct.convert("little: "
    "uint8_t   /* NONE, FILE or DIRECTORY */ "
    "uint32_t  /* filename length */ "
)
 
SIZE = xstruct.convert("little: "
    "uint32_t  /* file size */ "
)
 
DENTRY_NONE = 0
DENTRY_FILE = 1
DENTRY_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)):
            outf.write(struct.pack(DENTRY, DENTRY_FILE, len(name)))
            outf.write(xstruct.little_string(name))
            size = os.path.getsize(canon)
            rd = 0;
            outf.write(struct.pack(SIZE, size))
           
            inf = file(canon, "r")
            while (rd < size):
                data = inf.read(4096);
                outf.write(data)
                rd += len(data)
            inf.close()
       
        if (os.path.isdir(canon)):
            outf.write(struct.pack(DENTRY, DENTRY_DIRECTORY, len(name)))
            outf.write(xstruct.little_string(name))
            recursion(canon, outf)
            outf.write(struct.pack(DENTRY, DENTRY_NONE, 0))
 
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")
   
    outf.write(struct.pack(HEADER, "TMPFS"))
    recursion(path, outf)
    outf.write(struct.pack(DENTRY, DENTRY_NONE, 0))
 
    outf.close()
       
if __name__ == '__main__':
    main()
 

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

Subversion Repositories HelenOS

(root)/trunk/tools/mktmpfs.py – Rev 3259 → 3260