core.go 21.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
	"io/ioutil"
Jeromy's avatar
Jeromy committed
19
	"net"
20 21
	"os"
	"strings"
22
	"time"
23

24 25 26 27 28 29
	bstore "github.com/ipfs/go-ipfs/blocks/blockstore"
	bserv "github.com/ipfs/go-ipfs/blockservice"
	exchange "github.com/ipfs/go-ipfs/exchange"
	bitswap "github.com/ipfs/go-ipfs/exchange/bitswap"
	bsnet "github.com/ipfs/go-ipfs/exchange/bitswap/network"
	rp "github.com/ipfs/go-ipfs/exchange/reprovide"
30
	filestore "github.com/ipfs/go-ipfs/filestore"
31 32 33 34 35 36 37 38 39 40 41 42
	mount "github.com/ipfs/go-ipfs/fuse/mount"
	merkledag "github.com/ipfs/go-ipfs/merkledag"
	mfs "github.com/ipfs/go-ipfs/mfs"
	namesys "github.com/ipfs/go-ipfs/namesys"
	ipnsrp "github.com/ipfs/go-ipfs/namesys/republisher"
	path "github.com/ipfs/go-ipfs/path"
	pin "github.com/ipfs/go-ipfs/pin"
	repo "github.com/ipfs/go-ipfs/repo"
	config "github.com/ipfs/go-ipfs/repo/config"
	nilrouting "github.com/ipfs/go-ipfs/routing/none"
	offroute "github.com/ipfs/go-ipfs/routing/offline"
	ft "github.com/ipfs/go-ipfs/unixfs"
Jeromy's avatar
Jeromy committed
43

44
	routing "gx/ipfs/QmNdaQ8itUU9jEZUwTsG4gHMaPmRfi6FEe89QjQAFbep3M/go-libp2p-routing"
45 46
	ic "gx/ipfs/QmP1DfoUjiWH2ZBo1PBH6FupdBucbDepx3HpWmEY6JMUpY/go-libp2p-crypto"
	ipnet "gx/ipfs/QmPsBptED6X43GYg3347TAUruN3UfsAhaGTP9xbinYX7uf/go-libp2p-interface-pnet"
47
	mplex "gx/ipfs/QmQ3UABWTgK78utKeiVXaH9BrjC7Ydn1pRuwqnWHT3p4zh/go-smux-multiplex"
48 49 50 51 52
	discovery "gx/ipfs/QmQA5mdxru8Bh6dpC9PJfSkumqnmHgJX7knxSgBo5Lpime/go-libp2p/p2p/discovery"
	p2pbhost "gx/ipfs/QmQA5mdxru8Bh6dpC9PJfSkumqnmHgJX7knxSgBo5Lpime/go-libp2p/p2p/host/basic"
	rhost "gx/ipfs/QmQA5mdxru8Bh6dpC9PJfSkumqnmHgJX7knxSgBo5Lpime/go-libp2p/p2p/host/routed"
	identify "gx/ipfs/QmQA5mdxru8Bh6dpC9PJfSkumqnmHgJX7knxSgBo5Lpime/go-libp2p/p2p/protocol/identify"
	ping "gx/ipfs/QmQA5mdxru8Bh6dpC9PJfSkumqnmHgJX7knxSgBo5Lpime/go-libp2p/p2p/protocol/ping"
53
	mssmux "gx/ipfs/QmRVYfZ7tWNHPBzWiG6KWGzvT2hcGems8srihsQE29x1U5/go-smux-multistream"
Jeromy's avatar
Jeromy committed
54
	ds "gx/ipfs/QmRWDav6mzWseLWeYfVd5fvUKiVe9xNH29YfMF438fG364/go-datastore"
55
	dht "gx/ipfs/QmRmroYSdievxnjiuy99C8BzShNstdEWcEF3LQHF7fUbez/go-libp2p-kad-dht"
56
	goprocess "gx/ipfs/QmSF8fPo3jgVBAy8fpdjjYqgG87dkJgUprRBHRd2tmfgpP/goprocess"
