| package devices |
| |
| import ( |
| "gerrit.opencord.org/bbsim/api/openolt" |
| "github.com/looplab/fsm" |
| log "github.com/sirupsen/logrus" |
| ) |
| |
| var onuLogger = log.WithFields(log.Fields{ |
| "module": "ONU", |
| }) |
| |
| func CreateONU(olt OltDevice, pon PonPort, id uint32) Onu { |
| o := Onu{ |
| ID: id, |
| PonPortID: pon.ID, |
| PonPort: pon, |
| channel: make(chan Message), |
| } |
| o.SerialNumber = o.NewSN(olt.ID, pon.ID, o.ID) |
| |
| // NOTE this state machine is used to track the operational |
| // state as requested by VOLTHA |
| o.OperState = getOperStateFSM(func(e *fsm.Event) { |
| onuLogger.WithFields(log.Fields{ |
| "ID": o.ID, |
| }).Debugf("Changing ONU OperState from %s to %s", e.Src, e.Dst) |
| }) |
| |
| // NOTE this state machine is used to activate the OMCI, EAPOL and DHCP clients |
| o.InternalState = fsm.NewFSM( |
| "created", |
| fsm.Events{ |
| {Name: "discover", Src: []string{"created"}, Dst: "discovered"}, |
| {Name: "enable", Src: []string{"discovered"}, Dst: "enabled"}, |
| {Name: "start_omci", Src: []string{"enabled"}, Dst: "starting_openomci"}, |
| }, |
| fsm.Callbacks{ |
| "enter_state": func(e *fsm.Event) { |
| onuLogger.WithFields(log.Fields{ |
| "ID": o.ID, |
| }).Debugf("Changing ONU InternalState from %s to %s", e.Src, e.Dst) |
| }, |
| }, |
| ) |
| return o |
| } |
| |
| func (o Onu) processOnuMessages(stream openolt.Openolt_EnableIndicationServer) { |
| onuLogger.WithFields(log.Fields{ |
| "onuID": o.ID, |
| "onuSN": o.SerialNumber, |
| }).Debug("Started ONU Indication Channel") |
| |
| for message := range o.channel { |
| onuLogger.WithFields(log.Fields{ |
| "onuID": o.ID, |
| "onuSN": o.SerialNumber, |
| "messageType": message.Type, |
| }).Trace("Received message") |
| |
| switch message.Type { |
| case OnuDiscIndication: |
| msg, _ := message.Data.(OnuDiscIndicationMessage) |
| o.sendOnuDiscIndication(msg, stream) |
| case OnuIndication: |
| msg, _ := message.Data.(OnuIndicationMessage) |
| o.sendOnuIndication(msg, stream) |
| case OMCI: |
| o.InternalState.Event("start_omci") |
| onuLogger.Warn("Don't know how to handle OMCI Messages yet...") |
| default: |
| onuLogger.Warnf("Received unknown message data %v for type %v in OLT channel", message.Data, message.Type) |
| } |
| } |
| } |
| |
| func (o Onu) NewSN(oltid int, intfid uint32, onuid uint32) *openolt.SerialNumber { |
| |
| sn := new(openolt.SerialNumber) |
| |
| sn = new(openolt.SerialNumber) |
| sn.VendorId = []byte("BBSM") |
| sn.VendorSpecific = []byte{0, byte(oltid % 256), byte(intfid), byte(onuid)} |
| |
| return sn |
| } |
| |
| func (o Onu) sendOnuDiscIndication(msg OnuDiscIndicationMessage, stream openolt.Openolt_EnableIndicationServer) { |
| discoverData := &openolt.Indication_OnuDiscInd{OnuDiscInd: &openolt.OnuDiscIndication{ |
| IntfId: msg.Onu.PonPortID, |
| SerialNumber: msg.Onu.SerialNumber, |
| }} |
| if err := stream.Send(&openolt.Indication{Data: discoverData}); err != nil { |
| log.Error("Failed to send Indication_OnuDiscInd: %v", err) |
| } |
| o.InternalState.Event("discover") |
| onuLogger.WithFields(log.Fields{ |
| "IntfId": msg.Onu.PonPortID, |
| "SerialNumber": msg.Onu.SerialNumber, |
| }).Debug("Sent Indication_OnuDiscInd") |
| } |
| |
| func (o Onu) sendOnuIndication(msg OnuIndicationMessage, stream openolt.Openolt_EnableIndicationServer) { |
| // NOTE voltha returns an ID, but if we use that ID then it complains: |
| // expected_onu_id: 1, received_onu_id: 1024, event: ONU-id-mismatch, can happen if both voltha and the olt rebooted |
| // so we're using the internal ID that is 1 |
| // o.ID = msg.OnuID |
| o.OperState.Event("enable") |
| |
| indData := &openolt.Indication_OnuInd{OnuInd: &openolt.OnuIndication{ |
| IntfId: o.PonPortID, |
| OnuId: o.ID, |
| OperState: o.OperState.Current(), |
| AdminState: o.OperState.Current(), |
| SerialNumber: o.SerialNumber, |
| }} |
| if err := stream.Send(&openolt.Indication{Data: indData}); err != nil { |
| log.Error("Failed to send Indication_OnuInd: %v", err) |
| } |
| o.InternalState.Event("enable") |
| onuLogger.WithFields(log.Fields{ |
| "IntfId": o.PonPortID, |
| "OnuId": o.ID, |
| "OperState": msg.OperState.String(), |
| "AdminState": msg.OperState.String(), |
| "SerialNumber": o.SerialNumber, |
| }).Debug("Sent Indication_OnuInd") |
| } |