mlsr/mir/fibcontroller.go

//package mir
//@Author: Pei Xinyuan
//@Description:
//@Version: 0.1.0
//@Date: 2022/6/24 12:12:00
//@Copyright: MIN-Group;国家重大科技基础设施——未来网络北大实验室;深圳市信息论与未来网络重点实验室

package mir

import (
	"encoding/json"
	"errors"
	"fmt"
	"github.com/olekukonko/tablewriter"
	"minlib/common"
	"minlib/component"
	"minlib/logicface"
	"minlib/mgmt"
	"minlib/security"
	"mlsr/cmd"
	"os"
	"strconv"
)

const (
	TopPrefix      = "/min-mir/mgmt/localhost"
	IdentityDBPath = "/usr/local/.mir/identity/"
	DefaultId      = "/mlsr/min/daemonGo"
	defaultPasswd  = "pkusz"
	UnixPath       = "/tmp/mir.sock"
)

var controller *mgmt.MIRController

type NextHopInfo struct {
	LogicFaceId uint64
	Cost        uint64
}

type FibInfo struct {
	Identifier   string
	NextHopsInfo []NextHopInfo
}

//  init
//  @Description: 初始化password、keychain和controller
//
func init() {
	//password first
	passwd, err := cmd.AskPassword()
	if err != nil {
		common.LogFatal(err)
	}
	//keychain initialization
	Keychain := new(security.KeyChain)
	if err := Keychain.InitialKeyChainByPath(IdentityDBPath); err != nil {
		common.LogFatal(err)
	}
	if identity := Keychain.GetIdentityByName(DefaultId); identity == nil {
		common.LogFatal("Identity => " + "not exists")
	} else {
		if err := Keychain.SetCurrentIdentity(identity, passwd); err != nil {
			common.LogFatal(err)
		}
	}
	//controller initialization
	controller = mgmt.CreateMIRController(func() (mgmt.IMgmtLogicFace, error) {
		face := new(logicface.LogicFace)
		//unix connection
		if err := face.InitWithUnixSocket(UnixPath); err != nil {
			return nil, err
		}
		return face, nil
	}, true, Keychain)
}

//  AddFib
//  @Description: 添加下一跳路由
//  @param prefix
//  @param logicFaceId
//  @param cost
//  @return error
//

//add a para:pull/push
func AddFib(prefix string, logicFaceId uint64, cost uint64) error {
	//Create Control Parameters
	parameters := &component.ControlParameters{}
	identifier, err := component.CreateIdentifierByString(prefix)
	if err != nil {
		return err
	}
	parameters.SetPrefix(identifier)
	parameters.SetLogicFaceId(logicFaceId)
	parameters.SetCost(cost)

	//Create Command Executor

	commandExecutor, err := controller.PrepareCommandExecutor(mgmt.CreateFibAddCommand(TopPrefix, parameters))
	if err != nil {
		return err
	}

	commandExecutor.SetAutoShutdown(true)

	//Execute Command
	response, err := commandExecutor.Start()
	if err != nil {
		return err
	}
	//Success
	if response.Code != mgmt.ControlResponseCodeSuccess {
		//failure
		return errors.New(fmt.Sprintf("Add next hop for %s => %s failed! errMsg:%s", prefix, logicFaceId, response.Msg))
	}

	return nil
}

//  DelFib
//  @Description: 删除一个指定前缀和下一跳的路由
//  @param prefix
//  @param logicFaceId
//  @return error
//
func DelFib(prefix string, logicFaceId uint64) error {
	parametres := &component.ControlParameters{}
	identifier, err := component.CreateIdentifierByString(prefix)
	if err != nil {
		return err
	}
	parametres.SetPrefix(identifier)
	parametres.SetLogicFaceId(logicFaceId)

	//Create Command Executor
	commandExecutor, err := controller.PrepareCommandExecutor(mgmt.CreateFibDeleteCommand(TopPrefix, parametres))
	if err != nil {
		return err
	}
	commandExecutor.SetAutoShutdown(true)

	//Execute Command
	response, err := commandExecutor.Start()
	if err != nil {
		return err
	}

	//success
	if response.Code != mgmt.ControlResponseCodeSuccess {
		//Failure
		return errors.New(fmt.Sprintf("Delete next hop for %s => %s failed! errMsg:%s", prefix, logicFaceId, response.Msg))
	}
	return nil
}

//  ListFib
//  @Description:显示所有前缀对应的所有下一跳信息
//  @return error
//
func ListFib() error {
	//Create Command Executor
	commnadExecutor, err := controller.PrepareCommandExecutor(mgmt.CreateFibListCommand(TopPrefix))
	if err != nil {
		return err
	}
	commnadExecutor.SetAutoShutdown(true)
	//Execute Command
	response, err := commnadExecutor.Start()
	if err != nil {
		return err
	}
	//反序列化，输出结果
	var fibInfoList []FibInfo
	err = json.Unmarshal(response.GetBytes(), &fibInfoList)
	if err != nil {
		return err
	}
	//使用表格美化输出
	table := tablewriter.NewWriter(os.Stdout)
	for _, fibInfo := range fibInfoList {
		for _, nextHopInfo := range fibInfo.NextHopsInfo {
			table.Append([]string{fibInfo.Identifier, strconv.FormatUint(nextHopInfo.LogicFaceId, 10), strconv.FormatUint(nextHopInfo.Cost, 10)})
		}
	}
	table.SetHeader([]string{"Prefix", "LogicFaceId", "Cost"})
	table.SetHeaderColor(
		tablewriter.Colors{tablewriter.FgHiRedColor, tablewriter.Bold},
		tablewriter.Colors{tablewriter.FgHiRedColor, tablewriter.Bold},
		tablewriter.Colors{tablewriter.FgHiRedColor, tablewriter.Bold})
	table.SetCaption(true, "Fib Table Info")
	table.SetAlignment(tablewriter.ALIGN_CENTER)
	table.SetAutoMergeCellsByColumnIndex([]int{0})
	table.Render()

	return nil
}