github.com/minio/mc@v0.0.0-20240503112107-b471de8d1882/cmd/client-admin_test.go (about)

     1  // Copyright (c) 2015-2023 MinIO, Inc.
     2  //
     3  // # This file is part of MinIO Object Storage stack
     4  //
     5  // This program is free software: you can redistribute it and/or modify
     6  // it under the terms of the GNU Affero General Public License as published by
     7  // the Free Software Foundation, either version 3 of the License, or
     8  // (at your option) any later version.
     9  //
    10  // This program is distributed in the hope that it will be useful
    11  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    12  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    13  // GNU Affero General Public License for more details.
    14  //
    15  // You should have received a copy of the GNU Affero General Public License
    16  // along with this program.  If not, see <http://www.gnu.org/licenses/>.
    17  
    18  package cmd
    19  
    20  import (
    21  	"bytes"
    22  	"io"
    23  	"net/http"
    24  	"strconv"
    25  )
    26  
    27  type adminPolicyHandler struct {
    28  	endpoint string
    29  	name     string
    30  	policy   []byte
    31  }
    32  
    33  func (h adminPolicyHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    34  	if ak := r.Header.Get("Authorization"); len(ak) == 0 {
    35  		w.WriteHeader(http.StatusForbidden)
    36  		return
    37  	}
    38  	switch {
    39  	case r.Method == "PUT":
    40  		length, e := strconv.Atoi(r.Header.Get("Content-Length"))
    41  		if e != nil {
    42  			w.WriteHeader(http.StatusBadRequest)
    43  			return
    44  		}
    45  
    46  		var buffer bytes.Buffer
    47  		if _, e = io.CopyN(&buffer, r.Body, int64(length)); e != nil {
    48  			w.WriteHeader(http.StatusInternalServerError)
    49  			return
    50  		}
    51  
    52  		if len(h.policy) != buffer.Len() {
    53  			w.WriteHeader(http.StatusBadRequest)
    54  			return
    55  		}
    56  
    57  		w.Header().Set("Content-Length", "0")
    58  		w.WriteHeader(http.StatusOK)
    59  
    60  	default:
    61  		w.WriteHeader(http.StatusForbidden)
    62  	}
    63  }