123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219 |
- /*##############################################################################
- HPCC SYSTEMS software Copyright (C) 2012 HPCC Systems®.
- Licensed under the Apache License, Version 2.0 (the "License");
- you may not use this file except in compliance with the License.
- You may obtain a copy of the License at
- http://www.apache.org/licenses/LICENSE-2.0
- Unless required by applicable law or agreed to in writing, software
- distributed under the License is distributed on an "AS IS" BASIS,
- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- See the License for the specific language governing permissions and
- limitations under the License.
- ############################################################################## */
- #include "jlib.hpp"
- #include "environment.hpp"
- #include "jptree.hpp"
- #include "jexcept.hpp"
- #include "jiter.ipp"
- #include "jmisc.hpp"
- #include "jencrypt.hpp"
- #include "jutil.hpp"
- #include "mpbase.hpp"
- #include "daclient.hpp"
- #include "dadfs.hpp"
- #include "dafdesc.hpp"
- #include "dasds.hpp"
- #include "dalienv.hpp"
- #include "daqueue.hpp"
- #include <string>
- #include <unordered_map>
- #include <tuple>
- #define SDS_LOCK_TIMEOUT 30000
- #define DEFAULT_DROPZONE_INDEX 1
- #define DROPZONE_BY_MACHINE_SUFFIX "-dropzoneByMachine-"
- #define DROPZONE_SUFFIX "dropzone-"
- #define MACHINE_PREFIX "machine-"
- #define SPARKTHOR_SUFFIX "sparkthor-"
- static int environmentTraceLevel = 1;
- static Owned <IConstEnvironment> cache;
- class CLocalEnvironment;
- class CConstMachineInfoIterator : public CSimpleInterfaceOf<IConstMachineInfoIterator>
- {
- public:
- CConstMachineInfoIterator();
- virtual bool first() override;
- virtual bool next() override;
- virtual bool isValid() override;
- virtual IConstMachineInfo & query() override;
- virtual unsigned count() const override;
- protected:
- Owned<IConstMachineInfo> curr;
- Owned<CLocalEnvironment> constEnv;
- unsigned index = 1;
- unsigned maxIndex = 0;
- };
- class CConstDropZoneServerInfoIterator : public CSimpleInterfaceOf<IConstDropZoneServerInfoIterator>
- {
- public:
- CConstDropZoneServerInfoIterator(const IConstDropZoneInfo * dropZone);
- virtual bool first() override;
- virtual bool next() override;
- virtual bool isValid() override;
- virtual IConstDropZoneServerInfo & query() override;
- virtual unsigned count() const override;
- protected:
- Owned<IConstDropZoneServerInfo> curr;
- Owned<CLocalEnvironment> constEnv;
- Owned<IPropertyTreeIterator> serverListIt;
- unsigned maxIndex = 0;
- };
- class CConstDropZoneInfoIterator : public CSimpleInterfaceOf<IConstDropZoneInfoIterator>
- {
- public:
- CConstDropZoneInfoIterator();
- virtual bool first() override;
- virtual bool next() override;
- virtual bool isValid() override;
- virtual IConstDropZoneInfo & query() override;
- virtual unsigned count() const override;
- protected:
- Owned<IConstDropZoneInfo> curr;
- Owned<CLocalEnvironment> constEnv;
- unsigned index = 1;
- unsigned maxIndex = 0;
- };
- class CConstDfuQueueInfoIterator : public CSimpleInterfaceOf<IConstDfuQueueInfoIterator>
- {
- public:
- CConstDfuQueueInfoIterator();
- virtual bool first() override;
- virtual bool next() override;
- virtual bool isValid() override;
- virtual IConstDfuQueueInfo & query() override;
- virtual unsigned count() const override;
- protected:
- Owned<IConstDfuQueueInfo> curr;
- Owned<CLocalEnvironment> constEnv;
- unsigned index = 1;
- unsigned maxIndex = 0;
- };
- class CConstSparkThorInfoIterator : public CSimpleInterfaceOf<IConstSparkThorInfoIterator>
- {
- public:
- CConstSparkThorInfoIterator();
- virtual bool first() override;
- virtual bool next() override;
- virtual bool isValid() override;
- virtual IConstSparkThorInfo & query() override;
- virtual unsigned count() const override;
- protected:
- Owned<IConstSparkThorInfo> curr;
- Owned<CLocalEnvironment> constEnv;
- unsigned index = 1;
- unsigned maxIndex = 0;
- };
- class CConstInstanceInfoIterator : public CSimpleInterfaceOf<IConstInstanceInfoIterator>
- {
- public:
- CConstInstanceInfoIterator(const CLocalEnvironment * env, IPropertyTreeIterator * itr);
- virtual bool first() override;
- virtual bool next() override;
- virtual bool isValid() override;
- virtual IConstInstanceInfo & query() override;
- virtual unsigned count() const override;
- protected:
- Owned<IPropertyTreeIterator> instanceItr;
- Owned<IConstInstanceInfo> curr;
- const CLocalEnvironment* constEnv;
- unsigned index = 1;
- unsigned maxIndex = 0;
- };
- //==========================================================================================
- class CConstInstanceInfo;
- class CLocalEnvironment : implements IConstEnvironment, public CInterface
- {
- private:
- // NOTE - order is important - we need to construct before p and (especially) destruct after p
- Owned<IRemoteConnection> conn;
- Owned<IPropertyTree> p;
- mutable MapStringToMyClass<IConstEnvBase> cache; // NB: map of 'MappingStringToIInterface' that Link's the added IConstEnvBase, and Release's on element removal.
- mutable Mutex safeCache;
- mutable bool dropZoneCacheBuilt;
- mutable bool machineCacheBuilt;
- mutable bool sparkThorCacheBuilt;
- mutable bool clusterGroupKeyNameCache;
- StringBuffer fileAccessUrl;
- struct KeyPairMapEntity
- {
- std::string publicKey, privateKey;
- };
- mutable std::unordered_map<std::string, KeyPairMapEntity> keyPairMap;
- mutable std::unordered_map<std::string, std::string> keyGroupMap;
- StringBuffer xPath;
- mutable unsigned numOfMachines;
- mutable unsigned numOfDropZones;
- mutable unsigned numOfSparkThors;
- mutable bool isDropZoneRestrictionLoaded = false;
- mutable bool dropZoneRestrictionEnabled = true;
- void buildDfuQueueCache() const;
- mutable unsigned numOfDfuQueues = 0;
- mutable bool dfuQueueCacheBuilt = false;
- IConstEnvBase * getCache(const char *path) const;
- void setCache(const char *path, IConstEnvBase *value) const;
- void buildMachineCache() const;
- void buildDropZoneCache() const;
- void buildSparkThorCache() const;
- void init();
- void ensureClusterGroupKeyMap() const // keyPairMap and keyGroupMap it alters is mutable
- {
- if (!clusterGroupKeyNameCache)
- {
- StringBuffer keysDir;
- getConfigurationDirectory(nullptr, "keys",nullptr, nullptr, keysDir);
- Owned<IPropertyTreeIterator> keyPairIt = p->getElements("EnvSettings/Keys/KeyPair");
- ForEach(*keyPairIt)
- {
- IPropertyTree &keyPair = keyPairIt->query();
- const char *name = keyPair.queryProp("@name");
- const char *publicKeyPath = keyPair.queryProp("@publicKey");
- const char *privateKeyPath = keyPair.queryProp("@privateKey");
- if (isEmptyString(name))
- {
- WARNLOG("skipping invalid EnvSettings/Key/KeyPair entry, name not defined");
- continue;
- }
- if (isEmptyString(publicKeyPath) || isEmptyString(privateKeyPath))
- {
- WARNLOG("skipping invalid EnvSettings/Key/KeyPair entry, name=%s", name);
- continue;
- }
- StringBuffer absPublicKeyPath, absPrivateKeyPath;
- if (!isAbsolutePath(publicKeyPath))
- {
- absPublicKeyPath.append(keysDir);
- addPathSepChar(absPublicKeyPath);
- absPublicKeyPath.append(publicKeyPath);
- }
- else
- absPublicKeyPath.append(publicKeyPath);
- if (!isAbsolutePath(privateKeyPath))
- {
- absPrivateKeyPath.append(keysDir);
- addPathSepChar(absPrivateKeyPath);
- absPrivateKeyPath.append(privateKeyPath);
- }
- else
- absPrivateKeyPath.append(privateKeyPath);
- keyPairMap[name] = { absPublicKeyPath.str(), absPrivateKeyPath.str() };
- }
- /* From 7.0.0 until 7.0.6, the <Keys> section of the environment required
- * the mappings to be defined as "Cluster" instead of "ClusterGroup" - See: HPCC-21192
- */
- #define BKWRDCOMPAT_CLUSTER_VS_CLUSTERGROUP
- const char *groupKeysPath = "EnvSettings/Keys/ClusterGroup";
- #ifdef BKWRDCOMPAT_CLUSTER_VS_CLUSTERGROUP
- for (unsigned i=0; i<2; i++) // once for std. "ClusterGroup", 2nd time for legacy "Cluster"
- {
- #endif
- Owned<IPropertyTreeIterator> clusterGroupIter = p->getElements(groupKeysPath);
- #ifdef BKWRDCOMPAT_CLUSTER_VS_CLUSTERGROUP
- if (clusterGroupIter->first() && keyGroupMap.size()) // NB: always 0 1st time around.
- {
- WARNLOG("Invalid configuration: mixed 'Keys/ClusterGroup' definitions and legacy 'Keys/Cluster' definitions found, legacy 'Keys/Cluster' definition will be ignored.");
- break;
- }
- #endif
- ForEach(*clusterGroupIter)
- {
- IPropertyTree &clusterGroup = clusterGroupIter->query();
- const char *groupName = clusterGroup.queryProp("@name");
- if (isEmptyString(groupName))
- {
- WARNLOG("skipping %s entry with no name", groupKeysPath);
- continue;
- }
- if (clusterGroup.hasProp("@keyPairName"))
- {
- const char *keyPairName = clusterGroup.queryProp("@keyPairName");
- if (isEmptyString(keyPairName))
- {
- WARNLOG("skipping invalid %s entry, name=%s", groupKeysPath, groupName);
- continue;
- }
- keyGroupMap[groupName] = keyPairName;
- }
- }
- #ifdef BKWRDCOMPAT_CLUSTER_VS_CLUSTERGROUP
- groupKeysPath = "EnvSettings/Keys/Cluster";
- }
- #endif
- clusterGroupKeyNameCache = true;
- }
- }
- public:
- IMPLEMENT_IINTERFACE;
- CLocalEnvironment(IRemoteConnection *_conn, IPropertyTree *x=nullptr, const char* path="Environment");
- CLocalEnvironment(const char* path="config.xml");
- virtual ~CLocalEnvironment();
- virtual IStringVal & getName(IStringVal & str) const;
- virtual IStringVal & getXML(IStringVal & str) const;
- virtual IPropertyTree & getPTree() const;
- virtual IEnvironment& lock() const;
- virtual IConstDomainInfo * getDomain(const char * name) const;
- virtual IConstMachineInfo * getMachine(const char * name) const;
- virtual IConstMachineInfo * getMachineByAddress(const char * hostOrIP) const;
- virtual IConstMachineInfo * getMachineForLocalHost() const;
- virtual IConstDropZoneInfo * getDropZone(const char * name) const;
- virtual IConstInstanceInfo * getInstance(const char * type, const char * version, const char *domain) const;
- virtual CConstInstanceInfo * getInstanceByIP(const char *type, const char *version, IpAddress &ip) const;
- virtual IConstComputerTypeInfo * getComputerType(const char * name) const;
- virtual bool getRunInfo(IStringVal & path, IStringVal & dir, const char *type, const char *version, const char *machineaddr, const char *defprogname) const;
- virtual void preload();
- virtual IRemoteConnection* getConnection() const { return conn.getLink(); }
- void setXML(const char * logicalName);
- const char* getPath() const { return xPath.str(); }
- void unlockRemote();
- virtual bool isConstEnvironment() const { return true; }
- virtual void clearCache();
- virtual IConstMachineInfoIterator * getMachineIterator() const;
- virtual IConstDropZoneInfoIterator * getDropZoneIteratorByAddress(const char * address) const;
- virtual IConstDropZoneInfo * getDropZoneByAddressPath(const char * netaddress, const char *targetPath) const;
- virtual IConstDropZoneInfoIterator * getDropZoneIterator() const;
- unsigned getNumberOfMachines() const { buildMachineCache(); return numOfMachines; }
- IConstMachineInfo * getMachineByIndex(unsigned index) const;
- unsigned getNumberOfDropZones() const { buildDropZoneCache(); return numOfDropZones; }
- IConstDropZoneInfo * getDropZoneByIndex(unsigned index) const;
- bool isDropZoneRestrictionEnabled() const;
- unsigned getNumberOfDfuQueues() const { buildDfuQueueCache(); return numOfDfuQueues; }
- IConstDfuQueueInfo * getDfuQueueByIndex(unsigned index) const;
- virtual IConstDfuQueueInfoIterator * getDfuQueueIterator() const;
- bool isValidDfuQueueName(const char * queueName) const;
- virtual const char *getClusterGroupKeyPairName(const char *group) const override
- {
- synchronized procedure(safeCache);
- ensureClusterGroupKeyMap();
- auto it = keyGroupMap.find(group);
- if (it == keyGroupMap.end())
- return nullptr;
- else
- return it->second.c_str();
- }
- virtual const char *getPublicKeyPath(const char *keyPairName) const override
- {
- synchronized procedure(safeCache);
- ensureClusterGroupKeyMap();
- auto it = keyPairMap.find(keyPairName);
- if (it == keyPairMap.end())
- return nullptr;
- else
- return it->second.publicKey.c_str();
- }
- virtual const char *getPrivateKeyPath(const char *keyPairName) const override
- {
- synchronized procedure(safeCache);
- ensureClusterGroupKeyMap();
- auto it = keyPairMap.find(keyPairName);
- if (it == keyPairMap.end())
- return nullptr;
- else
- return it->second.privateKey.c_str();
- }
- virtual const char *getFileAccessUrl() const
- {
- synchronized procedure(safeCache);
- return fileAccessUrl.length() ? fileAccessUrl.str() : nullptr;
- }
- virtual IConstDaFileSrvInfo *getDaFileSrvGroupInfo(const char *name) const override;
- virtual IConstSparkThorInfo *getSparkThor(const char *name) const;
- virtual IConstSparkThorInfoIterator *getSparkThorIterator() const;
- unsigned getNumberOfSparkThors() const { buildSparkThorCache(); return numOfSparkThors; }
- IConstSparkThorInfo *getSparkThorByIndex(unsigned index) const;
- };
- class CLockedEnvironment : implements IEnvironment, public CInterface
- {
- public:
- //note that order of construction/destruction is important
- Owned<CLocalEnvironment> c;
- Owned<CLocalEnvironment> env;
- Owned<CLocalEnvironment> constEnv;
- IMPLEMENT_IINTERFACE;
- CLockedEnvironment(CLocalEnvironment *_c)
- {
- Owned<IRemoteConnection> connection = _c->getConnection();
- if (connection)
- {
- constEnv.set(_c); //save original constant environment
- //we only wish to allow one party to allow updating the environment.
- //
- //create a new /NewEnvironment subtree, locked for read/write access for self and entire subtree; delete on disconnect
- //
- StringBuffer newName("/New");
- newName.append(constEnv->getPath());
- const unsigned int mode = RTM_CREATE | RTM_CREATE_QUERY | RTM_LOCK_READ | RTM_LOCK_WRITE |
- RTM_LOCK_SUB | RTM_DELETE_ON_DISCONNECT;
- Owned<IRemoteConnection> conn = querySDS().connect(newName.str(), myProcessSession(), mode, SDS_LOCK_TIMEOUT);
- if (conn == nullptr)
- {
- if (environmentTraceLevel > 0)
- IERRLOG("Failed to create locked environment %s", newName.str());
- throw MakeStringException(-1, "Failed to get a lock on environment /%s", newName.str());
- }
- //save the locked environment
- env.setown(new CLocalEnvironment(conn, nullptr, newName.str()));
- //get a lock on the const environment
- const unsigned int mode2 = RTM_CREATE_QUERY | RTM_LOCK_READ | RTM_LOCK_WRITE | RTM_LOCK_SUB;
- Owned<IRemoteConnection> conn2 = querySDS().connect(constEnv->getPath(), myProcessSession(), mode2, SDS_LOCK_TIMEOUT);
- if (conn2 == nullptr)
- {
- if (environmentTraceLevel > 0)
- IERRLOG("Failed to lock environment %s", constEnv->getPath());
- throw MakeStringException(-1, "Failed to get a lock on environment /%s", constEnv->getPath());
- }
- //copy const environment to our member environment
- Owned<IPropertyTree> pSrc = conn2->getRoot();
- c.setown( new CLocalEnvironment(nullptr, createPTreeFromIPT(pSrc)));
- conn2->rollback();
- }
- else
- {
- c.set(_c);
- }
- }
- virtual ~CLockedEnvironment()
- {
- }
- virtual IStringVal & getName(IStringVal & str) const
- { return c->getName(str); }
- virtual IStringVal & getXML(IStringVal & str) const
- { return c->getXML(str); }
- virtual IPropertyTree & getPTree() const
- { return c->getPTree(); }
- virtual IConstDomainInfo * getDomain(const char * name) const
- { return c->getDomain(name); }
- virtual IConstMachineInfo * getMachine(const char * name) const
- { return c->getMachine(name); }
- virtual IConstMachineInfo * getMachineByAddress(const char * hostOrIP) const
- { return c->getMachineByAddress(hostOrIP); }
- virtual IConstMachineInfo * getMachineForLocalHost() const
- { return c->getMachineForLocalHost(); }
- virtual IConstDropZoneInfo * getDropZone(const char * name) const
- { return c->getDropZone(name); }
- virtual IConstInstanceInfo * getInstance(const char *type, const char *version, const char *domain) const
- { return c->getInstance(type, version, domain); }
- virtual bool getRunInfo(IStringVal & path, IStringVal & dir, const char *type, const char *version, const char *machineaddr,const char *defprogname) const
- { return c->getRunInfo(path, dir, type, version, machineaddr, defprogname); }
- virtual IConstComputerTypeInfo * getComputerType(const char * name) const
- { return c->getComputerType(name); }
- virtual IEnvironment & lock() const
- { ((CInterface*)this)->Link(); return *(IEnvironment*)this; }
- virtual void commit();
- virtual void rollback();
- virtual void setXML(const char * pstr)
- { c->setXML(pstr); }
- virtual void preload()
- { c->preload(); }
- virtual bool isConstEnvironment() const { return false; }
- virtual void clearCache() { c->clearCache(); }
- virtual IConstMachineInfoIterator * getMachineIterator() const
- { return c->getMachineIterator(); }
- virtual IConstDropZoneInfoIterator * getDropZoneIteratorByAddress(const char * address) const
- { return c->getDropZoneIteratorByAddress(address); }
- virtual IConstDropZoneInfo * getDropZoneByAddressPath(const char * netaddress, const char *targetPath) const
- { return c->getDropZoneByAddressPath(netaddress, targetPath); }
- virtual IConstDropZoneInfoIterator * getDropZoneIterator() const
- { return c->getDropZoneIterator(); }
- virtual bool isDropZoneRestrictionEnabled() const
- { return c->isDropZoneRestrictionEnabled(); }
- virtual const char *getClusterGroupKeyPairName(const char *cluster) const override
- { return c->getClusterGroupKeyPairName(cluster); }
- virtual const char *getPublicKeyPath(const char *keyPairName) const override
- { return c->getPublicKeyPath(keyPairName); }
- virtual const char *getPrivateKeyPath(const char *keyPairName) const override
- { return c->getPrivateKeyPath(keyPairName); }
- virtual const char *getFileAccessUrl() const
- { return c->getFileAccessUrl(); }
- virtual IConstDaFileSrvInfo *getDaFileSrvGroupInfo(const char *name) const override
- { return c->getDaFileSrvGroupInfo(name); }
- virtual IConstSparkThorInfo *getSparkThor(const char *name) const
- { return c->getSparkThor(name); }
- virtual IConstSparkThorInfoIterator *getSparkThorIterator() const
- { return c->getSparkThorIterator(); }
- virtual IConstDfuQueueInfoIterator * getDfuQueueIterator() const
- { return c->getDfuQueueIterator(); }
- virtual bool isValidDfuQueueName(const char * queueName) const
- { return c->isValidDfuQueueName(queueName); }
- };
- void CLockedEnvironment::commit()
- {
- if (constEnv)
- {
- //get a lock on const environment momentarily
- const unsigned int mode2 = RTM_CREATE_QUERY | RTM_LOCK_READ | RTM_LOCK_WRITE | RTM_LOCK_SUB;
- Owned<IRemoteConnection> conn2 = querySDS().connect(constEnv->getPath(), myProcessSession(), mode2, SDS_LOCK_TIMEOUT);
- if (conn2 == nullptr)
- {
- if (environmentTraceLevel > 0)
- IERRLOG("Failed to lock environment %s", constEnv->getPath());
- throw MakeStringException(-1, "Failed to get a lock on environment /%s", constEnv->getPath());
- }
- //copy locked environment to const environment
- Owned<IPropertyTree> pSrc = &getPTree();
- Owned<IPropertyTree> pDst = conn2->queryRoot()->getBranch(nullptr);
- // JCS - I think it could (and would be more efficient if it had kept the original read lock connection to Env
- // - instead of using NewEnv as lock point, still work on copy, then changeMode of original connect
- // - as opposed to current scheme, where it recoonects in write mode and has to lazy fetch original env to update.
- // ensures pDst is equal to pSrc, whilst minimizing changes to pDst
- try { synchronizePTree(pDst, pSrc); }
- catch (IException *) { conn2->rollback(); throw; }
- conn2->commit();
- }
- else
- {
- Owned<IRemoteConnection> conn = c->getConnection();
- conn->commit();
- }
- }
- void CLockedEnvironment::rollback()
- {
- if (constEnv)
- {
- //get a lock on const environment momentarily
- const unsigned int mode2 = RTM_CREATE_QUERY | RTM_LOCK_READ | RTM_LOCK_WRITE | RTM_LOCK_SUB;
- Owned<IRemoteConnection> conn2 = querySDS().connect(constEnv->getPath(), myProcessSession(), mode2, SDS_LOCK_TIMEOUT);
- if (conn2 == nullptr)
- {
- if (environmentTraceLevel > 0)
- IERRLOG("Failed to lock environment %s", constEnv->getPath());
- throw MakeStringException(-1, "Failed to get a lock on environment /%s", constEnv->getPath());
- }
- //copy const environment to locked environment (as it stands now) again losing any changes we made
- Owned<IPropertyTree> pSrc = conn2->getRoot();
- Owned<IPropertyTree> pDst = &getPTree();
- pDst->removeTree( pDst->queryPropTree("Hardware") );
- pDst->removeTree( pDst->queryPropTree("Software") );
- pDst->removeTree( pDst->queryPropTree("Programs") );
- pDst->removeTree( pDst->queryPropTree("Data") );
- mergePTree(pDst, pSrc);
- conn2->rollback();
- }
- else
- {
- Owned<IRemoteConnection> conn = c->getConnection();
- conn->rollback();
- }
- }
- //==========================================================================================
- // the following class implements notification handler for subscription to dali for environment
- // updates by other clients and is used by environment factory below. This also serves as
- // a sample self-contained implementation that can be easily tailored for other purposes.
- //==========================================================================================
- class CSdsSubscription : implements ISDSSubscription, public CInterface
- {
- public:
- CSdsSubscription(IEnvironmentFactory &_factory) : factory(_factory)
- {
- m_constEnvUpdated = false;
- sub_id = factory.subscribe(this);
- }
- virtual ~CSdsSubscription()
- {
- /* note that ideally, we would make this class automatically
- unsubscribe in this destructor. However, underlying dali client
- layer (CDaliSubscriptionManagerStub) links to this object and so
- object would not get destroyed just by an application releasing it.
- The application either needs to explicitly unsubscribe or close
- the environment which unsubscribes during close down. */
- }
- void unsubscribe()
- {
- synchronized block(m_mutexEnv);
- if (sub_id)
- {
- factory.unsubscribe(sub_id);
- sub_id = 0;
- }
- }
- IMPLEMENT_IINTERFACE;
- //another client (like configenv) may have updated the environment and we got notified
- //(thanks to our subscription) but don't just reload it yet since this notification is sent on
- //another thread asynchronously and we may be actively working with the old environment. Just
- //invoke handleEnvironmentChange() when we are ready to invalidate cache in environment factory.
- //
- void notify(SubscriptionId id, const char *xpath, SDSNotifyFlags flags, unsigned valueLen=0, const void *valueData=nullptr)
- {
- DBGLOG("Environment was updated by another client of Dali server. Invalidating cache.\n");
- synchronized block(m_mutexEnv);
- m_constEnvUpdated = true;
- }
- void handleEnvironmentChange()
- {
- synchronized block(m_mutexEnv);
- if (m_constEnvUpdated)
- {
- Owned<IConstEnvironment> constEnv = factory.openEnvironment();
- constEnv->clearCache();
- m_constEnvUpdated = false;
- }
- }
- private:
- SubscriptionId sub_id;
- Mutex m_mutexEnv;
- bool m_constEnvUpdated;
- IEnvironmentFactory &factory;
- };
- //==========================================================================================
- class CEnvironmentFactory : public CInterface,
- implements IEnvironmentFactory, implements IDaliClientShutdown
- {
- public:
- IMPLEMENT_IINTERFACE;
- typedef ArrayOf<SubscriptionId> SubscriptionIDs;
- SubscriptionIDs subIDs;
- Mutex mutex;
- Owned<CSdsSubscription> subscription;
- CEnvironmentFactory()
- {
- }
- virtual void clientShutdown();
- virtual ~CEnvironmentFactory()
- {
- close(); //just in case it was not explicitly closed
- }
- virtual IConstEnvironment* openEnvironment()
- {
- synchronized procedure(mutex);
- if (!cache)
- {
- Owned<IRemoteConnection> conn = querySDS().connect("/Environment", myProcessSession(), 0, SDS_LOCK_TIMEOUT);
- if (conn)
- cache.setown(new CLocalEnvironment(conn));
- }
- if (!cache)
- throw MakeStringException(0, "Failed to get environment information");
- return cache.getLink();
- }
- virtual IEnvironment* updateEnvironment()
- {
- Owned<IConstEnvironment> pConstEnv = openEnvironment();
- synchronized procedure(mutex);
- return &pConstEnv->lock();
- }
- virtual IEnvironment * loadLocalEnvironmentFile(const char * filename)
- {
- Owned<IPropertyTree> ptree = createPTreeFromXMLFile(filename, ipt_lowmem);
- Owned<CLocalEnvironment> pLocalEnv = new CLocalEnvironment(nullptr, ptree);
- return new CLockedEnvironment(pLocalEnv);
- }
- virtual IEnvironment * loadLocalEnvironment(const char * xml)
- {
- Owned<IPropertyTree> ptree = createPTreeFromXMLString(xml, ipt_lowmem);
- Owned<CLocalEnvironment> pLocalEnv = new CLocalEnvironment(nullptr, ptree);
- return new CLockedEnvironment(pLocalEnv);
- }
- void close()
- {
- SubscriptionIDs copySubIDs;
- {
- synchronized procedure(mutex);
- cache.clear();
- //save the active subscriptions in another array
- //so they can be unsubscribed without causing deadlock
- // since ~CSdsSubscription() would ask us to unsubscribe the
- //same requiring a mutex lock (copy is a little price for this
- //normally small/empty array).
- //
- ForEachItemIn(i, subIDs)
- copySubIDs.append(subIDs.item(i));
- subIDs.kill();
- }
- //now unsubscribe all outstanding subscriptions
- //
- subscription.clear();
- ForEachItemIn(i, copySubIDs)
- querySDS().unsubscribe( copySubIDs.item(i) );
- }
- virtual SubscriptionId subscribe(ISDSSubscription* pSubHandler)
- {
- SubscriptionId sub_id = querySDS().subscribe("/Environment", *pSubHandler);
- synchronized procedure(mutex);
- subIDs.append(sub_id);
- return sub_id;
- }
- virtual void unsubscribe(SubscriptionId sub_id)
- {
- synchronized procedure(mutex);
- aindex_t i = subIDs.find(sub_id);
- if (i != NotFound)
- {
- querySDS().unsubscribe(sub_id);
- subIDs.remove(i);
- }
- }
- virtual void validateCache()
- {
- if (!subscription)
- subscription.setown( new CSdsSubscription(*this) );
- subscription->handleEnvironmentChange();
- }
- private:
- IRemoteConnection* connect(const char *xpath, unsigned flags)
- {
- return querySDS().connect(xpath, myProcessSession(), flags, SDS_LOCK_TIMEOUT);
- }
- };
- static CEnvironmentFactory *factory=nullptr;
- void CEnvironmentFactory::clientShutdown()
- {
- closeEnvironment();
- }
- MODULE_INIT(INIT_PRIORITY_ENV_ENVIRONMENT)
- {
- return true;
- }
- MODULE_EXIT()
- {
- ::Release(factory);
- }
- //==========================================================================================
- class CConstEnvBase : public CInterface
- {
- protected:
- const CLocalEnvironment* env; // Not linked - would be circular....
- // That could cause problems
- Linked<IPropertyTree> root;
- public:
- CConstEnvBase(const CLocalEnvironment* _env, IPropertyTree *_root)
- : env(_env), root(_root)
- {
- }
- IStringVal& getXML(IStringVal &str) const
- {
- StringBuffer x;
- toXML(root->queryBranch("."), x);
- str.set(x.str());
- return str;
- };
- IStringVal& getName(IStringVal &str) const
- {
- str.set(root->queryProp("@name"));
- return str;
- }
- IPropertyTree& getPTree() const
- {
- return *LINK(root);
- }
- };
- #define IMPLEMENT_ICONSTENVBASE \
- virtual IStringVal& getXML(IStringVal &str) const { return CConstEnvBase::getXML(str); } \
- virtual IStringVal& getName(IStringVal &str) const { return CConstEnvBase::getName(str); } \
- virtual IPropertyTree& getPTree() const { return CConstEnvBase::getPTree(); }
- //==========================================================================================
- class CConstDomainInfo : public CConstEnvBase, implements IConstDomainInfo
- {
- public:
- IMPLEMENT_IINTERFACE;
- IMPLEMENT_ICONSTENVBASE;
- CConstDomainInfo(CLocalEnvironment *env, IPropertyTree *root) : CConstEnvBase(env, root) {}
- virtual void getAccountInfo(IStringVal &name, IStringVal &pw) const
- {
- if (root->hasProp("@username"))
- name.set(root->queryProp("@username"));
- else
- name.clear();
- if (root->hasProp("@password"))
- {
- StringBuffer pwd;
- decrypt(pwd, root->queryProp("@password"));
- pw.set(pwd.str());
- }
- else
- pw.clear();
- }
- virtual void getSnmpSecurityString(IStringVal & securityString) const
- {
- if (root->hasProp("@snmpSecurityString"))
- {
- StringBuffer sec_string;
- decrypt(sec_string, root->queryProp("@snmpSecurityString"));
- securityString.set(sec_string.str());
- }
- else
- securityString.set("");
- }
- virtual void getSSHAccountInfo(IStringVal &name, IStringVal &sshKeyFile, IStringVal& sshKeyPassphrase) const
- {
- if (root->hasProp("@username"))
- name.set(root->queryProp("@username"));
- else
- name.clear();
- if (root->hasProp("@sshKeyFile"))
- sshKeyFile.set(root->queryProp("@sshKeyFile"));
- else
- sshKeyFile.clear();
- if (root->hasProp("@sshKeyPassphrase"))
- sshKeyPassphrase.set(root->queryProp("@sshKeyPassphrase"));
- else
- sshKeyPassphrase.clear();
- }
- };
- //==========================================================================================
- struct mapOsEnums { EnvMachineOS val; const char *str; };
- static EnvMachineOS getEnum(IPropertyTree *p, const char *propname, mapOsEnums *map)
- {
- const char *v = p->queryProp(propname);
- if (v && *v)
- {
- while (map->str)
- {
- if (stricmp(v, map->str)==0)
- return map->val;
- map++;
- }
- throw MakeStringException(0, "Unknown operating system: \"%s\"", v);
- }
- return MachineOsUnknown;
- }
- struct mapStateEnums { EnvMachineState val; const char *str; };
- static EnvMachineState getEnum(IPropertyTree *p, const char *propname, mapStateEnums *map)
- {
- const char *v = p->queryProp(propname);
- if (v && *v)
- {
- while (map->str)
- {
- if (stricmp(v, map->str)==0)
- return map->val;
- map++;
- }
- assertex(!"Unexpected value in getEnum");
- }
- return MachineStateUnknown;
- }
- mapOsEnums OperatingSystems[] = {
- { MachineOsW2K, "W2K" },
- { MachineOsSolaris, "solaris" },
- { MachineOsLinux, "linux" },
- { MachineOsSize, nullptr }
- };
- mapStateEnums MachineStates[] = {
- { MachineStateAvailable, "Available" },
- { MachineStateUnavailable, "Unavailable" },
- { MachineStateUnknown, "Unknown" }
- };
- //==========================================================================================
- class CConstMachineInfo : public CConstEnvBase, implements IConstMachineInfo
- {
- public:
- IMPLEMENT_IINTERFACE;
- IMPLEMENT_ICONSTENVBASE;
- CConstMachineInfo(CLocalEnvironment *env, IPropertyTree *root) : CConstEnvBase(env, root) {}
- virtual IConstDomainInfo* getDomain() const
- {
- return env->getDomain(root->queryProp("@domain"));
- }
- virtual IStringVal& getNetAddress(IStringVal &str) const
- {
- str.set(root->queryProp("@netAddress"));
- return str;
- }
- virtual IStringVal& getDescription(IStringVal &str) const
- {
- UNIMPLEMENTED;
- }
- virtual unsigned getNicSpeedMbitSec() const
- {
- const char * v = root->queryProp("@nicSpeed");
- if (v && *v)
- return atoi(v);
- Owned<IConstComputerTypeInfo> type = env->getComputerType(root->queryProp("@computerType"));
- if (type)
- return type->getNicSpeedMbitSec();
- return 0;
- }
- virtual EnvMachineOS getOS() const
- {
- EnvMachineOS os = getEnum(root, "@opSys", OperatingSystems);
- if (os != MachineOsUnknown)
- return os;
- Owned<IConstComputerTypeInfo> type = env->getComputerType(root->queryProp("@computerType"));
- if (type)
- return type->getOS();
- return MachineOsUnknown;
- }
- virtual EnvMachineState getState() const
- {
- return getEnum(root, "@state", MachineStates);
- }
- };
- //==========================================================================================
- class CConstComputerTypeInfo : public CConstEnvBase, implements IConstComputerTypeInfo
- {
- public:
- IMPLEMENT_IINTERFACE;
- IMPLEMENT_ICONSTENVBASE;
- CConstComputerTypeInfo(CLocalEnvironment *env, IPropertyTree *root) : CConstEnvBase(env, root) {}
- virtual EnvMachineOS getOS() const
- {
- EnvMachineOS os = getEnum(root, "@opSys", OperatingSystems);
- if (os != MachineOsUnknown)
- return os;
- Owned<IConstComputerTypeInfo> type = env->getComputerType(root->queryProp("@computerType"));
- if (type && (type.get() != this))
- return type->getOS();
- return MachineOsUnknown;
- }
- virtual unsigned getNicSpeedMbitSec() const
- {
- const char * v = root->queryProp("@nicSpeed");
- if (v && *v)
- return atoi(v);
- Owned<IConstComputerTypeInfo> type = env->getComputerType(root->queryProp("@computerType"));
- if (type && (type.get() != this))
- return type->getNicSpeedMbitSec();
- return 0;
- }
- };
- //==========================================================================================
- extern ENVIRONMENT_API unsigned __int64 readSizeSetting(const char * sizeStr, const unsigned long defaultSize)
- {
- StringBuffer buf(sizeStr);
- buf.trim();
- if (buf.isEmpty())
- return defaultSize;
- const char* ptrStart = buf;
- const char* ptrAfterDigit = ptrStart;
- while (*ptrAfterDigit && isdigit(*ptrAfterDigit))
- ptrAfterDigit++;
- if (!*ptrAfterDigit)
- return atol(buf);
- const char* ptr = ptrAfterDigit;
- while (*ptr && (ptr[0] == ' '))
- ptr++;
- char c = ptr[0];
- buf.setLength(ptrAfterDigit - ptrStart);
- unsigned __int64 size = atoll(buf);
- switch (c)
- {
- case 'k':
- case 'K':
- size *= 1000;
- break;
- case 'm':
- case 'M':
- size *= 1000000;
- break;
- case 'g':
- case 'G':
- size *= 1000000000;
- break;
- case 't':
- case 'T':
- size *= 1000000000000;
- break;
- default:
- break;
- }
- return size;
- }
- class CConstInstanceInfo : public CConstEnvBase, implements IConstInstanceInfo
- {
- public:
- IMPLEMENT_IINTERFACE;
- IMPLEMENT_ICONSTENVBASE;
- CConstInstanceInfo(const CLocalEnvironment *env, IPropertyTree *root) : CConstEnvBase(env, root)
- {
- }
- virtual IConstMachineInfo * getMachine() const
- {
- return env->getMachine(root->queryProp("@computer"));
- }
- virtual IStringVal & getEndPoint(IStringVal & str) const
- {
- SCMStringBuffer ep;
- Owned<IConstMachineInfo> machine = getMachine();
- if (machine)
- {
- machine->getNetAddress(ep);
- const char *port = root->queryProp("@port");
- if (port)
- ep.s.append(':').append(port);
- }
- str.set(ep.str());
- return str;
- }
- virtual IStringVal & getExecutableDirectory(IStringVal & str) const
- {
- // this is the deploy directory so uses local path separators (I suspect this call is LEGACY now)
- SCMStringBuffer ep;
- Owned<IConstMachineInfo> machine = getMachine();
- if (machine)
- {
- machine->getNetAddress(ep);
- ep.s.insert(0, PATHSEPSTR PATHSEPSTR);
- }
- ep.s.append(PATHSEPCHAR).append(root->queryProp("@directory"));
- str.set(ep.str());
- return str;
- }
- virtual IStringVal & getDirectory(IStringVal & str) const
- {
- str.set(root->queryProp("@directory"));
- return str;
- }
- virtual bool doGetRunInfo(IStringVal & progpath, IStringVal & workdir, const char *defprogname, bool useprog) const
- {
- // this is remote path i.e. path should match *target* nodes format
- Owned<IConstMachineInfo> machine = getMachine();
- if (!machine)
- return false;
- char psep;
- bool appendexe;
- switch (machine->getOS())
- {
- case MachineOsSolaris:
- case MachineOsLinux:
- psep = '/';
- appendexe = false;
- break;
- default:
- psep = '\\';
- appendexe = true;
- }
- StringBuffer tmp;
- const char *program = useprog?root->queryProp("@program"):nullptr; // if program specified assume absolute
- if (!program||!*program)
- {
- SCMStringBuffer ep;
- machine->getNetAddress(ep);
- const char *dir = root->queryProp("@directory");
- if (dir)
- {
- if (isPathSepChar(*dir))
- dir++;
- if (!*dir)
- return false;
- tmp.append(psep).append(psep).append(ep.s).append(psep);
- do {
- if (isPathSepChar(*dir))
- tmp.append(psep);
- else
- tmp.append(*dir);
- dir++;
- } while (*dir);
- if (!isPathSepChar(tmp.charAt(tmp.length()-1)))
- tmp.append(psep);
- tmp.append(defprogname);
- size32_t l = strlen(defprogname);
- if (appendexe&&((l<5)||(stricmp(defprogname+l-4,".exe")!=0)))
- tmp.append(".exe");
- }
- program = tmp.str();
- }
- progpath.set(program);
- const char *workd = root->queryProp("@workdir"); // if program specified assume absolute
- workdir.set(workd?workd:"");
- return true;
- }
- virtual bool getRunInfo(IStringVal & progpath, IStringVal & workdir, const char *defprogname) const
- {
- return doGetRunInfo(progpath,workdir,defprogname,true);
- }
- virtual unsigned getPort() const
- {
- return root->getPropInt("@port", 0);
- }
- };
- class CConstDropZoneServerInfo : public CConstEnvBase, implements IConstDropZoneServerInfo
- {
- public:
- IMPLEMENT_IINTERFACE;
- IMPLEMENT_ICONSTENVBASE;
- CConstDropZoneServerInfo(CLocalEnvironment *env, IPropertyTree *root) : CConstEnvBase(env, root), prop(root) {}
- virtual StringBuffer & getName(StringBuffer & name) const
- {
- name.append(prop->queryProp("@name"));
- return name;
- }
- virtual StringBuffer & getServer(StringBuffer & server) const
- {
- server.append(prop->queryProp("@server"));
- return server;
- }
- private:
- IPropertyTree * prop;
- };
- class CConstDropZoneInfo : public CConstEnvBase, implements IConstDropZoneInfo
- {
- public:
- IMPLEMENT_IINTERFACE;
- IMPLEMENT_ICONSTENVBASE;
- CConstDropZoneInfo(CLocalEnvironment *env, IPropertyTree *root) : CConstEnvBase(env, root)
- {
- getStandardPosixPath(posixPath, root->queryProp("@directory"));
- }
- virtual IStringVal& getComputerName(IStringVal &str) const
- {
- str.set(root->queryProp("@computer"));
- return str;
- }
- virtual IStringVal& getDescription(IStringVal &str) const
- {
- str.set(root->queryProp("@description"));
- return str;
- }
- virtual IStringVal& getDirectory(IStringVal &str) const
- {
- str.set(posixPath.str());
- return str;
- }
- virtual IStringVal& getUMask(IStringVal &str) const
- {
- if (root->hasProp("@umask"))
- str.set(root->queryProp("@umask"));
- return str;
- }
- virtual bool isECLWatchVisible() const
- {
- return root->getPropBool("@ECLWatchVisible", true);
- }
- virtual IConstDropZoneServerInfoIterator * getServers() const
- {
- return new CConstDropZoneServerInfoIterator(this);
- }
- private:
- StringBuffer posixPath;
- };
- class CConstDfuQueueInfo : public CConstEnvBase, implements IConstDfuQueueInfo
- {
- public:
- IMPLEMENT_IINTERFACE;
- IMPLEMENT_ICONSTENVBASE;
- CConstDfuQueueInfo(CLocalEnvironment *env, IPropertyTree *root) : CConstEnvBase(env, root)
- {
- }
- virtual IStringVal& getDfuQueueName(IStringVal &str) const
- {
- str.set(root->queryProp("@queue"));
- return str;
- }
- };
- class CConstSparkThorInfo : public CConstEnvBase, implements IConstSparkThorInfo
- {
- public:
- IMPLEMENT_IINTERFACE;
- IMPLEMENT_ICONSTENVBASE;
- CConstSparkThorInfo(CLocalEnvironment *env, IPropertyTree *root) : CConstEnvBase(env, root) {}
- virtual IStringVal &getBuild(IStringVal &str) const
- {
- str.set(root->queryProp("@build"));
- return str;
- }
- virtual IStringVal &getThorClusterName(IStringVal &str) const
- {
- str.set(root->queryProp("@ThorClusterName"));
- return str;
- }
- virtual unsigned getSparkExecutorCores() const
- {
- return root->getPropInt("@SPARK_EXECUTOR_CORES", 0);
- }
- virtual unsigned __int64 getSparkExecutorMemory() const
- {
- return readSizeSetting(root->queryProp("@SPARK_EXECUTOR_MEMORY"), 0);
- }
- virtual unsigned getSparkMasterPort() const
- {
- return root->getPropInt("@SPARK_MASTER_PORT", 0);
- }
- virtual unsigned getSparkMasterWebUIPort() const
- {
- return root->getPropInt("@SPARK_MASTER_WEBUI_PORT", 0);
- }
- virtual unsigned getSparkWorkerCores() const
- {
- return root->getPropInt("@SPARK_WORKER_CORES", 0);
- }
- virtual unsigned __int64 getSparkWorkerMemory() const
- {
- return readSizeSetting(root->queryProp("@SPARK_WORKER_MEMORY"), 0);
- }
- virtual unsigned getSparkWorkerPort() const
- {
- return root->getPropInt("@SPARK_WORKER_PORT", 0);
- }
- virtual IConstInstanceInfoIterator *getInstanceIterator() const
- {
- return new CConstInstanceInfoIterator(env, root->getElements("Instance"));
- }
- };
- #if 0
- //==========================================================================================
- class CConstProcessInfo : public CConstEnvBase, implements IConstProcessInfo
- {
- IArrayOf<IConstInstanceInfo> w;
- CArrayIteratorOf<IInterface, IIterator> it;
- public:
- IMPLEMENT_IINTERFACE;
- IMPLEMENT_ICONSTENVBASE;
- CConstProcessInfo(CLocalEnvironment *env, IPropertyTree *root) : CConstEnvBase(env, root), it(w)
- {
- Owned<IPropertyTreeIterator> _it = root->getElements("*"); // MORE - should be instance
- for (_it->first(); _it->isValid(); _it->next())
- {
- IPropertyTree *rp = &_it->query();
- w.append(*new CConstInstanceInfo(env, rp)); // CConstInstanceInfo will link rp
- }
- }
- bool first() { return it.first(); }
- bool isValid() { return it.isValid(); }
- bool next() { return it.next(); }
- IConstInstanceInfo & query() { return (IConstInstanceInfo &) it.query();}
- virtual IConstInstanceInfo * getInstance(const char *domain)
- {
- for (int pass=0; pass<2; pass++)
- {
- ForEachItemIn(idx, w)
- {
- Owned<IConstMachineInfo> m = w.item(idx).getMachine();
- if (m)
- {
- Owned<IConstDomainInfo> dm = m->getDomain();
- if (dm)
- {
- StringBuffer thisdomain;
- //dm->getName(StringBufferAdaptor(thisdomain)); // confuses g++
- StringBufferAdaptor strval(thisdomain);
- dm->getName(strval);
- if (thisdomain.length() && strcmp(domain, thisdomain.str())==0)
- return LINK(&w.item(idx));
- }
- }
- }
- }
- return nullptr;
- }
- };
- #endif
- class CConstDaFileSrvInfo : public CConstEnvBase, implements IConstDaFileSrvInfo
- {
- public:
- IMPLEMENT_IINTERFACE;
- IMPLEMENT_ICONSTENVBASE;
- CConstDaFileSrvInfo(const CLocalEnvironment *env, IPropertyTree *root) : CConstEnvBase(env, root)
- {
- }
- virtual const char *getName() const override
- {
- return root->queryProp("@name");
- }
- virtual unsigned getPort() const override
- {
- return root->getPropInt("@rowServicePort");
- }
- virtual bool getSecure() const override
- {
- return root->getPropBool("@rowServiceSSL");
- }
- };
- //==========================================================================================
- CLocalEnvironment::CLocalEnvironment(const char* environmentFile)
- {
- if (environmentFile && *environmentFile)
- {
- IPropertyTree* root = createPTreeFromXMLFile(environmentFile);
- if (root)
- p.set(root);
- }
- init();
- }
- CLocalEnvironment::CLocalEnvironment(IRemoteConnection *_conn, IPropertyTree* root/*=nullptr*/,
- const char* path/*="/Environment"*/)
- : xPath(path)
- {
- conn.set(_conn);
- if (root)
- p.set(root);
- else
- p.setown(conn->getRoot());
- init();
- }
- void CLocalEnvironment::init()
- {
- machineCacheBuilt = false;
- dropZoneCacheBuilt = false;
- sparkThorCacheBuilt = false;
- dfuQueueCacheBuilt = false;
- numOfMachines = 0;
- numOfDropZones = 0;
- numOfSparkThors = 0;
- numOfDfuQueues = 0;
- isDropZoneRestrictionLoaded = false;
- clusterGroupKeyNameCache = false;
- ::getFileAccessUrl(fileAccessUrl);
- }
- CLocalEnvironment::~CLocalEnvironment()
- {
- if (conn)
- conn->rollback();
- }
- IEnvironment& CLocalEnvironment::lock() const
- {
- return *new CLockedEnvironment((CLocalEnvironment*)this);
- }
- IStringVal & CLocalEnvironment::getName(IStringVal & str) const
- {
- synchronized procedure(safeCache);
- str.set(p->queryProp("@name"));
- return str;
- }
- IStringVal & CLocalEnvironment::getXML(IStringVal & str) const
- {
- StringBuffer xml;
- {
- synchronized procedure(safeCache);
- toXML(p->queryBranch("."), xml);
- }
- str.set(xml.str());
- return str;
- }
- IPropertyTree & CLocalEnvironment::getPTree() const
- {
- synchronized procedure(safeCache);
- return *LINK(p);
- }
- IConstEnvBase * CLocalEnvironment::getCache(const char *path) const
- {
- IConstEnvBase * ret = cache.getValue(path);
- ::Link(ret);
- return ret;
- }
- void CLocalEnvironment::setCache(const char *path, IConstEnvBase *value) const
- {
- cache.setValue(path, value);
- }
- IConstDomainInfo * CLocalEnvironment::getDomain(const char * name) const
- {
- if (!name)
- return nullptr;
- StringBuffer xpath;
- xpath.appendf("Hardware/Domain[@name=\"%s\"]", name);
- synchronized procedure(safeCache);
- IConstEnvBase *cached = getCache(xpath.str());
- if (!cached)
- {
- IPropertyTree *d = p->queryPropTree(xpath.str());
- if (!d)
- return nullptr;
- cached = new CConstDomainInfo((CLocalEnvironment *) this, d);
- setCache(xpath.str(), cached);
- }
- return (IConstDomainInfo *) cached;
- }
- void CLocalEnvironment::buildMachineCache() const
- {
- synchronized procedure(safeCache);
- if (!machineCacheBuilt)
- {
- Owned<IPropertyTreeIterator> it = p->getElements("Hardware/Computer");
- ForEach(*it)
- {
- Owned<IConstEnvBase> cached = new CConstMachineInfo((CLocalEnvironment *) this, &it->query());
- const char *name = it->query().queryProp("@name");
- if (name)
- {
- StringBuffer x("Hardware/Computer[@name=\"");
- x.append(name).append("\"]");
- cache.setValue(x.str(), cached);
- }
- const char * netAddress = it->query().queryProp("@netAddress");
- if (netAddress)
- {
- StringBuffer x("Hardware/Computer[@netAddress=\"");
- x.append(netAddress).append("\"]");
- cache.setValue(x.str(), cached);
- IpAddress ip;
- ip.ipset(netAddress);
- if (ip.isLocal())
- cache.setValue("Hardware/Computer[@netAddress=\".\"]", cached);
- }
- numOfMachines++;
- StringBuffer x("Hardware/Computer[@id=\"");
- x.append(MACHINE_PREFIX).append(numOfMachines).append("\"]");
- cache.setValue(x.str(), cached);
- }
- machineCacheBuilt = true;
- }
- }
- void CLocalEnvironment::buildDropZoneCache() const
- {
- synchronized procedure(safeCache);
- if (!dropZoneCacheBuilt)
- {
- Owned<IPropertyTreeIterator> it = p->getElements("Software/DropZone");
- ForEach(*it)
- {
- const char *name = it->query().queryProp("@name");
- if (name)
- {
- StringBuffer x("Software/DropZone[@name=\"");
- x.append(name).append("\"]");
- Owned<IConstEnvBase> cached = new CConstDropZoneInfo((CLocalEnvironment *) this, &it->query());
- cache.setValue(x.str(), cached);
- }
- numOfDropZones++;
- StringBuffer x("Software/DropZone[@id=\"");
- x.append(DROPZONE_SUFFIX).append(numOfDropZones).append("\"]");
- Owned<IConstEnvBase> cached = new CConstDropZoneInfo((CLocalEnvironment *) this, &it->query());
- cache.setValue(x.str(), cached);
- }
- dropZoneCacheBuilt = true;
- }
- }
- void CLocalEnvironment::buildDfuQueueCache() const
- {
- synchronized procedure(safeCache);
- if (!dfuQueueCacheBuilt)
- {
- Owned<IPropertyTreeIterator> it = p->getElements("Software/DfuServerProcess");
- ForEach(*it)
- {
- const char *qname = it->query().queryProp("@queue");
- if (qname)
- {
- StringBuffer x("Software/DfuQueue[@qname=\"");
- x.append(qname).append("\"]");
- Owned<IConstEnvBase> cached = new CConstDfuQueueInfo((CLocalEnvironment *) this, &it->query());
- cache.setValue(x.str(), cached);
- }
- numOfDfuQueues++;
- StringBuffer x("Software/DfuQueue[@id=\"");
- x.append(numOfDfuQueues).append("\"]");
- Owned<IConstEnvBase> cached = new CConstDfuQueueInfo((CLocalEnvironment *) this, &it->query());
- cache.setValue(x.str(), cached);
- }
- dfuQueueCacheBuilt = true;
- }
- }
- IConstComputerTypeInfo * CLocalEnvironment::getComputerType(const char * name) const
- {
- if (!name)
- return nullptr;
- StringBuffer xpath;
- xpath.appendf("Hardware/ComputerType[@name=\"%s\"]", name);
- synchronized procedure(safeCache);
- IConstEnvBase *cached = getCache(xpath.str());
- if (!cached)
- {
- IPropertyTree *d = p->queryPropTree(xpath.str());
- if (!d)
- return nullptr;
- cached = new CConstComputerTypeInfo((CLocalEnvironment *) this, d);
- setCache(xpath.str(), cached);
- }
- return (CConstComputerTypeInfo *) cached;
- }
- IConstMachineInfo * CLocalEnvironment::getMachine(const char * name) const
- {
- if (!name)
- return nullptr;
- buildMachineCache();
- StringBuffer xpath;
- xpath.appendf("Hardware/Computer[@name=\"%s\"]", name);
- synchronized procedure(safeCache);
- IConstEnvBase *cached = getCache(xpath.str());
- if (!cached)
- {
- IPropertyTree *d = p->queryPropTree(xpath.str());
- if (!d)
- return nullptr;
- cached = new CConstMachineInfo((CLocalEnvironment *) this, d);
- setCache(xpath.str(), cached);
- }
- return (CConstMachineInfo *) cached;
- }
- IConstMachineInfo * CLocalEnvironment::getMachineByAddress(const char * hostOrIP) const
- {
- if (isEmptyString(hostOrIP))
- return nullptr;
- buildMachineCache();
- synchronized procedure(safeCache);
- VStringBuffer xpath("Hardware/Computer[@netAddress=\"%s\"]", hostOrIP);
- IConstEnvBase *cached = getCache(hostOrIP);
- if (cached)
- return (CConstMachineInfo *) cached;
- IPropertyTree *d = p->queryPropTree(xpath); // exact match
- if (!d && !isIPAddress(xpath)) // if not found and not an IP, resolve and match against resolved entries
- {
- IpAddress ip(hostOrIP);
- Owned<IPropertyTreeIterator> iter = p->getElements("Hardware/Computer");
- ForEach(*iter)
- {
- IPropertyTree &computer = iter->query();
- IpAddress computerIP;
- const char *computerNetAddress = computer.queryProp("@netAddress");
- if (!isEmptyString(computerNetAddress))
- {
- // NB: could 1st check if computerNetAddress isIPAddress() and not bother resolving here if it is.
- computerIP.ipset(computerNetAddress);
- if (ip.ipequals(computerIP))
- {
- d = &computer;
- break;
- }
- }
- }
- }
- if (!d)
- return nullptr;
- cached = new CConstMachineInfo((CLocalEnvironment *) this, d);
- setCache(xpath.str(), cached);
- return (CConstMachineInfo *) cached;
- }
- IConstMachineInfo * CLocalEnvironment::getMachineByIndex(unsigned index) const
- {
- if (!numOfMachines || (index == 0))
- return nullptr;
- buildMachineCache();
- if (index > numOfMachines)
- return nullptr;
- StringBuffer xpath("Hardware/Computer[@id=\"");
- xpath.append(MACHINE_PREFIX).append(index).append("\"]");
- synchronized procedure(safeCache);
- return (IConstMachineInfo *) getCache(xpath.str());
- }
- IConstMachineInfo * CLocalEnvironment::getMachineForLocalHost() const
- {
- buildMachineCache();
- synchronized procedure(safeCache);
- return (CConstMachineInfo *) getCache("Hardware/Computer[@netAddress=\".\"]");
- }
- IConstDropZoneInfo * CLocalEnvironment::getDropZone(const char * name) const
- {
- if (!name)
- return nullptr;
- buildDropZoneCache();
- VStringBuffer xpath("Software/DropZone[@name=\"%s\"]", name);
- synchronized procedure(safeCache);
- return (CConstDropZoneInfo *) getCache(xpath.str());
- }
- IConstDropZoneInfo * CLocalEnvironment::getDropZoneByIndex(unsigned index) const
- {
- if (!numOfDropZones || (index == 0))
- return nullptr;
- buildDropZoneCache();
- if (index > numOfDropZones)
- return nullptr;
- StringBuffer xpath("Software/DropZone[@id=\"");
- xpath.append(DROPZONE_SUFFIX).append(index).append("\"]");
- synchronized procedure(safeCache);
- return (CConstDropZoneInfo *) getCache(xpath.str());
- }
- IConstDfuQueueInfo * CLocalEnvironment::getDfuQueueByIndex(unsigned index) const
- {
- if (!numOfDfuQueues || (index == 0))
- return nullptr;
- buildDfuQueueCache();
- if (index > numOfDfuQueues)
- return nullptr;
- StringBuffer xpath("Software/DfuQueue[@id=\"");
- xpath.append(index).append("\"]");
- synchronized procedure(safeCache);
- return (CConstDfuQueueInfo *) getCache(xpath.str());
- }
- IConstInstanceInfo * CLocalEnvironment::getInstance(const char *type, const char *version, const char *domain) const
- {
- StringBuffer xpath("Software/");
- xpath.append(type);
- if (version)
- xpath.append("[@version='").append(version).append("']");
- xpath.append("/Instance");
- synchronized procedure(safeCache);
- Owned<IPropertyTreeIterator> _it = p->getElements(xpath);
- for (_it->first(); _it->isValid(); _it->next())
- {
- IPropertyTree *rp = &_it->query();
- Owned<CConstInstanceInfo> inst = new CConstInstanceInfo(this, rp); // CConstInstanceInfo will link rp
- Owned<IConstMachineInfo> m = inst->getMachine();
- if (m)
- {
- Owned<IConstDomainInfo> dm = m->getDomain();
- if (dm)
- {
- SCMStringBuffer thisdomain;
- dm->getName(thisdomain);
- if (thisdomain.length() && strcmp(domain, thisdomain.str())==0)
- return inst.getClear();
- }
- }
- }
- return nullptr;
- }
- CConstInstanceInfo * CLocalEnvironment::getInstanceByIP(const char *type, const char *version, IpAddress &ip) const
- {
- StringBuffer xpath("Software/");
- xpath.append(type);
- if (version)
- xpath.append("[@version='").append(version).append("']");
- xpath.append("/Instance");
- synchronized procedure(safeCache);
- assertex(p);
- Owned<IPropertyTreeIterator> _it = p->getElements(xpath);
- assertex(_it);
- for (_it->first(); _it->isValid(); _it->next())
- {
- IPropertyTree *rp = &_it->query();
- assertex(rp);
- Owned<CConstInstanceInfo> inst = new CConstInstanceInfo(this, rp); // CConstInstanceInfo will link rp
- Owned<IConstMachineInfo> m = inst->getMachine();
- if (m)
- {
- SCMStringBuffer eps;
- m->getNetAddress(eps);
- SocketEndpoint ep(eps.str());
- if (ep.ipequals(ip))
- return inst.getClear();
- }
- }
- return nullptr;
- }
- void CLocalEnvironment::unlockRemote()
- {
- #if 0
- conn->commit(true);
- conn->changeMode(0, SDS_LOCK_TIMEOUT);
- #else
- if (conn)
- {
- synchronized procedure(safeCache);
- p.clear();
- conn.setown(querySDS().connect(xPath.str(), myProcessSession(), 0, SDS_LOCK_TIMEOUT));
- p.setown(conn->getRoot());
- }
- #endif
- }
- void CLocalEnvironment::preload()
- {
- synchronized procedure(safeCache);
- p->queryBranch(".");
- }
- void CLocalEnvironment::setXML(const char *xml)
- {
- Owned<IPropertyTree> newRoot = createPTreeFromXMLString(xml, ipt_lowmem);
- synchronized procedure(safeCache);
- Owned<IPropertyTreeIterator> it = p->getElements("*");
- ForEach(*it)
- {
- p->removeTree(&it->query());
- }
- it.setown(newRoot->getElements("*"));
- ForEach(*it)
- {
- IPropertyTree *sub = &it->get();
- p->addPropTree(sub->queryName(), sub);
- }
- }
- bool CLocalEnvironment::getRunInfo(IStringVal & path, IStringVal & dir, const char * tag, const char * version, const char *machineaddr, const char *defprogname) const
- {
- try
- {
- // DBGLOG("getExecutablePath %s %s %s", tag, version, machineaddr);
- // first see if local machine with deployed on
- SocketEndpoint ep(machineaddr);
- Owned<CConstInstanceInfo> ipinstance = getInstanceByIP(tag, version, ep);
- if (ipinstance)
- {
- StringAttr testpath;
- StringAttrAdaptor teststrval(testpath);
- if (ipinstance->doGetRunInfo(teststrval,dir,defprogname,false))
- { // this returns full string
- RemoteFilename rfn;
- rfn.setRemotePath(testpath.get());
- Owned<IFile> file = createIFile(rfn);
- if (file->exists())
- {
- StringBuffer tmp;
- rfn.getLocalPath(tmp);
- path.set(tmp.str());
- return true;
- }
- }
- }
- Owned<IConstMachineInfo> machine = getMachineByAddress(machineaddr);
- if (!machine)
- {
- LOG(MCdebugInfo, unknownJob, "Unable to find machine for %s", machineaddr);
- return false;
- }
- StringAttr targetdomain;
- Owned<IConstDomainInfo> domain = machine->getDomain();
- if (!domain)
- {
- LOG(MCdebugInfo, unknownJob, "Unable to find domain for %s", machineaddr);
- return false;
- }
- //domain->getName(StringAttrAdaptor(targetdomain)); // confuses g++
- StringAttrAdaptor strval(targetdomain);
- domain->getName(strval);
- Owned<IConstInstanceInfo> instance = getInstance(tag, version, targetdomain);
- if (!instance)
- {
- LOG(MCdebugInfo, unknownJob, "Unable to find process %s for domain %s", tag, targetdomain.get());
- return false;
- }
- return instance->getRunInfo(path,dir,defprogname);
- }
- catch (IException * e)
- {
- EXCLOG(e, "Extracting slave version");
- e->Release();
- return false;
- }
- }
- void CLocalEnvironment::clearCache()
- {
- synchronized procedure(safeCache);
- if (conn)
- {
- p.clear();
- unsigned mode = 0;
- try
- {
- conn->reload();
- }
- catch (IException *e)
- {
- EXCLOG(e, "Failed to reload connection");
- e->Release();
- mode = conn->queryMode();
- conn.clear();
- }
- if (!conn)
- conn.setown(querySDS().connect(xPath, myProcessSession(), mode, SDS_LOCK_TIMEOUT));
- p.setown(conn->getRoot());
- }
- cache.kill();
- keyGroupMap.clear();
- keyPairMap.clear();
- init();
- }
- IConstDropZoneInfo * CLocalEnvironment::getDropZoneByAddressPath(const char * netaddress, const char *targetFilePath) const
- {
- IConstDropZoneInfo * dropZone = nullptr;
- IpAddress targetIp(netaddress);
- unsigned dropzonePathLen = _MAX_PATH + 1;
- #ifdef _DEBUG
- LOG(MCdebugInfo, unknownJob, "Netaddress: '%s', targetFilePath: '%s'", netaddress, targetFilePath);
- #endif
- // Check the directory path first
- Owned<IConstDropZoneInfoIterator> zoneIt = getDropZoneIterator();
- ForEach(*zoneIt)
- {
- SCMStringBuffer dropZoneDir;
- zoneIt->query().getDirectory(dropZoneDir);
- StringBuffer fullDropZoneDir(dropZoneDir.str());
- addPathSepChar(fullDropZoneDir);
- IConstDropZoneInfo * candidateDropZone = nullptr;
- if (strncmp(fullDropZoneDir, targetFilePath, fullDropZoneDir.length()) == 0)
- {
- candidateDropZone = &zoneIt->query();
- // The backward compatibility built in IConstDropZoneServerInfoIterator
- Owned<IConstDropZoneServerInfoIterator> dropzoneServerListIt = candidateDropZone->getServers();
- ForEach(*dropzoneServerListIt)
- {
- StringBuffer dropzoneServer;
- dropzoneServerListIt->query().getServer(dropzoneServer);
- // It can be a hostname or an IP -> get the IP
- IpAddress serverIP(dropzoneServer.str());
- #ifdef _DEBUG
- StringBuffer serverIpString;
- serverIP.getIpText(serverIpString);
- LOG(MCdebugInfo, unknownJob, "Listed server: '%s', IP: '%s'", dropzoneServer.str(), serverIpString.str());
- #endif
- if (strisame(netaddress, dropzoneServer) || targetIp.ipequals(serverIP))
- {
- // OK the target is a valid machine in the server list we have a right drop zone candidate
- // Keep this candidate drop zone if its directory path is shorter than we already have
- if (dropzonePathLen > fullDropZoneDir.length())
- {
- dropzonePathLen = fullDropZoneDir.length();
- dropZone = candidateDropZone;
- }
- break;
- }
- }
- }
- }
- return LINK(dropZone);
- }
- IConstDropZoneInfoIterator * CLocalEnvironment::getDropZoneIteratorByAddress(const char *addr) const
- {
- class CByAddrIter : public CSimpleInterfaceOf<IConstDropZoneInfoIterator>
- {
- IArrayOf<IConstDropZoneInfo> matches;
- unsigned cur = NotFound;
- public:
- CByAddrIter(IConstDropZoneInfoIterator *baseIter, const char *addr)
- {
- IpAddress toMatch(addr);
- ForEach(*baseIter)
- {
- IConstDropZoneInfo &dz = baseIter->query();
- Owned<IConstDropZoneServerInfoIterator> serverIter = dz.getServers();
- ForEach(*serverIter)
- {
- IConstDropZoneServerInfo &serverElem = serverIter->query();
- StringBuffer serverName;
- serverElem.getServer(serverName);
- IpAddress serverIp(serverName);
- if (strisame(addr, serverName) || serverIp.ipequals(toMatch))
- {
- matches.append(*LINK(&dz));
- break;
- }
- }
- }
- }
- virtual bool first() override
- {
- if (0 == matches.ordinality())
- {
- cur = NotFound;
- return false;
- }
- cur = 0;
- return true;
- }
- virtual bool next() override
- {
- if (cur+1==matches.ordinality())
- {
- cur = NotFound;
- return false;
- }
- ++cur;
- return true;
- }
- virtual bool isValid() override
- {
- return NotFound != cur;
- }
- virtual IConstDropZoneInfo &query() override
- {
- assertex(NotFound != cur);
- return matches.item(cur);
- }
- virtual unsigned count() const override
- {
- return matches.ordinality();
- }
- };
- Owned<IConstDropZoneInfoIterator> baseIter = new CConstDropZoneInfoIterator();
- return new CByAddrIter(baseIter, addr);
- }
- IConstDropZoneInfoIterator * CLocalEnvironment::getDropZoneIterator() const
- {
- return new CConstDropZoneInfoIterator();
- }
- IConstDfuQueueInfoIterator * CLocalEnvironment::getDfuQueueIterator() const
- {
- return new CConstDfuQueueInfoIterator();
- }
- bool CLocalEnvironment::isValidDfuQueueName(const char * queueName) const
- {
- bool retVal = false;
- if (!isEmptyString(queueName))
- {
- Owned<IConstDfuQueueInfoIterator> queueIt = getDfuQueueIterator();
- ForEach(*queueIt)
- {
- SCMStringBuffer _queueName;
- queueIt->query().getDfuQueueName(_queueName);
- retVal = streq(queueName, _queueName.str());
- }
- }
- return retVal;
- }
- IConstMachineInfoIterator * CLocalEnvironment::getMachineIterator() const
- {
- return new CConstMachineInfoIterator();
- }
- bool CLocalEnvironment::isDropZoneRestrictionEnabled() const
- {
- if (!isDropZoneRestrictionLoaded)
- {
- dropZoneRestrictionEnabled = queryEnvironmentConf().getPropBool("useDropZoneRestriction", true);
- isDropZoneRestrictionLoaded=true;
- }
- return dropZoneRestrictionEnabled;
- }
- IConstDaFileSrvInfo *CLocalEnvironment::getDaFileSrvGroupInfo(const char *name) const
- {
- if (!name)
- return nullptr;
- VStringBuffer xpath("Software/DafilesrvGroup[@name=\"%s\"]", name);
- synchronized procedure(safeCache);
- IConstEnvBase *cached = getCache(xpath.str());
- if (!cached)
- {
- IPropertyTree *d = p->queryPropTree(xpath.str());
- if (!d)
- return nullptr;
- cached = new CConstDaFileSrvInfo(this, d);
- setCache(xpath.str(), cached);
- }
- return (IConstDaFileSrvInfo *) cached;
- }
- IConstSparkThorInfo *CLocalEnvironment::getSparkThor(const char *name) const
- {
- if (isEmptyString(name))
- return nullptr;
- buildSparkThorCache();
- VStringBuffer xpath("Software/SparkThor[@name=\"%s\"]", name);
- synchronized procedure(safeCache);
- return (CConstSparkThorInfo *) getCache(xpath);
- }
- IConstSparkThorInfo *CLocalEnvironment::getSparkThorByIndex(unsigned index) const
- {
- if (index == 0)
- return nullptr;
- buildSparkThorCache();
- if (index > numOfSparkThors)
- return nullptr;
- StringBuffer xpath("Software/SparkThor[@id=\"");
- xpath.append(SPARKTHOR_SUFFIX).append(index).append("\"]");
- synchronized procedure(safeCache);
- return (CConstSparkThorInfo *) getCache(xpath);
- }
- IConstSparkThorInfoIterator *CLocalEnvironment::getSparkThorIterator() const
- {
- return new CConstSparkThorInfoIterator();
- }
- void CLocalEnvironment::buildSparkThorCache() const
- {
- synchronized procedure(safeCache);
- if (sparkThorCacheBuilt)
- return;
- Owned<IPropertyTreeIterator> it = p->getElements("Software/SparkThorProcess");
- ForEach(*it)
- {
- const char *name = it->query().queryProp("@name");
- if (!isEmptyString(name))
- {
- StringBuffer x("Software/SparkThor[@name=\"");
- x.append(name).append("\"]");
- Owned<IConstEnvBase> cached = new CConstSparkThorInfo((CLocalEnvironment *) this, &it->query());
- cache.setValue(x, cached);
- }
- numOfSparkThors++;
- StringBuffer x("Software/SparkThor[@id=\"");
- x.append(SPARKTHOR_SUFFIX).append(numOfSparkThors).append("\"]");
- Owned<IConstEnvBase> cached = new CConstSparkThorInfo((CLocalEnvironment *) this, &it->query());
- cache.setValue(x, cached);
- }
- sparkThorCacheBuilt = true;
- }
- //==========================================================================================
- // Iterators implementation
- CConstMachineInfoIterator::CConstMachineInfoIterator()
- {
- Owned<IEnvironmentFactory> factory = getEnvironmentFactory(true);
- constEnv.setown((CLocalEnvironment *)factory->openEnvironment());
- maxIndex = constEnv->getNumberOfMachines();
- }
- bool CConstMachineInfoIterator::first()
- {
- index = 1;
- curr.setown(constEnv->getMachineByIndex(index));
- return curr != nullptr;
- }
- bool CConstMachineInfoIterator::next()
- {
- if (index < maxIndex)
- {
- index++;
- curr.setown(constEnv->getMachineByIndex(index));
- }
- else
- curr.clear();
- return curr != nullptr;
- }
- bool CConstMachineInfoIterator::isValid()
- {
- return curr != nullptr;
- }
- IConstMachineInfo & CConstMachineInfoIterator::query()
- {
- return *curr;
- }
- unsigned CConstMachineInfoIterator::count() const
- {
- return maxIndex;
- }
- CConstDropZoneServerInfoIterator::CConstDropZoneServerInfoIterator(const IConstDropZoneInfo * dropZone)
- {
- Owned<IEnvironmentFactory> factory = getEnvironmentFactory(true);
- constEnv.setown((CLocalEnvironment *)factory->openEnvironment());
- // For backward compatibility
- SCMStringBuffer dropZoneMachineName;
- // Returns dropzone '@computer' value if it is exists
- dropZone->getComputerName(dropZoneMachineName);
- if (0 != dropZoneMachineName.length())
- {
- // Create a ServerList for legacy element.
- Owned<IPropertyTree> legacyServerList = createPTree(ipt_lowmem);
- Owned<IConstMachineInfo> machineInfo = constEnv->getMachine(dropZoneMachineName.str());
- if (machineInfo)
- {
- SCMStringBuffer dropZoneMachineNetAddress;
- machineInfo->getNetAddress(dropZoneMachineNetAddress);
- // Create a single ServerList record related to @computer
- //<ServerList name="ServerList" server="<IP_of_@computer>"/>
- IPropertyTree *newRecord = legacyServerList->addPropTree("ServerList");
- newRecord->setProp("@name", "ServerList");
- newRecord->setProp("@server", dropZoneMachineNetAddress.str());
- maxIndex = 1;
- }
- else
- {
- // Something is terrible wrong because there is no matching machine for DropZone @computer
- maxIndex = 0;
- }
- serverListIt.setown(legacyServerList->getElements("ServerList"));
- }
- else
- {
- Owned<IPropertyTree> pSrc = &dropZone->getPTree();
- serverListIt.setown(pSrc->getElements("ServerList"));
- maxIndex = pSrc->getCount("ServerList");
- }
- }
- bool CConstDropZoneServerInfoIterator::first()
- {
- bool hasFirst = serverListIt->first();
- if (hasFirst)
- curr.setown(new CConstDropZoneServerInfo(constEnv, &serverListIt->query()));
- else
- curr.clear();
- return hasFirst;
- }
- bool CConstDropZoneServerInfoIterator::next()
- {
- bool hasNext = serverListIt->next();
- if (hasNext)
- curr.setown(new CConstDropZoneServerInfo(constEnv, &serverListIt->query()));
- else
- curr.clear();
- return hasNext;
- }
- bool CConstDropZoneServerInfoIterator::isValid()
- {
- return nullptr != curr;
- }
- IConstDropZoneServerInfo & CConstDropZoneServerInfoIterator::query()
- {
- return *curr;
- }
- unsigned CConstDropZoneServerInfoIterator::count() const
- {
- return maxIndex;
- }
- //--------------------------------------------------
- CConstDropZoneInfoIterator::CConstDropZoneInfoIterator()
- {
- Owned<IEnvironmentFactory> factory = getEnvironmentFactory(true);
- constEnv.setown((CLocalEnvironment *)factory->openEnvironment());
- maxIndex = constEnv->getNumberOfDropZones();
- }
- bool CConstDropZoneInfoIterator::first()
- {
- index = 1;
- curr.setown(constEnv->getDropZoneByIndex(index));
- return curr != nullptr;
- }
- bool CConstDropZoneInfoIterator::next()
- {
- if (index < maxIndex)
- {
- index++;
- curr.setown(constEnv->getDropZoneByIndex(index));
- }
- else
- curr.clear();
- return curr != nullptr;
- }
- bool CConstDropZoneInfoIterator::isValid()
- {
- return curr != nullptr;
- }
- IConstDropZoneInfo & CConstDropZoneInfoIterator::query()
- {
- return *curr;
- }
- unsigned CConstDropZoneInfoIterator::count() const
- {
- return maxIndex;
- }
- //--------------------------------------------------
- CConstDfuQueueInfoIterator::CConstDfuQueueInfoIterator()
- {
- Owned<IEnvironmentFactory> factory = getEnvironmentFactory(true);
- constEnv.setown((CLocalEnvironment *)factory->openEnvironment());
- maxIndex = constEnv->getNumberOfDfuQueues();
- }
- bool CConstDfuQueueInfoIterator::first()
- {
- index = 1;
- curr.setown(constEnv->getDfuQueueByIndex(index));
- return curr != nullptr;
- }
- bool CConstDfuQueueInfoIterator::next()
- {
- if (index < maxIndex)
- {
- index++;
- curr.setown(constEnv->getDfuQueueByIndex(index));
- }
- else
- curr.clear();
- return curr != nullptr;
- }
- bool CConstDfuQueueInfoIterator::isValid()
- {
- return curr != nullptr;
- }
- IConstDfuQueueInfo & CConstDfuQueueInfoIterator::query()
- {
- return *curr;
- }
- unsigned CConstDfuQueueInfoIterator::count() const
- {
- return maxIndex;
- }
- //--------------------------------------------------
- CConstSparkThorInfoIterator::CConstSparkThorInfoIterator()
- {
- Owned<IEnvironmentFactory> factory = getEnvironmentFactory(true);
- constEnv.setown((CLocalEnvironment *)factory->openEnvironment());
- maxIndex = constEnv->getNumberOfSparkThors();
- }
- bool CConstSparkThorInfoIterator::first()
- {
- index = 1;
- curr.setown(constEnv->getSparkThorByIndex(index));
- return curr != nullptr;
- }
- bool CConstSparkThorInfoIterator::next()
- {
- if (index < maxIndex)
- {
- index++;
- curr.setown(constEnv->getSparkThorByIndex(index));
- }
- else
- curr.clear();
- return curr != nullptr;
- }
- bool CConstSparkThorInfoIterator::isValid()
- {
- return curr != nullptr;
- }
- IConstSparkThorInfo &CConstSparkThorInfoIterator::query()
- {
- return *curr;
- }
- unsigned CConstSparkThorInfoIterator::count() const
- {
- return maxIndex;
- }
- //--------------------------------------------------
- CConstInstanceInfoIterator::CConstInstanceInfoIterator(const CLocalEnvironment *env, IPropertyTreeIterator *itr)
- : constEnv(env)
- {
- instanceItr.setown(itr);
- maxIndex = 0;
- ForEach(*instanceItr)
- maxIndex++;
- }
- bool CConstInstanceInfoIterator::first()
- {
- index = 1;
- instanceItr->first();
- curr.setown(new CConstInstanceInfo(constEnv, &instanceItr->query()));
- return curr != nullptr;
- }
- bool CConstInstanceInfoIterator::next()
- {
- if (index < maxIndex)
- {
- index++;
- instanceItr->next();
- curr.setown(new CConstInstanceInfo(constEnv, &instanceItr->query()));
- }
- else
- curr.clear();
- return curr != nullptr;
- }
- bool CConstInstanceInfoIterator::isValid()
- {
- return curr != nullptr;
- }
- IConstInstanceInfo &CConstInstanceInfoIterator::query()
- {
- return *curr;
- }
- unsigned CConstInstanceInfoIterator::count() const
- {
- return maxIndex;
- }
- //==========================================================================================
- static CriticalSection getEnvSect;
- extern ENVIRONMENT_API IEnvironmentFactory * getEnvironmentFactory(bool update)
- {
- CriticalBlock block(getEnvSect);
- if (!factory)
- {
- factory = new CEnvironmentFactory();
- addShutdownHook(*factory);
- }
- if (update)
- factory->validateCache();
- return LINK(factory);
- }
- extern ENVIRONMENT_API void closeEnvironment()
- {
- try
- {
- CEnvironmentFactory* pFactory;
- {
- //this method is not meant to be invoked by multiple
- //threads concurrently but just in case...
- CriticalBlock block(getEnvSect);
- pFactory = factory;
- factory = nullptr;
- }
- if (pFactory)
- {
- removeShutdownHook(*pFactory);
- pFactory->close();
- pFactory->Release();
- }
- }
- catch (IException *e)
- {
- EXCLOG(e);
- }
- }
- unsigned getAccessibleServiceURLList(const char *serviceType, std::vector<std::string> &list)
- {
- unsigned added = 0;
- Owned<IEnvironmentFactory> factory = getEnvironmentFactory(true);
- Owned<IConstEnvironment> daliEnv = factory->openEnvironment();
- Owned<IPropertyTree> env = &daliEnv->getPTree();
- if (env.get())
- {
- StringBuffer fileMetaServiceUrl;
- StringBuffer espInstanceComputerName;
- StringBuffer bindingProtocol;
- StringBuffer xpath;
- StringBuffer instanceAddress;
- StringBuffer espServiceType;
- Owned<IPropertyTreeIterator> espProcessIter = env->getElements("Software/EspProcess");
- ForEach(*espProcessIter)
- {
- Owned<IPropertyTreeIterator> espBindingIter = espProcessIter->query().getElements("EspBinding");
- ForEach(*espBindingIter)
- {
- xpath.setf("Software/EspService[@name=\"%s\"]/Properties/@type", espBindingIter->query().queryProp("@service"));
- if (strisame(env->queryProp(xpath), serviceType))
- {
- if (espBindingIter->query().getProp("@protocol", bindingProtocol.clear()))
- {
- Owned<IPropertyTreeIterator> espInstanceIter = espProcessIter->query().getElements("Instance");
- ForEach(*espInstanceIter)
- {
- if (espInstanceIter->query().getProp("@computer", espInstanceComputerName.clear()))
- {
- xpath.setf("Hardware/Computer[@name=\"%s\"]/@netAddress", espInstanceComputerName.str());
- if (env->getProp(xpath.str(), instanceAddress.clear()))
- {
- fileMetaServiceUrl.setf("%s://%s:%d", bindingProtocol.str(), instanceAddress.str(), espBindingIter->query().getPropInt("@port",8010));
- list.push_back(fileMetaServiceUrl.str());
- ++added;
- }
- }
- }
- }
- }
- }//ESPBinding
- }//ESPProcess
- }
- return added;
- }
- //------------------- Moved from workunit.cpp -------------
- IPropertyTree *queryRoxieProcessTree(IPropertyTree *environment, const char *process)
- {
- if (!process || !*process)
- return NULL;
- VStringBuffer xpath("Software/RoxieCluster[@name=\"%s\"]", process);
- return environment->queryPropTree(xpath.str());
- }
- void getRoxieProcessServers(IPropertyTree *roxie, SocketEndpointArray &endpoints)
- {
- if (!roxie)
- return;
- Owned<IPropertyTreeIterator> servers = roxie->getElements("RoxieServerProcess");
- ForEach(*servers)
- {
- IPropertyTree &server = servers->query();
- const char *netAddress = server.queryProp("@netAddress");
- if (netAddress && *netAddress)
- {
- SocketEndpoint ep(netAddress, server.getPropInt("@port", 9876));
- endpoints.append(ep);
- }
- }
- }
- void getRoxieProcessServers(const char *process, SocketEndpointArray &servers)
- {
- Owned<IEnvironmentFactory> factory = getEnvironmentFactory(true);
- Owned<IConstEnvironment> env = factory->openEnvironment();
- Owned<IPropertyTree> root = &env->getPTree();
- getRoxieProcessServers(queryRoxieProcessTree(root, process), servers);
- }
- #define WUERR_MismatchClusterSize 5008
- class CEnvironmentClusterInfo: implements IConstWUClusterInfo, public CInterface
- {
- StringAttr name;
- StringAttr alias;
- StringAttr serverQueue;
- StringAttr agentQueue;
- StringAttr agentName;
- StringArray eclServerNames;
- bool legacyEclServer = false;
- StringAttr eclSchedulerName;
- StringAttr roxieProcess;
- SocketEndpointArray roxieServers;
- StringAttr thorQueue;
- StringArray thorProcesses;
- StringArray primaryThorProcesses;
- StringAttr prefix;
- StringAttr ldapUser;
- StringBuffer ldapPassword;
- ClusterType platform;
- unsigned clusterWidth;
- unsigned roxieRedundancy;
- unsigned channelsPerNode;
- unsigned numberOfSlaveLogs;
- int roxieReplicateOffset;
- public:
- IMPLEMENT_IINTERFACE;
- CEnvironmentClusterInfo(const char *_name, const char *_prefix, const char *_alias, IPropertyTree *agent,
- IArrayOf<IPropertyTree> &eclServers, bool _legacyEclServer, IPropertyTree *eclScheduler, IArrayOf<IPropertyTree> &thors, IPropertyTree *roxie)
- : name(_name), prefix(_prefix), alias(_alias), roxieRedundancy(0), channelsPerNode(0), numberOfSlaveLogs(0), roxieReplicateOffset(1), legacyEclServer(_legacyEclServer)
- {
- StringBuffer queue;
- if (thors.ordinality())
- {
- thorQueue.set(getClusterThorQueueName(queue.clear(), name));
- clusterWidth = 0;
- bool isMultiThor = (thors.length() > 1);
- ForEachItemIn(i,thors)
- {
- IPropertyTree &thor = thors.item(i);
- const char* thorName = thor.queryProp("@name");
- thorProcesses.append(thorName);
- if (!isMultiThor)
- primaryThorProcesses.append(thorName);
- else
- {
- const char *nodeGroup = thor.queryProp("@nodeGroup");
- if (!nodeGroup || strieq(nodeGroup, thorName))
- primaryThorProcesses.append(thorName);
- }
- unsigned nodes = thor.getCount("ThorSlaveProcess");
- unsigned slavesPerNode = thor.getPropInt("@slavesPerNode", 1);
- unsigned channelsPerSlave = thor.getPropInt("@channelsPerSlave", 1);
- unsigned ts = nodes * slavesPerNode * channelsPerSlave;
- numberOfSlaveLogs = nodes * slavesPerNode;
- if (clusterWidth && (ts!=clusterWidth))
- throw MakeStringException(WUERR_MismatchClusterSize,"CEnvironmentClusterInfo: mismatched thor sizes in cluster");
- clusterWidth = ts;
- }
- platform = ThorLCRCluster;
- }
- else if (roxie)
- {
- roxieProcess.set(roxie->queryProp("@name"));
- platform = RoxieCluster;
- getRoxieProcessServers(roxie, roxieServers);
- clusterWidth = roxieServers.length();
- ldapUser.set(roxie->queryProp("@ldapUser"));
- StringBuffer encPassword (roxie->queryProp("@ldapPassword"));
- if (encPassword.length())
- decrypt(ldapPassword, encPassword);
- const char *redundancyMode = roxie->queryProp("@slaveConfig");
- if (redundancyMode && *redundancyMode)
- {
- unsigned dataCopies = roxie->getPropInt("@numDataCopies", 1);
- if (strieq(redundancyMode, "overloaded"))
- channelsPerNode = roxie->getPropInt("@channelsPernode", 1);
- else if (strieq(redundancyMode, "full redundancy"))
- {
- roxieRedundancy = dataCopies-1;
- roxieReplicateOffset = 0;
- }
- else if (strieq(redundancyMode, "cyclic redundancy"))
- {
- roxieRedundancy = dataCopies-1;
- channelsPerNode = dataCopies;
- roxieReplicateOffset = roxie->getPropInt("@cyclicOffset", 1);
- }
- }
- }
- else
- {
- clusterWidth = 1;
- platform = HThorCluster;
- }
- #ifdef _CONTAINERIZED
- if (agent || roxie)
- {
- agentQueue.set(getClusterEclAgentQueueName(queue.clear(), name));
- if (agent)
- agentName.set(agent->queryProp("@name"));
- }
- #else
- if (agent)
- {
- assertex(!roxie);
- agentQueue.set(getClusterEclAgentQueueName(queue.clear(), name));
- agentName.set(agent->queryProp("@name"));
- }
- else if (roxie)
- agentQueue.set(getClusterRoxieQueueName(queue.clear(), name));
- #endif
- if (eclScheduler)
- eclSchedulerName.set(eclScheduler->queryProp("@name"));
- ForEachItemIn(j, eclServers)
- {
- const IPropertyTree &eclServer = eclServers.item(j);
- eclServerNames.append(eclServer.queryProp("@name"));
- }
- // MORE - does this need to be conditional?
- serverQueue.set(getClusterEclCCServerQueueName(queue.clear(), name));
- }
- IStringVal & getName(IStringVal & str) const
- {
- str.set(name.get());
- return str;
- }
- const char *getAlias() const
- {
- return alias;
- }
- IStringVal & getScope(IStringVal & str) const
- {
- str.set(prefix.get());
- return str;
- }
- IStringVal & getAgentQueue(IStringVal & str) const
- {
- str.set(agentQueue);
- return str;
- }
- IStringVal & getAgentName(IStringVal & str) const
- {
- str.set(agentName);
- return str;
- }
- const StringArray & getECLServerNames() const
- {
- return eclServerNames;
- }
- bool isLegacyEclServer() const
- {
- return legacyEclServer;
- }
- IStringVal & getECLSchedulerName(IStringVal & str) const
- {
- str.set(eclSchedulerName);
- return str;
- }
- virtual IStringVal & getServerQueue(IStringVal & str) const
- {
- str.set(serverQueue);
- return str;
- }
- IStringVal & getThorQueue(IStringVal & str) const
- {
- str.set(thorQueue);
- return str;
- }
- unsigned getSize() const
- {
- return clusterWidth;
- }
- unsigned getNumberOfSlaveLogs() const
- {
- return numberOfSlaveLogs;
- }
- virtual ClusterType getPlatform() const
- {
- return platform;
- }
- IStringVal & getRoxieProcess(IStringVal & str) const
- {
- str.set(roxieProcess.get());
- return str;
- }
- const StringArray & getThorProcesses() const
- {
- return thorProcesses;
- }
- const StringArray & getPrimaryThorProcesses() const
- {
- return primaryThorProcesses;
- }
- const SocketEndpointArray & getRoxieServers() const
- {
- return roxieServers;
- }
- unsigned getRoxieRedundancy() const
- {
- return roxieRedundancy;
- }
- unsigned getChannelsPerNode() const
- {
- return channelsPerNode;
- }
- int getRoxieReplicateOffset() const
- {
- return roxieReplicateOffset;
- }
- const char *getLdapUser() const
- {
- return ldapUser.get();
- }
- virtual const char *getLdapPassword() const
- {
- return ldapPassword.str();
- }
- };
- IStringVal &getProcessQueueNames(IStringVal &ret, const char *process, const char *type, const char *suffix)
- {
- if (process)
- {
- Owned<IEnvironmentFactory> factory = getEnvironmentFactory(true);
- Owned<IConstEnvironment> env = factory->openEnvironment();
- Owned<IPropertyTree> root = &env->getPTree();
- StringBuffer queueNames;
- StringBuffer xpath;
- xpath.appendf("%s[@process=\"%s\"]", type, process);
- Owned<IPropertyTreeIterator> targets = root->getElements("Software/Topology/Cluster");
- ForEach(*targets)
- {
- IPropertyTree &target = targets->query();
- if (target.hasProp(xpath))
- {
- if (queueNames.length())
- queueNames.append(',');
- queueNames.append(target.queryProp("@name")).append(suffix);
- }
- }
- ret.set(queueNames);
- }
- return ret;
- }
- extern void getDFUServerQueueNames(StringArray &ret, const char *process)
- {
- Owned<IEnvironmentFactory> factory = getEnvironmentFactory(true);
- Owned<IConstEnvironment> env = factory->openEnvironment();
- StringBuffer xpath ("Software/DfuServerProcess");
- if (!isEmptyString(process))
- xpath.appendf("[@name=\"%s\"]", process);
- Owned<IPropertyTree> root = &env->getPTree();
- Owned<IPropertyTreeIterator> targets = root->getElements(xpath.str());
- ForEach(*targets)
- {
- IPropertyTree &target = targets->query();
- if (target.hasProp("@queue"))
- ret.appendListUniq(target.queryProp("@queue"), ",");
- }
- return;
- }
- extern IStringVal &getEclCCServerQueueNames(IStringVal &ret, const char *process)
- {
- return getProcessQueueNames(ret, process, "EclCCServerProcess", ECLCCSERVER_QUEUE_EXT);
- }
- extern IStringVal &getEclServerQueueNames(IStringVal &ret, const char *process)
- {
- return getProcessQueueNames(ret, process, "EclServerProcess", ECLSERVER_QUEUE_EXT); // shares queue name with EclCCServer
- }
- extern IStringVal &getEclSchedulerQueueNames(IStringVal &ret, const char *process)
- {
- return getProcessQueueNames(ret, process, "EclSchedulerProcess", ECLSCHEDULER_QUEUE_EXT); // Shares deployment/config with EclCCServer
- }
- extern IStringVal &getAgentQueueNames(IStringVal &ret, const char *process)
- {
- return getProcessQueueNames(ret, process, "EclAgentProcess", ECLAGENT_QUEUE_EXT);
- }
- extern IStringVal &getRoxieQueueNames(IStringVal &ret, const char *process)
- {
- return getProcessQueueNames(ret, process, "RoxieCluster", ROXIE_QUEUE_EXT);
- }
- extern IStringVal &getThorQueueNames(IStringVal &ret, const char *process)
- {
- return getProcessQueueNames(ret, process, "ThorCluster", THOR_QUEUE_EXT);
- }
- extern StringBuffer &getClusterThorGroupName(StringBuffer &ret, const char *cluster)
- {
- Owned<IEnvironmentFactory> factory = getEnvironmentFactory(true);
- Owned<IConstEnvironment> env = factory->openEnvironment();
- Owned<IPropertyTree> root = &env->getPTree();
- StringBuffer path;
- path.append("Software/ThorCluster[@name=\"").append(cluster).append("\"]");
- IPropertyTree * child = root->queryPropTree(path);
- if (child)
- getClusterGroupName(*child, ret);
- return ret;
- }
- extern StringBuffer &getClusterGroupName(StringBuffer &ret, const char *cluster)
- {
- Owned<IEnvironmentFactory> factory = getEnvironmentFactory(true);
- Owned<IConstEnvironment> env = factory->openEnvironment();
- Owned<IPropertyTree> root = &env->getPTree();
- StringBuffer path;
- path.set("Software/ThorCluster[@name=\"").append(cluster).append("\"]");
- IPropertyTree * child = root->queryPropTree(path);
- if (child)
- {
- return getClusterGroupName(*child, ret);
- }
- path.set("Software/RoxieCluster[@name=\"").append(cluster).append("\"]");
- child = root->queryPropTree(path);
- if (child)
- {
- return getClusterGroupName(*child, ret);
- }
- path.set("Software/EclAgentProcess[@name=\"").append(cluster).append("\"]");
- child = root->queryPropTree(path);
- if (child)
- {
- return ret.setf("hthor__%s", cluster);
- }
- return ret;
- }
- extern ClusterType getClusterTypeByClusterName(const char *cluster)
- {
- Owned<IEnvironmentFactory> factory = getEnvironmentFactory(true);
- Owned<IConstEnvironment> env = factory->openEnvironment();
- Owned<IPropertyTree> root = &env->getPTree();
- StringBuffer path;
- path.set("Software/ThorCluster[@name=\"").append(cluster).append("\"]");
- if (root->hasProp(path))
- return ThorLCRCluster;
- path.set("Software/RoxieCluster[@name=\"").append(cluster).append("\"]");
- if (root->hasProp(path))
- return RoxieCluster;
- path.set("Software/EclAgentProcess[@name=\"").append(cluster).append("\"]");
- if (root->hasProp(path))
- return HThorCluster;
- return NoCluster;
- }
- extern IStringIterator *getTargetClusters(const char *processType, const char *processName)
- {
- Owned<CStringArrayIterator> ret = new CStringArrayIterator;
- Owned<IEnvironmentFactory> factory = getEnvironmentFactory(true);
- Owned<IConstEnvironment> env = factory->openEnvironment();
- Owned<IPropertyTree> root = &env->getPTree();
- StringBuffer xpath;
- xpath.appendf("%s", processType ? processType : "*");
- if (processName && *processName)
- xpath.appendf("[@process=\"%s\"]", processName);
- Owned<IPropertyTreeIterator> targets = root->getElements("Software/Topology/Cluster");
- ForEach(*targets)
- {
- IPropertyTree &target = targets->query();
- if (target.hasProp(xpath))
- {
- ret->append(target.queryProp("@name"));
- }
- }
- return ret.getClear();
- }
- extern bool isProcessCluster(const char *process)
- {
- if (!process || !*process)
- return false;
- Owned<IEnvironmentFactory> factory = getEnvironmentFactory(true);
- Owned<IConstEnvironment> env = factory->openEnvironment();
- Owned<IPropertyTree> root = &env->getPTree();
- VStringBuffer xpath("Software/*Cluster[@name=\"%s\"]", process);
- return root->hasProp(xpath.str());
- }
- extern bool isProcessCluster(const char *remoteDali, const char *process)
- {
- if (!remoteDali || !*remoteDali)
- return isProcessCluster(process);
- if (!process || !*process)
- return false;
- Owned<INode> remote = createINode(remoteDali, 7070);
- if (!remote)
- return false;
- //Cannot use getEnvironmentFactory() since it is using a remotedali
- VStringBuffer xpath("Environment/Software/*Cluster[@name=\"%s\"]/@name", process);
- try
- {
- Owned<IPropertyTreeIterator> clusters = querySDS().getElementsRaw(xpath, remote, 1000*60*1);
- return clusters->first();
- }
- catch (IException *E)
- {
- StringBuffer msg;
- E->errorMessage(msg);
- DBGLOG("Exception validating cluster %s/%s: %s", remoteDali, xpath.str(), msg.str());
- E->Release();
- }
- return true;
- }
- static void getTargetClusterProcesses(const IPropertyTree *environment, const IPropertyTree *cluster, const char *clustName, const char *processType, IArrayOf<IPropertyTree> &processes)
- {
- StringBuffer xpath;
- Owned<IPropertyTreeIterator> processItr = cluster->getElements(processType);
- ForEach(*processItr)
- {
- const char *processName = processItr->query().queryProp("@process");
- if (isEmptyString(processName))
- throw MakeStringException(-1, "Empty %s/@process for %s", processType, clustName);
- xpath.setf("Software/%s[@name=\"%s\"]", processType, processName);
- if (environment->hasProp(xpath))
- processes.append(*environment->getPropTree(xpath.str()));
- else
- WARNLOG("%s %s not found", processType, processName);
- }
- }
- IConstWUClusterInfo* getTargetClusterInfo(IPropertyTree *environment, IPropertyTree *cluster)
- {
- const char *clustname = cluster->queryProp("@name");
- // MORE - at the moment configenv specifies eclagent and thor queues by (in effect) placing an 'example' thor or eclagent in the topology
- // that uses the queue that will be used.
- // We should and I hope will change that, at which point the code below gets simpler
- StringBuffer prefix(cluster->queryProp("@prefix"));
- prefix.toLowerCase();
- StringBuffer xpath;
- StringBuffer querySetName;
- IPropertyTree *agent = NULL;
- const char *agentName = cluster->queryProp("EclAgentProcess/@process");
- if (!isEmptyString(agentName))
- {
- xpath.clear().appendf("Software/EclAgentProcess[@name=\"%s\"]", agentName);
- agent = environment->queryPropTree(xpath.str());
- }
- IPropertyTree *eclScheduler = nullptr;
- const char *eclSchedulerName = cluster->queryProp("EclSchedulerProcess/@process");
- if (!isEmptyString(eclSchedulerName))
- {
- xpath.setf("Software/EclSchedulerProcess[@name=\"%s\"]", eclSchedulerName);
- eclScheduler = environment->queryPropTree(xpath);
- }
- bool isLegacyEclServer = false;
- IArrayOf<IPropertyTree> eclServers;
- getTargetClusterProcesses(environment, cluster, clustname, "EclServerProcess", eclServers);
- if (eclServers.ordinality())
- isLegacyEclServer = true;
- else
- getTargetClusterProcesses(environment, cluster, clustname, "EclCCServerProcess", eclServers);
- Owned<IPropertyTreeIterator> ti = cluster->getElements("ThorCluster");
- IArrayOf<IPropertyTree> thors;
- ForEach(*ti)
- {
- const char *thorName = ti->query().queryProp("@process");
- if (thorName)
- {
- xpath.clear().appendf("Software/ThorCluster[@name=\"%s\"]", thorName);
- if (environment->hasProp(xpath.str()))
- thors.append(*environment->getPropTree(xpath.str()));
- }
- }
- const char *roxieName = cluster->queryProp("RoxieCluster/@process");
- return new CEnvironmentClusterInfo(clustname, prefix, cluster->queryProp("@alias"), agent, eclServers, isLegacyEclServer, eclScheduler, thors, queryRoxieProcessTree(environment, roxieName));
- }
- IPropertyTree* getTopologyCluster(Owned<IPropertyTree> &envRoot, const char *clustname)
- {
- if (!clustname || !*clustname)
- return NULL;
- Owned<IEnvironmentFactory> factory = getEnvironmentFactory(true);
- Owned<IConstEnvironment> env = factory->openEnvironment();
- envRoot.setown(&env->getPTree());
- StringBuffer xpath;
- xpath.appendf("Software/Topology/Cluster[@name=\"%s\"]", clustname);
- return envRoot->getPropTree(xpath.str());
- }
- bool validateTargetClusterName(const char *clustname)
- {
- Owned<IPropertyTree> envRoot;
- Owned<IPropertyTree> cluster = getTopologyCluster(envRoot, clustname);
- return (cluster.get()!=NULL);
- }
- IConstWUClusterInfo* getTargetClusterInfo(const char *clustname)
- {
- Owned<IPropertyTree> envRoot;
- Owned<IPropertyTree> cluster = getTopologyCluster(envRoot, clustname);
- if (!cluster)
- return NULL;
- return getTargetClusterInfo(envRoot, cluster);
- }
- unsigned getEnvironmentClusterInfo(CConstWUClusterInfoArray &clusters)
- {
- Owned<IEnvironmentFactory> factory = getEnvironmentFactory(true);
- Owned<IConstEnvironment> env = factory->openEnvironment();
- Owned<IPropertyTree> root = &env->getPTree();
- return getEnvironmentClusterInfo(root, clusters);
- }
- unsigned getEnvironmentClusterInfo(IPropertyTree* environmentRoot, CConstWUClusterInfoArray &clusters)
- {
- if (!environmentRoot)
- return 0;
- Owned<IPropertyTreeIterator> clusterIter = environmentRoot->getElements("Software/Topology/Cluster");
- ForEach(*clusterIter)
- {
- IPropertyTree &node = clusterIter->query();
- Owned<IConstWUClusterInfo> cluster = getTargetClusterInfo(environmentRoot, &node);
- clusters.append(*cluster.getClear());
- }
- return clusters.ordinality();
- }
- const char *getTargetClusterComponentName(const char *clustname, const char *processType, StringBuffer &name)
- {
- if (!clustname)
- return NULL;
- Owned<IEnvironmentFactory> factory = getEnvironmentFactory(true);
- Owned<IConstEnvironment> env = factory->openEnvironment();
- Owned<IPropertyTree> root = &env->getPTree();
- StringBuffer xpath;
- xpath.appendf("Software/Topology/Cluster[@name=\"%s\"]", clustname);
- Owned<IPropertyTree> cluster = root->getPropTree(xpath.str());
- if (!cluster)
- return NULL;
- StringBuffer xpath1;
- xpath1.appendf("%s/@process", processType);
- name.append(cluster->queryProp(xpath1.str()));
- return name.str();
- }
- unsigned getEnvironmentThorClusterNames(StringArray &thorNames, StringArray &groupNames, StringArray &targetNames, StringArray &queueNames)
- {
- Owned<IEnvironmentFactory> factory = getEnvironmentFactory(true);
- Owned<IConstEnvironment> env = factory->openEnvironment();
- Owned<IPropertyTree> root = &env->getPTree();
- Owned<IPropertyTreeIterator> allTargets = root->getElements("Software/Topology/Cluster");
- ForEach(*allTargets)
- {
- IPropertyTree &target = allTargets->query();
- const char *targetName = target.queryProp("@name");
- if (targetName && *targetName)
- {
- Owned<IPropertyTreeIterator> thorClusters = target.getElements("ThorCluster");
- ForEach(*thorClusters)
- {
- const char *thorName = thorClusters->query().queryProp("@process");
- VStringBuffer query("Software/ThorCluster[@name=\"%s\"]",thorName);
- IPropertyTree *thorCluster = root->queryPropTree(query.str());
- if (thorCluster)
- {
- const char *groupName = thorCluster->queryProp("@nodeGroup");
- if (!groupName||!*groupName)
- groupName = thorName;
- thorNames.append(thorName);
- groupNames.append(groupName);
- targetNames.append(targetName);
- StringBuffer queueName(targetName);
- queueNames.append(queueName.append(THOR_QUEUE_EXT));
- }
- }
- }
- }
- return thorNames.ordinality();
- }
|