From 7ddf3836d07665938c202a4c8f27daeec0571242 Mon Sep 17 00:00:00 2001
From: Henry <cryptix@riseup.net>
Date: Mon, 20 Oct 2014 15:33:46 +0200
Subject: [PATCH] implemented manual check and update (with signature
 verification)

---
 cmd/ipfs/equinox.yaml   |  6 +++++
 core/commands/update.go | 49 ++++++++++++++++++++++++++++++++++++++---
 daemon/daemon.go        |  2 ++
 updates/updates.go      | 31 +++++++++++++++++++++-----
 4 files changed, 79 insertions(+), 9 deletions(-)
 create mode 100644 cmd/ipfs/equinox.yaml

diff --git a/cmd/ipfs/equinox.yaml b/cmd/ipfs/equinox.yaml
new file mode 100644
index 000000000..9e764e05c
--- /dev/null
+++ b/cmd/ipfs/equinox.yaml
@@ -0,0 +1,6 @@
+---
+equinox-account: CHANGEME
+equinox-secret: CHANGEME
+equinox-app: CHANGEME
+channel: stable
+private-key: equinox-priv
diff --git a/core/commands/update.go b/core/commands/update.go
index f2b72c969..16e2f82d4 100644
--- a/core/commands/update.go
+++ b/core/commands/update.go
@@ -2,21 +2,64 @@ package commands
 
 import (
 	"errors"
+	"fmt"
 	"io"
+	"os"
 
 	"github.com/jbenet/go-ipfs/core"
+	"github.com/jbenet/go-ipfs/updates"
 )
 
 // UpdateApply applys an update of the ipfs binary and shuts down the node if successful
 func UpdateApply(n *core.IpfsNode, args []string, opts map[string]interface{}, out io.Writer) error {
-	return errors.New("TODOUpdateApply")
+	fmt.Fprintln(out, "Current Version:", updates.Version)
+	u, err := updates.CheckForUpdate()
+	if err != nil {
+		return err
+	}
+
+	if u == nil {
+		fmt.Fprintln(out, "No update available")
+		return nil
+	}
+	fmt.Fprintln(out, "New Version:", u.Version)
+
+	if err = updates.AbleToApply(); err != nil {
+		return fmt.Errorf("Can't apply update: %v", err)
+	}
+
+	if err, errRecover := u.Update(); err != nil {
+		err = fmt.Errorf("Update failed: %v\n", err)
+		if errRecover != nil {
+			err = fmt.Errorf("%s\nRecovery failed! Cause: %v\nYou may need to recover manually", err, errRecover)
+		}
+		fmt.Fprint(out, err.Error())
+		return err
+	}
+
+	fmt.Fprintln(out, "Updated applied! Shutting down.")
+	os.Exit(0)
+	return nil
 }
 
 // UpdateCheck checks wether there is an update available
 func UpdateCheck(n *core.IpfsNode, args []string, opts map[string]interface{}, out io.Writer) error {
-	return errors.New("TODOUpdateCheck")
+	fmt.Fprintln(out, "Current Version:", updates.Version)
+	u, err := updates.CheckForUpdate()
+	if err != nil {
+		return err
+	}
+
+	if u == nil {
+		fmt.Fprintln(out, "No update available")
+		return nil
+	}
+
+	fmt.Fprintln(out, "New Version:", u.Version)
+	return nil
 }
 
+// UpdateLog lists the version available online
 func UpdateLog(n *core.IpfsNode, args []string, opts map[string]interface{}, out io.Writer) error {
-	return errors.New("TODOUpdateLog")
+	return errors.New("Not yet implemented")
 }
diff --git a/daemon/daemon.go b/daemon/daemon.go
index 9e3ef370d..b4241e16b 100644
--- a/daemon/daemon.go
+++ b/daemon/daemon.go
@@ -141,6 +141,8 @@ func (dl *DaemonListener) handleConnection(conn manet.Conn) {
 		err = commands.Log(dl.node, command.Args, command.Opts, conn)
 	case "unpin":
 		err = commands.Unpin(dl.node, command.Args, command.Opts, conn)
+	case "updateApply":
+		err = commands.UpdateApply(dl.node, command.Args, command.Opts, conn)
 	default:
 		err = fmt.Errorf("Invalid Command: '%s'", command.Command)
 	}
diff --git a/updates/updates.go b/updates/updates.go
index 4b6480ed6..754f559ab 100644
--- a/updates/updates.go
+++ b/updates/updates.go
@@ -1,18 +1,26 @@
 package updates
 
 import (
+	"fmt"
 	"os"
 
+	u "github.com/jbenet/go-ipfs/util"
+
 	"github.com/jbenet/go-ipfs/Godeps/_workspace/src/github.com/coreos/go-semver/semver"
 	"github.com/jbenet/go-ipfs/Godeps/_workspace/src/github.com/inconshreveable/go-update"
 	"github.com/jbenet/go-ipfs/Godeps/_workspace/src/github.com/inconshreveable/go-update/check"
-	u "github.com/jbenet/go-ipfs/util"
 )
 
 const (
-	Version           = "0.1.0" // actual current application's version literal
-	UpdateEndpointURL = "https://api.equinox.io/1/Updates"
-	UpdateAppID       = "ap_ywkPmAR40q4EfdikN9Jh2hgIHi"
+	// Version is the current application's version literal
+	Version = "0.1.1"
+
+	updateEndpointURL = "https://api.equinox.io/1/Updates"
+	updateAppID       = "CHANGEME"
+
+	updatePubKey = `-----BEGIN RSA PUBLIC KEY-----
+CHANGEME
+-----END RSA PUBLIC KEY-----`
 )
 
 var log = u.Logger("updates")
@@ -32,12 +40,23 @@ func parseVersion() (*semver.Version, error) {
 	return semver.NewVersion(Version)
 }
 
+// CheckForUpdate checks the equinox.io api if there is an update available
 func CheckForUpdate() (*check.Result, error) {
 	param := check.Params{
 		AppVersion: Version,
-		AppId:      UpdateAppID,
+		AppId:      updateAppID,
 		Channel:    "stable",
 	}
 
-	return param.CheckForUpdate(UpdateEndpointURL, update.New())
+	up, err := update.New().VerifySignatureWithPEM([]byte(updatePubKey))
+	if err != nil {
+		return nil, fmt.Errorf("Failed to parse public key: %v", err)
+	}
+
+	return param.CheckForUpdate(updateEndpointURL, up)
+}
+
+// AbleToApply cheks if the running process is able to update itself
+func AbleToApply() error {
+	return update.New().CanUpdate()
 }