X-Git-Url: http://git.tuebingen.mpg.de/?p=paraslash.git;a=blobdiff_plain;f=afs.c;h=97c3d5694a8dd83fd24c753495bf1411011ae7b9;hp=6286806013471ea55278593d0cbf2de74f8934b7;hb=edeb499676e6d042ef1a913914a9fcb45a8cadde;hpb=8dab386f87d2998fd26126600ca7faf56d5a35d1

diff --git a/afs.c b/afs.c
index 62868060..97c3d569 100644
--- a/afs.c
+++ b/afs.c
@@ -233,7 +233,7 @@ int send_callback_request(callback_function *f, struct osl_object *query,
 	*(uint32_t *) buf = afs_socket_cookie;
 	*(int *) (buf + sizeof(afs_socket_cookie)) = query_shmid;
 
-	ret = create_remote_socket(conf.afs_socket_arg);
+	ret = connect_local_socket(conf.afs_socket_arg);
 	if (ret < 0)
 		goto out;
 	fd = ret;
@@ -709,15 +709,16 @@ static void signal_pre_select(struct sched *s, struct task *t)
 
 static void afs_signal_post_select(struct sched *s, struct task *t)
 {
-	struct signal_task *st = container_of(t, struct signal_task, task);
+	int signum;
+
 	if (getppid() == 1) {
 		PARA_EMERG_LOG("para_server died\n");
 		goto shutdown;
 	}
-	if (!FD_ISSET(st->fd, &s->rfds))
+	signum = para_next_signal(&s->rfds);
+	if (signum == 0)
 		return;
-	st->signum = para_next_signal();
-	if (st->signum == SIGHUP) {
+	if (signum == SIGHUP) {
 		close_afs_tables();
 		parse_config_or_die(1);
 		t->error = open_afs_tables();
@@ -726,7 +727,7 @@ static void afs_signal_post_select(struct sched *s, struct task *t)
 		init_admissible_files(current_mop);
 		return;
 	}
-	PARA_EMERG_LOG("terminating on signal %d\n", st->signum);
+	PARA_EMERG_LOG("terminating on signal %d\n", signum);
 shutdown:
 	sched_shutdown();
 	t->error = -E_AFS_SIGNAL;
@@ -842,57 +843,56 @@ static int call_callback(int fd, int query_shmid)
 	return shm_detach(query_shm);
 }
 
-static int execute_server_command(void)
+static int execute_server_command(fd_set *rfds)
 {
 	char buf[8];
-	int ret = recv_bin_buffer(server_socket, buf, sizeof(buf) - 1);
+	size_t n;
+	int ret = read_nonblock(server_socket, buf, sizeof(buf) - 1, rfds, &n);
 
-	if (ret <= 0) {
-		if (!ret)
-			ret = -ERRNO_TO_PARA_ERROR(ECONNRESET);
-		goto err;
-	}
-	buf[ret] = '\0';
-	PARA_DEBUG_LOG("received: %s\n", buf);
-	ret = -E_BAD_CMD;
+	if (ret < 0 || n == 0)
+		return ret;
+	buf[n] = '\0';
 	if (strcmp(buf, "new"))
-		goto err;
-	ret = open_next_audio_file();
-err:
-	return ret;
+		return -E_BAD_CMD;
+	return open_next_audio_file();
 }
 
-static void execute_afs_command(int fd, uint32_t expected_cookie)
+/* returns 0 if no data available, 1 else */
+static int execute_afs_command(int fd, fd_set *rfds, uint32_t expected_cookie)
 {
 	uint32_t cookie;
 	int query_shmid;
 	char buf[sizeof(cookie) + sizeof(query_shmid)];
-	int ret = recv_bin_buffer(fd, buf, sizeof(buf));
+	size_t n;
+	int ret = read_nonblock(fd, buf, sizeof(buf), rfds, &n);
 
 	if (ret < 0)
 		goto err;
-	if (ret != sizeof(buf)) {
+	if (n == 0)
+		return 0;
+	if (n != sizeof(buf)) {
 		PARA_NOTICE_LOG("short read (%d bytes, expected %lu)\n",
 			ret, (long unsigned) sizeof(buf));
-		return;
+		return 1;
 	}
 	cookie = *(uint32_t *)buf;
 	if (cookie != expected_cookie) {
-		PARA_NOTICE_LOG("received invalid cookie(got %u, expected %u)\n",
+		PARA_NOTICE_LOG("received invalid cookie (got %u, expected %u)\n",
 			(unsigned)cookie, (unsigned)expected_cookie);
-		return;
+		return 1;
 	}
 	query_shmid = *(int *)(buf + sizeof(cookie));
 	if (query_shmid < 0) {
 		PARA_WARNING_LOG("received invalid query shmid %d)\n",
 			query_shmid);
-		return;
+		return 1;
 	}
 	ret = call_callback(fd, query_shmid);
 	if (ret >= 0)
-		return;
+		return 1;
 err:
 	PARA_NOTICE_LOG("%s\n", para_strerror(-ret));
+	return 1;
 }
 
 /** Shutdown connection if query has not arrived until this many seconds. */
@@ -905,20 +905,16 @@ static void command_post_select(struct sched *s, struct task *t)
 	struct afs_client *client, *tmp;
 	int fd, ret;
 
-	if (FD_ISSET(server_socket, &s->rfds)) {
-		ret = execute_server_command();
-		if (ret < 0) {
-			PARA_EMERG_LOG("%s\n", para_strerror(-ret));
-			sched_shutdown();
-			return;
-		}
+	ret = execute_server_command(&s->rfds);
+	if (ret < 0) {
+		PARA_EMERG_LOG("%s\n", para_strerror(-ret));
+		sched_shutdown();
+		return;
 	}
-
 	/* Check the list of connected clients. */
 	list_for_each_entry_safe(client, tmp, &afs_client_list, node) {
-		if (FD_ISSET(client->fd, &s->rfds))
-			execute_afs_command(client->fd, ct->cookie);
-		else { /* prevent bogus connection flooding */
+		ret = execute_afs_command(client->fd, &s->rfds, ct->cookie);
+		if (ret == 0) { /* prevent bogus connection flooding */
 			struct timeval diff;
 			tv_diff(now, &client->connect_time, &diff);
 			if (diff.tv_sec < AFS_CLIENT_TIMEOUT)