request_signer.h

// Copyright 2014 Google Inc. All rights reserved.
//
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file or at
// https://developers.google.com/open-source/licenses/bsd
 
#ifndef PACKAGER_MEDIA_BASE_REQUEST_SIGNER_H_
#define PACKAGER_MEDIA_BASE_REQUEST_SIGNER_H_
 
#include <memory>
#include <string>
#include <vector>
 
#include "packager/base/macros.h"
 
namespace shaka {
namespace media {
 
class AesCbcEncryptor;
class RsaPrivateKey;
 
class RequestSigner {
 public:
  virtual ~RequestSigner();
 
  virtual bool GenerateSignature(const std::string& message,
                                 std::string* signature) = 0;
 
  const std::string& signer_name() const { return signer_name_; }
 
 protected:
  explicit RequestSigner(const std::string& signer_name);
 
 private:
  std::string signer_name_;
 
  DISALLOW_COPY_AND_ASSIGN(RequestSigner);
};
 
class AesRequestSigner : public RequestSigner {
 public:
  ~AesRequestSigner() override;
 
  static AesRequestSigner* CreateSigner(const std::string& signer_name,
                                        const std::vector<uint8_t>& aes_key,
                                        const std::vector<uint8_t>& iv);
 
  bool GenerateSignature(const std::string& message,
                         std::string* signature) override;
 
 private:
  AesRequestSigner(const std::string& signer_name,
                   std::unique_ptr<AesCbcEncryptor> encryptor);
 
  std::unique_ptr<AesCbcEncryptor> aes_cbc_encryptor_;
 
  DISALLOW_COPY_AND_ASSIGN(AesRequestSigner);
};
 
class RsaRequestSigner : public RequestSigner {
 public:
  ~RsaRequestSigner() override;
 
  static RsaRequestSigner* CreateSigner(const std::string& signer_name,
                                        const std::string& pkcs1_rsa_key);
 
  bool GenerateSignature(const std::string& message,
                         std::string* signature) override;
 
 private:
  RsaRequestSigner(const std::string& signer_name,
                   std::unique_ptr<RsaPrivateKey> rsa_private_key);
 
  std::unique_ptr<RsaPrivateKey> rsa_private_key_;
 
  DISALLOW_COPY_AND_ASSIGN(RsaRequestSigner);
};
 
}  // namespace media
}  // namespace shaka
 
#endif  // PACKAGER_MEDIA_BASE_REQUEST_SIGNER_H_