Install public key on dryad

This patch changes interface and communication between boruta server and
dryads. Key is generated on boruta server, public part is installed on
dryad and private part is stored internally. It is a preparation for
using user's public keys provided by an external service.

Change-Id: Ic6fb087aba02553c6b2b8f7cc13cc6bd67eff36a
Signed-off-by: Aleksander Mistewicz <a.mistewicz@samsung.com>

1 files changed, 11 insertions, 3 deletions

diff --git a/workers/workers.go b/workers/workers.go
index da22336..b77a01a 100644
--- a/workers/workers.go
+++ b/workers/workers.go
@@ -18,6 +18,7 @@
 package workers
 
 import (
+	"crypto/rand"
 	"crypto/rsa"
 	"fmt"
 	"math"
@@ -31,6 +32,10 @@ import (
 // UUID denotes a key in Capabilities where WorkerUUID is stored.
 const UUID string = "UUID"
 
+// sizeRSA is a length of the RSA key.
+// It is a variable for test purposes.
+var sizeRSA = 4096
+
 // mapWorker is used by WorkerList to store all
 // (public and private) structures representing Worker.
 type mapWorker struct {
@@ -440,8 +445,7 @@ func (wl *WorkerList) setState(worker WorkerUUID, state WorkerState) error {
 	return nil
 }
 
-// prepareKey delegates key generation to Dryad and sets up generated key in the
-// worker. In case of any failure it returns an error.
+// prepareKey generates key, installs public part on worker and stores private part in WorkerList.
 func (wl *WorkerList) prepareKey(worker WorkerUUID) error {
 	addr, err := wl.GetWorkerAddr(worker)
 	if err != nil {
@@ -453,7 +457,11 @@ func (wl *WorkerList) prepareKey(worker WorkerUUID) error {
 		return err
 	}
 	defer client.Close()
-	key, err := client.Prepare()
+	key, err := rsa.GenerateKey(rand.Reader, sizeRSA)
+	if err != nil {
+		return err
+	}
+	err = client.Prepare(&key.PublicKey)
 	if err != nil {
 		return err
 	}