core.go 28.2 KB
Newer Older
1 2 3 4
/*
Package core implements the IpfsNode object and related methods.

Packages underneath core/ provide a (relatively) stable, low-level API
5 6 7 8
to carry out most IPFS-related tasks.  For more details on the other
interfaces and how core/... fits into the bigger IPFS picture, see:

  $ godoc github.com/ipfs/go-ipfs
9
*/
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
10 11
package core

Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
12
import (
Jakub Sztandera's avatar
Jakub Sztandera committed
13
	"bytes"
14
	"context"
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
15
	"errors"
Juan Batiz-Benet's avatar
go fmt  
Juan Batiz-Benet committed
16
	"fmt"
17
	"io"
18 19 20
	"io/ioutil"
	"os"
	"strings"
21
	"time"
22

23
	version "github.com/ipfs/go-ipfs"
24
	rp "github.com/ipfs/go-ipfs/exchange/reprovide"
25
	filestore "github.com/ipfs/go-ipfs/filestore"
26 27 28
	mount "github.com/ipfs/go-ipfs/fuse/mount"
	namesys "github.com/ipfs/go-ipfs/namesys"
	ipnsrp "github.com/ipfs/go-ipfs/namesys/republisher"
Łukasz Magiera's avatar
Łukasz Magiera committed
29
	p2p "github.com/ipfs/go-ipfs/p2p"
30 31
	pin "github.com/ipfs/go-ipfs/pin"
	repo "github.com/ipfs/go-ipfs/repo"
32

Steven Allen's avatar
Steven Allen committed
33
	ma "gx/ipfs/QmNTCey11oxhb1AxDnQBRHtdhap6Ctud872NjAYPYYXPuc/go-multiaddr"
Steven Allen's avatar
Steven Allen committed
34 35 36
	ic "gx/ipfs/QmNiJiXwWE3kRhZrC5ej3kSjWHm337pYfhjLGSCDNKJP2s/go-libp2p-crypto"
	u "gx/ipfs/QmNohiVssaPw3KVLZik59DBVGTSm2dGvYT9eoXt5DQ36Yz/go-ipfs-util"
	exchange "gx/ipfs/QmP2g3VxmC7g7fyRJDj1VJ72KHZbJ9UW24YjSWEj1XTb4H/go-ipfs-exchange-interface"
Steven Allen's avatar
Steven Allen committed
37 38 39 40
	peer "gx/ipfs/QmPJxxDsX2UbchSHobbYuvz7qnyJTFKvaKMzE2rZWJ4x5B/go-libp2p-peer"
	quic "gx/ipfs/QmPxDT1mJcdVbPSGRrActszXdptSgcj9gtyMuhrPavXFCN/go-libp2p-quic-transport"
	pstore "gx/ipfs/QmQFFp4ntkd4C14sP3FaH9WJyBuetuGUVo6dShNHvnoEvC/go-libp2p-peerstore"
	circuit "gx/ipfs/QmQG8wJtY6KfsTH2tjhaThFPeYVJGm7cmRMxen73ipA4Z5/go-libp2p-circuit"
Steven Allen's avatar
Steven Allen committed
41
	mafilter "gx/ipfs/QmQgSnRC74nHoXrN9CShvfWUUSrgAMJ4unjbnuBVsxk2mw/go-maddr-filter"
Steven Allen's avatar
Steven Allen committed
42 43
	resolver "gx/ipfs/QmQiXYqcxU5AvpAJkfbXUEZgUYKog1Pd2Cv3WBiW2Hpe8M/go-path/resolver"
	merkledag "gx/ipfs/QmQvMsV5aPyd7eMd3U1hvAUhZEupG3rXbVZn7ppU5RE6bt/go-merkledag"
Steven Allen's avatar
Steven Allen committed
44
	cid "gx/ipfs/QmR8BauakNcBa3RbE4nbQu76PDiJgoQgz8AJdhJuiU4TAw/go-cid"
Steven Allen's avatar
Steven Allen committed
45
	nilrouting "gx/ipfs/QmRJvdmKJoDcQEhhTt5NYXJPQFnJYPo1kfapxtjZLfDDqH/go-ipfs-routing/none"
46
	ipld "gx/ipfs/QmRL22E4paat7ky7vx9MLpR97JHHbFPrg3ytFQw6qp1y1s/go-ipld-format"
Steven Allen's avatar
Steven Allen committed
47 48
	metrics "gx/ipfs/QmRN8cMBqfgLgrcaeK6vqUd7HuyvKbNnzaSj4TRBW9XvPQ/go-libp2p-metrics"
	routing "gx/ipfs/QmRjT8Bkut84fHf9nxMQBxGsqLAkqzMdFaemDK7e61dBNZ/go-libp2p-routing"
Steven Allen's avatar
Steven Allen committed
49
	autonat "gx/ipfs/QmRxFskwAFNMPNy9JLGYWm4HmVXAytDPYePWcJPXnjAJ99/go-libp2p-autonat-svc"
50
	bstore "gx/ipfs/QmS2aqUZLJp8kF1ihE5rvDGE5LvmKDPnx32w9Z1BW9xLV5/go-ipfs-blockstore"
Steven Allen's avatar
Steven Allen committed
51 52
	dht "gx/ipfs/QmS5Tvk8Adz1qPkCBCbiScty9KPbMSMCSTbFK4TVvatKqi/go-libp2p-kad-dht"
	dhtopts "gx/ipfs/QmS5Tvk8Adz1qPkCBCbiScty9KPbMSMCSTbFK4TVvatKqi/go-libp2p-kad-dht/opts"
53
	goprocess "gx/ipfs/QmSF8fPo3jgVBAy8fpdjjYqgG87dkJgUprRBHRd2tmfgpP/goprocess"
Jeromy's avatar
Jeromy committed
54
	mamask "gx/ipfs/QmSMZwvs3n4GBikZ7hKzT17c3bk65FmyZo2JqtJ16swqCv/multiaddr-filter"
Steven Allen's avatar
Steven Allen committed
55
	ft "gx/ipfs/QmSygPSC63Uka8z9PYokAS4thiMAor17vhXUTi4qmKHh6P/go-unixfs"
Steven Allen's avatar
Steven Allen committed
56 57
	config "gx/ipfs/QmTbcMKv6GU3fxhnNcbzYChdox9Fdd7VpucM3PQ7UWjX3D/go-ipfs-config"
	pubsub "gx/ipfs/QmWL6MKfes1HuSiRUNzGmwy9YyQDwcZF9V1NaA2keYKhtE/go-libp2p-pubsub"
Steven Allen's avatar
Steven Allen committed
58
	pnet "gx/ipfs/QmY4Q5JC4vxLEi8EpVxJM4rcRryEVtH1zRKVTAm6BKV1pg/go-libp2p-pnet"
Steven Allen's avatar
Steven Allen committed
59
	smux "gx/ipfs/QmY9JXR3FupnYAYJWK9aMr9bCpqWKcToQ1tz8DVGTrHpHw/go-stream-muxer"
Steven Allen's avatar
Steven Allen committed
60
	mplex "gx/ipfs/QmZsejKNkeFSQe5TcmYXJ8iq6qPL1FpsP4eAA8j7RfE7xg/go-smux-multiplex"
Steven Allen's avatar
Steven Allen committed
61 62
	bserv "gx/ipfs/QmZuPasxd7fSgtzRzCL7Z8J8QwDJML2fgBUExRbQCqb4BT/go-blockservice"
	mfs "gx/ipfs/QmavpuPmHfsY4JFTQFZZQJi8NfFRPPwLnxqdcgP3zTR4JV/go-mfs"
Steven Allen's avatar
Steven Allen committed
63 64
	connmgr "gx/ipfs/Qmb5KqwKh3iqcf91oLunTUXfV9PotzvCAdyrahhPq1uZyy/go-libp2p-connmgr"
	psrouter "gx/ipfs/QmcRQsJW4A5CSkQptLUEEEV5Rp3boMAPfKS9z3xrQTYvzT/go-libp2p-pubsub-router"
Steven Allen's avatar
Steven Allen committed
65
	logging "gx/ipfs/QmcuXC5cxs79ro2cUuHs4HQ2bkDLJUYokwL8aivcX6HW3C/go-log"
Steven Allen's avatar
Steven Allen committed
66
	yamux "gx/ipfs/Qmdps3CYh5htGQSrPvzg5PHouVexLmtpbuLCqc4vuej8PC/go-smux-yamux"
Steven Allen's avatar
Steven Allen committed
67
	ifconnmgr "gx/ipfs/QmebAt96MwXHnbJ5uns6KLm3eSVLPDaaCB4DU7phQUi9a3/go-libp2p-interface-connmgr"
Steven Allen's avatar
Steven Allen committed
68 69 70 71 72
	libp2p "gx/ipfs/QmebEmt23jQxrwnqBkFL4qbpE8EnnQunpv5U32LS5ESus1/go-libp2p"
	discovery "gx/ipfs/QmebEmt23jQxrwnqBkFL4qbpE8EnnQunpv5U32LS5ESus1/go-libp2p/p2p/discovery"
	p2pbhost "gx/ipfs/QmebEmt23jQxrwnqBkFL4qbpE8EnnQunpv5U32LS5ESus1/go-libp2p/p2p/host/basic"
	rhost "gx/ipfs/QmebEmt23jQxrwnqBkFL4qbpE8EnnQunpv5U32LS5ESus1/go-libp2p/p2p/host/routed"
	identify "gx/ipfs/QmebEmt23jQxrwnqBkFL4qbpE8EnnQunpv5U32LS5ESus1/go-libp2p/p2p/protocol/identify"
Steven Allen's avatar
Steven Allen committed
73 74
	record "gx/ipfs/QmexPd3srWxHC76gW2p5j5tQvwpPuCoW7b9vFhJ8BRPyh9/go-libp2p-record"
	rhelpers "gx/ipfs/Qmf2tThowTQbtNYtAohVDoY1oBxHgWxkVhauz1PJbVYZge/go-libp2p-routing-helpers"
75
	ds "gx/ipfs/Qmf4xQhNomPNhrtZc67qSnfJSjxjXs9LWvknJtSXwimPrM/go-datastore"
Steven Allen's avatar
Steven Allen committed
76 77
	bitswap "gx/ipfs/Qmf9HEDC317JbdByx7jUzDzcG8HtfEPYDaf8QRmovweuLT/go-bitswap"
	bsnet "gx/ipfs/Qmf9HEDC317JbdByx7jUzDzcG8HtfEPYDaf8QRmovweuLT/go-bitswap/network"
Steven Allen's avatar
Steven Allen committed
78
	p2phost "gx/ipfs/QmfRHxh8bt4jWLKRhNvR5fn7mFACrQBFLqV4wyoymEExKV/go-libp2p-host"
Łukasz Magiera's avatar
Łukasz Magiera committed
79
)
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
80

