aboutsummaryrefslogtreecommitdiffstats
path: root/p2p/peer_error.go
blob: 0ff4f4b43e74d0df3771c3cb7e6a27bb7df11514 (plain) (blame)
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
package p2p

import (
    "fmt"
)

const (
    errMagicTokenMismatch = iota
    errRead
    errWrite
    errMisc
    errInvalidMsgCode
    errInvalidMsg
    errP2PVersionMismatch
    errPubkeyInvalid
    errPubkeyForbidden
    errProtocolBreach
    errPingTimeout
    errInvalidNetworkId
    errInvalidProtocolVersion
)

var errorToString = map[int]string{
    errMagicTokenMismatch:     "magic token mismatch",
    errRead:                   "read error",
    errWrite:                  "write error",
    errMisc:                   "misc error",
    errInvalidMsgCode:         "invalid message code",
    errInvalidMsg:             "invalid message",
    errP2PVersionMismatch:     "P2P Version Mismatch",
    errPubkeyInvalid:          "public key invalid",
    errPubkeyForbidden:        "public key forbidden",
    errProtocolBreach:         "protocol Breach",
    errPingTimeout:            "ping timeout",
    errInvalidNetworkId:       "invalid network id",
    errInvalidProtocolVersion: "invalid protocol version",
}

type peerError struct {
    Code    int
    message string
}

func newPeerError(code int, format string, v ...interface{}) *peerError {
    desc, ok := errorToString[code]
    if !ok {
        panic("invalid error code")
    }
    err := &peerError{code, desc}
    if format != "" {
        err.message += ": " + fmt.Sprintf(format, v...)
    }
    return err
}

func (self *peerError) Error() string {
    return self.message
}

type DiscReason byte

const (
    DiscRequested DiscReason = iota
    DiscNetworkError
    DiscProtocolError
    DiscUselessPeer
    DiscTooManyPeers
    DiscAlreadyConnected
    DiscIncompatibleVersion
    DiscInvalidIdentity
    DiscQuitting
    DiscUnexpectedIdentity
    DiscSelf
    DiscReadTimeout
    DiscSubprotocolError
)

var discReasonToString = [...]string{
    DiscRequested:           "Disconnect requested",
    DiscNetworkError:        "Network error",
    DiscProtocolError:       "Breach of protocol",
    DiscUselessPeer:         "Useless peer",
    DiscTooManyPeers:        "Too many peers",
    DiscAlreadyConnected:    "Already connected",
    DiscIncompatibleVersion: "Incompatible P2P protocol version",
    DiscInvalidIdentity:     "Invalid node identity",
    DiscQuitting:            "Client quitting",
    DiscUnexpectedIdentity:  "Unexpected identity",
    DiscSelf:                "Connected to self",
    DiscReadTimeout:         "Read timeout",
    DiscSubprotocolError:    "Subprotocol error",
}

func (d DiscReason) String() string {
    if len(discReasonToString) < int(d) {
        return fmt.Sprintf("Unknown Reason(%d)", d)
    }
    return discReasonToString[d]
}

type discRequestedError DiscReason

func (err discRequestedError) Error() string {
    return fmt.Sprintf("disconnect requested: %v", DiscReason(err))
}

func discReasonForError(err error) DiscReason {
    if reason, ok := err.(discRequestedError); ok {
        return DiscReason(reason)
    }
    peerError, ok := err.(*peerError)
    if !ok {
        return DiscSubprotocolError
    }
    switch peerError.Code {
    case errP2PVersionMismatch:
        return DiscIncompatibleVersion
    case errPubkeyInvalid:
        return DiscInvalidIdentity
    case errPubkeyForbidden:
        return DiscUselessPeer
    case errInvalidMsgCode, errMagicTokenMismatch, errProtocolBreach:
        return DiscProtocolError
    case errPingTimeout:
        return DiscReadTimeout
    case errRead, errWrite:
        return DiscNetworkError
    default:
        return DiscSubprotocolError
    }
}