/*
* CMS Encoding Base
* (C) 1999-2007 Jack Lloyd
*
* Distributed under the terms of the Botan license
*/

#include <botan/cms_enc.h>
#include <botan/der_enc.h>
#include <botan/oids.h>
#include <botan/pem.h>

namespace Botan {

/*
* Setup the intitial layer of CMS data
*/
void CMS_Encoder::set_data(const byte buf[], size_t length)
   {
   if(!data.empty())
      throw Invalid_State("Cannot call CMS_Encoder::set_data here");

   data.resize(length);
   copy_mem(&data[0], buf, length);
   type = "CMS.DataContent";
   }

/*
* Setup the intitial layer of CMS data
*/
void CMS_Encoder::set_data(const std::string& str)
   {
   set_data(reinterpret_cast<const byte*>(str.c_str()), str.length());
   }

/*
* Finalize and return the CMS encoded data
*/
SecureVector<byte> CMS_Encoder::get_contents()
   {
   DER_Encoder encoder;

   encoder.start_cons(SEQUENCE).
      encode(OIDS::lookup(type)).
      start_explicit(0).
         raw_bytes(data).
      end_explicit().
   end_cons();

   data.clear();

   return encoder.get_contents();
   }

/*
* Add a new layer of encapsulation
*/
void CMS_Encoder::add_layer(const std::string& oid, DER_Encoder& new_layer)
   {
   data = new_layer.get_contents();
   type = oid;
   }

/*
* Return the PEM-encoded data
*/
std::string CMS_Encoder::PEM_contents()
   {
   return PEM_Code::encode(get_contents(), "PKCS7");
   }

/*
* Make an EncapsulatedContentInfo
*/
SecureVector<byte> CMS_Encoder::make_econtent(const SecureVector<byte>& data,
                                              const std::string& type)
   {
   return DER_Encoder().start_cons(SEQUENCE).
      encode(OIDS::lookup(type)).
      start_explicit(0).
         encode(data, OCTET_STRING).
      end_explicit().
   end_cons().
   get_contents();
   }

}
