Skip to content

Commit

Permalink
restrict the Muxer option to already constructed network.Multiplexers
Browse files Browse the repository at this point in the history
  • Loading branch information
marten-seemann committed Nov 8, 2022
1 parent 24948a1 commit a647731
Show file tree
Hide file tree
Showing 8 changed files with 20 additions and 593 deletions.
8 changes: 4 additions & 4 deletions config/config.go
Original file line number Diff line number Diff line change
Expand Up @@ -73,7 +73,7 @@ type Config struct {
PeerKey crypto.PrivKey

Transports []fx.Option
Muxers []MsMuxC
Muxers []Muxer
SecurityTransports []fx.Option
Insecure bool
PSK pnet.PSK
Expand Down Expand Up @@ -171,8 +171,8 @@ func (cfg *Config) addTransports(h host.Host) error {
}

muxers := make([]protocol.ID, 0, len(cfg.Muxers))
for _, muxc := range cfg.Muxers {
muxers = append(muxers, protocol.ID(muxc.ID))
for _, m := range cfg.Muxers {
muxers = append(muxers, m.ID)
}

var security []fx.Option
Expand All @@ -181,7 +181,7 @@ func (cfg *Config) addTransports(h host.Host) error {
} else {
security = cfg.SecurityTransports
}
muxer, err := makeMuxer(h, cfg.Muxers)
muxer, err := makeMuxer(cfg.Muxers)
if err != nil {
return err
}
Expand Down
88 changes: 0 additions & 88 deletions config/constructor_types.go

This file was deleted.

58 changes: 12 additions & 46 deletions config/muxer.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,61 +3,27 @@ package config
import (
"fmt"

"github.com/libp2p/go-libp2p/core/host"
"github.com/libp2p/go-libp2p/core/network"
"github.com/libp2p/go-libp2p/core/protocol"
msmux "github.com/libp2p/go-libp2p/p2p/muxer/muxer-multistream"
)

// MuxC is a stream multiplex transport constructor.
type MuxC func(h host.Host) (network.Multiplexer, error)

// MsMuxC is a tuple containing a multiplex transport constructor and a protocol
// ID.
type MsMuxC struct {
MuxC
ID string
type Muxer struct {
ID protocol.ID
Multiplexer network.Multiplexer
}

var muxArgTypes = newArgTypeSet(hostType, networkType, peerIDType, pstoreType)

// MuxerConstructor creates a multiplex constructor from the passed parameter
// using reflection.
func MuxerConstructor(m interface{}) (MuxC, error) {
// Already constructed?
if t, ok := m.(network.Multiplexer); ok {
return func(_ host.Host) (network.Multiplexer, error) {
return t, nil
}, nil
}

ctor, err := makeConstructor(m, muxType, muxArgTypes)
if err != nil {
return nil, err
}
return func(h host.Host) (network.Multiplexer, error) {
t, err := ctor(h, nil, nil, nil, nil, nil, nil)
if err != nil {
return nil, err
}
return t.(network.Multiplexer), nil
}, nil
}

func makeMuxer(h host.Host, tpts []MsMuxC) (network.Multiplexer, error) {
func makeMuxer(muxers []Muxer) (network.Multiplexer, error) {
muxMuxer := msmux.NewBlankTransport()
transportSet := make(map[string]struct{}, len(tpts))
for _, tptC := range tpts {
if _, ok := transportSet[tptC.ID]; ok {
return nil, fmt.Errorf("duplicate muxer transport: %s", tptC.ID)
transportSet := make(map[protocol.ID]struct{}, len(muxers))
for _, m := range muxers {
if _, ok := transportSet[m.ID]; ok {
return nil, fmt.Errorf("duplicate muxer transport: %s", m.ID)
}
transportSet[tptC.ID] = struct{}{}
transportSet[m.ID] = struct{}{}
}
for _, tptC := range tpts {
tpt, err := tptC.MuxC(h)
if err != nil {
return nil, err
}
muxMuxer.AddTransport(tptC.ID, tpt)
for _, m := range muxers {
muxMuxer.AddTransport(string(m.ID), m.Multiplexer)
}
return muxMuxer, nil
}
102 changes: 0 additions & 102 deletions config/muxer_test.go

This file was deleted.

Loading

0 comments on commit a647731

Please sign in to comment.