Support a message to be passed and logged from client to server (#67)

* Support a message to be passed and logged from client to server

1 files changed, 5 insertions, 2 deletions

diff --git a/cmd/cashier/main.go b/cmd/cashier/main.go
index f2d3bbe..78141c2 100644
--- a/cmd/cashier/main.go
+++ b/cmd/cashier/main.go
@@ -50,12 +50,15 @@ func main() {
 	fmt.Print("Enter token: ")
 	var token string
 	fmt.Scanln(&token)
+	var message string
+	fmt.Print("Enter message: ")
+	fmt.Scanln(&message)
 
 	var cert *ssh.Certificate
 	if *useGRPC {
-		cert, err = client.RPCSign(pub, token, c)
+		cert, err = client.RPCSign(pub, token, message, c)
 	} else {
-		cert, err = client.Sign(pub, token, c)
+		cert, err = client.Sign(pub, token, message, c)
 	}
 	if err != nil {
 		log.Fatalln(err)