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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
|
// Copyright 2018 Google LLC. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package mysql
import (
"bytes"
"database/sql"
"encoding/gob"
"fmt"
"time"
"github.com/google/trillian"
"github.com/google/trillian/storage/mysql/mysqlpb"
"google.golang.org/protobuf/types/known/anypb"
"google.golang.org/protobuf/types/known/durationpb"
"google.golang.org/protobuf/types/known/timestamppb"
)
// toMillisSinceEpoch converts a timestamp into milliseconds since epoch
func toMillisSinceEpoch(t time.Time) int64 {
return t.UnixNano() / 1000000
}
// fromMillisSinceEpoch converts
func fromMillisSinceEpoch(ts int64) time.Time {
return time.Unix(0, ts*1000000)
}
// setNullStringIfValid assigns src to dest if src is Valid.
func setNullStringIfValid(src sql.NullString, dest *string) {
if src.Valid {
*dest = src.String
}
}
// row defines a common interface between sql.Row and sql.Rows(!)
type row interface {
Scan(dest ...interface{}) error
}
// readTree takes a sql row and returns a tree
func readTree(r row) (*trillian.Tree, error) {
tree := &trillian.Tree{}
// Enums and Datetimes need an extra conversion step
var treeState, treeType, hashStrategy, hashAlgorithm, signatureAlgorithm string
var createMillis, updateMillis, maxRootDurationMillis int64
var displayName, description sql.NullString
var privateKey, publicKey []byte
var deleted sql.NullBool
var deleteMillis sql.NullInt64
err := r.Scan(
&tree.TreeId,
&treeState,
&treeType,
&hashStrategy,
&hashAlgorithm,
&signatureAlgorithm,
&displayName,
&description,
&createMillis,
&updateMillis,
&privateKey,
&publicKey,
&maxRootDurationMillis,
&deleted,
&deleteMillis,
)
if err != nil {
return nil, err
}
setNullStringIfValid(displayName, &tree.DisplayName)
setNullStringIfValid(description, &tree.Description)
// Convert all things!
if ts, ok := trillian.TreeState_value[treeState]; ok {
tree.TreeState = trillian.TreeState(ts)
} else {
return nil, fmt.Errorf("unknown TreeState: %v", treeState)
}
if tt, ok := trillian.TreeType_value[treeType]; ok {
tree.TreeType = trillian.TreeType(tt)
} else {
return nil, fmt.Errorf("unknown TreeType: %v", treeType)
}
if hashStrategy != "RFC6962_SHA256" {
return nil, fmt.Errorf("unknown HashStrategy: %v", hashStrategy)
}
// Let's make sure we didn't mismatch any of the casts above
ok := tree.TreeState.String() == treeState &&
tree.TreeType.String() == treeType
if !ok {
return nil, fmt.Errorf(
"mismatched enum: tree = %v, enums = [%v, %v, %v, %v, %v]",
tree,
treeState, treeType, hashStrategy, hashAlgorithm, signatureAlgorithm)
}
tree.CreateTime = timestamppb.New(fromMillisSinceEpoch(createMillis))
if err := tree.CreateTime.CheckValid(); err != nil {
return nil, fmt.Errorf("failed to parse create time: %w", err)
}
tree.UpdateTime = timestamppb.New(fromMillisSinceEpoch(updateMillis))
if err := tree.UpdateTime.CheckValid(); err != nil {
return nil, fmt.Errorf("failed to parse update time: %w", err)
}
tree.MaxRootDuration = durationpb.New(time.Duration(maxRootDurationMillis * int64(time.Millisecond)))
tree.Deleted = deleted.Valid && deleted.Bool
if tree.Deleted && deleteMillis.Valid {
tree.DeleteTime = timestamppb.New(fromMillisSinceEpoch(deleteMillis.Int64))
if err := tree.DeleteTime.CheckValid(); err != nil {
return nil, fmt.Errorf("failed to parse delete time: %w", err)
}
}
// We're going to try to interpret PublicKey as storageSettings, but it could be a
// public key from a really old tree, or an empty column from a tree created in the
// period between Trillian key material being removed and this column being used for
// storing settings.
buff := bytes.NewBuffer(publicKey)
dec := gob.NewDecoder(buff)
ss := &storageSettings{}
var o *mysqlpb.StorageOptions
if err := dec.Decode(ss); err != nil {
// If there are no storageSettings then this tree was created before settings
// were supported, and thus we have to populate the settings with the oldest
// settings for features.
o = &mysqlpb.StorageOptions{
SubtreeRevisions: true,
}
} else {
o = &mysqlpb.StorageOptions{
SubtreeRevisions: ss.Revisioned,
}
}
tree.StorageSettings, err = anypb.New(o)
if err != nil {
return nil, fmt.Errorf("failed to put StorageSettings into tree: %w", err)
}
return tree, nil
}
|
