def keyset_to_json(stub: testing_api_pb2_grpc.KeysetStub,
keyset: bytes) -> Text:
request = testing_api_pb2.KeysetToJsonRequest(keyset=keyset)
response = stub.ToJson(request)
if response.err:
raise tink.TinkError(response.err)
return response.json_keyset
def keyset_from_json(stub: testing_api_pb2_grpc.KeysetStub,
json_keyset: Text) -> bytes:
request = testing_api_pb2.KeysetFromJsonRequest(json_keyset=json_keyset)
response = stub.FromJson(request)
if response.err:
raise tink.TinkError(response.err)
return response.keyset
def public_keyset(stub: testing_api_pb2_grpc.KeysetStub,
private_keyset: bytes) -> bytes:
request = testing_api_pb2.KeysetPublicRequest(private_keyset=private_keyset)
response = stub.Public(request)
if response.err:
raise tink.TinkError(response.err)
return response.public_keyset
def key_template(stub: testing_api_pb2_grpc.KeysetStub,
template_name: str) -> tink_pb2.KeyTemplate:
request = testing_api_pb2.KeysetTemplateRequest(template_name=template_name)
response = stub.GetTemplate(request)
if response.err:
raise tink.TinkError(response.err)
return tink_pb2.KeyTemplate.FromString(response.key_template)
def new_keyset(stub: testing_api_pb2_grpc.KeysetStub,
key_template: tink_pb2.KeyTemplate) -> bytes:
gen_request = testing_api_pb2.KeysetGenerateRequest(
template=key_template.SerializeToString())
gen_response = stub.Generate(gen_request)
if gen_response.err:
raise tink.TinkError(gen_response.err)
return gen_response.keyset
def new_keyset_handle(stub: testing_api_pb2_grpc.KeysetStub,
key_template: tink_pb2.KeyTemplate) -> tink.KeysetHandle:
gen_request = testing_api_pb2.KeysetGenerateRequest(
template=key_template.SerializeToString())
gen_response = stub.Generate(gen_request)
if gen_response.err:
raise tink.TinkError(gen_response.err)
return cleartext_keyset_handle.read(
tink.BinaryKeysetReader(gen_response.keyset))
def public_keyset_handle(
stub: testing_api_pb2_grpc.KeysetStub,
private_keyset_handle: tink.KeysetHandle) -> tink.KeysetHandle:
request = testing_api_pb2.KeysetPublicRequest(
private_keyset=_keyset(private_keyset_handle))
response = stub.Public(request)
if response.err:
raise tink.TinkError(response.err)
return cleartext_keyset_handle.read(
tink.BinaryKeysetReader(response.public_keyset))
def keyset_write_encrypted(stub: testing_api_pb2_grpc.KeysetStub, keyset: bytes,
master_keyset: bytes,
associated_data: Optional[bytes]) -> bytes:
"""Writes an encrypted keyset."""
request = testing_api_pb2.KeysetWriteEncryptedRequest(
keyset=keyset, master_keyset=master_keyset)
if associated_data is not None:
request.associated_data.value = associated_data
response = stub.WriteEncrypted(request)
if response.err:
raise tink.TinkError(response.err)
return response.encrypted_keyset
def keyset_read_encrypted(stub: testing_api_pb2_grpc.KeysetStub,
encrypted_keyset: bytes, master_keyset: bytes,
associated_data: Optional[bytes],
keyset_reader_type: str) -> bytes:
"""Reads an encrypted keyset."""
request = testing_api_pb2.KeysetReadEncryptedRequest(
encrypted_keyset=encrypted_keyset,
master_keyset=master_keyset,
keyset_reader_type=testing_api_pb2.KeysetReaderType.Value(
keyset_reader_type))
if associated_data is not None:
request.associated_data.value = associated_data
response = stub.ReadEncrypted(request)
if response.err:
raise tink.TinkError(response.err)
return response.keyset
