RSM/rmrcgo/rmr_c_go_utils.go

   1 //
   2 // Copyright 2019 AT&T Intellectual Property
   3 // Copyright 2019 Nokia
   4 //
   5 // Licensed under the Apache License, Version 2.0 (the "License");
   6 // you may not use this file except in compliance with the License.
   7 // You may obtain a copy of the License at
   8 //
   9 //      http://www.apache.org/licenses/LICENSE-2.0
  10 //
  11 // Unless required by applicable law or agreed to in writing, software
  12 // distributed under the License is distributed on an "AS IS" BASIS,
  13 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14 // See the License for the specific language governing permissions and
  15 // limitations under the License.
  16
  17 //  This source code is part of the near-RT RIC (RAN Intelligent Controller)
  18 //  platform project (RICP).
  19
  20
  21 package rmrcgo
  22
  23 // #cgo LDFLAGS: -L/usr/local/lib -lrmr_nng -lnng
  24 // #include <rmr/rmr.h>
  25 // #include <stdlib.h>
  26 import "C"
  27 import (
  28         "bytes"
  29         "encoding/binary"
  30         "rsm/logger"
  31         "strings"
  32         "unsafe"
  33 )
  34
  35 func convertToMBuf(m *C.rmr_mbuf_t) *MBuf {
  36         payloadArr := C.GoBytes(unsafe.Pointer(m.payload),C.int(m.len))
  37         xActionArr := C.GoBytes(unsafe.Pointer(m.xaction),C.int(RMR_MAX_XACTION_LEN))
  38
  39         // Trim padding (space and 0)
  40         xActionStr :=  strings.TrimRight(string(xActionArr),"\040\000")
  41         xActionArr = []byte(xActionStr)
  42
  43         mbuf := &MBuf{
  44                 MType: int(m.mtype),
  45                 Len:   int(m.len),
  46                 Payload: &payloadArr,
  47                 XAction: &xActionArr,
  48         }
  49
  50         meidBuf := make([]byte, RMR_MAX_MEID_LEN)
  51         if meidCstr := C.rmr_get_meid(m, (*C.uchar)(unsafe.Pointer(&meidBuf[0]))); meidCstr != nil {
  52                 mbuf.Meid =     strings.TrimRight(string(meidBuf), "\000")
  53         }
  54
  55         return mbuf
  56 }
  57
  58 func (ctx *Context) getAllocatedCRmrMBuf(logger *logger.Logger, mBuf *MBuf, maxMsgSize int) (cMBuf *C.rmr_mbuf_t) {
  59         var xActionBuf [RMR_MAX_XACTION_LEN]byte
  60         var meidBuf[RMR_MAX_MEID_LEN]byte
  61
  62         cMBuf = C.rmr_alloc_msg(ctx.RmrCtx, C.int(maxMsgSize))
  63         cMBuf.mtype = C.int(mBuf.MType)
  64         cMBuf.len = C.int(mBuf.Len)
  65
  66         payloadLen := len(*mBuf.Payload)
  67         xActionLen := len(*mBuf.XAction)
  68
  69         //Add padding
  70         copy(xActionBuf[:], *mBuf.XAction)
  71         for i:= xActionLen; i < RMR_MAX_XACTION_LEN; i++{
  72                 xActionBuf[i] = '\040' //space
  73         }
  74
  75         // Add padding
  76         copy(meidBuf[:], mBuf.Meid)
  77         for i:= len(mBuf.Meid); i < RMR_MAX_MEID_LEN; i++{
  78                 meidBuf[i] = 0
  79         }
  80
  81         payloadArr := (*[1 << 30]byte)(unsafe.Pointer(cMBuf.payload))[:payloadLen:payloadLen]
  82         xActionArr := (*[1 << 30]byte)(unsafe.Pointer(cMBuf.xaction))[:RMR_MAX_XACTION_LEN:RMR_MAX_XACTION_LEN]
  83
  84         err := binary.Read(bytes.NewReader(*mBuf.Payload), binary.LittleEndian, payloadArr)
  85         if err != nil {
  86                 ctx.Logger.Errorf(
  87                         "#rmrCgoUtils.getAllocatedCRmrMBuf - Failed to read payload to allocated RMR message buffer")
  88         }
  89         err = binary.Read(bytes.NewReader(xActionBuf[:]), binary.LittleEndian, xActionArr)
  90         if err != nil {
  91                 ctx.Logger.Errorf(
  92                         "#rmrCgoUtils.getAllocatedCRmrMBuf - Failed to read xAction data to allocated RMR message buffer")
  93         }
  94         len := C.rmr_bytes2meid(cMBuf,  (*C.uchar)(unsafe.Pointer(&meidBuf[0])), C.int(RMR_MAX_MEID_LEN))
  95         if int(len) != RMR_MAX_MEID_LEN {
  96                 ctx.Logger.Errorf(
  97                         "#rmrCgoUtils.getAllocatedCRmrMBuf - Failed to copy meid data to allocated RMR message buffer")
  98         }
  99         return cMBuf
 100 }
 101
 102 //TODO: change to assert or return error
 103 func (ctx *Context) checkContextInitialized() {
 104         if ctx.RmrCtx == nil {
 105                 if ctx.Logger != nil {
 106                         ctx.Logger.DPanicf("#rmrCgoUtils.checkContextInitialized - The RMR router has not been initialized")
 107                 }
 108                 panic("#rmrCgoUtils.checkContextInitialized - The RMR router has not been initialized. To initialize router please call Init() method")
 109         }
 110 }