fix another case where cast clients could unexpectedly exit

1 files changed, 8 insertions, 2 deletions

diff --git a/src/cmd/cast.rs b/src/cmd/cast.rs
index 8cbbd96..5f8c44a 100644
--- a/src/cmd/cast.rs
+++ b/src/cmd/cast.rs
@@ -111,6 +111,8 @@ impl CastSession {
         &Self::poll_write_server,
     ];
 
+    // this should never return Err, because we don't want server
+    // communication issues to ever interrupt a running process
     fn poll_read_client(
         &mut self,
     ) -> Result<crate::component_future::Poll<()>> {
@@ -120,8 +122,12 @@ impl CastSession {
                     self.sent_remote = 0;
                     Ok(crate::component_future::Poll::DidWork)
                 }
-                crate::client::Event::ServerMessage(msg) => {
-                    Err(Error::UnexpectedMessage { message: msg })
+                crate::client::Event::ServerMessage(..) => {
+                    // we don't expect to ever see a server message once we
+                    // start casting, so if one comes through, assume
+                    // something is messed up and try again
+                    self.client.reconnect();
+                    Ok(crate::component_future::Poll::DidWork)
                 }
             },
             Ok(futures::Async::Ready(None)) => {