1
2
3
4
5
6
7 package sha3
8
9 import (
10 "crypto"
11 "crypto/internal/fips140/sha3"
12 "hash"
13 _ "unsafe"
14 )
15
16 func init() {
17 crypto.RegisterHash(crypto.SHA3_224, func() hash.Hash { return New224() })
18 crypto.RegisterHash(crypto.SHA3_256, func() hash.Hash { return New256() })
19 crypto.RegisterHash(crypto.SHA3_384, func() hash.Hash { return New384() })
20 crypto.RegisterHash(crypto.SHA3_512, func() hash.Hash { return New512() })
21 }
22
23
24 func Sum224(data []byte) [28]byte {
25 var out [28]byte
26 h := sha3.New224()
27 h.Write(data)
28 h.Sum(out[:0])
29 return out
30 }
31
32
33 func Sum256(data []byte) [32]byte {
34 var out [32]byte
35 h := sha3.New256()
36 h.Write(data)
37 h.Sum(out[:0])
38 return out
39 }
40
41
42 func Sum384(data []byte) [48]byte {
43 var out [48]byte
44 h := sha3.New384()
45 h.Write(data)
46 h.Sum(out[:0])
47 return out
48 }
49
50
51 func Sum512(data []byte) [64]byte {
52 var out [64]byte
53 h := sha3.New512()
54 h.Write(data)
55 h.Sum(out[:0])
56 return out
57 }
58
59
60
61 func SumSHAKE128(data []byte, length int) []byte {
62
63 out := make([]byte, 32)
64 return sumSHAKE128(out, data, length)
65 }
66
67 func sumSHAKE128(out, data []byte, length int) []byte {
68 if len(out) < length {
69 out = make([]byte, length)
70 } else {
71 out = out[:length]
72 }
73 h := sha3.NewShake128()
74 h.Write(data)
75 h.Read(out)
76 return out
77 }
78
79
80
81 func SumSHAKE256(data []byte, length int) []byte {
82
83 out := make([]byte, 64)
84 return sumSHAKE256(out, data, length)
85 }
86
87 func sumSHAKE256(out, data []byte, length int) []byte {
88 if len(out) < length {
89 out = make([]byte, length)
90 } else {
91 out = out[:length]
92 }
93 h := sha3.NewShake256()
94 h.Write(data)
95 h.Read(out)
96 return out
97 }
98
99
100 type SHA3 struct {
101 s sha3.Digest
102 }
103
104
105 func fips140hash_sha3Unwrap(sha3 *SHA3) *sha3.Digest {
106 return &sha3.s
107 }
108
109
110 func New224() *SHA3 {
111 return &SHA3{*sha3.New224()}
112 }
113
114
115 func New256() *SHA3 {
116 return &SHA3{*sha3.New256()}
117 }
118
119
120 func New384() *SHA3 {
121 return &SHA3{*sha3.New384()}
122 }
123
124
125 func New512() *SHA3 {
126 return &SHA3{*sha3.New512()}
127 }
128
129
130 func (s *SHA3) Write(p []byte) (n int, err error) {
131 return s.s.Write(p)
132 }
133
134
135 func (s *SHA3) Sum(b []byte) []byte {
136 return s.s.Sum(b)
137 }
138
139
140 func (s *SHA3) Reset() {
141 s.s.Reset()
142 }
143
144
145 func (s *SHA3) Size() int {
146 return s.s.Size()
147 }
148
149
150 func (s *SHA3) BlockSize() int {
151 return s.s.BlockSize()
152 }
153
154
155 func (s *SHA3) MarshalBinary() ([]byte, error) {
156 return s.s.MarshalBinary()
157 }
158
159
160 func (s *SHA3) AppendBinary(p []byte) ([]byte, error) {
161 return s.s.AppendBinary(p)
162 }
163
164
165 func (s *SHA3) UnmarshalBinary(data []byte) error {
166 return s.s.UnmarshalBinary(data)
167 }
168
169
170 func (d *SHA3) Clone() (hash.Cloner, error) {
171 r := *d
172 return &r, nil
173 }
174
175
176 type SHAKE struct {
177 s sha3.SHAKE
178 }
179
180
181 func NewSHAKE128() *SHAKE {
182 return &SHAKE{*sha3.NewShake128()}
183 }
184
185
186 func NewSHAKE256() *SHAKE {
187 return &SHAKE{*sha3.NewShake256()}
188 }
189
190
191
192
193
194
195 func NewCSHAKE128(N, S []byte) *SHAKE {
196 return &SHAKE{*sha3.NewCShake128(N, S)}
197 }
198
199
200
201
202
203
204 func NewCSHAKE256(N, S []byte) *SHAKE {
205 return &SHAKE{*sha3.NewCShake256(N, S)}
206 }
207
208
209
210
211 func (s *SHAKE) Write(p []byte) (n int, err error) {
212 return s.s.Write(p)
213 }
214
215
216
217
218 func (s *SHAKE) Read(p []byte) (n int, err error) {
219 return s.s.Read(p)
220 }
221
222
223 func (s *SHAKE) Reset() {
224 s.s.Reset()
225 }
226
227
228 func (s *SHAKE) BlockSize() int {
229 return s.s.BlockSize()
230 }
231
232
233 func (s *SHAKE) MarshalBinary() ([]byte, error) {
234 return s.s.MarshalBinary()
235 }
236
237
238 func (s *SHAKE) AppendBinary(p []byte) ([]byte, error) {
239 return s.s.AppendBinary(p)
240 }
241
242
243 func (s *SHAKE) UnmarshalBinary(data []byte) error {
244 return s.s.UnmarshalBinary(data)
245 }
246
View as plain text