From cddca44d348ac0259a3c71fd5ff4c0276ed1153e Mon Sep 17 00:00:00 2001
From: ruri <ruridev@protonmail.com>
Date: Wed, 24 Apr 2019 01:24:18 +0200
Subject: [PATCH] Implemented WebSocket frames support in TCP block.

---
 OpenBullet/LSDoc.xml                          |  2 +-
 .../Pages/StackerBlocks/PageBlockTCP.xaml     |  1 +
 RuriLib/Blocks/BlockTCP.cs                    | 67 ++++++++++++++++++-
 3 files changed, 66 insertions(+), 4 deletions(-)

diff --git a/OpenBullet/LSDoc.xml b/OpenBullet/LSDoc.xml
index 52b0a86..4853d02 100644
--- a/OpenBullet/LSDoc.xml
+++ b/OpenBullet/LSDoc.xml
@@ -367,7 +367,7 @@ TCP COMMAND [ARGUMENTS]
 
 Allowed Commands:
 CONNECT "HOST" "PORT" [UseSSL?] [-> VAR/CAP "NAME"]
-SEND "MESSAGE" [-> VAR/CAP "NAME"]
+SEND "MESSAGE" [WebSocket?] [-> VAR/CAP "NAME"]
 DISCONNECT
 
 Example:
diff --git a/OpenBullet/Pages/StackerBlocks/PageBlockTCP.xaml b/OpenBullet/Pages/StackerBlocks/PageBlockTCP.xaml
index 34ac6f9..5e3d8a0 100644
--- a/OpenBullet/Pages/StackerBlocks/PageBlockTCP.xaml
+++ b/OpenBullet/Pages/StackerBlocks/PageBlockTCP.xaml
@@ -57,6 +57,7 @@
                         <TextBox Text="{Binding Message, UpdateSourceTrigger=PropertyChanged}"/>
                     </DockPanel>
                     <Label Content="Remember to end your message with \r\n to let the server know it's over!" Foreground="{DynamicResource ForegroundCustom}"/>
+                    <CheckBox Content="WebSocket" IsChecked="{Binding WebSocket}" VerticalContentAlignment="Center"/>
                 </StackPanel>
             </TabItem>
         </TabControl>
diff --git a/RuriLib/Blocks/BlockTCP.cs b/RuriLib/Blocks/BlockTCP.cs
index 55340fa..ce175dc 100644
--- a/RuriLib/Blocks/BlockTCP.cs
+++ b/RuriLib/Blocks/BlockTCP.cs
@@ -6,6 +6,7 @@ using System.Linq;
 using System.Net.Security;
 using System.Net.Sockets;
 using System.Text;
+using System.Threading;
 using System.Threading.Tasks;
 using System.Windows.Media;
 
@@ -47,6 +48,10 @@ namespace RuriLib
         /// <summary>Whether the client will communicate over the Secure Sockets Layer.</summary>
         public bool UseSSL { get { return useSSL; } set { useSSL = value; OnPropertyChanged(); } }
 
+        private bool webSocket = false;
+        /// <summary>Whether to treat the message as a WebSocket payload (adds the frame overhead bytes).</summary>
+        public bool WebSocket { get { return webSocket; } set { webSocket = value; OnPropertyChanged(); } }
+
         private string message = "";
         /// <summary>The message sent to the host.</summary>
         public string Message { get { return message; } set { message = value; OnPropertyChanged(); } }
@@ -94,6 +99,9 @@ namespace RuriLib
 
                 case TCPCommand.Send:
                     Message = LineParser.ParseLiteral(ref input, "Message");
+
+                    if (LineParser.Lookahead(ref input) == TokenType.Boolean)
+                        LineParser.SetBool(ref input, this);
                     break;
 
                 default:
@@ -140,7 +148,8 @@ namespace RuriLib
 
                 case TCPCommand.Send:
                     writer
-                        .Literal(Message);
+                        .Literal(Message)
+                        .Boolean(WebSocket, "WebSocket");
                     break;
             }
 
@@ -185,7 +194,7 @@ namespace RuriLib
                              }
 
                         //Wait a bit and read the Stream if not Empty
-                        System.Threading.Thread.Sleep(200);
+                        Thread.Sleep(200);
                         if (net.DataAvailable)
                         {
                             if (UseSSL) bytes = ssl.Read(buffer, 0, buffer.Length);
@@ -232,7 +241,59 @@ namespace RuriLib
                     }
 
                     var msg = ReplaceValues(Message, data);
-                    var b = Encoding.ASCII.GetBytes(msg.Replace(@"\r\n", "\r\n"));
+                    byte[] b = { };
+                    var payload = Encoding.ASCII.GetBytes(msg.Replace(@"\r\n", "\r\n"));
+
+                    // Manual implementation of the WebSocket frame
+                    if (WebSocket)
+                    {
+                        List<byte> bl = new List<byte>();
+
+                        // (FIN=1) (RSV1=0) (RSV2=0) (RSV3=0) (OPCODE=0001) = 128 + 1 = 129
+                        bl.Add(129);
+
+                        ulong pllen = (ulong)payload.Length;
+
+                        // We add 128 because the mask bit (MSB) is always 1. In this case the payload len is 7 bits long
+                        if (pllen <= 125)
+                        {
+                            bl.Add((byte)(pllen + 128));
+                        }
+
+                        // Payload len set to 126 -> Next 2 bytes are payload len
+                        else if (pllen <= ushort.MaxValue)
+                        {
+                            bl.Add(126 + 128);
+                            bl.Add((byte)(pllen >> 8)); // Shift by 1 byte
+                            bl.Add((byte)(pllen % 255)); // Take LSB
+                        }
+
+                        // Payload len set to 127 -> Next 4 bytes are payload len
+                        else if (pllen <= ulong.MaxValue)
+                        {
+                            bl.Add(127 + 128);
+                            bl.Add((byte)(pllen >> 24)); // Shift by 3 bytes
+                            bl.Add((byte)((pllen >> 16) % 255)); // Shift by 2 bytes and take LSB
+                            bl.Add((byte)((pllen >> 8) % 255)); // Shift by 1 byte and take LSB
+                            bl.Add((byte)(pllen % 255)); // Take LSB
+                        }
+
+                        // Set the mask used for this message
+                        byte[] mask = new byte[4] { 61, 84, 35, 6 };
+                        bl.AddRange(mask);
+
+                        // Finally we add the payload XORed with the mask
+                        for (int i = 0; i < payload.Length; i++)
+                        {
+                            bl.Add((byte)(payload[i] ^ mask[i % 4]));
+                        }
+
+                        b = bl.ToArray();
+                    }
+                    else
+                    {
+                        b = payload;
+                    }
                     data.Log(new LogEntry("> " + msg, Colors.White));
 
                     if (data.TCPSSL)