wmic namespace + shell hook

src/defender-control/dcontrol.hpp

@@ -1,20 +1,20 @@
-#pragma once
+#pragma once
-
+
-#include <Windows.h>
+#include <Windows.h>
-#include <iostream>
+#include <iostream>
-
+
-#define DBG_MSG (1 << 0)
+#define DBG_MSG (1 << 0)
-
+
-namespace REG
+namespace REG
-{
+{
-  DWORD read_key(const wchar_t* root_name, const wchar_t* value_name, uint32_t flags = 0);
+  DWORD read_key(const wchar_t* root_name, const wchar_t* value_name, uint32_t flags = 0);
-  bool create_registry(const wchar_t* root_name, HKEY& hkey);
+  bool create_registry(const wchar_t* root_name, HKEY& hkey);
-  bool set_keyval(HKEY& hkey, const wchar_t* value_name, DWORD value);
+  bool set_keyval(HKEY& hkey, const wchar_t* value_name, DWORD value);
-  bool set_keyval_bin(HKEY& hkey, const wchar_t* value_name, DWORD value);
+  bool set_keyval_bin(HKEY& hkey, const wchar_t* value_name, DWORD value);
-}
+}
-
+
-namespace DCONTROL
+namespace DCONTROL
-{
+{
-  bool disable_defender();
+  bool disable_defender();
-  bool check_defender(uint32_t flags = 0);
+  bool check_defender(uint32_t flags = 0);
 }

src/defender-control/defender-control.vcxproj

@@ -143,9 +143,11 @@
   <ItemGroup>
     <ClCompile Include="dcontrol.cpp" />
     <ClCompile Include="main.cpp" />
+    <ClCompile Include="wmic.cpp" />
   </ItemGroup>
   <ItemGroup>
-    <ClInclude Include="dcontrol.h" />
+    <ClInclude Include="dcontrol.hpp" />
+    <ClInclude Include="wmic.hpp" />
   </ItemGroup>
   <Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" />
   <ImportGroup Label="ExtensionTargets">

src/defender-control/defender-control.vcxproj.filters

@@ -16,6 +16,9 @@
     <Filter Include="Source Files\defender-control">
       <UniqueIdentifier>{8a88e18b-d3f3-447e-a3b0-9867c153c3c1}</UniqueIdentifier>
     </Filter>
+    <Filter Include="Source Files\defender-control\wmic">
+      <UniqueIdentifier>{db30358b-f563-460e-92fb-eacafe3a68cb}</UniqueIdentifier>
+    </Filter>
   </ItemGroup>
   <ItemGroup>
     <ClCompile Include="main.cpp">
@@ -24,10 +27,16 @@
     <ClCompile Include="dcontrol.cpp">
       <Filter>Source Files\defender-control</Filter>
     </ClCompile>
+    <ClCompile Include="wmic.cpp">
+      <Filter>Source Files\defender-control\wmic</Filter>
+    </ClCompile>
   </ItemGroup>
   <ItemGroup>
-    <ClInclude Include="dcontrol.h">
+    <ClInclude Include="dcontrol.hpp">
       <Filter>Source Files\defender-control</Filter>
     </ClInclude>
+    <ClInclude Include="wmic.hpp">
+      <Filter>Source Files\defender-control\wmic</Filter>
+    </ClInclude>
   </ItemGroup>
 </Project>

src/defender-control/main.cpp

@@ -1,4 +1,5 @@
-#include "dcontrol.h"
+#include "dcontrol.hpp"
+#include "wmic.hpp"
 
 // to-do:
 // write argument parser
@@ -13,9 +14,7 @@ int main()
     "Windows defender is ACTIVE\n" :
     "Windows defender is OFF\n");
 
-  printf(DCONTROL::disable_defender() ?
+
-    "Defender disabled\n" :
-    "Failed to disable\n");
 
   system("pause");
 

src/defender-control/wmic.cpp

@@ -0,0 +1,7 @@
+// WMIC controls for windows defender module (cmdlet)
+//
+#include "wmic.hpp"
+
+namespace wmic
+{
+}

src/defender-control/wmic.hpp

@@ -0,0 +1,5 @@
+#pragma once
+
+namespace wmic
+{
+}