dsfx/shared/dnet/dnet.go

package dnet

import (
	"context"
	"crypto/ecdsa"
	"net"

	"koti.casa/numenor-labs/dsfx/shared/dlog"
)

// Dial ...
func Dial(
	ctx context.Context,
	identity *ecdsa.PrivateKey,
	laddr *Addr,
	raddr *Addr,
) (*Conn, error) {
	conn, err := net.DialTCP("tcp", laddr.TCPAddr(), raddr.TCPAddr())
	if err != nil {
		return nil, err
	}

	sessionKey, err := Handshake(ctx, conn, identity, raddr.PublicKey())
	if err != nil {
		return nil, err
	}

	return NewConn(conn, sessionKey, laddr.PublicKey(), raddr.PublicKey()), nil
}

// Listen ...
func Listen(
	ctx context.Context,
	identity *ecdsa.PrivateKey,
	laddr *Addr,
) (net.Listener, error) {
	tcpListener, err := net.ListenTCP("tcp", laddr.TCPAddr())
	if err != nil {
		return nil, err
	}

	logger := dlog.FromContext(ctx)

	return &Listener{logger, tcpListener, identity}, nil
}