@@ -87,9 +87,7 @@ def enc_setup(self, msg, key=None, auth_data=b"", **kwargs):
8787 try :
8888 _epk = kwargs ["epk" ]
8989 except KeyError :
90- _epk = ec .generate_private_key (
91- NIST2SEC [as_unicode (key .crv )], default_backend ()
92- )
90+ _epk = ec .generate_private_key (NIST2SEC [as_unicode (key .crv )], default_backend ())
9391 epk = ECKey ().load_key (_epk .public_key ())
9492 else :
9593 if isinstance (_epk , ec .EllipticCurvePrivateKey ):
@@ -116,15 +114,11 @@ def enc_setup(self, msg, key=None, auth_data=b"", **kwargs):
116114 except KeyError :
117115 raise ValueError ("Unknown key length for algorithm %s" % self .enc )
118116
119- cek = ecdh_derive_key (
120- _epk , key .pub_key , apu , apv , str (self .enc ).encode (), dk_len
121- )
117+ cek = ecdh_derive_key (_epk , key .pub_key , apu , apv , str (self .enc ).encode (), dk_len )
122118 elif self .alg in ["ECDH-ES+A128KW" , "ECDH-ES+A192KW" , "ECDH-ES+A256KW" ]:
123119 _pre , _post = self .alg .split ("+" )
124120 klen = int (_post [1 :4 ])
125- kek = ecdh_derive_key (
126- _epk , key .pub_key , apu , apv , str (_post ).encode (), klen
127- )
121+ kek = ecdh_derive_key (_epk , key .pub_key , apu , apv , str (_post ).encode (), klen )
128122 cek = self ._generate_key (self .enc , cek = cek )
129123 encrypted_key = aes_key_wrap (kek , cek , default_backend ())
130124 else :
@@ -163,12 +157,7 @@ def dec_setup(self, token, key=None, **kwargs):
163157 raise Exception ("Unknown key length for algorithm" )
164158
165159 self .cek = ecdh_derive_key (
166- key ,
167- epubkey .pub_key ,
168- apu ,
169- apv ,
170- str (self .headers ["enc" ]).encode (),
171- dk_len ,
160+ key , epubkey .pub_key , apu , apv , str (self .headers ["enc" ]).encode (), dk_len ,
172161 )
173162 elif self .headers ["alg" ] in [
174163 "ECDH-ES+A128KW" ,
@@ -177,9 +166,7 @@ def dec_setup(self, token, key=None, **kwargs):
177166 ]:
178167 _pre , _post = self .headers ["alg" ].split ("+" )
179168 klen = int (_post [1 :4 ])
180- kek = ecdh_derive_key (
181- key , epubkey .pub_key , apu , apv , str (_post ).encode (), klen
182- )
169+ kek = ecdh_derive_key (key , epubkey .pub_key , apu , apv , str (_post ).encode (), klen )
183170 self .cek = aes_key_unwrap (kek , token .encrypted_key (), default_backend ())
184171 else :
185172 raise Exception ("Unsupported algorithm %s" % self .headers ["alg" ])
0 commit comments