embeddable-dll-service: add more robust example for .NET 5

Signed-off-by: Jason A. Donenfeld <Jason@zx2c4.com>

5 files changed, 452 insertions, 0 deletions

diff --git a/embeddable-dll-service/csharp/DemoUI/MainWindow.Designer.cs b/embeddable-dll-service/csharp/DemoUI/MainWindow.Designer.cs
new file mode 100644
index 00000000..5fae35d3
--- /dev/null
+++ b/embeddable-dll-service/csharp/DemoUI/MainWindow.Designer.cs
@@ -0,0 +1,87 @@
+/* SPDX-License-Identifier: MIT
+ *
+ * Copyright (C) 2019-2021 WireGuard LLC. All Rights Reserved.
+ */
+
+namespace DemoUI
+{
+    partial class MainWindow
+    {
+        /// <summary>
+        ///  Required designer variable.
+        /// </summary>
+        private System.ComponentModel.IContainer components = null;
+
+        /// <summary>
+        ///  Clean up any resources being used.
+        /// </summary>
+        /// <param name="disposing">true if managed resources should be disposed; otherwise, false.</param>
+        protected override void Dispose(bool disposing)
+        {
+            if (disposing && (components != null))
+            {
+                components.Dispose();
+            }
+            base.Dispose(disposing);
+        }
+
+        #region Windows Form Designer generated code
+
+        /// <summary>
+        ///  Required method for Designer support - do not modify
+        ///  the contents of this method with the code editor.
+        /// </summary>
+        private void InitializeComponent()
+        {
+            this.connectButton = new System.Windows.Forms.Button();
+            this.logBox = new System.Windows.Forms.TextBox();
+            this.SuspendLayout();
+            // 
+            // connectButton
+            // 
+            this.connectButton.Anchor = ((System.Windows.Forms.AnchorStyles)(((System.Windows.Forms.AnchorStyles.Top | System.Windows.Forms.AnchorStyles.Left) 
+            | System.Windows.Forms.AnchorStyles.Right)));
+            this.connectButton.Location = new System.Drawing.Point(12, 12);
+            this.connectButton.Name = "connectButton";
+            this.connectButton.Size = new System.Drawing.Size(1137, 46);
+            this.connectButton.TabIndex = 5;
+            this.connectButton.Text = "&Connect";
+            this.connectButton.UseVisualStyleBackColor = true;
+            this.connectButton.Click += new System.EventHandler(this.connectButton_Click);
+            // 
+            // logBox
+            // 
+            this.logBox.Anchor = ((System.Windows.Forms.AnchorStyles)((((System.Windows.Forms.AnchorStyles.Top | System.Windows.Forms.AnchorStyles.Bottom) 
+            | System.Windows.Forms.AnchorStyles.Left) 
+            | System.Windows.Forms.AnchorStyles.Right)));
+            this.logBox.Location = new System.Drawing.Point(12, 64);
+            this.logBox.Multiline = true;
+            this.logBox.Name = "logBox";
+            this.logBox.ReadOnly = true;
+            this.logBox.ScrollBars = System.Windows.Forms.ScrollBars.Vertical;
+            this.logBox.Size = new System.Drawing.Size(1137, 561);
+            this.logBox.TabIndex = 4;
+            this.logBox.TabStop = false;
+            // 
+            // MainWindow
+            // 
+            this.AutoScaleDimensions = new System.Drawing.SizeF(13F, 32F);
+            this.AutoScaleMode = System.Windows.Forms.AutoScaleMode.Font;
+            this.ClientSize = new System.Drawing.Size(1161, 637);
+            this.Controls.Add(this.logBox);
+            this.Controls.Add(this.connectButton);
+            this.Name = "MainWindow";
+            this.Text = "WireGuard Demo Client";
+            this.FormClosing += new System.Windows.Forms.FormClosingEventHandler(this.MainWindow_FormClosing);
+            this.Load += new System.EventHandler(this.MainWindow_Load);
+            this.ResumeLayout(false);
+            this.PerformLayout();
+
+        }
+
+        #endregion
+        private System.Windows.Forms.Button connectButton;
+        private System.Windows.Forms.TextBox logBox;
+    }
+}
+
diff --git a/embeddable-dll-service/csharp/DemoUI/MainWindow.cs b/embeddable-dll-service/csharp/DemoUI/MainWindow.cs
new file mode 100644
index 00000000..13499fcb
--- /dev/null
+++ b/embeddable-dll-service/csharp/DemoUI/MainWindow.cs
@@ -0,0 +1,213 @@
+/* SPDX-License-Identifier: MIT
+ *
+ * Copyright (C) 2019-2021 WireGuard LLC. All Rights Reserved.
+ */
+
+using System;
+using System.Text;
+using System.Threading.Tasks;
+using System.IO;
+using System.Windows.Forms;
+using System.Threading;
+using System.IO.Pipes;
+using System.Diagnostics;
+using System.Net.Sockets;
+
+namespace DemoUI
+{
+    public partial class MainWindow : Form
+    {
+        private static readonly string userDirectory = Path.GetDirectoryName(Process.GetCurrentProcess().MainModule.FileName);
+        private static readonly string configFile = Path.Combine(userDirectory, "demobox.conf");
+        private static readonly string logFile = Path.Combine(userDirectory, "log.bin");
+
+        private Tunnel.Ringlogger log;
+        private Thread logPrintingThread, transferUpdateThread;
+        private volatile bool threadsRunning;
+        private bool connected;
+
+        public MainWindow()
+        {
+            InitializeComponent();
+            Application.ApplicationExit += Application_ApplicationExit;
+
+            try { File.Delete(logFile); } catch { }
+            log = new Tunnel.Ringlogger(logFile, "GUI");
+            logPrintingThread = new Thread(new ThreadStart(tailLog));
+            transferUpdateThread = new Thread(new ThreadStart(tailTransfer));
+        }
+
+        private void tailLog()
+        {
+            var cursor = Tunnel.Ringlogger.CursorAll;
+            while (threadsRunning)
+            {
+                var lines = log.FollowFromCursor(ref cursor);
+                foreach (var line in lines)
+                    logBox.Invoke(new Action<string>(logBox.AppendText), new object[] { line + "\r\n" });
+                try
+                {
+                    Thread.Sleep(300);
+                }
+                catch
+                {
+                    break;
+                }
+            }
+        }
+
+        private void tailTransfer()
+        {
+            NamedPipeClientStream stream = null;
+            try
+            {
+                while (threadsRunning)
+                {
+                    while (threadsRunning)
+                    {
+                        try
+                        {
+                            stream = Tunnel.Service.GetPipe(configFile);
+                            stream.Connect();
+                            break;
+                        }
+                        catch { }
+                        Thread.Sleep(1000);
+                    }
+
+                    var reader = new StreamReader(stream);
+                    stream.Write(Encoding.UTF8.GetBytes("get=1\n\n"));
+                    ulong rx = 0, tx = 0;
+                    while (threadsRunning)
+                    {
+                        var line = reader.ReadLine();
+                        if (line == null)
+                            break;
+                        line = line.Trim();
+                        if (line.Length == 0)
+                            break;
+                        if (line.StartsWith("rx_bytes="))
+                            rx += ulong.Parse(line.Substring(9));
+                        else if (line.StartsWith("tx_bytes="))
+                            tx += ulong.Parse(line.Substring(9));
+                    }
+                    Invoke(new Action<ulong, ulong>(updateTransferTitle), new object[] { rx, tx });
+                    stream.Close();
+                    Thread.Sleep(1000);
+                }
+            }
+            catch { }
+            finally
+            {
+                if (stream != null && stream.IsConnected)
+                    stream.Close();
+            }
+        }
+
+        private void Application_ApplicationExit(object sender, EventArgs e)
+        {
+            Tunnel.Service.Remove(configFile, true);
+            try { File.Delete(logFile); } catch { }
+            try { File.Delete(configFile); } catch { }
+        }
+
+        private void MainWindow_Load(object sender, EventArgs e)
+        {
+            threadsRunning = true;
+            logPrintingThread.Start();
+            transferUpdateThread.Start();
+        }
+
+        private void MainWindow_FormClosing(object sender, FormClosingEventArgs e)
+        {
+            threadsRunning = false;
+            logPrintingThread.Interrupt();
+            transferUpdateThread.Interrupt();
+            try { logPrintingThread.Join(); } catch { }
+            try { transferUpdateThread.Join(); } catch { }
+        }
+
+        private static string formatBytes(ulong bytes)
+        {
+            decimal d = bytes;
+            string selectedUnit = null;
+            foreach (string unit in new string[] { "B", "KiB", "MiB", "GiB", "TiB" })
+            {
+                selectedUnit = unit;
+                if (d < 1024)
+                    break;
+                d /= 1024;
+            }
+            return string.Format("{0:0.##} {1}", d, selectedUnit);
+        }
+
+        private void updateTransferTitle(ulong rx, ulong tx)
+        {
+            var titleBase = Text;
+            var idx = titleBase.IndexOf(" - ");
+            if (idx != -1)
+                titleBase = titleBase.Substring(0, idx);
+            if (rx == 0 && tx == 0)
+                Text = titleBase;
+            else
+                Text = string.Format("{0} - rx: {1}, tx: {2}", titleBase, formatBytes(rx), formatBytes(tx));
+        }
+
+        private async Task<string> generateNewConfig()
+        {
+            log.Write("Generating keys");
+            var keys = Tunnel.Keypair.Generate();
+            log.Write("Exchanging keys with demo server");
+            var client = new TcpClient();
+            await client.ConnectAsync("demo.wireguard.com", 42912);
+            var stream = client.GetStream();
+            var reader = new StreamReader(stream, Encoding.UTF8);
+            var pubKeyBytes = Encoding.UTF8.GetBytes(keys.Public + "\n");
+            await stream.WriteAsync(pubKeyBytes, 0, pubKeyBytes.Length);
+            await stream.FlushAsync();
+            var ret = (await reader.ReadLineAsync()).Split(':');
+            client.Close();
+            var status = ret.Length >= 1 ? ret[0] : "";
+            var serverPubkey = ret.Length >= 2 ? ret[1] : "";
+            var serverPort = ret.Length >= 3 ? ret[2] : "";
+            var internalIP = ret.Length >= 4 ? ret[3] : "";
+            if (status != "OK")
+                throw new InvalidOperationException(string.Format("Server status is {0}", status));
+            return string.Format("[Interface]\nPrivateKey = {0}\nAddress = {1}/24\nDNS = 8.8.8.8, 8.8.4.4\n\n[Peer]\nPublicKey = {2}\nEndpoint = demo.wireguard.com:{3}\nAllowedIPs = 0.0.0.0/0\n", keys.Private, internalIP, serverPubkey, serverPort);
+        }
+
+        private async void connectButton_Click(object sender, EventArgs e)
+        {
+            if (connected)
+            {
+                connectButton.Enabled = false;
+                await Task.Run(() =>
+                {
+                    Tunnel.Service.Remove(configFile, true);
+                    try { File.Delete(configFile); } catch { }
+                });
+                updateTransferTitle(0, 0);
+                connectButton.Text = "&Connect";
+                connectButton.Enabled = true;
+                connected = false;
+                return;
+            }
+
+            connectButton.Enabled = false;
+            try
+            {
+                var config = await generateNewConfig();
+                await File.WriteAllBytesAsync(configFile, Encoding.UTF8.GetBytes(config));
+                await Task.Run(() => Tunnel.Service.Add(configFile, true));
+                connected = true;
+                connectButton.Text = "&Disconnect";
+            }
+            catch (Exception ex)
+            {
+                log.Write(ex.Message);
+                try { File.Delete(configFile); } catch { }
+            }
+            connectButton.Enabled = true;
+        }
+    }
+}
diff --git a/embeddable-dll-service/csharp/DemoUI/MainWindow.resx b/embeddable-dll-service/csharp/DemoUI/MainWindow.resx
new file mode 100644
index 00000000..f298a7be
--- /dev/null
+++ b/embeddable-dll-service/csharp/DemoUI/MainWindow.resx
@@ -0,0 +1,60 @@
+<root>
+  <xsd:schema id="root" xmlns="" xmlns:xsd="http://www.w3.org/2001/XMLSchema" xmlns:msdata="urn:schemas-microsoft-com:xml-msdata">
+    <xsd:import namespace="http://www.w3.org/XML/1998/namespace" />
+    <xsd:element name="root" msdata:IsDataSet="true">
+      <xsd:complexType>
+        <xsd:choice maxOccurs="unbounded">
+          <xsd:element name="metadata">
+            <xsd:complexType>
+              <xsd:sequence>
+                <xsd:element name="value" type="xsd:string" minOccurs="0" />
+              </xsd:sequence>
+              <xsd:attribute name="name" use="required" type="xsd:string" />
+              <xsd:attribute name="type" type="xsd:string" />
+              <xsd:attribute name="mimetype" type="xsd:string" />
+              <xsd:attribute ref="xml:space" />
+            </xsd:complexType>
+          </xsd:element>
+          <xsd:element name="assembly">
+            <xsd:complexType>
+              <xsd:attribute name="alias" type="xsd:string" />
+              <xsd:attribute name="name" type="xsd:string" />
+            </xsd:complexType>
+          </xsd:element>
+          <xsd:element name="data">
+            <xsd:complexType>
+              <xsd:sequence>
+                <xsd:element name="value" type="xsd:string" minOccurs="0" msdata:Ordinal="1" />
+                <xsd:element name="comment" type="xsd:string" minOccurs="0" msdata:Ordinal="2" />
+              </xsd:sequence>
+              <xsd:attribute name="name" type="xsd:string" use="required" msdata:Ordinal="1" />
+              <xsd:attribute name="type" type="xsd:string" msdata:Ordinal="3" />
+              <xsd:attribute name="mimetype" type="xsd:string" msdata:Ordinal="4" />
+              <xsd:attribute ref="xml:space" />
+            </xsd:complexType>
+          </xsd:element>
+          <xsd:element name="resheader">
+            <xsd:complexType>
+              <xsd:sequence>
+                <xsd:element name="value" type="xsd:string" minOccurs="0" msdata:Ordinal="1" />
+              </xsd:sequence>
+              <xsd:attribute name="name" type="xsd:string" use="required" />
+            </xsd:complexType>
+          </xsd:element>
+        </xsd:choice>
+      </xsd:complexType>
+    </xsd:element>
+  </xsd:schema>
+  <resheader name="resmimetype">
+    <value>text/microsoft-resx</value>
+  </resheader>
+  <resheader name="version">
+    <value>2.0</value>
+  </resheader>
+  <resheader name="reader">
+    <value>System.Resources.ResXResourceReader, System.Windows.Forms, Version=4.0.0.0, Culture=neutral, PublicKeyToken=b77a5c561934e089</value>
+  </resheader>
+  <resheader name="writer">
+    <value>System.Resources.ResXResourceWriter, System.Windows.Forms, Version=4.0.0.0, Culture=neutral, PublicKeyToken=b77a5c561934e089</value>
+  </resheader>
+</root>
\ No newline at end of file
diff --git a/embeddable-dll-service/csharp/DemoUI/Program.cs b/embeddable-dll-service/csharp/DemoUI/Program.cs
new file mode 100644
index 00000000..8c420973
--- /dev/null
+++ b/embeddable-dll-service/csharp/DemoUI/Program.cs
@@ -0,0 +1,44 @@
+/* SPDX-License-Identifier: MIT
+ *
+ * Copyright (C) 2019-2021 WireGuard LLC. All Rights Reserved.
+ */
+
+using System;
+using System.Threading;
+using System.Diagnostics;
+using System.Windows.Forms;
+
+namespace DemoUI
+{
+    static class Program
+    {
+        [STAThread]
+        static void Main(string[] args)
+        {
+            if (args.Length == 3 && args[0] == "/service")
+            {
+                var t = new Thread(() =>
+                {
+                    try
+                    {
+                        var currentProcess = Process.GetCurrentProcess();
+                        var uiProcess = Process.GetProcessById(int.Parse(args[2]));
+                        if (uiProcess.MainModule.FileName != currentProcess.MainModule.FileName)
+                            return;
+                        uiProcess.WaitForExit();
+                        Tunnel.Service.Remove(args[1], false);
+                    }
+                    catch { }
+                });
+                t.Start();
+                Tunnel.Service.Run(args[1]);
+                t.Interrupt();
+                return;
+            }
+            Application.SetHighDpiMode(HighDpiMode.SystemAware);
+            Application.EnableVisualStyles();
+            Application.SetCompatibleTextRenderingDefault(false);
+            Application.Run(new MainWindow());
+        }
+    }
+}
diff --git a/embeddable-dll-service/csharp/DemoUI/app.manifest b/embeddable-dll-service/csharp/DemoUI/app.manifest
new file mode 100644
index 00000000..616ab5b9
--- /dev/null
+++ b/embeddable-dll-service/csharp/DemoUI/app.manifest
@@ -0,0 +1,48 @@
+<?xml version="1.0" encoding="utf-8"?>
+<assembly manifestVersion="1.0" xmlns="urn:schemas-microsoft-com:asm.v1">
+  <assemblyIdentity version="1.0.0.0" name="demo-client"/>
+  <trustInfo xmlns="urn:schemas-microsoft-com:asm.v2">
+    <security>
+      <requestedPrivileges xmlns="urn:schemas-microsoft-com:asm.v3">
+        <requestedExecutionLevel level="requireAdministrator" uiAccess="false" />
+      </requestedPrivileges>
+    </security>
+  </trustInfo>
+
+  <compatibility xmlns="urn:schemas-microsoft-com:compatibility.v1">
+    <application>
+      <!-- Windows 7 -->
+      <supportedOS Id="{35138b9a-5d96-4fbd-8e2d-a2440225f93a}" />
+
+      <!-- Windows 8 -->
+      <supportedOS Id="{4a2f28e3-53b9-4441-ba9c-d69d4a4a6e38}" />
+
+      <!-- Windows 8.1 -->
+      <supportedOS Id="{1f676c76-80e1-4239-95bb-83d0f6d0da78}" />
+
+      <!-- Windows 10 -->
+      <supportedOS Id="{8e0f7a12-bfb3-4fe8-b9a5-48fd50a15a9a}" />
+
+    </application>
+  </compatibility>
+
+  <application xmlns="urn:schemas-microsoft-com:asm.v3">
+    <windowsSettings>
+      <dpiAware xmlns="http://schemas.microsoft.com/SMI/2005/WindowsSettings">true</dpiAware>
+    </windowsSettings>
+  </application>
+
+  <dependency>
+    <dependentAssembly>
+      <assemblyIdentity
+          type="win32"
+          name="Microsoft.Windows.Common-Controls"
+          version="6.0.0.0"
+          processorArchitecture="*"
+          publicKeyToken="6595b64144ccf1df"
+          language="*"
+        />
+    </dependentAssembly>
+  </dependency>
+
+</assembly>