zishuo/p2p-forwarder
1
0
Fork 0
mirror of https://github.com/nickname76/p2p-forwarder.git synced 2024-05-03 11:04:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix

Browse Source
This commit is contained in:
nickname32 2020-12-21 18:09:30 +03:00
parent b90e22d249
commit 7253b607e5
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
protportssub.go
View File

@ -3,6 +3,7 @@ package p2pforwarder
import (
"context"
"encoding/binary"
"errors"
"fmt"
"io"
@ -130,6 +131,9 @@ func (f *Forwarder) sendPortsManifestToSubscriber(peerid peer.ID, b []byte) {
f.portsSubscribersMux.Unlock()
}
// ErrConnReset = error Connection reset
var ErrConnReset = errors.New("Connection reset")
func (f *Forwarder) sendOpenPortsManifestBytes(peerid peer.ID, b []byte) error {
s, err := f.host.NewStream(context.Background(), peerid, portssubProtID)
if err != nil {
@ -148,12 +152,17 @@ func (f *Forwarder) sendOpenPortsManifestBytes(peerid peer.ID, b []byte) error {
}
// Test, if connection have been reset or not
_, err = io.ReadFull(s, make([]byte, 1))
n, err := io.ReadFull(s, make([]byte, 1))
if err != nil {
s.Reset()
return fmt.Errorf("sendOpenPortsManifestBytes: %s", err)
}
if n == 0 {
s.Reset()
return fmt.Errorf("sendOpenPortsManifestBytes: %s", ErrConnReset)
}
s.Close()
return nil