luzifer/vault-unseal
1
0
Fork 0
mirror of https://github.com/Luzifer/vault-unseal.git synced 2024-12-22 22:01:20 +00:00
Code Issues Releases Wiki Activity

Use log package for logging to have timestamps

Browse source
This commit is contained in:
Knut Ahlers 2016-03-31 16:00:50 +02:00
parent a206b1d39b
commit 817a128ea3
1 changed files with 10 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
main.go
View file

@ -3,7 +3,7 @@ package main // import "github.com/Jimdo/vault-unseal"
import ( import (
"bytes" "bytes"
"encoding/json" "encoding/json"
"fmt" "log"
"net/http" "net/http"
"os" "os"
"strings" "strings"
@ -22,12 +22,12 @@ var config = struct {
func init() { func init() {
if err := rconfig.Parse(&config); err != nil { if err := rconfig.Parse(&config); err != nil {
fmt.Printf("Unable to parse CLI parameters: %s\n", err) log.Printf("Unable to parse CLI parameters: %s\n", err)
os.Exit(1) os.Exit(1)
} }
if len(config.SealTokensRaw) == 0 { if len(config.SealTokensRaw) == 0 {
fmt.Println("You must provide at least one token.") log.Println("You must provide at least one token.")
os.Exit(1) os.Exit(1)
} }
@ -39,19 +39,19 @@ func main() {
s := sealStatus{} s := sealStatus{}
r, err := http.Get(config.VaultInstance + "/v1/sys/seal-status") r, err := http.Get(config.VaultInstance + "/v1/sys/seal-status")
if err != nil { if err != nil {
fmt.Printf("An error ocurred while reading seal-status: %s\n", err) log.Printf("An error ocurred while reading seal-status: %s\n", err)
os.Exit(1) os.Exit(1)
} }
defer r.Body.Close() defer r.Body.Close()
if err := json.NewDecoder(r.Body).Decode(&s); err != nil { if err := json.NewDecoder(r.Body).Decode(&s); err != nil {
fmt.Printf("Unable to decode seal-status: %s\n", err) log.Printf("Unable to decode seal-status: %s\n", err)
os.Exit(1) os.Exit(1)
} }
if s.Sealed { if s.Sealed {
for _, token := range config.SealTokens { for _, token := range config.SealTokens {
fmt.Printf("Vault instance is sealed (missing %d tokens), trying to unlock...\n", s.T-s.Progress) log.Printf("Vault instance is sealed (missing %d tokens), trying to unlock...\n", s.T-s.Progress)
body := bytes.NewBuffer([]byte{}) body := bytes.NewBuffer([]byte{})
json.NewEncoder(body).Encode(map[string]interface{}{ json.NewEncoder(body).Encode(map[string]interface{}{
"key": token, "key": token,
@ -59,24 +59,24 @@ func main() {
r, _ := http.NewRequest("PUT", config.VaultInstance+"/v1/sys/unseal", body) r, _ := http.NewRequest("PUT", config.VaultInstance+"/v1/sys/unseal", body)
resp, err := http.DefaultClient.Do(r) resp, err := http.DefaultClient.Do(r)
if err != nil { if err != nil {
fmt.Printf("An error ocurred while doing unseal: %s\n", err) log.Printf("An error ocurred while doing unseal: %s\n", err)
os.Exit(1) os.Exit(1)
} }
defer resp.Body.Close() defer resp.Body.Close()
if err := json.NewDecoder(resp.Body).Decode(&s); err != nil { if err := json.NewDecoder(resp.Body).Decode(&s); err != nil {
fmt.Printf("Unable to decode seal-status: %s\n", err) log.Printf("Unable to decode seal-status: %s\n", err)
os.Exit(1) os.Exit(1)
} }
if !s.Sealed { if !s.Sealed {
fmt.Printf("Unseal successfully finished.\n") log.Printf("Unseal successfully finished.\n")
break break
} }
} }
if s.Sealed { if s.Sealed {
fmt.Printf("Vault instance is still sealed (missing %d tokens), I don't have any more tokens.\n", s.T-s.Progress) log.Printf("Vault instance is still sealed (missing %d tokens), I don't have any more tokens.\n", s.T-s.Progress)
} }
} }