Src/DirScan.cpp

   1 /**
   2  *  @file DirScan.cpp
   3  *
   4  *  @brief Implementation of DirScan (q.v.) and helper functions
   5  */
   6
   7 #include "pch.h"
   8 #include "DirScan.h"
   9 #include <cassert>
  10 #include <memory>
  11 #define POCO_NO_UNWINDOWS 1
  12 #include <Poco/Semaphore.h>
  13 #include <Poco/Notification.h>
  14 #include <Poco/NotificationQueue.h>
  15 #include <Poco/Environment.h>
  16 #include <Poco/ThreadPool.h>
  17 #include <Poco/Thread.h>
  18 #include <Poco/Runnable.h>
  19 #include <Poco/Mutex.h>
  20 #include <Poco/AutoPtr.h>
  21 #include <Poco/Stopwatch.h>
  22 #include <Poco/Format.h>
  23 #include "DiffThread.h"
  24 #include "UnicodeString.h"
  25 #include "DiffWrapper.h"
  26 #include "CompareStats.h"
  27 #include "FolderCmp.h"
  28 #include "FileFilterHelper.h"
  29 #include "IAbortable.h"
  30 #include "DirItem.h"
  31 #include "DirTravel.h"
  32 #include "paths.h"
  33 #include "Plugins.h"
  34 #include "MergeAppCOMClass.h"
  35 #include "MergeApp.h"
  36 #include "OptionsDef.h"
  37 #include "OptionsMgr.h"
  38 #include "PathContext.h"
  39 #include "DebugNew.h"
  40
  41 using Poco::NotificationQueue;
  42 using Poco::Notification;
  43 using Poco::AutoPtr;
  44 using Poco::Thread;
  45 using Poco::ThreadPool;
  46 using Poco::Runnable;
  47 using Poco::Environment;
  48 using Poco::Stopwatch;
  49
  50 // Static functions (ie, functions only used locally)
  51 static void CompareDiffItem(FolderCmp &fc, DIFFITEM &di);
  52 static DIFFITEM *AddToList(const String &sLeftDir, const String &sRightDir, const DirItem *lent, const DirItem *rent,
  53         unsigned code, DiffFuncStruct *myStruct, DIFFITEM *parent);
  54 static DIFFITEM *AddToList(const String &sDir1, const String &sDir2, const String &sDir3, const DirItem *ent1, const DirItem *ent2, const DirItem *ent3,
  55         unsigned code, DiffFuncStruct *myStruct, DIFFITEM *parent, int nItems = 3);
  56 static void UpdateDiffItem(DIFFITEM &di, bool &bExists, CDiffContext *pCtxt);
  57 static int CompareItems(NotificationQueue &queue, DiffFuncStruct *myStruct, DIFFITEM *parentdiffpos);
  58 static unsigned GetDirCompareFlags3Way(const DIFFITEM& di);
  59
  60 class WorkNotification: public Poco::Notification
  61 {
  62 public:
  63         WorkNotification(DIFFITEM& di, NotificationQueue& queueResult): m_di(di), m_queueResult(queueResult) {}
  64         DIFFITEM& data() const { return m_di; }
  65         NotificationQueue& queueResult() const { return m_queueResult; }
  66 private:
  67         DIFFITEM& m_di;
  68         NotificationQueue& m_queueResult;
  69 };
  70
  71 class WorkCompletedNotification: public Poco::Notification
  72 {
  73 public:
  74         explicit WorkCompletedNotification(DIFFITEM& di): m_di(di) {}
  75         DIFFITEM& data() const { return m_di; }
  76 private:
  77         DIFFITEM& m_di;
  78 };
  79
  80 class DiffWorker: public Runnable
  81 {
  82 public:
  83         DiffWorker(NotificationQueue& queue, CDiffContext *pCtxt, int id):
  84           m_queue(queue), m_pCtxt(pCtxt), m_id(id) {}
  85
  86         void run()
  87         {
  88                 FolderCmp fc(m_pCtxt);
  89                 // keep the scripts alive during the Rescan
  90                 // when we exit the thread, we delete this and release the scripts
  91                 CAssureScriptsForThread scriptsForRescan(new MergeAppCOMClass());
  92
  93                 AutoPtr<Notification> pNf(m_queue.waitDequeueNotification());
  94                 while (pNf.get() != nullptr)
  95                 {
  96                         WorkNotification* pWorkNf = dynamic_cast<WorkNotification*>(pNf.get());
  97                         if (pWorkNf != nullptr) {
  98                                 m_pCtxt->m_pCompareStats->BeginCompare(&pWorkNf->data(), m_id);
  99                                 if (!m_pCtxt->ShouldAbort())
 100                                         CompareDiffItem(fc, pWorkNf->data());
 101                                 pWorkNf->queueResult().enqueueNotification(new WorkCompletedNotification(pWorkNf->data()));
 102                         }
 103                         if (m_pCtxt->m_pCompareStats->IsIdleCompareThread(m_id))
 104                         {
 105                                 m_pCtxt->m_pCompareStats->BeginCompare(nullptr, m_id);
 106                                 while (!m_pCtxt->ShouldAbort() && m_pCtxt->m_pCompareStats->IsIdleCompareThread(m_id))
 107                                         Poco::Thread::sleep(10);
 108                         }
 109
 110                         pNf = m_queue.waitDequeueNotification();
 111                 }
 112         }
 113
 114 private:
 115         NotificationQueue& m_queue;
 116         CDiffContext *m_pCtxt;
 117         int m_id;
 118 };
 119
 120 typedef std::shared_ptr<DiffWorker> DiffWorkerPtr;
 121
 122 /**
 123  * @brief Collect file- and folder-names to list.
 124  * This function walks given folders and adds found subfolders and files into
 125  * lists. There are two modes, determined by the @p depth:
 126  * - in non-recursive mode we walk only given folders, and add files
 127  *   contained. Subfolders are added as folder items, not walked into.
 128  * - in recursive mode we walk all subfolders and add the files they
 129  *   contain into list.
 130  *
 131  * Items are tested against file filters in this function.
 132  *
 133  * @param [in] paths Root paths of compare
 134  * @param [in] leftsubdir Left side subdirectory under root path
 135  * @param [in] bLeftUniq Is left-side folder unique folder?
 136  * @param [in] rightsubdir Right side subdirectory under root path
 137  * @param [in] bRightUniq Is right-side folder unique folder?
 138  * @param [in] myStruct Compare-related data, like context etc.
 139  * @param [in] casesensitive Is filename compare case sensitive?
 140  * @param [in] depth Levels of subdirectories to scan, -1 scans all
 141  * @param [in] parent Folder diff item to be scanned
 142  * @param [in] bUniques If true, walk into unique folders.
 143  * @return 1 normally, -1 if compare was aborted
 144  */
 145 int DirScan_GetItems(const PathContext &paths, const String subdir[],
 146                 DiffFuncStruct *myStruct,
 147                 bool casesensitive, int depth, DIFFITEM *parent,
 148                 bool bUniques)
 149 {
 150         static const tchar_t backslash[] = _T("\\");
 151         int nDirs = paths.GetSize();
 152         CDiffContext *pCtxt = myStruct->context;
 153         String sDir[3];
 154         String subprefix[3];
 155
 156         std::copy(paths.begin(), paths.end(), sDir);
 157
 158         if (!subdir[0].empty())
 159         {
 160                 for (int nIndex = 0; nIndex < paths.GetSize(); nIndex++)
 161                 {
 162                         sDir[nIndex] = paths::ConcatPath(sDir[nIndex], subdir[nIndex]);
 163                         subprefix[nIndex] = subdir[nIndex] + backslash;
 164                 }
 165         }
 166
 167         DirItemArray dirs[3], aFiles[3];
 168         for (int nIndex = 0; nIndex < nDirs; nIndex++)
 169                 LoadAndSortFiles(sDir[nIndex], &dirs[nIndex], &aFiles[nIndex], casesensitive);
 170
 171         // Allow user to abort scanning
 172         if (pCtxt->ShouldAbort())
 173                 return -1;
 174
 175         // Handle directories
 176         // i points to current directory in left list (leftDirs)
 177         // j points to current directory in right list (rightDirs)
 178
 179         {
 180                 int nIndex;
 181                 for (nIndex = 0; nIndex < nDirs; nIndex++)
 182                         if (dirs[nIndex].size() != 0 || aFiles[nIndex].size() != 0) break;
 183                 if (nIndex == nDirs)
 184                         return 0;
 185         }
 186
 187         DirItemArray::size_type i=0, j=0, k=0;
 188         while (true)
 189         {
 190                 if (pCtxt->ShouldAbort())
 191                         return -1;
 192
 193                 if (i >= dirs[0].size() && j >= dirs[1].size() && (nDirs < 3 || k >= dirs[2].size()))
 194                         break;
 195
 196                 unsigned nDiffCode = DIFFCODE::DIR;
 197                 // Comparing directories leftDirs[i].name to rightDirs[j].name
 198                 if (i<dirs[0].size() && (j==dirs[1].size() || collstr(dirs[0][i].filename, dirs[1][j].filename, casesensitive)<0)
 199                         && (nDirs < 3 ||      (k==dirs[2].size() || collstr(dirs[0][i].filename, dirs[2][k].filename, casesensitive)<0) ))
 200                 {
 201                         nDiffCode |= DIFFCODE::FIRST;
 202                 }
 203                 else if (j<dirs[1].size() && (i==dirs[0].size() || collstr(dirs[1][j].filename, dirs[0][i].filename, casesensitive)<0)
 204                         && (nDirs < 3 ||      (k==dirs[2].size() || collstr(dirs[1][j].filename, dirs[2][k].filename, casesensitive)<0) ))
 205                 {
 206                         nDiffCode |= DIFFCODE::SECOND;
 207                 }
 208                 else if (nDirs < 3)
 209                 {
 210                         nDiffCode |= DIFFCODE::BOTH;
 211                 }
 212                 else
 213                 {
 214                         if (k<dirs[2].size() && (i==dirs[0].size() || collstr(dirs[2][k].filename, dirs[0][i].filename, casesensitive)<0)
 215                                 &&                     (j==dirs[1].size() || collstr(dirs[2][k].filename, dirs[1][j].filename, casesensitive)<0) )
 216                         {
 217                                 nDiffCode |= DIFFCODE::THIRD;
 218                         }
 219                         else if ((i<dirs[0].size() && j<dirs[1].size() && collstr(dirs[0][i].filename, dirs[1][j].filename, casesensitive) == 0)
 220                                 && (k==dirs[2].size() || collstr(dirs[2][k].filename, dirs[0][i].filename, casesensitive) != 0))
 221                         {
 222                                 nDiffCode |= DIFFCODE::FIRST | DIFFCODE::SECOND;
 223                         }
 224                         else if ((i<dirs[0].size() && k<dirs[2].size() && collstr(dirs[0][i].filename, dirs[2][k].filename, casesensitive) == 0)
 225                                 && (j==dirs[1].size() || collstr(dirs[1][j].filename, dirs[2][k].filename, casesensitive) != 0))
 226                         {
 227                                 nDiffCode |= DIFFCODE::FIRST | DIFFCODE::THIRD;
 228                         }
 229                         else if ((j<dirs[1].size() && k<dirs[2].size() && collstr(dirs[1][j].filename, dirs[2][k].filename, casesensitive) == 0)
 230                                 && (i==dirs[0].size() || collstr(dirs[0][i].filename, dirs[1][j].filename, casesensitive) != 0))
 231                         {
 232                                 nDiffCode |= DIFFCODE::SECOND | DIFFCODE::THIRD;
 233                         }
 234                         else
 235                         {
 236                                 nDiffCode |= DIFFCODE::ALL;
 237                         }
 238                 }
 239
 240                 String leftnewsub;
 241                 String rightnewsub;
 242                 String middlenewsub;
 243                 if (nDirs < 3)
 244                 {
 245                         leftnewsub  = (nDiffCode & DIFFCODE::FIRST)  ? subprefix[0] + dirs[0][i].filename.get() : subprefix[0] + dirs[1][j].filename.get();
 246                         rightnewsub = (nDiffCode & DIFFCODE::SECOND) ? subprefix[1] + dirs[1][j].filename.get() : subprefix[1] + dirs[0][i].filename.get();
 247
 248                         // Test against filter so we don't include contents of filtered out directories
 249                         // Also this is only place we can test for both-sides directories in recursive compare
 250                         if ((pCtxt->m_piFilterGlobal!=nullptr && !pCtxt->m_piFilterGlobal->includeDir(leftnewsub, rightnewsub)) ||
 251                                 (pCtxt->m_bIgnoreReparsePoints && (
 252                                 (nDiffCode & DIFFCODE::FIRST) && (dirs[0][i].flags.attributes & FILE_ATTRIBUTE_REPARSE_POINT) ||
 253                                         (nDiffCode & DIFFCODE::SECOND) && (dirs[1][j].flags.attributes & FILE_ATTRIBUTE_REPARSE_POINT))
 254                                         )
 255                                 )
 256                                 nDiffCode |= DIFFCODE::SKIPPED;
 257                 }
 258                 else
 259                 {
 260                         leftnewsub   = subprefix[0];
 261                         if (nDiffCode & DIFFCODE::FIRST)       leftnewsub += dirs[0][i].filename;
 262                         else if (nDiffCode & DIFFCODE::SECOND) leftnewsub += dirs[1][j].filename;
 263                         else if (nDiffCode & DIFFCODE::THIRD)  leftnewsub += dirs[2][k].filename;
 264                         middlenewsub = subprefix[1];
 265                         if (nDiffCode & DIFFCODE::SECOND)      middlenewsub += dirs[1][j].filename;
 266                         else if (nDiffCode & DIFFCODE::FIRST)  middlenewsub += dirs[0][i].filename;
 267                         else if (nDiffCode & DIFFCODE::THIRD)  middlenewsub += dirs[2][k].filename;
 268                         rightnewsub  = subprefix[2];
 269                         if (nDiffCode & DIFFCODE::THIRD)       rightnewsub += dirs[2][k].filename;
 270                         else if (nDiffCode & DIFFCODE::FIRST)  rightnewsub += dirs[0][i].filename;
 271                         else if (nDiffCode & DIFFCODE::SECOND) rightnewsub += dirs[1][j].filename;
 272
 273                         // Test against filter so we don't include contents of filtered out directories
 274                         // Also this is only place we can test for both-sides directories in recursive compare
 275                         if ((pCtxt->m_piFilterGlobal!=nullptr && !pCtxt->m_piFilterGlobal->includeDir(leftnewsub, middlenewsub, rightnewsub)) ||
 276                                 (pCtxt->m_bIgnoreReparsePoints && (
 277                                   (nDiffCode & DIFFCODE::FIRST)  && (dirs[0][i].flags.attributes & FILE_ATTRIBUTE_REPARSE_POINT) ||
 278                                   (nDiffCode & DIFFCODE::SECOND) && (dirs[1][j].flags.attributes & FILE_ATTRIBUTE_REPARSE_POINT) ||
 279                                   (nDiffCode & DIFFCODE::THIRD)  && (dirs[2][k].flags.attributes & FILE_ATTRIBUTE_REPARSE_POINT))
 280                                 )
 281                            )
 282                                 nDiffCode |= DIFFCODE::SKIPPED;
 283                 }
 284
 285                 // add to list
 286                 if (!depth)
 287                 {
 288                         if (nDirs < 3)
 289                                 AddToList(subdir[0], subdir[1],
 290                                         (nDiffCode & DIFFCODE::FIRST ) ? &dirs[0][i] : nullptr,
 291                                         (nDiffCode & DIFFCODE::SECOND) ? &dirs[1][j] : nullptr,
 292                                         nDiffCode, myStruct, parent);
 293                         else
 294                                 AddToList(subdir[0], subdir[1], subdir[2],
 295                                         (nDiffCode & DIFFCODE::FIRST ) ? &dirs[0][i] : nullptr,
 296                                         (nDiffCode & DIFFCODE::SECOND) ? &dirs[1][j] : nullptr,
 297                                         (nDiffCode & DIFFCODE::THIRD ) ? &dirs[2][k] : nullptr,
 298                                         nDiffCode, myStruct, parent);
 299                 }
 300                 else
 301                 {
 302                         // Recursive compare
 303                         assert(pCtxt->m_bRecursive);
 304                         if (nDirs < 3)
 305                         {
 306                                 DIFFITEM *me = AddToList(subdir[0], subdir[1],
 307                                         (nDiffCode & DIFFCODE::FIRST ) ? &dirs[0][i] : nullptr,
 308                                         (nDiffCode & DIFFCODE::SECOND) ? &dirs[1][j] : nullptr,
 309                                         nDiffCode, myStruct, parent);
 310                                 if ((nDiffCode & DIFFCODE::SKIPPED) == 0 && ((nDiffCode & DIFFCODE::SIDEFLAGS) == DIFFCODE::BOTH || bUniques))
 311                                 {
 312                                         // Scan recursively all subdirectories too, we are not adding folders
 313                                         String newsubdir[3] = {leftnewsub, rightnewsub};
 314                                         int result = DirScan_GetItems(paths, newsubdir, myStruct, casesensitive,
 315                                                         depth - 1, me, bUniques);
 316                                         if (result == -1)
 317                                                 return -1;
 318                                 }
 319                         }
 320                         else
 321                         {
 322                                 DIFFITEM *me = AddToList(subdir[0], subdir[1], subdir[2],
 323                                         (nDiffCode & DIFFCODE::FIRST ) ? &dirs[0][i] : nullptr,
 324                                         (nDiffCode & DIFFCODE::SECOND) ? &dirs[1][j] : nullptr,
 325                                         (nDiffCode & DIFFCODE::THIRD ) ? &dirs[2][k] : nullptr,
 326                                         nDiffCode, myStruct, parent);
 327                                 if ((nDiffCode & DIFFCODE::SKIPPED) == 0 && ((nDiffCode & DIFFCODE::SIDEFLAGS) == DIFFCODE::ALL || bUniques))
 328                                 {
 329                                         // Scan recursively all subdirectories too, we are not adding folders
 330                                         String newsubdir[3] = {leftnewsub, middlenewsub, rightnewsub};
 331                                         int result = DirScan_GetItems(paths, newsubdir, myStruct, casesensitive,
 332                                                         depth - 1, me, bUniques);
 333                                         if (result == -1)
 334                                                 return -1;
 335                                 }
 336                         }
 337                 }
 338                 if (nDiffCode & DIFFCODE::FIRST)
 339                         i++;
 340                 if (nDiffCode & DIFFCODE::SECOND)
 341                         j++;
 342                 if (nDiffCode & DIFFCODE::THIRD)
 343                         k++;
 344         }
 345         // Handle files
 346         // i points to current file in left list (aFiles[0])
 347         // j points to current file in right list (aFiles[1])
 348         i=0, j=0, k=0;
 349         while (true)
 350         {
 351                 if (pCtxt->ShouldAbort())
 352                         return -1;
 353
 354
 355                 // Comparing file aFiles[0][i].name to aFiles[1][j].name
 356                 if (i<aFiles[0].size() && (j==aFiles[1].size() ||
 357                                 collstr(aFiles[0][i].filename, aFiles[1][j].filename, casesensitive) < 0)
 358                         && (nDirs < 3 ||
 359                                 (k==aFiles[2].size() || collstr(aFiles[0][i].filename, aFiles[2][k].filename, casesensitive)<0) ))
 360                 {
 361                         if (nDirs < 3)
 362                         {
 363                                 const unsigned nDiffCode = DIFFCODE::FIRST | DIFFCODE::FILE;
 364                                 AddToList(subdir[0], subdir[1], &aFiles[0][i], nullptr, nDiffCode, myStruct, parent);
 365                         }
 366                         else
 367                         {
 368                                 const unsigned nDiffCode = DIFFCODE::FIRST | DIFFCODE::FILE;
 369                                 AddToList(subdir[0], subdir[1], subdir[2], &aFiles[0][i], nullptr, nullptr, nDiffCode, myStruct, parent);
 370                         }
 371                         // Advance left pointer over left-only entry, and then retest with new pointers
 372                         ++i;
 373                         continue;
 374                 }
 375                 if (j<aFiles[1].size() && (i==aFiles[0].size() ||
 376                                 collstr(aFiles[0][i].filename, aFiles[1][j].filename, casesensitive) > 0)
 377                         && (nDirs < 3 ||
 378                                 (k==aFiles[2].size() || collstr(aFiles[1][j].filename, aFiles[2][k].filename, casesensitive)<0) ))
 379                 {
 380                         const unsigned nDiffCode = DIFFCODE::SECOND | DIFFCODE::FILE;
 381                         if (nDirs < 3)
 382                                 AddToList(subdir[0], subdir[1], nullptr, &aFiles[1][j], nDiffCode, myStruct, parent);
 383                         else
 384                                 AddToList(subdir[0], subdir[1], subdir[2], nullptr, &aFiles[1][j], nullptr, nDiffCode, myStruct, parent);
 385                         // Advance right pointer over right-only entry, and then retest with new pointers
 386                         ++j;
 387                         continue;
 388                 }
 389                 if (nDirs == 3)
 390                 {
 391                         if (k<aFiles[2].size() && (i==aFiles[0].size() ||
 392                                         collstr(aFiles[2][k].filename, aFiles[0][i].filename, casesensitive)<0)
 393                                 && (j==aFiles[1].size() || collstr(aFiles[2][k].filename, aFiles[1][j].filename, casesensitive)<0) )
 394                         {
 395                                 const unsigned nDiffCode = DIFFCODE::THIRD | DIFFCODE::FILE;
 396                                 AddToList(subdir[0], subdir[1], subdir[2], nullptr, nullptr, &aFiles[2][k], nDiffCode, myStruct, parent);
 397                                 ++k;
 398                                 // Advance right pointer over right-only entry, and then retest with new pointers
 399                                 continue;
 400                         }
 401
 402                         if ((i<aFiles[0].size() && j<aFiles[1].size() && collstr(aFiles[0][i].filename, aFiles[1][j].filename, casesensitive) == 0)
 403                             && (k==aFiles[2].size() || collstr(aFiles[0][i].filename, aFiles[2][k].filename, casesensitive) != 0))
 404                         {
 405                                 const unsigned nDiffCode = DIFFCODE::FIRST | DIFFCODE::SECOND | DIFFCODE::FILE;
 406                                 AddToList(subdir[0], subdir[1], subdir[2], &aFiles[0][i], &aFiles[1][j], nullptr, nDiffCode, myStruct, parent);
 407                                 ++i;
 408                                 ++j;
 409                                 continue;
 410                         }
 411                         else if ((i<aFiles[0].size() && k<aFiles[2].size() && collstr(aFiles[0][i].filename, aFiles[2][k].filename, casesensitive) == 0)
 412                             && (j==aFiles[1].size() || collstr(aFiles[1][j].filename, aFiles[2][k].filename, casesensitive) != 0))
 413                         {
 414                                 const unsigned nDiffCode = DIFFCODE::FIRST | DIFFCODE::THIRD | DIFFCODE::FILE;
 415                                 AddToList(subdir[0], subdir[1], subdir[2], &aFiles[0][i], nullptr, &aFiles[2][k], nDiffCode, myStruct, parent);
 416                                 ++i;
 417                                 ++k;
 418                                 continue;
 419                         }
 420                         else if ((j<aFiles[1].size() && k<aFiles[2].size() && collstr(aFiles[1][j].filename, aFiles[2][k].filename, casesensitive) == 0)
 421                             && (i==aFiles[0].size() || collstr(aFiles[0][i].filename, aFiles[1][j].filename, casesensitive) != 0))
 422                         {
 423                                 const unsigned nDiffCode = DIFFCODE::SECOND | DIFFCODE::THIRD | DIFFCODE::FILE;
 424                                 AddToList(subdir[0], subdir[1], subdir[2], nullptr, &aFiles[1][j], &aFiles[2][k], nDiffCode, myStruct, parent);
 425                                 ++j;
 426                                 ++k;
 427                                 continue;
 428                         }
 429                 }
 430                 if (i<aFiles[0].size())
 431                 {
 432                         if (nDirs < 3)
 433                         {
 434                                 assert(j<aFiles[1].size());
 435                                 const unsigned nDiffCode = DIFFCODE::BOTH | DIFFCODE::FILE;
 436                                 AddToList(subdir[0], subdir[1], &aFiles[0][i], &aFiles[1][j], nDiffCode, myStruct, parent);
 437                                 ++i;
 438                                 ++j;
 439                                 continue;
 440                         }
 441                         else
 442                         {
 443                                 assert(j<aFiles[1].size());
 444                                 assert(k<aFiles[2].size());
 445                                 const unsigned nDiffCode = DIFFCODE::ALL | DIFFCODE::FILE;
 446                                 AddToList(subdir[0], subdir[1], subdir[2], &aFiles[0][i], &aFiles[1][j], &aFiles[2][k], nDiffCode, myStruct, parent);
 447                                 ++i;
 448                                 ++j;
 449                                 ++k;
 450                                 continue;
 451                         }
 452                 }
 453                 break;
 454         }
 455
 456         if (parent != nullptr)
 457         {
 458                 for (int nIndex = 0; nIndex < nDirs; ++nIndex)
 459                         if (parent->diffcode.exists(nIndex) && parent->diffFileInfo[nIndex].size == DirItem::FILE_SIZE_NONE)
 460                                 parent->diffFileInfo[nIndex].size = 0;
 461
 462                 DIFFITEM *dic = parent->GetFirstChild();
 463                 while (dic)
 464                 {
 465                         for (int nIndex = 0; nIndex < nDirs; ++nIndex)
 466                         {
 467                                 if (dic->diffFileInfo[nIndex].size != DirItem::FILE_SIZE_NONE)
 468                                         parent->diffFileInfo[nIndex].size += dic->diffFileInfo[nIndex].size;
 469                         }
 470                         dic = dic->GetFwdSiblingLink();
 471                 }
 472         }
 473
 474         return 1;
 475 }
 476
 477 /**
 478  * @brief Compare DiffItems in list and add results to compare context.
 479  *
 480  * @param myStruct [in] A structure containing compare-related data.
 481  * @param parentdiffpos [in] Position of parent diff item
 482  * @return >= 0 number of diff items, -1 if compare was aborted
 483  */
 484 int DirScan_CompareItems(DiffFuncStruct *myStruct, DIFFITEM *parentdiffpos)
 485 {
 486         const int compareMethod = myStruct->context->GetCompareMethod();
 487         int nworkers = 1;
 488
 489         if (compareMethod == CMP_CONTENT || compareMethod == CMP_QUICK_CONTENT)
 490         {
 491                 nworkers = GetOptionsMgr()->GetInt(OPT_CMP_COMPARE_THREADS);
 492                 if (nworkers <= 0)
 493                         nworkers += Environment::processorCount();
 494                 nworkers = std::clamp(nworkers, 1, static_cast<int>(Environment::processorCount()));
 495         }
 496
 497         ThreadPool threadPool(nworkers, nworkers);
 498         std::vector<DiffWorkerPtr> workers;
 499         NotificationQueue queue;
 500         myStruct->context->m_pCompareStats->SetCompareThreadCount(nworkers);
 501         for (int i = 0; i < nworkers; ++i)
 502         {
 503                 workers.emplace_back(std::make_shared<DiffWorker>(queue, myStruct->context, i));
 504                 threadPool.start(*workers[i]);
 505         }
 506
 507         int res = CompareItems(queue, myStruct, parentdiffpos);
 508
 509         myStruct->context->m_pCompareStats->SetIdleCompareThreadCount(0);
 510         Thread::sleep(100);
 511         queue.wakeUpAll();
 512         threadPool.joinAll();
 513
 514         return res;
 515 }
 516
 517 static int CompareItems(NotificationQueue& queue, DiffFuncStruct *myStruct, DIFFITEM *parentdiffpos)
 518 {
 519         NotificationQueue queueResult;
 520         Stopwatch stopwatch;
 521         CDiffContext *pCtxt = myStruct->context;
 522         int res = 0;
 523         int count = 0;
 524         bool bCompareFailure = false;
 525         if (parentdiffpos == nullptr)
 526                 myStruct->pSemaphore->wait();
 527         stopwatch.start();
 528         int nDirs = pCtxt->GetCompareDirs();
 529         DIFFITEM *pos = pCtxt->GetFirstChildDiffPosition(parentdiffpos);
 530         while (pos != nullptr)
 531         {
 532                 if (pCtxt->ShouldAbort())
 533                         break;
 534
 535                 if (stopwatch.elapsed() > 2000000)
 536                 {
 537                         int event = CDiffThread::EVENT_COMPARE_PROGRESSED;
 538                         myStruct->m_listeners.notify(myStruct, event);
 539                         stopwatch.restart();
 540                 }
 541                 myStruct->pSemaphore->wait();
 542                 DIFFITEM *curpos = pos;
 543                 DIFFITEM &di = pCtxt->GetNextSiblingDiffRefPosition(pos);
 544                 bool existsalldirs = di.diffcode.existAll();
 545                 if (di.diffcode.isDirectory() && pCtxt->m_bRecursive)
 546                 {
 547                         if ((di.diffcode.diffcode & DIFFCODE::CMPERR) != DIFFCODE::CMPERR)
 548                         {       // Only clear DIFF|SAME flags if not CMPERR (eg. both flags together)
 549                                 di.diffcode.diffcode &= ~(DIFFCODE::DIFF | DIFFCODE::SAME);
 550                         }
 551                         int ndiff = CompareItems(queue, myStruct, curpos);
 552                         // Propagate sub-directory status to this directory
 553                         if (ndiff > 0)
 554                         {       // There were differences in the sub-directories
 555                                 if (existsalldirs || pCtxt->m_bWalkUniques)
 556                                         di.diffcode.diffcode |= DIFFCODE::DIFF;
 557                                 res += ndiff;
 558                         }
 559                         else
 560                         if (ndiff == 0)
 561                         {       // Sub-directories were identical
 562                                 if (existsalldirs)
 563                                         di.diffcode.diffcode |= DIFFCODE::SAME;
 564                                 else if (pCtxt->m_bWalkUniques && !di.diffcode.isResultFiltered())
 565                                         di.diffcode.diffcode |= DIFFCODE::DIFF;
 566                         }
 567                         else
 568                         if (ndiff == -1)
 569                         {       // There were file IO-errors during sub-directory comparison.
 570                                 di.diffcode.diffcode |= DIFFCODE::CMPERR;
 571                                 bCompareFailure = true;
 572                         }
 573
 574                         if (nDirs == 3 && (di.diffcode.diffcode & DIFFCODE::COMPAREFLAGS) == DIFFCODE::DIFF && !di.diffcode.isResultFiltered())
 575                         {
 576                                 di.diffcode.diffcode &= ~DIFFCODE::COMPAREFLAGS3WAY;
 577                                 di.diffcode.diffcode |= GetDirCompareFlags3Way(di);
 578                         }
 579                 }
 580                 if (existsalldirs)
 581                         queue.enqueueUrgentNotification(new WorkNotification(di, queueResult));
 582                 else
 583                         queue.enqueueNotification(new WorkNotification(di, queueResult));
 584                 ++count;
 585                 pos = curpos;
 586                 pCtxt->GetNextSiblingDiffRefPosition(pos);
 587         }
 588
 589         while (count > 0)
 590         {
 591                 AutoPtr<Notification> pNf(queueResult.waitDequeueNotification());
 592                 if (pNf.get() == nullptr)
 593                         break;
 594                 WorkCompletedNotification* pWorkCompletedNf = dynamic_cast<WorkCompletedNotification*>(pNf.get());
 595                 if (pWorkCompletedNf != nullptr) {
 596                         DIFFITEM &di = pWorkCompletedNf->data();
 597                         if (di.diffcode.isResultError()) {
 598                                 DIFFITEM *diParent = di.GetParentLink();
 599                                 assert(diParent != nullptr);
 600                                 if (diParent != nullptr)
 601                                 {
 602                                         diParent->diffcode.diffcode |= DIFFCODE::CMPERR;
 603                                         bCompareFailure = true;
 604                                 }
 605                         }
 606
 607                         if (di.diffcode.isResultDiff() ||
 608                                 (!di.diffcode.existAll() && !di.diffcode.isResultFiltered()))
 609                                 res++;
 610                 }
 611                 --count;
 612         }
 613
 614         return bCompareFailure || pCtxt->ShouldAbort() ? -1 : res;
 615 }
 616
 617 /**
 618  * @brief Compare DiffItems in context marked for rescan.
 619  *
 620  * @param myStruct [in,out] A structure containing compare-related data.
 621  * @param parentdiffpos [in] Position of parent diff item
 622  * @return >= 0 number of diff items, -1 if compare was aborted
 623  */
 624 static int CompareRequestedItems(DiffFuncStruct *myStruct, DIFFITEM *parentdiffpos)
 625 {
 626         CDiffContext *pCtxt = myStruct->context;
 627         FolderCmp fc(pCtxt);
 628         int res = 0;
 629         bool bCompareFailure = false;
 630         bool bCompareIndeterminate = false;
 631         if (parentdiffpos == nullptr)
 632                 myStruct->pSemaphore->wait();
 633
 634         // Since the collect thread deletes the DiffItems in the rescan by "Refresh selected",
 635         // the compare thread process should not be executed until the collect thread process is completed
 636         // to avoid accessing  the deleted DiffItems.
 637         assert(myStruct->nCollectThreadState == CDiffThread::THREAD_COMPLETED);
 638
 639         int nDirs = pCtxt->GetCompareDirs();
 640         DIFFITEM *pos = pCtxt->GetFirstChildDiffPosition(parentdiffpos);
 641         while (pos != nullptr)
 642         {
 643                 if (pCtxt->ShouldAbort())
 644                 {
 645                         res = -1;
 646                         break;
 647                 }
 648
 649                 DIFFITEM *curpos = pos;
 650                 DIFFITEM &di = pCtxt->GetNextSiblingDiffRefPosition(pos);
 651                 bool existsalldirs = di.diffcode.existAll();
 652                 if (di.diffcode.isDirectory())
 653                 {
 654                         if (pCtxt->m_bRecursive)
 655                         {
 656                                 di.diffcode.diffcode &= ~(DIFFCODE::DIFF | DIFFCODE::SAME);
 657                                 int ndiff = CompareRequestedItems(myStruct, curpos);
 658                                 if (ndiff > 0)
 659                                 {
 660                                         if (existsalldirs || pCtxt->m_bWalkUniques)
 661                                                 di.diffcode.diffcode |= DIFFCODE::DIFF;
 662                                         res += ndiff;
 663                                 }
 664                                 else
 665                                 if (ndiff == 0)
 666                                 {
 667                                         if (existsalldirs)
 668                                                 di.diffcode.diffcode |= DIFFCODE::SAME;
 669                                         else if (pCtxt->m_bWalkUniques && !di.diffcode.isResultFiltered())
 670                                                 di.diffcode.diffcode |= DIFFCODE::DIFF;
 671                                 }
 672                                 else
 673                                 if (ndiff == -1)
 674                                 {       // There were file IO-errors during sub-directory comparison.
 675                                         di.diffcode.diffcode |= DIFFCODE::CMPERR;
 676                                         bCompareFailure = true;
 677                                 }
 678                                 else
 679                                 if (ndiff == -2)
 680                                 {       // There were files that have not been compared
 681                                         bCompareIndeterminate = true;
 682                                 }
 683
 684                                 if (nDirs == 3 && (di.diffcode.diffcode & DIFFCODE::COMPAREFLAGS) == DIFFCODE::DIFF && !di.diffcode.isResultFiltered())
 685                                 {
 686                                         di.diffcode.diffcode &= ~DIFFCODE::COMPAREFLAGS3WAY;
 687                                         di.diffcode.diffcode |= GetDirCompareFlags3Way(di);
 688                                 }
 689                         }
 690                 }
 691                 else
 692                 {
 693                         if (di.diffcode.isScanNeeded())
 694                         {
 695                                 CompareDiffItem(fc, di);
 696                                 if (di.diffcode.isResultError())
 697                                 {
 698                                         DIFFITEM *diParent = di.GetParentLink();
 699                                         assert(diParent != nullptr);
 700                                         if (diParent != nullptr)
 701                                         {
 702                                                 diParent->diffcode.diffcode |= DIFFCODE::CMPERR;
 703                                                 bCompareFailure = true;
 704                                         }
 705                                 }
 706                         }
 707                         else
 708                         {
 709                                 if (di.diffcode.isResultError())
 710                                         bCompareFailure = true;
 711                                 else if (di.diffcode.isResultNone() || di.diffcode.isResultAbort())
 712                                         bCompareIndeterminate = true;
 713                         }
 714                 }
 715                 if (di.diffcode.isResultDiff() ||
 716                         (!existsalldirs && !di.diffcode.isResultFiltered()))
 717                         res++;
 718         }
 719         return bCompareIndeterminate ? -2 : (bCompareFailure ? -1 : res);
 720 }
 721
 722 int DirScan_CompareRequestedItems(DiffFuncStruct *myStruct, DIFFITEM *parentdiffpos)
 723 {
 724         return CompareRequestedItems(myStruct, parentdiffpos);
 725 }
 726
 727 static int markChildrenForRescan(CDiffContext *pCtxt, DIFFITEM *parentdiffpos)
 728 {
 729         int ncount = 0;
 730         DIFFITEM *pos = pCtxt->GetFirstChildDiffPosition(parentdiffpos);
 731         while (pos != nullptr)
 732         {
 733                 DIFFITEM *curpos = pos;
 734                 DIFFITEM &di = pCtxt->GetNextSiblingDiffRefPosition(pos);
 735                 if (di.diffcode.isDirectory())
 736                         ncount += markChildrenForRescan(pCtxt, curpos);
 737                 else
 738                 {
 739                         di.diffcode.diffcode |= DIFFCODE::NEEDSCAN;
 740                         ++ncount;
 741                 }
 742         }
 743         return ncount;
 744 }
 745
 746 int DirScan_UpdateMarkedItems(DiffFuncStruct *myStruct, DIFFITEM *parentdiffpos)
 747 {
 748         CDiffContext *pCtxt = myStruct->context;
 749         DIFFITEM *pos = pCtxt->GetFirstChildDiffPosition(parentdiffpos);
 750         int ncount = 0;
 751
 752         while (pos != nullptr)
 753         {
 754                 if (pCtxt->ShouldAbort())
 755                         break;
 756                 DIFFITEM *curpos = pos;
 757                 DIFFITEM &di = pCtxt->GetNextSiblingDiffRefPosition(pos);
 758                 if (di.diffcode.isScanNeeded())
 759                 {
 760                         bool bItemsExist = true;
 761                         UpdateDiffItem(di, bItemsExist, pCtxt);
 762                         if (!bItemsExist)
 763                         {
 764                                 di.DelinkFromSiblings();        // delink from list of Siblings
 765                                 delete &di;                                     // Also delete all Children items
 766                                 continue;                                       // (... because `di` is now invalid)
 767                         }
 768                         if (!di.diffcode.isDirectory())
 769                                 ++ncount;
 770                 }
 771                 if (di.diffcode.isDirectory() && pCtxt->m_bRecursive)
 772                 {
 773                         for (int i = 0; i < pCtxt->GetCompareDirs(); ++i)
 774                                 if (di.diffcode.exists(i))
 775                                         di.diffFileInfo[i].size = 0;
 776                         if (di.diffcode.isScanNeeded() && !di.diffcode.isResultFiltered())
 777                         {
 778                                 di.RemoveChildren();
 779                                 di.diffcode.diffcode &= ~DIFFCODE::NEEDSCAN;
 780
 781                                 bool casesensitive = false;
 782                                 int depth = myStruct->context->m_bRecursive ? -1 : 0;
 783                                 String subdir[3];
 784                                 PathContext paths = myStruct->context->GetNormalizedPaths();
 785                                 for (int i = 0; i < pCtxt->GetCompareDirs(); ++i)
 786                                         subdir[i] = di.diffFileInfo[i].GetFile();
 787                                 DirScan_GetItems(paths, subdir, myStruct,
 788                                         casesensitive, depth, &di, myStruct->context->m_bWalkUniques);
 789                                 ncount += markChildrenForRescan(myStruct->context, curpos);
 790                         }
 791                         else
 792                         {
 793                                 ncount += DirScan_UpdateMarkedItems(myStruct, curpos);
 794                         }
 795                 }
 796                 if (parentdiffpos != nullptr && pCtxt->m_bRecursive)
 797                 {
 798                         for (int nIndex = 0; nIndex < pCtxt->GetCompareDirs(); ++nIndex)
 799                                 if (curpos->diffFileInfo[nIndex].size != DirItem::FILE_SIZE_NONE)
 800                                         parentdiffpos->diffFileInfo[nIndex].size += curpos->diffFileInfo[nIndex].size;
 801                 }
 802         }
 803         return ncount;
 804 }
 805 /**
 806  * @brief Update diffitem file/dir infos.
 807  *
 808  * Re-tests dirs/files if sides still exists, and updates infos for
 809  * existing sides. This assumes filenames, or paths are not changed.
 810  * Since in normal situations (I can think of) they cannot change
 811  * after first compare.
 812  *
 813  * @param [in,out] di DiffItem to update.
 814  * @param [out] bExists Set to
 815  *  - true if one of items exists so diffitem is valid
 816  *  - false if items were deleted, so diffitem is not valid
 817  * @param [in] pCtxt Compare context
 818  */
 819 static void UpdateDiffItem(DIFFITEM &di, bool & bExists, CDiffContext *pCtxt)
 820 {
 821         bExists = false;
 822         di.diffcode.setSideNone();
 823         for (int i = 0; i < pCtxt->GetCompareDirs(); ++i)
 824         {
 825                 di.diffFileInfo[i].ClearPartial();
 826                 if (pCtxt->UpdateInfoFromDiskHalf(di, i))
 827                 {
 828                         bool bUpdated = false;
 829                         if (di.diffFileInfo[i].IsDirectory() == di.diffcode.isDirectory())
 830                         {
 831                                 String filepath = paths::ConcatPath(pCtxt->GetPath(i), di.diffFileInfo[i].GetFile());
 832                                 if (di.diffFileInfo[i].UpdateFileName(filepath)) {
 833                                         di.diffcode.diffcode |= DIFFCODE::FIRST << i;
 834                                         bExists = true;
 835                                         bUpdated = true;
 836                                 }
 837                         }
 838                         if (!bUpdated)
 839                                 di.diffFileInfo[i].ClearPartial();
 840                 }
 841         }
 842 }
 843
 844 /**
 845  * @brief Compare two diffitems and add results to difflist in context.
 846  *
 847  * This function does the actual compare for previously gathered list of
 848  * items. Basically we:
 849  * - ignore items matching file filters
 850  * - add non-ignored directories (no compare for directory items)
 851  * - add unique files
 852  * - compare files
 853  *
 854  * @param [in] di DiffItem to compare
 855  * @param [in,out] pCtxt Compare context: contains difflist, encoding info etc.
 856  * @todo For date compare, maybe we should use creation date if modification
 857  * date is missing?
 858  */
 859 static void CompareDiffItem(FolderCmp &fc, DIFFITEM &di)
 860 {
 861         CDiffContext *const pCtxt = fc.m_pCtxt;
 862         int nDirs = pCtxt->GetCompareDirs();
 863         // Clear rescan-request flag (not set by all codepaths)
 864         di.diffcode.diffcode &= ~DIFFCODE::NEEDSCAN;
 865         // Is it a directory?
 866         if (di.diffcode.isDirectory())
 867         {
 868                 // We don't actually 'compare' directories, just add non-ignored
 869                 // directories to list.
 870         }
 871         else
 872         {
 873                 // 1. Test against filters
 874                 if (pCtxt->m_piFilterGlobal==nullptr ||
 875                         (nDirs == 2 && pCtxt->m_piFilterGlobal->includeFile(
 876                                 paths::ConcatPath(di.diffFileInfo[0].path, di.diffFileInfo[0].filename),
 877                                 paths::ConcatPath(di.diffFileInfo[1].path, di.diffFileInfo[1].filename)
 878                         )) ||
 879                         (nDirs == 3 && pCtxt->m_piFilterGlobal->includeFile(
 880                                 paths::ConcatPath(di.diffFileInfo[0].path, di.diffFileInfo[0].filename),
 881                                 paths::ConcatPath(di.diffFileInfo[1].path, di.diffFileInfo[1].filename),
 882                                 paths::ConcatPath(di.diffFileInfo[2].path, di.diffFileInfo[2].filename)
 883                         )))
 884                 {
 885                         di.diffcode.diffcode |= DIFFCODE::INCLUDED;
 886                         di.diffcode.diffcode |= fc.prepAndCompareFiles(di);
 887                         di.nsdiffs = fc.m_ndiffs;
 888                         di.nidiffs = fc.m_ntrivialdiffs;
 889
 890                         for (int i = 0; i < nDirs; ++i)
 891                         {
 892                                 // Set text statistics
 893                                 if (di.diffcode.exists(i))
 894                                 {
 895                                         di.diffFileInfo[i].m_textStats = fc.m_diffFileData.m_textStats[i];
 896                                         di.diffFileInfo[i].encoding = fc.m_diffFileData.m_FileLocation[i].encoding;
 897                                 }
 898                         }
 899                 }
 900                 else
 901                 {
 902                         di.diffcode.diffcode |= DIFFCODE::SKIPPED;
 903                 }
 904         }
 905         pCtxt->m_pCompareStats->AddItem(di.diffcode.diffcode);
 906 }
 907
 908 /**
 909  * @brief Add one compare item to list.
 910  * @param [in] sLeftDir Left subdirectory.
 911  * @param [in] sRightDir Right subdirectory.
 912  * @param [in] lent Left item data to add.
 913  * @param [in] rent Right item data to add.
 914  * @param [in] pCtxt Compare context.
 915  * @param [in] parent Parent of item to be added
 916  */
 917 static DIFFITEM *AddToList(const String& sLeftDir, const String& sRightDir,
 918         const DirItem * lent, const DirItem * rent,
 919         unsigned code, DiffFuncStruct *myStruct, DIFFITEM *parent)
 920 {
 921         return AddToList(sLeftDir, sRightDir, sLeftDir, lent, rent, nullptr, code, myStruct, parent, 2);
 922 }
 923
 924 /**
 925  * @brief Add one compare item to list.
 926  */
 927 static DIFFITEM *AddToList(const String& sDir1, const String& sDir2, const String& sDir3,
 928         const DirItem *ent1, const DirItem *ent2, const DirItem *ent3,
 929         unsigned code, DiffFuncStruct *myStruct, DIFFITEM *parent, int nItems /*= 3*/)
 930 {
 931         // We must store both paths - we cannot get paths later
 932         // and we need unique item paths for example when items
 933         // change to identical
 934         DIFFITEM *di = myStruct->context->AddNewDiff(parent);
 935
 936         di->diffFileInfo[0].path = sDir1;
 937         di->diffFileInfo[1].path = sDir2;
 938         di->diffFileInfo[2].path = sDir3;
 939
 940         if (ent1 != nullptr)
 941         {
 942                 di->diffFileInfo[0].filename = ent1->filename;
 943                 di->diffFileInfo[0].mtime = ent1->mtime;
 944                 di->diffFileInfo[0].ctime = ent1->ctime;
 945                 di->diffFileInfo[0].size = ent1->size;
 946                 di->diffFileInfo[0].flags.attributes = ent1->flags.attributes;
 947         }
 948         else
 949         {
 950                 // Don't break CDirView::DoCopyRightToLeft()
 951                 if (ent3 != nullptr)
 952                         di->diffFileInfo[0].filename = ent3->filename;
 953                 else if (ent2 != nullptr)
 954                         di->diffFileInfo[0].filename = ent2->filename;
 955         }
 956
 957         if (ent2 != nullptr)
 958         {
 959                 di->diffFileInfo[1].filename = ent2->filename;
 960                 di->diffFileInfo[1].mtime = ent2->mtime;
 961                 di->diffFileInfo[1].ctime = ent2->ctime;
 962                 di->diffFileInfo[1].size = ent2->size;
 963                 di->diffFileInfo[1].flags.attributes = ent2->flags.attributes;
 964         }
 965         else
 966         {
 967                 // Don't break CDirView::DoCopyLeftToRight()
 968                 if (ent1 != nullptr)
 969                         di->diffFileInfo[1].filename = ent1->filename;
 970                 else if (ent3 != nullptr)
 971                         di->diffFileInfo[1].filename = ent3->filename;
 972         }
 973
 974         if (ent3 != nullptr)
 975         {
 976                 di->diffFileInfo[2].filename = ent3->filename;
 977                 di->diffFileInfo[2].mtime = ent3->mtime;
 978                 di->diffFileInfo[2].ctime = ent3->ctime;
 979                 di->diffFileInfo[2].size = ent3->size;
 980                 di->diffFileInfo[2].flags.attributes = ent3->flags.attributes;
 981         }
 982         else
 983         {
 984                 // Don't break CDirView::DoCopyLeftToRight()
 985                 if (ent1 != nullptr)
 986                         di->diffFileInfo[2].filename = ent1->filename;
 987                 else if (ent2 != nullptr)
 988                         di->diffFileInfo[2].filename = ent2->filename;
 989         }
 990
 991         if (nItems == 2)
 992                 di->diffcode.diffcode = code;
 993         else
 994                 di->diffcode.diffcode = code | DIFFCODE::THREEWAY;
 995
 996         if (!myStruct->bMarkedRescan && myStruct->m_fncCollect)
 997         {
 998                 myStruct->context->m_pCompareStats->IncreaseTotalItems();
 999                 myStruct->pSemaphore->set();
1000         }
1001         return di;
1002 }
1003
1004 /**
1005  * @brief Get the 3-way compare flag of the specified directory item.
1006  * @param [in] di Directory Item to get the 3-way compare flag.
1007  * @return 3-way compare flag of the specified directory item.
1008  */
1009 static unsigned GetDirCompareFlags3Way(const DIFFITEM& di)
1010 {
1011         assert(di.diffcode.isDirectory());
1012         assert((di.diffcode.diffcode & DIFFCODE::COMPAREFLAGS) == DIFFCODE::DIFF && !di.diffcode.isResultFiltered());
1013
1014         bool bSet = false;
1015         unsigned code = DIFFCODE::DIFFALL;
1016
1017         if (di.diffcode.isSideFirstOnly() || di.diffcode.isMissingFirstOnly())
1018         {
1019                 code = DIFFCODE::DIFF1STONLY;
1020                 bSet = true;
1021         }
1022         else if (di.diffcode.isSideSecondOnly() || di.diffcode.isMissingSecondOnly())
1023         {
1024                 code = DIFFCODE::DIFF2NDONLY;
1025                 bSet = true;
1026         }
1027         else if (di.diffcode.isSideThirdOnly() || di.diffcode.isMissingThirdOnly())
1028         {
1029                 code = DIFFCODE::DIFF3RDONLY;
1030                 bSet = true;
1031         }
1032
1033         if (di.HasChildren())
1034         {
1035                 for (DIFFITEM* pdi = di.GetFirstChild(); pdi != nullptr; pdi = pdi->GetFwdSiblingLink())
1036                 {
1037                         if ((pdi->diffcode.diffcode & DIFFCODE::COMPAREFLAGS) == DIFFCODE::DIFF && !pdi->diffcode.isResultFiltered())
1038                         {
1039                                 if (!bSet)
1040                                 {
1041                                         code = (pdi->diffcode.diffcode & DIFFCODE::COMPAREFLAGS3WAY);
1042                                         bSet = true;
1043                                 }
1044                                 else
1045                                 {
1046                                         switch (code)
1047                                         {
1048                                         case DIFFCODE::DIFF1STONLY:
1049                                                 if ((pdi->diffcode.diffcode & DIFFCODE::COMPAREFLAGS3WAY) != DIFFCODE::DIFF1STONLY)
1050                                                         code = DIFFCODE::DIFFALL;
1051                                                 break;
1052                                         case DIFFCODE::DIFF2NDONLY:
1053                                                 if ((pdi->diffcode.diffcode & DIFFCODE::COMPAREFLAGS3WAY) != DIFFCODE::DIFF2NDONLY)
1054                                                         code = DIFFCODE::DIFFALL;
1055                                                 break;
1056                                         case DIFFCODE::DIFF3RDONLY:
1057                                                 if ((pdi->diffcode.diffcode & DIFFCODE::COMPAREFLAGS3WAY) != DIFFCODE::DIFF3RDONLY)
1058                                                         code = DIFFCODE::DIFFALL;
1059                                                 break;
1060                                         default:
1061                                                 code = DIFFCODE::DIFFALL;
1062                                         }
1063                                 }
1064                         }
1065                 }
1066         }
1067
1068         return code;
1069 }