Jeromy's avatar
Jeromy committed
57
	mamask "gx/ipfs/QmSMZwvs3n4GBikZ7hKzT17c3bk65FmyZo2JqtJ16swqCv/multiaddr-filter"
Jeromy's avatar
Jeromy committed
58
	logging "gx/ipfs/QmSpJByNKFX1sCsHBEp3R73FL4NF6FnQTEGyNAXHm2GS52/go-log"
Jeromy's avatar
Jeromy committed
59
	b58 "gx/ipfs/QmT8rehPR3F6bmwL6zjUN8XpiDBFFpMP2myPdC6ApsWfJf/go-base58"
Lars Gierth's avatar
Lars Gierth committed
60
	pnet "gx/ipfs/QmTJoXQ24GqDf9MqAUwf3vW38HG6ahE9S7GzZoRMEeE8Kc/go-libp2p-pnet"
61 62 63
	floodsub "gx/ipfs/QmUpeULWfmtsgCnfuRN3BHsfhHvBxNphoYh4La4CMxGt2Z/floodsub"
	p2phost "gx/ipfs/QmUywuGNZoUKV8B9iyvup9bPkLiMrhTsyVMkeSXW5VxAfC/go-libp2p-host"
	swarm "gx/ipfs/QmVkDnNm71vYyY6s6rXwtmyDYis3WkKyrEhMECwT6R12uJ/go-libp2p-swarm"
64
	u "gx/ipfs/QmWbjfz3u6HkAdPh34dgPchGbQjob6LXLhAeCGii2TX69n/go-ipfs-util"
65
	pstore "gx/ipfs/QmXZSd1qR5BxZkPyuwfT5jpqQFScZccoZvDneXsKzCNHWX/go-libp2p-peerstore"
66 67
	cid "gx/ipfs/QmYhQaCYEcaPPjxJX7YcPcVKkQfRy6sJ7B3XmGFk82XYdQ/go-cid"
	addrutil "gx/ipfs/QmbH3urJHTrZSUETgvQRriWM6mMFqyNSwCqnhknxfSGVWv/go-addr-util"
68
	yamux "gx/ipfs/Qmbn7RYyWzBVXiUp9jZ1dA4VADHy9DtS7iZLwfhEUQvm3U/go-smux-yamux"
69 70
	ma "gx/ipfs/QmcyqRMCAXVtYPS4DiBrA7sezL9rRGfW8Ctx7cywL4TXJj/go-multiaddr"
	peer "gx/ipfs/QmdS9KpbDyPrieswibZhkod1oXqRwZJrUPzxCofAMWpFGq/go-libp2p-peer"
71
	metrics "gx/ipfs/QmdibiN2wzuuXXz4JvqQ1ZGW3eUkoAy1AWznHFau6iePCc/go-libp2p-metrics"
72
	smux "gx/ipfs/QmeZBgYBHvxMukGK5ojg28BCNLB9SeXqT7XXg6o7r2GbJy/go-stream-muxer"
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
73 74
)

Jeromy's avatar
Jeromy committed
75
const IpnsValidatorTag = "ipns"
76

77
const kReprovideFrequency = time.Hour * 12
78
const discoveryConnTimeout = time.Second * 30
Jeromy's avatar
Jeromy committed
79

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

82 83 84 85
type mode int

const (
	// zero value is not a valid mode, must be explicitly set
86
	localMode mode = iota
87 88 89 90
	offlineMode
	onlineMode
)

91 92 93 94
func init() {
	identify.ClientVersion = "go-ipfs/" + config.CurrentVersionNumber + "/" + config.CurrentCommit
}

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

98
	// Self
99
	Identity peer.ID // the local node's identity
100

101
	Repo repo.Repo
102 103

	// Local node
Jakub Sztandera's avatar
Jakub Sztandera committed
104 105 106 107
	Pinning        pin.Pinner // the pinning manager
	Mounts         Mounts     // current mount state, if any.
	PrivateKey     ic.PrivKey // the local node's private Key
	PNetFingerpint []byte     // fingerprint of private network
108 109

	// Services
Jeromy's avatar
Jeromy committed
110
	Peerstore  pstore.Peerstore     // storage for other Peer instances
