yggdrasil-network/yggdrasil-go
1
0
Fork 0
mirror of https://github.com/yggdrasil-network/yggdrasil-go.git synced 2025-04-30 07:05:06 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Tweaks

Browse source
This commit is contained in:
Neil Alexander 2020-07-02 13:39:38 +01:00
parent 7b0d315675
commit 326fdd4cf9
No known key found for this signature in database
GPG key ID: A02A2019A2BB0944
2 changed files with 19 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/yggdrasil/api.go
View file

@ -235,10 +235,12 @@ func (c *Core) PacketConn() *PacketConn {
// Resolve takes a masked node ID and performs a search, returning the complete
// node ID and the node's public key.
func (c *Core) Resolve(nodeID, nodeMask *crypto.NodeID) (fullNodeID *crypto.NodeID, boxPubKey *crypto.BoxPubKey, err error) {
fmt.Println("**** START RESOLVE")
defer fmt.Println("**** END RESOLVE")
done := make(chan struct{})
c.router.Act(c, func() {
_, isIn := c.router.searches.searches[*nodeID]
if !isIn {
if _, searching := c.router.searches.searches[*nodeID]; !searching {
searchCompleted := func(sinfo *sessionInfo, e error) {
select {
case <-done:
@ -257,8 +259,7 @@ func (c *Core) Resolve(nodeID, nodeMask *crypto.NodeID) (fullNodeID *crypto.Node
close(done)
}
}
sinfo := c.router.searches.newIterSearch(nodeID, nodeMask, searchCompleted)
sinfo.startSearch()
c.router.searches.newIterSearch(nodeID, nodeMask, searchCompleted).startSearch()
} else {
err = errors.New("search already exists")
close(done)

21
src/yggdrasil/packetconn.go
View file

@ -2,6 +2,7 @@ package yggdrasil
import (
"errors"
"fmt"
"net"
"time"
@ -73,27 +74,33 @@ func (c *PacketConn) WriteTo(b []byte, addr net.Addr) (int, error) {
nodeMask[i] = 0xFF
}
var err error
var session *sessionInfo
phony.Block(c.sessions.router, func() {
var ok bool
session, ok = c.sessions.getByTheirPerm(boxPubKey)
if !ok {
c.sessions.router.core.Resolve(nodeID, nodeMask)
session, _ = c.sessions.getByTheirPerm(boxPubKey)
nodeID, boxPubKey, err = c.sessions.router.core.Resolve(nodeID, nodeMask)
if err == nil {
session, _ = c.sessions.getByTheirPerm(boxPubKey)
}
}
})
if err != nil {
return 0, fmt.Errorf("failed to find session/start search: %w", err)
}
if session == nil {
return 0, errors.New("expected a session but there was none")
}
err := make(chan error, 1)
sendErr := make(chan error, 1)
msg := FlowKeyMessage{Message: b}
session.Act(c, func() {
// Check if the packet is small enough to go through this session
sessionMTU := session._getMTU()
if types.MTU(len(b)) > sessionMTU {
err <- PacketConnError{maxsize: int(sessionMTU)}
sendErr <- PacketConnError{maxsize: int(sessionMTU)}
return
}
@ -125,11 +132,11 @@ func (c *PacketConn) WriteTo(b []byte, addr net.Addr) (int, error) {
default: // Don't do anything, to keep traffic throttled
}
err <- nil
sendErr <- nil
})
e := <-err
return len(b), e
err = <-sendErr
return len(b), err
}
// implements net.PacketConn