Compare commits

...

3 Commits

Author SHA1 Message Date
Derrick Hammer fadbd7c37e
fix: use RUnlock 2024-03-11 17:48:50 -04:00
Derrick Hammer 605b6a6a09
fix: add lock to GetChallenge 2024-03-11 17:48:27 -04:00
Derrick Hammer 7173abb54f
fix: use Lock 2024-03-11 17:47:56 -04:00
1 changed files with 5 additions and 3 deletions

View File

@ -130,7 +130,7 @@ func (p *WebSocketPeer) EndForAbuse() error {
return nil
}
func (p *WebSocketPeer) SetId(id *encoding.NodeId) {
p.BasePeer.lock.RLock()
p.BasePeer.lock.Lock()
defer p.BasePeer.lock.Unlock()
p.id = id
}
@ -142,12 +142,14 @@ func (p *WebSocketPeer) SetChallenge(challenge []byte) {
}
func (p *WebSocketPeer) GetChallenge() []byte {
p.BasePeer.lock.RLock()
defer p.BasePeer.lock.RUnlock()
return p.challenge
}
func (p *WebSocketPeer) GetIP() net.Addr {
p.BasePeer.lock.RLock()
defer p.BasePeer.lock.Unlock()
defer p.BasePeer.lock.RUnlock()
if p.ip != nil {
return p.ip
}
@ -174,6 +176,6 @@ func (b *WebSocketPeer) GetIPString() string {
func (p *WebSocketPeer) Abuser() bool {
p.BasePeer.lock.RLock()
defer p.BasePeer.lock.Unlock()
defer p.BasePeer.lock.RUnlock()
return p.abuser
}