/* * MinIO Cloud Storage, (C) 2017 MinIO, Inc. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package cmd import ( "bytes" "fmt" "io" "io/ioutil" "net/http" "reflect" "runtime" "strconv" "strings" "time" xnet "github.com/minio/minio/pkg/net" trace "github.com/minio/minio/pkg/trace" ) // recordRequest - records the first recLen bytes // of a given io.Reader type recordRequest struct { // Data source to record io.Reader // Response body should be logged logBody bool // Internal recording buffer buf bytes.Buffer } func (r *recordRequest) Read(p []byte) (n int, err error) { n, err = r.Reader.Read(p) if r.logBody { r.buf.Write(p[:n]) } if err != nil { return n, err } return n, err } // Return the bytes that were recorded. func (r *recordRequest) Data() []byte { return r.buf.Bytes() } // recordResponseWriter - records the first recLen bytes // of a given http.ResponseWriter type recordResponseWriter struct { // Data source to record http.ResponseWriter // Response body should be logged logBody bool // Internal recording buffer headers bytes.Buffer body bytes.Buffer // The status code of the current HTTP request statusCode int // Indicate if headers are written in the log headersLogged bool } // Write the headers into the given buffer func writeHeaders(w io.Writer, statusCode int, headers http.Header) { fmt.Fprintf(w, "%d %s\n", statusCode, http.StatusText(statusCode)) for k, v := range headers { fmt.Fprintf(w, "%s: %s\n", k, v[0]) } } // Record the headers. func (r *recordResponseWriter) WriteHeader(i int) { r.statusCode = i if !r.headersLogged { writeHeaders(&r.headers, i, r.ResponseWriter.Header()) r.headersLogged = true } r.ResponseWriter.WriteHeader(i) } func (r *recordResponseWriter) Write(p []byte) (n int, err error) { n, err = r.ResponseWriter.Write(p) if !r.headersLogged { // We assume the response code to be '200 OK' when WriteHeader() is not called, // that way following Golang HTTP response behavior. writeHeaders(&r.headers, http.StatusOK, r.ResponseWriter.Header()) r.headersLogged = true } if (r.statusCode != http.StatusOK && r.statusCode != http.StatusPartialContent && r.statusCode != 0) || r.logBody { // Always logging error responses. r.body.Write(p) } return n, err } // Calls the underlying Flush. func (r *recordResponseWriter) Flush() { r.ResponseWriter.(http.Flusher).Flush() } // Return response body. func (r *recordResponseWriter) Body() []byte { return r.body.Bytes() } // Trace gets trace of http request func Trace(f http.HandlerFunc, logBody bool, w http.ResponseWriter, r *http.Request) trace.Info { name := runtime.FuncForPC(reflect.ValueOf(f).Pointer()).Name() name = strings.TrimPrefix(name, "github.com/minio/minio/cmd.") name = strings.TrimSuffix(name, "Handler-fm") bodyPlaceHolder := []byte("") var reqBodyRecorder *recordRequest t := trace.Info{FuncName: name} reqBodyRecorder = &recordRequest{Reader: r.Body, logBody: logBody} r.Body = ioutil.NopCloser(reqBodyRecorder) host, err := xnet.ParseHost(GetLocalPeer(globalEndpoints)) if err == nil { t.NodeName = host.Name } rq := trace.RequestInfo{Time: time.Now().UTC(), Method: r.Method, Path: r.URL.Path, RawQuery: r.URL.RawQuery} rq.Headers = cloneHeader(r.Header) rq.Headers.Set("Content-Length", strconv.Itoa(int(r.ContentLength))) rq.Headers.Set("Host", r.Host) for _, enc := range r.TransferEncoding { rq.Headers.Add("Transfer-Encoding", enc) } if logBody { // If body logging is disabled then we print as a placeholder // for the actual body. rq.Body = reqBodyRecorder.Data() } else { rq.Body = bodyPlaceHolder } // Setup a http response body recorder respBodyRecorder := &recordResponseWriter{ResponseWriter: w, logBody: logBody} f(respBodyRecorder, r) rs := trace.ResponseInfo{Time: time.Now().UTC()} rs.Headers = cloneHeader(respBodyRecorder.Header()) rs.StatusCode = respBodyRecorder.statusCode if rs.StatusCode == 0 { rs.StatusCode = http.StatusOK } bodyContents := respBodyRecorder.Body() if bodyContents != nil { rs.Body = bodyContents } if !logBody { // If there was no error response and body logging is disabled // then we print as a placeholder for the actual body. rs.Body = bodyPlaceHolder } t.ReqInfo = rq t.RespInfo = rs return t }