port number can be specified by user

Author: cnbatch

README.md

@@ -2,6 +2,7 @@
 Socks5 Server Demo implemented in C++20 coroutine with ASIO library
 
 [简体中文版](#简体中文版)
+
 [繁體中文版](#繁體中文版)
 
 ## Supported Features
@@ -25,6 +26,18 @@ For example, username is ‘user’, password is ‘door’:
 ./socks5demo user door
 ```
 
+### Specify Port Number
+Default port number is 1080. If the port number must be changed, for example, use 1180 instead:
+```
+./socks5demo 1180
+```
+
+### Specify Port Number + Username / Password Authentication
+
+```
+./socks5demo 1180 user door
+```
+
 ## Requirements
 - `ASIO` library must be installed first.
 - Compiler that supports C++20
@@ -67,6 +80,18 @@ cmake --build .
 ./socks5demo user door
 ```
 
+### 指定端口号
+默认端口号是 1080，如果需要使用其他端口号，可以自行指定，例如使用 1180 端口号
+```
+./socks5demo 1180
+```
+
+### 指定端口号 + 用户名 / 密码 认证模式
+
+```
+./socks5demo 1180 user door
+```
+
 ## 编译前置要求
 - 必须先安装 `ASIO` 库
 - 支持C++20的编译器
@@ -111,6 +136,18 @@ cmake --build .
 ./socks5demo user door
 ```
 
+### 指定通訊埠號
+預設通訊埠號是 1080，如果需要使用其他通訊埠號，可以自行指定，例如使用 1180 通訊埠號
+```
+./socks5demo 1180
+```
+
+### 指定通訊埠號 + 用戶名稱 / 密碼 認證模式
+
+```
+./socks5demo 1180 user door
+```
+
 ## 編譯前置要求
 - 必須事先裝好 C++庫 `ASIO`
 - 支援C++20的編譯器

src/main.cpp

@@ -843,25 +843,39 @@ awaitable<void> socks5_access(tcp_socket client_socket, const char *username, co
 	}
 }
 
-awaitable<void> listener_v4(const char *username, const char *password)
+awaitable<void> listener_ipv4(const char *username, const char *password, uint16_t port = 1080)
 {
 	asio::any_io_executor executor = co_await this_coro::executor;
-	tcp_acceptor acceptor(executor, { tcp::v4(), 1080 });
-	while (true)
+	try
+	{
+		tcp_acceptor acceptor(executor, { tcp::v4(), port });
+		while (true)
+		{
+			tcp_socket socket = co_await acceptor.async_accept();
+			co_spawn(executor, socks5_access(std::move(socket), username, password), detached);
+		}
+	}
+	catch (std::exception &e)
 	{
-		tcp_socket socket = co_await acceptor.async_accept();
-		co_spawn(executor, socks5_access(std::move(socket), username, password), detached);
+		std::printf("IPv4 socks5_listen Exception: %s\n", e.what());
 	}
 }
 
-awaitable<void> listener_v6(const char *username, const char *password)
+awaitable<void> listener_ipv6(const char *username, const char *password, uint16_t port = 1080)
 {
 	asio::any_io_executor executor = co_await this_coro::executor;
-	tcp_acceptor acceptor(executor, { tcp::v6(), 1080 });
-	while (true)
+	try
+	{
+		tcp_acceptor acceptor(executor, { tcp::v6(), port });
+		while (true)
+		{
+			tcp_socket socket = co_await acceptor.async_accept();
+			co_spawn(executor, socks5_access(std::move(socket), username, password), detached);
+		}
+	}
+	catch (std::exception &e)
 	{
-		tcp_socket socket = co_await acceptor.async_accept();
-		co_spawn(executor, socks5_access(std::move(socket), username, password), detached);
+		std::printf("IPv6 socks5_listen Exception: %s\n", e.what());
 	}
 }
 
@@ -874,15 +888,42 @@ int main(int argc, char *argv[])
 		asio::signal_set signals(io_context, SIGINT, SIGTERM);
 		signals.async_wait([&](auto, auto) { io_context.stop(); });
 
-		if (argc == 3)
+		if (argc == 1)
+		{
+			co_spawn(io_context, listener_ipv4(nullptr, nullptr), detached);
+			co_spawn(io_context, listener_ipv6(nullptr, nullptr), detached);
+		}
+		else if (argc == 2)
+		{
+			int port = std::stoi(argv[1]);
+			if (port < 1 || port > 65535)
+			{
+				std::printf("Incorrect port number: %d\n", port);
+				return 1;
+			}
+			co_spawn(io_context, listener_ipv4(nullptr, nullptr, (uint16_t)port), detached);
+			co_spawn(io_context, listener_ipv6(nullptr, nullptr, (uint16_t)port), detached);
+		}
+		else if (argc == 3)
+		{
+			co_spawn(io_context, listener_ipv4(argv[1], argv[2]), detached);
+			co_spawn(io_context, listener_ipv6(argv[1], argv[2]), detached);
+		}
+		else if (argc == 4)
 		{
-			co_spawn(io_context, listener_v4(argv[1], argv[2]), detached);
-			co_spawn(io_context, listener_v6(argv[1], argv[2]), detached);
+			int port = std::stoi(argv[1]);
+			if (port < 1 || port > 65535)
+			{
+				std::printf("Incorrect port number: %d\n", (uint16_t)port);
+				return 1;
+			}
+			co_spawn(io_context, listener_ipv4(argv[2], argv[3], port), detached);
+			co_spawn(io_context, listener_ipv6(argv[2], argv[3], port), detached);
 		}
 		else
 		{
-			co_spawn(io_context, listener_v4(nullptr, nullptr), detached);
-			co_spawn(io_context, listener_v6(nullptr, nullptr), detached);
+			std::printf("Incorrect arguments\n");
+			return 1;
 		}
 
 		io_context.run();