X-Git-Url: http://git.tuebingen.mpg.de/?a=blobdiff_plain;f=client_common.c;h=fdd04e5acf450073a0ee73683e9818d0a42f38e4;hb=742c8b3f79a59278ced71e1855da1846b157b494;hp=94a6e86544e63824664b114cc400161d4e223684;hpb=c20e3e67fc2be1d5f8778b9c482271e023bbdcb8;p=paraslash.git

diff --git a/client_common.c b/client_common.c
index 94a6e865..fdd04e5a 100644
--- a/client_common.c
+++ b/client_common.c
@@ -68,13 +68,13 @@ static void client_pre_select(struct sched *s, void *context)
 	case CL_CONNECTED:
 	case CL_SENT_AUTH:
 	case CL_SENT_CH_RESPONSE:
-		para_fd_set(ct->scc.fd, &s->rfds, &s->max_fileno);
+		sched_monitor_readfd(ct->scc.fd, s);
 		return;
 
 	case CL_RECEIVED_WELCOME:
 	case CL_RECEIVED_PROCEED:
 	case CL_RECEIVED_CHALLENGE:
-		para_fd_set(ct->scc.fd, &s->wfds, &s->max_fileno);
+		sched_monitor_writefd(ct->scc.fd, s);
 		return;
 
 	case CL_SENDING:
@@ -83,7 +83,7 @@ static void client_pre_select(struct sched *s, void *context)
 			if (ret < 0)
 				sched_min_delay(s);
 			else if (ret > 0)
-				para_fd_set(ct->scc.fd, &s->wfds, &s->max_fileno);
+				sched_monitor_writefd(ct->scc.fd, s);
 		}
 		__attribute__ ((fallthrough));
 	case CL_EXECUTING:
@@ -92,7 +92,7 @@ static void client_pre_select(struct sched *s, void *context)
 			if (ret < 0)
 				sched_min_delay(s);
 			else if (ret > 0)
-				para_fd_set(ct->scc.fd, &s->rfds, &s->max_fileno);
+				sched_monitor_readfd(ct->scc.fd, s);
 		}
 		return;
 	}
@@ -125,8 +125,7 @@ static int send_sb(struct client_task *ct, int channel, void *buf, size_t numbyt
 	return 0;
 }
 
-static int recv_sb(struct client_task *ct, fd_set *rfds,
-		struct sb_buffer *result)
+static int recv_sb(struct client_task *ct, struct sb_buffer *result)
 {
 	int ret;
 	size_t n;
@@ -134,8 +133,6 @@ static int recv_sb(struct client_task *ct, fd_set *rfds,
 	void *trafo_context;
 	struct iovec iov;
 
-	if (!FD_ISSET(ct->scc.fd, rfds))
-		return 0;
 	if (ct->status < CL_SENT_CH_RESPONSE)
 		trafo = trafo_context = NULL;
 	else {
@@ -146,7 +143,7 @@ static int recv_sb(struct client_task *ct, fd_set *rfds,
 		ct->sbc[0] = sb_new_recv(0, trafo, trafo_context);
 again:
 	sb_get_recv_buffer(ct->sbc[0], &iov);
-	ret = read_nonblock(ct->scc.fd, iov.iov_base, iov.iov_len, rfds, &n);
+	ret = read_nonblock(ct->scc.fd, iov.iov_base, iov.iov_len, &n);
 	if (ret < 0) {
 		sb_free(ct->sbc[0]);
 		ct->sbc[0] = NULL;
@@ -288,7 +285,7 @@ static int client_post_select(struct sched *s, void *context)
 		return 0;
 	switch (ct->status) {
 	case CL_CONNECTED: /* receive welcome message */
-		ret = read_nonblock(ct->scc.fd, buf, sizeof(buf), &s->rfds, &n);
+		ret = read_nonblock(ct->scc.fd, buf, sizeof(buf), &n);
 		if (ret < 0 || n == 0)
 			goto out;
 		ct->features = parse_features(buf);
@@ -302,7 +299,7 @@ static int client_post_select(struct sched *s, void *context)
 		 * 0.8.0 we no longer need to request the feature.
 		 */
 		bool has_sha256;
-		if (!FD_ISSET(ct->scc.fd, &s->wfds))
+		if (!sched_write_ok(ct->scc.fd, s))
 			return 0;
 		has_sha256 = has_feature("sha256", ct);
 		sprintf(buf, AUTH_REQUEST_MSG "%s%s", ct->user, has_sha256?
@@ -324,7 +321,7 @@ static int client_post_select(struct sched *s, void *context)
 		unsigned char crypt_buf[1024];
 		struct sb_buffer sbb;
 
-		ret = recv_sb(ct, &s->rfds, &sbb);
+		ret = recv_sb(ct, &sbb);
 		if (ret <= 0)
 			goto out;
 		if (sbb.band != SBD_CHALLENGE) {
@@ -371,7 +368,7 @@ static int client_post_select(struct sched *s, void *context)
 	case CL_SENT_CH_RESPONSE: /* read server response */
 		{
 		struct sb_buffer sbb;
-		ret = recv_sb(ct, &s->rfds, &sbb);
+		ret = recv_sb(ct, &sbb);
 		if (ret <= 0)
 			goto out;
 		free(sbb.iov.iov_base);
@@ -383,7 +380,7 @@ static int client_post_select(struct sched *s, void *context)
 		}
 	case CL_RECEIVED_PROCEED: /* concat args and send command */
 		{
-		if (!FD_ISSET(ct->scc.fd, &s->wfds))
+		if (!sched_write_ok(ct->scc.fd, s))
 			return 0;
 		ret = send_sb_command(ct);
 		if (ret <= 0)
@@ -405,7 +402,7 @@ static int client_post_select(struct sched *s, void *context)
 			}
 			if (ret < 0)
 				goto close1;
-			if (ret > 0 && FD_ISSET(ct->scc.fd, &s->wfds)) {
+			if (ret > 0 && sched_write_ok(ct->scc.fd, s)) {
 				sz = btr_next_buffer(ct->btrn[1], &buf2);
 				assert(sz);
 				ret = send_sb(ct, 1, buf2, sz, SBD_BLOB_DATA, true);
@@ -421,9 +418,9 @@ static int client_post_select(struct sched *s, void *context)
 			ret = btr_node_status(ct->btrn[0], 0, BTR_NT_ROOT);
 			if (ret < 0)
 				goto close0;
-			if (ret > 0 && FD_ISSET(ct->scc.fd, &s->rfds)) {
+			if (ret > 0 && sched_read_ok(ct->scc.fd, s)) {
 				struct sb_buffer sbb;
-				ret = recv_sb(ct, &s->rfds, &sbb);
+				ret = recv_sb(ct, &sbb);
 				if (ret < 0)
 					goto close0;
 				if (ret > 0) {
@@ -452,8 +449,7 @@ out:
 		return 0;
 	btr_remove_node(&ct->btrn[0]);
 	btr_remove_node(&ct->btrn[1]);
-	if (ret != -E_SERVER_CMD_SUCCESS && ret != -E_SERVER_CMD_FAILURE)
-		PARA_ERROR_LOG("%s\n", para_strerror(-ret));
+	PARA_NOTICE_LOG("closing connection (%s)\n", para_strerror(-ret));
 	if (ct->scc.fd >= 0) {
 		close(ct->scc.fd);
 		ct->scc.fd = -1;