add message signing again

This adds again the possibility to sign messages through the API.

1 files changed, 33 insertions, 20 deletions

diff --git a/main.go b/main.go
index f5971eb..d7e152c 100644
--- a/main.go
+++ b/main.go
@@ -1,7 +1,11 @@
 package main
 
 import (
+  "crypto"
+  "encoding/base64"
   "fmt"
+  "io"
+  "io/ioutil"
   "os"
   "path/filepath"
 
@@ -12,19 +16,6 @@ var (
   EmptyByteArray = make([]byte, 0)
 )
 
-//const (
-//  RsaLowerLength = 2048
-//  RsaUpperLength = 4096
-//  TypeLabelRSA   = "RSA PRIVATE KEY"
-//  TypeLabelECDSA = "EC PRIVATE KEY"
-//  TypeLabelCSR   = "CERTIFICATE REQUEST"
-//  TypeLabelPubKey = "PUBLIC KEY"
-//)
-//
-//var (
-//  EcdsaLength = []int{224, 256, 384, 521}
-//)
-//
 func main() {
   if len(os.Args) == 1 {
     crash_with_help(1, "No module selected!")
@@ -32,12 +23,12 @@ func main() {
   switch os.Args[1] {
   case "create-private":   create_private_key()
   case "create-public":    create_public_key()
-  case "help":             print_modules()
-//  case "info":             info_on_file()
-//  case "sign-input":       sign_input()
+  case "sign-input":       sign_input()
 //  case "verify-signature": verify_signature()
 //  case "create-cert-sign": create_sign_request()
 //  case "sign-request":     sign_request()
+  case "help":             print_modules()
+//  case "info":             info_on_file()
   default: crash_with_help(1, "Command not supported!")
   }
 }
@@ -78,24 +69,46 @@ func create_public_key() {
   if err != nil { os.Exit(2) }
 }
 
+func sign_input() {
+  fs := NewFlags("sign-input")
+  fs.AddPrivateKey()
+  fs.AddOutput()
+  fs.AddInput()
+  err := fs.Parse(program_args())
+  if err != nil { os.Exit(2) }
+
+  message, err := ioutil.ReadAll(fs.Flags.Input)
+  if err != nil { crash_with_help(2, "Error reading input: %s", err) }
+  signature, err := fs.Flags.PrivateKey.Sign(message, crypto.SHA256)
+  if err != nil { crash_with_help(2, "Could not compute signature: %s", err) }
+  _, err = io.WriteString(fs.Flags.Output, base64.StdEncoding.EncodeToString(signature))
+  if err != nil { crash_with_help(2, "Could not write to output: %s", err) }
+
+  // if we print to stderr, send a final line break to make the output nice
+  if fs.Flags.Output == os.Stdout {
+    // we can ignore the result, as either Stdout did work or not
+    _, _ = io.WriteString(fs.Flags.Output, "\n")
+  }
+}
+
 // print the module help
 func print_modules() {
   fmt.Printf(`Usage: %s command args
 where 'command' is one of:
     create-private    create a new private key
     create-public     create a public key from a private one
+    sign-input        sign a message with a private key
+    verify-signature  verify a signature
     create-cert-sign  create a new certificate sign request
+    sign-request      sign a certificate request
     help              show this help
     info              get info on a file
-    sign-request      sign a certificate request
-    sign-input        sign a message with a private key
-    verify-signature  verify a signature
 `, filepath.Base(os.Args[0]))
   fmt.Println()
 }
 
 // crash and provide a helpful message
-func crash_with_help(code int, message string) {
+func crash_with_help(code int, message string, args ...interface{}) {
   fmt.Fprintln(os.Stderr, message)
   print_modules()
   os.Exit(code)