111
	Blockstore bstore.GCBlockstore  // the block store (lower level)
112 113 114
	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
115
	Blocks     bserv.BlockService   // the block service, get/add blocks.
116 117
	DAG        merkledag.DAGService // the merkle dag service, get/add objects.
	Resolver   *path.Resolver       // the path resolution system
Jeromy's avatar
Jeromy committed
118
	Reporter   metrics.Reporter
119
	Discovery  discovery.Service
Jeromy's avatar
Jeromy committed
120
	FilesRoot  *mfs.Root
121 122

	// Online
123 124 125 126 127
	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
Jeromy's avatar
Jeromy committed
128 129
	Ping         *ping.PingService
	Reprovider   *rp.Reprovider // the value reprovider system
Jeromy's avatar
Jeromy committed
130
	IpnsRepub    *ipnsrp.Republisher
131

Jeromy's avatar
Jeromy committed
132 133
	Floodsub *floodsub.PubSub

134 135
	proc goprocess.Process
	ctx  context.Context
136

Jeromy's avatar
Jeromy committed
137
	mode         mode
138
	localModeSet bool
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
139 140
}

141 142 143 144 145 146 147 148
// 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
}

149
func (n *IpfsNode) startOnlineServices(ctx context.Context, routingOption RoutingOption, hostOption HostOption, do DiscoveryOption, pubsub, mplex bool) error {
150 151

	if n.PeerHost != nil { // already online.
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
152
		return errors.New("node already online")
153 154 155
	}

	// load private key
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
156
	if err := n.LoadPrivateKey(); err != nil {
157 158 159
		return err
	}

Jeromy's avatar
Jeromy committed
160
	// get undialable addrs from config
161 162 163 164
	cfg, err := n.Repo.Config()
	if err != nil {
		return err
	}
Jeromy's avatar
Jeromy committed
165
	var addrfilter []*net.IPNet
166
	for _, s := range cfg.Swarm.AddrFilters {
Jeromy's avatar
Jeromy committed
167 168
		f, err := mamask.NewMask(s)
		if err != nil {
169
			return fmt.Errorf("incorrectly formatted address filter in config: %s", s)
Jeromy's avatar
Jeromy committed
170 171 172 173
		}
		addrfilter = append(addrfilter, f)
	}

174 175 176 177 178
	if !cfg.Swarm.DisableBandwidthMetrics {
		// Set reporter
		n.Reporter = metrics.NewBandwidthCounter()
	}

179 180
	tpt := makeSmuxTransport(mplex)

Jakub Sztandera's avatar
Jakub Sztandera committed
181 182 183 184 185 186 187 188 189 190 191 192
	swarmkey, err := n.Repo.SwarmKey()
	if err != nil {
		return err
	}

	var protec ipnet.Protector
	if swarmkey != nil {
		protec, err = pnet.NewProtector(bytes.NewReader(swarmkey))
		if err != nil {
			return err
		}
		n.PNetFingerpint = protec.Fingerprint()
193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210
		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
				}
			}
		}()
Jakub Sztandera's avatar
Jakub Sztandera committed
211 212 213
	}

	peerhost, err := hostOption(ctx, n.Identity, n.Peerstore, n.Reporter,
Kevin Atkinson's avatar
Kevin Atkinson committed
214
		addrfilter, tpt, protec, &ConstructPeerHostOpts{DisableNatPortMap: cfg.Swarm.DisableNatPortMap})
215
	if err != nil {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
216
		return err
217 218
	}

219
	if err := n.startOnlineServicesWithHost(ctx, peerhost, routingOption); err != nil {
220
		return err
221 222 223
	}

	// Ok, now we're ready to listen.
224
	if err := startListening(ctx, n.PeerHost, cfg); err != nil {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
225
		return err
226
	}
227

228
	n.Reprovider = rp.NewReprovider(n.Routing, n.Blockstore)
229 230 231 232 233 234 235 236 237 238 239 240 241 242

	if cfg.Reprovider.Interval != "0" {
		interval := kReprovideFrequency
		if cfg.Reprovider.Interval != "" {
			dur, err := time.ParseDuration(cfg.Reprovider.Interval)
			if err != nil {
				return err
			}

			interval = dur
		}

		go n.Reprovider.ProvideEvery(ctx, interval)
	}