Jeromy's avatar
Jeromy committed
81
const IpnsValidatorTag = "ipns"
82

83
const kReprovideFrequency = time.Hour * 12
84
const discoveryConnTimeout = time.Second * 30
85
const DefaultIpnsCacheSize = 128
Jeromy's avatar
Jeromy committed
86

Jeromy's avatar
Jeromy committed
87
var log = logging.Logger("core")
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
88

89 90 91 92
type mode int

const (
	// zero value is not a valid mode, must be explicitly set
93
	localMode mode = iota
94 95 96 97
	offlineMode
	onlineMode
)

98
func init() {
99
	identify.ClientVersion = "go-ipfs/" + version.CurrentVersionNumber + "/" + version.CurrentCommit
100 101
}

Juan Batiz-Benet's avatar
go lint  
Juan Batiz-Benet committed
102
// IpfsNode is IPFS Core module. It represents an IPFS instance.
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
103 104
type IpfsNode struct {

105
	// Self
106
	Identity peer.ID // the local node's identity
107

108
	Repo repo.Repo
109 110

	// Local node
111 112 113 114
	Pinning         pin.Pinner // the pinning manager
	Mounts          Mounts     // current mount state, if any.
	PrivateKey      ic.PrivKey // the local node's private Key
	PNetFingerprint []byte     // fingerprint of private network
115 116

	// Services
117 118 119 120 121 122 123 124 125 126 127 128
	Peerstore       pstore.Peerstore     // storage for other Peer instances
	Blockstore      bstore.GCBlockstore  // the block store (lower level)
	Filestore       *filestore.Filestore // the filestore blockstore
	BaseBlocks      bstore.Blockstore    // the raw blockstore, no filestore wrapping
	GCLocker        bstore.GCLocker      // the locker used to protect the blockstore during gc
	Blocks          bserv.BlockService   // the block service, get/add blocks.
	DAG             ipld.DAGService      // the merkle dag service, get/add objects.
	Resolver        *resolver.Resolver   // the path resolution system
	Reporter        metrics.Reporter
	Discovery       discovery.Service
	FilesRoot       *mfs.Root
	RecordValidator record.Validator
129 130

	// Online
131 132 133 134 135
	PeerHost     p2phost.Host        // the network host (server+client)
	Bootstrapper io.Closer           // the periodic bootstrapper
	Routing      routing.IpfsRouting // the routing system. recommend ipfs-dht
	Exchange     exchange.Interface  // the block exchange + strategy (bitswap)
	Namesys      namesys.NameSystem  // the name system, resolves paths to hashes
136
	Reprovider   *rp.Reprovider      // the value reprovider system
Jeromy's avatar
Jeromy committed
137
	IpnsRepub    *ipnsrp.Republisher
138

vyzo's avatar
vyzo committed
139
	AutoNAT  *autonat.AutoNATService
Steven Allen's avatar
Steven Allen committed
140
	PubSub   *pubsub.PubSub
141
	PSRouter *psrouter.PubsubValueStore
142
	DHT      *dht.IpfsDHT
Łukasz Magiera's avatar
Łukasz Magiera committed
143
	P2P      *p2p.P2P
Jeromy's avatar
Jeromy committed
144

145 146
	proc goprocess.Process
	ctx  context.Context
147

Jeromy's avatar
Jeromy committed
148
	mode         mode
149
	localModeSet bool
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
150 151
}

