/*
 * Copyright 2018 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef REMOTE_KEYMASTER_H_
#define REMOTE_KEYMASTER_H_

#include <keymaster/android_keymaster_messages.h>

#include "common/libs/security/keymaster_channel_sharedfd.h"

namespace keymaster {

class RemoteKeymaster {
 private:
  cuttlefish::SharedFdKeymasterChannel* channel_;
  const int32_t message_version_;

  void ForwardCommand(AndroidKeymasterCommand command, const Serializable& req,
                      KeymasterResponse* rsp);

 public:
  RemoteKeymaster(cuttlefish::SharedFdKeymasterChannel*,
                  int32_t message_version = kDefaultMessageVersion);
  ~RemoteKeymaster();
  bool Initialize();
  void GetVersion(const GetVersionRequest& request,
                  GetVersionResponse* response);
  void SupportedAlgorithms(const SupportedAlgorithmsRequest& request,
                           SupportedAlgorithmsResponse* response);
  void SupportedBlockModes(const SupportedBlockModesRequest& request,
                           SupportedBlockModesResponse* response);
  void SupportedPaddingModes(const SupportedPaddingModesRequest& request,
                             SupportedPaddingModesResponse* response);
  void SupportedDigests(const SupportedDigestsRequest& request,
                        SupportedDigestsResponse* response);
  void SupportedImportFormats(const SupportedImportFormatsRequest& request,
                              SupportedImportFormatsResponse* response);
  void SupportedExportFormats(const SupportedExportFormatsRequest& request,
                              SupportedExportFormatsResponse* response);
  void AddRngEntropy(const AddEntropyRequest& request,
                     AddEntropyResponse* response);
  void Configure(const ConfigureRequest& request, ConfigureResponse* response);
  void GenerateKey(const GenerateKeyRequest& request,
                   GenerateKeyResponse* response);
  void GenerateRkpKey(const GenerateRkpKeyRequest& request,
                      GenerateRkpKeyResponse* response);
  void GenerateCsr(const GenerateCsrRequest& request,
                   GenerateCsrResponse* response);
  void GenerateCsrV2(const GenerateCsrV2Request& request,
                     GenerateCsrV2Response* response);
  void GetKeyCharacteristics(const GetKeyCharacteristicsRequest& request,
                             GetKeyCharacteristicsResponse* response);
  void ImportKey(const ImportKeyRequest& request, ImportKeyResponse* response);
  void ImportWrappedKey(const ImportWrappedKeyRequest& request,
                        ImportWrappedKeyResponse* response);
  void ExportKey(const ExportKeyRequest& request, ExportKeyResponse* response);
  void AttestKey(const AttestKeyRequest& request, AttestKeyResponse* response);
  void UpgradeKey(const UpgradeKeyRequest& request,
                  UpgradeKeyResponse* response);
  void DeleteKey(const DeleteKeyRequest& request, DeleteKeyResponse* response);
  void DeleteAllKeys(const DeleteAllKeysRequest& request,
                     DeleteAllKeysResponse* response);
  void BeginOperation(const BeginOperationRequest& request,
                      BeginOperationResponse* response);
  void UpdateOperation(const UpdateOperationRequest& request,
                       UpdateOperationResponse* response);
  void FinishOperation(const FinishOperationRequest& request,
                       FinishOperationResponse* response);
  void AbortOperation(const AbortOperationRequest& request,
                      AbortOperationResponse* response);
  GetHmacSharingParametersResponse GetHmacSharingParameters();
  ComputeSharedHmacResponse ComputeSharedHmac(
      const ComputeSharedHmacRequest& request);
  VerifyAuthorizationResponse VerifyAuthorization(
      const VerifyAuthorizationRequest& request);
  DeviceLockedResponse DeviceLocked(const DeviceLockedRequest& request);
  EarlyBootEndedResponse EarlyBootEnded();
  ConfigureVendorPatchlevelResponse ConfigureVendorPatchlevel(
      const ConfigureVendorPatchlevelRequest& request);
  ConfigureBootPatchlevelResponse ConfigureBootPatchlevel(
      const ConfigureBootPatchlevelRequest& request);
  ConfigureVerifiedBootInfoResponse ConfigureVerifiedBootInfo(
      const ConfigureVerifiedBootInfoRequest& request);
  void GenerateTimestampToken(GenerateTimestampTokenRequest& request,
                              GenerateTimestampTokenResponse* response);
  GetRootOfTrustResponse GetRootOfTrust(const GetRootOfTrustRequest& request);
  GetHwInfoResponse GetHwInfo();
  SetAttestationIdsResponse SetAttestationIds(
      const SetAttestationIdsRequest& request);
  SetAttestationIdsKM3Response SetAttestationIdsKM3(
      const SetAttestationIdsKM3Request& request);

  // CF HAL and remote sides are always compiled together, so will never
  // disagree about message versions.
  int32_t message_version() { return message_version_; }
};

}  // namespace keymaster

#endif  // REMOTE_KEYMASTER_H_