-
Notifications
You must be signed in to change notification settings - Fork 9
/
jwe_key_encryption_decryptor.go
108 lines (94 loc) · 3.24 KB
/
jwe_key_encryption_decryptor.go
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
// Copyright 2024 Thales Group
//
// Permission is hereby granted, free of charge, to any person obtaining
// a copy of this software and associated documentation files (the
// "Software"), to deal in the Software without restriction, including
// without limitation the rights to use, copy, modify, merge, publish,
// distribute, sublicense, and/or sell copies of the Software, and to
// permit persons to whom the Software is furnished to do so, subject to
// the following conditions:
//
// The above copyright notice and this permission notice shall be
// included in all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
// LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
// OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
package gose
import (
"crypto/aes"
"crypto/cipher"
"github.com/ThalesGroup/gose/jose"
)
// JweRsaKeyEncryptionDecryptorImpl implements RSA Key Encryption CEK mode.
type JweRsaKeyEncryptionDecryptorImpl struct {
keystore AsymmetricDecryptionKeyStore
}
// Decrypt decrypts the given JWE returning the contained plaintext and any additional authentic associated data.
func (d *JweRsaKeyEncryptionDecryptorImpl) Decrypt(jwe string) (plaintext, aad []byte, err error) {
var jweStruct jose.Jwe
if err = jweStruct.Unmarshal(jwe); err != nil {
return
}
// We do not support zip compression
if jweStruct.Header.Zip != "" {
err = ErrZipCompressionNotSupported
return
}
var key AsymmetricDecryptionKey
key, err = d.keystore.Get(jweStruct.Header.Kid)
if err != nil {
return
}
// Check alg is as expected
if jweStruct.Header.Alg != key.Algorithm() {
err = ErrInvalidAlgorithm
return
}
// Check the content encryption is a support algorithm
switch jweStruct.Header.Enc {
case jose.EncA128GCM, jose.EncA192GCM, jose.EncA256GCM:
// All good.
default:
err = ErrInvalidEncryption
return
}
// First decrypt the content encryption key.
var cekBytes []byte
cekBytes, err = key.Decrypt(jose.KeyOpsDecrypt, jweStruct.EncryptedKey)
if err != nil {
return
}
var block cipher.Block
block, err = aes.NewCipher(cekBytes)
if err != nil {
return
}
var aead cipher.AEAD
aead, err = cipher.NewGCM(block)
if err != nil {
return
}
// Decrypt the JWE payload.
ctAndTag := make([]byte, len(jweStruct.Ciphertext) + len(jweStruct.Tag))
copy(ctAndTag[:len(jweStruct.Ciphertext)], jweStruct.Ciphertext)
copy(ctAndTag[len(jweStruct.Ciphertext):], jweStruct.Tag)
plaintext, err = aead.Open(nil, jweStruct.Iv, ctAndTag, jweStruct.MarshalledHeader)
if err != nil {
return
}
if jweStruct.Header.OtherAad != nil {
aad = jweStruct.Header.OtherAad.Bytes()
}
return
}
// NewJweRsaKeyEncryptionDecryptorImpl creates an instance of JweRsaKeyEncryptionDecryptorImpl with the given keystore.
func NewJweRsaKeyEncryptionDecryptorImpl(keystore AsymmetricDecryptionKeyStore) *JweRsaKeyEncryptionDecryptorImpl {
return &JweRsaKeyEncryptionDecryptorImpl{
keystore: keystore,
}
}