152 153 154 155 156 157 158 159
// Mounts defines what the node's mount state is. This should
// perhaps be moved to the daemon or mount. It's here because
// it needs to be accessible across daemon requests.
type Mounts struct {
	Ipfs mount.Mount
	Ipns mount.Mount
}

vyzo's avatar
vyzo committed
160
func (n *IpfsNode) startOnlineServices(ctx context.Context, routingOption RoutingOption, hostOption HostOption, do DiscoveryOption, pubsub, ipnsps, mplex bool) error {
161
	if n.PeerHost != nil { // already online.
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
162
		return errors.New("node already online")
163 164
	}

165 166 167 168
	if n.PrivateKey == nil {
		return fmt.Errorf("private key not available")
	}

Jeromy's avatar
Jeromy committed
169
	// get undialable addrs from config
170 171 172 173
	cfg, err := n.Repo.Config()
	if err != nil {
		return err
	}
Steven Allen's avatar
Steven Allen committed
174 175

	var libp2pOpts []libp2p.Option
176
	for _, s := range cfg.Swarm.AddrFilters {
Jeromy's avatar
Jeromy committed
177 178
		f, err := mamask.NewMask(s)
		if err != nil {
179
			return fmt.Errorf("incorrectly formatted address filter in config: %s", s)
Jeromy's avatar
Jeromy committed
180
		}
Steven Allen's avatar
Steven Allen committed
181
		libp2pOpts = append(libp2pOpts, libp2p.FilterAddresses(f))
Jeromy's avatar
Jeromy committed
182 183
	}

184 185 186
	if !cfg.Swarm.DisableBandwidthMetrics {
		// Set reporter
		n.Reporter = metrics.NewBandwidthCounter()
Steven Allen's avatar
Steven Allen committed
187
		libp2pOpts = append(libp2pOpts, libp2p.BandwidthReporter(n.Reporter))
188 189
	}

Jakub Sztandera's avatar
Jakub Sztandera committed
190 191 192 193 194 195
	swarmkey, err := n.Repo.SwarmKey()
	if err != nil {
		return err
	}

	if swarmkey != nil {
Steven Allen's avatar
Steven Allen committed
196
		protec, err := pnet.NewProtector(bytes.NewReader(swarmkey))
Jakub Sztandera's avatar
Jakub Sztandera committed
197
		if err != nil {
198
			return fmt.Errorf("failed to configure private network: %s", err)
Jakub Sztandera's avatar
Jakub Sztandera committed
199
		}
200
		n.PNetFingerprint = protec.Fingerprint()
201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218
		go func() {
			t := time.NewTicker(30 * time.Second)
			<-t.C // swallow one tick
			for {
				select {
				case <-t.C:
					if ph := n.PeerHost; ph != nil {
						if len(ph.Network().Peers()) == 0 {
							log.Warning("We are in private network and have no peers.")
							log.Warning("This might be configuration mistake.")
						}
					}
				case <-n.Process().Closing():
					t.Stop()
					return
				}
			}
		}()
Steven Allen's avatar
Steven Allen committed
219 220

		libp2pOpts = append(libp2pOpts, libp2p.PrivateNetwork(protec))
Jakub Sztandera's avatar
Jakub Sztandera committed
221 222
	}

223 224 225 226
	addrsFactory, err := makeAddrsFactory(cfg.Addresses)
	if err != nil {
		return err
	}
Steven Allen's avatar
Steven Allen committed
227 228 229 230
	if !cfg.Swarm.DisableRelay {
		addrsFactory = composeAddrsFactory(addrsFactory, filterRelayAddrs)
	}
	libp2pOpts = append(libp2pOpts, libp2p.AddrsFactory(addrsFactory))
231

Steven Allen's avatar
Steven Allen committed
232
	connm, err := constructConnMgr(cfg.Swarm.ConnMgr)
Jeromy's avatar
Jeromy committed
233 234 235
	if err != nil {
		return err
	}
Steven Allen's avatar
Steven Allen committed
236 237 238
	libp2pOpts = append(libp2pOpts, libp2p.ConnectionManager(connm))

	libp2pOpts = append(libp2pOpts, makeSmuxTransportOption(mplex))
Jeromy's avatar
Jeromy committed
239

Steven Allen's avatar
Steven Allen committed
240 241
	if !cfg.Swarm.DisableNatPortMap {
		libp2pOpts = append(libp2pOpts, libp2p.NATPortMap())
242
	}
Steven Allen's avatar
Steven Allen committed
243

244 245 246
	// disable the default listen addrs
	libp2pOpts = append(libp2pOpts, libp2p.NoListenAddrs)

247 248 249 250 251 252 253 254 255 256 257
	if cfg.Swarm.DisableRelay {
		// Enabled by default.
		libp2pOpts = append(libp2pOpts, libp2p.DisableRelay())
	} else {
		relayOpts := []circuit.RelayOpt{circuit.OptDiscovery}
		if cfg.Swarm.EnableRelayHop {
			relayOpts = append(relayOpts, circuit.OptHop)
		}
		libp2pOpts = append(libp2pOpts, libp2p.EnableRelay(relayOpts...))
	}

258 259 260
	// explicitly enable the default transports
	libp2pOpts = append(libp2pOpts, libp2p.DefaultTransports)

Marten Seemann's avatar
Marten Seemann committed
261 262 263 264
	if cfg.Experimental.QUIC {
		libp2pOpts = append(libp2pOpts, libp2p.Transport(quic.NewTransport))
	}

vyzo's avatar
vyzo committed
265 266 267 268 269 270 271 272
	// enable routing
	libp2pOpts = append(libp2pOpts, libp2p.Routing(func(h p2phost.Host) (routing.PeerRouting, error) {
		r, err := routingOption(ctx, h, n.Repo.Datastore(), n.RecordValidator)
		n.Routing = r
		return r, err
	}))

	// enable autorelay
vyzo's avatar
vyzo committed
273
	if cfg.Swarm.EnableAutoRelay {
vyzo's avatar
vyzo committed
274
		libp2pOpts = append(libp2pOpts, libp2p.EnableAutoRelay())
vyzo's avatar
vyzo committed
275
	}
276

Steven Allen's avatar
Steven Allen committed
277
	peerhost, err := hostOption(ctx, n.Identity, n.Peerstore, libp2pOpts...)
vyzo's avatar
vyzo committed
278

279
	if err != nil {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
280
		return err
281 282
	}

283 284
	n.PeerHost = peerhost

285
	if err := n.startOnlineServicesWithHost(ctx, routingOption, pubsub, ipnsps); err != nil {
286
		return err
287 288 289
	}

	// Ok, now we're ready to listen.
Łukasz Magiera's avatar
Łukasz Magiera committed
290
	if err := startListening(n.PeerHost, cfg); err != nil {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
291
		return err
292
	}
293

Łukasz Magiera's avatar
Łukasz Magiera committed
294
	n.P2P = p2p.NewP2P(n.Identity, n.PeerHost, n.Peerstore)
295

296
	// setup local discovery
Jeromy's avatar
Jeromy committed
297
	if do != nil {
Jeromy's avatar
Jeromy committed
298
		service, err := do(ctx, n.PeerHost)
Jeromy's avatar
Jeromy committed
299
		if err != nil {
Jeromy's avatar
Jeromy committed
300 301 302 303
			log.Error("mdns error: ", err)
		} else {
			service.RegisterNotifee(n)
			n.Discovery = service
Jeromy's avatar
Jeromy committed
304
		}
305 306
	}

307
	return n.Bootstrap(DefaultBootstrapConfig)
308 309
}

