Skip to content

wsssh
wsssh
Commit 96207bd8 authored by Stefy Lanza (nextime / spora )'s avatar Stefy Lanza (nextime / spora )
Browse files
Options

Fix wssshc tunnel handling - remove SSH protocol logic, implement raw TCP forwarding

parent 1114c673
Expand all Show whitespace changes
Inline Side-by-side
Showing with 173 additions and 197 deletions
wssshtools/tunnel.c
View file @ 96207bd8
This diff is collapsed.
wssshtools/tunnel.h
View file @ 96207bd8
......@@ -58,7 +58,7 @@ void *forward_tcp_to_ws(void *arg);
void *forward_ws_to_local(void *arg);
void *forward_ws_to_ssh_server(void *arg);
void *tunnel_thread(void *arg);
void handle_tunnel_request(SSL *ssl, const char *request_id, int debug);
void handle_tunnel_request(SSL *ssl, const char *request_id, int debug, const char *ssh_host, int ssh_port);
void handle_tunnel_data(SSL *ssl, const char *request_id, const char *data_hex, int debug);
void handle_tunnel_close(SSL *ssl, const char *request_id, int debug);
void send_tunnel_close(SSL *ssl, const char *request_id, int debug);
......
wssshtools/wsssh.c
View file @ 96207bd8
......@@ -236,6 +236,7 @@ char **modify_ssh_args(int argc, char *argv[], const char *original_host, int lo
int main(int argc, char *argv[]) {
// Read config
char *config_domain = read_config_value("domain");
......
wssshtools/wssshc.c
View file @ 96207bd8
......@@ -41,8 +41,10 @@ int global_debug = 0;
typedef struct {
char *server_ip;
int port;
char *wssshd_server;
int wssshd_port;
char *ssh_host;
int ssh_port;
char *client_id;
char *password;
int interval;
......@@ -73,12 +75,18 @@ void load_config_file(const char *config_path, wssshc_config_t *config) {
while (end > value && *end == ' ') *end-- = 0;
if (strcmp(key, "password") == 0 && !config->password) {
config->password = strdup(value);
} else if (strcmp(key, "server-ip") == 0 && !config->server_ip) {
config->server_ip = strdup(value);
} else if (strcmp(key, "domain") == 0 && !config->server_ip) {
config->server_ip = strdup(value);
} else if (strcmp(key, "port") == 0) {
config->port = atoi(value);
} else if (strcmp(key, "wssshd-server") == 0 && !config->wssshd_server) {
config->wssshd_server = strdup(value);
} else if (strcmp(key, "domain") == 0 && !config->wssshd_server) {
config->wssshd_server = strdup(value);
} else if (strcmp(key, "wssshd-port") == 0) {
config->wssshd_port = atoi(value);
} else if (strcmp(key, "port") == 0 && config->wssshd_port == 9898) {
config->wssshd_port = atoi(value);
} else if (strcmp(key, "ssh-host") == 0 && !config->ssh_host) {
config->ssh_host = strdup(value);
} else if (strcmp(key, "ssh-port") == 0) {
config->ssh_port = atoi(value);
} else if (strcmp(key, "id") == 0 && !config->client_id) {
config->client_id = strdup(value);
} else if (strcmp(key, "interval") == 0) {
......@@ -117,8 +125,10 @@ void print_usage(const char *program_name) {
fprintf(stderr, "Protect the dolls!\n\n");
fprintf(stderr, "Options:\n");
fprintf(stderr, " --config FILE Configuration file path (overrides default hierarchy)\n");
fprintf(stderr, " --server-ip IP Server IP address\n");
fprintf(stderr, " --port PORT Server port (default: %d)\n", DEFAULT_PORT);
fprintf(stderr, " --wssshd-server HOST WSSSHD server host (default: mbeted.nexlab.net)\n");
fprintf(stderr, " --wssshd-port PORT WSSSHD server port (default: 9898)\n");
fprintf(stderr, " --ssh-host HOST SSH host to forward tunnel data to (default: 127.0.0.1)\n");
fprintf(stderr, " --ssh-port PORT SSH port to forward tunnel data to (default: 22)\n");
fprintf(stderr, " --id ID Client identifier\n");
fprintf(stderr, " --password PASS Registration password\n");
fprintf(stderr, " --interval SEC Reconnection interval (default: 30)\n");
......@@ -136,8 +146,10 @@ void print_usage(const char *program_name) {
int parse_args(int argc, char *argv[], wssshc_config_t *config) {
static struct option long_options[] = {
{"config", required_argument, 0, 'c'},
{"server-ip", required_argument, 0, 's'},
{"port", required_argument, 0, 'p'},
{"wssshd-server", required_argument, 0, 's'},
{"wssshd-port", required_argument, 0, 'p'},
{"ssh-host", required_argument, 0, 'H'},
{"ssh-port", required_argument, 0, 'P'},
{"id", required_argument, 0, 'i'},
{"password", required_argument, 0, 'w'},
{"interval", required_argument, 0, 't'},
......@@ -149,17 +161,24 @@ int parse_args(int argc, char *argv[], wssshc_config_t *config) {
int opt;
char *custom_config = NULL;
while ((opt = getopt_long(argc, argv, "c:s:p:i:w:t:dh", long_options, NULL)) != -1) {
while ((opt = getopt_long(argc, argv, "c:s:p:H:P:i:w:t:dh", long_options, NULL)) != -1) {
switch (opt) {
case 'c':
custom_config = optarg;
break;
case 's':
if (config->server_ip) free(config->server_ip);
config->server_ip = strdup(optarg);
if (config->wssshd_server) free(config->wssshd_server);
config->wssshd_server = strdup(optarg);
break;
case 'p':
config->port = atoi(optarg);
config->wssshd_port = atoi(optarg);
break;
case 'H':
if (config->ssh_host) free(config->ssh_host);
config->ssh_host = strdup(optarg);
break;
case 'P':
config->ssh_port = atoi(optarg);
break;
case 'i':
if (config->client_id) free(config->client_id);
......@@ -215,7 +234,7 @@ int connect_to_server(const wssshc_config_t *config) {
cleanup_tunnel(config->debug);
// Resolve hostname
if ((he = gethostbyname(config->server_ip)) == NULL) {
if ((he = gethostbyname(config->wssshd_server)) == NULL) {
herror("gethostbyname");
return 1;
}
......@@ -228,7 +247,7 @@ int connect_to_server(const wssshc_config_t *config) {
memset(&server_addr, 0, sizeof(server_addr));
server_addr.sin_family = AF_INET;
server_addr.sin_port = htons(config->port);
server_addr.sin_port = htons(config->wssshd_port);
server_addr.sin_addr = *((struct in_addr *)he->h_addr);
// Connect to server
......@@ -257,7 +276,7 @@ int connect_to_server(const wssshc_config_t *config) {
}
// Perform WebSocket handshake
if (!websocket_handshake(ssl, config->server_ip, config->port, "/")) {
if (!websocket_handshake(ssl, config->wssshd_server, config->wssshd_port, "/")) {
SSL_free(ssl);
SSL_CTX_free(ssl_ctx);
close(sock);
......@@ -456,7 +475,7 @@ int connect_to_server(const wssshc_config_t *config) {
printf("[DEBUG - WebSockets] Received tunnel_request for ID: %s\n", id_start);
fflush(stdout);
}
handle_tunnel_request(ssl, id_start, config->debug);
handle_tunnel_request(ssl, id_start, config->debug, config->ssh_host, config->ssh_port);
}
}
}
......@@ -648,8 +667,10 @@ int connect_to_server(const wssshc_config_t *config) {
int main(int argc, char *argv[]) {
wssshc_config_t config = {
.server_ip = NULL,
.port = DEFAULT_PORT,
.wssshd_server = NULL,
.wssshd_port = 9898,
.ssh_host = NULL,
.ssh_port = 22,
.client_id = NULL,
.password = NULL,
.interval = 30,
......@@ -667,11 +688,20 @@ int main(int argc, char *argv[]) {
return 1;
}
// Set defaults for optional fields
if (!config.wssshd_server) {
config.wssshd_server = strdup("mbeted.nexlab.net");
}
if (!config.ssh_host) {
config.ssh_host = strdup("127.0.0.1");
}
// Validate required arguments
if (!config.server_ip || !config.client_id || !config.password) {
fprintf(stderr, "Error: --server-ip, --id, and --password are required\n");
if (!config.client_id || !config.password) {
fprintf(stderr, "Error: --id and --password are required\n");
print_usage(argv[0]);
if (config.server_ip) free(config.server_ip);
if (config.wssshd_server) free(config.wssshd_server);
if (config.ssh_host) free(config.ssh_host);
if (config.client_id) free(config.client_id);
if (config.password) free(config.password);
pthread_mutex_destroy(&tunnel_mutex);
......@@ -679,7 +709,19 @@ int main(int argc, char *argv[]) {
}
global_debug = config.debug;
// Print configured options
printf("WebSocket SSH Client starting...\n");
printf("Configuration:\n");
printf(" WSSSHD Server: %s\n", config.wssshd_server ? config.wssshd_server : "(null)");
printf(" WSSSHD Port: %d\n", config.wssshd_port);
printf(" SSH Host: %s\n", config.ssh_host ? config.ssh_host : "(null)");
printf(" SSH Port: %d\n", config.ssh_port);
printf(" Client ID: %s\n", config.client_id ? config.client_id : "(null)");
printf(" Password: %s\n", config.password ? "***" : "(null)");
printf(" Reconnection Interval: %d seconds\n", config.interval);
printf(" Debug Mode: %s\n", config.debug ? "enabled" : "disabled");
printf("\n");
while (1) {
int result = connect_to_server(&config);
......@@ -699,7 +741,8 @@ int main(int argc, char *argv[]) {
}
// Cleanup
free(config.server_ip);
free(config.wssshd_server);
free(config.ssh_host);
free(config.client_id);
free(config.password);
pthread_mutex_destroy(&tunnel_mutex);
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment