Rev 3262 | 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 |
| 3263 | decky | 61 | variable = None |
| 3259 | decky | 62 | for token in tokens[1:]: |
| 3260 | decky | 63 | if (comment): |
| 64 | if (token == "*/"): |
||
| 65 | comment = False |
||
| 66 | continue |
||
| 67 | |||
| 3263 | decky | 68 | if (token == "/*"): |
| 69 | comment = True |
||
| 3262 | decky | 70 | continue |
| 71 | |||
| 3263 | decky | 72 | if (variable != None): |
| 73 | subtokens = token.split("[") |
||
| 74 | |||
| 75 | if (len(subtokens) > 1): |
||
| 76 | format += "%d" % int(subtokens[1].split("]")[0]) |
||
| 77 | |||
| 78 | format += variable |
||
| 79 | |||
| 80 | inst.__dict__[subtokens[0]] = None |
||
| 81 | list.append(subtokens[0]) |
||
| 82 | |||
| 83 | variable = None |
||
| 84 | continue |
||
| 85 | |||
| 86 | if (token[0:8] == "padding["): |
||
| 3262 | decky | 87 | size = token[8:].split("]")[0] |
| 88 | format += "%dx" % int(size) |
||
| 3263 | decky | 89 | continue |
| 90 | |||
| 91 | variable = { |
||
| 92 | "char": lambda: "s", |
||
| 93 | "uint8_t": lambda: "B", |
||
| 94 | "uint16_t": lambda: "H", |
||
| 95 | "uint32_t": lambda: "L", |
||
| 96 | "uint64_t": lambda: "Q", |
||
| 97 | |||
| 98 | "int8_t": lambda: "b", |
||
| 99 | "int16_t": lambda: "h", |
||
| 100 | "int32_t": lambda: "l", |
||
| 101 | "int64_t": lambda: "q" |
||
| 102 | }[token]() |
||
| 3259 | decky | 103 | |
| 3262 | decky | 104 | inst.__dict__['_format_'] = format |
| 105 | inst.__dict__['_list_'] = list |
||
| 106 | return inst |