def delete_flow(self, onu_id, intf_id, flow_id, is_downstream): try: obj = bal_pb2.BalCfg() # Fill Header details obj.device_id = self.device_id.encode('ascii', 'ignore') obj.hdr.obj_type = bal_model_ids_pb2.BAL_OBJ_ID_FLOW # Fill Access Terminal Details # To-DO flow ID need to be retrieved from flow details obj.flow.key.flow_id = flow_id if is_downstream is False: obj.flow.key.flow_type = \ bal_model_types_pb2.BAL_FLOW_TYPE_UPSTREAM else: obj.flow.key.flow_type = \ bal_model_types_pb2.BAL_FLOW_TYPE_DOWNSTREAM obj.flow.data.admin_state = bal_model_types_pb2.BAL_STATE_DOWN obj.flow.data.access_int_id = intf_id # obj.flow.data.network_int_id = intf_id obj.flow.data.sub_term_id = onu_id self.log.info('deleting-flows-from-OLT-Device', flow_details=obj) yield self.stub.BalCfgSet(obj) except Exception as e: self.log.info('delete_flow-exception', flow_id, onu_id, exc=str(e)) return
def packet_out(self, pkt, pkt_info): obj = bal_pb2.BalCfg() obj.device_id = self.device_id.encode('ascii', 'ignore') obj.hdr.obj_type = bal_model_ids_pb2.BAL_OBJ_ID_PACKET if pkt_info['dest_type'] == 'onu': # Set the destination ONU info obj.packet.key.packet_send_dest.type = bal_model_types_pb2.BAL_DEST_TYPE_SUB_TERM obj.packet.key.packet_send_dest.sub_term.sub_term_id = pkt_info[ 'onu_id'] # TODO: Need to provide correct values for sub_term_uni and int_id #obj.packet.key.packet_send_dest.sub_term.sub_term_uni = egress_port obj.packet.key.packet_send_dest.sub_term.intf_id = pkt_info[ 'intf_id'] obj.packet.data.intf_type = bal_model_types_pb2.BAL_INTF_TYPE_PON elif pkt_info['dest_type'] == 'gem_port': obj.packet.key.packet_send_dest.type = bal_model_types_pb2.BAL_DEST_TYPE_SVC_PORT obj.packet.key.packet_send_dest.svc_port.svc_port_id = pkt_info[ 'gem_port'] obj.packet.key.packet_send_dest.svc_port.intf_id = pkt_info[ 'intf_id'] obj.packet.data.intf_type = bal_model_types_pb2.BAL_INTF_TYPE_PON elif pkt_info['dest_type'] == 'nni': obj.packet.key.packet_send_dest.type = bal_model_types_pb2.BAL_DEST_TYPE_NNI obj.packet.key.packet_send_dest.nni.intf_id = pkt_info['intf_id'] else: self.log.error('unsupported-dest-type', dest_type=pkt_info['dest_type']) # Set the Packet-out info # TODO: Need to provide correct value for intf_id obj.packet.data.pkt = pkt self.log.info('sending-packet-out', packet_out_details=obj) yield self.stub.BalCfgSet(obj)
def create_scheduler(self, id, direction, owner_info, num_priority, rate_info=None): try: obj = bal_pb2.BalCfg() # Fill Header details obj.device_id = self.device_id.encode('ascii', 'ignore') obj.hdr.obj_type = bal_model_ids_pb2.BAL_OBJ_ID_TM_SCHED # Fill Access Terminal Details if direction == 'downstream': obj.tm_sched_cfg.key.dir = \ bal_model_types_pb2.BAL_TM_SCHED_DIR_DS else: obj.tm_sched_cfg.key.dir = \ bal_model_types_pb2.BAL_TM_SCHED_DIR_US obj.tm_sched_cfg.key.id = id if owner_info['type'] == 'agg_port': obj.tm_sched_cfg.data.owner.type = \ bal_model_types_pb2.BAL_TM_SCHED_OWNER_TYPE_AGG_PORT obj.tm_sched_cfg.data.owner.agg_port.presence_mask = 0 obj.tm_sched_cfg.data.owner.agg_port.intf_id = \ owner_info['intf_id'] obj.tm_sched_cfg.data.owner.agg_port.presence_mask |= \ bal_model_types_pb2.BAL_TM_SCHED_OWNER_AGG_PORT_ID_INTF_ID obj.tm_sched_cfg.data.owner.agg_port.sub_term_id = \ owner_info['onu_id'] obj.tm_sched_cfg.data.owner.agg_port.presence_mask |= \ bal_model_types_pb2.BAL_TM_SCHED_OWNER_AGG_PORT_ID_SUB_TERM_ID obj.tm_sched_cfg.data.owner.agg_port.agg_port_id = \ owner_info['alloc_id'] obj.tm_sched_cfg.data.owner.agg_port.presence_mask |= \ bal_model_types_pb2.BAL_TM_SCHED_OWNER_AGG_PORT_ID_AGG_PORT_ID else: self.log.error('Not-supported-scheduling-type', sched_type=owner_info['type']) return #obj.tm_sched_cfg.data.sched_type = \ # bal_model_types_pb2.BAL_TM_SCHED_TYPE_SP_WFQ #obj.tm_sched_cfg.data.num_priorities = num_priority if rate_info is not None: obj.tm_sched_cfg.data.rate.presence_mask = \ bal_model_types_pb2.BAL_TM_SHAPING_ID_ALL obj.tm_sched_cfg.data.rate.cir = rate_info['cir'] obj.tm_sched_cfg.data.rate.pir = rate_info['pir'] obj.tm_sched_cfg.data.rate.burst = rate_info['burst'] self.log.info('Creating-Scheduler', scheduler_details=obj) yield self.stub.BalCfgSet(obj, timeout=GRPC_TIMEOUT) except Exception as e: self.log.info('creat-scheduler-exception', olt=self.olt.olt_id, sched_id=id, direction=direction, owner=owner_info, rate=rate_info, exc=str(e)) return
def deactivate_eapol_flow(self, flow_id, is_downstream, onu_id=None, intf_id=None, network_int_id=None, gemport_id=None, stag=None, dba_sched_id=None, queue_id=None, queue_sched_id=None): try: obj = bal_pb2.BalCfg() # Fill Header details obj.device_id = self.device_id.encode('ascii', 'ignore') obj.hdr.obj_type = bal_model_ids_pb2.BAL_OBJ_ID_FLOW # Fill Access Terminal Details # To-DO flow ID need to be retrieved from flow details obj.flow.key.flow_id = flow_id obj.flow.data.admin_state = bal_model_types_pb2.BAL_STATE_DOWN if intf_id is not None: obj.flow.data.access_int_id = intf_id if network_int_id is not None: obj.flow.data.network_int_id = network_int_id if onu_id is not None: obj.flow.data.sub_term_id = onu_id if gemport_id is not None: obj.flow.data.svc_port_id = gemport_id if is_downstream is True: obj.flow.key.flow_type = \ bal_model_types_pb2.BAL_FLOW_TYPE_DOWNSTREAM else: obj.flow.key.flow_type = \ bal_model_types_pb2.BAL_FLOW_TYPE_UPSTREAM obj.flow.data.classifier.pkt_tag_type = \ bal_model_types_pb2.BAL_PKT_TAG_TYPE_SINGLE_TAG obj.flow.data.classifier.presence_mask |= \ bal_model_types_pb2.BAL_CLASSIFIER_ID_PKT_TAG_TYPE obj.flow.data.classifier.o_vid = stag obj.flow.data.classifier.presence_mask |= \ bal_model_types_pb2.BAL_CLASSIFIER_ID_O_VID obj.flow.data.action.cmds_bitmask |= \ bal_model_types_pb2.BAL_ACTION_CMD_ID_TRAP_TO_HOST obj.flow.data.action.presence_mask |= \ bal_model_types_pb2.BAL_ACTION_ID_CMDS_BITMASK if dba_sched_id: obj.flow.data.dba_tm_sched_id = dba_sched_id if queue_id: obj.flow.data.queue.queue_id = queue_id if queue_sched_id: obj.flow.data.queue.sched_id = queue_sched_id self.log.info('deactivating-eapol-flows-from-OLT-Device', flow_details=obj) yield self.stub.BalCfgSet(obj, timeout=GRPC_TIMEOUT) except Exception as e: self.log.exception('deactivate-eapol-flow-exception', flow_id, onu_id, exc=str(e)) return
def set_access_terminal_admin_state(self, admin_state): self.log.info('setting-admin-state', admin_state=admin_state, device_id=self.device_id) obj = bal_pb2.BalCfg() obj.device_id = self.device_id.encode('ascii', 'ignore') obj.hdr.obj_type = bal_model_ids_pb2.BAL_OBJ_ID_ACCESS_TERMINAL obj.cfg.key.access_term_id = 0 obj.cfg.data.admin_state = admin_state yield self.stub.BalCfgSet(obj)
def set_access_terminal_admin_state(self, admin_state): obj = bal_pb2.BalCfg() obj.device_id = self.device_id.encode('ascii', 'ignore') obj.hdr.obj_type = bal_model_ids_pb2.BAL_OBJ_ID_ACCESS_TERMINAL obj.cfg.key.access_term_id = 0 obj.cfg.data.admin_state = admin_state self.log.info('Admin-stage-change-Access-Terminal-Device', admin_state=admin_state, device_id=self.device_id, access_terminal_details=obj) yield self.stub.BalCfgSet(obj, timeout=GRPC_TIMEOUT)
def create_queue(self, id, direction, sched_id, priority=None, weight=None, rate_info=None): try: obj = bal_pb2.BalCfg() # Fill Header details obj.device_id = self.device_id.encode('ascii', 'ignore') obj.hdr.obj_type = bal_model_ids_pb2.BAL_OBJ_ID_TM_QUEUE # Fill Queue Cfg Details if direction == 'downstream': obj.tm_queue_cfg.key.sched_dir = \ bal_model_types_pb2.BAL_TM_SCHED_DIR_DS else: obj.tm_queue_cfg.key.sched_dir = \ bal_model_types_pb2.BAL_TM_SCHED_DIR_US obj.tm_queue_cfg.key.id = id obj.tm_queue_cfg.key.sched_id = sched_id ''' TO-DO:By default the schedular created is of type sp_wfq, which requires either priority or weight but not both. Need to fetch schd_type then assign either one of them ''' if weight is not None: obj.tm_queue_cfg.data.weight = weight if rate_info is not None: obj.tm_queue_cfg.data.rate.presence_mask = \ bal_model_types_pb2.BAL_TM_SHAPING_ID_ALL obj.tm_queue_cfg.data.rate.cir = rate_info['cir'] obj.tm_queue_cfg.data.rate.pir = rate_info['pir'] obj.tm_queue_cfg.data.rate.burst = rate_info['burst'] else: obj.tm_queue_cfg.data.rate.presence_mask = \ bal_model_types_pb2.BAL_TM_SHAPING_ID_NONE obj.tm_queue_cfg.data.creation_mode = \ bal_model_types_pb2.BAL_TM_CREATION_MODE_MANUAL obj.tm_queue_cfg.data.ref_count = 0 self.log.info('Creating-Queue', scheduler_details=obj) yield self.stub.BalCfgSet(obj, timeout=GRPC_TIMEOUT) except Exception as e: self.log.info('create-queue-exception', olt=self.olt.olt_id, queue_id=id, direction=direction, sched_id=sched_id, priority=priority, weight=weight, rate_info=rate_info, exc=str(e)) return
def packet_out(self, onu_id, egress_port, pkt): obj = bal_pb2.BalCfg() # Set the destination ONU info obj.packet.key.dest.packet_send_dest.sub_term.sub_term_id = onu_id # TODO: Need to provide correct values for sub_term_uni and int_id obj.packet.key.dest.packet_send_dest.sub_term.sub_term_uni = egress_port obj.packet.key.dest.packet_send_dest.sub_term.int_id = egress_port # Set the Packet-out info obj.packet.data.flow_type = bal_model_types_pb2.BAL_FLOW_TYPE_DOWNSTREAM # TODO: Need to provide correct value for intf_id obj.packet.data.intf_id = egress_port obj.packet.data.pkt = pkt self.log.info('packet-out', packet_out_details=obj) yield self.stub.BalCfgSet(obj)
def deactivate_pon_port(self, olt_no, pon_port, transceiver_type): try: obj = bal_pb2.BalCfg() # Fill Header details obj.device_id = self.device_id.encode('ascii', 'ignore') obj.hdr.obj_type = bal_model_ids_pb2.BAL_OBJ_ID_INTERFACE # Fill Access Terminal Details obj.interface.key.intf_id = pon_port obj.interface.key.intf_type = bal_model_types_pb2.BAL_INTF_TYPE_PON obj.interface.data.admin_state = bal_model_types_pb2.BAL_STATE_DOWN obj.interface.data.transceiver_type = transceiver_type self.log.info('deactivating-pon-port-in-olt', olt=olt_no, pon_port=pon_port, pon_port_details=obj, transceiver_type=transceiver_type) except Exception as e: self.log.info('deactivating-pon-port in olt-exception', exc=str(e)) return
def activate_pon_port(self, olt_no, pon_port): self.log.info('activating-pon-port in olt', olt=olt_no, pon_port=pon_port) try: obj = bal_pb2.BalCfg() # Fill Header details obj.device_id = self.device_id.encode('ascii', 'ignore') obj.hdr.obj_type = bal_model_ids_pb2.BAL_OBJ_ID_INTERFACE # Fill Access Terminal Details obj.interface.key.intf_id = pon_port obj.interface.key.intf_type = bal_model_types_pb2.BAL_INTF_TYPE_PON obj.interface.data.admin_state = bal_model_types_pb2.BAL_STATE_UP obj.interface.data.transceiver_type = \ bal_model_types_pb2.BAL_TRX_TYPE_XGPON_LTH_7226_PC yield self.stub.BalCfgSet(obj) except Exception as e: self.log.info('activating-pon-port in olt-exception', exc=str(e)) return
def send_omci_request_message(self, proxy_address, msg): try: obj = bal_pb2.BalCfg() # Fill Header details obj.device_id = self.device_id.encode('ascii', 'ignore') obj.hdr.obj_type = bal_model_ids_pb2.BAL_OBJ_ID_PACKET # Fill packet Details obj.packet.key.packet_send_dest.type = \ bal_model_types_pb2.BAL_DEST_TYPE_ITU_OMCI_CHANNEL obj.packet.key.packet_send_dest.itu_omci_channel.sub_term_id = \ proxy_address.onu_id obj.packet.key.packet_send_dest.itu_omci_channel.intf_id = \ proxy_address.channel_id obj.packet.data.pkt = msg self.log.info('send_omci_request_message', proxy_address=proxy_address.channel_id, omci_msg_details=obj) yield self.stub.BalCfgSet(obj) except Exception as e: self.log.info('send-proxied_message-exception', exc=str(e)) return
def deactivate_onu(self, onu_info): try: obj = bal_pb2.BalCfg() # Fill Header details obj.device_id = self.device_id.encode('ascii', 'ignore') obj.hdr.obj_type = bal_model_ids_pb2.BAL_OBJ_ID_SUBSCRIBER_TERMINAL # Fill Access Terminal Details obj.terminal.key.intf_id = onu_info['pon_id'] obj.terminal.key.sub_term_id = onu_info['onu_id'] obj.terminal.data.admin_state = bal_model_types_pb2.BAL_STATE_DOWN obj.terminal.data.serial_number.vendor_id = onu_info['vendor'] obj.terminal.data.serial_number.vendor_specific = \ onu_info['vendor_specific'] obj.terminal.data.registration_id = onu_info['reg_id'] self.log.info('deactivating-ONU-in-olt', onu_details=obj) yield self.stub.BalCfgSet(obj, timeout=GRPC_TIMEOUT) except Exception as e: self.log.info('deactivating-ONU-exception', onu_info['onu_id'], exc=str(e)) return
def activate_onu(self, onu_info): self.log.info('activating-ONU in olt', olt=self.olt.olt_id, onu_id=onu_info['onu_id']) try: obj = bal_pb2.BalCfg() # Fill Header details obj.device_id = self.device_id.encode('ascii', 'ignore') obj.hdr.obj_type = bal_model_ids_pb2.BAL_OBJ_ID_SUBSCRIBER_TERMINAL # Fill Access Terminal Details obj.terminal.key.intf_id = onu_info['pon_id'] obj.terminal.key.sub_term_id = onu_info['onu_id'] obj.terminal.data.admin_state = bal_model_types_pb2.BAL_STATE_UP obj.terminal.data.serial_number.vendor_id = onu_info['vendor'] obj.terminal.data.serial_number.vendor_specific = \ onu_info['vendor_specific'] obj.terminal.data.registration_id = \ '202020202020202020202020202020202020202020202020202020202020202020202020' yield self.stub.BalCfgSet(obj) except Exception as e: self.log.info('activating-ONU-exception', onu_info['onu_id'], exc=str(e)) return
def add_flow(self, onu_id, intf_id, flow_id, gem_port, classifier_info, is_downstream, action_info=None, sched_id=None): try: obj = bal_pb2.BalCfg() # Fill Header details obj.device_id = self.device_id.encode('ascii', 'ignore') obj.hdr.obj_type = bal_model_ids_pb2.BAL_OBJ_ID_FLOW # Fill Access Terminal Details # To-DO flow ID need to be retrieved from flow details obj.flow.key.flow_id = flow_id if is_downstream is False: obj.flow.key.flow_type = \ bal_model_types_pb2.BAL_FLOW_TYPE_UPSTREAM obj.flow.data.dba_tm_sched_id = sched_id else: obj.flow.key.flow_type = \ bal_model_types_pb2.BAL_FLOW_TYPE_DOWNSTREAM obj.flow.data.admin_state = bal_model_types_pb2.BAL_STATE_UP obj.flow.data.access_int_id = intf_id # obj.flow.data.network_int_id = intf_id obj.flow.data.sub_term_id = onu_id obj.flow.data.svc_port_id = gem_port obj.flow.data.classifier.presence_mask = 0 if 'eth_type' in classifier_info: obj.flow.data.classifier.ether_type = \ classifier_info['eth_type'] obj.flow.data.classifier.presence_mask |= \ bal_model_types_pb2.BAL_CLASSIFIER_ID_ETHER_TYPE if 'ip_proto' in classifier_info: obj.flow.data.classifier.ip_proto = \ classifier_info['ip_proto'] obj.flow.data.classifier.presence_mask |= \ bal_model_types_pb2.BAL_CLASSIFIER_ID_IP_PROTO if 'vlan_vid' in classifier_info: obj.flow.data.classifier.o_vid = \ classifier_info['vlan_vid'] obj.flow.data.classifier.presence_mask |= \ bal_model_types_pb2.BAL_CLASSIFIER_ID_O_VID if 'vlan_pcp' in classifier_info: obj.flow.data.classifier.o_pbits = \ classifier_info['vlan_pcp'] obj.flow.data.classifier.presence_mask |= \ bal_model_types_pb2.BAL_CLASSIFIER_ID_O_PBITS if 'udp_src' in classifier_info: obj.flow.data.classifier.src_port = \ classifier_info['udp_src'] obj.flow.data.classifier.presence_mask |= \ bal_model_types_pb2.BAL_CLASSIFIER_ID_SRC_PORT if 'udp_dst' in classifier_info: obj.flow.data.classifier.dst_port = \ classifier_info['udp_dst'] obj.flow.data.classifier.presence_mask |= \ bal_model_types_pb2.BAL_CLASSIFIER_ID_DST_PORT if 'ipv4_dst' in classifier_info: obj.flow.data.classifier.dst_ip = \ classifier_info['ipv4_dst'] obj.flow.data.classifier.presence_mask |= \ bal_model_types_pb2.BAL_CLASSIFIER_ID_DST_IP if 'ipv4_src' in classifier_info: obj.flow.data.classifier.src_ip = \ classifier_info['ipv4_src'] obj.flow.data.classifier.presence_mask |= \ bal_model_types_pb2.BAL_CLASSIFIER_ID_SRC_IP if 'metadata' in classifier_info: obj.flow.data.classifier.i_vid = \ classifier_info['metadata'] obj.flow.data.classifier.presence_mask |= \ bal_model_types_pb2.BAL_CLASSIFIER_ID_I_VID if 'pkt_tag_type' in classifier_info: if classifier_info['pkt_tag_type'] == 'single_tag': obj.flow.data.classifier.pkt_tag_type = \ bal_model_types_pb2.BAL_PKT_TAG_TYPE_SINGLE_TAG elif classifier_info['pkt_tag_type'] == 'double_tag': obj.flow.data.classifier.pkt_tag_type = \ bal_model_types_pb2.BAL_PKT_TAG_TYPE_DOUBLE_TAG elif classifier_info['pkt_tag_type'] == 'untagged': obj.flow.data.classifier.pkt_tag_type = \ bal_model_types_pb2.BAL_PKT_TAG_TYPE_UNTAGGED else: obj.flow.data.classifier.pkt_tag_type = \ bal_model_types_pb2.BAL_PKT_TAG_TYPE_NONE obj.flow.data.classifier.presence_mask |= \ bal_model_types_pb2.BAL_CLASSIFIER_ID_PKT_TAG_TYPE if action_info is not None: obj.flow.data.action.presence_mask = 0 obj.flow.data.action.cmds_bitmask = 0 if 'pop_vlan' in action_info: obj.flow.data.action.o_vid = action_info['vlan_vid'] obj.flow.data.action.cmds_bitmask |= \ bal_model_types_pb2.BAL_ACTION_CMD_ID_REMOVE_OUTER_TAG obj.flow.data.action.presence_mask |= \ bal_model_types_pb2.BAL_ACTION_ID_CMDS_BITMASK obj.flow.data.action.presence_mask |= \ bal_model_types_pb2.BAL_ACTION_ID_O_VID elif 'push_vlan' in action_info: obj.flow.data.action.o_vid = action_info['vlan_vid'] obj.flow.data.action.cmds_bitmask |= \ bal_model_types_pb2.BAL_ACTION_CMD_ID_ADD_OUTER_TAG obj.flow.data.action.presence_mask |= \ bal_model_types_pb2.BAL_ACTION_ID_CMDS_BITMASK obj.flow.data.action.presence_mask |= \ bal_model_types_pb2.BAL_ACTION_ID_O_VID elif 'trap_to_host' in action_info: obj.flow.data.action.cmds_bitmask |= \ bal_model_types_pb2.BAL_ACTION_CMD_ID_TRAP_TO_HOST obj.flow.data.action.presence_mask |= \ bal_model_types_pb2.BAL_ACTION_ID_CMDS_BITMASK else: self.log.info('Invalid-action-field') return self.log.info('adding-flow-to-OLT-Device', flow_details=obj) yield self.stub.BalCfgSet(obj) except Exception as e: self.log.info('add_flow-exception', flow_id, onu_id, exc=str(e)) return