|
11 | 11 | import com.nimbusds.jwt.SignedJWT; |
12 | 12 | import io.grpc.StatusRuntimeException; |
13 | 13 | import io.grpc.Status; |
14 | | -import io.opentdf.platform.kas.AccessServiceClient; |
15 | | -import io.opentdf.platform.kas.PublicKeyRequest; |
16 | | -import io.opentdf.platform.kas.PublicKeyResponse; |
17 | | -import io.opentdf.platform.kas.RewrapRequest; |
18 | | -import io.opentdf.platform.kas.RewrapResponse; |
| 14 | +import io.opentdf.platform.generated.kas.AccessServiceClient; |
| 15 | +import io.opentdf.platform.generated.kas.PublicKeyRequest; |
| 16 | +import io.opentdf.platform.generated.kas.PublicKeyResponse; |
| 17 | +import io.opentdf.platform.generated.kas.RewrapRequest; |
| 18 | +import io.opentdf.platform.generated.kas.RewrapResponse; |
19 | 19 | import io.opentdf.platform.sdk.Config.KASInfo; |
20 | 20 | import io.opentdf.platform.sdk.nanotdf.ECKeyPair; |
21 | 21 | import io.opentdf.platform.sdk.nanotdf.NanoTDFType; |
|
31 | 31 | import java.security.NoSuchAlgorithmException; |
32 | 32 | import java.time.Duration; |
33 | 33 | import java.time.Instant; |
| 34 | +import java.util.Collections; |
34 | 35 | import java.util.Date; |
35 | 36 | import java.util.HashMap; |
36 | 37 | import java.util.function.Function; |
@@ -71,6 +72,12 @@ public KASClient(Function<String, AccessServiceClient> clientFactory, RSAKey dpo |
71 | 72 | this.kasKeyCache = new KASKeyCache(); |
72 | 73 | } |
73 | 74 |
|
| 75 | + public KASClient(Function<String, AccessServiceClient> channelFactory, boolean usePlaintext, RSASSASigner signer) { |
| 76 | + this.channelFactory = channelFactory; |
| 77 | + this.usePlaintext = usePlaintext; |
| 78 | + this.signer = signer; |
| 79 | + } |
| 80 | + |
74 | 81 | @Override |
75 | 82 | public KASInfo getECPublicKey(Config.KASInfo kasInfo, NanoTDFType.ECCurve curve) { |
76 | 83 | var req = PublicKeyRequest.newBuilder().setAlgorithm(format("ec:%s", curve.toString())).build(); |
@@ -221,7 +228,7 @@ public byte[] unwrap(Manifest.KeyAccess keyAccess, String policy, KeyType sessi |
221 | 228 | .build(); |
222 | 229 | RewrapResponse response; |
223 | 230 | try { |
224 | | - var req = getStub(keyAccess.url).rewrapBlocking(request, MapsKt.mapOf()).execute(); |
| 231 | + var req = getStub(keyAccess.url).rewrapBlocking(request, Collections.emptyMap()).execute(); |
225 | 232 | try { |
226 | 233 | response = getOrThrow(req); |
227 | 234 | } catch (Exception e) { |
@@ -290,7 +297,7 @@ public byte[] unwrapNanoTDF(NanoTDFType.ECCurve curve, String header, String kas |
290 | 297 | .setSignedRequestToken(jwt.serialize()) |
291 | 298 | .build(); |
292 | 299 |
|
293 | | - var request = getStub(keyAccess.url).rewrapBlocking(req, MapsKt.mapOf()).execute(); |
| 300 | + var request = getStub(keyAccess.url).rewrapBlocking(req, Collections.emptyMap()).execute(); |
294 | 301 | RewrapResponse response; |
295 | 302 | try { |
296 | 303 | response = getOrThrow(request); |
|
0 commit comments