From a1e1922e62419de75d2909dfae1ee7caa9ed487f Mon Sep 17 00:00:00 2001
From: Ben Harris <ben@tilde.team>
Date: Sat, 16 May 2020 22:08:40 -0400
Subject: Add xmldocs, and handle null bytes

---
 IRCTokens/StatefulDecoder.cs | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'IRCTokens/StatefulDecoder.cs')

diff --git a/IRCTokens/StatefulDecoder.cs b/IRCTokens/StatefulDecoder.cs
index 82630f6..619253b 100644
--- a/IRCTokens/StatefulDecoder.cs
+++ b/IRCTokens/StatefulDecoder.cs
@@ -59,6 +59,15 @@ namespace IRCTokens
 
             _buffer = _buffer == null ? Array.Empty<byte>() : _buffer.Concat(data.Take(bytesReceived)).ToArray();
 
+            // truncate message at NUL if found
+            if (_buffer.Contains((byte) 0))
+            {
+                _buffer = _buffer
+                    .Take(Array.IndexOf(_buffer, (byte) 0))
+                    .Concat(new []{(byte) '\n'})
+                    .ToArray();
+            }
+
             var listLines = _buffer.Split((byte) '\n').Select(l => l.Trim((byte) '\r')).ToList();
             _buffer = listLines.LastOrDefault() ?? Array.Empty<byte>();
 
-- 
cgit 1.4.1