mirror of
				https://github.com/yggdrasil-network/yggdrasil-go.git
				synced 2025-11-04 11:15:07 +03:00 
			
		
		
		
	work-in-progress on buffering overhaul
This commit is contained in:
		
							parent
							
								
									dc128121e5
								
							
						
					
					
						commit
						052de98f12
					
				
					 5 changed files with 202 additions and 111 deletions
				
			
		| 
						 | 
				
			
			@ -47,7 +47,7 @@ type linkInterfaceMsgIO interface {
 | 
			
		|||
}
 | 
			
		||||
 | 
			
		||||
type linkInterface struct {
 | 
			
		||||
	name           string
 | 
			
		||||
	lname          string
 | 
			
		||||
	link           *link
 | 
			
		||||
	peer           *peer
 | 
			
		||||
	msgIO          linkInterfaceMsgIO
 | 
			
		||||
| 
						 | 
				
			
			@ -125,7 +125,7 @@ func (l *link) listen(uri string) error {
 | 
			
		|||
func (l *link) create(msgIO linkInterfaceMsgIO, name, linkType, local, remote string, incoming, force bool) (*linkInterface, error) {
 | 
			
		||||
	// Technically anything unique would work for names, but let's pick something human readable, just for debugging
 | 
			
		||||
	intf := linkInterface{
 | 
			
		||||
		name:  name,
 | 
			
		||||
		lname: name,
 | 
			
		||||
		link:  l,
 | 
			
		||||
		msgIO: msgIO,
 | 
			
		||||
		info: linkInfo{
 | 
			
		||||
| 
						 | 
				
			
			@ -178,7 +178,7 @@ func (intf *linkInterface) handler() error {
 | 
			
		|||
	}
 | 
			
		||||
	base := version_getBaseMetadata()
 | 
			
		||||
	if meta.ver > base.ver || meta.ver == base.ver && meta.minorVer > base.minorVer {
 | 
			
		||||
		intf.link.core.log.Errorln("Failed to connect to node: " + intf.name + " version: " + fmt.Sprintf("%d.%d", meta.ver, meta.minorVer))
 | 
			
		||||
		intf.link.core.log.Errorln("Failed to connect to node: " + intf.lname + " version: " + fmt.Sprintf("%d.%d", meta.ver, meta.minorVer))
 | 
			
		||||
		return errors.New("failed to connect: wrong version")
 | 
			
		||||
	}
 | 
			
		||||
	// Check if we're authorized to connect to this key / IP
 | 
			
		||||
| 
						 | 
				
			
			@ -217,23 +217,9 @@ func (intf *linkInterface) handler() error {
 | 
			
		|||
	intf.link.mutex.Unlock()
 | 
			
		||||
	// Create peer
 | 
			
		||||
	shared := crypto.GetSharedKey(myLinkPriv, &meta.link)
 | 
			
		||||
	out := func(msgs [][]byte) {
 | 
			
		||||
		// nil to prevent it from blocking if the link is somehow frozen
 | 
			
		||||
		// this is safe because another packet won't be sent until the link notifies
 | 
			
		||||
		//  the peer that it's ready for one
 | 
			
		||||
		intf.writer.sendFrom(nil, msgs, false)
 | 
			
		||||
	}
 | 
			
		||||
	linkOut := func(bs []byte) {
 | 
			
		||||
		// nil to prevent it from blocking if the link is somehow frozen
 | 
			
		||||
		// FIXME this is hypothetically not safe, the peer shouldn't be sending
 | 
			
		||||
		//  additional packets until this one finishes, otherwise this could leak
 | 
			
		||||
		//  memory if writing happens slower than link packets are generated...
 | 
			
		||||
		//  that seems unlikely, so it's a lesser evil than deadlocking for now
 | 
			
		||||
		intf.writer.sendFrom(nil, [][]byte{bs}, true)
 | 
			
		||||
	}
 | 
			
		||||
	phony.Block(&intf.link.core.peers, func() {
 | 
			
		||||
		// FIXME don't use phony.Block, it's bad practice, even if it's safe here
 | 
			
		||||
		intf.peer = intf.link.core.peers._newPeer(&meta.box, &meta.sig, shared, intf, func() { intf.msgIO.close() }, out, linkOut)
 | 
			
		||||
		intf.peer = intf.link.core.peers._newPeer(&meta.box, &meta.sig, shared, intf)
 | 
			
		||||
	})
 | 
			
		||||
	if intf.peer == nil {
 | 
			
		||||
		return errors.New("failed to create peer")
 | 
			
		||||
| 
						 | 
				
			
			@ -275,6 +261,58 @@ func (intf *linkInterface) handler() error {
 | 
			
		|||
 | 
			
		||||
////////////////////////////////////////////////////////////////////////////////
 | 
			
		||||
 | 
			
		||||
// linkInterface needs to match the peerInterface type needed by the peers
 | 
			
		||||
 | 
			
		||||
func (intf *linkInterface) out(bss [][]byte) {
 | 
			
		||||
	intf.Act(nil, func() {
 | 
			
		||||
		// nil to prevent it from blocking if the link is somehow frozen
 | 
			
		||||
		// this is safe because another packet won't be sent until the link notifies
 | 
			
		||||
		//  the peer that it's ready for one
 | 
			
		||||
		intf.writer.sendFrom(nil, bss, false)
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (intf *linkInterface) linkOut(bs []byte) {
 | 
			
		||||
	intf.Act(nil, func() {
 | 
			
		||||
		// nil to prevent it from blocking if the link is somehow frozen
 | 
			
		||||
		// FIXME this is hypothetically not safe, the peer shouldn't be sending
 | 
			
		||||
		//  additional packets until this one finishes, otherwise this could leak
 | 
			
		||||
		//  memory if writing happens slower than link packets are generated...
 | 
			
		||||
		//  that seems unlikely, so it's a lesser evil than deadlocking for now
 | 
			
		||||
		intf.writer.sendFrom(nil, [][]byte{bs}, true)
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (intf *linkInterface) notifyQueued(seq uint64) {
 | 
			
		||||
	// This is the part where we want non-nil 'from' fields
 | 
			
		||||
	intf.Act(intf.peer, func() {
 | 
			
		||||
		if !intf.isIdle {
 | 
			
		||||
			intf.peer.dropFromQueue(intf, seq)
 | 
			
		||||
		}
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (intf *linkInterface) close() {
 | 
			
		||||
	intf.Act(nil, func() { intf.msgIO.close() })
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (intf *linkInterface) name() string {
 | 
			
		||||
	return intf.lname
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (intf *linkInterface) local() string {
 | 
			
		||||
	return intf.info.local
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (intf *linkInterface) remote() string {
 | 
			
		||||
	return intf.info.remote
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (intf *linkInterface) interfaceType() string {
 | 
			
		||||
	return intf.info.linkType
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
////////////////////////////////////////////////////////////////////////////////
 | 
			
		||||
const (
 | 
			
		||||
	sendTime      = 1 * time.Second    // How long to wait before deciding a send is blocked
 | 
			
		||||
	keepAliveTime = 2 * time.Second    // How long to wait before sending a keep-alive response if we have no real traffic to send
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue