1 files changed, 8 insertions, 1 deletions
diff --git a/main.c b/main.c
index e8b0026..800b807 100644
--- a/main.c
+++ b/main.c
@@ -34,9 +34,16 @@ int main(int argc, char *argv[])
 	if (argc == 1)
 		errx(EXIT_FAILURE, "bad usage");
 
+	if (!strcmp(argv[1], "server")) {
+		if (!tunnel_server())
+			errx(EXIT_FAILURE, "start server failed");
+
+		return 0;
+	}
+
 	int sock = tunnel_client();
 	if (sock < 0)
-		errx(EXIT_FAILURE, "tunnel_client failed");
+		errx(EXIT_FAILURE, "init client failed");
 
 	sendcmd(sock, argc - 1, argv + 1);