Rev 3260 | Rev 3263 | Go to most recent revision | Details | Compare with Previous | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line |
|---|---|---|---|
| 3259 | decky | 1 | # |
| 2 | # Copyright (c) 2008 Martin Decky |
||
| 3 | # All rights reserved. |
||
| 4 | # |
||
| 5 | # Redistribution and use in source and binary forms, with or without |
||
| 6 | # modification, are permitted provided that the following conditions |
||
| 7 | # are met: |
||
| 8 | # |
||
| 9 | # - Redistributions of source code must retain the above copyright |
||
| 10 | # notice, this list of conditions and the following disclaimer. |
||
| 11 | # - Redistributions in binary form must reproduce the above copyright |
||
| 12 | # notice, this list of conditions and the following disclaimer in the |
||
| 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 |
||
| 15 | # derived from this software without specific prior written permission. |
||
| 16 | # |
||
| 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 |
||
| 19 | # OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. |
||
| 20 | # IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, |
||
| 21 | # INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT |
||
| 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 |
||
| 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 |
||
| 26 | # THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||
| 27 | # |
||
| 28 | """ |
||
| 3262 | decky | 29 | Convert descriptive structure definitions to structure object |
| 3259 | decky | 30 | """ |
| 31 | |||
| 32 | import struct |
||
| 33 | |||
| 3262 | decky | 34 | class Struct: |
| 35 | def size(self): |
||
| 36 | return struct.calcsize(self._format_) |
||
| 3259 | decky | 37 | |
| 3262 | decky | 38 | def pack(self): |
| 39 | list = [] |
||
| 40 | for variable in self._list_: |
||
| 41 | list.append(self.__dict__[variable]) |
||
| 42 | |||
| 43 | return struct.pack(self._format_, *list) |
||
| 44 | |||
| 45 | def create(definition): |
||
| 46 | "Create structure object" |
||
| 47 | |||
| 3259 | decky | 48 | tokens = definition.split(None) |
| 49 | |||
| 50 | # Initial byte order tag |
||
| 3262 | decky | 51 | format = { |
| 3259 | decky | 52 | "little:": lambda: "<", |
| 53 | "big:": lambda: ">", |
||
| 54 | "network:": lambda: "!" |
||
| 55 | }[tokens[0]]() |
||
| 3262 | decky | 56 | inst = Struct() |
| 57 | list = [] |
||
| 3260 | decky | 58 | |
| 3259 | decky | 59 | # Member tags |
| 3260 | decky | 60 | comment = False |
| 3262 | decky | 61 | variable = False |
| 3259 | decky | 62 | for token in tokens[1:]: |
| 3260 | decky | 63 | if (comment): |
| 64 | if (token == "*/"): |
||
| 65 | comment = False |
||
| 66 | continue |
||
| 67 | |||
| 3262 | decky | 68 | if (variable): |
| 69 | inst.__dict__[token] = None |
||
| 70 | list.append(token) |
||
| 71 | variable = False |
||
| 72 | continue |
||
| 73 | |||
| 3260 | decky | 74 | if (token == "/*"): |
| 75 | comment = True |
||
| 3262 | decky | 76 | elif (token[0:8] == "padding["): |
| 77 | size = token[8:].split("]")[0] |
||
| 78 | format += "%dx" % int(size) |
||
| 3260 | decky | 79 | elif (token[0:5] == "char["): |
| 3259 | decky | 80 | size = token[5:].split("]")[0] |
| 3262 | decky | 81 | format += "%ds" % int(size) |
| 82 | variable = True |
||
| 3259 | decky | 83 | else: |
| 3262 | decky | 84 | format += { |
| 3259 | decky | 85 | "uint8_t": lambda: "B", |
| 86 | "uint16_t": lambda: "H", |
||
| 87 | "uint32_t": lambda: "L", |
||
| 88 | "uint64_t": lambda: "Q", |
||
| 89 | |||
| 90 | "int8_t": lambda: "b", |
||
| 91 | "int16_t": lambda: "h", |
||
| 92 | "int32_t": lambda: "l", |
||
| 93 | "int64_t": lambda: "q" |
||
| 94 | }[token]() |
||
| 3262 | decky | 95 | variable = True |
| 3259 | decky | 96 | |
| 3262 | decky | 97 | inst.__dict__['_format_'] = format |
| 98 | inst.__dict__['_list_'] = list |
||
| 99 | return inst |