From ed6c980665ca79caa4bfcbec661db338c1c7ae7c Mon Sep 17 00:00:00 2001
From: Oleksii Shevchuk <alxchk@gmail.com>
Date: Thu, 15 Nov 2018 19:13:24 +0200
Subject: [PATCH] Fix https://github.com/n1nj4sec/pupy/issues/710

---
 pupy/pupylib/PupyModule.py | 2 +-
 pupy/pupylib/PupyWeb.py    | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/pupy/pupylib/PupyModule.py b/pupy/pupylib/PupyModule.py
index 86ea720b..b01dc82e 100644
--- a/pupy/pupylib/PupyModule.py
+++ b/pupy/pupylib/PupyModule.py
@@ -448,7 +448,7 @@ class PupyModule(object):
         if not self.client.pupsrv.start_webserver():
             return None
         else:
-            return self.client.pupsrv.pupweb.start_webplugin(self.web_handlers)
+            return self.client.pupsrv.pupweb.start_webplugin('rdesktop', self.web_handlers)
 
     @classmethod
     def is_compatible_with(cls, client):
diff --git a/pupy/pupylib/PupyWeb.py b/pupy/pupylib/PupyWeb.py
index f6b263df..4af3103b 100644
--- a/pupy/pupylib/PupyWeb.py
+++ b/pupy/pupylib/PupyWeb.py
@@ -56,7 +56,7 @@ class RequestHandler(TornadoRequestHandler):
             self.finish(log_msg)
             return
 
-        super(WebSocketHandler, self).prepare(*args, **kwargs)
+        super(RequestHandler, self).prepare(*args, **kwargs)
 
 class StaticTextHandler(TornadoRequestHandler):
     def set_default_headers(self):
@@ -214,7 +214,7 @@ class PupyWebServer(object):
         self._ioloop = None
         self._thread = None
 
-        for _, _, cleanup in self._registered:
+        for (_, _, cleanup) in self._registered.itervalues():
             if cleanup:
                 cleanup()