243

Jeromy's avatar
Jeromy committed
244 245 246
	if pubsub {
		n.Floodsub = floodsub.NewFloodSub(ctx, peerhost)
	}
Jeromy's avatar
Jeromy committed
247

248
	// setup local discovery
Jeromy's avatar
Jeromy committed
249
	if do != nil {
Jeromy's avatar
Jeromy committed
250
		service, err := do(ctx, n.PeerHost)
Jeromy's avatar
Jeromy committed
251
		if err != nil {
Jeromy's avatar
Jeromy committed
252 253 254 255
			log.Error("mdns error: ", err)
		} else {
			service.RegisterNotifee(n)
			n.Discovery = service
Jeromy's avatar
Jeromy committed
256
		}
257 258
	}

259
	return n.Bootstrap(DefaultBootstrapConfig)
260 261
}

262 263 264 265 266 267 268 269 270 271 272 273
func makeSmuxTransport(mplexExp bool) smux.Transport {
	mstpt := mssmux.NewBlankTransport()

	ymxtpt := &yamux.Transport{
		AcceptBacklog:          8192,
		ConnectionWriteTimeout: time.Second * 10,
		KeepAliveInterval:      time.Second * 30,
		EnableKeepAlive:        true,
		MaxStreamWindowSize:    uint32(1024 * 512),
		LogOutput:              ioutil.Discard,
	}

274 275 276 277
	if os.Getenv("YAMUX_DEBUG") != "" {
		ymxtpt.LogOutput = os.Stderr
	}

278 279 280 281 282 283 284 285 286 287 288 289 290 291
	mstpt.AddTransport("/yamux/1.0.0", ymxtpt)

	if mplexExp {
		mstpt.AddTransport("/mplex/6.7.0", mplex.DefaultTransport)
	}

	// Allow muxer preference order overriding
	if prefs := os.Getenv("LIBP2P_MUX_PREFS"); prefs != "" {
		mstpt.OrderPreference = strings.Fields(prefs)
	}

	return mstpt
}

Jeromy's avatar
Jeromy committed
292 293
func setupDiscoveryOption(d config.Discovery) DiscoveryOption {
	if d.MDNS.Enabled {
Jeromy's avatar
Jeromy committed
294
		return func(ctx context.Context, h p2phost.Host) (discovery.Service, error) {
Jeromy's avatar
Jeromy committed
295 296 297
			if d.MDNS.Interval == 0 {
				d.MDNS.Interval = 5
			}
Jeromy's avatar
Jeromy committed
298
			return discovery.NewMdnsService(ctx, h, time.Duration(d.MDNS.Interval)*time.Second)
Jeromy's avatar
Jeromy committed
299 300 301 302 303
		}
	}
	return nil
}

Jeromy's avatar
Jeromy committed
304
func (n *IpfsNode) HandlePeerFound(p pstore.PeerInfo) {
305
	log.Warning("trying peer info: ", p)
306
	ctx, cancel := context.WithTimeout(n.Context(), discoveryConnTimeout)
rht's avatar
rht committed
307
	defer cancel()
308
	if err := n.PeerHost.Connect(ctx, p); err != nil {
309 310 311 312
		log.Warning("Failed to connect to peer found by discovery: ", err)
	}
}

