Windows - support DLLs

pwnshop/challenges/__init__.py

@@ -329,6 +329,7 @@ class WindowsChallenge(Challenge):
     CFG = True
     DYNAMIC_BASE = True
     ASLR_HIGH_ENTROPY = True
+    MAKE_DLL = False
 
     def build_compiler_cmd(self):
         cmd = [self.COMPILER]
@@ -349,6 +350,9 @@ def build_compiler_cmd(self):
         else:
             cmd.append("/guard:cf-")
 
+        if self.MAKE_DLL:
+            cmd.append("/LD")
+
         return cmd
         # Linker options
         cmd.append('/LINK')
@@ -374,7 +378,11 @@ def build_binary(self, source=None):
         if self.build_image is None:
             with tempfile.TemporaryDirectory(prefix='pwnshop-') as workdir:
                 src_path = f"{workdir}/{self.__class__.__name__.lower()}.c"
-                bin_path = f"{workdir}/{self.__class__.__name__.lower()}.exe"
+
+                if self.MAKE_DLL:
+                    bin_path = f"{workdir}/{self.__class__.__name__.lower()}.dll"
+                else:
+                    bin_path = f"{workdir}/{self.__class__.__name__.lower()}.exe"
                 pdb_path = f"{workdir}/{self.__class__.__name__.lower()}.pdb"
                 with open(src_path, 'w') as f:
                     f.write(source)