Blame view

go/src/themaru/vendor/github.com/gin-gonic/gin/recovery.go 4.24 KB
476d2547e   김태훈   태마루 시스템 설정 / 업데이트...
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
  // Copyright 2014 Manu Martinez-Almeida.  All rights reserved.
  // Use of this source code is governed by a MIT style
  // license that can be found in the LICENSE file.
  
  package gin
  
  import (
  	"bytes"
  	"fmt"
  	"io"
  	"io/ioutil"
  	"log"
  	"net"
  	"net/http"
  	"net/http/httputil"
  	"os"
  	"runtime"
  	"strings"
  	"time"
  )
  
  var (
  	dunno     = []byte("???")
  	centerDot = []byte("·")
  	dot       = []byte(".")
  	slash     = []byte("/")
  )
  
  // Recovery returns a middleware that recovers from any panics and writes a 500 if there was one.
  func Recovery() HandlerFunc {
  	return RecoveryWithWriter(DefaultErrorWriter)
  }
  
  // RecoveryWithWriter returns a middleware for a given writer that recovers from any panics and writes a 500 if there was one.
  func RecoveryWithWriter(out io.Writer) HandlerFunc {
  	var logger *log.Logger
  	if out != nil {
  		logger = log.New(out, "
  
  \x1b[31m", log.LstdFlags)
  	}
  	return func(c *Context) {
  		defer func() {
  			if err := recover(); err != nil {
  				// Check for a broken connection, as it is not really a
  				// condition that warrants a panic stack trace.
  				var brokenPipe bool
  				if ne, ok := err.(*net.OpError); ok {
  					if se, ok := ne.Err.(*os.SyscallError); ok {
  						if strings.Contains(strings.ToLower(se.Error()), "broken pipe") || strings.Contains(strings.ToLower(se.Error()), "connection reset by peer") {
  							brokenPipe = true
  						}
  					}
  				}
  				if logger != nil {
  					stack := stack(3)
  					httpRequest, _ := httputil.DumpRequest(c.Request, false)
  					headers := strings.Split(string(httpRequest), "\r
  ")
  					for idx, header := range headers {
  						current := strings.Split(header, ":")
  						if current[0] == "Authorization" {
  							headers[idx] = current[0] + ": *"
  						}
  					}
  					if brokenPipe {
  						logger.Printf("%s
  %s%s", err, string(httpRequest), reset)
  					} else if IsDebugging() {
  						logger.Printf("[Recovery] %s panic recovered:
  %s
  %s
  %s%s",
  							timeFormat(time.Now()), strings.Join(headers, "\r
  "), err, stack, reset)
  					} else {
  						logger.Printf("[Recovery] %s panic recovered:
  %s
  %s%s",
  							timeFormat(time.Now()), err, stack, reset)
  					}
  				}
  
  				// If the connection is dead, we can't write a status to it.
  				if brokenPipe {
  					c.Error(err.(error)) // nolint: errcheck
  					c.Abort()
  				} else {
  					c.AbortWithStatus(http.StatusInternalServerError)
  				}
  			}
  		}()
  		c.Next()
  	}
  }
  
  // stack returns a nicely formatted stack frame, skipping skip frames.
  func stack(skip int) []byte {
  	buf := new(bytes.Buffer) // the returned data
  	// As we loop, we open files and read them. These variables record the currently
  	// loaded file.
  	var lines [][]byte
  	var lastFile string
  	for i := skip; ; i++ { // Skip the expected number of frames
  		pc, file, line, ok := runtime.Caller(i)
  		if !ok {
  			break
  		}
  		// Print this much at least.  If we can't find the source, it won't show.
  		fmt.Fprintf(buf, "%s:%d (0x%x)
  ", file, line, pc)
  		if file != lastFile {
  			data, err := ioutil.ReadFile(file)
  			if err != nil {
  				continue
  			}
  			lines = bytes.Split(data, []byte{'
  '})
  			lastFile = file
  		}
  		fmt.Fprintf(buf, "\t%s: %s
  ", function(pc), source(lines, line))
  	}
  	return buf.Bytes()
  }
  
  // source returns a space-trimmed slice of the n'th line.
  func source(lines [][]byte, n int) []byte {
  	n-- // in stack trace, lines are 1-indexed but our array is 0-indexed
  	if n < 0 || n >= len(lines) {
  		return dunno
  	}
  	return bytes.TrimSpace(lines[n])
  }
  
  // function returns, if possible, the name of the function containing the PC.
  func function(pc uintptr) []byte {
  	fn := runtime.FuncForPC(pc)
  	if fn == nil {
  		return dunno
  	}
  	name := []byte(fn.Name())
  	// The name includes the path name to the package, which is unnecessary
  	// since the file name is already included.  Plus, it has center dots.
  	// That is, we see
  	//	runtime/debug.*T·ptrmethod
  	// and want
  	//	*T.ptrmethod
  	// Also the package path might contains dot (e.g. code.google.com/...),
  	// so first eliminate the path prefix
  	if lastSlash := bytes.LastIndex(name, slash); lastSlash >= 0 {
  		name = name[lastSlash+1:]
  	}
  	if period := bytes.Index(name, dot); period >= 0 {
  		name = name[period+1:]
  	}
  	name = bytes.Replace(name, centerDot, dot, -1)
  	return name
  }
  
  func timeFormat(t time.Time) string {
  	var timeString = t.Format("2006/01/02 - 15:04:05")
  	return timeString
  }