aboutsummaryrefslogtreecommitdiff
path: root/generate.go
blob: b289b9955f6672be9dbe6f0aef5ef8b0fca2db59 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
package main

import (
	"encoding"
	"flag"
	"os"

	"dim13.org/signify/ask"
	"dim13.org/signify/file"
	"dim13.org/signify/key"
)

// Usage: signify -G [-n] [-c comment] -p pubkey -s seckey

func generate(args []string) error {
	opts := flag.NewFlagSet("generate", flag.ExitOnError)
	var (
		nopass  = opts.Bool("n", false, "No key passphrase")
		comment = opts.String("c", "signify", "Comment")
		pubFile = opts.String("p", "", "Public key file (required)")
		encFile = opts.String("s", "", "Secret key file (required)")
	)
	opts.Parse(args)
	if *pubFile == "" || *encFile == "" {
		opts.Usage()
		return nil
	}
	if err := file.CheckNames(*pubFile, *encFile); err != nil {
		return err
	}

	var deriver key.KeyDeriver
	if !*nopass {
		deriver = ask.Confirmed{}
	}

	pubKey, encKey, err := key.NewKey(deriver)
	if err != nil {
		return err
	}

	if err := writeFile(*encFile, *comment+" secret key", encKey, file.EncMode); err != nil {
		return err
	}
	if err := writeFile(*pubFile, *comment+" public key", pubKey, file.PubMode); err != nil {
		return err
	}
	return nil
}

func writeFile(fname, comment string, v encoding.BinaryMarshaler, mode os.FileMode) error {
	raw, err := v.MarshalBinary()
	if err != nil {
		return err
	}
	block := &file.Block{
		Comment: comment,
		Bytes:   raw,
	}
	return file.EncodeFile(fname, mode, block)
}