313 314
// startOnlineServicesWithHost  is the set of services which need to be
// initialized with the host and _before_ we start listening.
315
func (n *IpfsNode) startOnlineServicesWithHost(ctx context.Context, host p2phost.Host, routingOption RoutingOption) error {
316
	// setup diagnostics service
Jeromy's avatar
Jeromy committed
317
	n.Ping = ping.NewPingService(host)
318 319

	// setup routing service
320
	r, err := routingOption(ctx, host, n.Repo.Datastore())
Jeromy's avatar
Jeromy committed
321
	if err != nil {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
322
		return err
323
	}
Jeromy's avatar
Jeromy committed
324
	n.Routing = r
325

326 327 328
	// Wrap standard peer host with routing system to allow unknown peer lookups
	n.PeerHost = rhost.Wrap(host, n.Routing)

329 330
	// setup exchange service
	const alwaysSendToPeer = true // use YesManStrategy
331
	bitswapNetwork := bsnet.NewFromIpfsHost(n.PeerHost, n.Routing)
332 333
	n.Exchange = bitswap.New(ctx, n.Identity, bitswapNetwork, n.Blockstore, alwaysSendToPeer)

334 335 336 337 338
	size, err := n.getCacheSize()
	if err != nil {
		return err
	}

339
	// setup name system
340
	n.Namesys = namesys.NewNameSystem(n.Routing, n.Repo.Datastore(), size)
341

Jeromy's avatar
Jeromy committed
342
	// setup ipns republishing
343
	return n.setupIpnsRepublisher()
344 345
}

346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362
// 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 {
		cs = 128
	}
	if cs < 0 {
		return 0, fmt.Errorf("cannot specify negative resolve cache size")
	}
	return cs, nil
}

363
func (n *IpfsNode) setupIpnsRepublisher() error {
Jeromy's avatar
Jeromy committed
364 365 366 367
	cfg, err := n.Repo.Config()
	if err != nil {
		return err
	}
368 369 370 371

	n.IpnsRepub = ipnsrp.NewRepublisher(n.Routing, n.Repo.Datastore(), n.Peerstore)
	n.IpnsRepub.AddName(n.Identity)

Jeromy's avatar
Jeromy committed
372 373 374 375 376 377
	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)
		}

378
		if !u.Debug && (d < time.Minute || d > (time.Hour*24)) {
Jeromy's avatar
Jeromy committed
379 380 381 382 383 384
			return fmt.Errorf("config setting IPNS.RepublishPeriod is not between 1min and 1day: %s", d)
		}

		n.IpnsRepub.Interval = d
	}

385 386 387 388 389 390 391 392 393
	if cfg.Ipns.RecordLifetime != "" {
		d, err := time.ParseDuration(cfg.Ipns.RepublishPeriod)
		if err != nil {
			return fmt.Errorf("failure to parse config setting IPNS.RecordLifetime: %s", err)
		}

		n.IpnsRepub.RecordLifetime = d
	}

Jeromy's avatar
Jeromy committed
394 395
	n.Process().Go(n.IpnsRepub.Run)

396 397 398
	return nil
}

399 400 401 402 403 404 405 406 407 408 409 410
// 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 {
411 412 413
	if n.ctx == nil {
		n.ctx = context.TODO()
	}
414 415 416
	return n.ctx
}

417 418
// teardown closes owned children. If any errors occur, this function returns
// the first error.
Brian Tiger Chow's avatar
Brian Tiger Chow committed
419
func (n *IpfsNode) teardown() error {
420
	log.Debug("core is shutting down...")
421 422
	// 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
423 424
	var closers []io.Closer

425
	// NOTE: The order that objects are added(closed) matters, if an object
Jeromy's avatar
Jeromy committed
426 427 428 429 430
	// 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
431
	}
432

433 434 435 436
	if n.Exchange != nil {
		closers = append(closers, n.Exchange)
	}

437
	if n.Mounts.Ipfs != nil && !n.Mounts.Ipfs.IsActive() {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
438 439
		closers = append(closers, mount.Closer(n.Mounts.Ipfs))
	}
440
	if n.Mounts.Ipns != nil && !n.Mounts.Ipns.IsActive() {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
441 442 443
		closers = append(closers, mount.Closer(n.Mounts.Ipns))
	}

Jeromy's avatar
Jeromy committed
444 445 446 447
	if dht, ok := n.Routing.(*dht.IpfsDHT); ok {
		closers = append(closers, dht.Process())
	}

Jeromy's avatar
Jeromy committed
448 449 450 451
	if n.Blocks != nil {
		closers = append(closers, n.Blocks)
	}

Jeromy's avatar
Jeromy committed
452 453
	if n.Bootstrapper != nil {
		closers = append(closers, n.Bootstrapper)
454 455
	}

Jeromy's avatar
Jeromy committed
456 457
	if n.PeerHost != nil {
		closers = append(closers, n.PeerHost)
458
	}
