wip

Eyepea · May 7, 2018 · 29e15fa · ovv · May 7, 2018 · ovv
1 parent 8a2a81e
commit 29e15fa
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 1 deletion.
diff --git a/aiosip/application.py b/aiosip/application.py
@@ -128,6 +128,21 @@ async def prepare(self, status_code, *args, **kwargs):
 
                 return dialog
 
+            async def unauthorized(self, msg):
+                dialog = self._create_dialog()
+                await dialog.unauthorized(msg)
+                return dialog
+
+            def validate_auth(self, msg, password):
+                dialog = self._create_dialog()
+                if msg.auth and msg.auth.validate(password, dialog._nonce):
+                    dialog._nonce = None
+                    return True
+                elif msg.method == 'CANCEL':
+                    return True
+                else:
+                    return False
+
         request = Request()
         await route(request, msg)
 

diff --git a/aiosip/dialog.py b/aiosip/dialog.py
@@ -64,7 +64,7 @@ def dialog_id(self):
     def _receive_response(self, msg):
 
         if 'tag' not in self.to_details['params']:
-            del self.app._dialogs[self.dialog_id]
+            # del self.app._dialogs[self.dialog_id]
             self.to_details['params']['tag'] = msg.to_details['params']['tag']
             self.app._dialogs[self.dialog_id] = self
 

diff --git a/aiosip/peers.py b/aiosip/peers.py
@@ -78,6 +78,9 @@ def _create_dialog(self, method, from_details, to_details, contact_details=None,
 
         LOG.debug('Creating: %s', dialog)
         self._app._dialogs[dialog.dialog_id] = dialog
+        self._app._dialogs[
+            frozenset((dialog.original_msg.to_details['params'].get('tag'), None, dialog.call_id))
+        ] = dialog
         return dialog
 
     async def request(self, method, from_details, to_details, contact_details=None, password=None, call_id=None,