From 0d760a94f84d23a13b950586bfb92af120033758 Mon Sep 17 00:00:00 2001
From: Juan Batiz-Benet <juan@benet.ai>
Date: Sun, 21 Dec 2014 13:46:49 -0800
Subject: [PATCH] core: cleanup initIdentity

---
 core/core.go      | 67 +++++++++++++++++++++++++++--------------------
 peer/peerstore.go |  4 +++
 2 files changed, 43 insertions(+), 28 deletions(-)

diff --git a/core/core.go b/core/core.go
index 4dd06ac34..03ff5c459 100644
--- a/core/core.go
+++ b/core/core.go
@@ -97,13 +97,18 @@ func NewIpfsNode(ctx context.Context, cfg *config.Config, online bool) (n *IpfsN
 		return nil, debugerror.Wrap(err)
 	}
 
-	// setup peerstore + local peer identity
-	n.Peerstore = peer.NewPeerstore()
-	n.Identity, n.PrivateKey, err = initIdentity(&n.Config.Identity, n.Peerstore, online)
+	// setup local peer identity
+	n.Identity, n.PrivateKey, err = initIdentity(&n.Config.Identity, online)
 	if err != nil {
 		return nil, debugerror.Wrap(err)
 	}
 
+	// setup Peerstore
+	n.Peerstore = peer.NewPeerstore()
+	if n.PrivateKey != nil {
+		n.Peerstore.AddPrivKey(n.Identity, n.PrivateKey)
+	}
+
 	blockstore, err := bstore.WriteCached(bstore.NewBlockstore(n.Datastore), kSizeBlockstoreWriteCache)
 	n.Exchange = offline.Exchange(blockstore)
 
@@ -178,41 +183,47 @@ func (n *IpfsNode) OnlineMode() bool {
 	return n.onlineMode
 }
 
-func initIdentity(cfg *config.Identity, peers peer.Peerstore, online bool) (p peer.ID, sk ic.PrivKey, err error) {
+func initIdentity(cfg *config.Identity, online bool) (peer.ID, ic.PrivKey, error) {
+
 	if cfg.PeerID == "" {
-		err = debugerror.New("Identity was not set in config (was ipfs init run?)")
-		return
+		return "", nil, debugerror.New("Identity was not set in config (was ipfs init run?)")
 	}
 
 	if len(cfg.PeerID) == 0 {
-		err = debugerror.New("No peer ID in config! (was ipfs init run?)")
-		return
+		return "", nil, debugerror.New("No peer ID in config! (was ipfs init run?)")
 	}
 
 	id := peer.ID(b58.Decode(cfg.PeerID))
 
 	// when not online, don't need to parse private keys (yet)
-	if online {
-		var sk2 ic.PrivKey
-		sk2, err = cfg.DecodePrivateKey("passphrase todo!")
-		if err != nil {
-			return
-		}
-
-		var id2 peer.ID
-		id2, err = peer.IDFromPrivateKey(sk2)
-		if err != nil {
-			return
-		}
-		if id2 != id {
-			err = fmt.Errorf("private key in config does not match id: %s != %s", id, id2)
-			return
-		}
-
-		sk = sk2
+	if !online {
+		return id, nil, nil
 	}
-	p = id
-	return
+
+	sk, err := loadPrivateKey(cfg, id)
+	if err != nil {
+		return "", nil, err
+	}
+
+	return id, sk, nil
+}
+
+func loadPrivateKey(cfg *config.Identity, id peer.ID) (ic.PrivKey, error) {
+	sk, err := cfg.DecodePrivateKey("passphrase todo!")
+	if err != nil {
+		return nil, err
+	}
+
+	id2, err := peer.IDFromPrivateKey(sk)
+	if err != nil {
+		return nil, err
+	}
+
+	if id2 != id {
+		return nil, fmt.Errorf("private key in config does not match id: %s != %s", id, id2)
+	}
+
+	return sk, nil
 }
 
 func listenAddresses(cfg *config.Config) ([]ma.Multiaddr, error) {
diff --git a/peer/peerstore.go b/peer/peerstore.go
index c79bfed44..46de50f0d 100644
--- a/peer/peerstore.go
+++ b/peer/peerstore.go
@@ -168,6 +168,10 @@ func (kb *keybook) PrivKey(p ID) ic.PrivKey {
 
 func (kb *keybook) AddPrivKey(p ID, sk ic.PrivKey) error {
 
+	if sk == nil {
+		return errors.New("sk is nil (PrivKey)")
+	}
+
 	// check it's correct first
 	if !p.MatchesPrivateKey(sk) {
 		return errors.New("ID does not match PrivateKey")