#include "donar_client.h" void load_onion_services(struct donar_client_ctx* ctx, char* onion_file, int ports_count) { tor_os_create (&(ctx->tos), onion_file, NULL, ports_count); tor_os_read (&(ctx->tos)); } void init_tcp_client(struct donar_client_ctx* ctx, int i) { ctx->ports[i] = 7500 + i; int sock1, err; while (1) { sock1 = create_tcp_client("127.0.0.1", "9050"); if (sock1 < 0) goto failed_socks5; ctx->client_sock[i].fd = sock1; ctx->client_sock[i].state = SOCKS5_STATE_NEW; evt_core_add_fd (&(ctx->evts), "configure-socks5", sock1); //@FIXME: We suppose that we will be able to do the whole write at once which is wrong too err = socks5_handshake_syn(sock1); if (err) goto failed_socks5; break; failed_socks5: fprintf(stderr, "Failed connection for socket %d/%d. Sleeping 2 seconds...\n",i+1, CLIENT_PORT_SIZE); close(sock1); sleep(2); } } void configure_tcp_clients(struct evt_core_ctx* ctx, struct evt_core_cat* cat, int fd) { int err, pos; struct donar_client_ctx* app_ctx = (struct donar_client_ctx*) cat->app_ctx; pos = -1; for (int i = 0; i < CLIENT_PORT_SIZE; i++) { if (app_ctx->client_sock[i].fd == fd) { pos = i; } } char target_host[255]; if (strlen(app_ctx->tos.keys[pos].pub) > 254) { fprintf(stderr, "Domain name is too long\n"); exit(EXIT_FAILURE); } sprintf(target_host, "%s.onion", app_ctx->tos.keys[pos].pub); switch (app_ctx->client_sock[pos].state) { case SOCKS5_STATE_NEW: //@FIXME: We suppose that we will be able to do the whole read at once which is wrong err = socks5_handshake_ack (fd); if (err < 0) goto on_socks5_err; //@FIXME: We suppose that we will be able to do the whole write at once which is wrong too err = socks5_connect_dns(fd, target_host, app_ctx->ports[pos]); if (err < 0) goto on_socks5_err; app_ctx->client_sock[pos].state = SOCKS5_STATE_ACK; printf("Socket %d/%d %s:%d is connecting...\n", pos+1, CLIENT_PORT_SIZE, target_host, app_ctx->ports[pos]); break; case SOCKS5_STATE_ACK: //@FIXME: We suppose that we will be able to do the whole read at once which is wrong too err = socks5_reply (fd); if (err < 0) goto on_socks5_err; app_ctx->client_sock[pos].state = SOCKS5_STATE_RDY; int sock1, sock2; sock1 = dup(fd); sock2 = dup(fd); if (sock1 < 0 || sock2 < 0) goto on_socks5_err; void* fdcat = evt_core_rm_fd (ctx, fd); if (fdcat == NULL) goto on_socks5_err; evt_core_add_fd (ctx, "tcp-write", sock1); evt_core_add_fd (ctx, "tcp-read", sock2); printf("Socket %d/%d %s:%d has been added to the pool!\n", pos+1, CLIENT_PORT_SIZE, target_host, app_ctx->ports[pos]); break; case SOCKS5_STATE_RDY: goto on_socks5_err; break; case SOCKS5_STATE_ERR: goto on_socks5_err; break; } return; on_socks5_err: perror("An error occured while connecting to an Onion Service"); app_ctx->client_sock[pos].state = SOCKS5_STATE_ERR; evt_core_rm_fd (ctx, fd); init_tcp_client (app_ctx, pos); } void init_udp_server(struct donar_client_ctx* ctx, char* port) { int sock1, sock2; sock1 = create_udp_server(port); if (sock1 < 0) goto socket_failed; sock2 = dup(sock1); if (sock2 < 0) goto socket_failed; evt_core_add_fd (&(ctx->evts), "udp-read", sock1); evt_core_add_fd (&(ctx->evts), "udp-write", sock2); return; socket_failed: fprintf(stderr, "Socket initialization for the UDP server failed\n"); exit(EXIT_FAILURE); } void donar_client(struct donar_client_ctx* ctx, struct algo_skel* algo, char* onion_file, char* port) { evt_core_init (&(ctx->evts)); struct evt_core_cat init_socks5 = { .app_ctx = ctx, .free_app_ctx = NULL, .cb = configure_tcp_clients, .name = "configure-socks5", .flags = EPOLLIN | EPOLLET, .socklist = NULL }; evt_core_add_cat (&(ctx->evts), &init_socks5); evt_core_add_cat (&(ctx->evts), &(algo->on_tcp_co)); evt_core_add_cat (&(ctx->evts), &(algo->on_udp_read)); evt_core_add_cat (&(ctx->evts), &(algo->on_tcp_read)); evt_core_add_cat (&(ctx->evts), &(algo->on_udp_write)); evt_core_add_cat (&(ctx->evts), &(algo->on_tcp_write)); printf("--- Categories created\n"); load_onion_services (ctx, onion_file, CLIENT_PORT_SIZE); printf("--- Onion services loaded\n"); for (int i = 0; i < CLIENT_PORT_SIZE; i++) { init_tcp_client(ctx, i); } printf("--- TCP Clients Connected\n"); init_udp_server (ctx, port); printf("--- UDP Server is listening\n"); evt_core_loop(&(ctx->evts)); tor_os_free (&(ctx->tos)); }