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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
package store
import (
"encoding/json"
"fmt"
"time"
"github.com/nsheridan/cashier/lib"
"github.com/nsheridan/cashier/server/config"
"golang.org/x/crypto/ssh"
)
// New returns a new configured database.
func New(c config.Database) (CertStorer, error) {
switch c["type"] {
case "mysql", "sqlite":
return newSQLStore(c)
case "mem":
return newMemoryStore(), nil
}
return nil, fmt.Errorf("unable to create store with driver %s", c["type"])
}
// CertStorer records issued certs in a persistent store for audit and
// revocation purposes.
type CertStorer interface {
Get(id string) (*CertRecord, error)
SetRecord(record *CertRecord) error
List(includeExpired bool) ([]*CertRecord, error)
Revoke(id []string) error
GetRevoked() ([]*CertRecord, error)
Close() error
}
// A CertRecord is a representation of a ssh certificate used by a CertStorer.
type CertRecord struct {
ID int `json:"-" db:"id"`
KeyID string `json:"key_id" db:"key_id"`
Principals StringSlice `json:"principals" db:"principals"`
CreatedAt time.Time `json:"created_at" db:"created_at"`
Expires time.Time `json:"expires" db:"expires_at"`
Revoked bool `json:"revoked" db:"revoked"`
Raw string `json:"-" db:"raw_key"`
Message string `json:"message" db:"message"`
}
// MarshalJSON implements the json.Marshaler interface for the CreatedAt and
// Expires fields.
// The resulting string looks like "2017-04-11 10:00:00 +0000"
func (c *CertRecord) MarshalJSON() ([]byte, error) {
type Alias CertRecord
f := "2006-01-02 15:04:05 -0700"
return json.Marshal(&struct {
*Alias
CreatedAt string `json:"created_at"`
Expires string `json:"expires"`
}{
Alias: (*Alias)(c),
CreatedAt: c.CreatedAt.Format(f),
Expires: c.Expires.Format(f),
})
}
func parseTime(t uint64) time.Time {
return time.Unix(int64(t), 0)
}
// MakeRecord converts a Certificate to a CertRecord
func MakeRecord(cert *ssh.Certificate) *CertRecord {
return &CertRecord{
KeyID: cert.KeyId,
Principals: StringSlice(cert.ValidPrincipals),
CreatedAt: parseTime(cert.ValidAfter),
Expires: parseTime(cert.ValidBefore),
Raw: string(lib.GetPublicKey(cert)),
}
}
|
