diff options
Diffstat (limited to 'bin/live-md5check/live-md5check.c')
| -rw-r--r-- | bin/live-md5check/live-md5check.c | 270 |
1 files changed, 270 insertions, 0 deletions
diff --git a/bin/live-md5check/live-md5check.c b/bin/live-md5check/live-md5check.c new file mode 100644 index 0000000..8dfd577 --- /dev/null +++ b/bin/live-md5check/live-md5check.c @@ -0,0 +1,270 @@ +/* casper-md5check - a tool to check md5sums and talk to usplash + (C) Canonical Ltd 2006 + Written by Tollef Fog Heen <tfheen@ubuntu.com> + + This program is free software; you can redistribute it and/or + modify it under the terms of the GNU General Public License as + published by the Free Software Foundation; either version 2 of the + License, or (at your option) any later version. + + This program is distributed in the hope that it will be useful, but + WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + General Public License for more details. + + You should have received a copy of the GNU General Public License + along with this program; if not, write to the Free Software + Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 + USA. */ + +#define _GNU_SOURCE +#include <stdio.h> +#include <unistd.h> +#include <sys/reboot.h> +#include <linux/reboot.h> +#include <string.h> +#include <errno.h> +#include <stdarg.h> +#include <sys/types.h> +#include <sys/stat.h> +#include <fcntl.h> +#include <stdlib.h> +#include <math.h> +#include <termios.h> + +#define USPLASH_FIFO "/dev/.initramfs/usplash_fifo" +#define MAXTRIES 5 +#include "md5.h" +#define DEBUG + +int write_and_retry(int fd, char *s) { + int try = 0, ret = 0; + char *end; + +#ifdef DEBUG + fprintf(stderr, "-> %s\n", s); +#endif + + end = s + strlen(s)+1; + + ret = write(fd, s, end - s); + while (s + ret < end && try < MAXTRIES) { + sleep(1); + s += ret; + ret = write(fd, s, strlen(s)+1); + try++; + } + return (s+ret == end ? 0 : 1); +} + +void usplash_timeout(int fd, int timeout) { + char *s; + + asprintf(&s, "TIMEOUT %d", timeout); + + write_and_retry(fd, s); + + free(s); + +} + +void usplash_failure(int fd, char *format, ...) { + char *s, *s1; + va_list argp; + + va_start(argp, format); + vasprintf(&s, format, argp); + va_end(argp); + + asprintf(&s1, "FAILURE %s", s); + + write_and_retry(fd, s1); + + free(s); + free(s1); +} + +void usplash_text(int fd, char *format, ...) { + char *s, *s1; + va_list argp; + + va_start(argp, format); + vasprintf(&s, format, argp); + va_end(argp); + + asprintf(&s1, "TEXT %s", s); + + write_and_retry(fd, s1); + + free(s); + free(s1); +} + +void usplash_urgent(int fd, char *format, ...) { + char *s, *s1; + va_list argp; + + va_start(argp, format); + vasprintf(&s, format, argp); + va_end(argp); + + asprintf(&s1, "TEXT-URGENT %s", s); + + write_and_retry(fd, s1); + + free(s); + free(s1); +} + + +void usplash_success(int fd, char *format, ...) { + char *s, *s1; + va_list argp; + + va_start(argp, format); + vasprintf(&s, format, argp); + va_end(argp); + + asprintf(&s1, "SUCCESS %s", s); + + write_and_retry(fd, s1); + + free(s); + free(s1); +} + +void usplash_progress(int fd, int progress) { + static int prevprogress = -1; + char *s; + + if (progress == prevprogress) + return; + prevprogress = progress; + + asprintf(&s, "PROGRESS %d", progress); + + write_and_retry(fd, s); + + free(s); +} + +int set_nocanonical_tty(int fd) { + struct termios t; + + if (tcgetattr(fd, &t) == -1) { + perror("tcgetattr"); + } + t.c_lflag &= ~ICANON; + t.c_cc[VMIN] = 1; + t.c_cc[VTIME] = 0; + return tcsetattr(fd, TCSANOW, &t); +} + +int main(int argc, char **argv) { + + int pipe_fd, check_fd; + int failed = 0; + + FILE *md5_file; + md5_state_t state; + md5_byte_t digest[16]; + char hex_output[16*2 + 1]; + char *checksum, *checkfile; + ssize_t tsize, csize; + + tsize = 0; + csize = 0; + + if (argc != 3) { + fprintf(stderr,"Wrong number of arguments\n"); + fprintf(stderr,"%s <root directory> <md5sum file>\n", argv[0]); + exit(1); + } + + if (chdir(argv[1]) != 0) { + perror("chdir"); + exit(1); + } + + pipe_fd = open(USPLASH_FIFO, O_WRONLY|O_NONBLOCK); + + if (pipe_fd == -1) { + /* We can't really do anything useful here */ + perror("Opening pipe"); + exit(1); + } + + + usplash_progress(pipe_fd, 0); + usplash_urgent(pipe_fd, "Checking integrity, this may take some time"); + md5_file = fopen(argv[2], "r"); + if (!md5_file) { + perror("fopen md5_file"); + exit(1); + } + while (fscanf(md5_file, "%as %as", &checksum, &checkfile) == 2) { + struct stat statbuf; + + if (stat(checkfile, &statbuf) == 0) { + tsize += statbuf.st_size; + } + + free(checksum); + free(checkfile); + } + + rewind(md5_file); + while (fscanf(md5_file, "%as %as", &checksum, &checkfile) == 2) { + char buf[BUFSIZ]; + ssize_t rsize; + int i; + + md5_init(&state); + + usplash_text(pipe_fd, "Checking %s", checkfile); + + check_fd = open(checkfile, O_RDONLY); + if (check_fd < 0) { + usplash_timeout(pipe_fd, 300); + usplash_failure(pipe_fd, "%s", strerror(errno)); + sleep(10); + } + + rsize = read(check_fd, buf, sizeof(buf)); + + while (rsize > 0) { + csize += rsize; + usplash_progress(pipe_fd, floorl(100*((long double)csize)/tsize)); + + md5_append(&state, (const md5_byte_t *)buf, rsize); + rsize = read(check_fd, buf, sizeof(buf)); + } + + close(check_fd); + md5_finish(&state, digest); + + for (i = 0; i < 16; i++) + sprintf(hex_output + i * 2, "%02x", digest[i]); + + if (strncmp(hex_output, checksum, strlen(hex_output)) == 0) { + usplash_success(pipe_fd, "OK"); + } else { + usplash_failure(pipe_fd, "mismatch"); + failed++; + } + free(checksum); + free(checkfile); + } + if (failed) { + usplash_urgent(pipe_fd, "Check finished: errors found in %d files!", failed); + } else { + usplash_urgent(pipe_fd, "Check finished: no errors found"); + } + usplash_urgent(pipe_fd, "Press any key to reboot your system"); + usplash_timeout(pipe_fd, 86400); + set_nocanonical_tty(0); + getchar(); + reboot(LINUX_REBOOT_CMD_RESTART); + return 0; + +} |