X-Git-Url: https://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?a=blobdiff_plain;f=src%2Ffsckd%2Ffsckd.c;h=91b8df03a80e26355a7e11cd65fcc1f1e3212d5a;hb=867ce6a516b8978dfbef479392687767787dff8f;hp=77808608ac6a2dd7260c1a995d4488a6170a6df4;hpb=df4573e85190759e39b417d4d4a8c73eda6a05e5;p=elogind.git diff --git a/src/fsckd/fsckd.c b/src/fsckd/fsckd.c index 77808608a..91b8df03a 100644 --- a/src/fsckd/fsckd.c +++ b/src/fsckd/fsckd.c @@ -47,6 +47,7 @@ #define IDLE_TIME_SECONDS 30 #define PLYMOUTH_REQUEST_KEY "K\2\2\3" +#define CLIENTS_MAX 128 struct Manager; @@ -64,6 +65,8 @@ typedef struct Client { size_t buflen; bool cancelled; + sd_event_source *event_source; + LIST_FIELDS(struct Client, clients); } Client; @@ -71,11 +74,15 @@ typedef struct Manager { sd_event *event; LIST_HEAD(Client, clients); + unsigned n_clients; + + size_t clear; - int clear; int connection_fd; + sd_event_source *connection_event_source; + + bool show_status_console; - FILE *console; double percent; int numdevices; @@ -86,11 +93,41 @@ typedef struct Manager { bool cancel_requested; } Manager; -static int connect_plymouth(Manager *m); -static int update_global_progress(Manager *m); +static void client_free(Client *c); static void manager_free(Manager *m); + +DEFINE_TRIVIAL_CLEANUP_FUNC(Client*, client_free); DEFINE_TRIVIAL_CLEANUP_FUNC(Manager*, manager_free); -#define _cleanup_manager_free_ _cleanup_(manager_freep) + +static int manager_write_console(Manager *m, const char *message) { + _cleanup_fclose_ FILE *console = NULL; + int l; + size_t j; + + assert(m); + + if (!m->show_status_console) + return 0; + + /* Reduce the SAK window by opening and closing console on every request */ + console = fopen("/dev/console", "we"); + if (!console) + return -errno; + + if (message) { + fprintf(console, "\r%s\r%n", message, &l); + if (m->clear < (size_t)l) + m->clear = (size_t)l; + } else { + fputc('\r', console); + for (j = 0; j < m->clear; j++) + fputc(' ', console); + fputc('\r', console); + } + fflush(console); + + return 0; +} static double compute_percent(int pass, size_t cur, size_t max) { /* Values stolen from e2fsck */ @@ -137,14 +174,18 @@ static int client_request_cancel(Client *c) { static void client_free(Client *c) { assert(c); - if (c->manager) + if (c->manager) { LIST_REMOVE(clients, c->manager->clients, c); + c->manager->n_clients--; + } + + sd_event_source_unref(c->event_source); safe_close(c->fd); free(c); } -static void plymouth_disconnect(Manager *m) { +static void manager_disconnect_plymouth(Manager *m) { assert(m); m->plymouth_event_source = sd_event_source_unref(m->plymouth_event_source); @@ -152,7 +193,7 @@ static void plymouth_disconnect(Manager *m) { m->plymouth_cancel_sent = false; } -static int plymouth_feedback_handler(sd_event_source *s, int fd, uint32_t revents, void *userdata) { +static int manager_plymouth_feedback_handler(sd_event_source *s, int fd, uint32_t revents, void *userdata) { Manager *m = userdata; Client *current; char buffer[6]; @@ -163,11 +204,11 @@ static int plymouth_feedback_handler(sd_event_source *s, int fd, uint32_t revent l = read(m->plymouth_fd, buffer, sizeof(buffer)); if (l < 0) { log_warning_errno(errno, "Got error while reading from plymouth: %m"); - plymouth_disconnect(m); + manager_disconnect_plymouth(m); return -errno; } if (l == 0) { - plymouth_disconnect(m); + manager_disconnect_plymouth(m); return 0; } @@ -186,7 +227,37 @@ static int plymouth_feedback_handler(sd_event_source *s, int fd, uint32_t revent return 0; } -static int send_message_plymouth_socket(int plymouth_fd, const char *message, bool update) { +static int manager_connect_plymouth(Manager *m) { + union sockaddr_union sa = PLYMOUTH_SOCKET; + int r; + + /* try to connect or reconnect if sending a message */ + if (m->plymouth_fd >= 0) + return 0; + + m->plymouth_fd = socket(AF_UNIX, SOCK_STREAM|SOCK_CLOEXEC, 0); + if (m->plymouth_fd < 0) + return log_warning_errno(errno, "Connection to plymouth socket failed: %m"); + + if (connect(m->plymouth_fd, &sa.sa, offsetof(struct sockaddr_un, sun_path) + 1 + strlen(sa.un.sun_path+1)) < 0) { + r = log_warning_errno(errno, "Couldn't connect to plymouth: %m"); + goto fail; + } + + r = sd_event_add_io(m->event, &m->plymouth_event_source, m->plymouth_fd, EPOLLIN, manager_plymouth_feedback_handler, m); + if (r < 0) { + log_warning_errno(r, "Can't listen to plymouth socket: %m"); + goto fail; + } + + return 1; + +fail: + manager_disconnect_plymouth(m); + return r; +} + +static int plymouth_send_message(int plymouth_fd, const char *message, bool update) { _cleanup_free_ char *packet = NULL; int n; char mode = 'M'; @@ -200,11 +271,11 @@ static int send_message_plymouth_socket(int plymouth_fd, const char *message, bo return loop_write(plymouth_fd, packet, n + 1, true); } -static int send_message_plymouth(Manager *m, const char *message) { +static int manager_send_plymouth_message(Manager *m, const char *message) { const char *plymouth_cancel_message = NULL; int r; - r = connect_plymouth(m); + r = manager_connect_plymouth(m); if (r < 0) return r; @@ -219,7 +290,7 @@ static int send_message_plymouth(Manager *m, const char *message) { plymouth_cancel_message = strjoina("fsckd-cancel-msg:", _("Press Ctrl+C to cancel all filesystem checks in progress")); - r = send_message_plymouth_socket(m->plymouth_fd, plymouth_cancel_message, false); + r = plymouth_send_message(m->plymouth_fd, plymouth_cancel_message, false); if (r < 0) log_warning_errno(r, "Can't send filesystem cancel message to plymouth: %m"); @@ -227,23 +298,23 @@ static int send_message_plymouth(Manager *m, const char *message) { m->plymouth_cancel_sent = false; - r = send_message_plymouth_socket(m->plymouth_fd, "", false); + r = plymouth_send_message(m->plymouth_fd, "", false); if (r < 0) log_warning_errno(r, "Can't clear plymouth filesystem cancel message: %m"); } - r = send_message_plymouth_socket(m->plymouth_fd, message, true); + r = plymouth_send_message(m->plymouth_fd, message, true); if (r < 0) return log_warning_errno(r, "Couldn't send \"%s\" to plymouth: %m", message); return 0; } -static int update_global_progress(Manager *m) { +static int manager_update_global_progress(Manager *m) { Client *current = NULL; _cleanup_free_ char *console_message = NULL; _cleanup_free_ char *fsck_message = NULL; - int current_numdevices = 0, l = 0, r; + int current_numdevices = 0, r; double current_percent = 100; /* get the overall percentage */ @@ -270,61 +341,28 @@ static int update_global_progress(Manager *m) { if (asprintf(&fsck_message, "fsckd:%d:%3.1f:%s", m->numdevices, m->percent, console_message) < 0) return -ENOMEM; - /* write to console */ - if (m->console) { - fprintf(m->console, "\r%s\r%n", console_message, &l); - fflush(m->console); - } + r = manager_write_console(m, console_message); + if (r < 0) + return r; /* try to connect to plymouth and send message */ - r = send_message_plymouth(m, fsck_message); + r = manager_send_plymouth_message(m, fsck_message); if (r < 0) log_debug("Couldn't send message to plymouth"); - if (l > m->clear) - m->clear = l; - } - return 0; -} - -static int connect_plymouth(Manager *m) { - union sockaddr_union sa = PLYMOUTH_SOCKET; - int r; - - /* try to connect or reconnect if sending a message */ - if (m->plymouth_fd >= 0) - return 0; - - m->plymouth_fd = socket(AF_UNIX, SOCK_STREAM|SOCK_CLOEXEC, 0); - if (m->plymouth_fd < 0) - return log_warning_errno(errno, "Connection to plymouth socket failed: %m"); - - if (connect(m->plymouth_fd, &sa.sa, offsetof(struct sockaddr_un, sun_path) + 1 + strlen(sa.un.sun_path+1)) < 0) { - r = log_warning_errno(errno, "Couldn't connect to plymouth: %m"); - goto fail; - } - - r = sd_event_add_io(m->event, &m->plymouth_event_source, m->plymouth_fd, EPOLLIN, plymouth_feedback_handler, m); - if (r < 0) { - log_warning_errno(r, "Can't listen to plymouth socket: %m"); - goto fail; } - return 0; - -fail: - plymouth_disconnect(m); - return r; } -static int progress_handler(sd_event_source *s, int fd, uint32_t revents, void *userdata) { +static int client_progress_handler(sd_event_source *s, int fd, uint32_t revents, void *userdata) { Client *client = userdata; - Manager *m = NULL; FsckProgress fsck_data; size_t buflen; + Manager *m; int r; assert(client); + m = client->manager; /* check first if we need to cancel this client */ @@ -340,7 +378,7 @@ static int progress_handler(sd_event_source *s, int fd, uint32_t revents, void * else { log_warning("Closing bad behaving fsck client connection at fd %d", client->fd); client_free(client); - r = update_global_progress(m); + r = manager_update_global_progress(m); if (r < 0) log_warning_errno(r, "Couldn't update global progress: %m"); } @@ -366,17 +404,18 @@ static int progress_handler(sd_event_source *s, int fd, uint32_t revents, void * } else log_error_errno(r, "Unknown error while trying to read fsck data: %m"); - r = update_global_progress(m); + r = manager_update_global_progress(m); if (r < 0) log_warning_errno(r, "Couldn't update global progress: %m"); return 0; } -static int new_connection_handler(sd_event_source *s, int fd, uint32_t revents, void *userdata) { +static int manager_new_connection_handler(sd_event_source *s, int fd, uint32_t revents, void *userdata) { + _cleanup_(client_freep) Client *c = NULL; + _cleanup_close_ int new_client_fd = -1; Manager *m = userdata; - Client *client = NULL; - int new_client_fd, r; + int r; assert(m); @@ -385,23 +424,37 @@ static int new_connection_handler(sd_event_source *s, int fd, uint32_t revents, if (new_client_fd < 0) return log_error_errno(errno, "Couldn't accept a new connection: %m"); + if (m->n_clients >= CLIENTS_MAX) { + log_error("Too many clients, refusing connection."); + return 0; + } + log_debug("New fsck client connected to fd: %d", new_client_fd); - client = new0(Client, 1); - if (!client) - return log_oom(); - client->fd = new_client_fd; - client->manager = m; - LIST_PREPEND(clients, m->clients, client); - r = sd_event_add_io(m->event, NULL, client->fd, EPOLLIN, progress_handler, client); + c = new0(Client, 1); + if (!c) { + log_oom(); + return 0; + } + + c->fd = new_client_fd; + new_client_fd = -1; + + r = sd_event_add_io(m->event, &c->event_source, c->fd, EPOLLIN, client_progress_handler, c); if (r < 0) { - client_free(client); - return r; + log_oom(); + return 0; } + + LIST_PREPEND(clients, m->clients, c); + m->n_clients++; + c->manager = m; + /* only request the client to cancel now in case the request is dropped by the client (chance to recancel) */ if (m->cancel_requested) - client_request_cancel(client); + client_request_cancel(c); + c = NULL; return 0; } @@ -410,33 +463,23 @@ static void manager_free(Manager *m) { return; /* clear last line */ - if (m->console && m->clear > 0) { - unsigned j; - - fputc('\r', m->console); - for (j = 0; j < (unsigned) m->clear; j++) - fputc(' ', m->console); - fputc('\r', m->console); - fflush(m->console); - } - - plymouth_disconnect(m); + manager_write_console(m, NULL); + sd_event_source_unref(m->connection_event_source); safe_close(m->connection_fd); - if (m->console) - fclose(m->console); - while (m->clients) client_free(m->clients); + manager_disconnect_plymouth(m); + sd_event_unref(m->event); free(m); } static int manager_new(Manager **ret, int fd) { - _cleanup_manager_free_ Manager *m = NULL; + _cleanup_(manager_freep) Manager *m = NULL; int r; assert(ret); @@ -445,19 +488,21 @@ static int manager_new(Manager **ret, int fd) { if (!m) return -ENOMEM; + m->plymouth_fd = -1; + m->connection_fd = fd; + m->percent = 100; + r = sd_event_default(&m->event); if (r < 0) return r; - m->connection_fd = fd; - if (access("/run/systemd/show-status", F_OK) >= 0) { - m->console = fopen("/dev/console", "we"); - if (!m->console) - return log_warning_errno(errno, "Can't connect to /dev/console: %m"); - } - m->percent = 100; + if (access("/run/systemd/show-status", F_OK) >= 0) + m->show_status_console = true; + + r = sd_event_add_io(m->event, &m->connection_event_source, fd, EPOLLIN, manager_new_connection_handler, m); + if (r < 0) + return r; - m->plymouth_fd = -1; *ret = m; m = NULL; @@ -548,7 +593,7 @@ static int parse_argv(int argc, char *argv[]) { } int main(int argc, char *argv[]) { - _cleanup_manager_free_ Manager *m = NULL; + _cleanup_(manager_freep) Manager *m = NULL; int fd = -1; int r, n; @@ -582,12 +627,6 @@ int main(int argc, char *argv[]) { goto finish; } - r = sd_event_add_io(m->event, NULL, fd, EPOLLIN, new_connection_handler, m); - if (r < 0) { - log_error_errno(r, "Can't listen to connection socket: %m"); - goto finish; - } - r = run_event_loop_with_timeout(m->event, IDLE_TIME_SECONDS * USEC_PER_SEC); if (r < 0) { log_error_errno(r, "Failed to run event loop: %m");