#include <security_cdsa_client/cryptoclient.h>
using namespace CssmClient;
Crypt::Crypt(const CSP &csp, CSSM_ALGORITHMS alg)
: Context(csp, alg), mMode(CSSM_ALGMODE_NONE), mInitVector(NULL),
mPadding(CSSM_PADDING_NONE)
{
}
void Crypt::key(const Key &key)
{
mKey = key;
set(CSSM_ATTRIBUTE_KEY, static_cast<const CssmKey &>(key));
}
void
Crypt::activate()
{
StLock<Mutex> _(mActivateMutex);
if (!mActive)
{
if (!mKey && mAlgorithm != CSSM_ALGID_NONE)
CssmError::throwMe(CSSMERR_CSP_MISSING_ATTR_KEY);
if (!mKey || mKey->keyClass() == CSSM_KEYCLASS_SESSION_KEY)
{ check(CSSM_CSP_CreateSymmetricContext(attachment()->handle(), mAlgorithm,
mMode, neededCred(), mKey, mInitVector, mPadding, NULL,
&mHandle));
}
else
{
check(CSSM_CSP_CreateAsymmetricContext(attachment()->handle(), mAlgorithm,
neededCred(), mKey, mPadding, &mHandle));
}
mActive = true;
}
}
CSSM_SIZE
Encrypt::encrypt(const CssmData *in, uint32 inCount,
CssmData *out, uint32 outCount, CssmData &remData)
{
unstaged();
CSSM_SIZE total;
check(CSSM_EncryptData(handle(), in, inCount, out, outCount, &total, &remData));
return total;
}
void
Encrypt::init()
{
check(CSSM_EncryptDataInit(handle()));
mStaged = true;
}
CSSM_SIZE
Encrypt::encrypt(const CssmData *in, uint32 inCount,
CssmData *out, uint32 outCount)
{
staged();
CSSM_SIZE total;
check(CSSM_EncryptDataUpdate(handle(), in, inCount, out, outCount, &total));
return total;
}
void
Encrypt::final(CssmData &remData)
{
staged();
check(CSSM_EncryptDataFinal(handle(), &remData));
mStaged = false;
}
CSSM_SIZE
Decrypt::decrypt(const CssmData *in, uint32 inCount,
CssmData *out, uint32 outCount, CssmData &remData)
{
unstaged();
CSSM_SIZE total;
check(CSSM_DecryptData(handle(), in, inCount, out, outCount, &total, &remData));
return total;
}
void
Decrypt::init()
{
check(CSSM_DecryptDataInit(handle()));
mStaged = true;
}
CSSM_SIZE
Decrypt::decrypt(const CssmData *in, uint32 inCount,
CssmData *out, uint32 outCount)
{
staged();
CSSM_SIZE total;
check(CSSM_DecryptDataUpdate(handle(), in, inCount, out, outCount, &total));
return total;
}
void
Decrypt::final(CssmData &remData)
{
staged();
check(CSSM_DecryptDataFinal(handle(), &remData));
mStaged = false;
}