Jeromy's avatar
Jeromy committed
310 311
func constructConnMgr(cfg config.ConnMgr) (ifconnmgr.ConnManager, error) {
	switch cfg.Type {
312 313 314 315
	case "":
		// 'default' value is the basic connection manager
		return connmgr.NewConnManager(config.DefaultConnMgrLowWater, config.DefaultConnMgrHighWater, config.DefaultConnMgrGracePeriod), nil
	case "none":
Jeromy's avatar
Jeromy committed
316 317 318 319 320 321 322 323 324 325 326 327 328
		return nil, nil
	case "basic":
		grace, err := time.ParseDuration(cfg.GracePeriod)
		if err != nil {
			return nil, fmt.Errorf("parsing Swarm.ConnMgr.GracePeriod: %s", err)
		}

		return connmgr.NewConnManager(cfg.LowWater, cfg.HighWater, grace), nil
	default:
		return nil, fmt.Errorf("unrecognized ConnMgr.Type: %q", cfg.Type)
	}
}

Łukasz Magiera's avatar
Łukasz Magiera committed
329 330 331 332 333 334
func (n *IpfsNode) startLateOnlineServices(ctx context.Context) error {
	cfg, err := n.Repo.Config()
	if err != nil {
		return err
	}

335
	var keyProvider rp.KeyChanFunc
Łukasz Magiera's avatar
Łukasz Magiera committed
336 337 338 339 340 341 342 343 344 345 346

	switch cfg.Reprovider.Strategy {
	case "all":
		fallthrough
	case "":
		keyProvider = rp.NewBlockstoreProvider(n.Blockstore)
	case "roots":
		keyProvider = rp.NewPinnedProvider(n.Pinning, n.DAG, true)
	case "pinned":
		keyProvider = rp.NewPinnedProvider(n.Pinning, n.DAG, false)
	default:
347
		return fmt.Errorf("unknown reprovider strategy '%s'", cfg.Reprovider.Strategy)
Łukasz Magiera's avatar
Łukasz Magiera committed
348
	}
349
	n.Reprovider = rp.NewReprovider(ctx, n.Routing, keyProvider)
Łukasz Magiera's avatar
Łukasz Magiera committed
350

351 352 353 354 355
	reproviderInterval := kReprovideFrequency
	if cfg.Reprovider.Interval != "" {
		dur, err := time.ParseDuration(cfg.Reprovider.Interval)
		if err != nil {
			return err
Łukasz Magiera's avatar
Łukasz Magiera committed
356 357
		}

358
		reproviderInterval = dur
Łukasz Magiera's avatar
Łukasz Magiera committed
359 360
	}

361 362
	go n.Reprovider.Run(reproviderInterval)

Łukasz Magiera's avatar
Łukasz Magiera committed
363 364 365
	return nil
}

366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411
func makeAddrsFactory(cfg config.Addresses) (p2pbhost.AddrsFactory, error) {
	var annAddrs []ma.Multiaddr
	for _, addr := range cfg.Announce {
		maddr, err := ma.NewMultiaddr(addr)
		if err != nil {
			return nil, err
		}
		annAddrs = append(annAddrs, maddr)
	}

	filters := mafilter.NewFilters()
	noAnnAddrs := map[string]bool{}
	for _, addr := range cfg.NoAnnounce {
		f, err := mamask.NewMask(addr)
		if err == nil {
			filters.AddDialFilter(f)
			continue
		}
		maddr, err := ma.NewMultiaddr(addr)
		if err != nil {
			return nil, err
		}
		noAnnAddrs[maddr.String()] = true
	}

	return func(allAddrs []ma.Multiaddr) []ma.Multiaddr {
		var addrs []ma.Multiaddr
		if len(annAddrs) > 0 {
			addrs = annAddrs
		} else {
			addrs = allAddrs
		}

		var out []ma.Multiaddr
		for _, maddr := range addrs {
			// check for exact matches
			ok, _ := noAnnAddrs[maddr.String()]
			// check for /ipcidr matches
			if !ok && !filters.AddrBlocked(maddr) {
				out = append(out, maddr)
			}
		}
		return out
	}, nil
}

