X-Git-Url: http://git.tuebingen.mpg.de/?a=blobdiff_plain;f=client_common.c;h=ba419786ee036745972825ec49629d9f1cf342ab;hb=e4db7671a91a7552c642acc979f0eb278f8d467f;hp=a7115fd75ac33495d9dd36862ff446757bd9666c;hpb=b62e2796b85c7d7f3138fe729f4637853e0fafe0;p=paraslash.git diff --git a/client_common.c b/client_common.c index a7115fd7..ba419786 100644 --- a/client_common.c +++ b/client_common.c @@ -9,7 +9,6 @@ #include #include #include -#include #include "para.h" #include "error.h" @@ -40,8 +39,10 @@ void client_close(struct client_task *ct) { if (!ct) return; - if (ct->rc4c.fd >= 0) - close(ct->rc4c.fd); + if (ct->scc.fd >= 0) + close(ct->scc.fd); + sc_free(ct->scc.recv); + sc_free(ct->scc.send); free(ct->user); free(ct->config_file); free(ct->key_file); @@ -69,19 +70,19 @@ static void client_pre_select(struct sched *s, struct task *t) struct client_task *ct = container_of(t, struct client_task, task); struct btr_node *btrn = ct->btrn; - if (ct->rc4c.fd < 0) + if (ct->scc.fd < 0) return; switch (ct->status) { case CL_CONNECTED: case CL_SENT_AUTH: case CL_SENT_CH_RESPONSE: case CL_SENT_COMMAND: - para_fd_set(ct->rc4c.fd, &s->rfds, &s->max_fileno); + para_fd_set(ct->scc.fd, &s->rfds, &s->max_fileno); return; case CL_RECEIVED_WELCOME: case CL_RECEIVED_PROCEED: - para_fd_set(ct->rc4c.fd, &s->wfds, &s->max_fileno); + para_fd_set(ct->scc.fd, &s->wfds, &s->max_fileno); return; case CL_RECEIVING: @@ -90,7 +91,7 @@ static void client_pre_select(struct sched *s, struct task *t) if (ret < 0) sched_min_delay(s); else - para_fd_set(ct->rc4c.fd, &s->rfds, + para_fd_set(ct->scc.fd, &s->rfds, &s->max_fileno); } return; @@ -100,7 +101,7 @@ static void client_pre_select(struct sched *s, struct task *t) if (ret < 0) sched_min_delay(s); else - para_fd_set(ct->rc4c.fd, &s->wfds, + para_fd_set(ct->scc.fd, &s->wfds, &s->max_fileno); } return; @@ -113,12 +114,12 @@ static int client_recv_buffer(struct client_task *ct, fd_set *rfds, int ret; if (ct->status < CL_SENT_CH_RESPONSE) - return read_nonblock(ct->rc4c.fd, buf, sz, rfds, n); + return read_nonblock(ct->scc.fd, buf, sz, rfds, n); *n = 0; - ret = rc4_recv_buffer(&ct->rc4c, buf, sz); + ret = sc_recv_buffer(&ct->scc, buf, sz); /* - * rc4_recv_buffer is used with blocking fds elsewhere, so it + * sc_recv_buffer is used with blocking fds elsewhere, so it * does not use the nonblock-API. Therefore we need to * check for EOF and EAGAIN. */ @@ -154,7 +155,7 @@ static void client_post_select(struct sched *s, struct task *t) char buf[CLIENT_BUFSIZE]; t->error = 0; - if (ct->rc4c.fd < 0) + if (ct->scc.fd < 0) return; switch (ct->status) { case CL_CONNECTED: /* receive welcome message */ @@ -166,20 +167,20 @@ static void client_post_select(struct sched *s, struct task *t) case CL_RECEIVED_WELCOME: /* send auth command */ sprintf(buf, AUTH_REQUEST_MSG "%s", ct->user); PARA_INFO_LOG("--> %s\n", buf); - if (!FD_ISSET(ct->rc4c.fd, &s->wfds)) + if (!FD_ISSET(ct->scc.fd, &s->wfds)) return; - ret = send_buffer(ct->rc4c.fd, buf); + ret = send_buffer(ct->scc.fd, buf); if (ret < 0) goto out; ct->status = CL_SENT_AUTH; return; case CL_SENT_AUTH: /* - * Receive challenge and rc4 keys, decrypt the challenge and + * Receive challenge and session keys, decrypt the challenge and * send back the hash of the decrypted challenge. */ { - /* decrypted challenge/rc4 buffer */ + /* decrypted challenge/session key buffer */ unsigned char crypt_buf[1024]; /* the SHA1 of the decrypted challenge */ unsigned char challenge_sha1[HASH_SIZE]; @@ -188,18 +189,17 @@ static void client_post_select(struct sched *s, struct task *t) if (ret < 0 || n == 0) goto out; PARA_INFO_LOG("<-- [challenge] (%zu bytes)\n", n); - ret = para_decrypt_buffer(ct->key_file, crypt_buf, + ret = priv_decrypt(ct->key_file, crypt_buf, (unsigned char *)buf, n); if (ret < 0) goto out; sha1_hash((char *)crypt_buf, CHALLENGE_SIZE, challenge_sha1); - RC4_set_key(&ct->rc4c.send_key, RC4_KEY_LEN, - crypt_buf + CHALLENGE_SIZE); - RC4_set_key(&ct->rc4c.recv_key, RC4_KEY_LEN, - crypt_buf + CHALLENGE_SIZE + RC4_KEY_LEN); + ct->scc.send = sc_new(crypt_buf + CHALLENGE_SIZE, RC4_KEY_LEN); + ct->scc.recv = sc_new(crypt_buf + CHALLENGE_SIZE + RC4_KEY_LEN, + RC4_KEY_LEN); hash_to_asc(challenge_sha1, buf); PARA_INFO_LOG("--> %s\n", buf); - ret = send_bin_buffer(ct->rc4c.fd, (char *)challenge_sha1, + ret = send_bin_buffer(ct->scc.fd, (char *)challenge_sha1, HASH_SIZE); if (ret < 0) goto out; @@ -224,7 +224,7 @@ static void client_post_select(struct sched *s, struct task *t) { int i; char *command = NULL; - if (!FD_ISSET(ct->rc4c.fd, &s->wfds)) + if (!FD_ISSET(ct->scc.fd, &s->wfds)) return; for (i = 0; i < ct->conf.inputs_num; i++) { char *tmp = command; @@ -234,7 +234,7 @@ static void client_post_select(struct sched *s, struct task *t) } command = para_strcat(command, EOC_MSG "\n"); PARA_DEBUG_LOG("--> %s\n", command); - ret = rc4_send_buffer(&ct->rc4c, command); + ret = sc_send_buffer(&ct->scc, command); free(command); if (ret < 0) goto out; @@ -268,10 +268,10 @@ static void client_post_select(struct sched *s, struct task *t) goto out; if (ret == 0) return; - if (!FD_ISSET(ct->rc4c.fd, &s->wfds)) + if (!FD_ISSET(ct->scc.fd, &s->wfds)) return; sz = btr_next_buffer(btrn, &buf2); - ret = rc4_send_bin_buffer(&ct->rc4c, buf2, sz); + ret = sc_send_bin_buffer(&ct->scc, buf2, sz); if (ret < 0) goto out; btr_consume(btrn, sz); @@ -289,7 +289,7 @@ static void client_post_select(struct sched *s, struct task *t) * The FD_ISSET() is not strictly necessary, but is allows us * to skip the malloc below if there is nothing to read anyway. */ - if (!FD_ISSET(ct->rc4c.fd, &s->rfds)) + if (!FD_ISSET(ct->scc.fd, &s->rfds)) return; buf2 = para_malloc(CLIENT_BUFSIZE); ret = client_recv_buffer(ct, &s->rfds, buf2, CLIENT_BUFSIZE, &n); @@ -315,14 +315,14 @@ static int client_connect(struct client_task *ct) { int ret; - ct->rc4c.fd = -1; + ct->scc.fd = -1; ret = para_connect_simple(IPPROTO_TCP, ct->conf.hostname_arg, ct->conf.server_port_arg); if (ret < 0) return ret; - ct->rc4c.fd = ret; + ct->scc.fd = ret; ct->status = CL_CONNECTED; - ret = mark_fd_nonblocking(ct->rc4c.fd); + ret = mark_fd_nonblocking(ct->scc.fd); if (ret < 0) goto err_out; ct->task.pre_select = client_pre_select; @@ -331,8 +331,8 @@ static int client_connect(struct client_task *ct) register_task(&ct->task); return 1; err_out: - close(ct->rc4c.fd); - ct->rc4c.fd = -1; + close(ct->scc.fd); + ct->scc.fd = -1; return ret; } @@ -363,7 +363,7 @@ int client_open(int argc, char *argv[], struct client_task **ct_ptr, ct->btrn = btr_new_node(&(struct btr_node_description) EMBRACE(.name = "client", .parent = parent, .child = child)); *ct_ptr = ct; - ct->rc4c.fd = -1; + ct->scc.fd = -1; ret = -E_CLIENT_SYNTAX; if (client_cmdline_parser(argc, argv, &ct->conf)) goto out;