//
// Copyright 2013 Google Inc. All Rights Reserved.
//

#ifndef WV_DRM_PLUGIN_H_
#define WV_DRM_PLUGIN_H_

#include "media/drm/DrmAPI.h"
#include "media/stagefright/foundation/ABase.h"
#include "media/stagefright/foundation/AString.h"
#include "utils/Errors.h"
#include "utils/KeyedVector.h"
#include "utils/List.h"
#include "utils/String8.h"
#include "utils/Vector.h"
#include "wv_content_decryption_module.h"

namespace wvdrm {

using android::KeyedVector;
using android::List;
using android::status_t;
using android::String8;
using android::Vector;

class WVDrmPlugin : public android::DrmPlugin {
 public:
  WVDrmPlugin(wvcdm::WvContentDecryptionModule* cdm);
  virtual ~WVDrmPlugin() {}

  virtual status_t openSession(Vector<uint8_t>& sessionId);

  virtual status_t closeSession(const Vector<uint8_t>& sessionId);

  virtual status_t getKeyRequest(
      const Vector<uint8_t>& sessionId,
      const Vector<uint8_t>& initData,
      const String8& mimeType,
      KeyType keyType,
      const KeyedVector<String8, String8>& optionalParameters,
      Vector<uint8_t>& request,
      String8& defaultUrl);

  virtual status_t provideKeyResponse(const Vector<uint8_t>& sessionId,
                                      const Vector<uint8_t>& response,
                                      Vector<uint8_t>& keySetId);

  virtual status_t removeKeys(const Vector<uint8_t>& keySetId);

  virtual status_t restoreKeys(const Vector<uint8_t>& sessionId,
                               const Vector<uint8_t>& keySetId);

  virtual status_t queryKeyStatus(
      const Vector<uint8_t>& sessionId,
      KeyedVector<String8, String8>& infoMap) const;

  virtual status_t getProvisionRequest(Vector<uint8_t>& request,
                                       String8& defaultUrl);

  virtual status_t provideProvisionResponse(const Vector<uint8_t>& response);

  virtual status_t getSecureStops(List<Vector<uint8_t> >& secureStops);

  virtual status_t releaseSecureStops(const Vector<uint8_t>& ssRelease);

  virtual status_t getPropertyString(const String8& name, String8& value) const;

  virtual status_t getPropertyByteArray(const String8& name,
                                        Vector<uint8_t>& value) const;

  virtual status_t setPropertyString(const String8& name, const String8& value);

  virtual status_t setPropertyByteArray(const String8& name,
                                        const Vector<uint8_t>& value);
    virtual status_t setCipherAlgorithm(Vector<uint8_t> const &sessionId,
                                        String8 const &algorithm);

    virtual status_t setMacAlgorithm(Vector<uint8_t> const &sessionId,
                                     String8 const &algorithm);

    virtual status_t encrypt(Vector<uint8_t> const &sessionId,
                             Vector<uint8_t> const &keyId,
                             Vector<uint8_t> const &input,
                             Vector<uint8_t> const &iv,
                             Vector<uint8_t> &output);

    virtual status_t decrypt(Vector<uint8_t> const &sessionId,
                             Vector<uint8_t> const &keyId,
                             Vector<uint8_t> const &input,
                             Vector<uint8_t> const &iv,
                             Vector<uint8_t> &output);

    virtual status_t sign(Vector<uint8_t> const &sessionId,
                          Vector<uint8_t> const &keyId,
                          Vector<uint8_t> const &message,
                          Vector<uint8_t> &signature);

    virtual status_t verify(Vector<uint8_t> const &sessionId,
                            Vector<uint8_t> const &keyId,
                            Vector<uint8_t> const &message,
                            Vector<uint8_t> const &signature,
                            bool &match);

 private:
  DISALLOW_EVIL_CONSTRUCTORS(WVDrmPlugin);

  wvcdm::WvContentDecryptionModule* mCDM;
};

} // namespace wvdrm

#endif // WV_DRM_PLUGIN_H_