Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,125 @@ | ||
| //go:build linux | ||
| // +build linux | ||
|
|
||
| package core | ||
|
|
||
| import ( | ||
| "context" | ||
| "net" | ||
| "net/url" | ||
| "strconv" | ||
| "strings" | ||
|
|
||
| "github.com/Arceliar/phony" | ||
| sctp "github.com/vikulin/sctp" | ||
| ) | ||
|
|
||
| type linkSCTP struct { | ||
| phony.Inbox | ||
| *links | ||
| listener *net.ListenConfig | ||
| _listeners map[*Listener]context.CancelFunc | ||
| } | ||
|
|
||
| func (l *links) newLinkSCTP() *linkSCTP { | ||
| lt := &linkSCTP{ | ||
| links: l, | ||
| listener: &net.ListenConfig{ | ||
| KeepAlive: -1, | ||
| }, | ||
| _listeners: map[*Listener]context.CancelFunc{}, | ||
| } | ||
| return lt | ||
| } | ||
|
|
||
| func (l *linkSCTP) dial(tx context.Context, url *url.URL, info linkInfo, options linkOptions) (net.Conn, error) { | ||
| host, port, err := net.SplitHostPort(url.Host) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| dst, err := net.ResolveIPAddr("ip", host) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| raddress := l.getAddress(dst.String() + ":" + port) | ||
| var conn net.Conn | ||
| laddress := l.getAddress("0.0.0.0:0") | ||
| conn, err = sctp.NewSCTPConnection(laddress, laddress.AddressFamily, sctp.InitMsg{NumOstreams: 2, MaxInstreams: 2, MaxAttempts: 2, MaxInitTimeout: 5}, sctp.OneToOne, false) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| err = conn.(*sctp.SCTPConn).Connect(raddress) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| //conn.(*sctp.SCTPConn).SetWriteBuffer(324288) | ||
| //conn.(*sctp.SCTPConn).SetReadBuffer(324288) | ||
| //wbuf, _ := conn.(*sctp.SCTPConn).GetWriteBuffer() | ||
| //rbuf, _ := conn.(*sctp.SCTPConn).GetReadBuffer() | ||
|
|
||
| //l.core.log.Printf("Read buffer %d", rbuf) | ||
| //l.core.log.Printf("Write buffer %d", wbuf) | ||
| err = conn.(*sctp.SCTPConn).SetEvents(sctp.SCTP_EVENT_DATA_IO) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| return conn, nil | ||
| } | ||
|
|
||
| func (l *linkSCTP) listen(ctx context.Context, url *url.URL, sintf string) (net.Listener, error) { | ||
| addr := l.getAddress(url.Host) | ||
| listener, err := sctp.NewSCTPListener(addr, sctp.InitMsg{NumOstreams: 2, MaxInstreams: 2, MaxAttempts: 2, MaxInitTimeout: 5}, sctp.OneToOne, false) | ||
|
|
||
| if err != nil { | ||
| //cancel() | ||
| return nil, err | ||
| } | ||
| err = listener.SetEvents(sctp.SCTP_EVENT_DATA_IO) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| return listener, nil | ||
| } | ||
|
|
||
| // Returns the address of the listener. | ||
| // | ||
| //nolint:unused | ||
| func (l *linkSCTP) getAddr() *net.TCPAddr { | ||
| // TODO: Fix this, because this will currently only give a single address | ||
| // to multicast.go, which obviously is not great, but right now multicast.go | ||
| // doesn't have the ability to send more than one address in a packet either | ||
| var addr *net.TCPAddr | ||
| phony.Block(l, func() { | ||
| for listener := range l._listeners { | ||
| addr = listener.Addr().(*net.TCPAddr) | ||
| } | ||
| }) | ||
| return addr | ||
| } | ||
|
|
||
| // SCTP infrastructure | ||
| func (l *linkSCTP) getAddress(host string) *sctp.SCTPAddr { | ||
|
|
||
| //sctp supports multihoming but current implementation reuires only one path | ||
| ips := []net.IPAddr{} | ||
| ip, port, err := net.SplitHostPort(host) | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
| for _, i := range strings.Split(ip, ",") { | ||
| if a, err := net.ResolveIPAddr("ip", i); err == nil { | ||
| l.core.log.Printf("Resolved address '%s' to %s", i, a) | ||
| ips = append(ips, *a) | ||
| } else { | ||
| l.core.log.Errorln("Error resolving address '%s': %v", i, err) | ||
| } | ||
| } | ||
| p, _ := strconv.Atoi(port) | ||
| addr := &sctp.SCTPAddr{ | ||
| IPAddrs: ips, | ||
| Port: p, | ||
| } | ||
| return addr | ||
| } |