From d05cac8b6c4bfa0751722b41d4fc97a88f116bcf Mon Sep 17 00:00:00 2001
From: Chris Boden <cboden@gmail.com>
Date: Fri, 11 May 2012 11:59:18 -0400
Subject: [PATCH] [Io] Handle Application Exceptions

---
 src/Ratchet/Server/IoServer.php | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/Ratchet/Server/IoServer.php b/src/Ratchet/Server/IoServer.php
index 7c80f18..daf9b12 100644
--- a/src/Ratchet/Server/IoServer.php
+++ b/src/Ratchet/Server/IoServer.php
@@ -75,11 +75,19 @@ class IoServer {
     }
 
     public function handleData($data, $conn) {
-        $this->app->onMessage($conn->decor, $data);
+        try {
+            $this->app->onMessage($conn->decor, $data);
+        } catch (\Exception $e) {
+            $this->handleError($e, $conn);
+        }
     }
 
     public function handleEnd($conn) {
-        $this->app->onClose($conn->decor);
+        try {
+            $this->app->onClose($conn->decor);
+        } catch (\Exception $e) {
+            $this->handleError($e, $conn);
+        }
     }
 
     public function handleError(\Exception $e, $conn) {