dfuxreflib.cpp 93 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759
  1. /*##############################################################################
  2. Copyright (C) 2011 HPCC Systems.
  3. All rights reserved. This program is free software: you can redistribute it and/or modify
  4. it under the terms of the GNU Affero General Public License as
  5. published by the Free Software Foundation, either version 3 of the
  6. License, or (at your option) any later version.
  7. This program is distributed in the hope that it will be useful,
  8. but WITHOUT ANY WARRANTY; without even the implied warranty of
  9. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  10. GNU Affero General Public License for more details.
  11. You should have received a copy of the GNU Affero General Public License
  12. along with this program. If not, see <http://www.gnu.org/licenses/>.
  13. ############################################################################## */
  14. // DFU XREF Program
  15. #include "platform.h"
  16. #include "jlib.hpp"
  17. #include "jmisc.hpp"
  18. #include "jptree.hpp"
  19. #include "mpbase.hpp"
  20. #include "mpcomm.hpp"
  21. #include "daclient.hpp"
  22. #include "dadiags.hpp"
  23. #include "danqs.hpp"
  24. #include "dadfs.hpp"
  25. #include "dasds.hpp"
  26. #include "dalienv.hpp"
  27. #include "daft.hpp"
  28. #include "rmtfile.hpp"
  29. #include "dautils.hpp"
  30. #include "jptree.hpp"
  31. #include "XRefNodeManager.hpp"
  32. //#define PARTS_SIZE_NEEDED
  33. //#define CROSSLINK_CHECK_NEEDED
  34. static bool fixSizes = false;
  35. #define TESTXML
  36. #define SDS_CONNECT_TIMEOUT (1000*60*60*2) // better than infinite
  37. #define SDS_DFS_ROOT "Files" // followed by scope/name
  38. //#define CONNECT_EACH_PATH
  39. #include "dfuxreflib.hpp"
  40. extern IPropertyTree *getDirectory(const char * directory, INode * node, unsigned short port);
  41. void testGetDir()
  42. {
  43. SocketEndpoint ep("10.173.72.1");
  44. Owned<INode> node = createINode(ep);
  45. Owned<IPropertyTree> results = getDirectory("/c$/thordata;/d$/thordata",node,getDaliServixPort());
  46. if (results) {
  47. PROGLOG("--------------------------------------------------------");
  48. PROGLOG("DIR");
  49. StringBuffer dirs;
  50. toXML(results, dirs, 2);
  51. PROGLOG("\n%s\n----------------------------------------------------------",dirs.str());
  52. }
  53. }
  54. #define CFBcluster (0x00000001)
  55. #define CFBname (0x00000002)
  56. #define CFBnumparts (0x00000004)
  57. #define CFBpartslost (0x00000008)
  58. #define CFBprimarylost (0x00000010)
  59. #define CFBreplicatedlost (0x00000040)
  60. #define CFBmodified (0x00000080)
  61. #define CFBsize (0x00000100)
  62. #define CFBmismatchedsizes (0x00000400)
  63. #define CFBpartnode (0x00010000)
  64. #define CFBpartnum (0x00020000)
  65. #define CFBpartreplicate (0x00040000)
  66. #define CFBpartprimary (0x00080000)
  67. #define CFBpartmask (CFBpartnode|CFBpartnum|CFBpartreplicate|CFBpartprimary)
  68. static IPropertyTree *addBranch(IPropertyTree *dst,const char *name)
  69. {
  70. return dst->addPropTree(name,createPTree());
  71. }
  72. struct CFileEntry;
  73. struct CDfuDirEntry;
  74. class COrphanEntry;
  75. struct CLogicalNameEntry;
  76. typedef CFileEntry *CFileEntryPtr;
  77. typedef CDfuDirEntry *CDfuDirEntryPtr;
  78. typedef COrphanEntry *COrphanEntryPtr;
  79. typedef CLogicalNameEntry *CLogicalNameEntryPtr;
  80. typedef MapStringTo<CLogicalNameEntryPtr> CLogicalNameMap;
  81. //typedef MapStringTo<CFileEntryPtr> CFileEntryMap;
  82. typedef MapStringTo<CDfuDirEntryPtr> CDfuDirEntryMap;
  83. typedef MapStringTo<COrphanEntryPtr> COrphanEntryMap;
  84. Owned <IFileIOStream> outfileio;
  85. void outf(const char *fmt, ...) __attribute__((format(printf, 1, 2)))
  86. {
  87. va_list args;
  88. va_start(args, fmt);
  89. StringBuffer buf;
  90. buf.valist_appendf(fmt,args);
  91. va_end(args);
  92. if (outfileio)
  93. outfileio->write(buf.length(),buf.str());
  94. else
  95. printf("%s",buf.str());
  96. }
  97. #define FEF_RESOLVED 0x01
  98. #define FEF_REPLICATE 0x02
  99. struct CFileEntry: public CInterface
  100. {
  101. CFileEntry(const char *_fname,CLogicalNameEntry *_owner,unsigned _part,bool _replicate,__int64 _size, bool compresskludge, __int64 _compsize);
  102. StringBuffer &getLogicalName(StringBuffer &buf);
  103. unsigned queryHash()
  104. {
  105. return keyhash;
  106. }
  107. bool comparePath(const char *toname)
  108. {
  109. return strcmp(toname,fname)==0;
  110. }
  111. inline bool resolved() { return (flags&FEF_RESOLVED)!=0; }
  112. inline bool replicate() { return (flags&FEF_REPLICATE)!=0; }
  113. //IpAddress ip;
  114. //StringAttr dir; // TBD
  115. //StringAttr tail;
  116. StringAttr fname;
  117. CLogicalNameEntry *owner;
  118. #ifdef PARTS_SIZE_NEEDED
  119. __int64 size;
  120. __int64 expsize;
  121. #endif
  122. #ifdef CROSSLINK_CHECK_NEEDED
  123. Owned<CFileEntry> crosslink;
  124. #endif
  125. unsigned keyhash;
  126. unsigned short part;
  127. byte flags;
  128. //bool replicate;
  129. //bool resolved;
  130. };
  131. class CFileEntryMap : public SuperHashTableOf<CFileEntry, const char>
  132. {
  133. public:
  134. ~CFileEntryMap()
  135. {
  136. releaseAll();
  137. }
  138. virtual void onAdd(void *e)
  139. {
  140. }
  141. virtual void onRemove(void *e)
  142. {
  143. CFileEntry &elem=*(CFileEntry *)e;
  144. elem.Release();
  145. }
  146. virtual unsigned getHashFromElement(const void *e) const
  147. {
  148. return ((CFileEntry *) e)->queryHash();
  149. }
  150. virtual unsigned getHashFromFindParam(const void *fp) const
  151. {
  152. return hashc((const unsigned char *)fp, strlen((const char *)fp), 0);
  153. }
  154. virtual const void *getFindParam(const void *e) const
  155. {
  156. return ((CFileEntry *) e)->fname;
  157. }
  158. virtual bool matchesFindParam(const void *e, const void *fp, unsigned fphash) const
  159. {
  160. return (0 == strcmp(((CFileEntry *) e)->fname, (const char *)fp));
  161. }
  162. };
  163. StringBuffer &substnum(StringBuffer &str,const char *sub,unsigned n)
  164. {
  165. StringBuffer out;
  166. const char *s=str.str();
  167. const char *p=sub;
  168. while (*s) {
  169. if (*s==*p) {
  170. p++;
  171. if (!*p) {
  172. out.append(n);
  173. p = sub;
  174. }
  175. }
  176. else {
  177. if (p!=sub) {
  178. out.append(p-sub,sub);
  179. p = sub;
  180. }
  181. out.append(*s);
  182. }
  183. s++;
  184. }
  185. str.swapWith(out);
  186. return str;
  187. }
  188. static StringBuffer &makeScopeQuery(const char *scope,StringBuffer &query)
  189. {
  190. const char *s=scope;
  191. loop {
  192. const char *e=strstr(s,"::");
  193. if (s!=scope)
  194. query.append('/');
  195. StringBuffer tail;
  196. if (e)
  197. tail.append(e-s,s);
  198. else
  199. tail.append(s);
  200. query.append("Scope[@name=\"").append(tail.toLowerCase().str()).append("\"]");
  201. if (!e)
  202. break;
  203. s = e+2;
  204. }
  205. return query;
  206. }
  207. static const char *splitScope(const char *name,StringBuffer &scope)
  208. {
  209. const char *s=strstr(name,"::");
  210. if (s) {
  211. loop {
  212. const char *ns = strstr(s+2,"::");
  213. if (!ns)
  214. break;
  215. s = ns;
  216. }
  217. StringBuffer str;
  218. str.append(s-name,name);
  219. scope.append(str.toLowerCase().str());
  220. return s+2;
  221. }
  222. scope.append(".");
  223. return name;
  224. }
  225. static StringBuffer &makeFullnameQuery(const char *lname,StringBuffer &query)
  226. {
  227. StringBuffer scope;
  228. StringBuffer tail(splitScope(lname,scope));
  229. return makeScopeQuery(scope.str(),query).append("/File[@name=\"").append(tail.toLowerCase().str()).append("\"]");
  230. }
  231. static StringBuffer &makeAbsoluteFullnameQuery(const char *lname,StringBuffer &query)
  232. {
  233. query.append(SDS_DFS_ROOT "/");
  234. return makeFullnameQuery(lname,query);
  235. }
  236. struct CLogicalNameEntry;
  237. class CEndpointItem: public CInterface
  238. {
  239. public:
  240. IMPLEMENT_IINTERFACE;
  241. SocketEndpoint ep;
  242. rank_t rank;
  243. };
  244. struct CDfuDirEntry: public CInterface
  245. {
  246. CDfuDirEntry(const char *_name,unsigned _clustsize)
  247. : name(_name)
  248. {
  249. size = 0;
  250. num = 0;
  251. numdir = 0;
  252. maxsize = 0;
  253. minsize = -1;
  254. clustersize = _clustsize;
  255. }
  256. StringAttr name;
  257. __int64 size;
  258. __int64 minsize;
  259. IpAddress minip;
  260. __int64 maxsize;
  261. IpAddress maxip;
  262. unsigned num;
  263. unsigned numdir;
  264. unsigned clustersize;
  265. StringBuffer &getskew(StringBuffer &skew)
  266. {
  267. if (clustersize&&size&&(minsize<maxsize)) {
  268. __int64 av = size/(__int64)clustersize;
  269. if (av) {
  270. unsigned pcp = (unsigned)(maxsize*100/av);
  271. unsigned pcn = (unsigned)(minsize*100/av);
  272. if ((pcp>100)||(pcn<100))
  273. skew.appendf("+%d%%/-%d%%",pcp-100,100-pcn);
  274. }
  275. }
  276. return skew;
  277. }
  278. };
  279. static unsigned short getDafsPort(const SocketEndpoint &ep,unsigned &numfails,CriticalSection *sect)
  280. {
  281. if (sect) {
  282. CriticalBlock block(*sect);
  283. if (numfails>5)
  284. return 0;
  285. }
  286. else if (numfails>5)
  287. return 0;
  288. if (testDaliServixPresent(ep))
  289. return ep.port?ep.port:getDaliServixPort();
  290. StringBuffer err("Failed to connect to DaFileSrv on ");
  291. ep.getUrlStr(err);
  292. #ifdef _WIN32
  293. ERRLOG("%s",err.str());
  294. if (sect) {
  295. CriticalBlock block(*sect);
  296. numfails++;
  297. }
  298. else
  299. numfails++;
  300. #else
  301. throw MakeStringException(-1, "%s", err.str());
  302. #endif
  303. return 0;
  304. }
  305. class CEndpointTable: public SuperHashTableOf<CEndpointItem,SocketEndpoint>
  306. {
  307. public:
  308. ~CEndpointTable()
  309. {
  310. releaseAll();
  311. }
  312. void onAdd(void *)
  313. {
  314. // not used
  315. }
  316. void onRemove(void *e)
  317. {
  318. CEndpointItem &elem=*(CEndpointItem *)e;
  319. elem.Release();
  320. }
  321. unsigned getHashFromElement(const void *e) const
  322. {
  323. const CEndpointItem &elem=*(const CEndpointItem *)e;
  324. return elem.ep.hash(0);
  325. }
  326. unsigned getHashFromFindParam(const void *fp) const
  327. {
  328. return ((const SocketEndpoint *)fp)->hash(0);
  329. }
  330. const void * getFindParam(const void *p) const
  331. {
  332. const CEndpointItem &elem=*(const CEndpointItem *)p;
  333. return &elem.ep;
  334. }
  335. bool matchesFindParam(const void * et, const void *fp, unsigned) const
  336. {
  337. return ((CEndpointItem *)et)->ep.equals(*(SocketEndpoint *)fp);
  338. }
  339. IMPLEMENT_SUPERHASHTABLEOF_REF_FIND(CEndpointItem,SocketEndpoint);
  340. };
  341. class CXRefManagerBase
  342. {
  343. protected:
  344. struct cMessage: public CInterface
  345. {
  346. StringAttr lname;
  347. StringAttr msg;
  348. cMessage(const char *_lname,const char *_msg)
  349. : lname(_lname), msg(_msg)
  350. {
  351. }
  352. };
  353. public:
  354. IArrayOf<IGroup> knowngroups;
  355. StringAttrArray knowngroupnames;
  356. Linked<IXRefProgressCallback> msgcallback;
  357. CriticalSection logsect;
  358. unsigned totalCompressed;
  359. __int64 totalUncompressedSize;
  360. __int64 totalCompressedSize;
  361. __int64 totalSizeOrphans;
  362. unsigned totalNumOrphans;
  363. CIArrayOf<CLogicalNameEntry> logicalnamelist;
  364. CLogicalNameMap logicalnamemap;
  365. CLogicalNameMap logicaldirmap;
  366. CFileEntryMap filemap;
  367. CDfuDirEntryMap dirmap;
  368. CIArrayOf<CDfuDirEntry> dirlist;
  369. COrphanEntryMap orphanmap;
  370. CIArrayOf<COrphanEntry> orphanlist;
  371. CEndpointTable EndpointTable;
  372. CIArrayOf<cMessage> errors;
  373. CIArrayOf<cMessage> warnings;
  374. CriticalSection inprogresssect;
  375. SocketEndpointArray inprogress;
  376. CXRefManagerBase()
  377. {
  378. totalCompressed=0;
  379. totalUncompressedSize=0;
  380. totalCompressedSize=0;
  381. }
  382. virtual ~CXRefManagerBase();
  383. IGroup *resolveGroup(IGroup *_grp,StringBuffer &gname) // takes ownership of _grp
  384. {
  385. StringBuffer name;
  386. IGroup *grp = _grp;
  387. ForEachItemIn(i,knowngroups) {
  388. GroupRelation gr = _grp->compare(&knowngroups.item(i));
  389. const char *gn = knowngroupnames.item(i).text.get();
  390. if ((gr==GRidentical)||(gn&&*gn&&((gr==GRbasesubset)||(gr==GRwrappedsuperset)))) {
  391. _grp->Release();
  392. _grp=NULL;
  393. if (gn)
  394. gname.append(gn);
  395. grp = &knowngroups.item(i);
  396. grp->Link();
  397. break;
  398. }
  399. }
  400. if (_grp) {
  401. if (queryNamedGroupStore().find(_grp,gname)) {
  402. if (gname.length()) {
  403. _grp->Release();
  404. grp=queryNamedGroupStore().lookup(gname.str());
  405. }
  406. }
  407. else {
  408. name.clear();
  409. }
  410. knowngroupnames.append(* new StringAttrItem(gname.str()));
  411. knowngroups.append(*LINK(grp));
  412. }
  413. return grp;
  414. }
  415. void log(const char * format, ...) __attribute__((format(printf, 2, 3)))
  416. {
  417. CriticalBlock block(logsect);
  418. va_list args;
  419. va_start(args, format);
  420. StringBuffer line;
  421. line.valist_appendf(format, args);
  422. va_end(args);
  423. if (msgcallback) {
  424. msgcallback->progress(line.str());
  425. }
  426. else {
  427. PROGLOG("%s",line.str());
  428. }
  429. }
  430. void error(const char *lname,const char * format, ...) __attribute__((format(printf, 3, 4)))
  431. {
  432. CriticalBlock block(logsect);
  433. va_list args;
  434. va_start(args, format);
  435. StringBuffer line;
  436. line.valist_appendf(format, args);
  437. va_end(args);
  438. if (errors.ordinality()<1000) {
  439. errors.append(*new cMessage(lname,line.str()));
  440. if (errors.ordinality()==1000)
  441. errors.append(*new cMessage("","error limit exceeded (1000), truncating"));
  442. }
  443. if (msgcallback) {
  444. StringBuffer cbline("ERROR: ");
  445. cbline.append(lname).append(": ").append(line);
  446. msgcallback->progress(cbline.str());
  447. }
  448. else {
  449. ERRLOG("%s: %s",lname,line.str());
  450. }
  451. }
  452. void warn(const char *lname,const char * format, ...) __attribute__((format(printf, 3, 4)))
  453. {
  454. CriticalBlock block(logsect);
  455. va_list args;
  456. va_start(args, format);
  457. StringBuffer line;
  458. line.valist_appendf(format, args);
  459. va_end(args);
  460. if (warnings.ordinality()<1000) {
  461. warnings.append(*new cMessage(lname,line.str()));
  462. if (warnings.ordinality()==1000)
  463. warnings.append(*new cMessage("","warning limit (1000) exceeded, truncating"));
  464. }
  465. if (msgcallback) {
  466. StringBuffer cbline("WARNING: ");
  467. cbline.append(lname).append(": ").append(line);
  468. msgcallback->progress(cbline.str());
  469. }
  470. else {
  471. WARNLOG("%s: %s",lname,line.str());
  472. }
  473. }
  474. void addNodeInProgress(INode &node)
  475. {
  476. CriticalBlock block(inprogresssect);
  477. SocketEndpoint ep = node.endpoint();
  478. inprogress.append(ep);
  479. }
  480. void removeNodeInProgress(INode &node)
  481. {
  482. CriticalBlock block(inprogresssect);
  483. SocketEndpoint ep = node.endpoint();
  484. inprogress.zap(ep);
  485. if (inprogress.ordinality()==0)
  486. return;
  487. StringBuffer msg("Waiting for ");
  488. ForEachItemIn(i,inprogress) {
  489. if (i)
  490. msg.append(", ");
  491. inprogress.item(i).getIpText(msg);
  492. }
  493. if (msgcallback)
  494. msgcallback->progress(msg.str());
  495. else
  496. PROGLOG("%s",msg.str());
  497. }
  498. void incDirSize(const IpAddress &ip,const char *dir,__int64 sz,unsigned num,unsigned numdir,unsigned clustsize)
  499. {
  500. CDfuDirEntryPtr *entryp= dirmap.getValue(dir);
  501. CDfuDirEntryPtr entry;
  502. if (entryp)
  503. entry = *entryp;
  504. else {
  505. entry = new CDfuDirEntry(dir,clustsize);
  506. dirmap.setValue(dir,entry);
  507. dirlist.append(*entry);
  508. }
  509. entry->num+=num;
  510. entry->numdir+=numdir;
  511. entry->size+=sz;
  512. if ((entry->minsize==-1)||(entry->minsize>sz)) {
  513. entry->minsize=sz;
  514. entry->minip.ipset(ip);
  515. }
  516. if (entry->maxsize<sz) {
  517. entry->maxsize=sz;
  518. entry->maxip.ipset(ip);
  519. }
  520. }
  521. };
  522. struct CLogicalNameEntry: public CInterface
  523. {
  524. CLogicalNameEntry(CXRefManagerBase &_manager, const char *_lname,IPropertyTree &file) // takes ownership of grp
  525. : manager(_manager),lname(_lname)
  526. {
  527. replicated = false;
  528. outsidedir = 0;
  529. primarynum = 0;
  530. replicatenum = 0;
  531. max = file.getPropInt("@numparts");
  532. missinggrp = false;
  533. unknowngrp = false;
  534. const char *s=file.queryProp("@group"); // TBD - Handling for multiple clusters?
  535. if (s&&*s) {
  536. grpname.append(s);
  537. if (isAnonCluster(s))
  538. manager.warn(_lname,"File has anonymous cluster");
  539. if (strchr(s,','))
  540. manager.error(_lname,"XREF can't handle multi-file clusters yet!");
  541. }
  542. s = file.queryProp("@partmask");
  543. if (s&&*s)
  544. pmask.set(s);
  545. s = file.queryProp("@modified");
  546. if (s&&*s)
  547. {
  548. modified.set(s);
  549. }
  550. done = false;
  551. wrongwidth = false;
  552. primaryresolved = NULL;
  553. replicateresolved = NULL;
  554. nummismatchedsizes=0;
  555. mismatchedsizeinfo = NULL;
  556. totalsize=0;
  557. recordsize = file.getPropInt("Attr/@recordSize",-1);
  558. compressed = (file.getPropInt("Attr/@rowCompressed", 0)!=0)||(file.getPropInt("Attr/@blockCompressed", 0)!=0);
  559. grouped = file.getPropInt("Attr/@grouped", 0)!=0;
  560. const char *partmask = file.queryProp("@partmask");
  561. StringBuffer tmp;
  562. if (partmask&&*partmask) {
  563. if (!containsPathSepChar(partmask)) {
  564. const char *dir = file.queryProp("@directory");
  565. if (dir&&*dir)
  566. tmp.append(dir).append(getPathSepChar(dir));
  567. }
  568. tmp.append(partmask);
  569. }
  570. tmp.toLowerCase();
  571. substnum(tmp,"$n$",max);
  572. dirpartmask.set(tmp.str());
  573. }
  574. ~CLogicalNameEntry()
  575. {
  576. free(primaryresolved);
  577. free(replicateresolved);
  578. }
  579. void setGroup(IGroup *_grp,IGroup *lgrp) // note incoming group does *not* override cluster
  580. // takes ownerhip of lgrp if set
  581. {
  582. StringBuffer name;
  583. unsigned n = 0;
  584. if (_grp) {
  585. n = _grp->ordinality();
  586. grp.setown(manager.resolveGroup(_grp,name));
  587. if (!grp) {
  588. SocketListCreator cr;
  589. SocketEndpointArray epa;
  590. ForEachNodeInGroup(i,*grp) {
  591. SocketEndpoint ep = grp->queryNode(i).endpoint();
  592. epa.append(ep);
  593. }
  594. cr.addSockets(epa);
  595. manager.warn(lname.get(),"Cluster group %s did not resolve",cr.getText());
  596. }
  597. }
  598. if (grpname.length()) {
  599. if(!_grp||(stricmp(grpname.str(),name.str())!=0)) {
  600. if (!lgrp)
  601. lgrp = queryNamedGroupStore().lookup(grpname.str());
  602. if (lgrp) {
  603. if (name.length())
  604. mismatchgrp.set(name.str());
  605. grp.setown(lgrp);
  606. }
  607. else {
  608. unknowngrp = true;
  609. grpname.swapWith(name);
  610. manager.warn(lname.get(),"Cluster group %s not found",grpname.str());
  611. }
  612. }
  613. }
  614. else {
  615. grpname.swapWith(name);
  616. if (n>1)
  617. missinggrp = true;
  618. }
  619. primaryresolved=(CFileEntry * *)calloc(max,sizeof(CFileEntry *));
  620. replicateresolved=(CFileEntry * *)calloc(max,sizeof(CFileEntry *));
  621. if (!grp)
  622. manager.warn(lname.get(),"No cluster group set");
  623. }
  624. bool remove()
  625. {
  626. IDistributedFileDirectory &fdir = queryDistributedFileDirectory();
  627. Owned<IDistributedFile> file = fdir.lookup(lname.get());
  628. if (!file)
  629. return false;
  630. file->detach();
  631. outf("Removed %s from DDFS\n",lname.get());
  632. return true;
  633. }
  634. IGroup *queryGroup()
  635. {
  636. return grp.get();
  637. }
  638. unsigned queryNumParts()
  639. {
  640. return max;
  641. }
  642. void addCrossLink(CLogicalNameEntry *entry)
  643. {
  644. if (entry) {
  645. ForEachItemIn(i,crosslinked) {
  646. if (&crosslinked.item(i)==entry)
  647. return;
  648. }
  649. crosslinked.append(*LINK(entry));
  650. }
  651. }
  652. void addMismatchedSize(__int64 expected, __int64 actual, __int64 expanded, CFileEntry *entry)
  653. {
  654. unsigned i = nummismatchedsizes++;
  655. mismatchedsizeinfo = (MMSinfo *)(mismatchedsizeinfo?realloc(mismatchedsizeinfo,nummismatchedsizes*sizeof(MMSinfo)):malloc(sizeof(MMSinfo)));
  656. mismatchedsizeinfo[i].expected = expected;
  657. mismatchedsizeinfo[i].actual = actual;
  658. mismatchedsizeinfo[i].expanded = expanded;
  659. mismatchedsizeinfo[i].entry = entry;
  660. }
  661. bool incomplete()
  662. {
  663. for (unsigned j=0;j<max;j++)
  664. if (!primaryresolved[j]&&!replicateresolved[j])
  665. return true;
  666. return false;
  667. }
  668. void resolve(CFileEntry *entry);
  669. IPropertyTree *addFileBranch(IPropertyTree *dst,unsigned flags);
  670. StringAttr lname;
  671. StringAttr pmask;
  672. StringAttr modified;
  673. unsigned primarynum;
  674. unsigned replicatenum;
  675. unsigned max;
  676. unsigned outsidedir;
  677. StringBuffer errdir;
  678. Owned<IGroup> grp;
  679. SocketEndpointArray outsidenodes;
  680. CIArrayOf<CLogicalNameEntry> crosslinked;
  681. bool unknowngrp;
  682. StringAttr mismatchgrp;
  683. bool missinggrp;
  684. bool wrongwidth;
  685. struct MMSinfo {
  686. __int64 expected;
  687. __int64 actual;
  688. __int64 expanded;
  689. CFileEntry *entry;
  690. } *mismatchedsizeinfo;
  691. unsigned nummismatchedsizes;
  692. StringBuffer grpname;
  693. bool replicated;
  694. bool done;
  695. CFileEntry **primaryresolved;
  696. CFileEntry **replicateresolved;
  697. __int64 totalsize;
  698. size32_t recordsize;
  699. bool compressed;
  700. bool grouped;
  701. StringAttr dirpartmask;
  702. CXRefManagerBase &manager;
  703. };
  704. CFileEntry::CFileEntry(const char *_fname,CLogicalNameEntry *_owner,unsigned _part,bool _replicate,__int64 _size, bool compresskludge, __int64 _compsize)
  705. {
  706. fname.set(_fname);
  707. keyhash = hashc((const unsigned char *)fname.get(), fname.length(), 0);
  708. flags = 0;
  709. if (_replicate)
  710. flags |= FEF_REPLICATE;
  711. part = _part;
  712. #ifdef PARTS_SIZE_NEEDED
  713. size = (_compsize>0)?_compsize:_size;
  714. expsize = (_compsize>0)?_size:-1;
  715. #endif
  716. owner = _owner;
  717. if (compresskludge)
  718. owner->compressed = true;
  719. }
  720. StringBuffer &CFileEntry::getLogicalName(StringBuffer &buf)
  721. {
  722. buf.append(owner->lname.get()).append('[').append(part);
  723. if (replicate())
  724. buf.append('R');
  725. return buf.append(']');
  726. }
  727. CXRefManagerBase::~CXRefManagerBase()
  728. {
  729. ForEachItemIn(i,logicalnamelist) {
  730. CLogicalNameEntry &item = logicalnamelist.item(i);
  731. item.crosslinked.kill();
  732. }
  733. }
  734. static bool constructLogicalName(const char *fullname,const char *basedir,StringBuffer &logicalname,unsigned &num,unsigned &max)
  735. {
  736. max = 0;
  737. num = 0;
  738. // mask filename with $P$ extension or normal filename to logical name and num (0 for $P$) and max
  739. while (*fullname&&(toupper(*fullname)==toupper(*basedir))) {
  740. fullname++;
  741. basedir++;
  742. }
  743. if (isPathSepChar(*fullname))
  744. fullname++;
  745. if (!*fullname)
  746. return false;
  747. const char *s=fullname;
  748. loop {
  749. const char *e=s;
  750. while (*e&&!isPathSepChar(*e))
  751. e++;
  752. if (!*e)
  753. break;
  754. if (logicalname.length())
  755. logicalname.append("::");
  756. logicalname.append(e-s,s);
  757. s = e+1;
  758. }
  759. const char *ext = strchr(s,'.');
  760. if (!ext)
  761. return false;
  762. loop {
  763. const char *ne = strchr(ext+1,'.');
  764. if (!ne)
  765. break;
  766. ext = ne;
  767. }
  768. const char *es=ext;
  769. if (memicmp(es,"._$P$_of_",9)==0)
  770. es += 9;
  771. else {
  772. if (memcmp(es,"._",2)!=0)
  773. return false;
  774. es+=2;
  775. StringBuffer n;
  776. while (*es!='_') {
  777. if (!*es)
  778. return false;
  779. n.append(*(es++));
  780. }
  781. num = atoi(n.str());
  782. if (memicmp(es,"_of_",4)!=0)
  783. return false;
  784. es += 4;
  785. }
  786. max = atoi(es);
  787. if (logicalname.length())
  788. logicalname.append("::");
  789. logicalname.append(ext-s,s);
  790. logicalname.toLowerCase();
  791. return true;
  792. }
  793. static void constructPartname(const char *filename,unsigned n, StringBuffer &pn,bool replicate)
  794. {
  795. StringBuffer repdir;
  796. // mask filename with $P$ extension to paticular part filename
  797. if (replicate) {
  798. setReplicateDir(filename,repdir);
  799. filename = repdir.str();
  800. }
  801. while (*filename) {
  802. if (memicmp(filename,"_$P$_",5)==0) {
  803. filename = filename+5;
  804. pn.append('_').append(n).append('_');
  805. }
  806. else {
  807. pn.append(*filename);
  808. filename++;
  809. }
  810. }
  811. }
  812. static bool parseFileName(const char *name,StringBuffer &mname,unsigned &num,unsigned &max,bool &replicate)
  813. {
  814. // takes filename and creates mask filename with $P$ extension
  815. StringBuffer nonrepdir;
  816. replicate = setReplicateDir(name,nonrepdir,false);
  817. if (replicate)
  818. name = nonrepdir.str();
  819. num = 0;
  820. max = 0;
  821. loop {
  822. char c=*name;
  823. if (!c)
  824. break;
  825. if ((c=='.')&&(name[1]=='_')) {
  826. unsigned pn = 0;
  827. const char *s = name+2;
  828. while (*s&&isdigit(*s)) {
  829. pn = pn*10+(*s-'0');
  830. s++;
  831. }
  832. if (pn&&(memicmp(s,"_of_",4)==0)) {
  833. unsigned mn = 0;
  834. s += 4;
  835. while (*s&&isdigit(*s)) {
  836. mn = mn*10+(*s-'0');
  837. s++;
  838. }
  839. if ((mn!=0)&&((*s==0)||(*s=='.'))&&(mn>=pn)) { // NB allow trailing extension
  840. mname.append("._$P$_of_").append(mn);
  841. if (*s)
  842. mname.append(s);
  843. num = pn;
  844. max = mn;
  845. return true;
  846. }
  847. }
  848. }
  849. mname.append(c);
  850. name++;
  851. }
  852. return false;
  853. }
  854. class COrphanEntry: public CInterface
  855. {
  856. public:
  857. StringAttr basedir;
  858. StringAttr fname;
  859. StringAttr modified;
  860. unsigned max;
  861. __int64 size;
  862. SocketEndpointArray primaryepa;
  863. UnsignedArray primarypna;
  864. SocketEndpointArray replicateepa;
  865. UnsignedArray replicatepna;
  866. bool dirfailed;
  867. CLogicalNameEntry *misplaced;
  868. CXRefManagerBase &manager;
  869. byte incompletestate; // 0 unknown, 1 ignore, 2 incomplete
  870. COrphanEntry(CXRefManagerBase &_manager, const char *_fname,const char *_basedir,unsigned _max,const char *_modified,CLogicalNameEntry *_misplaced)
  871. : manager(_manager),fname(_fname), basedir(_basedir), modified(_modified)
  872. {
  873. dirfailed = false;
  874. max = _max;
  875. size = 0;
  876. misplaced = _misplaced;
  877. incompletestate = 0;
  878. }
  879. void add(SocketEndpoint &ep,unsigned pn,bool replicate,__int64 sz)
  880. {
  881. if (max&&(pn>=max)) {
  882. return;
  883. }
  884. size += sz;
  885. if (replicate) {
  886. replicateepa.append(ep);
  887. replicatepna.append(pn);
  888. }
  889. else {
  890. primaryepa.append(ep);
  891. primarypna.append(pn);
  892. }
  893. }
  894. bool complete()
  895. {
  896. if (incompletestate)
  897. return (incompletestate>1);
  898. if (!max) {
  899. // check matches mask
  900. return true;
  901. }
  902. if (max==num()) {
  903. // check for a group here
  904. incompletestate = 2;
  905. return true;
  906. }
  907. incompletestate = 1;
  908. return false;
  909. }
  910. bool isSingleton()
  911. {
  912. if (primaryepa.ordinality()==1) {
  913. if (replicateepa.ordinality()>1)
  914. return false;
  915. if (primarypna.item(0)>0)
  916. return false;
  917. }
  918. if (replicateepa.ordinality()==1) {
  919. if (primaryepa.ordinality()>1)
  920. return false;
  921. if (replicatepna.item(0)>0)
  922. return false;
  923. }
  924. return true;
  925. }
  926. bool singletonName(StringBuffer &name,bool replicate)
  927. {
  928. SocketEndpoint ep;
  929. if (replicate) {
  930. if (replicateepa.ordinality()==0)
  931. return false;
  932. ep = replicateepa.item(0);
  933. }
  934. else {
  935. if (primaryepa.ordinality()==0)
  936. return false;
  937. ep = primaryepa.item(0);
  938. }
  939. StringBuffer partname;
  940. constructPartname(fname.get(),1,partname,replicate);
  941. RemoteFilename rfn;
  942. rfn.setPath(ep,partname.str());
  943. rfn.getRemotePath(name);
  944. return true;
  945. }
  946. unsigned num()
  947. {
  948. if (!max)
  949. return 1;
  950. bool *done = (bool *)calloc(sizeof(bool),max);
  951. ForEachItemIn(i1,primarypna)
  952. done[primarypna.item(i1)] = true;
  953. ForEachItemIn(i2,replicatepna)
  954. done[replicatepna.item(i2)] = true;
  955. unsigned ret = 0;
  956. for (unsigned i=0;i<max;i++)
  957. if (done[i])
  958. ret++;
  959. free(done);
  960. return ret;
  961. }
  962. void getEps(StringBuffer &out,bool replicate)
  963. {
  964. SocketEndpointArray &epa=(replicate?replicateepa:primaryepa);
  965. UnsignedArray &pna=(replicate?replicatepna:primarypna);
  966. unsigned *sorted = new unsigned[epa.ordinality()];
  967. // its expected to be in order so just do insertion sort
  968. ForEachItemIn(i1,pna) {
  969. unsigned i = i1;
  970. unsigned pn = pna.item(i1);
  971. while (i&&(pna.item(sorted[i-1])>pn))
  972. i--;
  973. if (i<i1)
  974. memmove(sorted+i+1,sorted+i,sizeof(unsigned)*(i1-i));
  975. sorted[i] = i1;
  976. }
  977. StringBuffer prefix;
  978. ForEachItemIn(i2,epa) {
  979. if (i2)
  980. out.append(", ");
  981. SocketEndpoint &item = epa.item(sorted[i2]);
  982. StringBuffer cur;
  983. item.getUrlStr(cur);
  984. const char *s1 = prefix.str();
  985. const char *s2 = cur.str();
  986. if (prefix.length()&&(memcmp(s1,s2,prefix.length())==0))
  987. out.append(s2+prefix.length());
  988. else {
  989. unsigned n = 2;
  990. out.append(s2);
  991. prefix.clear();
  992. while (*s2&&n) {
  993. prefix.append(*s2);
  994. if (*s2=='.')
  995. n--;
  996. s2++;
  997. }
  998. }
  999. out.append('[').append(pna.item(sorted[i2])+1).append(']');
  1000. }
  1001. }
  1002. IPropertyTree *addFileBranch(IPropertyTree *dst,unsigned flags)
  1003. {
  1004. unsigned i;
  1005. unsigned numparts = max?max:1;
  1006. StringBuffer buf;
  1007. IPropertyTree *out;
  1008. if (flags&CFBpartmask) {
  1009. unsigned np = (flags&CFBpartprimary)?primarypna.ordinality():0;
  1010. unsigned nr = (flags&CFBpartreplicate)?replicatepna.ordinality():0;
  1011. if (np+nr==0)
  1012. return NULL;
  1013. out = addBranch(dst,"File");
  1014. if (fname.get())
  1015. {
  1016. StringBuffer expfn;
  1017. StringBuffer repfn;
  1018. const char *pm = fname.get();
  1019. if (np+nr==1) {
  1020. expandMask(expfn, fname.get(), (np?primarypna.item(0):replicatepna.item(0)), numparts);
  1021. pm = expfn.str();
  1022. if ((np==0)&&setReplicateDir(pm,repfn))
  1023. pm = repfn.str();
  1024. }
  1025. else if (((flags&CFBpartprimary)==0) && setReplicateDir(pm,repfn)) // not sure if this used
  1026. pm = repfn.str();
  1027. out->setProp("Partmask",pm);
  1028. }
  1029. }
  1030. else
  1031. out = addBranch(dst,"File");
  1032. if (flags&CFBnumparts) {
  1033. out->setPropInt("Numparts",numparts);
  1034. }
  1035. if (flags&CFBmodified) {
  1036. if (modified.get())
  1037. out->setProp("Modified",modified.get());
  1038. }
  1039. if (flags&CFBcluster) {
  1040. // don't support cluster resolution on orphans
  1041. flags |= (CFBpartnum|CFBpartnode);
  1042. }
  1043. if (flags&CFBsize) {
  1044. out->setPropInt64("Size",size);
  1045. }
  1046. if (flags&CFBpartmask) {
  1047. for (int copy=0;copy<=1;copy++) {
  1048. if ((!copy&&(flags&CFBpartprimary))||
  1049. (copy&&(flags&CFBpartreplicate))) {
  1050. UnsignedArray *parts = new UnsignedArray[numparts];
  1051. if (copy) {
  1052. ForEachItemIn(i2,replicatepna)
  1053. parts[replicatepna.item(i2)].append(i2);
  1054. }
  1055. else {
  1056. ForEachItemIn(i1,primarypna)
  1057. parts[primarypna.item(i1)].append(i1);
  1058. }
  1059. for (i=0;i<numparts;i++) {
  1060. if (parts[i].ordinality()) {
  1061. StringBuffer xpath;
  1062. xpath.appendf("Part[Num=\"%d\"]",i+1);
  1063. IPropertyTree *b = out->queryPropTree(xpath.str());
  1064. if (!b)
  1065. b = addBranch(out,"Part");
  1066. if (flags&CFBpartnode) {
  1067. ForEachItemIn(i3,parts[i]) {
  1068. unsigned p = parts[i].item(i3);
  1069. if (copy) {
  1070. replicateepa.item(p).getUrlStr(buf.clear());
  1071. b->addProp("RNode",buf.str());
  1072. }
  1073. else {
  1074. primaryepa.item(p).getUrlStr(buf.clear());
  1075. b->addProp("Node",buf.str());
  1076. }
  1077. }
  1078. }
  1079. if (flags&CFBpartnum)
  1080. b->setPropInt("Num",i+1);
  1081. }
  1082. }
  1083. delete [] parts;
  1084. }
  1085. }
  1086. }
  1087. return out;
  1088. }
  1089. };
  1090. void CLogicalNameEntry::resolve(CFileEntry *entry)
  1091. {
  1092. unsigned part=entry->part-1;
  1093. if (part>=max) { // MORE
  1094. manager.error(lname.get(),"Part %d: Invalid entry (greater than max %d)",part+1,max);
  1095. return;
  1096. }
  1097. if (entry->replicate()) {
  1098. if (replicateresolved[part]) {
  1099. manager.error(lname.get(),"Part %d: Multiple replicated part entry %d",part+1);
  1100. }
  1101. else {
  1102. replicatenum++;
  1103. replicateresolved[part] = entry;
  1104. }
  1105. }
  1106. else {
  1107. if (primaryresolved[part]) {
  1108. manager.error(lname.get(),"Part %d: Multiple primary part entry",part+1);
  1109. }
  1110. else {
  1111. primarynum++;
  1112. primaryresolved[part] = entry;
  1113. #ifdef PARTS_SIZE_NEEDED
  1114. if (entry->size>0) {
  1115. totalsize += entry->size;
  1116. if (((int)recordsize>0)&&
  1117. ((entry->size%recordsize)!=0)&&
  1118. !compressed&&
  1119. (!grouped||(entry->size%(recordsize+1)!=0))) {
  1120. manager.error(lname.get(),"Part %d: Record size %d not multiple of file size %"I64F"d\n",part+1,recordsize,entry->size);
  1121. }
  1122. }
  1123. #endif
  1124. }
  1125. }
  1126. #ifdef PARTS_SIZE_NEEDED
  1127. if (primaryresolved[part]&&replicateresolved[part]&&(primaryresolved[part]->size!=replicateresolved[part]->size)) {
  1128. manager.error(lname.get(),"Part %d: primary size %"I64F"d is different to replicate size %"I64F"d",part+1,primaryresolved[part]->size,replicateresolved[part]->size);
  1129. }
  1130. #endif
  1131. }
  1132. IPropertyTree *CLogicalNameEntry::addFileBranch(IPropertyTree *dst,unsigned flags)
  1133. {
  1134. unsigned i;
  1135. StringBuffer buf;
  1136. IPropertyTree *out = addBranch(dst,"File");
  1137. if (flags&CFBname) {
  1138. out->setProp("Name",lname.get());
  1139. }
  1140. if (flags&CFBpartmask) {
  1141. if (pmask.get()) {
  1142. StringBuffer tmp(pmask.get());
  1143. tmp.toLowerCase();
  1144. out->setProp("Partmask",tmp.str());
  1145. }
  1146. }
  1147. if (flags&CFBnumparts) {
  1148. out->setPropInt("Numparts",max);
  1149. }
  1150. if (flags&CFBpartslost) {
  1151. unsigned n=0;
  1152. for (i=0;i<max;i++)
  1153. if ((primaryresolved[i]==NULL)&&(replicateresolved[i]==NULL))
  1154. n++;
  1155. out->setPropInt("Partslost",n);
  1156. }
  1157. if (flags&CFBprimarylost) {
  1158. if (primarynum!=max)
  1159. out->setPropInt("Primarylost",max-primarynum);
  1160. }
  1161. if (flags&CFBreplicatedlost) {
  1162. if (replicatenum!=max)
  1163. out->setPropInt("Replicatedlost",max-replicatenum);
  1164. }
  1165. if (flags&CFBmodified) {
  1166. if (modified.get())
  1167. out->setProp("Modified",modified.get());
  1168. }
  1169. if (flags&CFBcluster) {
  1170. if (grpname.length())
  1171. out->setProp("Cluster",grpname.str());
  1172. }
  1173. #ifdef PARTS_SIZE_NEEDED
  1174. if (flags&CFBsize) {
  1175. out->setPropInt64("Size",totalsize);
  1176. }
  1177. #endif
  1178. if ((flags&CFBpartmask)&&grp&&grp->ordinality()) {
  1179. unsigned n=0;
  1180. unsigned nc = (flags&CFBpartreplicate)?2:1;
  1181. unsigned rep;
  1182. for (i=0;i<max;i++) {
  1183. for (rep=0;rep<nc;rep++) {
  1184. CFileEntry *e=rep?replicateresolved[i]:primaryresolved[i];
  1185. if ((flags&CFBpartslost)&&e)
  1186. continue;
  1187. IPropertyTree *part = addBranch(out,"Part");
  1188. if (flags&CFBpartnode) {
  1189. grp->queryNode((i+rep)%grp->ordinality()).endpoint().getUrlStr(buf.clear()); // TBD check grp==cluster TBD
  1190. part->setProp("Node",buf.str());
  1191. }
  1192. if (flags&CFBpartnum) {
  1193. part->setPropInt("Num",i+1);
  1194. }
  1195. if (flags&CFBpartreplicate) {
  1196. if (rep)
  1197. part->setPropInt("Replicate",1);
  1198. }
  1199. if (flags&CFBmismatchedsizes) {
  1200. unsigned k = nummismatchedsizes;
  1201. while (k&&(mismatchedsizeinfo[k-1].entry!=e))
  1202. k--;
  1203. if (k) {
  1204. part->setPropInt64("Recordedsize",mismatchedsizeinfo[k-1].expected);
  1205. part->setPropInt64("Actualsize",mismatchedsizeinfo[k-1].actual);
  1206. }
  1207. }
  1208. }
  1209. }
  1210. }
  1211. return out;
  1212. }
  1213. struct TimedBlock
  1214. {
  1215. char *msg;
  1216. unsigned start;
  1217. unsigned limit;
  1218. unsigned ln;
  1219. TimedBlock(const char *_msg,unsigned _limit,unsigned _ln)
  1220. {
  1221. msg = strdup(_msg);
  1222. ln = _ln;
  1223. start = msTick();
  1224. limit = _limit;
  1225. }
  1226. ~TimedBlock()
  1227. {
  1228. unsigned elapsed=msTick()-start;
  1229. if (elapsed>limit)
  1230. PrintLog("TIME: %s took %dms - line(%d)",msg,elapsed,ln);
  1231. free(msg);
  1232. }
  1233. };
  1234. #define TIMEDBLOCK(name,msg,lim) TimedBlock name(msg,lim,__LINE__);
  1235. void loadFromDFS(CXRefManagerBase &manager,IGroup *grp,unsigned numdirs,const char **dirbaselist,const char* Cluster)
  1236. {
  1237. rank_t r=grp->ordinality();
  1238. while (r--) {
  1239. SocketEndpoint ep=grp->queryNode(r).endpoint();
  1240. CEndpointItem *item=manager.EndpointTable.find(ep);
  1241. if (!item) {
  1242. item = new CEndpointItem;
  1243. item->ep = ep;
  1244. item->rank = r;
  1245. manager.EndpointTable.add(*item);
  1246. }
  1247. }
  1248. class Cscanner: public CSDSFileScanner
  1249. {
  1250. SocketEndpointArray epa;
  1251. CLogicalNameEntry* lnentry; // set during processFile
  1252. StringBuffer localname;
  1253. StringBuffer remotename;
  1254. unsigned pass;
  1255. unsigned numdirs;
  1256. const char **dirbaselist;
  1257. CXRefManagerBase &manager;
  1258. IGroup *testgroup;
  1259. Owned<IGroup> cachedgroup;
  1260. public:
  1261. Cscanner(CXRefManagerBase &_manager, IGroup *_testgroup, unsigned _numdirs,const char **_dirbaselist)
  1262. : manager(_manager)
  1263. {
  1264. numdirs = _numdirs;
  1265. dirbaselist = _dirbaselist;
  1266. testgroup = _testgroup;
  1267. }
  1268. bool checkGroupOk(const char *grpname)
  1269. {
  1270. if (!grpname)
  1271. return true;
  1272. cachedgroup.setown(queryNamedGroupStore().lookup(grpname));
  1273. if (!cachedgroup)
  1274. return true;
  1275. ForEachNodeInGroup(i,*cachedgroup)
  1276. if (testgroup->isMember(&cachedgroup->queryNode(i))) // would be better to have hash here TBD
  1277. return true;
  1278. return false;
  1279. }
  1280. virtual bool checkFileOk(IPropertyTree &file,const char *filename)
  1281. {
  1282. StringArray groups;
  1283. getFileGroups(&file,groups);
  1284. ForEachItemIn(i,groups) {
  1285. if (checkGroupOk(groups.item(i)))
  1286. return true;
  1287. }
  1288. return false;
  1289. }
  1290. void processParts(IPropertyTree &root)
  1291. {
  1292. Owned<IPropertyTreeIterator> iter;
  1293. unsigned numparts = root.getPropInt("@numparts");
  1294. MemoryBuffer mb;
  1295. if (root.getPropBin("Parts",mb)) {
  1296. iter.setown(deserializePartAttrIterator(mb)); // this itterator is in order
  1297. ForEach(*iter) {
  1298. IPropertyTree &part = iter->query();
  1299. unsigned partno = part.getPropInt("@num",0);
  1300. SocketEndpoint ep;
  1301. const char *eps = part.queryProp("@node");
  1302. if (eps&&*eps)
  1303. ep.set(eps);
  1304. processPart(root,part,partno,numparts,ep);
  1305. }
  1306. }
  1307. else { // slow but should be going anyway
  1308. iter.setown(root.getElements("Part")); // use parts
  1309. IArrayOf<IPropertyTree> parts;
  1310. Owned<IPropertyTree> empty = createPTree("Attr");
  1311. unsigned i;
  1312. for (i=0;i<numparts;i++)
  1313. parts.append(*empty.getLink());
  1314. unsigned lastpartno=0;
  1315. ForEach(*iter) {
  1316. IPropertyTree &part = iter->query();
  1317. unsigned partno = part.getPropInt("@num",0);
  1318. if (partno&&(partno!=lastpartno)&&(partno<=numparts)) {
  1319. parts.replace(*createPTreeFromIPT(&part),partno-1);
  1320. lastpartno = partno;
  1321. }
  1322. }
  1323. for (i=0;i<numparts;i++) {
  1324. IPropertyTree &part = parts.item(i);
  1325. SocketEndpoint ep;
  1326. const char *eps = part.queryProp("@node");
  1327. if (eps&&*eps)
  1328. ep.set(eps);
  1329. processPart(root,part,i+1,numparts,ep);
  1330. }
  1331. }
  1332. }
  1333. void processFile(IPropertyTree &file,StringBuffer &name)
  1334. {
  1335. if (!manager.msgcallback.get())
  1336. DBGLOG("processFile %s",name.str());
  1337. Owned<CLogicalNameEntry> lnentrybase = new CLogicalNameEntry(manager,name.str(),file);
  1338. lnentry = lnentrybase;
  1339. epa.kill();
  1340. pass=0;
  1341. processParts(file); // could improve
  1342. manager.logicalnamemap.setValue(name.str(), lnentry);
  1343. lnentry->Link();
  1344. manager.logicalnamelist.append(*lnentry);
  1345. manager.logicaldirmap.setValue(lnentry->dirpartmask.get(), lnentry);
  1346. lnentry->setGroup(epa.ordinality()?createIGroup(epa):NULL,cachedgroup.getLink());
  1347. pass = 2;
  1348. processParts(file);
  1349. }
  1350. void processFile(StringBuffer &name,StringBuffer &fullpath)
  1351. {
  1352. Owned<IRemoteConnection> conn = querySDS().connect(fullpath.str(),myProcessSession(),RTM_LOCK_READ|RTM_SUB, SDS_CONNECT_TIMEOUT);
  1353. if (conn) // there is at least one duff entry that this is needed for
  1354. processFile(*conn->queryRoot(),name);
  1355. }
  1356. void processPart(IPropertyTree &file,IPropertyTree &part,unsigned partno,unsigned numparts,SocketEndpoint &ep)
  1357. {
  1358. if (pass==0) {
  1359. partno--;
  1360. if (!ep.isNull()) {
  1361. SocketEndpoint nullep;
  1362. while (partno>=epa.ordinality())
  1363. epa.append(nullep);
  1364. epa.item(partno) = ep;
  1365. }
  1366. }
  1367. else {
  1368. bool replicate=false;
  1369. const char *partname = part.queryProp("@name");
  1370. const char *partmask = file.queryProp("@partmask");
  1371. const char *partdir = file.queryProp("@directory");
  1372. int replicateoffset = file.getPropInt("@replicateOffset",1);
  1373. loop {
  1374. RemoteFilename rfn;
  1375. IGroup *grp = lnentry->queryGroup();
  1376. if (!grp) {
  1377. manager.warn(lnentry->lname.get(),"No group found, ignoring logical file");
  1378. return;
  1379. }
  1380. constructPartFilename(grp,partno,numparts,partname,partmask,partdir,replicate,replicateoffset,rfn);
  1381. SocketEndpoint rep=rfn.queryEndpoint();
  1382. if (manager.EndpointTable.find(rep)!=NULL) {
  1383. rfn.getLocalPath(localname.clear());
  1384. bool dirmatch=false;
  1385. unsigned k;
  1386. for (k=0;k<numdirs;k++)
  1387. {
  1388. if (memicmp(localname.str(),dirbaselist[k],strlen(dirbaselist[k]))==0) {
  1389. dirmatch = true;
  1390. break;
  1391. }
  1392. }
  1393. if (dirmatch) {
  1394. rfn.getRemotePath(remotename.clear());
  1395. remotename.toLowerCase();
  1396. CFileEntry *entry= new CFileEntry(remotename.str(),lnentry,partno,replicate,part.getPropInt64("@size", -1),part.getPropInt("@rowCompression", 0)!=0,part.getPropInt("@compressedSize", -1));
  1397. CFileEntry *oldentry= manager.filemap.find(remotename.str());
  1398. if (oldentry)
  1399. {
  1400. if (oldentry->owner!=lnentry) {
  1401. #ifdef CROSSLINK_CHECK_NEEDED
  1402. StringBuffer s1;
  1403. StringBuffer s2;
  1404. entry->getLogicalName(s1);
  1405. oldentry->getLogicalName(s2);
  1406. //outf("CROSSLINK: %s contains same file as %s\n",s1.str(),s2.str());
  1407. entry->owner->addCrossLink(oldentry->owner);
  1408. oldentry->owner->addCrossLink(entry->owner);
  1409. oldentry->crosslink.setown(entry);
  1410. #endif
  1411. }
  1412. }
  1413. else {
  1414. manager.filemap.add(*entry);
  1415. }
  1416. }
  1417. else {
  1418. lnentry->outsidedir++;
  1419. if (lnentry->errdir.length()==0)
  1420. lnentry->errdir.append(localname.str());
  1421. }
  1422. }
  1423. else {
  1424. if (lnentry->outsidenodes.find(rep)==NotFound)
  1425. lnentry->outsidenodes.append(rep);
  1426. }
  1427. if (replicate)
  1428. break;
  1429. replicate = true;
  1430. }
  1431. }
  1432. }
  1433. } scanner(manager,grp,numdirs,dirbaselist);
  1434. manager.log("Loading Files branch from SDS");
  1435. Owned<IRemoteConnection> conn = querySDS().connect(SDS_DFS_ROOT,myProcessSession(),RTM_LOCK_READ, INFINITE);
  1436. if (!conn) {
  1437. throw MakeStringException(-1,"Could not connect to Files");
  1438. }
  1439. conn->changeMode(RTM_NONE);
  1440. manager.log("Files loaded, scanning");
  1441. scanner.scan(conn);
  1442. manager.log("Scanning done");
  1443. }
  1444. class CPhysicalXREF
  1445. {
  1446. unsigned numdirs;
  1447. const char **dirbaselist;
  1448. CriticalSection xrefsect;
  1449. CXRefManagerBase &manager;
  1450. public:
  1451. CPhysicalXREF(CXRefManagerBase &_manager, unsigned _numdirs,const char **_dirbaselist)
  1452. : manager(_manager)
  1453. {
  1454. numdirs = _numdirs;
  1455. dirbaselist = _dirbaselist;
  1456. setDaliServixSocketCaching(true);
  1457. }
  1458. void getBaseDir(const char *name,StringBuffer &basedir)
  1459. {
  1460. unsigned i;
  1461. for (i=0;i<numdirs;i++) {
  1462. size32_t l = strlen(dirbaselist[i]);
  1463. if ((memicmp(dirbaselist[i],name,l)==0)&&isPathSepChar(name[l])) {
  1464. basedir.append(dirbaselist[i]).toLowerCase();
  1465. return;
  1466. }
  1467. }
  1468. }
  1469. void xreffile(SocketEndpoint &ep,const char *fullname,IPropertyTree *file)
  1470. {
  1471. RemoteFilename rname;
  1472. StringBuffer name;
  1473. rname.setPath(ep,fullname);
  1474. rname.getRemotePath(name);
  1475. name.toLowerCase();
  1476. CFileEntry *entry= manager.filemap.find(name.str());
  1477. __int64 sz = file->getPropInt64("@size", -1);
  1478. if (entry) {
  1479. do {
  1480. entry->flags |= FEF_RESOLVED;
  1481. #ifdef PARTS_SIZE_NEEDED
  1482. if (entry->owner->compressed) {
  1483. if ((entry->expsize>=0)&&((sz!=entry->size)||((entry->size==0)&&(entry->expsize>0)))) {
  1484. manager.error(entry->owner->lname.get(),"Part %d: %s size mismatch: recorded size %"I64F"d, actual size %"I64F"d, expanded size %"I64F"d",entry->part,entry->replicate?"replicate":"primary",entry->size,sz,entry->expsize);
  1485. entry->owner->addMismatchedSize(entry->size,sz,entry->expsize,entry);
  1486. }
  1487. if (entry->size>=0) {
  1488. manager.totalCompressed++;
  1489. manager.totalCompressedSize+=sz;
  1490. manager.totalUncompressedSize+=entry->size;
  1491. }
  1492. }
  1493. else if ((entry->size>=0)&&(entry->size!=sz)) {
  1494. if (!entry->replicate) {
  1495. entry->owner->addMismatchedSize(entry->size,sz,-1,entry);
  1496. }
  1497. manager.error(entry->owner->lname.get(),"Part %d: %s size mismatch: recorded size %"I64F"d, actual size %"I64F"d",entry->part,entry->replicate?"replicate":"primary",entry->size,sz);
  1498. entry->size=sz; // set to actual size for fix
  1499. }
  1500. #endif
  1501. entry->owner->resolve(entry);
  1502. #ifdef CROSSLINK_CHECK_NEEDED
  1503. entry = entry->crosslink;
  1504. #else
  1505. entry = NULL;
  1506. #endif
  1507. } while (entry);
  1508. }
  1509. else {
  1510. manager.totalSizeOrphans += sz;
  1511. manager.totalNumOrphans++;
  1512. StringBuffer orphanname;
  1513. unsigned m;
  1514. unsigned n;
  1515. bool replicate;
  1516. parseFileName(fullname,orphanname,n,m,replicate);
  1517. if (n>m)
  1518. manager.error(fullname, "Part %d: number greater than max %d",n+1,m);
  1519. else {
  1520. orphanname.toLowerCase();
  1521. COrphanEntryPtr *entryp = manager.orphanmap.getValue(orphanname.str());
  1522. COrphanEntryPtr entry;
  1523. if (entryp)
  1524. entry = *entryp;
  1525. else {
  1526. StringBuffer basedir;
  1527. getBaseDir(orphanname.str(),basedir);
  1528. CLogicalNameEntryPtr *parentp = manager.logicaldirmap.getValue(orphanname.str());
  1529. entry = new COrphanEntry(manager,orphanname.str(),basedir.str(),m,file->queryProp("@modified"),parentp?*parentp:NULL);
  1530. manager.orphanmap.setValue(orphanname.str(),entry);
  1531. manager.orphanlist.append(*entry);
  1532. }
  1533. if (n)
  1534. n--;
  1535. entry->add(ep,n,replicate,sz);
  1536. }
  1537. }
  1538. }
  1539. void xrefdir(SocketEndpoint &ep,StringBuffer &path,IPropertyTree *dirtree,unsigned clustsize)
  1540. {
  1541. unsigned fnum=0;
  1542. unsigned dnum=0;
  1543. size32_t pathlen = path.length();
  1544. Owned<IPropertyTreeIterator> dirs= dirtree->getElements("directory");
  1545. if (dirs->first()) {
  1546. do {
  1547. IPropertyTree &dir = dirs->query();
  1548. if (path.length()&&!isPathSepChar(path.charAt(path.length()-1)))
  1549. path.append(getPathSepChar(path.str()));
  1550. path.append(dir.queryProp("@name"));
  1551. xrefdir(ep,path,&dir,clustsize);
  1552. path.setLength(pathlen);
  1553. } while (dirs->next());
  1554. dnum++;
  1555. }
  1556. Owned<IPropertyTreeIterator> files= dirtree->getElements("file");
  1557. if (files->first()) {
  1558. do {
  1559. IPropertyTree &file = files->query();
  1560. const char *fname=file.queryProp("@name");
  1561. size32_t l = strlen(fname);
  1562. if ((l>4)&&(stricmp(fname+l-4,".crc")==0))
  1563. continue;
  1564. if (path.length()&&!isPathSepChar(path.charAt(path.length()-1)))
  1565. path.append(getPathSepChar(path.str()));
  1566. path.append(fname);
  1567. xreffile(ep,path.str(),&file);
  1568. path.setLength(pathlen);
  1569. fnum++;
  1570. } while (files->next());
  1571. }
  1572. manager.incDirSize(ep,path.str(),dirtree->getPropInt64("@size", 0),fnum,dnum,clustsize);
  1573. }
  1574. void xref(IPropertyTree *machine,unsigned clustsize)
  1575. {
  1576. CriticalBlock block(xrefsect);
  1577. StringBuffer path;
  1578. SocketEndpoint ep(machine->queryProp("@ip"));
  1579. ep.port = 0;
  1580. xrefdir(ep,path,machine,clustsize);
  1581. }
  1582. };
  1583. class CXRefManager: public CXRefManagerBase
  1584. {
  1585. #ifdef PARTS_SIZE_NEEDED
  1586. void fixSizeInDFS()
  1587. {
  1588. StringBuffer name;
  1589. __int64 sz;
  1590. ForEachItemIn(i,logicalnamelist) {
  1591. CLogicalNameEntry &item = logicalnamelist.item(i);
  1592. if (!item.done&&item.nummismatchedsizes) {
  1593. Owned<IDistributedFile> file = queryDistributedFileDirectory().lookup(item.lname.get());
  1594. if (file) {
  1595. outf("checking %s\n",item.lname.get());
  1596. Owned<IDistributedFilePartIterator> partiter = file->getIterator();
  1597. unsigned partno=0;
  1598. __int64 total = 0;
  1599. ForEach(*partiter) {
  1600. partno++;
  1601. Owned<IDistributedFilePart> part = &partiter->get();
  1602. RemoteFilename rname;
  1603. part->getFilename(rname);
  1604. rname.getRemotePath(name.clear());
  1605. name.toLowerCase();
  1606. CFileEntry *entry= filemap.find(name.str());
  1607. if (entry) {
  1608. __int64 sz = part->queryProperties().getPropInt64("@size", -1);
  1609. if (sz!=entry->size) {
  1610. StringBuffer s1;
  1611. entry->getLogicalName(s1);
  1612. outf("SIZEFIX: Changing size for %s from %"I64F"d to %"I64F"d\n",s1.str(),sz,entry->size);
  1613. part->lockProperties().setPropInt64("@size", entry->size);
  1614. part->unlockProperties();
  1615. }
  1616. if (total!=-1)
  1617. total += entry->size;
  1618. }
  1619. else
  1620. total = -1;
  1621. }
  1622. sz = file->queryProperties().getPropInt64("@size", -1);
  1623. if (sz!=total) {
  1624. outf("SIZEFIX: Changing total size for %s from %"I64F"d to %"I64F"d\n",item.lname.get(),sz,total);
  1625. if (total!=-1)
  1626. file->lockProperties().setPropInt64("@size", total);
  1627. else
  1628. file->lockProperties().removeProp("@size");
  1629. file->unlockProperties();
  1630. }
  1631. }
  1632. }
  1633. }
  1634. outf("Size fix completed\n");
  1635. }
  1636. #endif
  1637. void xrefRemoteDirectories(IGroup *g,unsigned numdirs,const char **dirbaselist,unsigned numthreads)
  1638. {
  1639. if (numthreads<=1)
  1640. numthreads = 10; // should be safe now
  1641. StringBuffer dirlist;
  1642. unsigned j;
  1643. for (j=0;j<numdirs;j++) {
  1644. const char *basedir = dirbaselist[j];
  1645. if (!basedir||!*basedir)
  1646. continue;
  1647. if (dirlist.length())
  1648. dirlist.append(';');
  1649. dirlist.append(basedir);
  1650. }
  1651. CPhysicalXREF cxref(*this,numdirs,dirbaselist);
  1652. unsigned numfails=0;
  1653. class casyncfor: public CAsyncFor
  1654. {
  1655. CPhysicalXREF &cxref;
  1656. IGroup *g;
  1657. const char *dirlist;
  1658. bool abort;
  1659. CXRefManagerBase &manager;
  1660. CriticalSection crit;
  1661. unsigned numfails;
  1662. public:
  1663. casyncfor(IGroup *_g, const char *_dirlist, CPhysicalXREF &_cxref, CXRefManagerBase *_manager)
  1664. : cxref(_cxref), manager(*_manager)
  1665. {
  1666. g = _g;
  1667. dirlist = _dirlist;
  1668. abort = false;
  1669. numfails = 0;
  1670. }
  1671. void Do(unsigned idx)
  1672. {
  1673. if (abort)
  1674. return;
  1675. CriticalBlock block(manager.logsect); // only the get directory is async
  1676. try {
  1677. StringBuffer msg;
  1678. INode &node = g->queryNode(idx);
  1679. node.endpoint().getUrlStr(msg);
  1680. manager.log("Getting directories for %s",msg.str());
  1681. manager.addNodeInProgress(node);
  1682. Owned<IPropertyTree> results;
  1683. {
  1684. CriticalUnblock unblock(manager.logsect);
  1685. unsigned short port = getDafsPort(node.endpoint(),numfails,&crit);
  1686. results.setown(getDirectory(dirlist,&node,port));
  1687. }
  1688. manager.log("Crossreferencing %s",msg.str());
  1689. cxref.xref(results,g->ordinality());
  1690. manager.removeNodeInProgress(node);
  1691. }
  1692. catch (IException *)
  1693. {
  1694. abort = true;
  1695. throw;
  1696. }
  1697. }
  1698. } afor(g,dirlist.str(),cxref,this);
  1699. afor.For(g->ordinality(), numthreads,false,true);
  1700. ForEachItemIn(i1,logicalnamelist) {
  1701. CLogicalNameEntry &item = logicalnamelist.item(i1);
  1702. if (item.nummismatchedsizes) {
  1703. error(item.lname.get(),"%d part%s physical size differ from recorded size\n",item.nummismatchedsizes,item.nummismatchedsizes>1?"s":"");
  1704. }
  1705. if (item.crosslinked.ordinality()) {
  1706. StringBuffer to;
  1707. ForEachItemIn(cli1,item.crosslinked) {
  1708. if (cli1)
  1709. to.append(", ");
  1710. to.append(item.crosslinked.item(cli1).lname.get());
  1711. }
  1712. error(item.lname.get(),"Crosslinked with %s",to.str());
  1713. }
  1714. }
  1715. }
  1716. void listDir(IGroup *g,unsigned numdirs,const char **dirbaselist)
  1717. {
  1718. class cDirScan
  1719. {
  1720. public:
  1721. char pathsepchar;
  1722. void scanfile(SocketEndpoint &ep,const char *fullname,IPropertyTree *file)
  1723. {
  1724. const char * tail = fullname;
  1725. loop {
  1726. const char *s = strchr(tail,pathsepchar);
  1727. if (!s)
  1728. break;
  1729. tail = s+1;
  1730. }
  1731. if (memicmp(tail+3,".HISTORY.",9)!=0)
  1732. return;
  1733. RemoteFilename rname;
  1734. StringBuffer name;
  1735. rname.setPath(ep,fullname);
  1736. rname.getRemotePath(name);
  1737. Owned<IFile> f= createIFile(name.str());
  1738. outf("%12"I64F"d %s\n",f->size(),name.str());
  1739. }
  1740. void scandir(SocketEndpoint &ep,StringBuffer &path,IPropertyTree *dirtree)
  1741. {
  1742. size32_t pathlen = path.length();
  1743. Owned<IPropertyTreeIterator> dirs= dirtree->getElements("directory");
  1744. if (dirs->first()) {
  1745. do {
  1746. IPropertyTree &dir = dirs->query();
  1747. if (path.length()&&(path.charAt(path.length()-1)!=pathsepchar))
  1748. path.append(pathsepchar);
  1749. path.append(dir.queryProp("@name"));
  1750. scandir(ep,path,&dir);
  1751. path.setLength(pathlen);
  1752. } while (dirs->next());
  1753. }
  1754. Owned<IPropertyTreeIterator> files= dirtree->getElements("file");
  1755. if (files->first()) {
  1756. do {
  1757. IPropertyTree &file = files->query();
  1758. const char *fname=file.queryProp("@name");
  1759. size32_t l = strlen(fname);
  1760. // if ((l>4)&&(stricmp(fname+l-4,".crc")==0))
  1761. // continue;
  1762. if (path.length()&&(path.charAt(path.length()-1)!=pathsepchar))
  1763. path.append(pathsepchar);
  1764. path.append(fname);
  1765. scanfile(ep,path.str(),&file);
  1766. path.setLength(pathlen);
  1767. } while (files->next());
  1768. }
  1769. }
  1770. void scan(IPropertyTree *machine)
  1771. {
  1772. StringBuffer path;
  1773. SocketEndpoint ep(machine->queryProp("@ip"));
  1774. ep.port = 0;
  1775. scandir(ep,path,machine);
  1776. }
  1777. } scandir;
  1778. StringBuffer dirlist;
  1779. unsigned j;
  1780. for (j=0;j<numdirs;j++) {
  1781. const char *basedir = dirbaselist[j];
  1782. if (!basedir||!*basedir)
  1783. continue;
  1784. if (dirlist.length())
  1785. dirlist.append(';');
  1786. dirlist.append(basedir);
  1787. }
  1788. unsigned numfails = 0;
  1789. for (unsigned i=0;i<g->ordinality();i++) {
  1790. INode &node = g->queryNode(i);
  1791. scandir.pathsepchar = getPathSepChar(dirlist.str());
  1792. unsigned short port = getDafsPort(node.endpoint(),numfails,NULL);
  1793. Owned<IPropertyTree> results = getDirectory(dirlist.str(),&node,port);
  1794. scandir.scan(results);
  1795. }
  1796. }
  1797. CriticalSection xrefsect;
  1798. void addn(StringBuffer &b,unsigned p,unsigned &firstp,unsigned &lastp)
  1799. {
  1800. if (p) {
  1801. if (firstp==0) {
  1802. firstp = p;
  1803. lastp = p;
  1804. }
  1805. else if (lastp+1==p)
  1806. lastp = p;
  1807. else {
  1808. addn(b,0,firstp,lastp);
  1809. firstp = p;
  1810. lastp = p;
  1811. }
  1812. }
  1813. else if (firstp!=0) {
  1814. if (b.length())
  1815. b.append(',');
  1816. b.append(firstp);
  1817. if (lastp!=firstp) {
  1818. b.append('-');
  1819. b.append(lastp);
  1820. }
  1821. firstp = 0;
  1822. }
  1823. }
  1824. const char *plural(const char *s)
  1825. {
  1826. if ((strchr(s,'-')!=NULL)||(strchr(s,',')!=NULL))
  1827. return "s";
  1828. else return "";
  1829. }
  1830. void outputTextReport(const char *filename)
  1831. {
  1832. Owned<IFile> file = createIFile(filename);
  1833. Owned<IFileIO> fileio = file->open(IFOcreate);
  1834. if (!fileio) {
  1835. printf("ERROR cannot create %s\n",filename);
  1836. return;
  1837. }
  1838. outfileio.setown(createIOStream(fileio)); SuperHashIteratorOf<CFileEntry> fileiter(filemap);
  1839. if (errors.ordinality()) {
  1840. outf("\n--------------------------------------------------------\nERRORS\n");
  1841. ForEachItemIn(i,errors) {
  1842. cMessage &item = errors.item(i);
  1843. outf("%s: %s\n",item.lname.get(),item.msg.get());
  1844. }
  1845. }
  1846. if (warnings.ordinality()) {
  1847. outf("\n--------------------------------------------------------\nWARNINGS:\n");
  1848. ForEachItemIn(i,warnings) {
  1849. cMessage &item = warnings.item(i);
  1850. outf("%s: %s\n",item.lname.get(),item.msg.get());
  1851. }
  1852. }
  1853. {
  1854. outf("\n--------------------------------------------------------\nUNRESOLVED PRIMARY PARTS:\n");
  1855. ForEach(fileiter) {
  1856. CFileEntry &item = fileiter.query();
  1857. if (!item.resolved()&&!item.replicate()) {
  1858. StringBuffer s1;
  1859. item.getLogicalName(s1);
  1860. outf("%s (%s)\n",s1.str(),item.fname.get());
  1861. }
  1862. }
  1863. }
  1864. {
  1865. outf("\n--------------------------------------------------------\nUNRESOLVED REPLICATE PARTS:\n");
  1866. ForEach(fileiter) {
  1867. CFileEntry &item = fileiter.query();
  1868. if (!item.resolved()&&item.replicate()) {
  1869. StringBuffer s1;
  1870. item.getLogicalName(s1);
  1871. outf("%s (%s)\n",s1.str(),item.fname.get());
  1872. }
  1873. }
  1874. }
  1875. outf("\n--------------------------------------------------------\nMISPLACED PARTS:\n");
  1876. {
  1877. ForEachItemIn(i,orphanlist) {
  1878. COrphanEntry &item = orphanlist.item(i);
  1879. if (!item.complete()&&item.misplaced) {
  1880. if (item.max>0)
  1881. outf("%s %s (found %d, size %"I64F"d, modified %s)\n",item.fname.get(),item.misplaced->lname.get(),item.num(),item.size,item.modified.get());
  1882. else
  1883. outf("%s (size %"I64F"d, modified %s)\n",item.fname.get(),item.size,item.modified.get());
  1884. StringBuffer s1;
  1885. item.getEps(s1,false);
  1886. outf(" primary: %s\n",s1.str());
  1887. item.getEps(s1.clear(),true);
  1888. outf(" replicate: %s\n",s1.str());
  1889. }
  1890. }
  1891. }
  1892. outf("\n--------------------------------------------------------\nORPHANS PARTIAL:\n");
  1893. {
  1894. ForEachItemIn(i,orphanlist) {
  1895. COrphanEntry &item = orphanlist.item(i);
  1896. if (!item.complete()&&!item.misplaced) {
  1897. if (item.max>0)
  1898. outf("%s (found %d, size %"I64F"d, modified %s)\n",item.fname.get(),item.num(),item.size,item.modified.get());
  1899. else
  1900. outf("%s (size %"I64F"d, modified %s)\n",item.fname.get(),item.size,item.modified.get());
  1901. StringBuffer s1;
  1902. item.getEps(s1,false);
  1903. outf(" primary: %s\n",s1.str());
  1904. item.getEps(s1.clear(),true);
  1905. outf(" replicate: %s\n",s1.str());
  1906. }
  1907. }
  1908. }
  1909. outf("\n--------------------------------------------------------\nSINGLETON ORPHANS:\n");
  1910. {
  1911. ForEachItemIn(i,orphanlist) {
  1912. COrphanEntry &item = orphanlist.item(i);
  1913. if (item.complete()) {
  1914. StringBuffer sname;
  1915. if (item.isSingleton()) {
  1916. if (item.singletonName(sname,false))
  1917. outf("%s (size %"I64F"d)\n",sname.str(),item.size);
  1918. if (item.singletonName(sname.clear(),true))
  1919. outf("%s (size %"I64F"d)\n",sname.str(),item.size);
  1920. }
  1921. }
  1922. }
  1923. }
  1924. outf("\n--------------------------------------------------------\nORPHANS COMPLETE:\n");
  1925. {
  1926. ForEachItemIn(i,orphanlist) {
  1927. COrphanEntry &item = orphanlist.item(i);
  1928. if (item.complete()) {
  1929. StringBuffer sname;
  1930. if (!item.isSingleton()) {
  1931. outf("%s %s(size %"I64F"d)\n",item.fname.get(),item.misplaced?item.misplaced->lname.get():"",item.size);
  1932. StringBuffer s1;
  1933. item.getEps(s1,false);
  1934. outf(" primary: %s\n",s1.str());
  1935. item.getEps(s1.clear(),true);
  1936. outf(" replicate: %s\n",s1.str());
  1937. }
  1938. }
  1939. }
  1940. }
  1941. outf("\n--------------------------------------------------------\nOUTSIDE DIRECTORIES:\n");
  1942. {
  1943. ForEachItemIn(i,logicalnamelist) {
  1944. CLogicalNameEntry &item = logicalnamelist.item(i);
  1945. if (!item.done&&item.outsidedir) {
  1946. outf("%s (directory %s, %d of %d)\n",item.lname.get(),item.errdir.str(),item.outsidedir,item.max);
  1947. item.done = true;
  1948. }
  1949. }
  1950. }
  1951. outf("\n--------------------------------------------------------\nOUTSIDE CLUSTERS:\n");
  1952. {
  1953. ForEachItemIn(i,logicalnamelist) {
  1954. CLogicalNameEntry &item = logicalnamelist.item(i);
  1955. if (!item.done&&item.outsidenodes.ordinality()) {
  1956. outf("%s (%d of %d)\n",item.lname.get(),item.outsidenodes.ordinality(),item.max);
  1957. SocketListCreator cr;
  1958. cr.addSockets(item.outsidenodes);
  1959. outf(" %s\n",cr.getText());
  1960. item.done = true;
  1961. }
  1962. }
  1963. }
  1964. outf("\n--------------------------------------------------------\nINCOMPLETE FILES:\n");
  1965. {
  1966. ForEachItemIn(i,logicalnamelist) {
  1967. CLogicalNameEntry &item = logicalnamelist.item(i);
  1968. if (!item.done&&item.incomplete()) {
  1969. outf("%s (found %d of %d",item.lname.get(),item.primarynum,item.max);
  1970. unsigned m=item.max-item.primarynum;
  1971. if (m<=10) {
  1972. StringBuffer unres;
  1973. for (unsigned j=0;j<item.max;j++) {
  1974. if (!item.primaryresolved[j]&&!item.replicateresolved[j]) {
  1975. if (unres.length()!=0)
  1976. unres.append(',');
  1977. unres.append(j+1);
  1978. }
  1979. }
  1980. outf(" missing parts: %s",unres.str());
  1981. }
  1982. outf(")\n");
  1983. item.done = true;
  1984. }
  1985. }
  1986. }
  1987. #ifdef PARTS_SIZE_NEEDED
  1988. outf("\n--------------------------------------------------------\nMISMATCHED SIZES:\n");
  1989. {
  1990. if (fixSizes)
  1991. fixSizeInDFS();
  1992. ForEachItemIn(i,logicalnamelist) {
  1993. CLogicalNameEntry &item = logicalnamelist.item(i);
  1994. if (!item.done&&item.nummismatchedsizes) {
  1995. outf("%s (%d of %d)\n",item.lname.get(),item.nummismatchedsizes,item.max);
  1996. }
  1997. }
  1998. }
  1999. #endif
  2000. outf("\n--------------------------------------------------------\nPRIMARY MISSING:\n");
  2001. {
  2002. ForEachItemIn(i,logicalnamelist) {
  2003. CLogicalNameEntry &item = logicalnamelist.item(i);
  2004. if (!item.done&&(item.primarynum!=item.max)) {
  2005. outf("%s (found %d of %d",item.lname.get(),item.primarynum,item.max);
  2006. unsigned m=item.max-item.primarynum;
  2007. if (m<=10) {
  2008. StringBuffer unres;
  2009. unsigned j0a;
  2010. for (j0a=0;j0a<item.max;j0a++) {
  2011. if (!item.primaryresolved[j0a]&&item.replicateresolved[j0a]) {
  2012. if (unres.length()!=0)
  2013. unres.append(',');
  2014. unres.append(j0a+1);
  2015. }
  2016. }
  2017. outf(" missing parts: %s",unres.str());
  2018. }
  2019. outf(")\n");
  2020. item.done = true;
  2021. }
  2022. }
  2023. }
  2024. outf("\n--------------------------------------------------------\nSECONDARY MISSING:\n");
  2025. {
  2026. ForEachItemIn(i,logicalnamelist) {
  2027. CLogicalNameEntry &item = logicalnamelist.item(i);
  2028. if (!item.done&&(item.replicatenum!=item.max)) {
  2029. outf("%s (found %d of %d",item.lname.get(),item.replicatenum,item.max);
  2030. unsigned m=item.max-item.primarynum;
  2031. if (m<=10) {
  2032. StringBuffer unres;
  2033. unsigned j0a;
  2034. for (j0a=0;j0a<item.max;j0a++) {
  2035. if (item.primaryresolved[j0a]&&!item.replicateresolved[j0a]) {
  2036. if (unres.length()!=0)
  2037. unres.append(',');
  2038. unres.append(j0a+1);
  2039. }
  2040. }
  2041. outf(" missing parts: %s",unres.str());
  2042. }
  2043. outf(")\n");
  2044. item.done = true;
  2045. }
  2046. }
  2047. }
  2048. outf("\n--------------------------------------------------------\nCROSSLINKED:\n");
  2049. {
  2050. ForEachItemIn(i1,logicalnamelist) {
  2051. CLogicalNameEntry &item = logicalnamelist.item(i1);
  2052. if (!item.done&&item.crosslinked.ordinality()) {
  2053. StringBuffer to;
  2054. ForEachItemIn(cli1,item.crosslinked) {
  2055. if (cli1)
  2056. to.append(", ");
  2057. to.append(item.crosslinked.item(cli1).lname.get());
  2058. }
  2059. outf("%s to %s\n",item.lname.get(),to.str());
  2060. item.done = true;
  2061. }
  2062. }
  2063. }
  2064. outf("\n--------------------------------------------------------\nGROUPS:\n");
  2065. {
  2066. ForEachItemIn(i,logicalnamelist) {
  2067. CLogicalNameEntry &item = logicalnamelist.item(i);
  2068. if (item.unknowngrp||item.mismatchgrp.get()||item.missinggrp) {
  2069. Owned<IDistributedFile> file = queryDistributedFileDirectory().lookup(item.lname.get());
  2070. if (file) {
  2071. if (item.missinggrp)
  2072. outf("WARNING: Missing group for %s\n",item.lname.get());
  2073. else if (item.mismatchgrp.get()) {
  2074. // ** TBD check queryClusterName correct in following
  2075. StringBuffer tmp;
  2076. outf("ERROR: Group mismatch for %s, Group says %s but nodes match %s\n",item.lname.get(),file->getClusterName(0,tmp).str(),item.mismatchgrp.get());
  2077. }
  2078. else if (item.unknowngrp) {
  2079. // ** TBD check queryClusterName correct in following
  2080. StringBuffer tmp;
  2081. outf("WARNING: Unknown group %s for %s\n",file->getClusterName(0,tmp).str(),item.lname.get());
  2082. }
  2083. }
  2084. }
  2085. }
  2086. }
  2087. {
  2088. outf("\n--------------------------------------------------------\nDIRECTORIES:\n");
  2089. ForEachItemIn(i,dirlist) {
  2090. CDfuDirEntry &item = dirlist.item(i);
  2091. if (item.minsize<0)
  2092. item.minsize = 0;
  2093. StringBuffer s1;
  2094. if (!item.minip.isNull())
  2095. item.minip.getIpText(s1);
  2096. StringBuffer s2;
  2097. if (!item.maxip.isNull())
  2098. item.maxip.getIpText(s2);
  2099. StringBuffer skew;
  2100. item.getskew(skew);
  2101. outf("%s numfiles=%u totalsize=%"CF"d minsize=%"CF"d(%s) maxsize=%"CF"d(%s), skew=%s\n",item.name.get(),item.num,item.size,
  2102. item.minsize,s1.str(),item.maxsize,s2.str(),skew.str());
  2103. }
  2104. }
  2105. outf("\n--------------------------------------------------------\nOK:\n");
  2106. {
  2107. ForEachItemIn(i,logicalnamelist) {
  2108. CLogicalNameEntry &item = logicalnamelist.item(i);
  2109. if (!item.done) {
  2110. outf("%s\n",item.lname.get());
  2111. }
  2112. }
  2113. }
  2114. outf("TOTAL ORPHANS: %d files %"I64F"d bytes\n",totalNumOrphans,totalSizeOrphans);
  2115. outf("Row Compression: %d files %"I64F"d compressed %"I64F"d uncompressed\n",totalCompressed,totalCompressedSize, totalUncompressedSize);
  2116. outfileio.clear();
  2117. }
  2118. void outputCsvReport(const char *filename)
  2119. {
  2120. // more needs doing to make 'proper' csv
  2121. Owned<IFile> file = createIFile(filename);
  2122. Owned<IFileIO> fileio = file->open(IFOcreate);
  2123. if (!fileio) {
  2124. printf("ERROR cannot create %s\n",filename);
  2125. return;
  2126. }
  2127. outfileio.setown(createIOStream(fileio));
  2128. outf("===============================================================\n");
  2129. outf("\n--------------------------------------------------------\nORPHANS PARTIAL:\n");
  2130. {
  2131. ForEachItemIn(i,orphanlist) {
  2132. COrphanEntry &item = orphanlist.item(i);
  2133. if ((!item.complete())&&!item.misplaced) {
  2134. outf("%s,%d,%"I64F"d,%s\n",item.fname.get(),item.num(),item.size,item.modified.get());
  2135. }
  2136. }
  2137. }
  2138. outf("\n--------------------------------------------------------\nORPHANS PARTIAL FILES:\n");
  2139. {
  2140. ForEachItemIn(i,orphanlist) {
  2141. COrphanEntry &item = orphanlist.item(i);
  2142. if (!item.complete()) {
  2143. SocketEndpoint nullep;
  2144. ForEachItemIn(j1,item.primaryepa) {
  2145. SocketEndpoint ep=item.primaryepa.item(j1);
  2146. unsigned pn=item.primarypna.item(j1);
  2147. if (!ep.equals(nullep)) {
  2148. StringBuffer partname;
  2149. constructPartname(item.fname.get(),pn+1,partname,false);
  2150. RemoteFilename rfn;
  2151. rfn.setPath(ep,partname.str());
  2152. rfn.getRemotePath(partname.clear());
  2153. outf("%s,%s\n",partname.str(),item.misplaced?item.misplaced->lname.get():"");
  2154. }
  2155. }
  2156. ForEachItemIn(j2,item.replicateepa) {
  2157. SocketEndpoint ep=item.replicateepa.item(j2);
  2158. unsigned pn=item.replicatepna.item(j2);
  2159. if (!ep.equals(nullep)) {
  2160. StringBuffer partname;
  2161. constructPartname(item.fname.get(),pn+1,partname,true);
  2162. RemoteFilename rfn;
  2163. rfn.setPath(ep,partname.str());
  2164. rfn.getRemotePath(partname.clear());
  2165. outf("%s,%s\n",partname.str(),item.misplaced?item.misplaced->lname.get():"");
  2166. }
  2167. }
  2168. }
  2169. }
  2170. }
  2171. outf("\n--------------------------------------------------------\nORPHANS COMPLETE:\n");
  2172. {
  2173. ForEachItemIn(i,orphanlist) {
  2174. COrphanEntry &item = orphanlist.item(i);
  2175. if (item.complete()&&!item.isSingleton()) {
  2176. outf("%s,%d,%"I64F"d,%s %s\n",item.fname.get(),item.num(),item.size,item.modified.get(),item.misplaced?item.misplaced->lname.get():"");
  2177. }
  2178. }
  2179. }
  2180. outf("\n--------------------------------------------------------\nINCOMPLETE FILES:\n");
  2181. {
  2182. ForEachItemIn(i,logicalnamelist) {
  2183. CLogicalNameEntry &item = logicalnamelist.item(i);
  2184. item.done = false;
  2185. bool incomplete = false;
  2186. unsigned j0a;
  2187. for (j0a=0;j0a<item.max;j0a++)
  2188. if (!item.primaryresolved[j0a]&&!item.replicateresolved[j0a])
  2189. incomplete = true;
  2190. if (incomplete&&(!item.outsidedir)&&(item.outsidenodes.ordinality()==0)&&(item.primarynum!=item.max)) {
  2191. outf("%s,%d,%d,%d,%"I64F"d\n",item.lname.get(),item.primarynum,item.replicatenum,item.max,item.totalsize);
  2192. item.done = true;
  2193. }
  2194. }
  2195. }
  2196. outf("\n--------------------------------------------------------\nPRIMARY MISSING:\n");
  2197. {
  2198. ForEachItemIn(i,logicalnamelist) {
  2199. CLogicalNameEntry &item = logicalnamelist.item(i);
  2200. if (!item.outsidedir&&(item.outsidenodes.ordinality()==0)&&(item.primarynum!=item.max)) {
  2201. outf("%s,%d,%d,%"I64F"d\n",item.lname.get(),item.primarynum,item.max,item.totalsize);
  2202. item.done = true;
  2203. }
  2204. }
  2205. }
  2206. outf("\n--------------------------------------------------------\nSECONDARY MISSING:\n");
  2207. {
  2208. ForEachItemIn(i,logicalnamelist) {
  2209. CLogicalNameEntry &item = logicalnamelist.item(i);
  2210. if (!item.outsidedir&&(item.outsidenodes.ordinality()==0)&&(item.replicatenum!=item.max)) {
  2211. outf("%s,%d,%d,%"I64F"d\n",item.lname.get(),item.replicatenum,item.max,item.totalsize);
  2212. item.done = true;
  2213. }
  2214. }
  2215. }
  2216. outfileio.clear();
  2217. }
  2218. void outputBackupReport()
  2219. {
  2220. ForEachItemIn(i,logicalnamelist) {
  2221. CLogicalNameEntry &item = logicalnamelist.item(i);
  2222. if ((!item.outsidedir)&&(item.outsidenodes.ordinality()==0)) {
  2223. if ((item.primarynum!=item.max)||(item.replicatenum!=item.max)) {
  2224. StringBuffer cdrv;
  2225. StringBuffer ddrv;
  2226. StringBuffer missing;
  2227. unsigned firstc=0;
  2228. unsigned firstd=0;
  2229. unsigned firstm=0;
  2230. unsigned lastc=0;
  2231. unsigned lastd=0;
  2232. unsigned lastm=0;
  2233. unsigned j;
  2234. for (j=0;j<item.max;j++) {
  2235. if (!item.primaryresolved[j]) {
  2236. if (!item.replicateresolved[j]) {
  2237. addn(missing,j+1,firstm,lastm);
  2238. }
  2239. else {
  2240. addn(cdrv,j+1,firstc,lastc);
  2241. }
  2242. }
  2243. else if (!item.replicateresolved[j]) {
  2244. addn(ddrv,j+1,firstd,lastd);
  2245. }
  2246. }
  2247. addn(cdrv,0,firstc,lastc);
  2248. addn(ddrv,0,firstd,lastd);
  2249. addn(missing,0,firstm,lastm);
  2250. printf("%s\n",item.lname.get());
  2251. if (cdrv.length())
  2252. printf(" C: missing part%s %s\n",plural(cdrv.str()),cdrv.str());
  2253. if (ddrv.length())
  2254. printf(" D: missing part%s %s\n",plural(ddrv.str()),ddrv.str());
  2255. if (missing.length())
  2256. printf(" C: and D: missing part%s %s\n",plural(missing.str()),missing.str());
  2257. printf("\n");
  2258. }
  2259. }
  2260. }
  2261. }
  2262. static int compareDirectory(CInterface **le, CInterface **re)
  2263. {
  2264. const CDfuDirEntry *l = (const CDfuDirEntry *)*le;
  2265. const CDfuDirEntry *r = (const CDfuDirEntry *)*re;
  2266. __int64 dif = l->size-r->size;
  2267. if (dif<0)
  2268. return 1;
  2269. if (dif>0)
  2270. return -1;
  2271. return stricmp(l->name.get(),r->name.get());
  2272. }
  2273. IPropertyTree * outputTree()
  2274. {
  2275. log("Collating output");
  2276. IPropertyTree *out = createPTree("XREF");
  2277. IPropertyTree *orphans = addBranch(out,"Orphans");
  2278. IPropertyTree *found = addBranch(out,"Found");
  2279. IPropertyTree *lost = addBranch(out,"Lost");
  2280. IPropertyTree *message = addBranch(out,"Messages");
  2281. IPropertyTree *directories = addBranch(out,"Directories");
  2282. // Lost
  2283. {
  2284. DBGLOG("// Lost");
  2285. ForEachItemIn(i,logicalnamelist)
  2286. {
  2287. CLogicalNameEntry &item = logicalnamelist.item(i);
  2288. if ((!item.outsidedir)&&(item.outsidenodes.ordinality()==0)) {
  2289. if (item.incomplete()) { // need check for if replicated here
  2290. item.addFileBranch(lost,CFBname|CFBcluster|CFBsize|CFBnumparts|CFBpartslost|CFBprimarylost|CFBreplicatedlost|CFBmodified|CFBpartnode|CFBpartnum|CFBpartprimary|CFBpartreplicate);
  2291. item.done = true;
  2292. }
  2293. }
  2294. }
  2295. }
  2296. // Found
  2297. {
  2298. DBGLOG("// Found");
  2299. ForEachItemIn(i,orphanlist) {
  2300. COrphanEntry &item = orphanlist.item(i);
  2301. if (item.complete()) {
  2302. item.addFileBranch(found,CFBpartnode|CFBpartnum|CFBpartreplicate|CFBpartprimary|CFBnumparts|CFBcluster|CFBmodified|CFBsize);
  2303. }
  2304. }
  2305. }
  2306. // Orphans
  2307. {
  2308. DBGLOG("// Orphans");
  2309. ForEachItemIn(i,orphanlist) {
  2310. COrphanEntry &item = orphanlist.item(i);
  2311. if (!item.complete()) {
  2312. item.addFileBranch(orphans,CFBpartnode|CFBpartnum|CFBpartprimary|CFBnumparts|CFBmodified|CFBsize|CFBpartnode|CFBpartnum);
  2313. item.addFileBranch(orphans,CFBpartnode|CFBpartnum|CFBpartreplicate|CFBnumparts|CFBmodified|CFBsize|CFBpartnode|CFBpartnum);
  2314. }
  2315. }
  2316. }
  2317. // Messages
  2318. {
  2319. DBGLOG("// Messages");
  2320. {
  2321. ForEachItemIn(i,errors) {
  2322. cMessage &item = errors.item(i);
  2323. IPropertyTree *t = addBranch(message,"Error");
  2324. t->addProp("File",item.lname.get());
  2325. t->addProp("Text",item.msg.get());
  2326. }
  2327. }
  2328. {
  2329. ForEachItemIn(i,warnings) {
  2330. cMessage &item = warnings.item(i);
  2331. IPropertyTree *t = addBranch(message,"Warning");
  2332. t->addProp("File",item.lname.get());
  2333. t->addProp("Text",item.msg.get());
  2334. }
  2335. }
  2336. }
  2337. // Directories
  2338. {
  2339. dirlist.sort(compareDirectory);
  2340. DBGLOG("// Directories");
  2341. {
  2342. ForEachItemIn(i,dirlist) {
  2343. CDfuDirEntry &item = dirlist.item(i);
  2344. if (item.minsize<0)
  2345. item.minsize = 0;
  2346. if ((item.num==0)&&(item.size==0)&&(item.numdir>0)) // exclude intermediate empty dirs
  2347. continue;
  2348. IPropertyTree *t = addBranch(directories,"Directory");
  2349. t->addProp("Name",item.name.get());
  2350. t->addPropInt("Num",item.num);
  2351. t->addPropInt64("Size",item.size);
  2352. if (item.size) {
  2353. t->addPropInt64("MaxSize",item.maxsize);
  2354. StringBuffer s1;
  2355. if (!item.maxip.isNull())
  2356. item.maxip.getIpText(s1);
  2357. t->addProp("MaxIP",s1.str());
  2358. t->addPropInt64("MinSize",item.minsize);
  2359. s1.clear();
  2360. if (!item.minip.isNull())
  2361. item.minip.getIpText(s1);
  2362. t->addProp("MinIP",s1.str());
  2363. item.getskew(s1.clear());
  2364. if (s1.length())
  2365. t->addProp("Skew",s1.str());
  2366. }
  2367. }
  2368. }
  2369. }
  2370. return out;
  2371. }
  2372. public:
  2373. IPropertyTree *process(unsigned nclusters,const char **clusters,unsigned numdirs,const char **dirbaselist,unsigned flags,IXRefProgressCallback *_msgcallback,unsigned numthreads)
  2374. {
  2375. CriticalBlock block(xrefsect);
  2376. msgcallback.set(_msgcallback);
  2377. IPropertyTree *out=NULL;
  2378. Owned<IGroup> g;
  2379. unsigned j;
  2380. if (!nclusters) {
  2381. error("XREF","No clusters specified\n");
  2382. return NULL;
  2383. }
  2384. if (!numdirs) {
  2385. error("XREF","No directories specified\n");
  2386. return NULL;
  2387. }
  2388. for (j=0;j<nclusters;j++) {
  2389. Owned<IGroup> gsub = queryNamedGroupStore().lookup(clusters[j]);
  2390. if (!gsub) {
  2391. error(clusters[j],"Could not find cluster group");
  2392. return NULL;
  2393. }
  2394. if (!g)
  2395. g.set(gsub.get());
  2396. else
  2397. g.setown(g->combine(gsub.get()));
  2398. }
  2399. totalSizeOrphans =0;
  2400. totalNumOrphans = 0;
  2401. logicalnamelist.kill();
  2402. dirlist.kill();
  2403. orphanlist.kill();
  2404. const char* cluster = clusters[0];
  2405. loadFromDFS(*this,g,numdirs,dirbaselist,cluster);
  2406. xrefRemoteDirectories(g,numdirs,dirbaselist,numthreads);
  2407. StringBuffer filename;
  2408. filename.clear().append("xrefrpt");
  2409. addFileTimestamp(filename, true);
  2410. filename.append(".txt");
  2411. if (flags&PMtextoutput)
  2412. outputTextReport(filename.str());
  2413. filename.clear().append("xrefrpt");
  2414. addFileTimestamp(filename, true);
  2415. filename.append(".txt");
  2416. if (flags&PMcsvoutput)
  2417. outputCsvReport(filename.str());
  2418. if (flags&PMbackupoutput)
  2419. outputBackupReport();
  2420. if (flags&PMtreeoutput)
  2421. out = outputTree();
  2422. logicalnamemap.kill();
  2423. filemap.kill();
  2424. orphanmap.kill();
  2425. dirmap.kill();
  2426. logicaldirmap.kill();
  2427. log("Complete");
  2428. DBGLOG("Finished...");
  2429. return out;
  2430. }
  2431. };
  2432. IPropertyTree * runXRef(unsigned nclusters,const char **clusters,IXRefProgressCallback *callback,unsigned numthreads)
  2433. {
  2434. if (nclusters==0)
  2435. return NULL;
  2436. CXRefManager xrefmanager;
  2437. const char *dirs[2];
  2438. unsigned numdirs;
  2439. #ifdef _WIN32
  2440. bool islinux = false;
  2441. #else
  2442. bool islinux = true;
  2443. #endif
  2444. // assume all nodes same OS
  2445. Owned<IGroup> group = queryNamedGroupStore().lookup(clusters[0]);
  2446. if (group)
  2447. islinux = queryOS(group->queryNode(0).endpoint())==MachineOsLinux;
  2448. dirs[0] = queryBaseDirectory(false,islinux?DFD_OSunix:DFD_OSwindows);
  2449. dirs[1] = queryBaseDirectory(true,islinux?DFD_OSunix:DFD_OSwindows);
  2450. numdirs = 2;
  2451. IPropertyTree *ret=NULL;
  2452. try {
  2453. ret = xrefmanager.process(nclusters,clusters,numdirs,dirs,PMtreeoutput,callback,numthreads);
  2454. }
  2455. catch (IException *e) {
  2456. StringBuffer s;
  2457. e->errorMessage(s);
  2458. if (callback)
  2459. callback->error(s.str());
  2460. else
  2461. ERRLOG("%s",s.str());
  2462. }
  2463. return ret;
  2464. }
  2465. IPropertyTree * runXRefCluster(const char *cluster,IXRefNode *nodeToUpdate)
  2466. {
  2467. DBGLOG("runXRefCluster starting for cluster %s",cluster);
  2468. CXRefManager xrefmanager;
  2469. IPropertyTree *ret=NULL;
  2470. IXRefProgressCallback* callback = dynamic_cast<IXRefProgressCallback*>(nodeToUpdate);
  2471. try {
  2472. const char *clusters[1];
  2473. clusters[0] = cluster;
  2474. ret = runXRef(1, clusters,callback,4); // only single thread for time being
  2475. // xrefmanager.process(1,clusters,4,dirs,PMtreeoutput,callback);
  2476. }
  2477. catch (IException *e) {
  2478. StringBuffer s;
  2479. e->errorMessage(s);
  2480. if (callback)
  2481. callback->error(s.str());
  2482. else
  2483. ERRLOG("%s",s.str());
  2484. }
  2485. if(ret)
  2486. {
  2487. DBGLOG("runXRefCluster building DFU node for cluster %s",cluster);
  2488. nodeToUpdate->BuildXRefData(*ret,cluster);
  2489. nodeToUpdate->commit();
  2490. }
  2491. return ret;
  2492. }
  2493. IPropertyTree * RunProcess(unsigned nclusters,const char **clusters,unsigned numdirs,const char **dirbaselist,unsigned flags,IXRefProgressCallback *_msgcallback,unsigned numthreads)
  2494. {
  2495. //Provide a wrapper for the command line
  2496. if (flags & PMupdateeclwatch) {
  2497. if (nclusters==1) {
  2498. const char *cluster = *clusters;
  2499. CXRefNodeManager nodemanager;
  2500. Owned<IPropertyTree> tree = runXRef(nclusters,clusters,NULL,numthreads);
  2501. if (tree) {
  2502. Owned<IXRefNode> xRefNode = nodemanager.getXRefNode(cluster);
  2503. if (!xRefNode.get())
  2504. xRefNode.setown( nodemanager.CreateXRefNode(cluster));
  2505. xRefNode->setCluster(cluster);
  2506. xRefNode->BuildXRefData(*tree.get(),cluster);
  2507. xRefNode->commit();
  2508. }
  2509. }
  2510. else {
  2511. // do clusters 1 at time
  2512. for (unsigned i = 0; i<nclusters; i++)
  2513. RunProcess(1,clusters+i,numdirs,dirbaselist,flags,_msgcallback,numthreads);
  2514. }
  2515. return NULL;
  2516. }
  2517. CXRefManager xrefmanager;
  2518. return xrefmanager.process(nclusters,clusters,numdirs,dirbaselist,flags,_msgcallback,numthreads);
  2519. }