459

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

463
	var errs []error
464
	for _, closer := range closers {
465 466
		if err := closer.Close(); err != nil {
			errs = append(errs, err)
467 468 469 470
		}
	}
	if len(errs) > 0 {
		return errs[0]
Brian Tiger Chow's avatar
Brian Tiger Chow committed
471 472
	}
	return nil
Brian Tiger Chow's avatar
Brian Tiger Chow committed
473 474
}

Brian Tiger Chow's avatar
Brian Tiger Chow committed
475
func (n *IpfsNode) OnlineMode() bool {
476 477 478 479 480 481
	switch n.mode {
	case onlineMode:
		return true
	default:
		return false
	}
Brian Tiger Chow's avatar
Brian Tiger Chow committed
482 483
}

484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503
func (n *IpfsNode) SetLocal(isLocal bool) {
	if isLocal {
		n.mode = localMode
	}
	n.localModeSet = true
}

func (n *IpfsNode) LocalMode() bool {
	if !n.localModeSet {
		// programmer error should not happen
		panic("local mode not set")
	}
	switch n.mode {
	case localMode:
		return true
	default:
		return false
	}
}

504
func (n *IpfsNode) Bootstrap(cfg BootstrapConfig) error {
505 506

	// TODO what should return value be when in offlineMode?
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
507 508 509 510
	if n.Routing == nil {
		return nil
	}

511 512 513 514 515 516 517
	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
518
		cfg.BootstrapPeers = func() []pstore.PeerInfo {
519
			ps, err := n.loadBootstrapPeers()
520
			if err != nil {
521
				log.Warning("failed to parse bootstrap peers from config")
522 523 524 525 526 527 528 529 530
				return nil
			}
			return ps
		}
	}

	var err error
	n.Bootstrapper, err = Bootstrap(n, cfg)
	return err
531 532
}

533 534
func (n *IpfsNode) loadID() error {
	if n.Identity != "" {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
535
		return errors.New("identity already loaded")
536 537
	}

538 539 540 541 542 543
	cfg, err := n.Repo.Config()
	if err != nil {
		return err
	}

	cid := cfg.Identity.PeerID
544
	if cid == "" {
545
		return errors.New("identity was not set in config (was 'ipfs init' run?)")
546 547
	}
	if len(cid) == 0 {
548
		return errors.New("no peer ID in config! (was 'ipfs init' run?)")
549 550
	}

551 552 553
	n.Identity = peer.ID(b58.Decode(cid))
	return nil
}
554

555 556 557 558 559 560 561 562
func (n *IpfsNode) GetKey(name string) (ic.PrivKey, error) {
	if name == "self" {
		return n.PrivateKey, nil
	} else {
		return n.Repo.Keystore().Get(name)
	}
}

Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
563
func (n *IpfsNode) LoadPrivateKey() error {
564
	if n.Identity == "" || n.Peerstore == nil {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
565
		return errors.New("loaded private key out of order.")
566 567
	}

568
	if n.PrivateKey != nil {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
569
		return errors.New("private key already loaded")
570 571
	}

572 573 574 575 576 577
	cfg, err := n.Repo.Config()
	if err != nil {
		return err
	}

	sk, err := loadPrivateKey(&cfg.Identity, n.Identity)
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
578
	if err != nil {
579
		return err
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
580
	}
581

582 583
	n.PrivateKey = sk
	n.Peerstore.AddPrivKey(n.Identity, n.PrivateKey)
Jeromy's avatar
Jeromy committed
584 585 586 587
	n.Peerstore.AddPubKey(n.Identity, sk.GetPublic())
	return nil
}

Jeromy's avatar
Jeromy committed
588
func (n *IpfsNode) loadBootstrapPeers() ([]pstore.PeerInfo, error) {
589 590 591 592 593 594
	cfg, err := n.Repo.Config()
	if err != nil {
		return nil, err
	}

	parsed, err := cfg.BootstrapPeers()
595 596 597 598 599 600
	if err != nil {
		return nil, err
	}
	return toPeerInfos(parsed), nil
}

