test/e2t/e2t.go

   1 /*
   2 ==================================================================================
   3   Copyright (c) 2019 AT&T Intellectual Property.
   4   Copyright (c) 2019 Nokia
   5
   6    Licensed under the Apache License, Version 2.0 (the "License");
   7    you may not use this file except in compliance with the License.
   8    You may obtain a copy of the License at
   9
  10        http://www.apache.org/licenses/LICENSE-2.0
  11
  12    Unless required by applicable law or agreed to in writing, software
  13    distributed under the License is distributed on an "AS IS" BASIS,
  14    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15    See the License for the specific language governing permissions and
  16    limitations under the License.
  17 ==================================================================================
  18 */
  19
  20 package main
  21
  22 /*
  23 #include <rmr/RIC_message_types.h>
  24 #include <rmr/rmr.h>
  25
  26 #cgo CFLAGS: -I../
  27 #cgo LDFLAGS: -lrmr_nng -lnng
  28 */
  29 import "C"
  30
  31 import (
  32         "encoding/hex"
  33         "errors"
  34         submgr "gerrit.o-ran-sc.org/r/ric-plt/submgr/pkg/control"
  35         "gerrit.o-ran-sc.org/r/ric-plt/xapp-frame/pkg/xapp"
  36         "github.com/spf13/viper"
  37         "strconv"
  38 )
  39
  40 type E2t struct {
  41         submgr.E2ap
  42 }
  43
  44 var c chan xapp.RMRParams = make(chan xapp.RMRParams, 1)
  45
  46 var requestRawData string
  47 var deleteRawData string
  48
  49 func init() {
  50         viper.AutomaticEnv()
  51         viper.SetEnvPrefix("e2t")
  52         viper.AllowEmptyEnv(true)
  53         requestRawData = viper.GetString("rawdata")
  54         if requestRawData == "" {
  55                 requestRawData = "20c9001d000003ea7e00050000010002ea6300020003ea6c000700ea6d40020004"
  56         }
  57         xapp.Logger.Info("Initial RAW Data: %v", requestRawData)
  58         deleteRawData = viper.GetString("rawdata")
  59         if deleteRawData == "" {
  60                 deleteRawData = "20ca0012000002ea7e00050000010002ea6300020003"
  61         }
  62         xapp.Logger.Info("Initial RAW Data: %v", deleteRawData)
  63 }
  64
  65 func (e *E2t) GenerateRequestPayload(subId uint16) (payload []byte, err error) {
  66         skeleton, err := hex.DecodeString(requestRawData)
  67         if err != nil {
  68                 return make([]byte, 0), errors.New("unable to decode data provided in \"RCO_RAWDATA\" environment variable")
  69         }
  70         payload, err = e.SetSubscriptionResponseSequenceNumber(skeleton, subId)
  71         return
  72 }
  73
  74 func (e *E2t) GenerateDeletePayload(subId uint16) (payload []byte, err error) {
  75         skeleton, err := hex.DecodeString(deleteRawData)
  76         if err != nil {
  77                 return make([]byte, 0), errors.New("unable to decode data provided in \"RCO_RAWDATA\" environment variable")
  78         }
  79         payload, err = e.SetSubscriptionDeleteResponseSequenceNumber(skeleton, subId)
  80         return
  81 }
  82
  83 func (e E2t) Consume(rp *xapp.RMRParams) (err error) {
  84         switch rp.Mtype {
  85         case C.RIC_SUB_REQ:
  86                 err = e.handleSubscriptionRequest(rp)
  87         case C.RIC_SUB_DEL_REQ:
  88                 err = e.handleSubscriptionDeleteRequest(rp)
  89         default:
  90                 err = errors.New("Message Type " + strconv.Itoa(rp.Mtype) + " is discarded")
  91                 xapp.Logger.Error("Unknown message type: %v", err)
  92         }
  93         return
  94 }
  95
  96 func (e E2t) handleSubscriptionRequest(request *xapp.RMRParams) (err error) {
  97         payloadSeqNum, err := e.GetSubscriptionRequestSequenceNumber(request.Payload)
  98         if err != nil {
  99                 xapp.Logger.Error("Unable to get Subscription Sequence Number from Payload due to: " + err.Error())
 100         }
 101         xapp.Logger.Info("Subscription Request Received: RMR SUBSCRIPTION_ID: %v | PAYLOAD SEQUENCE_NUMBER: %v", payloadSeqNum, payloadSeqNum)
 102         payload, err := e.GenerateRequestPayload(payloadSeqNum)
 103         if err != nil {
 104                 xapp.Logger.Debug(err.Error())
 105                 return
 106         }
 107         request.Payload = payload
 108         request.Mtype = 12011
 109         c <- *request
 110         return
 111 }
 112
 113 func (e E2t) handleSubscriptionDeleteRequest(request *xapp.RMRParams) (err error) {
 114         payloadSeqNum, err := e.GetSubscriptionDeleteRequestSequenceNumber(request.Payload)
 115         if err != nil {
 116                 xapp.Logger.Error("Unable to get Subscription Sequence Number from Payload due to: " + err.Error())
 117         }
 118         xapp.Logger.Info("Subscription Delete Request Received: RMR SUBSCRIPTION_ID: %v | PAYLOAD SEQUENCE_NUMBER: %v", payloadSeqNum, payloadSeqNum)
 119         payload, err := e.GenerateDeletePayload(payloadSeqNum)
 120         if err != nil {
 121                 xapp.Logger.Debug(err.Error())
 122                 return
 123         }
 124         request.Payload = payload
 125         request.Mtype = 12021
 126         c <- *request
 127         return
 128 }
 129
 130 func (e *E2t) Run() {
 131         for {
 132                 message := <-c
 133                 var payloadSeqNum uint16
 134                 var err error
 135                 if message.Mtype == 12011 {
 136                         payloadSeqNum, err = e.GetSubscriptionResponseSequenceNumber(message.Payload)
 137                 } else if message.Mtype == 12021 {
 138                         payloadSeqNum, err = e.GetSubscriptionDeleteResponseSequenceNumber(message.Payload)
 139                 } else {
 140                         err = errors.New("OH MY GOD")
 141                 }
 142                 if err != nil {
 143                         xapp.Logger.Debug("Unable to get Subscription Sequence Number from Payload due to: " + err.Error())
 144                 }
 145                 xapp.Logger.Info("Sending Message: TYPE: %v | RMR SUBSCRIPTION_ID: %v | PAYLOAD SEQUENCE_NUMBER: %v)", message.Mtype, message.SubId, payloadSeqNum)
 146                 xapp.Rmr.Send(&message, true)
 147         }
 148 }
 149
 150 func main() {
 151         e2t := E2t{}
 152         go e2t.Run()
 153         xapp.Run(e2t)
 154 }