X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?p=elogind.git;a=blobdiff_plain;f=src%2Ffsckd%2Ffsckd.c;h=1415b53dd487f9302ed55f409392d85aca7c458c;hp=14fa994008edd09f077ecd244c017ca5f654718d;hb=57b394b50579feeee90a4cb12243d5785a9176a6;hpb=f8824a512904449fa60333c500fc63c66b2937e5 diff --git a/src/fsckd/fsckd.c b/src/fsckd/fsckd.c index 14fa99400..1415b53dd 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,9 +74,12 @@ typedef struct Manager { sd_event *event; LIST_HEAD(Client, clients); + unsigned n_clients; int clear; + int connection_fd; + sd_event_source *connection_event_source; FILE *console; double percent; @@ -86,11 +92,11 @@ 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 double compute_percent(int pass, size_t cur, size_t max) { /* Values stolen from e2fsck */ @@ -110,36 +116,45 @@ static double compute_percent(int pass, size_t cur, size_t max) { (double) cur / max; } -static int request_cancel_client(Client *current) { +static int client_request_cancel(Client *c) { FsckdMessage cancel_msg = { .cancel = 1, }; ssize_t n; - n = send(current->fd, &cancel_msg, sizeof(FsckdMessage), 0); + assert(c); + + if (c->cancelled) + return 0; + + n = send(c->fd, &cancel_msg, sizeof(FsckdMessage), 0); if (n < 0) - return log_warning_errno(errno, "Cannot send cancel to fsck on (%u:%u): %m", major(current->devnum), minor(current->devnum)); + return log_warning_errno(errno, "Cannot send cancel to fsck on (%u:%u): %m", major(c->devnum), minor(c->devnum)); if ((size_t) n < sizeof(FsckdMessage)) { - log_warning("Short send when sending cancel to fsck on (%u:%u).", major(current->devnum), minor(current->devnum)); + log_warning("Short send when sending cancel to fsck on (%u:%u).", major(c->devnum), minor(c->devnum)); return -EIO; } - current->cancelled = true; - return 0; + c->cancelled = true; + return 1; } 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); @@ -147,7 +162,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]; @@ -158,30 +173,60 @@ 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; } - if (buffer[0] == '\15') + if (l > 1 && buffer[0] == '\15') log_error("Message update to plymouth wasn't delivered successfully"); /* the only answer support type we requested is a key interruption */ - if (buffer[0] == '\2' && buffer[5] == '\3') { + if (l > 2 && buffer[0] == '\2' && buffer[5] == '\3') { m->cancel_requested = true; /* cancel all connected clients */ LIST_FOREACH(clients, current, m->clients) - request_cancel_client(current); + client_request_cancel(current); } 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'; @@ -195,11 +240,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; @@ -214,7 +259,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"); @@ -222,19 +267,19 @@ 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; @@ -272,7 +317,7 @@ static int update_global_progress(Manager *m) { } /* 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"); @@ -282,51 +327,20 @@ static int update_global_progress(Manager *m) { 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 */ - if (m->cancel_requested) { - if (!client->cancelled) - request_cancel_client(client); - } + if (m->cancel_requested) + client_request_cancel(client); /* ensure we have enough data to read */ r = ioctl(fd, FIONREAD, &buflen); @@ -337,7 +351,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"); } @@ -363,17 +377,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); @@ -382,23 +397,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) - request_cancel_client(client); + client_request_cancel(c); + c = NULL; return 0; } @@ -417,23 +446,24 @@ static void manager_free(Manager *m) { fflush(m->console); } - plymouth_disconnect(m); - + 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); + + if (m->console) + fclose(m->console); + 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); @@ -442,19 +472,24 @@ 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"); + return -errno; } - m->percent = 100; - m->plymouth_fd = -1; + r = sd_event_add_io(m->event, &m->connection_event_source, fd, EPOLLIN, manager_new_connection_handler, m); + if (r < 0) + return r; + *ret = m; m = NULL; @@ -545,7 +580,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; @@ -579,12 +614,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");