fake server: fix screen not initialized, add log terminating statement

GoSecure · Dec 20, 2022 · 5ed64f6 · 5ed64f6
1 parent f7583f9
commit 5ed64f6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyrdp/mitm/FakeServer.py b/pyrdp/mitm/FakeServer.py
@@ -308,7 +308,7 @@ def set_username(self, username: str):
             time.sleep(0.1)
         if self.fakeLoginScreen is not None:
             self.fakeLoginScreen.set_username(username)
-        self.check_submit()
+            self.check_submit()
 
     def set_password(self, password: str):
         self.fakeLoginScreen.set_password(password)
@@ -318,5 +318,6 @@ def terminate(self):
         # TODO: the user sees "An internal error has occurred."
         for proc in (self.rdp_server_process, self.rdp_client_process):
             if not isinstance(proc, subprocess.CompletedProcess):
+                self.log.info("Killing process %(name)s", {"name": " ".join(proc.args)})
                 proc.kill()
         self.display.stop()