Steven Allen's avatar
Steven Allen committed
412 413 414
func makeSmuxTransportOption(mplexExp bool) libp2p.Option {
	const yamuxID = "/yamux/1.0.0"
	const mplexID = "/mplex/6.7.0"
415 416

	ymxtpt := &yamux.Transport{
Steven Allen's avatar
Steven Allen committed
417
		AcceptBacklog:          512,
418 419 420 421 422 423 424
		ConnectionWriteTimeout: time.Second * 10,
		KeepAliveInterval:      time.Second * 30,
		EnableKeepAlive:        true,
		MaxStreamWindowSize:    uint32(1024 * 512),
		LogOutput:              ioutil.Discard,
	}

425 426 427 428
	if os.Getenv("YAMUX_DEBUG") != "" {
		ymxtpt.LogOutput = os.Stderr
	}

Steven Allen's avatar
Steven Allen committed
429
	muxers := map[string]smux.Transport{yamuxID: ymxtpt}
430
	if mplexExp {
Steven Allen's avatar
Steven Allen committed
431
		muxers[mplexID] = mplex.DefaultTransport
432 433 434
	}

	// Allow muxer preference order overriding
Steven Allen's avatar
Steven Allen committed
435
	order := []string{yamuxID, mplexID}
436
	if prefs := os.Getenv("LIBP2P_MUX_PREFS"); prefs != "" {
Steven Allen's avatar
Steven Allen committed
437 438 439 440 441 442 443 444 445 446 447 448
		order = strings.Fields(prefs)
	}

	opts := make([]libp2p.Option, 0, len(order))
	for _, id := range order {
		tpt, ok := muxers[id]
		if !ok {
			log.Warning("unknown or duplicate muxer in LIBP2P_MUX_PREFS: %s", id)
			continue
		}
		delete(muxers, id)
		opts = append(opts, libp2p.Muxer(id, tpt))
449 450
	}

Steven Allen's avatar
Steven Allen committed
451
	return libp2p.ChainOptions(opts...)
452 453
}

Jeromy's avatar
Jeromy committed
454 455
func setupDiscoveryOption(d config.Discovery) DiscoveryOption {
	if d.MDNS.Enabled {
Jeromy's avatar
Jeromy committed
456
		return func(ctx context.Context, h p2phost.Host) (discovery.Service, error) {
Jeromy's avatar
Jeromy committed
457 458 459
			if d.MDNS.Interval == 0 {
				d.MDNS.Interval = 5
			}
Jeromy's avatar
Jeromy committed
460
			return discovery.NewMdnsService(ctx, h, time.Duration(d.MDNS.Interval)*time.Second, discovery.ServiceTag)
Jeromy's avatar
Jeromy committed
461 462 463 464 465
		}
	}
	return nil
}

466 467
// HandlePeerFound attempts to connect to peer from `PeerInfo`, if it fails
// logs a warning log.
Jeromy's avatar
Jeromy committed
468
func (n *IpfsNode) HandlePeerFound(p pstore.PeerInfo) {
469
	log.Warning("trying peer info: ", p)
470
	ctx, cancel := context.WithTimeout(n.Context(), discoveryConnTimeout)
rht's avatar
rht committed
471
	defer cancel()
472
	if err := n.PeerHost.Connect(ctx, p); err != nil {
473 474 475 476
		log.Warning("Failed to connect to peer found by discovery: ", err)
	}
}

477 478
// startOnlineServicesWithHost  is the set of services which need to be
// initialized with the host and _before_ we start listening.
479
func (n *IpfsNode) startOnlineServicesWithHost(ctx context.Context, routingOption RoutingOption, enablePubsub bool, enableIpnsps bool) error {
vyzo's avatar
vyzo committed
480 481 482 483 484 485
	cfg, err := n.Repo.Config()
	if err != nil {
		return err
	}

	if cfg.Swarm.EnableAutoNATService {
vyzo's avatar
vyzo committed
486 487 488 489 490
		var opts []libp2p.Option
		if cfg.Experimental.QUIC {
			opts = append(opts, libp2p.DefaultTransports, libp2p.Transport(quic.NewTransport))
		}

491
		svc, err := autonat.NewAutoNATService(ctx, n.PeerHost, opts...)
492 493 494
		if err != nil {
			return err
		}
vyzo's avatar
vyzo committed
495 496
		n.AutoNAT = svc
	}
497

vyzo's avatar
vyzo committed
498
	if enablePubsub || enableIpnsps {
Steven Allen's avatar
Steven Allen committed
499
		var service *pubsub.PubSub
500

Steven Allen's avatar
Steven Allen committed
501 502 503 504 505 506 507 508 509
		var pubsubOptions []pubsub.Option
		if cfg.Pubsub.DisableSigning {
			pubsubOptions = append(pubsubOptions, pubsub.WithMessageSigning(false))
		}

		if cfg.Pubsub.StrictSignatureVerification {
			pubsubOptions = append(pubsubOptions, pubsub.WithStrictSignatureVerification(true))
		}

510 511 512 513
		switch cfg.Pubsub.Router {
		case "":
			fallthrough
		case "floodsub":
514
			service, err = pubsub.NewFloodSub(ctx, n.PeerHost, pubsubOptions...)
515 516

		case "gossipsub":
517
			service, err = pubsub.NewGossipSub(ctx, n.PeerHost, pubsubOptions...)
518 519 520 521 522

		default:
			err = fmt.Errorf("Unknown pubsub router %s", cfg.Pubsub.Router)
		}

523 524 525
		if err != nil {
			return err
		}
Steven Allen's avatar
Steven Allen committed
526
		n.PubSub = service
527 528
	}

529 530
	// this code is necessary just for tests: mock network constructions
	// ignore the libp2p constructor options that actually construct the routing!
531
	if n.Routing == nil {
532
		r, err := routingOption(ctx, n.PeerHost, n.Repo.Datastore(), n.RecordValidator)
533 534 535 536
		if err != nil {
			return err
		}
		n.Routing = r
537
		n.PeerHost = rhost.Wrap(n.PeerHost, n.Routing)
538 539
	}

540 541 542 543 544 545 546 547 548 549 550 551 552
	// TODO: I'm not a fan of type assertions like this but the
	// `RoutingOption` system doesn't currently provide access to the
	// IpfsNode.
	//
	// Ideally, we'd do something like:
	//
	// 1. Add some fancy method to introspect into tiered routers to extract
	//    things like the pubsub router or the DHT (complicated, messy,
	//    probably not worth it).
	// 2. Pass the IpfsNode into the RoutingOption (would also remove the
	//    PSRouter case below.
	// 3. Introduce some kind of service manager? (my personal favorite but
	//    that requires a fair amount of work).
553
	if dht, ok := n.Routing.(*dht.IpfsDHT); ok {
554 555 556
		n.DHT = dht
	}

Steven Allen's avatar
Steven Allen committed
557
	if enableIpnsps {
558 559
		n.PSRouter = psrouter.NewPubsubValueStore(
			ctx,
560
			n.PeerHost,
561
			n.Routing,
Steven Allen's avatar
Steven Allen committed
562
			n.PubSub,
563
			n.RecordValidator,
564 565
		)
		n.Routing = rhelpers.Tiered{
Łukasz Magiera's avatar
Łukasz Magiera committed
566 567 568 569 570 571 572
			Routers: []routing.IpfsRouting{
				// Always check pubsub first.
				&rhelpers.Compose{
					ValueStore: &rhelpers.LimitedValueStore{
						ValueStore: n.PSRouter,
						Namespaces: []string{"ipns"},
					},
573
				},
Łukasz Magiera's avatar
Łukasz Magiera committed
574
				n.Routing,
575
			},
Łukasz Magiera's avatar
Łukasz Magiera committed
576
			Validator: n.RecordValidator,
577 578 579
		}
	}

580
	// setup exchange service
581
	bitswapNetwork := bsnet.NewFromIpfsHost(n.PeerHost, n.Routing)
Łukasz Magiera's avatar
Łukasz Magiera committed
582
	n.Exchange = bitswap.New(ctx, bitswapNetwork, n.Blockstore)
583

584 585 586 587 588
	size, err := n.getCacheSize()
	if err != nil {
		return err
	}

589
	// setup name system
590
	n.Namesys = namesys.NewNameSystem(n.Routing, n.Repo.Datastore(), size)
591

Jeromy's avatar
Jeromy committed
592
	// setup ipns republishing
593
	return n.setupIpnsRepublisher()
594 595
}

