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
|
// Copyright (C) 2019-2023 Algorand, Inc.
// This file is part of go-algorand
//
// go-algorand is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as
// published by the Free Software Foundation, either version 3 of the
// License, or (at your option) any later version.
//
// go-algorand is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with go-algorand. If not, see <https://www.gnu.org/licenses/>.
package algod
import (
"bytes"
"fmt"
"os"
"runtime"
"sync"
"github.com/algorand/go-deadlock"
"github.com/algorand/go-algorand/logging"
)
type deadlockLogger struct {
logging.Logger
*bytes.Buffer
bufferSync chan struct{}
panic func()
reportDeadlock sync.Once
}
// Panic is defined here just so we can emulate the usage of the deadlockLogger
func (logger *deadlockLogger) Panic() {
logger.Logger.Panic("potential deadlock detected")
}
// Write implements the io.Writer interface, ensuring that the write is syncronized.
func (logger *deadlockLogger) Write(p []byte) (n int, err error) {
logger.bufferSync <- struct{}{}
n, err = logger.Buffer.Write(p)
<-logger.bufferSync
return
}
// captureCallstack captures the callstack and return a byte array of the output.
func captureCallstack() []byte {
// Capture all goroutine stacks
var buf []byte
bufferSize := 256 * 1024
for {
buf = make([]byte, bufferSize)
if writtenBytes := runtime.Stack(buf, true); writtenBytes < bufferSize {
buf = buf[:writtenBytes]
break
}
bufferSize *= 2
}
return buf
}
// onPotentialDeadlock is the handler to be used by the deadlock library.
func (logger *deadlockLogger) onPotentialDeadlock() {
// The deadlock reporting is done only once; this would prevent recursive deadlock issues.
// in practive, once we report the deadlock, we panic and abort anyway, so it won't be an issue.
logger.reportDeadlock.Do(func() {
// Capture all goroutine stacks
buf := captureCallstack()
logger.bufferSync <- struct{}{}
loggedString := logger.String()
<-logger.bufferSync
fmt.Fprintln(os.Stderr, string(buf))
// logging the logged string to the logger has to happen in a separate go-routine, since the
// logger itself ( for instance, the CyclicLogWriter ) is using a mutex of it's own.
go func() {
logger.Error(loggedString)
logger.panic()
}()
})
}
func setupDeadlockLogger() *deadlockLogger {
logger := &deadlockLogger{
Logger: logging.Base(),
Buffer: bytes.NewBuffer(make([]byte, 0)),
bufferSync: make(chan struct{}, 1),
}
logger.panic = logger.Panic
deadlock.Opts.LogBuf = logger
deadlock.Opts.OnPotentialDeadlock = logger.onPotentialDeadlock
return logger
}
|