#include "process.h"
#include "server.h"
#include "session.h"
#include "authority.h"
Process::Process(Port servicePort, TaskPort taskPort, const char *identity, uid_t uid, gid_t gid)
: session(Session::find(servicePort)), mBusyCount(0), mDying(false),
mTaskPort(taskPort), mUid(uid), mGid(gid)
{
mPid = mTaskPort.pid();
session.addProcess(this);
mClientCode = CodeSigning::OSXCode::decode(identity);
debug("SS", "New process %p(%d) uid=%d gid=%d session=%p TP=%d for %s",
this, mPid, mUid, mGid, &session,
mTaskPort.port(), identity ? identity : "(unknown)");
}
#if 0
Process::Process(Process &prior)
: session(Session::find(prior.mTaskPort.bootstrap())), mBusyCount(0), mDying(false),
mTaskPort(prior.mTaskPort), mUid(prior.mUid), mGid(prior.mGid)
{
mPid = prior.mPid;
session.addProcess(this);
mClientCode = prior.mClientCode;
prior.mTaskPort = Port();
debug("SS", "Process %p(%d) recloned uid=%d gid=%d session=%p",
this, mPid, mUid, mGid, &session);
}
#endif
Process::~Process()
{
assert(mBusyCount == 0);
IFDEBUG(if (!mAuthorizations.empty())
debug("SS", "Process %p(%d) clearing %d authorizations",
this, mPid, int(mAuthorizations.size())));
for (AuthorizationSet::iterator it = mAuthorizations.begin();
it != mAuthorizations.end(); it++) {
AuthorizationToken *auth = *it;
if (removeAuthorization(auth))
delete auth;
}
IFDEBUG(if (!mDatabases.empty())
debug("SS", "Process %p(%d) clearing %d database handles",
this, mPid, int(mDatabases.size())));
for (DatabaseSet::iterator it = mDatabases.begin();
it != mDatabases.end(); it++)
delete *it;
debug("SS", "Process %p(%d) has died", this, mPid);
if (mTaskPort)
mTaskPort.destroy();
if (session.removeProcess(this))
delete &session;
}
bool Process::kill()
{
if (mBusyCount == 0) {
return true; } else {
debug("SS", "Process %p(%d) destruction deferred for %d busy connections",
this, mPid, int(mBusyCount));
mDying = true;
return false; }
}
void Process::beginConnection(Connection &)
{
StLock<Mutex> _(mLock);
mBusyCount++;
}
bool Process::endConnection(Connection &)
{
StLock<Mutex> _(mLock);
return --mBusyCount == 0 && mDying;
}
void Process::addDatabase(Database *database)
{
StLock<Mutex> _(mLock);
mDatabases.insert(database);
}
void Process::removeDatabase(Database *database)
{
StLock<Mutex> _(mLock);
assert(mDatabases.find(database) != mDatabases.end());
mDatabases.erase(database);
}
bool Process::verifyCodeSignature(const CodeSigning::Signature *signature)
{
if (mClientCode)
return Server::signer().verify(*mClientCode, signature);
else
return false; }
void Process::addAuthorization(AuthorizationToken *auth)
{
assert(auth);
StLock<Mutex> _(mLock);
mAuthorizations.insert(auth);
auth->addProcess(*this);
}
bool Process::removeAuthorization(AuthorizationToken *auth)
{
assert(auth);
StLock<Mutex> _(mLock);
typedef AuthorizationSet::iterator Iter;
pair<Iter, Iter> range = mAuthorizations.equal_range(auth);
assert(range.first != mAuthorizations.end());
Iter next = range.first; next++; mAuthorizations.erase(range.first); if (next == range.second) { if (auth->endProcess(*this)) return true; }
return false; }
void Process::requestNotifications(Port port, Listener::Domain domain, Listener::EventMask events)
{
new Listener(*this, port, domain, events);
}
void Process::stopNotifications(Port port)
{
if (!Listener::remove(port))
CssmError::throwMe(CSSMERR_CSSM_INVALID_HANDLE_USAGE); }
void Process::postNotification(Listener::Domain domain, Listener::Event event, const CssmData &data)
{
Listener::notify(domain, event, data);
}