#include "SSDatabase.h"
#include "KeySchema.h"
using namespace CssmClient;
using namespace SecurityServer;
const char *const SSDatabaseImpl::DBBlobRelationName = "DBBlob";
SSDatabaseImpl::SSDatabaseImpl(ClientSession &inClientSession, const CssmClient::DL &dl,
const char *inDbName, const CSSM_NET_ADDRESS *inDbLocation)
: Db::Impl(dl, inDbName, inDbLocation), mClientSession(inClientSession), mSSDbHandle(noDb)
{
}
SSDatabaseImpl::~SSDatabaseImpl()
{
if (mSSDbHandle != noDb)
mClientSession.releaseDb(mSSDbHandle);
}
SSUniqueRecord
SSDatabaseImpl::insert(CSSM_DB_RECORDTYPE recordType,
const CSSM_DB_RECORD_ATTRIBUTE_DATA *attributes,
const CSSM_DATA *data, bool)
{
SSUniqueRecord uniqueId(SSDatabase(this));
check(CSSM_DL_DataInsert(handle(), recordType,
attributes,
data, uniqueId));
uniqueId->activate();
return uniqueId;
}
void
SSDatabaseImpl::authenticate(CSSM_DB_ACCESS_TYPE inAccessRequest,
const CSSM_ACCESS_CREDENTIALS *inAccessCredentials)
{
mClientSession.authenticateDb(dbHandle(), inAccessRequest,
AccessCredentials::overlay(inAccessCredentials));
}
void
SSDatabaseImpl::lock()
{
mClientSession.lock(dbHandle());
}
void
SSDatabaseImpl::unlock()
{
mClientSession.unlock(dbHandle());
}
void
SSDatabaseImpl::unlock(const CSSM_DATA &password)
{
mClientSession.unlock(dbHandle(), CssmData::overlay(password));
}
void
SSDatabaseImpl::getSettings(uint32 &outIdleTimeout, bool &outLockOnSleep)
{
DBParameters parameters;
mClientSession.getDbParameters(dbHandle(), parameters);
outIdleTimeout = parameters.idleTimeout;
outLockOnSleep = parameters.lockOnSleep;
}
void
SSDatabaseImpl::setSettings(uint32 inIdleTimeout, bool inLockOnSleep)
{
DBParameters parameters;
parameters.idleTimeout = inIdleTimeout;
parameters.lockOnSleep = inLockOnSleep;
mClientSession.setDbParameters(dbHandle(), parameters);
CssmDataContainer dbb(allocator());
mClientSession.encodeDb(mSSDbHandle, dbb, allocator());
getDbBlobId(NULL)->modify(DBBlobRelationID, NULL, &dbb, CSSM_DB_MODIFY_ATTRIBUTE_NONE);
}
bool
SSDatabaseImpl::isLocked()
{
return mClientSession.isLocked(dbHandle());
}
void
SSDatabaseImpl::changePassphrase(const CSSM_ACCESS_CREDENTIALS *cred)
{
mClientSession.changePassphrase(dbHandle(), AccessCredentials::overlay(cred));
CssmDataContainer dbb(allocator());
mClientSession.encodeDb(mSSDbHandle, dbb, allocator());
getDbBlobId(NULL)->modify(DBBlobRelationID, NULL, &dbb, CSSM_DB_MODIFY_ATTRIBUTE_NONE);
}
DbHandle
SSDatabaseImpl::dbHandle()
{
activate();
if (mForked()) {
CssmDataContainer dbb(allocator());
getDbBlobId(&dbb);
mSSDbHandle = mClientSession.decodeDb(mIdentifier,
AccessCredentials::overlay(accessCredentials()), dbb);
}
return mSSDbHandle;
}
void
SSDatabaseImpl::create(const DLDbIdentifier &dlDbIdentifier)
{
mIdentifier = dlDbIdentifier;
DbImpl::create();
try
{
createRelation(DBBlobRelationID, DBBlobRelationName,
0, (CSSM_DB_SCHEMA_ATTRIBUTE_INFO *)42,
0, (CSSM_DB_SCHEMA_INDEX_INFO *)42);
createRelation(CSSM_DL_DB_RECORD_PUBLIC_KEY, "CSSM_DL_DB_RECORD_PUBLIC_KEY",
KeySchema::KeySchemaAttributeCount, KeySchema::KeySchemaAttributeList,
KeySchema::KeySchemaIndexCount, KeySchema::KeySchemaIndexList);
createRelation(CSSM_DL_DB_RECORD_PRIVATE_KEY, "CSSM_DL_DB_RECORD_PRIVATE_KEY",
KeySchema::KeySchemaAttributeCount, KeySchema::KeySchemaAttributeList,
KeySchema::KeySchemaIndexCount, KeySchema::KeySchemaIndexList);
createRelation(CSSM_DL_DB_RECORD_SYMMETRIC_KEY, "CSSM_DL_DB_RECORD_SYMMETRIC_KEY",
KeySchema::KeySchemaAttributeCount, KeySchema::KeySchemaAttributeList,
KeySchema::KeySchemaIndexCount, KeySchema::KeySchemaIndexList);
DBParameters dbParameters;
memset(&dbParameters, 0, sizeof(DBParameters));
dbParameters.idleTimeout = kDefaultIdleTimeout;
dbParameters.lockOnSleep = kDefaultLockOnSleep;
const AccessCredentials *cred = NULL;
const AclEntryInput *owner = NULL;
if (resourceControlContext())
{
cred = AccessCredentials::overlay(resourceControlContext()->AccessCred);
owner = &AclEntryInput::overlay(resourceControlContext()->InitialAclEntry);
}
mSSDbHandle = mClientSession.createDb(dlDbIdentifier, cred, owner, dbParameters);
CssmDataContainer dbb(allocator());
mClientSession.encodeDb(mSSDbHandle, dbb, allocator());
Db::Impl::insert(DBBlobRelationID, NULL, &dbb);
}
catch(...)
{
DbImpl::deleteDb();
throw;
}
}
void
SSDatabaseImpl::open(const DLDbIdentifier &dlDbIdentifier)
{
mIdentifier = dlDbIdentifier;
Db::Impl::open();
CssmDataContainer dbb(allocator());
getDbBlobId(&dbb);
mSSDbHandle = mClientSession.decodeDb(dlDbIdentifier, AccessCredentials::overlay(accessCredentials()), dbb);
}
DbUniqueRecordImpl *
SSDatabaseImpl::newDbUniqueRecord()
{
return new SSUniqueRecordImpl(SSDatabase(this));
}
CssmClient::DbUniqueRecord
SSDatabaseImpl::getDbBlobId(CssmDataContainer *dbb)
{
CssmClient::DbUniqueRecord dbBlobId;
DbCursor cursor(SSDatabase(this));
cursor->recordType(DBBlobRelationID);
if (!cursor->next(NULL, dbb, dbBlobId))
CssmError::throwMe(CSSMERR_DL_DATABASE_CORRUPT);
return dbBlobId;
}
SSUniqueRecordImpl::SSUniqueRecordImpl(const SSDatabase &db)
: DbUniqueRecord::Impl(db)
{
}
SSUniqueRecordImpl::~SSUniqueRecordImpl()
{
}
SSDatabase
SSUniqueRecordImpl::database() const
{
return parent<SSDatabase>();
}