| OLD | NEW |
| (Empty) |
| 1 // Common/Wildcard.cpp | |
| 2 | |
| 3 #include "StdAfx.h" | |
| 4 | |
| 5 #include "Wildcard.h" | |
| 6 | |
| 7 bool g_CaseSensitive = | |
| 8 #ifdef _WIN32 | |
| 9 false; | |
| 10 #else | |
| 11 true; | |
| 12 #endif | |
| 13 | |
| 14 static const wchar_t kAnyCharsChar = L'*'; | |
| 15 static const wchar_t kAnyCharChar = L'?'; | |
| 16 | |
| 17 #ifdef _WIN32 | |
| 18 static const wchar_t kDirDelimiter1 = L'\\'; | |
| 19 #endif | |
| 20 static const wchar_t kDirDelimiter2 = L'/'; | |
| 21 | |
| 22 static const UString kWildCardCharSet = L"?*"; | |
| 23 | |
| 24 static const UString kIllegalWildCardFileNameChars= | |
| 25 L"\x1\x2\x3\x4\x5\x6\x7\x8\x9\xA\xB\xC\xD\xE\xF" | |
| 26 L"\x10\x11\x12\x13\x14\x15\x16\x17\x18\x19\x1A\x1B\x1C\x1D\x1E\x1F" | |
| 27 L"\"/:<>\\|"; | |
| 28 | |
| 29 | |
| 30 static inline bool IsCharDirLimiter(wchar_t c) | |
| 31 { | |
| 32 return ( | |
| 33 #ifdef _WIN32 | |
| 34 c == kDirDelimiter1 || | |
| 35 #endif | |
| 36 c == kDirDelimiter2); | |
| 37 } | |
| 38 | |
| 39 int CompareFileNames(const UString &s1, const UString &s2) | |
| 40 { | |
| 41 if (g_CaseSensitive) | |
| 42 return s1.Compare(s2); | |
| 43 return s1.CompareNoCase(s2); | |
| 44 } | |
| 45 | |
| 46 // ----------------------------------------- | |
| 47 // this function compares name with mask | |
| 48 // ? - any char | |
| 49 // * - any char or empty | |
| 50 | |
| 51 static bool EnhancedMaskTest(const wchar_t *mask, const wchar_t *name) | |
| 52 { | |
| 53 for (;;) | |
| 54 { | |
| 55 wchar_t m = *mask; | |
| 56 wchar_t c = *name; | |
| 57 if (m == 0) | |
| 58 return (c == 0); | |
| 59 if (m == kAnyCharsChar) | |
| 60 { | |
| 61 if (EnhancedMaskTest(mask + 1, name)) | |
| 62 return true; | |
| 63 if (c == 0) | |
| 64 return false; | |
| 65 } | |
| 66 else | |
| 67 { | |
| 68 if (m == kAnyCharChar) | |
| 69 { | |
| 70 if (c == 0) | |
| 71 return false; | |
| 72 } | |
| 73 else if (m != c) | |
| 74 if (g_CaseSensitive || MyCharUpper(m) != MyCharUpper(c)) | |
| 75 return false; | |
| 76 mask++; | |
| 77 } | |
| 78 name++; | |
| 79 } | |
| 80 } | |
| 81 | |
| 82 // -------------------------------------------------- | |
| 83 // Splits path to strings | |
| 84 | |
| 85 void SplitPathToParts(const UString &path, UStringVector &pathParts) | |
| 86 { | |
| 87 pathParts.Clear(); | |
| 88 UString name; | |
| 89 int len = path.Length(); | |
| 90 if (len == 0) | |
| 91 return; | |
| 92 for (int i = 0; i < len; i++) | |
| 93 { | |
| 94 wchar_t c = path[i]; | |
| 95 if (IsCharDirLimiter(c)) | |
| 96 { | |
| 97 pathParts.Add(name); | |
| 98 name.Empty(); | |
| 99 } | |
| 100 else | |
| 101 name += c; | |
| 102 } | |
| 103 pathParts.Add(name); | |
| 104 } | |
| 105 | |
| 106 void SplitPathToParts(const UString &path, UString &dirPrefix, UString &name) | |
| 107 { | |
| 108 int i; | |
| 109 for (i = path.Length() - 1; i >= 0; i--) | |
| 110 if (IsCharDirLimiter(path[i])) | |
| 111 break; | |
| 112 dirPrefix = path.Left(i + 1); | |
| 113 name = path.Mid(i + 1); | |
| 114 } | |
| 115 | |
| 116 UString ExtractDirPrefixFromPath(const UString &path) | |
| 117 { | |
| 118 int i; | |
| 119 for (i = path.Length() - 1; i >= 0; i--) | |
| 120 if (IsCharDirLimiter(path[i])) | |
| 121 break; | |
| 122 return path.Left(i + 1); | |
| 123 } | |
| 124 | |
| 125 UString ExtractFileNameFromPath(const UString &path) | |
| 126 { | |
| 127 int i; | |
| 128 for (i = path.Length() - 1; i >= 0; i--) | |
| 129 if (IsCharDirLimiter(path[i])) | |
| 130 break; | |
| 131 return path.Mid(i + 1); | |
| 132 } | |
| 133 | |
| 134 | |
| 135 bool CompareWildCardWithName(const UString &mask, const UString &name) | |
| 136 { | |
| 137 return EnhancedMaskTest(mask, name); | |
| 138 } | |
| 139 | |
| 140 bool DoesNameContainWildCard(const UString &path) | |
| 141 { | |
| 142 return (path.FindOneOf(kWildCardCharSet) >= 0); | |
| 143 } | |
| 144 | |
| 145 | |
| 146 // ----------------------------------------------------------' | |
| 147 // NWildcard | |
| 148 | |
| 149 namespace NWildcard { | |
| 150 | |
| 151 | |
| 152 /* | |
| 153 M = MaskParts.Size(); | |
| 154 N = TestNameParts.Size(); | |
| 155 | |
| 156 File Dir | |
| 157 ForFile req M<=N [N-M, N) - | |
| 158 nonreq M=N [0, M) - | |
| 159 | |
| 160 ForDir req M<N [0, M) ... [N-M-1, N-1) same as ForBoth-File | |
| 161 nonreq [0, M) same as ForBoth-File | |
| 162 | |
| 163 ForBoth req m<=N [0, M) ... [N-M, N) same as ForBoth-File | |
| 164 nonreq [0, M) same as ForBoth-File | |
| 165 | |
| 166 */ | |
| 167 | |
| 168 bool CItem::CheckPath(const UStringVector &pathParts, bool isFile) const | |
| 169 { | |
| 170 if (!isFile && !ForDir) | |
| 171 return false; | |
| 172 int delta = (int)pathParts.Size() - (int)PathParts.Size(); | |
| 173 if (delta < 0) | |
| 174 return false; | |
| 175 int start = 0; | |
| 176 int finish = 0; | |
| 177 if (isFile) | |
| 178 { | |
| 179 if (!ForDir && !Recursive && delta !=0) | |
| 180 return false; | |
| 181 if (!ForFile && delta == 0) | |
| 182 return false; | |
| 183 if (!ForDir && Recursive) | |
| 184 start = delta; | |
| 185 } | |
| 186 if (Recursive) | |
| 187 { | |
| 188 finish = delta; | |
| 189 if (isFile && !ForFile) | |
| 190 finish = delta - 1; | |
| 191 } | |
| 192 for (int d = start; d <= finish; d++) | |
| 193 { | |
| 194 int i; | |
| 195 for (i = 0; i < PathParts.Size(); i++) | |
| 196 if (!CompareWildCardWithName(PathParts[i], pathParts[i + d])) | |
| 197 break; | |
| 198 if (i == PathParts.Size()) | |
| 199 return true; | |
| 200 } | |
| 201 return false; | |
| 202 } | |
| 203 | |
| 204 int CCensorNode::FindSubNode(const UString &name) const | |
| 205 { | |
| 206 for (int i = 0; i < SubNodes.Size(); i++) | |
| 207 if (CompareFileNames(SubNodes[i].Name, name) == 0) | |
| 208 return i; | |
| 209 return -1; | |
| 210 } | |
| 211 | |
| 212 void CCensorNode::AddItemSimple(bool include, CItem &item) | |
| 213 { | |
| 214 if (include) | |
| 215 IncludeItems.Add(item); | |
| 216 else | |
| 217 ExcludeItems.Add(item); | |
| 218 } | |
| 219 | |
| 220 void CCensorNode::AddItem(bool include, CItem &item) | |
| 221 { | |
| 222 if (item.PathParts.Size() <= 1) | |
| 223 { | |
| 224 AddItemSimple(include, item); | |
| 225 return; | |
| 226 } | |
| 227 const UString &front = item.PathParts.Front(); | |
| 228 if (DoesNameContainWildCard(front)) | |
| 229 { | |
| 230 AddItemSimple(include, item); | |
| 231 return; | |
| 232 } | |
| 233 int index = FindSubNode(front); | |
| 234 if (index < 0) | |
| 235 index = SubNodes.Add(CCensorNode(front, this)); | |
| 236 item.PathParts.Delete(0); | |
| 237 SubNodes[index].AddItem(include, item); | |
| 238 } | |
| 239 | |
| 240 void CCensorNode::AddItem(bool include, const UString &path, bool recursive, boo
l forFile, bool forDir) | |
| 241 { | |
| 242 CItem item; | |
| 243 SplitPathToParts(path, item.PathParts); | |
| 244 item.Recursive = recursive; | |
| 245 item.ForFile = forFile; | |
| 246 item.ForDir = forDir; | |
| 247 AddItem(include, item); | |
| 248 } | |
| 249 | |
| 250 bool CCensorNode::NeedCheckSubDirs() const | |
| 251 { | |
| 252 for (int i = 0; i < IncludeItems.Size(); i++) | |
| 253 { | |
| 254 const CItem &item = IncludeItems[i]; | |
| 255 if (item.Recursive || item.PathParts.Size() > 1) | |
| 256 return true; | |
| 257 } | |
| 258 return false; | |
| 259 } | |
| 260 | |
| 261 bool CCensorNode::AreThereIncludeItems() const | |
| 262 { | |
| 263 if (IncludeItems.Size() > 0) | |
| 264 return true; | |
| 265 for (int i = 0; i < SubNodes.Size(); i++) | |
| 266 if (SubNodes[i].AreThereIncludeItems()) | |
| 267 return true; | |
| 268 return false; | |
| 269 } | |
| 270 | |
| 271 bool CCensorNode::CheckPathCurrent(bool include, const UStringVector &pathParts,
bool isFile) const | |
| 272 { | |
| 273 const CObjectVector<CItem> &items = include ? IncludeItems : ExcludeItems; | |
| 274 for (int i = 0; i < items.Size(); i++) | |
| 275 if (items[i].CheckPath(pathParts, isFile)) | |
| 276 return true; | |
| 277 return false; | |
| 278 } | |
| 279 | |
| 280 bool CCensorNode::CheckPath(UStringVector &pathParts, bool isFile, bool &include
) const | |
| 281 { | |
| 282 if (CheckPathCurrent(false, pathParts, isFile)) | |
| 283 { | |
| 284 include = false; | |
| 285 return true; | |
| 286 } | |
| 287 include = true; | |
| 288 bool finded = CheckPathCurrent(true, pathParts, isFile); | |
| 289 if (pathParts.Size() == 1) | |
| 290 return finded; | |
| 291 int index = FindSubNode(pathParts.Front()); | |
| 292 if (index >= 0) | |
| 293 { | |
| 294 UStringVector pathParts2 = pathParts; | |
| 295 pathParts2.Delete(0); | |
| 296 if (SubNodes[index].CheckPath(pathParts2, isFile, include)) | |
| 297 return true; | |
| 298 } | |
| 299 return finded; | |
| 300 } | |
| 301 | |
| 302 bool CCensorNode::CheckPath(const UString &path, bool isFile, bool &include) con
st | |
| 303 { | |
| 304 UStringVector pathParts; | |
| 305 SplitPathToParts(path, pathParts); | |
| 306 return CheckPath(pathParts, isFile, include); | |
| 307 } | |
| 308 | |
| 309 bool CCensorNode::CheckPath(const UString &path, bool isFile) const | |
| 310 { | |
| 311 bool include; | |
| 312 if (CheckPath(path, isFile, include)) | |
| 313 return include; | |
| 314 return false; | |
| 315 } | |
| 316 | |
| 317 bool CCensorNode::CheckPathToRoot(bool include, UStringVector &pathParts, bool i
sFile) const | |
| 318 { | |
| 319 if (CheckPathCurrent(include, pathParts, isFile)) | |
| 320 return true; | |
| 321 if (Parent == 0) | |
| 322 return false; | |
| 323 pathParts.Insert(0, Name); | |
| 324 return Parent->CheckPathToRoot(include, pathParts, isFile); | |
| 325 } | |
| 326 | |
| 327 /* | |
| 328 bool CCensorNode::CheckPathToRoot(bool include, const UString &path, bool isFile
) const | |
| 329 { | |
| 330 UStringVector pathParts; | |
| 331 SplitPathToParts(path, pathParts); | |
| 332 return CheckPathToRoot(include, pathParts, isFile); | |
| 333 } | |
| 334 */ | |
| 335 | |
| 336 void CCensorNode::AddItem2(bool include, const UString &path, bool recursive) | |
| 337 { | |
| 338 if (path.IsEmpty()) | |
| 339 return; | |
| 340 bool forFile = true; | |
| 341 bool forFolder = true; | |
| 342 UString path2 = path; | |
| 343 if (IsCharDirLimiter(path[path.Length() - 1])) | |
| 344 { | |
| 345 path2.Delete(path.Length() - 1); | |
| 346 forFile = false; | |
| 347 } | |
| 348 AddItem(include, path2, recursive, forFile, forFolder); | |
| 349 } | |
| 350 | |
| 351 void CCensorNode::ExtendExclude(const CCensorNode &fromNodes) | |
| 352 { | |
| 353 ExcludeItems += fromNodes.ExcludeItems; | |
| 354 for (int i = 0; i < fromNodes.SubNodes.Size(); i++) | |
| 355 { | |
| 356 const CCensorNode &node = fromNodes.SubNodes[i]; | |
| 357 int subNodeIndex = FindSubNode(node.Name); | |
| 358 if (subNodeIndex < 0) | |
| 359 subNodeIndex = SubNodes.Add(CCensorNode(node.Name, this)); | |
| 360 SubNodes[subNodeIndex].ExtendExclude(node); | |
| 361 } | |
| 362 } | |
| 363 | |
| 364 int CCensor::FindPrefix(const UString &prefix) const | |
| 365 { | |
| 366 for (int i = 0; i < Pairs.Size(); i++) | |
| 367 if (CompareFileNames(Pairs[i].Prefix, prefix) == 0) | |
| 368 return i; | |
| 369 return -1; | |
| 370 } | |
| 371 | |
| 372 void CCensor::AddItem(bool include, const UString &path, bool recursive) | |
| 373 { | |
| 374 UStringVector pathParts; | |
| 375 SplitPathToParts(path, pathParts); | |
| 376 bool forFile = true; | |
| 377 if (pathParts.Back().IsEmpty()) | |
| 378 { | |
| 379 forFile = false; | |
| 380 pathParts.DeleteBack(); | |
| 381 } | |
| 382 const UString &front = pathParts.Front(); | |
| 383 bool isAbs = false; | |
| 384 if (front.IsEmpty()) | |
| 385 isAbs = true; | |
| 386 else if (front.Length() == 2 && front[1] == L':') | |
| 387 isAbs = true; | |
| 388 else | |
| 389 { | |
| 390 for (int i = 0; i < pathParts.Size(); i++) | |
| 391 { | |
| 392 const UString &part = pathParts[i]; | |
| 393 if (part == L".." || part == L".") | |
| 394 { | |
| 395 isAbs = true; | |
| 396 break; | |
| 397 } | |
| 398 } | |
| 399 } | |
| 400 int numAbsParts = 0; | |
| 401 if (isAbs) | |
| 402 if (pathParts.Size() > 1) | |
| 403 numAbsParts = pathParts.Size() - 1; | |
| 404 else | |
| 405 numAbsParts = 1; | |
| 406 UString prefix; | |
| 407 for (int i = 0; i < numAbsParts; i++) | |
| 408 { | |
| 409 const UString &front = pathParts.Front(); | |
| 410 if (DoesNameContainWildCard(front)) | |
| 411 break; | |
| 412 prefix += front; | |
| 413 prefix += WCHAR_PATH_SEPARATOR; | |
| 414 pathParts.Delete(0); | |
| 415 } | |
| 416 int index = FindPrefix(prefix); | |
| 417 if (index < 0) | |
| 418 index = Pairs.Add(CPair(prefix)); | |
| 419 | |
| 420 CItem item; | |
| 421 item.PathParts = pathParts; | |
| 422 item.ForDir = true; | |
| 423 item.ForFile = forFile; | |
| 424 item.Recursive = recursive; | |
| 425 Pairs[index].Head.AddItem(include, item); | |
| 426 } | |
| 427 | |
| 428 bool CCensor::CheckPath(const UString &path, bool isFile) const | |
| 429 { | |
| 430 bool finded = false; | |
| 431 for (int i = 0; i < Pairs.Size(); i++) | |
| 432 { | |
| 433 bool include; | |
| 434 if (Pairs[i].Head.CheckPath(path, isFile, include)) | |
| 435 { | |
| 436 if (!include) | |
| 437 return false; | |
| 438 finded = true; | |
| 439 } | |
| 440 } | |
| 441 return finded; | |
| 442 } | |
| 443 | |
| 444 void CCensor::ExtendExclude() | |
| 445 { | |
| 446 int i; | |
| 447 for (i = 0; i < Pairs.Size(); i++) | |
| 448 if (Pairs[i].Prefix.IsEmpty()) | |
| 449 break; | |
| 450 if (i == Pairs.Size()) | |
| 451 return; | |
| 452 int index = i; | |
| 453 for (i = 0; i < Pairs.Size(); i++) | |
| 454 if (index != i) | |
| 455 Pairs[i].Head.ExtendExclude(Pairs[index].Head); | |
| 456 } | |
| 457 | |
| 458 } | |
| OLD | NEW |