IPropertyTree *getPkgSetRegistry(const char *process, bool readonly)
{
Owned<IRemoteConnection> globalLock = querySDS().connect("/PackageSets/", myProcessSession(), RTM_LOCK_WRITE|RTM_CREATE_QUERY, SDS_LOCK_TIMEOUT);
if (!globalLock)
throw MakeStringException(PKG_DALI_LOOKUP_ERROR, "Unable to connect to PackageSet information in dali /PackageSets");
IPropertyTree *pkgSets = globalLock->queryRoot();
if (!pkgSets)
throw MakeStringException(PKG_DALI_LOOKUP_ERROR, "Unable to open PackageSet information in dali /PackageSets");
if (!process || !*process)
process = "*";
StringBuffer id;
buildPkgSetId(id, process);
//Only lock the branch for the target we're interested in.
VStringBuffer xpath("/PackageSets/PackageSet[@id='%s']", id.str());
Owned<IRemoteConnection> conn = querySDS().connect(xpath.str(), myProcessSession(), readonly ? RTM_LOCK_READ : RTM_LOCK_WRITE, SDS_LOCK_TIMEOUT);
if (!conn)
{
if (readonly)
return NULL;
Owned<IPropertyTree> pkgSet = createPTree();
pkgSet->setProp("@id", id.str());
pkgSet->setProp("@process", process);
pkgSets->addPropTree("PackageSet", pkgSet.getClear());
globalLock->commit();
conn.setown(querySDS().connect(xpath.str(), myProcessSession(), RTM_LOCK_WRITE, SDS_LOCK_TIMEOUT));
}
return (conn) ? conn->getRoot() : NULL;
}
bool WuResubmit(const char *wuid)
{
Owned<IWorkUnitFactory> factory = getWorkUnitFactory();
Owned<IWorkUnit> wu = factory->updateWorkUnit(wuid);
if (!wu)
{
ERRLOG("WuResubmit(%s): could not find workunit",wuid);
return false;
}
if (wu->getState()!=WUStateFailed)
{
ERRLOG("WuResubmit(%s): could not resubmit as workunit state is '%s'", wuid, wu->queryStateDesc());
return false;
}
SCMStringBuffer token;
wu->getSecurityToken(token);
SCMStringBuffer user;
SCMStringBuffer password;
extractToken(token.str(), wuid, user, password);
wu->resetWorkflow();
wu->setState(WUStateSubmitted);
wu->commit();
wu.clear();
submitWorkUnit(wuid,user.str(),password.str());
PROGLOG("WuResubmit(%s): resubmitted",wuid);
return true;
}
void cleanupWorkUnitSchedule()
{
Owned<IRemoteConnection> conn = querySDS().connect("/Schedule", myProcessSession(), RTM_LOCK_WRITE, connectionTimeout);
if(!conn) return;
Owned<IPropertyTree> root(conn->queryRoot()->getBranch("."));
recursiveCleanup(root, 2);
conn->commit();
cleanupSchedulerList(root);
}
IXRefNode * CXRefNodeManager::CreateXRefNode(const char* NodeName)
{
Owned<IRemoteConnection> conn = querySDS().connect("/DFU/XREF",myProcessSession(),RTM_CREATE_QUERY|RTM_LOCK_WRITE ,INFINITE);
IPropertyTree* xref_ptree = conn->queryRoot();
IPropertyTree* cluster_ptree = xref_ptree->addPropTree("Cluster", createPTree());
cluster_ptree->setProp("@name",NodeName);
conn->commit();
conn->changeMode(RTM_NONE);
return new CXRefNode(NodeName,conn);
}
int CEclAgentExecutionServer::executeWorkunit(const char * wuid)
{
//build eclagent command line
StringBuffer command;
#ifdef _WIN32
command.append(".\\eclagent.exe");
#else
command.append("start_eclagent");
#endif
StringBuffer cmdLine = command;
cmdLine.append(" WUID=").append(wuid).append(" DALISERVERS=").append(daliServers);
DWORD runcode;
PROGLOG("AgentExec: Executing '%s'", cmdLine.str());
#ifdef _WIN32
bool success = invoke_program(cmdLine.str(), runcode, false, NULL, NULL);
#else
//specify "wait" to eliminate linux zombies. Waits for the startup script to
//complete (not eclagent), because script starts eclagent in the background
bool success = invoke_program(cmdLine.str(), runcode, true, NULL, NULL);
#endif
if (success)
{
if (runcode != 0)
PROGLOG("Process failed during execution: %s error(%"I64F"i)", cmdLine.str(), (unsigned __int64) runcode);
else
PROGLOG("Execution started");
}
else
{
Owned<IWorkUnitFactory> factory = getWorkUnitFactory();
Owned<IWorkUnit> workunit = factory->updateWorkUnit(wuid);
if (workunit)
{
workunit->setState(WUStateFailed);
workunit->commit();
}
PROGLOG("Process failed to start: %s", cmdLine.str());
}
return success && runcode == 0;
}
IXRefNode * CXRefNodeManager::getXRefNode(const char* NodeName)
{
//DBGLOG("Node Name %s:",NodeName);
StringBuffer xpath;
xpath.appendf("Cluster[@name=\"%s\"]", NodeName);
//if not exists, add DFU/XREF/ClusterName branch to SDS
//not linked
Owned<IRemoteConnection> conn = querySDS().connect("/DFU/XREF",myProcessSession(),RTM_CREATE_QUERY|RTM_NONE ,INFINITE);
IPropertyTree* cluster_ptree = conn->queryRoot()->queryPropTree(xpath.str());
conn->commit();
if (cluster_ptree==0)
{
DBGLOG("Cluster[@name=%s] can't be found under /DFU/XREF", NodeName);
return 0 ;
}
return new CXRefNode(NodeName,conn);
}
bool CWsDfuXRefEx::onDFUXRefArrayAction(IEspContext &context, IEspDFUXRefArrayActionRequest &req, IEspDFUXRefArrayActionResponse &resp)
{
try
{
StringBuffer username;
context.getUserID(username);
DBGLOG("CWsDfuXRefEx::onDFUXRefArrayAction User=%s",username.str());
Owned<IUserDescriptor> userdesc;
if(username.length() > 0)
{
const char* passwd = context.queryPassword();
userdesc.setown(createUserDescriptor());
userdesc->set(username.str(), passwd);
}
if(*req.getAction() == 0 || *req.getType() == 0 || *req.getCluster() == 0)
{
ERRLOG("Invalid Parameters into CWsDfuXRefEx::onDFUXRefArrayAction");
throw MakeStringExceptionDirect(ECLWATCH_INVALID_INPUT, "Action, cluster, or type not defined.");
}
Owned<IXRefNode> xRefNode = XRefNodeManager->getXRefNode(req.getCluster());
if (xRefNode.get() == 0)
{
ERRLOG("Unable to resolve XRef cluster name %s",req.getCluster());
throw MakeStringException(ECLWATCH_CANNOT_RESOLVE_CLUSTER_NAME, "Unable to resolve cluster name %s",req.getCluster());
}
Owned<IXRefFilesNode> _fileNode = getFileNodeInterface(*xRefNode.get(),req.getType());
if (_fileNode.get() == 0)
{
ERRLOG("Unable to find a suitable IXRefFilesNode interface for %s",req.getType());
throw MakeStringException(ECLWATCH_CANNOT_FIND_IXREFFILESNODE, "Unable to find a suitable IXRefFilesNode interface for %s",req.getType());
}
if (!context.validateFeatureAccess(FEATURE_URL, SecAccess_Full, false))
throw MakeStringException(ECLWATCH_DFU_XREF_ACCESS_DENIED, "Failed to run Xref command. Permission denied.");
StringBuffer returnStr,UserName;
const char* ActionType = req.getAction();
DBGLOG("Running Xref Command %s for user %s",ActionType,context.getUserID(UserName).str());
for(unsigned i = 0; i < req.getXRefFiles().length();i++)
{
DBGLOG("CWsDfuXRefEx::onDFUXRefArrayAction %s file %s for User=%s", ActionType, req.getXRefFiles().item(i), username.str());
StringBuffer errstr;
if (strcmp("Delete" ,ActionType) == 0)
{
if (_fileNode->RemovePhysical(req.getXRefFiles().item(i),userdesc,req.getCluster(),errstr))
appendReplyMessage(returnStr,NULL,"Removed Physical part %s",req.getXRefFiles().item(i));
else
appendReplyMessage(returnStr,NULL,"Error(s) removing physical part %s\n%s",req.getXRefFiles().item(i),errstr.str());
}
else if (strcmp("Attach" ,ActionType) == 0)
{
if(_fileNode->AttachPhysical(req.getXRefFiles().item(i),userdesc,req.getCluster(),errstr) )
appendReplyMessage(returnStr,NULL,"Reattached Physical part %s",req.getXRefFiles().item(i));
else
appendReplyMessage(returnStr,NULL,"Error(s) attaching physical part %s\n%s",req.getXRefFiles().item(i),errstr.str());
}
if (strcmp("DeleteLogical" ,ActionType) == 0)
{
// Note we don't want to physically delete 'lost' files - this will end up with orphans on next time round but that is safer
if (_fileNode->RemoveLogical(req.getXRefFiles().item(i),userdesc,req.getCluster(),errstr)) {
appendReplyMessage(returnStr,NULL,"Removed Logical File %s",req.getXRefFiles().item(i));
}
else
appendReplyMessage(returnStr,NULL,"Error(s) removing File %s\n%s",req.getXRefFiles().item(i),errstr.str());
}
}
xRefNode->commit();
resp.setDFUXRefArrayActionResult(returnStr.str());
}
catch(IException* e)
{
FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
}
return true;
}