Rich Lane | a06d0c3 | 2013-03-25 08:52:03 -0700 | [diff] [blame] | 1 | # Copyright 2013, Big Switch Networks, Inc. |
| 2 | # |
| 3 | # LoxiGen is licensed under the Eclipse Public License, version 1.0 (EPL), with |
| 4 | # the following special exception: |
| 5 | # |
| 6 | # LOXI Exception |
| 7 | # |
| 8 | # As a special exception to the terms of the EPL, you may distribute libraries |
| 9 | # generated by LoxiGen (LoxiGen Libraries) under the terms of your choice, provided |
| 10 | # that copyright and licensing notices generated by LoxiGen are not altered or removed |
| 11 | # from the LoxiGen Libraries and the notice provided below is (i) included in |
| 12 | # the LoxiGen Libraries, if distributed in source code form and (ii) included in any |
| 13 | # documentation for the LoxiGen Libraries, if distributed in binary form. |
| 14 | # |
| 15 | # Notice: "Copyright 2013, Big Switch Networks, Inc. This library was generated by the LoxiGen Compiler." |
| 16 | # |
| 17 | # You may not use this file except in compliance with the EPL or LOXI Exception. You may obtain |
| 18 | # a copy of the EPL at: |
| 19 | # |
| 20 | # http://www.eclipse.org/legal/epl-v10.html |
| 21 | # |
| 22 | # Unless required by applicable law or agreed to in writing, software |
| 23 | # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT |
| 24 | # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the |
| 25 | # EPL for the specific language governing permissions and limitations |
| 26 | # under the EPL. |
| 27 | |
Rich Lane | 8a22cda | 2013-06-19 13:20:07 -0700 | [diff] [blame] | 28 | from collections import namedtuple |
| 29 | |
| 30 | OFTypeData = namedtuple("OFTypeData", ["init", "pack", "unpack"]) |
| 31 | |
Rich Lane | c854bae | 2013-06-19 14:39:51 -0700 | [diff] [blame] | 32 | # Map from LOXI type name to an object with templates for init, pack, and unpack |
| 33 | # Most types are defined using the convenience code below. This dict should |
| 34 | # only be used directly for special cases such as primitive types. |
Rich Lane | da11f8b | 2013-06-19 15:53:25 -0700 | [diff] [blame] | 35 | type_data_map = { |
Rich Lane | 8a22cda | 2013-06-19 13:20:07 -0700 | [diff] [blame] | 36 | 'char': OFTypeData( |
| 37 | init='0', |
| 38 | pack='struct.pack("!B", %s)', |
| 39 | unpack='%s.read("!B")[0]'), |
| 40 | |
| 41 | 'uint8_t': OFTypeData( |
| 42 | init='0', |
| 43 | pack='struct.pack("!B", %s)', |
| 44 | unpack='%s.read("!B")[0]'), |
| 45 | |
| 46 | 'uint16_t': OFTypeData( |
| 47 | init='0', |
| 48 | pack='struct.pack("!H", %s)', |
| 49 | unpack='%s.read("!H")[0]'), |
| 50 | |
| 51 | 'uint32_t': OFTypeData( |
| 52 | init='0', |
| 53 | pack='struct.pack("!L", %s)', |
| 54 | unpack='%s.read("!L")[0]'), |
| 55 | |
| 56 | 'uint64_t': OFTypeData( |
| 57 | init='0', |
| 58 | pack='struct.pack("!Q", %s)', |
| 59 | unpack='%s.read("!Q")[0]'), |
| 60 | |
| 61 | 'of_port_no_t': OFTypeData( |
| 62 | init='0', |
| 63 | pack='util.pack_port_no(%s)', |
| 64 | unpack='util.unpack_port_no(%s)'), |
| 65 | |
| 66 | 'of_fm_cmd_t': OFTypeData( |
| 67 | init='0', |
| 68 | pack='util.pack_fm_cmd(%s)', |
| 69 | unpack='util.unpack_fm_cmd(%s)'), |
| 70 | |
| 71 | 'of_wc_bmap_t': OFTypeData( |
| 72 | init='util.init_wc_bmap()', |
| 73 | pack='util.pack_wc_bmap(%s)', |
| 74 | unpack='util.unpack_wc_bmap(%s)'), |
| 75 | |
| 76 | 'of_match_bmap_t': OFTypeData( |
| 77 | init='util.init_match_bmap()', |
| 78 | pack='util.pack_match_bmap(%s)', |
| 79 | unpack='util.unpack_match_bmap(%s)'), |
| 80 | |
Andreas Wundsam | b566a16 | 2013-07-18 19:30:23 -0700 | [diff] [blame] | 81 | 'of_ipv4_t': OFTypeData( |
| 82 | init='0', |
| 83 | pack='struct.pack("!L", %s)', |
| 84 | unpack='%s.read("!L")[0]'), |
| 85 | |
Rich Lane | 8a22cda | 2013-06-19 13:20:07 -0700 | [diff] [blame] | 86 | 'of_ipv6_t': OFTypeData( |
| 87 | init="'\\x00\\x00\\x00\\x00\\x00\\x00\\x00\\x00\\x00\\x00\\x00\\x00\\x00\\x00\\x00\\x00'", |
| 88 | pack='struct.pack("!16s", %s)', |
| 89 | unpack="%s.read('!16s')[0]"), |
| 90 | |
| 91 | 'of_mac_addr_t': OFTypeData( |
| 92 | init='[0,0,0,0,0,0]', |
| 93 | pack='struct.pack("!6B", *%s)', |
| 94 | unpack="list(%s.read('!6B'))"), |
| 95 | |
| 96 | 'of_octets_t': OFTypeData( |
| 97 | init="''", |
| 98 | pack='%s', |
| 99 | unpack='str(%s.read_all())'), |
| 100 | |
Rich Lane | 8a22cda | 2013-06-19 13:20:07 -0700 | [diff] [blame] | 101 | # HACK need the match_v3 length field |
| 102 | 'list(of_oxm_t)': OFTypeData( |
| 103 | init='[]', |
| 104 | pack='util.pack_list(%s)', |
| 105 | unpack='oxm.unpack_list(%s.slice(_length-4))'), |
| 106 | |
Rich Lane | be90eae | 2013-07-22 16:44:26 -0700 | [diff] [blame] | 107 | 'of_oxm_t': OFTypeData( |
| 108 | init='None', |
| 109 | pack='%s.pack()', |
| 110 | unpack='oxm.unpack(%s)'), |
| 111 | |
Rich Lane | c854bae | 2013-06-19 14:39:51 -0700 | [diff] [blame] | 112 | # TODO implement unpack |
| 113 | 'list(of_table_features_t)': OFTypeData( |
Rich Lane | 8a22cda | 2013-06-19 13:20:07 -0700 | [diff] [blame] | 114 | init='[]', |
| 115 | pack='util.pack_list(%s)', |
Rich Lane | c854bae | 2013-06-19 14:39:51 -0700 | [diff] [blame] | 116 | unpack=None), |
Rich Lane | 8a22cda | 2013-06-19 13:20:07 -0700 | [diff] [blame] | 117 | |
Rich Lane | c854bae | 2013-06-19 14:39:51 -0700 | [diff] [blame] | 118 | # TODO implement unpack |
Rich Lane | 8a22cda | 2013-06-19 13:20:07 -0700 | [diff] [blame] | 119 | 'list(of_action_id_t)': OFTypeData( |
| 120 | init='[]', |
| 121 | pack='util.pack_list(%s)', |
| 122 | unpack=None), |
| 123 | |
Rich Lane | c854bae | 2013-06-19 14:39:51 -0700 | [diff] [blame] | 124 | # TODO implement unpack |
Rich Lane | 8a22cda | 2013-06-19 13:20:07 -0700 | [diff] [blame] | 125 | 'list(of_table_feature_prop_t)': OFTypeData( |
| 126 | init='[]', |
| 127 | pack='util.pack_list(%s)', |
| 128 | unpack=None), |
Rich Lane | 8a22cda | 2013-06-19 13:20:07 -0700 | [diff] [blame] | 129 | } |
Rich Lane | a06d0c3 | 2013-03-25 08:52:03 -0700 | [diff] [blame] | 130 | |
Rich Lane | c854bae | 2013-06-19 14:39:51 -0700 | [diff] [blame] | 131 | ## Fixed length strings |
| 132 | |
| 133 | # Map from class name to length |
| 134 | fixed_length_strings = { |
| 135 | 'of_port_name_t': 16, |
| 136 | 'of_table_name_t': 32, |
| 137 | 'of_serial_num_t': 32, |
| 138 | 'of_desc_str_t': 256, |
| 139 | } |
| 140 | |
| 141 | for (cls, length) in fixed_length_strings.items(): |
Rich Lane | da11f8b | 2013-06-19 15:53:25 -0700 | [diff] [blame] | 142 | type_data_map[cls] = OFTypeData( |
Rich Lane | c854bae | 2013-06-19 14:39:51 -0700 | [diff] [blame] | 143 | init='""', |
| 144 | pack='struct.pack("!%ds", %%s)' % length, |
| 145 | unpack='%%s.read("!%ds")[0].rstrip("\\x00")' % length) |
| 146 | |
| 147 | ## Embedded structs |
| 148 | |
| 149 | # Map from class name to Python class name |
| 150 | embedded_structs = { |
| 151 | 'of_match_t': 'common.match', |
| 152 | 'of_port_desc_t': 'common.port_desc', |
| 153 | 'of_meter_features_t': 'common.meter_features', |
| 154 | 'of_bsn_vport_q_in_q_t': 'common.bsn_vport_q_in_q', |
| 155 | } |
| 156 | |
| 157 | for (cls, pyclass) in embedded_structs.items(): |
Rich Lane | da11f8b | 2013-06-19 15:53:25 -0700 | [diff] [blame] | 158 | type_data_map[cls] = OFTypeData( |
Rich Lane | c854bae | 2013-06-19 14:39:51 -0700 | [diff] [blame] | 159 | init='%s()' % pyclass, |
| 160 | pack='%s.pack()', |
| 161 | unpack='%s.unpack(%%s)' % pyclass) |
| 162 | |
| 163 | ## Variable element length lists |
| 164 | |
| 165 | # Map from list class name to list deserializer |
| 166 | variable_elem_len_lists = { |
| 167 | 'list(of_action_t)': 'action.unpack_list', |
| 168 | 'list(of_bucket_t)': 'common.unpack_list_bucket', |
| 169 | 'list(of_flow_stats_entry_t)': 'common.unpack_list_flow_stats_entry', |
| 170 | 'list(of_group_desc_stats_entry_t)': 'common.unpack_list_group_desc_stats_entry', |
| 171 | 'list(of_group_stats_entry_t)': 'common.unpack_list_group_stats_entry', |
| 172 | 'list(of_hello_elem_t)': 'common.unpack_list_hello_elem', |
| 173 | 'list(of_instruction_t)': 'instruction.unpack_list', |
| 174 | 'list(of_meter_band_t)': 'meter_band.unpack_list', |
| 175 | 'list(of_meter_stats_t)': 'common.unpack_list_meter_stats', |
| 176 | 'list(of_packet_queue_t)': 'common.unpack_list_packet_queue', |
| 177 | 'list(of_queue_prop_t)': 'common.unpack_list_queue_prop', |
| 178 | } |
| 179 | |
| 180 | for (cls, deserializer) in variable_elem_len_lists.items(): |
Rich Lane | da11f8b | 2013-06-19 15:53:25 -0700 | [diff] [blame] | 181 | type_data_map[cls] = OFTypeData( |
Rich Lane | c854bae | 2013-06-19 14:39:51 -0700 | [diff] [blame] | 182 | init='[]', |
| 183 | pack='util.pack_list(%s)', |
| 184 | unpack='%s(%%s)' % deserializer) |
| 185 | |
| 186 | ## Fixed element length lists |
| 187 | |
| 188 | # Map from list class name to list element deserializer |
| 189 | fixed_elem_len_lists = { |
| 190 | 'list(of_bsn_interface_t)': 'common.bsn_interface.unpack', |
| 191 | 'list(of_bucket_counter_t)': 'common.bucket_counter.unpack', |
| 192 | 'list(of_meter_band_stats_t)': 'common.meter_band_stats.unpack', |
| 193 | 'list(of_port_desc_t)': 'common.port_desc.unpack', |
| 194 | 'list(of_port_stats_entry_t)': 'common.port_stats_entry.unpack', |
| 195 | 'list(of_queue_stats_entry_t)': 'common.queue_stats_entry.unpack', |
| 196 | 'list(of_table_stats_entry_t)': 'common.table_stats_entry.unpack', |
| 197 | 'list(of_uint32_t)': 'common.uint32.unpack', |
| 198 | 'list(of_uint8_t)': 'common.uint8.unpack', |
| 199 | } |
| 200 | |
| 201 | for (cls, element_deserializer) in fixed_elem_len_lists.items(): |
Rich Lane | da11f8b | 2013-06-19 15:53:25 -0700 | [diff] [blame] | 202 | type_data_map[cls] = OFTypeData( |
Rich Lane | c854bae | 2013-06-19 14:39:51 -0700 | [diff] [blame] | 203 | init='[]', |
| 204 | pack='util.pack_list(%s)', |
| 205 | unpack='loxi.generic_util.unpack_list(%%s, %s)' % element_deserializer) |
| 206 | |
Rich Lane | da11f8b | 2013-06-19 15:53:25 -0700 | [diff] [blame] | 207 | ## Public interface |
Rich Lane | c854bae | 2013-06-19 14:39:51 -0700 | [diff] [blame] | 208 | |
Rich Lane | da11f8b | 2013-06-19 15:53:25 -0700 | [diff] [blame] | 209 | # Return an initializer expression for the given oftype |
| 210 | def gen_init_expr(oftype): |
| 211 | type_data = type_data_map.get(oftype) |
| 212 | if type_data and type_data.init: |
| 213 | return type_data.init |
| 214 | else: |
| 215 | return "loxi.unimplemented('init %s')" % oftype |
Rich Lane | a06d0c3 | 2013-03-25 08:52:03 -0700 | [diff] [blame] | 216 | |
Rich Lane | da11f8b | 2013-06-19 15:53:25 -0700 | [diff] [blame] | 217 | # Return a pack expression for the given oftype |
| 218 | # |
| 219 | # 'value_expr' is a string of Python code which will evaluate to |
| 220 | # the value to be packed. |
| 221 | def gen_pack_expr(oftype, value_expr): |
| 222 | type_data = type_data_map.get(oftype) |
| 223 | if type_data and type_data.pack: |
| 224 | return type_data.pack % value_expr |
| 225 | else: |
| 226 | return "loxi.unimplemented('pack %s')" % oftype |
Rich Lane | a06d0c3 | 2013-03-25 08:52:03 -0700 | [diff] [blame] | 227 | |
Rich Lane | da11f8b | 2013-06-19 15:53:25 -0700 | [diff] [blame] | 228 | # Return an unpack expression for the given oftype |
| 229 | # |
| 230 | # 'reader_expr' is a string of Python code which will evaluate to |
| 231 | # the OFReader instance used for deserialization. |
| 232 | def gen_unpack_expr(oftype, reader_expr): |
| 233 | type_data = type_data_map.get(oftype) |
| 234 | if type_data and type_data.unpack: |
| 235 | return type_data.unpack % reader_expr |
| 236 | else: |
| 237 | return "loxi.unimplemented('unpack %s')" % oftype |