Lock tuning and registry release fix.
[ric-plt/submgr.git] / test / rco / rco.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 import (
23         "encoding/hex"
24         "errors"
25         submgr "gerrit.o-ran-sc.org/r/ric-plt/submgr/pkg/control"
26         "gerrit.o-ran-sc.org/r/ric-plt/xapp-frame/pkg/xapp"
27         "github.com/spf13/viper"
28         "os"
29         "strconv"
30         "time"
31 )
32
33 type Rco struct {
34         submgr.E2ap
35 }
36
37 var c = make(chan submgr.RmrDatagram, 1)
38 var params xapp.RMRParams
39
40 var requestRawData string
41 var deleteRawData string
42 var seedSN uint16
43 var deleteSeedSN uint16
44
45 func init() {
46         viper.AutomaticEnv()
47         viper.SetEnvPrefix("rco")
48         viper.AllowEmptyEnv(true)
49         requestRawData = viper.GetString("rawdata")
50         if requestRawData == "" {
51                 requestRawData = "00c90020000003ea7e00050000010002ea6300020003ea81000a000000ea6b4003000440"
52         }
53         deleteRawData = viper.GetString("deleterawdata")
54         if deleteRawData == "" {
55                 deleteRawData = "00ca0012000002ea7e00050000010002ea6300020003"
56         }
57         xapp.Logger.Info("Initial RAW DATA: %v", requestRawData)
58         xapp.Logger.Info("Initial DELETE RAW DATA: %v", deleteRawData)
59         seedSN = uint16(viper.GetInt("seed_sn"))
60         if seedSN == 0 || seedSN > 65535 {
61                 seedSN = 12345
62         }
63         deleteSeedSN = uint16(viper.GetInt("delete_seed_sn"))
64         if deleteSeedSN == 0 || deleteSeedSN > 65535 {
65                 deleteSeedSN = seedSN
66         }
67
68         xapp.Logger.Info("Initial SEQUENCE NUMBER: %v", seedSN)
69 }
70
71 func (r *Rco) GeneratePayload(subId uint16) (payload []byte, err error) {
72         skeleton, err := hex.DecodeString(requestRawData)
73         if err != nil {
74                 return make([]byte, 0), errors.New("nable to decode data provided in \"RCO_RAWDATA\" environment variable")
75         }
76         payload, err = r.SetSubscriptionRequestSequenceNumber(skeleton, subId)
77         return
78 }
79
80 func (r *Rco) GenerateDeletePayload(subId uint16) (payload []byte, err error) {
81         skeleton, err := hex.DecodeString(deleteRawData)
82         if err != nil {
83                 return make([]byte, 0), errors.New("unable to decode data provided in \"RCO_DELETERAWDATA\" environment variable")
84         }
85         payload, err = r.SetSubscriptionDeleteRequestSequenceNumber(skeleton, subId)
86         return
87 }
88
89 func (r Rco) Consume(params *xapp.RMRParams) (err error) {
90         switch params.Mtype {
91         case xapp.RICMessageTypes["RIC_SUB_RESP"]:
92                 payloadSeqNum, err := r.GetSubscriptionResponseSequenceNumber(params.Payload)
93                 if err != nil {
94                         xapp.Logger.Error("SUBRESP: Unable to get Subscription Sequence Number from Payload due to: " + err.Error())
95                 }
96                 xapp.Logger.Info("Subscription Response Message Received: RMR SUBSCRIPTION_ID: %v | PAYLOAD SEQUENCE_NUMBER: %v", params.SubId, payloadSeqNum)
97                 return err
98         case xapp.RICMessageTypes["RIC_SUB_DEL_RESP"]:
99                 payloadSeqNum, err := r.GetSubscriptionDeleteResponseSequenceNumber(params.Payload)
100                 if err != nil {
101                         xapp.Logger.Error("DELRESP: Unable to get Subscription Sequence Number from Payload due to: " + err.Error())
102                 }
103                 xapp.Logger.Info("Subscription Delete Response Message Received: RMR SUBSCRIPTION_ID: %v | PAYLOAD SEQUENCE_NUMBER: %v", params.SubId, payloadSeqNum)
104                 return err
105         default:
106                 err = errors.New("Message Type " + strconv.Itoa(params.Mtype) + " is discarded")
107                 xapp.Logger.Error("Unknown message type: %v", err)
108                 return
109         }
110 }
111
112 func (r *Rco) SendRequests() (err error) {
113         message, err := r.GeneratePayload(seedSN)
114         if err != nil {
115                 xapp.Logger.Debug(err.Error())
116                 return
117         }
118         deletemessage, err := r.GenerateDeletePayload(deleteSeedSN)
119         if err != nil {
120                 xapp.Logger.Debug(err.Error())
121                 return
122         }
123         for {
124                 time.Sleep(5 * time.Second)
125                 c <- submgr.RmrDatagram{MessageType: 12010, SubscriptionId: seedSN, Payload: message}
126                 seedSN++
127                 time.Sleep(5 * time.Second)
128                 c <- submgr.RmrDatagram{MessageType: 12020, SubscriptionId: deleteSeedSN, Payload: deletemessage}
129                 deleteSeedSN++
130         }
131 }
132
133 func (r *Rco) Run() {
134         for {
135                 message := <-c
136                 payloadSeqNum, err := r.GetSubscriptionRequestSequenceNumber(message.Payload)
137                 if err != nil {
138                         xapp.Logger.Debug("Unable to get Subscription Sequence Number from Payload due to: " + err.Error())
139                 }
140                 params.SubId = int(message.SubscriptionId)
141                 params.Mtype = message.MessageType
142                 params.PayloadLen = len(message.Payload)
143                 params.Payload = message.Payload
144                 xapp.Logger.Info("Sending Message: TYPE: %v | RMR SUBSCRIPTION_ID: %v | PAYLOAD SEQUENCE_NUMBER: %v)", message.MessageType, message.SubscriptionId, payloadSeqNum)
145                 xapp.Rmr.Send(&params, false)
146         }
147 }
148
149 func (r *Rco) sendInvalidTestMessages() {
150         for {
151                 time.Sleep(7 * time.Second)
152                 c <- submgr.RmrDatagram{MessageType: 10000, SubscriptionId: 0, Payload: make([]byte, 1)}
153                 time.Sleep(7 * time.Second)
154                 c <- submgr.RmrDatagram{MessageType: 12010, SubscriptionId: 0, Payload: make([]byte, 1)}
155         }
156 }
157
158 func main() {
159         rco := Rco{}
160         go xapp.Rmr.Start(rco)
161         go rco.Run()
162         go rco.sendInvalidTestMessages()
163         err := rco.SendRequests()
164         if err != nil {
165                 xapp.Logger.Info("Error: %v", err)
166                 os.Exit(1)
167         }
168 }