596 597 598 599 600 601 602 603 604
// getCacheSize returns cache life and cache size
func (n *IpfsNode) getCacheSize() (int, error) {
	cfg, err := n.Repo.Config()
	if err != nil {
		return 0, err
	}

	cs := cfg.Ipns.ResolveCacheSize
	if cs == 0 {
605
		cs = DefaultIpnsCacheSize
606 607 608 609 610 611 612
	}
	if cs < 0 {
		return 0, fmt.Errorf("cannot specify negative resolve cache size")
	}
	return cs, nil
}

613
func (n *IpfsNode) setupIpnsRepublisher() error {
Jeromy's avatar
Jeromy committed
614 615 616 617
	cfg, err := n.Repo.Config()
	if err != nil {
		return err
	}
618

619
	n.IpnsRepub = ipnsrp.NewRepublisher(n.Namesys, n.Repo.Datastore(), n.PrivateKey, n.Repo.Keystore())
620

Jeromy's avatar
Jeromy committed
621 622 623 624 625 626
	if cfg.Ipns.RepublishPeriod != "" {
		d, err := time.ParseDuration(cfg.Ipns.RepublishPeriod)
		if err != nil {
			return fmt.Errorf("failure to parse config setting IPNS.RepublishPeriod: %s", err)
		}

627
		if !u.Debug && (d < time.Minute || d > (time.Hour*24)) {
Jeromy's avatar
Jeromy committed
628 629 630 631 632 633
			return fmt.Errorf("config setting IPNS.RepublishPeriod is not between 1min and 1day: %s", d)
		}

		n.IpnsRepub.Interval = d
	}

634
	if cfg.Ipns.RecordLifetime != "" {
635
		d, err := time.ParseDuration(cfg.Ipns.RecordLifetime)
636 637 638 639 640 641 642
		if err != nil {
			return fmt.Errorf("failure to parse config setting IPNS.RecordLifetime: %s", err)
		}

		n.IpnsRepub.RecordLifetime = d
	}

Jeromy's avatar
Jeromy committed
643 644
	n.Process().Go(n.IpnsRepub.Run)

645 646 647
	return nil
}

648 649 650 651 652 653 654 655 656 657 658 659
// Process returns the Process object
func (n *IpfsNode) Process() goprocess.Process {
	return n.proc
}

// Close calls Close() on the Process object
func (n *IpfsNode) Close() error {
	return n.proc.Close()
}

// Context returns the IpfsNode context
func (n *IpfsNode) Context() context.Context {
660 661 662
	if n.ctx == nil {
		n.ctx = context.TODO()
	}
663 664 665
	return n.ctx
}

666 667
// teardown closes owned children. If any errors occur, this function returns
// the first error.
Brian Tiger Chow's avatar
Brian Tiger Chow committed
668
func (n *IpfsNode) teardown() error {
669
	log.Debug("core is shutting down...")
670 671
	// owned objects are closed in this teardown to ensure that they're closed
	// regardless of which constructor was used to add them to the node.
Jeromy's avatar
Jeromy committed
672 673
	var closers []io.Closer

674
	// NOTE: The order that objects are added(closed) matters, if an object
Jeromy's avatar
Jeromy committed
675 676 677 678 679
	// needs to use another during its shutdown/cleanup process, it should be
	// closed before that other object

	if n.FilesRoot != nil {
		closers = append(closers, n.FilesRoot)
Jeromy's avatar
Jeromy committed
680
	}
681

682 683 684 685
	if n.Exchange != nil {
		closers = append(closers, n.Exchange)
	}

686
	if n.Mounts.Ipfs != nil && !n.Mounts.Ipfs.IsActive() {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
687 688
		closers = append(closers, mount.Closer(n.Mounts.Ipfs))
	}
689
	if n.Mounts.Ipns != nil && !n.Mounts.Ipns.IsActive() {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
690 691 692
		closers = append(closers, mount.Closer(n.Mounts.Ipns))
	}

693 694
	if n.DHT != nil {
		closers = append(closers, n.DHT.Process())
Jeromy's avatar
Jeromy committed
695 696
	}

Jeromy's avatar
Jeromy committed
697 698 699 700
	if n.Blocks != nil {
		closers = append(closers, n.Blocks)
	}

Jeromy's avatar
Jeromy committed
701 702
	if n.Bootstrapper != nil {
		closers = append(closers, n.Bootstrapper)
703 704
	}

Jeromy's avatar
Jeromy committed
705 706
	if n.PeerHost != nil {
		closers = append(closers, n.PeerHost)
707
	}
708

Jeromy's avatar
Jeromy committed
709 710 711
	// Repo closed last, most things need to preserve state here
	closers = append(closers, n.Repo)

712
	var errs []error
713
	for _, closer := range closers {
714 715
		if err := closer.Close(); err != nil {
			errs = append(errs, err)
716 717 718 719
		}
	}
	if len(errs) > 0 {
		return errs[0]
Brian Tiger Chow's avatar
Brian Tiger Chow committed
720 721
	}
	return nil
Brian Tiger Chow's avatar
Brian Tiger Chow committed
722 723
}

