OrcCommand: UtilitiesMain: add EnumOption

DFIR-ORC · May 5, 2023 · 466a30f · 466a30f
1 parent 0435d20
commit 466a30f
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/OrcCommand/UtilitiesMain.h b/src/OrcCommand/UtilitiesMain.h
@@ -614,6 +614,14 @@ class UtilitiesMain
 
     bool ToggleBooleanOption(LPCWSTR szArg, LPCWSTR szOption, bool& bOption);
 
+    template <typename _EnumT>
+    bool EnumOption(LPCWSTR szArg, LPCWSTR szOption, _EnumT& eOption, _EnumT eValue) {
+        if (_wcsnicmp(szArg, szOption, wcslen(szOption)))
+            return false;
+        eOption = eValue;
+        return true;
+    }
+
     bool ShadowsOption(
         LPCWSTR szArg,
         LPCWSTR szOption,