frostfs-node/pkg/services/object/acl/eacl/v2/headers.go

package v2

import (
	"context"
	"errors"
	"fmt"

	"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/acl"
	objectV2 "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/object"
	refsV2 "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/refs"
	"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/session"
	cid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id"
	eaclSDK "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/eacl"
	objectSDK "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object"
	oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id"
	"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/user"
)

type Option func(*cfg)

type cfg struct {
	storage ObjectStorage

	msg XHeaderSource

	cnr cid.ID
	obj *oid.ID
}

type ObjectStorage interface {
	Head(context.Context, oid.Address) (*objectSDK.Object, error)
}

type Request interface {
	GetMetaHeader() *session.RequestMetaHeader
}

type Response interface {
	GetMetaHeader() *session.ResponseMetaHeader
}

type headerSource struct {
	requestHeaders []eaclSDK.Header
	objectHeaders  []eaclSDK.Header

	incompleteObjectHeaders bool
}

func NewMessageHeaderSource(os ObjectStorage, xhs XHeaderSource, cnrID cid.ID, opts ...Option) (eaclSDK.TypedHeaderSource, error) {
	cfg := &cfg{
		storage: os,
		cnr:     cnrID,
		msg:     xhs,
	}

	for i := range opts {
		opts[i](cfg)
	}

	if cfg.msg == nil {
		return nil, errors.New("message is not provided")
	}

	var res headerSource

	err := cfg.readObjectHeaders(&res)
	if err != nil {
		return nil, err
	}

	res.requestHeaders = cfg.msg.GetXHeaders()

	return res, nil
}

func (h headerSource) HeadersOfType(typ eaclSDK.FilterHeaderType) ([]eaclSDK.Header, bool) {
	switch typ {
	default:
		return nil, true
	case eaclSDK.HeaderFromRequest:
		return h.requestHeaders, true
	case eaclSDK.HeaderFromObject:
		return h.objectHeaders, !h.incompleteObjectHeaders
	}
}

type xHeader session.XHeader

func (x xHeader) Key() string {
	return (*session.XHeader)(&x).GetKey()
}

func (x xHeader) Value() string {
	return (*session.XHeader)(&x).GetValue()
}

var errMissingOID = errors.New("object ID is missing")

func (h *cfg) readObjectHeaders(dst *headerSource) error {
	switch m := h.msg.(type) {
	default:
		panic(fmt.Sprintf("unexpected message type %T", h.msg))
	case requestXHeaderSource:
		return h.readObjectHeadersFromRequestXHeaderSource(m, dst)
	case responseXHeaderSource:
		return h.readObjectHeadersResponseXHeaderSource(m, dst)
	}
}

func (h *cfg) readObjectHeadersFromRequestXHeaderSource(m requestXHeaderSource, dst *headerSource) error {
	switch req := m.req.(type) {
	case
		*objectV2.GetRequest,
		*objectV2.HeadRequest:
		if h.obj == nil {
			return errMissingOID
		}

		objHeaders, completed := h.localObjectHeaders(h.cnr, h.obj)

		dst.objectHeaders = objHeaders
		dst.incompleteObjectHeaders = !completed
	case
		*objectV2.GetRangeRequest,
		*objectV2.GetRangeHashRequest,
		*objectV2.DeleteRequest:
		if h.obj == nil {
			return errMissingOID
		}

		dst.objectHeaders = addressHeaders(h.cnr, h.obj)
	case *objectV2.PutRequest:
		if v, ok := req.GetBody().GetObjectPart().(*objectV2.PutObjectPartInit); ok {
			oV2 := new(objectV2.Object)
			oV2.SetObjectID(v.GetObjectID())
			oV2.SetHeader(v.GetHeader())

			dst.objectHeaders = headersFromObject(objectSDK.NewFromV2(oV2), h.cnr, h.obj)
		}
	case *objectV2.PutSingleRequest:
		dst.objectHeaders = headersFromObject(objectSDK.NewFromV2(req.GetBody().GetObject()), h.cnr, h.obj)
	case *objectV2.SearchRequest:
		cnrV2 := req.GetBody().GetContainerID()
		var cnr cid.ID

		if cnrV2 != nil {
			if err := cnr.ReadFromV2(*cnrV2); err != nil {
				return fmt.Errorf("can't parse container ID: %w", err)
			}
		}

		dst.objectHeaders = []eaclSDK.Header{cidHeader(cnr)}
	}
	return nil
}

func (h *cfg) readObjectHeadersResponseXHeaderSource(m responseXHeaderSource, dst *headerSource) error {
	switch resp := m.resp.(type) {
	default:
		objectHeaders, completed := h.localObjectHeaders(h.cnr, h.obj)

		dst.objectHeaders = objectHeaders
		dst.incompleteObjectHeaders = !completed
	case *objectV2.GetResponse:
		if v, ok := resp.GetBody().GetObjectPart().(*objectV2.GetObjectPartInit); ok {
			oV2 := new(objectV2.Object)
			oV2.SetObjectID(v.GetObjectID())
			oV2.SetHeader(v.GetHeader())

			dst.objectHeaders = headersFromObject(objectSDK.NewFromV2(oV2), h.cnr, h.obj)
		}
	case *objectV2.HeadResponse:
		oV2 := new(objectV2.Object)

		var hdr *objectV2.Header

		switch v := resp.GetBody().GetHeaderPart().(type) {
		case *objectV2.ShortHeader:
			hdr = new(objectV2.Header)

			var idV2 refsV2.ContainerID
			h.cnr.WriteToV2(&idV2)

			hdr.SetContainerID(&idV2)
			hdr.SetVersion(v.GetVersion())
			hdr.SetCreationEpoch(v.GetCreationEpoch())
			hdr.SetOwnerID(v.GetOwnerID())
			hdr.SetObjectType(v.GetObjectType())
			hdr.SetPayloadLength(v.GetPayloadLength())
		case *objectV2.HeaderWithSignature:
			hdr = v.GetHeader()
		}

		oV2.SetHeader(hdr)

		dst.objectHeaders = headersFromObject(objectSDK.NewFromV2(oV2), h.cnr, h.obj)
	}
	return nil
}

func (h *cfg) localObjectHeaders(cnr cid.ID, idObj *oid.ID) ([]eaclSDK.Header, bool) {
	if idObj != nil {
		var addr oid.Address
		addr.SetContainer(cnr)
		addr.SetObject(*idObj)

		obj, err := h.storage.Head(context.TODO(), addr)
		if err == nil {
			return headersFromObject(obj, cnr, idObj), true
		}
	}

	return addressHeaders(cnr, idObj), false
}

func cidHeader(idCnr cid.ID) sysObjHdr {
	return sysObjHdr{
		k: acl.FilterObjectContainerID,
		v: idCnr.EncodeToString(),
	}
}

func oidHeader(obj oid.ID) sysObjHdr {
	return sysObjHdr{
		k: acl.FilterObjectID,
		v: obj.EncodeToString(),
	}
}

func ownerIDHeader(ownerID user.ID) sysObjHdr {
	return sysObjHdr{
		k: acl.FilterObjectOwnerID,
		v: ownerID.EncodeToString(),
	}
}

func addressHeaders(cnr cid.ID, oid *oid.ID) []eaclSDK.Header {
	hh := make([]eaclSDK.Header, 0, 2)
	hh = append(hh, cidHeader(cnr))

	if oid != nil {
		hh = append(hh, oidHeader(*oid))
	}

	return hh
}