From cfcc8933e4169c771ec7d53a223bfb5df146c8f4 Mon Sep 17 00:00:00 2001
From: Denis Kirillov <denis@nspcc.ru>
Date: Wed, 9 Mar 2022 16:15:17 +0300
Subject: [PATCH] [#195] Add response lock headers

Signed-off-by: Denis Kirillov <denis@nspcc.ru>
---
 api/handler/get.go  | 13 ++++++++++-
 api/handler/head.go | 54 ++++++++++++++++++++++++++++++++++++++++++++-
 2 files changed, 65 insertions(+), 2 deletions(-)

diff --git a/api/handler/get.go b/api/handler/get.go
index 5452504f4..eb70f76a1 100644
--- a/api/handler/get.go
+++ b/api/handler/get.go
@@ -115,7 +115,12 @@ func (h *handler) GetObjectHandler(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	if err = h.checkBucketOwner(r, reqInfo.BucketName); err != nil {
+	bktInfo, err := h.obj.GetBucketInfo(r.Context(), reqInfo.BucketName)
+	if err != nil {
+		h.logAndSendError(w, "could not get bucket info", reqInfo, err)
+		return
+	}
+	if err = checkOwner(bktInfo, r.Header.Get(api.AmzExpectedBucketOwner)); err != nil {
 		h.logAndSendError(w, "expected owner doesn't match", reqInfo, err)
 		return
 	}
@@ -150,6 +155,12 @@ func (h *handler) GetObjectHandler(w http.ResponseWriter, r *http.Request) {
 	if layer.IsAuthenticatedRequest(r.Context()) {
 		overrideResponseHeaders(w.Header(), reqInfo.URL.Query())
 	}
+
+	if err = h.setLockingHeaders(r.Context(), bktInfo, info, w.Header()); err != nil {
+		h.logAndSendError(w, "could not get locking info", reqInfo, err)
+		return
+	}
+
 	writeHeaders(w.Header(), info, len(tagSet))
 	if params != nil {
 		writeRangeHeaders(w, params, info.Size)
diff --git a/api/handler/head.go b/api/handler/head.go
index 2f9c662b6..43cac716f 100644
--- a/api/handler/head.go
+++ b/api/handler/head.go
@@ -2,6 +2,7 @@ package handler
 
 import (
 	"bytes"
+	"context"
 	"net/http"
 
 	"github.com/nspcc-dev/neofs-s3-gw/api"
@@ -33,7 +34,12 @@ func (h *handler) HeadObjectHandler(w http.ResponseWriter, r *http.Request) {
 		reqInfo = api.GetReqInfo(r.Context())
 	)
 
-	if err = h.checkBucketOwner(r, reqInfo.BucketName); err != nil {
+	bktInfo, err := h.obj.GetBucketInfo(r.Context(), reqInfo.BucketName)
+	if err != nil {
+		h.logAndSendError(w, "could not get bucket info", reqInfo, err)
+		return
+	}
+	if err = checkOwner(bktInfo, r.Header.Get(api.AmzExpectedBucketOwner)); err != nil {
 		h.logAndSendError(w, "expected owner doesn't match", reqInfo, err)
 		return
 	}
@@ -69,6 +75,11 @@ func (h *handler) HeadObjectHandler(w http.ResponseWriter, r *http.Request) {
 		info.ContentType = http.DetectContentType(buffer.Bytes())
 	}
 
+	if err = h.setLockingHeaders(r.Context(), bktInfo, info, w.Header()); err != nil {
+		h.logAndSendError(w, "could not get locking info", reqInfo, err)
+		return
+	}
+
 	writeHeaders(w.Header(), info, len(tagSet))
 	w.WriteHeader(http.StatusOK)
 }
@@ -89,3 +100,44 @@ func (h *handler) HeadBucketHandler(w http.ResponseWriter, r *http.Request) {
 	w.Header().Set(api.ContainerID, bktInfo.CID.String())
 	api.WriteResponse(w, http.StatusOK, nil, api.MimeNone)
 }
+
+func (h *handler) setLockingHeaders(ctx context.Context, bktInfo *data.BucketInfo, objInfo *data.ObjectInfo, header http.Header) error {
+	if !bktInfo.ObjectLockEnabled {
+		return nil
+	}
+
+	legalHold := &data.LegalHold{Status: legalHoldOff}
+	legalHoldInfo, err := h.obj.HeadSystemObject(ctx, bktInfo, objInfo.LegalHoldObject())
+	if err != nil && !errors.IsS3Error(err, errors.ErrNoSuchKey) {
+		return err
+	}
+
+	if legalHoldInfo != nil {
+		legalHold.Status = legalHoldOn
+	}
+
+	retention := &data.Retention{Mode: governanceMode}
+	retentionInfo, err := h.obj.HeadSystemObject(ctx, bktInfo, objInfo.RetentionObject())
+	if err != nil && !errors.IsS3Error(err, errors.ErrNoSuchKey) {
+		return err
+	}
+
+	if retentionInfo != nil {
+		retention.RetainUntilDate = retentionInfo.Headers[layer.AttributeRetainUntil]
+		if retentionInfo.Headers[layer.AttributeComplianceMode] != "" {
+			retention.Mode = complianceMode
+		}
+	}
+
+	writeLockHeaders(header, legalHold, retention)
+	return nil
+}
+
+func writeLockHeaders(h http.Header, legalHold *data.LegalHold, retention *data.Retention) {
+	h.Set(api.AmzObjectLockLegalHold, legalHold.Status)
+
+	if retention.RetainUntilDate != "" {
+		h.Set(api.AmzObjectLockRetainUntilDate, retention.RetainUntilDate)
+		h.Set(api.AmzObjectLockMode, retention.Mode)
+	}
+}