more_to_read

src/communication.c

@@ -545,7 +545,11 @@ struct ipc_error ipc_wait_event (struct ipc_ctx *ctx, struct ipc_event *event, i
 
 	for (size_t i = 0; i <= ctx->size; i++) {
 		// Something to read or connection.
-		if (ctx->pollfd[i].revents & POLLIN) {
+		if (ctx->pollfd[i].revents & POLLIN || ctx->cinfos[i].more_to_read == 1) {
+
+			// Avoiding loops.
+			ctx->cinfos[i].more_to_read = 0;
+
 			// In case there is something to read for the server socket: new client.
 			if (ctx->cinfos[i].type == IPC_CONNECTION_TYPE_SERVER) {
 				return ipc_accept_add (event, ctx, i);

src/ipc.h

@@ -231,6 +231,7 @@ enum ipc_connection_type {
 
 struct ipc_connection_info {
 	enum ipc_connection_type type;
+	short int more_to_read;
 	char *spath; // max size: PATH_MAX
 };
 
@@ -250,9 +251,9 @@ struct ipc_messages {
 struct ipc_switching {
 	int orig;
 	int dest;
-	enum ipccb (*orig_in)  (int origin_fd, struct ipc_message *m);
+	enum ipccb (*orig_in)  (int origin_fd, struct ipc_message *m, short int *more_to_read);
 	enum ipccb (*orig_out) (int origin_fd, struct ipc_message *m);
-	enum ipccb (*dest_in)  (int origin_fd, struct ipc_message *m);
+	enum ipccb (*dest_in)  (int origin_fd, struct ipc_message *m, short int *more_to_read);
 	enum ipccb (*dest_out) (int origin_fd, struct ipc_message *m);
 };
 
@@ -404,11 +405,11 @@ int ipc_switching_del (struct ipc_switchings *is, int fd);
 int ipc_switching_get (struct ipc_switchings *is, int fd);
 void ipc_switching_free (struct ipc_switchings *is);
 void ipc_switching_callbacks_ (struct ipc_ctx *ctx, int fd
-	, enum ipccb (*cb_in )(int fd, struct ipc_message *m));
+	, enum ipccb (*cb_in )(int fd, struct ipc_message *m, short int *more_to_read));
 void ipc_switching_callbacks (
 	  struct ipc_ctx *ctx
 	, int fd
-	, enum ipccb (*cb_in )(int fd, struct ipc_message *m)
+	, enum ipccb (*cb_in )(int fd, struct ipc_message *m, short int *more_to_read)
 	, enum ipccb (*cb_out)(int fd, struct ipc_message *m));
 
 int ipc_ctx_fd_type (struct ipc_ctx *ctx, int fd, enum ipc_connection_type type);

src/network.c

@@ -197,9 +197,10 @@ void ipc_switching_free (struct ipc_switchings *is)
 }
 
 enum ipccb
-default_cb_in(int fd, struct ipc_message *m)
+default_cb_in(int fd, struct ipc_message *m, short int *more_to_read)
 {
-	// TODO: fix buffer size for switching messages
+	*more_to_read = 0;
+
 	size_t msize = IPC_MAX_MESSAGE_SIZE;
 	SECURE_BUFFER_DECLARATION (char, buf, msize);
 	char *pbuf = buf;
@@ -251,7 +252,7 @@ default_cb_out(int fd, struct ipc_message *m)
 }
 
 void ipc_switching_callbacks_ (struct ipc_ctx *ctx, int fd
-	, enum ipccb (*cb_in )(int fd, struct ipc_message *m))
+	, enum ipccb (*cb_in )(int fd, struct ipc_message *m, short int *more_to_read))
 {
 	ipc_switching_callbacks (ctx, fd, cb_in, NULL);
 }
@@ -259,7 +260,7 @@ void ipc_switching_callbacks_ (struct ipc_ctx *ctx, int fd
 void ipc_switching_callbacks (
 	  struct ipc_ctx *ctx
 	, int fd
-	, enum ipccb (*cb_in )(int fd, struct ipc_message *m)
+	, enum ipccb (*cb_in )(int fd, struct ipc_message *m, short int *more_to_read)
 	, enum ipccb (*cb_out)(int fd, struct ipc_message *m))
 {
 	struct ipc_switching *sw = NULL;
@@ -297,6 +298,7 @@ struct ipc_error fd_switching_read (struct ipc_event *event, struct ipc_ctx *ctx
 
 	enum ipccb r;
 	int is_valid = 0;
+	short int more_to_read = 0;
 
 	is_valid = ipc_switching_get_ (&ctx->switchdb, talkingfd, &sw);
 
@@ -305,22 +307,24 @@ struct ipc_error fd_switching_read (struct ipc_event *event, struct ipc_ctx *ctx
 	if (sw->orig == talkingfd) {
 		dest_fd = sw->dest;
 		if (sw->orig_in == NULL) {
-			r = default_cb_in (talkingfd, &m);
+			r = default_cb_in (talkingfd, &m, &more_to_read);
 		}
 		else {
-			r = (*sw->orig_in)(talkingfd, &m);
+			r = (*sw->orig_in)(talkingfd, &m, &more_to_read);
 		}
 	}
 	else {
 		dest_fd = sw->orig;
 		if (sw->dest_in == NULL) {
-			r = default_cb_in (talkingfd, &m);
+			r = default_cb_in (talkingfd, &m, &more_to_read);
 		}
 		else {
-			r = (*sw->dest_in)(talkingfd, &m);
+			r = (*sw->dest_in)(talkingfd, &m, &more_to_read);
 		}
 	}
 
+	ctx->cinfos[index].more_to_read = more_to_read;
+
 	// Message reception OK: reading the message and put it in the list of messages to send.
 	if (r == IPC_CB_NO_ERROR) {
 		// In case of message reception: