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


#
# 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.
#
"""
Convert descriptive structure definitions to structure object
"""
 
import struct
 
class Struct:
    def size(self):
        return struct.calcsize(self._format_)
   
    def pack(self):
        args = []
        for variable in self._args_:
            if (isinstance(self.__dict__[variable], list)):
                for item in self.__dict__[variable]:
                    args.append(item)
            else:
                args.append(self.__dict__[variable])
       
        return struct.pack(self._format_, *args)
 
def create(definition):
    "Create structure object"
   
    tokens = definition.split(None)
   
    # Initial byte order tag
    format = {
        "little:":  lambda: "<",
        "big:":     lambda: ">",
        "network:": lambda: "!"
    }[tokens[0]]()
    inst = Struct()
    args = []
   
    # Member tags
    comment = False
    variable = None
    for token in tokens[1:]:
        if (comment):
            if (token == "*/"):
                comment = False
            continue
       
        if (token == "/*"):
            comment = True
            continue
       
        if (variable != None):
            subtokens = token.split("[")
           
            if (len(subtokens) > 1):
                format += "%d" % int(subtokens[1].split("]")[0])
           
            format += variable
           
            inst.__dict__[subtokens[0]] = None
            args.append(subtokens[0])
           
            variable = None
            continue
       
        if (token[0:8] == "padding["):
            size = token[8:].split("]")[0]
            format += "%dx" % int(size)
            continue
       
        variable = {
            "char":     lambda: "s",
            "uint8_t":  lambda: "B",
            "uint16_t": lambda: "H",
            "uint32_t": lambda: "L",
            "uint64_t": lambda: "Q",
           
            "int8_t":   lambda: "b",
            "int16_t":  lambda: "h",
            "int32_t":  lambda: "l",
            "int64_t":  lambda: "q"
        }[token]()
   
    inst.__dict__['_format_'] = format
    inst.__dict__['_args_'] = args
    return inst
 

Rev 3263	Rev 3264
1	#	1	#
2	# Copyright (c) 2008 Martin Decky	2	# Copyright (c) 2008 Martin Decky
3	# All rights reserved.	3	# All rights reserved.
4	#	4	#
5	# Redistribution and use in source and binary forms, with or without	5	# Redistribution and use in source and binary forms, with or without
6	# modification, are permitted provided that the following conditions	6	# modification, are permitted provided that the following conditions
7	# are met:	7	# are met:
8	#	8	#
9	# - Redistributions of source code must retain the above copyright	9	# - Redistributions of source code must retain the above copyright
10	# notice, this list of conditions and the following disclaimer.	10	# notice, this list of conditions and the following disclaimer.
11	# - Redistributions in binary form must reproduce the above copyright	11	# - Redistributions in binary form must reproduce the above copyright
12	# notice, this list of conditions and the following disclaimer in the	12	# notice, this list of conditions and the following disclaimer in the
13	# documentation and/or other materials provided with the distribution.	13	# documentation and/or other materials provided with the distribution.
14	# - The name of the author may not be used to endorse or promote products	14	# - The name of the author may not be used to endorse or promote products
15	# derived from this software without specific prior written permission.	15	# derived from this software without specific prior written permission.
16	#	16	#
17	# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR	17	# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18	# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES	18	# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19	# OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.	19	# OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20	# IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,	20	# IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21	# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT	21	# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22	# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,	22	# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23	# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY	23	# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24	# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT	24	# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25	# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF	25	# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26	# THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.	26	# THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27	#	27	#
28	"""	28	"""
29	Convert descriptive structure definitions to structure object	29	Convert descriptive structure definitions to structure object
30	"""	30	"""
31		31
32	import struct	32	import struct
33		33
34	class Struct:	34	class Struct:
35	def size(self):	35	def size(self):
36	return struct.calcsize(self._format_)	36	return struct.calcsize(self._format_)
37		37
38	def pack(self):	38	def pack(self):
39	list = []	39	args = []
40	for variable in self._list_:	40	for variable in self._args_:
-		41	if (isinstance(self.__dict__[variable], list)):
-		42	for item in self.__dict__[variable]:
-		43	args.append(item)
-		44	else:
41	list.append(self.__dict__[variable])	45	args.append(self.__dict__[variable])
42		46
43	return struct.pack(self._format_, *list)	47	return struct.pack(self._format_, *args)
44		48
45	def create(definition):	49	def create(definition):
46	"Create structure object"	50	"Create structure object"
47		51
48	tokens = definition.split(None)	52	tokens = definition.split(None)
49		53
50	# Initial byte order tag	54	# Initial byte order tag
51	format = {	55	format = {
52	"little:": lambda: "<",	56	"little:": lambda: "<",
53	"big:": lambda: ">",	57	"big:": lambda: ">",
54	"network:": lambda: "!"	58	"network:": lambda: "!"
55	}[tokens[0]]()	59	}[tokens[0]]()
56	inst = Struct()	60	inst = Struct()
57	list = []	61	args = []
58		62
59	# Member tags	63	# Member tags
60	comment = False	64	comment = False
61	variable = None	65	variable = None
62	for token in tokens[1:]:	66	for token in tokens[1:]:
63	if (comment):	67	if (comment):
64	if (token == "*/"):	68	if (token == "*/"):
65	comment = False	69	comment = False
66	continue	70	continue
67		71
68	if (token == "/*"):	72	if (token == "/*"):
69	comment = True	73	comment = True
70	continue	74	continue
71		75
72	if (variable != None):	76	if (variable != None):
73	subtokens = token.split("[")	77	subtokens = token.split("[")
74		78
75	if (len(subtokens) > 1):	79	if (len(subtokens) > 1):
76	format += "%d" % int(subtokens[1].split("]")[0])	80	format += "%d" % int(subtokens[1].split("]")[0])
77		81
78	format += variable	82	format += variable
79		83
80	inst.__dict__[subtokens[0]] = None	84	inst.__dict__[subtokens[0]] = None
81	list.append(subtokens[0])	85	args.append(subtokens[0])
82		86
83	variable = None	87	variable = None
84	continue	88	continue
85		89
86	if (token[0:8] == "padding["):	90	if (token[0:8] == "padding["):
87	size = token[8:].split("]")[0]	91	size = token[8:].split("]")[0]
88	format += "%dx" % int(size)	92	format += "%dx" % int(size)
89	continue	93	continue
90		94
91	variable = {	95	variable = {
92	"char": lambda: "s",	96	"char": lambda: "s",
93	"uint8_t": lambda: "B",	97	"uint8_t": lambda: "B",
94	"uint16_t": lambda: "H",	98	"uint16_t": lambda: "H",
95	"uint32_t": lambda: "L",	99	"uint32_t": lambda: "L",
96	"uint64_t": lambda: "Q",	100	"uint64_t": lambda: "Q",
97		101
98	"int8_t": lambda: "b",	102	"int8_t": lambda: "b",
99	"int16_t": lambda: "h",	103	"int16_t": lambda: "h",
100	"int32_t": lambda: "l",	104	"int32_t": lambda: "l",
101	"int64_t": lambda: "q"	105	"int64_t": lambda: "q"
102	}[token]()	106	}[token]()
103		107
104	inst.__dict__['_format_'] = format	108	inst.__dict__['_format_'] = format
105	inst.__dict__['_list_'] = list	109	inst.__dict__['_args_'] = args
106	return inst	110	return inst
107		111

Subversion Repositories HelenOS

(root)/trunk/tools/xstruct.py – Rev 3263 → 3264