ziti/router/xgress_common/messages.go

package xgress_common

import (
	"github.com/golang/protobuf/proto"
	"github.com/openziti/channel"
	"github.com/openziti/channel/protobufs"
	"github.com/openziti/edge/pb/edge_ctrl_pb"
	"github.com/pkg/errors"
)

const (
	PayloadFlagsHeader uint8 = 0x10
)

func GetResultOrFailure(msg *channel.Message, err error, result protobufs.TypedMessage) error {
	if err != nil {
		return err
	}

	if msg.ContentType == int32(edge_ctrl_pb.ContentType_ErrorType) {
		msg := string(msg.Body)
		if msg == "" {
			msg = "error state returned from controller with no message"
		}
		return errors.New(msg)
	}

	if msg.ContentType != result.GetContentType() {
		return errors.Errorf("unexpected response type %v to request. expected %v", msg.ContentType, result.GetContentType())
	}

	return proto.Unmarshal(msg.Body, result)
}

func CheckForFailureResult(msg *channel.Message, err error, successType edge_ctrl_pb.ContentType) error {
	if err != nil {
		return err
	}

	if msg.ContentType == int32(edge_ctrl_pb.ContentType_ErrorType) {
		msg := string(msg.Body)
		if msg == "" {
			msg = "error state returned from controller with no message"
		}
		return errors.New(msg)
	}

	if msg.ContentType != int32(successType) {
		return errors.Errorf("unexpected response type %v to request. expected %v", msg.ContentType, successType)
	}

	return nil
}

func GetFinHeaders() map[uint8][]byte {
	return map[uint8][]byte{
		PayloadFlagsHeader: {0x1, 0, 0, 0},
	}
}