Handle index out of bound exceptions when parsing invalid messages

kapetan · Apr 1, 2018 · 71df004 · 71df004
1 parent edf1eb8
commit 71df004
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/DNS/Client/ClientRequest.cs b/DNS/Client/ClientRequest.cs
@@ -86,6 +86,8 @@ public class ClientRequest : IRequest {
                 return response;
             } catch (ArgumentException e) {
                 throw new ResponseException("Invalid response", e);
+            } catch (IndexOutOfRangeException e) {
+                throw new ResponseException("Invalid response", e);
             }
         }
     }

diff --git a/DNS/Server/DnsServer.cs b/DNS/Server/DnsServer.cs
@@ -143,6 +143,7 @@ await udp
             }
             catch (SocketException e) { OnErrored(e); }
             catch (ArgumentException e) { OnErrored(e); }
+            catch (IndexOutOfRangeException e) { OnErrored(e); }
             catch (OperationCanceledException e) { OnErrored(e); }
             catch (IOException e) { OnErrored(e); }
             catch (ObjectDisposedException e) { OnErrored(e); }