724
// OnlineMode returns whether or not the IpfsNode is in OnlineMode.
Brian Tiger Chow's avatar
Brian Tiger Chow committed
725
func (n *IpfsNode) OnlineMode() bool {
726
	return n.mode == onlineMode
Brian Tiger Chow's avatar
Brian Tiger Chow committed
727 728
}

729
// SetLocal will set the IpfsNode to local mode
730 731 732 733 734 735 736
func (n *IpfsNode) SetLocal(isLocal bool) {
	if isLocal {
		n.mode = localMode
	}
	n.localModeSet = true
}

737
// LocalMode returns whether or not the IpfsNode is in LocalMode
738 739 740 741 742
func (n *IpfsNode) LocalMode() bool {
	if !n.localModeSet {
		// programmer error should not happen
		panic("local mode not set")
	}
743
	return n.mode == localMode
744 745
}

746
// Bootstrap will set and call the IpfsNodes bootstrap function.
747
func (n *IpfsNode) Bootstrap(cfg BootstrapConfig) error {
748
	// TODO what should return value be when in offlineMode?
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
749 750 751 752
	if n.Routing == nil {
		return nil
	}

753 754 755 756 757 758 759
	if n.Bootstrapper != nil {
		n.Bootstrapper.Close() // stop previous bootstrap process.
	}

	// if the caller did not specify a bootstrap peer function, get the
	// freshest bootstrap peers from config. this responds to live changes.
	if cfg.BootstrapPeers == nil {
Jeromy's avatar
Jeromy committed
760
		cfg.BootstrapPeers = func() []pstore.PeerInfo {
761
			ps, err := n.loadBootstrapPeers()
762
			if err != nil {
763
				log.Warning("failed to parse bootstrap peers from config")
764 765 766 767 768 769 770 771 772
				return nil
			}
			return ps
		}
	}

	var err error
	n.Bootstrapper, err = Bootstrap(n, cfg)
	return err
773 774
}

775 776
func (n *IpfsNode) loadID() error {
	if n.Identity != "" {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
777
		return errors.New("identity already loaded")
778 779
	}

780 781 782 783 784 785
	cfg, err := n.Repo.Config()
	if err != nil {
		return err
	}

	cid := cfg.Identity.PeerID
786
	if cid == "" {
787
		return errors.New("identity was not set in config (was 'ipfs init' run?)")
788 789
	}
	if len(cid) == 0 {
790
		return errors.New("no peer ID in config! (was 'ipfs init' run?)")
791 792
	}

Steven Allen's avatar
Steven Allen committed
793 794 795 796 797 798
	id, err := peer.IDB58Decode(cid)
	if err != nil {
		return fmt.Errorf("peer ID invalid: %s", err)
	}

	n.Identity = id
799 800
	return nil
}
801

802
// GetKey will return a key from the Keystore with name `name`.
803 804
func (n *IpfsNode) GetKey(name string) (ic.PrivKey, error) {
	if name == "self" {
805 806 807
		if n.PrivateKey == nil {
			return nil, fmt.Errorf("private key not available")
		}
808 809 810 811 812 813
		return n.PrivateKey, nil
	} else {
		return n.Repo.Keystore().Get(name)
	}
}

814 815
// loadPrivateKey loads the private key *if* available
func (n *IpfsNode) loadPrivateKey() error {
816
	if n.Identity == "" || n.Peerstore == nil {
Łukasz Magiera's avatar
Łukasz Magiera committed
817
		return errors.New("loaded private key out of order")
818 819
	}

820
	if n.PrivateKey != nil {
Kejie Zhang's avatar
Kejie Zhang committed
821 822
		log.Warning("private key already loaded")
		return nil
823 824
	}

825 826 827 828 829
	cfg, err := n.Repo.Config()
	if err != nil {
		return err
	}

830 831 832 833
	if cfg.Identity.PrivKey == "" {
		return nil
	}

834
	sk, err := loadPrivateKey(&cfg.Identity, n.Identity)
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
835
	if err != nil {
836
		return err
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
837
	}
838

839 840
	n.PrivateKey = sk
	n.Peerstore.AddPrivKey(n.Identity, n.PrivateKey)
Jeromy's avatar
Jeromy committed
841 842 843 844
	n.Peerstore.AddPubKey(n.Identity, sk.GetPublic())
	return nil
}

Jeromy's avatar
Jeromy committed
845
func (n *IpfsNode) loadBootstrapPeers() ([]pstore.PeerInfo, error) {
846 847 848 849 850 851
	cfg, err := n.Repo.Config()
	if err != nil {
		return nil, err
	}

	parsed, err := cfg.BootstrapPeers()
852 853 854 855 856 857
	if err != nil {
		return nil, err
	}
	return toPeerInfos(parsed), nil
}

