Blame SOURCES/fapolicyd-fgets-update-thread.patch

fa39dc
diff -up ./src/cli/fapolicyd-cli.c.upgrade-thread ./src/cli/fapolicyd-cli.c
fa39dc
--- ./src/cli/fapolicyd-cli.c.upgrade-thread	2022-08-03 18:00:02.374999369 +0200
fa39dc
+++ ./src/cli/fapolicyd-cli.c	2022-08-03 18:00:09.802830497 +0200
fa39dc
@@ -482,7 +482,7 @@ static int do_update(void)
fa39dc
 		}
fa39dc
 	}
fa39dc
 
fa39dc
-	ssize_t ret = write(fd, "1", 2);
fa39dc
+	ssize_t ret = write(fd, "1\n", 3);
fa39dc
 
fa39dc
 	if (ret == -1) {
fa39dc
 		fprintf(stderr, "Write: %s -> %s\n", _pipe, strerror(errno));
fa39dc
diff -up ./src/library/database.c.upgrade-thread ./src/library/database.c
fa39dc
--- ./src/library/database.c.upgrade-thread	2022-06-21 16:55:47.000000000 +0200
fa39dc
+++ ./src/library/database.c	2022-08-03 17:58:04.034689808 +0200
fa39dc
@@ -34,6 +34,7 @@
fa39dc
 #include <errno.h>
fa39dc
 #include <unistd.h>
fa39dc
 #include <fcntl.h>
fa39dc
+#include <ctype.h>
fa39dc
 #include <gcrypt.h>
fa39dc
 #include <signal.h>
fa39dc
 #include <sys/stat.h>
fa39dc
@@ -43,6 +44,7 @@
fa39dc
 #include "message.h"
fa39dc
 #include "llist.h"
fa39dc
 #include "file.h"
fa39dc
+#include "fd-fgets.h"
fa39dc
 
fa39dc
 #include "fapolicyd-backend.h"
fa39dc
 #include "backend-manager.h"
fa39dc
@@ -1181,6 +1183,7 @@ static void *update_thread_main(void *ar
fa39dc
 			return NULL;
fa39dc
 	}
fa39dc
 
fa39dc
+	fcntl(ffd[0].fd, F_SETFL, O_NONBLOCK);
fa39dc
 	ffd[0].events = POLLIN;
fa39dc
 
fa39dc
 	while (!stop) {
fa39dc
@@ -1200,97 +1203,102 @@ static void *update_thread_main(void *ar
fa39dc
 			} else {
fa39dc
 				msg(LOG_ERR, "Update poll error (%s)",
fa39dc
 				    strerror_r(errno, err_buff, BUFFER_SIZE));
fa39dc
-				goto err_out;
fa39dc
+				goto finalize;
fa39dc
 			}
fa39dc
 		} else if (rc == 0) {
fa39dc
 #ifdef DEBUG
fa39dc
 			msg(LOG_DEBUG, "Update poll timeout expired");
fa39dc
 #endif
fa39dc
-			if (db_operation != DB_NO_OP)
fa39dc
-				goto handle_db_ops;
fa39dc
 			continue;
fa39dc
 		} else {
fa39dc
 			if (ffd[0].revents & POLLIN) {
fa39dc
-				ssize_t count = read(ffd[0].fd, buff,
fa39dc
-						     BUFFER_SIZE-1);
fa39dc
 
fa39dc
-				if (count == -1) {
fa39dc
-					msg(LOG_ERR,
fa39dc
-					   "Failed to read from a pipe %s (%s)",
fa39dc
-					   fifo_path,
fa39dc
-					   strerror_r(errno, err_buff,
fa39dc
-						      BUFFER_SIZE));
fa39dc
-					goto err_out;
fa39dc
-				}
fa39dc
+				do {
fa39dc
+					fd_fgets_rewind();
fa39dc
+					int res = fd_fgets(buff, sizeof(buff), ffd[0].fd);
fa39dc
 
fa39dc
-				if (count == 0) {
fa39dc
-#ifdef DEBUG
fa39dc
-					msg(LOG_DEBUG,
fa39dc
-					    "Buffer contains zero bytes!");
fa39dc
-#endif
fa39dc
-					continue;
fa39dc
-				} else // Manually terminate buff
fa39dc
-					buff[count] = 0;
fa39dc
-#ifdef DEBUG
fa39dc
-				msg(LOG_DEBUG, "Buffer contains: \"%s\"", buff);
fa39dc
-#endif
fa39dc
-				for (int i = 0 ; i < count ; i++) {
fa39dc
-					// assume file name
fa39dc
-					// operation = 0
fa39dc
-					if (buff[i] == '/') {
fa39dc
-						db_operation = ONE_FILE;
fa39dc
+					// nothing to read
fa39dc
+					if (res == -1)
fa39dc
 						break;
fa39dc
-					}
fa39dc
+					else if (res > 0) {
fa39dc
+						char* end  = strchr(buff, '\n');
fa39dc
 
fa39dc
-					if (buff[i] == '1') {
fa39dc
-						db_operation = RELOAD_DB;
fa39dc
-						break;
fa39dc
+						if (end == NULL) {
fa39dc
+							msg(LOG_ERR, "Too long line?");
fa39dc
+							continue;
fa39dc
+						}
fa39dc
+
fa39dc
+						int count = end - buff;
fa39dc
+
fa39dc
+						*end = '\0';
fa39dc
+
fa39dc
+						for (int i = 0 ; i < count ; i++) {
fa39dc
+							// assume file name
fa39dc
+							// operation = 0
fa39dc
+							if (buff[i] == '/') {
fa39dc
+								db_operation = ONE_FILE;
fa39dc
+								break;
fa39dc
+							}
fa39dc
+
fa39dc
+							if (buff[i] == '1') {
fa39dc
+								db_operation = RELOAD_DB;
fa39dc
+								break;
fa39dc
+							}
fa39dc
+
fa39dc
+							if (buff[i] == '2') {
fa39dc
+								db_operation = FLUSH_CACHE;
fa39dc
+								break;
fa39dc
+							}
fa39dc
+
fa39dc
+							if (isspace(buff[i]))
fa39dc
+								continue;
fa39dc
+
fa39dc
+							msg(LOG_ERR, "Cannot handle data \"%s\" from pipe", buff);
fa39dc
+							break;
fa39dc
+						}
fa39dc
+
fa39dc
+						*end = '\n';
fa39dc
+
fa39dc
+						// got "1" -> reload db
fa39dc
+						if (db_operation == RELOAD_DB) {
fa39dc
+							db_operation = DB_NO_OP;
fa39dc
+							msg(LOG_INFO,
fa39dc
+								"It looks like there was an update of the system... Syncing DB.");
fa39dc
+
fa39dc
+							backend_close();
fa39dc
+							backend_init(config);
fa39dc
+							backend_load(config);
fa39dc
+
fa39dc
+							if ((rc = update_database(config))) {
fa39dc
+								msg(LOG_ERR,
fa39dc
+									"Cannot update trust database!");
fa39dc
+								close(ffd[0].fd);
fa39dc
+								backend_close();
fa39dc
+								unlink_fifo();
fa39dc
+								exit(rc);
fa39dc
+							}
fa39dc
+
fa39dc
+							msg(LOG_INFO, "Updated");
fa39dc
+
fa39dc
+							// Conserve memory
fa39dc
+							backend_close();
fa39dc
+							// got "2" -> flush cache
fa39dc
+						} else if (db_operation == FLUSH_CACHE) {
fa39dc
+							db_operation = DB_NO_OP;
fa39dc
+							needs_flush = true;
fa39dc
+						} else if (db_operation == ONE_FILE) {
fa39dc
+							db_operation = DB_NO_OP;
fa39dc
+							if (handle_record(buff))
fa39dc
+								continue;
fa39dc
+						}
fa39dc
 					}
fa39dc
 
fa39dc
-					if (buff[i] == '2') {
fa39dc
-						db_operation = FLUSH_CACHE;
fa39dc
-						break;
fa39dc
-					}
fa39dc
-				}
fa39dc
-
fa39dc
-handle_db_ops:
fa39dc
-				// got "1" -> reload db
fa39dc
-				if (db_operation == RELOAD_DB) {
fa39dc
-					db_operation = DB_NO_OP;
fa39dc
-					msg(LOG_INFO,
fa39dc
-	    "It looks like there was an update of the system... Syncing DB.");
fa39dc
-
fa39dc
-					backend_close();
fa39dc
-					backend_init(config);
fa39dc
-					backend_load(config);
fa39dc
-
fa39dc
-					if ((rc = update_database(config))) {
fa39dc
-						msg(LOG_ERR,
fa39dc
-						   "Cannot update trust database!");
fa39dc
-						close(ffd[0].fd);
fa39dc
-						backend_close();
fa39dc
-						unlink_fifo();
fa39dc
-						exit(rc);
fa39dc
-					} else
fa39dc
-						msg(LOG_INFO, "Updated");
fa39dc
-
fa39dc
-					// Conserve memory
fa39dc
-					backend_close();
fa39dc
-				// got "2" -> flush cache
fa39dc
-				} else if (db_operation == FLUSH_CACHE) {
fa39dc
-					db_operation = DB_NO_OP;
fa39dc
-					needs_flush = true;
fa39dc
-				} else if (db_operation == ONE_FILE) {
fa39dc
-					db_operation = DB_NO_OP;
fa39dc
-					if (handle_record(buff))
fa39dc
-						continue;
fa39dc
-				}
fa39dc
+				} while(!fd_fgets_eof());
fa39dc
 			}
fa39dc
 		}
fa39dc
-
fa39dc
 	}
fa39dc
 
fa39dc
-err_out:
fa39dc
+finalize:
fa39dc
 	close(ffd[0].fd);
fa39dc
 	unlink_fifo();
fa39dc