🎒 CLI: legacy backend

zedifen · Mar 6, 2021 · a8f0f58 · a8f0f58
1 parent 77d49cf
commit a8f0f58
Show file tree

Hide file tree

Showing 7 changed files with 172 additions and 19 deletions.
diff --git a/Shadowsocks.CLI/Backend.cs b/Shadowsocks.CLI/Backend.cs
@@ -0,0 +1,10 @@
+namespace Shadowsocks.CLI
+{
+    public enum Backend
+    {
+        SsRust,
+        V2Ray,
+        Legacy,
+        Pipelines,
+    }
+}
diff --git a/Shadowsocks.CLI/Client/Legacy.cs b/Shadowsocks.CLI/Client/Legacy.cs
@@ -0,0 +1,57 @@
+using Shadowsocks.Models;
+using Shadowsocks.Net;
+using System;
+using System.Collections.Generic;
+using System.Diagnostics;
+using System.Linq;
+using System.Net;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace Shadowsocks.CLI.Client
+{
+    public class Legacy
+    {
+        private TCPListener? _tcpListener;
+        private UDPListener? _udpListener;
+
+        public void Start(string listenSocks, string serverAddress, int serverPort, string method, string password, string? plugin, string? pluginOpts, string? pluginArgs)
+        {
+            var localEP = IPEndPoint.Parse(listenSocks);
+            var server = new Server()
+            {
+                Host = serverAddress,
+                Port = serverPort,
+                Method = method,
+                Password = password,
+                Plugin = plugin,
+                PluginOpts = pluginOpts,
+            };
+            if (!string.IsNullOrEmpty(plugin) && !string.IsNullOrEmpty(pluginArgs))
+            {
+                var processStartInfo = new ProcessStartInfo(plugin, pluginArgs);
+                server.PluginArgs = processStartInfo.ArgumentList.ToList();
+            }
+
+            var tcpRelay = new TCPRelay(server);
+            _tcpListener = new TCPListener(localEP, new List<IStreamService>()
+                        {
+                            tcpRelay,
+                        });
+            _tcpListener.Start();
+
+            var udpRelay = new UDPRelay(server);
+            _udpListener = new UDPListener(localEP, new List<IDatagramService>()
+                        {
+                            udpRelay,
+                        });
+            _udpListener.Start();
+        }
+
+        public void Stop()
+        {
+            _tcpListener?.Stop();
+            _udpListener?.Stop();
+        }
+    }
+}
diff --git a/Shadowsocks.CLI/Client/Pipelines.cs b/Shadowsocks.CLI/Client/Pipelines.cs
@@ -0,0 +1,29 @@
+using Shadowsocks.Protocol;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Net;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace Shadowsocks.CLI.Client
+{
+    public class Pipelines
+    {
+        private TcpPipeListener? _tcpPipeListener;
+
+        public Task Start(string listenSocks, string serverAddress, int serverPort, string method, string? password, string? key, string? plugin, string? pluginOpts, string? pluginArgs)
+        {
+            // TODO
+            var localEP = IPEndPoint.Parse(listenSocks);
+            var remoteEp = new DnsEndPoint(serverAddress, serverPort);
+            byte[]? mainKey = null;
+            if (!string.IsNullOrEmpty(key))
+                mainKey = Encoding.UTF8.GetBytes(key);
+            _tcpPipeListener = new(localEP);
+            return _tcpPipeListener.Start(localEP, remoteEp, method, password, mainKey);
+        }
+
+        public void Stop() => _tcpPipeListener?.Stop();
+    }
+}
diff --git a/Shadowsocks.CLI/Program.cs b/Shadowsocks.CLI/Program.cs
@@ -1,8 +1,11 @@
-using Shadowsocks.Protocol;
+using Shadowsocks.Models;
+using Splat;
 using System;
 using System.Collections.Generic;
 using System.CommandLine;
 using System.CommandLine.Invocation;
+using System.Diagnostics;
+using System.Linq;
 using System.Net;
 using System.Text;
 using System.Threading;
