WebSVN – HelenOS – Diff – /trunk/tools/mkhord.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.
#
"""
HORD encapsulator
"""
 
import sys
import os
 
import xstruct
 
HEADER = """little:
    char[4] tag            /* 'HORD' */
    uint8_t version        /* version */
    uint8_t encoding       /* encoding */
    uint32_t header_size   /* header size */
    uint64_t payload_size  /* payload size */
"""
 
 
HORD_LSB = 1
 
def align_up(size, alignment):
    "Align upwards to a given alignment"
    return (((size) + ((alignment) - 1)) & ~((alignment) - 1))
 
def usage(prname):
    "Print usage syntax"
    print prname + " <ALIGNMENT> <FS_IMAGE> <HORD_IMAGE>"
 
def main():
    if (len(sys.argv) < 4):
        usage(sys.argv[0])
        return
   
    if (not sys.argv[1].isdigit()):
        print "<ALIGNMENT> must be a number"
        return
   
    align = int(sys.argv[1], 0)
    if (align <= 0):
        print "<ALIGNMENT> must be positive"
        return
   
    fs_image = os.path.abspath(sys.argv[2])
    if (not os.path.isfile(fs_image)):
        print "<FS_IMAGE> must be a file"
        return
   
    inf = file(fs_image, "rb")
    outf = file(sys.argv[3], "wb")
   
    header = xstruct.create(HEADER)
   
    header_size = header.size()
    payload_size = os.path.getsize(fs_image)
   
    header_size_aligned = align_up(header_size, align)
    payload_size_aligned = align_up(payload_size, align)
   
    header.tag = "HORD"
    header.version = 1
    header.encoding = HORD_LSB
    header.header_size = header_size_aligned
    header.payload_size = payload_size_aligned
   
    outf.write(header.pack())
    outf.write(xstruct.create("little: padding[%d]" % (header_size_aligned - header_size)).pack())
   
    outf.write(inf.read())
   
    padding = payload_size_aligned - payload_size
    if (padding > 0):
        outf.write(xstruct.create("little: padding[%d]" % padding).pack())
   
    inf.close()
    outf.close()
 
if __name__ == '__main__':
    main()
 

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

Subversion Repositories HelenOS

(root)/trunk/tools/mkhord.py – Rev 3260 → 3262