Jeromy's avatar
Jeromy committed
858
func (n *IpfsNode) loadFilesRoot() error {
Jeromy's avatar
Jeromy committed
859
	dsk := ds.NewKey("/local/filesroot")
860
	pf := func(ctx context.Context, c cid.Cid) error {
Jeromy's avatar
Jeromy committed
861
		return n.Repo.Datastore().Put(dsk, c.Bytes())
Jeromy's avatar
Jeromy committed
862 863
	}

864
	var nd *merkledag.ProtoNode
Jeromy's avatar
Jeromy committed
865 866 867 868
	val, err := n.Repo.Datastore().Get(dsk)

	switch {
	case err == ds.ErrNotFound || val == nil:
869
		nd = ft.EmptyDirNode()
870
		err := n.DAG.Add(n.Context(), nd)
Jeromy's avatar
Jeromy committed
871 872 873 874
		if err != nil {
			return fmt.Errorf("failure writing to dagstore: %s", err)
		}
	case err == nil:
875
		c, err := cid.Cast(val)
Jeromy's avatar
Jeromy committed
876 877 878 879
		if err != nil {
			return err
		}

880
		rnd, err := n.DAG.Get(n.Context(), c)
Jeromy's avatar
Jeromy committed
881 882 883
		if err != nil {
			return fmt.Errorf("error loading filesroot from DAG: %s", err)
		}
884 885 886 887 888 889 890

		pbnd, ok := rnd.(*merkledag.ProtoNode)
		if !ok {
			return merkledag.ErrNotProtobuf
		}

		nd = pbnd
Jeromy's avatar
Jeromy committed
891 892 893 894 895 896 897 898 899 900 901 902 903
	default:
		return err
	}

	mr, err := mfs.NewRoot(n.Context(), n.DAG, nd, pf)
	if err != nil {
		return err
	}

	n.FilesRoot = mr
	return nil
}

904 905
func loadPrivateKey(cfg *config.Identity, id peer.ID) (ic.PrivKey, error) {
	sk, err := cfg.DecodePrivateKey("passphrase todo!")
906 907 908
	if err != nil {
		return nil, err
	}
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
909

910 911 912 913
	id2, err := peer.IDFromPrivateKey(sk)
	if err != nil {
		return nil, err
	}
914

915 916
	if id2 != id {
		return nil, fmt.Errorf("private key in config does not match id: %s != %s", id, id2)
917 918
	}

919
	return sk, nil
920
}
921

922
func listenAddresses(cfg *config.Config) ([]ma.Multiaddr, error) {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
923 924 925
	var listen []ma.Multiaddr
	for _, addr := range cfg.Addresses.Swarm {
		maddr, err := ma.NewMultiaddr(addr)
926
		if err != nil {
Łukasz Magiera's avatar
Łukasz Magiera committed
927
			return nil, fmt.Errorf("failure to parse config.Addresses.Swarm: %s", cfg.Addresses.Swarm)
928
		}
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
929
		listen = append(listen, maddr)
930 931 932 933
	}

	return listen, nil
}
934

Kevin Atkinson's avatar
Kevin Atkinson committed
935
type ConstructPeerHostOpts struct {
936
	AddrsFactory      p2pbhost.AddrsFactory
vyzo's avatar
vyzo committed
937 938 939
	DisableNatPortMap bool
	DisableRelay      bool
	EnableRelayHop    bool
Jeromy's avatar
Jeromy committed
940
	ConnectionManager ifconnmgr.ConnManager
Kevin Atkinson's avatar
Kevin Atkinson committed
941 942
}

Steven Allen's avatar
Steven Allen committed
943
type HostOption func(ctx context.Context, id peer.ID, ps pstore.Peerstore, options ...libp2p.Option) (p2phost.Host, error)
Jeromy's avatar
Jeromy committed
944 945 946

var DefaultHostOption HostOption = constructPeerHost

947
// isolates the complex initialization steps
Steven Allen's avatar
Steven Allen committed
948 949 950 951
func constructPeerHost(ctx context.Context, id peer.ID, ps pstore.Peerstore, options ...libp2p.Option) (p2phost.Host, error) {
	pkey := ps.PrivKey(id)
	if pkey == nil {
		return nil, fmt.Errorf("missing private key for node ID: %s", id.Pretty())
952
	}
Steven Allen's avatar
Steven Allen committed
953 954
	options = append([]libp2p.Option{libp2p.Identity(pkey), libp2p.Peerstore(ps)}, options...)
	return libp2p.New(ctx, options...)
955 956
}

957 958 959 960 961 962 963 964 965 966 967 968
func filterRelayAddrs(addrs []ma.Multiaddr) []ma.Multiaddr {
	var raddrs []ma.Multiaddr
	for _, addr := range addrs {
		_, err := addr.ValueForProtocol(circuit.P_CIRCUIT)
		if err == nil {
			continue
		}
		raddrs = append(raddrs, addr)
	}
	return raddrs
}

969 970 971 972 973 974
func composeAddrsFactory(f, g p2pbhost.AddrsFactory) p2pbhost.AddrsFactory {
	return func(addrs []ma.Multiaddr) []ma.Multiaddr {
		return f(g(addrs))
	}
}

975
// startListening on the network addresses
Łukasz Magiera's avatar
Łukasz Magiera committed
976
func startListening(host p2phost.Host, cfg *config.Config) error {
977 978
	listenAddrs, err := listenAddresses(cfg)
	if err != nil {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
979
		return err
980 981 982
	}

	// Actually start listening:
Steven Allen's avatar
Steven Allen committed
983
	if err := host.Network().Listen(listenAddrs...); err != nil {
984
		return err
985 986
	}

987
	// list out our addresses
988
	addrs, err := host.Network().InterfaceListenAddresses()
989
	if err != nil {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
990
		return err
991
	}
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
992
	log.Infof("Swarm listening at: %s", addrs)
993
	return nil
994
}
995

996 997 998 999 1000 1001
func constructDHTRouting(ctx context.Context, host p2phost.Host, dstore ds.Batching, validator record.Validator) (routing.IpfsRouting, error) {
	return dht.New(
		ctx, host,
		dhtopts.Datastore(dstore),
		dhtopts.Validator(validator),
	)
1002
}
Jeromy's avatar
Jeromy committed
1003

1004 1005 1006 1007 1008 1009 1010
func constructClientDHTRouting(ctx context.Context, host p2phost.Host, dstore ds.Batching, validator record.Validator) (routing.IpfsRouting, error) {
	return dht.New(
		ctx, host,
		dhtopts.Client(true),
		dhtopts.Datastore(dstore),
		dhtopts.Validator(validator),
	)
Jeromy's avatar
Jeromy committed
1011 1012
}

1013
type RoutingOption func(context.Context, p2phost.Host, ds.Batching, record.Validator) (routing.IpfsRouting, error)
Jeromy's avatar
Jeromy committed
1014

Jeromy's avatar
Jeromy committed
1015
type DiscoveryOption func(context.Context, p2phost.Host) (discovery.Service, error)
Jeromy's avatar
Jeromy committed
1016

1017
var DHTOption RoutingOption = constructDHTRouting
Jeromy's avatar
Jeromy committed
1018
var DHTClientOption RoutingOption = constructClientDHTRouting
1019
var NilRouterOption RoutingOption = nilrouting.ConstructNilRouting