From 69cf40161e33fbc671bfe3f6279304cfbca68acf Mon Sep 17 00:00:00 2001
From: gibbed <rick@gibbed.us>
Date: Thu, 19 Jan 2017 18:13:07 -0600
Subject: [PATCH] Added support for '?' to FS wildcard handling.

---
 src/xenia/base/filesystem_wildcard.cc | 32 +++++++++++++++++++++------
 src/xenia/base/filesystem_wildcard.h  |  5 +++--
 2 files changed, 28 insertions(+), 9 deletions(-)

diff --git a/src/xenia/base/filesystem_wildcard.cc b/src/xenia/base/filesystem_wildcard.cc
index 312e9296f..b2b4fc62c 100644
--- a/src/xenia/base/filesystem_wildcard.cc
+++ b/src/xenia/base/filesystem_wildcard.cc
@@ -8,19 +8,22 @@
  */
 
 #include "xenia/base/filesystem_wildcard.h"
+#include "xenia/base/assert.h"
 
 #include <algorithm>
 
 namespace xe {
 namespace filesystem {
 
-WildcardFlags WildcardFlags::FIRST(true, false);
-WildcardFlags WildcardFlags::LAST(false, true);
+WildcardFlags WildcardFlags::FIRST(true, false, false);
+WildcardFlags WildcardFlags::LAST(false, true, false);
+WildcardFlags WildcardFlags::ANY(false, false, true);
 
-WildcardFlags::WildcardFlags() : FromStart(false), ToEnd(false) {}
+WildcardFlags::WildcardFlags()
+    : FromStart(false), ToEnd(false), ExactLength(false) {}
 
-WildcardFlags::WildcardFlags(bool start, bool end)
-    : FromStart(start), ToEnd(end) {}
+WildcardFlags::WildcardFlags(bool start, bool end, bool exact_length)
+    : FromStart(start), ToEnd(end), ExactLength(exact_length) {}
 
 WildcardRule::WildcardRule(const std::string& str_match,
                            const WildcardFlags& flags)
@@ -38,6 +41,11 @@ bool WildcardRule::Check(const std::string& str_lower,
     return false;
   }
 
+  if (rules.ExactLength) {
+    *offset += match.size();
+    return true;
+  }
+
   std::string::size_type result(str_lower.find(match, *offset));
 
   if (result != std::string::npos) {
@@ -62,12 +70,22 @@ void WildcardEngine::PreparePattern(const std::string& pattern) {
   WildcardFlags flags(WildcardFlags::FIRST);
   size_t n = 0;
   size_t last = 0;
-  while ((n = pattern.find_first_of('*', last)) != pattern.npos) {
+  while ((n = pattern.find_first_of("*?", last)) != pattern.npos) {
     if (last != n) {
       std::string str_str(pattern.substr(last, n - last));
       rules.push_back(WildcardRule(str_str, flags));
     }
-    last = n + 1;
+    if (pattern[n] == '?') {
+      auto end = pattern.find_first_not_of('?', n + 1);
+      auto count = end == pattern.npos ? (pattern.size() - n) : (end - n);
+      rules.push_back(
+          WildcardRule(pattern.substr(n, count), WildcardFlags::ANY));
+      last = n + count;
+    } else if (pattern[n] == '*') {
+      last = n + 1;
+    } else {
+      assert_always();
+    }
     flags = WildcardFlags();
   }
   if (last != pattern.size()) {
diff --git a/src/xenia/base/filesystem_wildcard.h b/src/xenia/base/filesystem_wildcard.h
index e9519ffe8..1ecd452f2 100644
--- a/src/xenia/base/filesystem_wildcard.h
+++ b/src/xenia/base/filesystem_wildcard.h
@@ -22,13 +22,14 @@ namespace filesystem {
 
 class WildcardFlags {
  public:
-  bool FromStart : 1, ToEnd : 1;
+  bool FromStart : 1, ToEnd : 1, ExactLength : 1;
 
   WildcardFlags();
-  WildcardFlags(bool start, bool end);
+  WildcardFlags(bool start, bool end, bool exact_length);
 
   static WildcardFlags FIRST;
   static WildcardFlags LAST;
+  static WildcardFlags ANY;
 };
 
 class WildcardRule {