@@ -16,28 +19,81 @@ private static Task<int> Main(string[] args)
         {
             var clientCommand = new Command("client", "Shadowsocks client.");
             clientCommand.AddAlias("c");
-            clientCommand.AddOption(new Option<string?>("--listen", "The address and port to listen on for both SOCKS5 and HTTP proxy."));
-            clientCommand.AddOption(new Option<string?>("--listen-socks", "The address and port to listen on for SOCKS5 proxy."));
-            clientCommand.AddOption(new Option<string?>("--listen-http", "The address and port to listen on for HTTP proxy."));
+            clientCommand.AddOption(new Option<Backend>("--backend", "Shadowsocks backend to use. Available backends: shadowsocks-rust, v2ray, legacy, pipelines."));
+            clientCommand.AddOption(new Option<string?>("--listen", "Address and port to listen on for both SOCKS5 and HTTP proxy."));
+            clientCommand.AddOption(new Option<string?>("--listen-socks", "Address and port to listen on for SOCKS5 proxy."));
+            clientCommand.AddOption(new Option<string?>("--listen-http", "Address and port to listen on for HTTP proxy."));
             clientCommand.AddOption(new Option<string>("--server-address", "Address of the remote Shadowsocks server to connect to."));
             clientCommand.AddOption(new Option<int>("--server-port", "Port of the remote Shadowsocks server to connect to."));
-            clientCommand.AddOption(new Option<string>("--method", "Encryption method to use for the remote Shadowsocks server."));
-            clientCommand.AddOption(new Option<string?>("--password", "Password to use for the remote Shadowsocks server."));
-            clientCommand.AddOption(new Option<string?>("--key", "Encryption key (NOT password!) to use for the remote Shadowsocks server."));
+            clientCommand.AddOption(new Option<string>("--method", "Encryption method to use for remote Shadowsocks server."));
+            clientCommand.AddOption(new Option<string?>("--password", "Password to use for remote Shadowsocks server."));
+            clientCommand.AddOption(new Option<string?>("--key", "Encryption key (NOT password!) to use for remote Shadowsocks server."));
             clientCommand.AddOption(new Option<string?>("--plugin", "Plugin binary path."));
             clientCommand.AddOption(new Option<string?>("--plugin-opts", "Plugin options."));
             clientCommand.AddOption(new Option<string?>("--plugin-args", "Plugin startup arguments."));
             clientCommand.Handler = CommandHandler.Create(
-                async (string? listen, string? listenSocks, string? listenHttp, string serverAddress, int serverPort, string method, string? password, string? key, string? plugin, string? pluginOpts, string? pluginArgs) =>
+                async (Backend backend, string? listen, string? listenSocks, string? listenHttp, string serverAddress, int serverPort, string method, string? password, string? key, string? plugin, string? pluginOpts, string? pluginArgs, CancellationToken cancellationToken) =>
                 {
-                    // TODO
-                    var localEP = IPEndPoint.Parse(listenSocks);
-                    var remoteEp = new DnsEndPoint(serverAddress, serverPort);
-                    byte[]? mainKey = null;
-                    if (!string.IsNullOrEmpty(key))
-                        mainKey = Encoding.UTF8.GetBytes(key);
-                    var tcpPipeListener = new TcpPipeListener(localEP);
-                    tcpPipeListener.Start(localEP, remoteEp, method, password, mainKey).Wait();
+                    Locator.CurrentMutable.RegisterConstant<ConsoleLogger>(new());
+                    if (string.IsNullOrEmpty(listenSocks))
+                    {
+                        LogHost.Default.Error("You must specify SOCKS5 listen address and port.");
+                        return;
+                    }
+
+                    Client.Legacy? legacyClient = null;
+                    Client.Pipelines? pipelinesClient = null;
+
+                    switch (backend)
+                    {
+                        case Backend.SsRust:
+                            LogHost.Default.Error("Not implemented.");
+                            break;
+                        case Backend.V2Ray:
+                            LogHost.Default.Error("Not implemented.");
+                            break;
+                        case Backend.Legacy:
+                            if (!string.IsNullOrEmpty(password))
+                            {
+                                legacyClient = new();
+                                legacyClient.Start(listenSocks, serverAddress, serverPort, method, password, plugin, pluginOpts, pluginArgs);
+                            }
+                            else
+                                LogHost.Default.Error("The legacy backend requires password.");
+                            break;
+                        case Backend.Pipelines:
+                            pipelinesClient = new();
+                            await pipelinesClient.Start(listenSocks, serverAddress, serverPort, method, password, key, plugin, pluginOpts, pluginArgs);
+                            break;
+                        default:
+                            LogHost.Default.Error("Not implemented.");
+                            break;
+                    }
+
+                    while (!cancellationToken.IsCancellationRequested)
+                    {
+                        await Task.Delay(TimeSpan.FromHours(1.00), cancellationToken);
+                        Console.WriteLine("An hour has passed.");
+                    }
+
+                    switch (backend)
+                    {
+                        case Backend.SsRust:
+                            LogHost.Default.Error("Not implemented.");
+                            break;
+                        case Backend.V2Ray:
+                            LogHost.Default.Error("Not implemented.");
+                            break;
+                        case Backend.Legacy:
+                            legacyClient?.Stop();
+                            break;
+                        case Backend.Pipelines:
+                            pipelinesClient?.Stop();
+                            break;
+                        default:
+                            LogHost.Default.Error("Not implemented.");
+                            break;
+                    }
                 });
 
             var serverCommand = new Command("server", "Shadowsocks server.");

diff --git a/Shadowsocks.CLI/Shadowsocks.CLI.csproj b/Shadowsocks.CLI/Shadowsocks.CLI.csproj
@@ -34,6 +34,7 @@
 
   <ItemGroup>
     <ProjectReference Include="..\Shadowsocks.Interop\Shadowsocks.Interop.csproj" />
+    <ProjectReference Include="..\Shadowsocks.Net\Shadowsocks.Net.csproj" />
     <ProjectReference Include="..\Shadowsocks.Protocol\Shadowsocks.Protocol.csproj" />
   </ItemGroup>
 

diff --git a/Shadowsocks.Net/TCPRelay.cs b/Shadowsocks.Net/TCPRelay.cs
@@ -16,7 +16,7 @@
 
 namespace Shadowsocks.Net
 {
-    class TCPRelay : StreamService, IEnableLogger
+    public class TCPRelay : StreamService, IEnableLogger
     {
         public event EventHandler<SSTCPConnectedEventArgs> OnConnected;
         public event EventHandler<SSTransmitEventArgs> OnInbound;
@@ -182,7 +182,7 @@ public SSTCPConnectedEventArgs(Server server, TimeSpan latency) : base(server)
         }
     }
 
-    internal class TCPHandler : IEnableLogger
+    public class TCPHandler : IEnableLogger
     {
         public event EventHandler<SSTCPConnectedEventArgs> OnConnected;
         public event EventHandler<SSTransmitEventArgs> OnInbound;

diff --git a/Shadowsocks.Net/UDPRelay.cs b/Shadowsocks.Net/UDPRelay.cs
@@ -12,7 +12,7 @@
 
 namespace Shadowsocks.Net
 {
-    class UDPRelay : DatagramService
+    public class UDPRelay : DatagramService
     {
         Server _server;
         // TODO: choose a smart number