mirrors/dendrite
1
0
Fork 0
mirror of https://github.com/matrix-org/dendrite.git synced 2025-12-20 05:13:11 -06:00
Code Issues Packages Projects Releases Wiki Activity

Determinate yamux roles

Browse source
This commit is contained in:
Neil Alexander 2020-06-10 09:32:25 +01:00
parent 51dfb3aaf9
commit 450f4d17b7
1 changed files with 13 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
cmd/dendrite-demo-yggdrasil/yggconn/session.go
View file

@ -3,6 +3,7 @@ package yggconn
import (
"context"
"net"
"strings"
"time"
"github.com/libp2p/go-yamux"
@ -24,9 +25,13 @@ func (n *Node) listenFromYgg() {
n.log.Println("n.listener.Accept:", err)
return
}
session, err := yamux.Server(conn, n.yamuxConfig())
var session *yamux.Session
if strings.Compare(n.EncryptionPublicKey(), conn.RemoteAddr().String()) < 0 {
session, err = yamux.Client(conn, n.yamuxConfig())
} else {
session, err = yamux.Server(conn, n.yamuxConfig())
}
if err != nil {
n.log.Println("yamux.Server:", err)
return
}
go n.listenFromYggConn(session)
@ -77,15 +82,19 @@ func (n *Node) DialContext(ctx context.Context, network, address string) (net.Co
n.log.Println("n.dialer.DialContext:", err)
return nil, err
}
session, err = yamux.Client(conn, n.yamuxConfig())
if strings.Compare(n.EncryptionPublicKey(), address) < 0 {
session, err = yamux.Client(conn, n.yamuxConfig())
} else {
session, err = yamux.Server(conn, n.yamuxConfig())
}
if err != nil {
n.log.Println("yamux.Client.AcceptStream:", err)
return nil, err
}
go n.listenFromYggConn(session)
}
st, err := session.OpenStream()
if err != nil {
n.log.Println("session.OpenStream:", err)
return nil, err
}
return st, nil