| /* Disk protection for HP machines. |
| * |
| * Copyright 2008 Eric Piel |
| * Copyright 2009 Pavel Machek <pavel@suse.cz> |
| * |
| * GPLv2. |
| */ |
| |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include <unistd.h> |
| #include <fcntl.h> |
| #include <sys/stat.h> |
| #include <sys/types.h> |
| #include <string.h> |
| #include <stdint.h> |
| #include <errno.h> |
| #include <signal.h> |
| |
| void write_int(char *path, int i) |
| { |
| char buf[1024]; |
| int fd = open(path, O_RDWR); |
| if (fd < 0) { |
| perror("open"); |
| exit(1); |
| } |
| sprintf(buf, "%d", i); |
| if (write(fd, buf, strlen(buf)) != strlen(buf)) { |
| perror("write"); |
| exit(1); |
| } |
| close(fd); |
| } |
| |
| void set_led(int on) |
| { |
| write_int("/sys/class/leds/hp::hddprotect/brightness", on); |
| } |
| |
| void protect(int seconds) |
| { |
| write_int("/sys/block/sda/device/unload_heads", seconds*1000); |
| } |
| |
| int on_ac(void) |
| { |
| // /sys/class/power_supply/AC0/online |
| } |
| |
| int lid_open(void) |
| { |
| // /proc/acpi/button/lid/LID/state |
| } |
| |
| void ignore_me(void) |
| { |
| protect(0); |
| set_led(0); |
| |
| } |
| |
| int main(int argc, char* argv[]) |
| { |
| int fd, ret; |
| |
| fd = open("/dev/freefall", O_RDONLY); |
| if (fd < 0) { |
| perror("open"); |
| return EXIT_FAILURE; |
| } |
| |
| signal(SIGALRM, ignore_me); |
| |
| for (;;) { |
| unsigned char count; |
| |
| ret = read(fd, &count, sizeof(count)); |
| alarm(0); |
| if ((ret == -1) && (errno == EINTR)) { |
| /* Alarm expired, time to unpark the heads */ |
| continue; |
| } |
| |
| if (ret != sizeof(count)) { |
| perror("read"); |
| break; |
| } |
| |
| protect(21); |
| set_led(1); |
| if (1 || on_ac() || lid_open()) { |
| alarm(2); |
| } else { |
| alarm(20); |
| } |
| } |
| |
| close(fd); |
| return EXIT_SUCCESS; |
| } |