Jeromy's avatar
Jeromy committed
601
func (n *IpfsNode) loadFilesRoot() error {
Jeromy's avatar
Jeromy committed
602
	dsk := ds.NewKey("/local/filesroot")
Jeromy's avatar
Jeromy committed
603 604
	pf := func(ctx context.Context, c *cid.Cid) error {
		return n.Repo.Datastore().Put(dsk, c.Bytes())
Jeromy's avatar
Jeromy committed
605 606
	}

607
	var nd *merkledag.ProtoNode
Jeromy's avatar
Jeromy committed
608 609 610 611
	val, err := n.Repo.Datastore().Get(dsk)

	switch {
	case err == ds.ErrNotFound || val == nil:
612
		nd = ft.EmptyDirNode()
Jeromy's avatar
Jeromy committed
613 614 615 616 617
		_, err := n.DAG.Add(nd)
		if err != nil {
			return fmt.Errorf("failure writing to dagstore: %s", err)
		}
	case err == nil:
Jeromy's avatar
Jeromy committed
618 619 620 621 622
		c, err := cid.Cast(val.([]byte))
		if err != nil {
			return err
		}

623
		rnd, err := n.DAG.Get(n.Context(), c)
Jeromy's avatar
Jeromy committed
624 625 626
		if err != nil {
			return fmt.Errorf("error loading filesroot from DAG: %s", err)
		}
627 628 629 630 631 632 633

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

		nd = pbnd
Jeromy's avatar
Jeromy committed
634 635 636 637 638 639 640 641 642 643 644 645 646
	default:
		return err
	}

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

	n.FilesRoot = mr
	return nil
}

Jeromy's avatar
Jeromy committed
647 648 649
// SetupOfflineRouting loads the local nodes private key and
// uses it to instantiate a routing system in offline mode.
// This is primarily used for offline ipns modifications.
Jeromy's avatar
Jeromy committed
650
func (n *IpfsNode) SetupOfflineRouting() error {
651 652 653 654
	if n.Routing != nil {
		// Routing was already set up
		return nil
	}
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
655
	err := n.LoadPrivateKey()
Jeromy's avatar
Jeromy committed
656 657 658 659 660
	if err != nil {
		return err
	}

	n.Routing = offroute.NewOfflineRouter(n.Repo.Datastore(), n.PrivateKey)
661

662 663 664 665 666 667
	size, err := n.getCacheSize()
	if err != nil {
		return err
	}

	n.Namesys = namesys.NewNameSystem(n.Routing, n.Repo.Datastore(), size)
668

669
	return nil
670 671 672 673
}

func loadPrivateKey(cfg *config.Identity, id peer.ID) (ic.PrivKey, error) {
	sk, err := cfg.DecodePrivateKey("passphrase todo!")
674 675 676
	if err != nil {
		return nil, err
	}
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
677

678 679 680 681
	id2, err := peer.IDFromPrivateKey(sk)
	if err != nil {
		return nil, err
	}
682

683 684
	if id2 != id {
		return nil, fmt.Errorf("private key in config does not match id: %s != %s", id, id2)
685 686
	}

687
	return sk, nil
688
}
689

690
func listenAddresses(cfg *config.Config) ([]ma.Multiaddr, error) {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
691 692 693
	var listen []ma.Multiaddr
	for _, addr := range cfg.Addresses.Swarm {
		maddr, err := ma.NewMultiaddr(addr)
694
		if err != nil {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
695
			return nil, fmt.Errorf("Failure to parse config.Addresses.Swarm: %s", cfg.Addresses.Swarm)
696
		}
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
697
		listen = append(listen, maddr)
698 699 700 701
	}

	return listen, nil
}
702

Kevin Atkinson's avatar
Kevin Atkinson committed
703 704 705 706 707
type ConstructPeerHostOpts struct {
	DisableNatPortMap bool
}

type HostOption func(ctx context.Context, id peer.ID, ps pstore.Peerstore, bwr metrics.Reporter, fs []*net.IPNet, tpt smux.Transport, protc ipnet.Protector, opts *ConstructPeerHostOpts) (p2phost.Host, error)
Jeromy's avatar
Jeromy committed
708 709 710

