diff options
Diffstat (limited to 'proxy.c')
-rw-r--r-- | proxy.c | 7 |
1 files changed, 7 insertions, 0 deletions
@@ -51,6 +51,7 @@ proxy_client_data_out(struct cfg *cfg, struct uring_task *task, int res) return; } + uring_task_set_fd(&proxy->clienttask, proxy->cfd); uring_read(cfg, task, proxy->clientbuf, sizeof(proxy->clientbuf), 0, proxy_client_data_in); } @@ -64,6 +65,7 @@ proxy_client_data_in(struct cfg *cfg, struct uring_task *task, int res) return; proxy->clientlen = res; + uring_task_set_fd(&proxy->clienttask, proxy->sfd); uring_write(cfg, task, proxy->clientbuf, res, proxy_client_data_out); } @@ -83,6 +85,7 @@ proxy_server_data_out(struct cfg *cfg, struct uring_task *task, int res) return; } + uring_task_set_fd(&proxy->servertask, proxy->sfd); uring_read(cfg, task, proxy->serverbuf, sizeof(proxy->serverbuf), 0, proxy_server_data_in); } @@ -96,6 +99,7 @@ proxy_server_data_in(struct cfg *cfg, struct uring_task *task, int res) return; proxy->serverlen = res; + uring_task_set_fd(&proxy->servertask, proxy->cfd); uring_write(cfg, task, proxy->serverbuf, res, proxy_server_data_out); } @@ -152,6 +156,7 @@ again: goto again; } + proxy->sfd = sfd; uring_task_set_fd(&proxy->servertask, sfd); uring_connect(cfg, &proxy->servertask, &proxy->server, proxy_server_connected); } @@ -167,6 +172,8 @@ proxy_new(struct cfg *cfg, struct server *scfg, struct sockaddr_in46 *client, in return NULL; } + proxy->sfd = -1; + proxy->cfd = fd; proxy->scfg = scfg; proxy->client = *client; sockaddr_to_str(&proxy->client, proxy->clientstr, sizeof(proxy->clientstr)); |