var DefaultHostOption HostOption = constructPeerHost

711
// isolates the complex initialization steps
Kevin Atkinson's avatar
Kevin Atkinson committed
712
func constructPeerHost(ctx context.Context, id peer.ID, ps pstore.Peerstore, bwr metrics.Reporter, fs []*net.IPNet, tpt smux.Transport, protec ipnet.Protector, opts *ConstructPeerHostOpts) (p2phost.Host, error) {
713 714

	// no addresses to begin with. we'll start later.
Jakub Sztandera's avatar
Jakub Sztandera committed
715
	swrm, err := swarm.NewSwarmWithProtector(ctx, nil, id, ps, protec, tpt, bwr)
716
	if err != nil {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
717
		return nil, err
718
	}
719

720 721
	network := (*swarm.Network)(swrm)

722 723 724 725
	for _, f := range fs {
		network.Swarm().Filters.AddDialFilter(f)
	}

Kevin Atkinson's avatar
Kevin Atkinson committed
726 727 728 729 730 731
	hostOpts := []interface{}{bwr}
	if !opts.DisableNatPortMap {
		hostOpts = append(hostOpts, p2pbhost.NATPortMap)
	}

	host := p2pbhost.New(network, hostOpts...)
Jeromy's avatar
Jeromy committed
732

733 734 735 736 737 738 739
	return host, nil
}

// startListening on the network addresses
func startListening(ctx context.Context, host p2phost.Host, cfg *config.Config) error {
	listenAddrs, err := listenAddresses(cfg)
	if err != nil {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
740
		return err
741 742
	}

743
	// make sure we error out if our config does not have addresses we can use
744
	log.Debugf("Config.Addresses.Swarm:%s", listenAddrs)
745
	filteredAddrs := addrutil.FilterUsableAddrs(listenAddrs)
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
746
	log.Debugf("Config.Addresses.Swarm:%s (filtered)", filteredAddrs)
747
	if len(filteredAddrs) < 1 {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
748
		return fmt.Errorf("addresses in config not usable: %s", listenAddrs)
749 750
	}

751 752 753
	// Actually start listening:
	if err := host.Network().Listen(filteredAddrs...); err != nil {
		return err
754 755
	}

756
	// list out our addresses
757
	addrs, err := host.Network().InterfaceListenAddresses()
758
	if err != nil {
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
759
		return err
760
	}
Juan Batiz-Benet's avatar
Juan Batiz-Benet committed
761
	log.Infof("Swarm listening at: %s", addrs)
762
	return nil
763
}
764

Jeromy's avatar
Jeromy committed
765
func constructDHTRouting(ctx context.Context, host p2phost.Host, dstore repo.Datastore) (routing.IpfsRouting, error) {
766
	dhtRouting := dht.NewDHT(ctx, host, dstore)
767
	dhtRouting.Validator[IpnsValidatorTag] = namesys.IpnsRecordValidator
768
	dhtRouting.Selector[IpnsValidatorTag] = namesys.IpnsSelectorFunc
769 770
	return dhtRouting, nil
}
Jeromy's avatar
Jeromy committed
771

Jeromy's avatar
Jeromy committed
772 773 774 775 776 777 778
func constructClientDHTRouting(ctx context.Context, host p2phost.Host, dstore repo.Datastore) (routing.IpfsRouting, error) {
	dhtRouting := dht.NewDHTClient(ctx, host, dstore)
	dhtRouting.Validator[IpnsValidatorTag] = namesys.IpnsRecordValidator
	dhtRouting.Selector[IpnsValidatorTag] = namesys.IpnsSelectorFunc
	return dhtRouting, nil
}

Jeromy's avatar
Jeromy committed
779
type RoutingOption func(context.Context, p2phost.Host, repo.Datastore) (routing.IpfsRouting, error)
Jeromy's avatar
Jeromy committed
780

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

783
var DHTOption RoutingOption = constructDHTRouting
Jeromy's avatar
Jeromy committed
784
var DHTClientOption RoutingOption = constructClientDHTRouting
785
var NilRouterOption RoutingOption = nilrouting.ConstructNilRouting