mirror of
https://git.kernel.org/pub/scm/linux/kernel/git/stable/linux.git
synced 2024-08-23 01:10:04 +00:00
![Pali Rohár](/assets/img/avatar_default.png)
For security reasons I stopped using gmail account and kernel address is now up-to-date alias to my personal address. People periodically send me emails to address which they found in source code of drivers, so this change reflects state where people can contact me. [ Added .mailmap entry as per Joe Perches - Linus ] Signed-off-by: Pali Rohár <pali@kernel.org> Signed-off-by: Andrew Morton <akpm@linux-foundation.org> Cc: Greg Kroah-Hartman <gregkh@linuxfoundation.org> Cc: Joe Perches <joe@perches.com> Link: http://lkml.kernel.org/r/20200307104237.8199-1-pali@kernel.org Signed-off-by: Linus Torvalds <torvalds@linux-foundation.org>
173 lines
3.1 KiB
C
173 lines
3.1 KiB
C
// SPDX-License-Identifier: GPL-2.0-only
|
|
/* Disk protection for HP/DELL machines.
|
|
*
|
|
* Copyright 2008 Eric Piel
|
|
* Copyright 2009 Pavel Machek <pavel@ucw.cz>
|
|
* Copyright 2012 Sonal Santan
|
|
* Copyright 2014 Pali Rohár <pali@kernel.org>
|
|
*/
|
|
|
|
#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>
|
|
#include <sys/mman.h>
|
|
#include <sched.h>
|
|
#include <syslog.h>
|
|
|
|
static int noled;
|
|
static char unload_heads_path[64];
|
|
static char device_path[32];
|
|
static const char app_name[] = "FREE FALL";
|
|
|
|
static int set_unload_heads_path(char *device)
|
|
{
|
|
if (strlen(device) <= 5 || strncmp(device, "/dev/", 5) != 0)
|
|
return -EINVAL;
|
|
strncpy(device_path, device, sizeof(device_path) - 1);
|
|
|
|
snprintf(unload_heads_path, sizeof(unload_heads_path) - 1,
|
|
"/sys/block/%s/device/unload_heads", device+5);
|
|
return 0;
|
|
}
|
|
|
|
static int valid_disk(void)
|
|
{
|
|
int fd = open(unload_heads_path, O_RDONLY);
|
|
|
|
if (fd < 0) {
|
|
perror(unload_heads_path);
|
|
return 0;
|
|
}
|
|
|
|
close(fd);
|
|
return 1;
|
|
}
|
|
|
|
static 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);
|
|
}
|
|
|
|
static void set_led(int on)
|
|
{
|
|
if (noled)
|
|
return;
|
|
write_int("/sys/class/leds/hp::hddprotect/brightness", on);
|
|
}
|
|
|
|
static void protect(int seconds)
|
|
{
|
|
const char *str = (seconds == 0) ? "Unparked" : "Parked";
|
|
|
|
write_int(unload_heads_path, seconds*1000);
|
|
syslog(LOG_INFO, "%s %s disk head\n", str, device_path);
|
|
}
|
|
|
|
static int on_ac(void)
|
|
{
|
|
/* /sys/class/power_supply/AC0/online */
|
|
return 1;
|
|
}
|
|
|
|
static int lid_open(void)
|
|
{
|
|
/* /proc/acpi/button/lid/LID/state */
|
|
return 1;
|
|
}
|
|
|
|
static void ignore_me(int signum)
|
|
{
|
|
protect(0);
|
|
set_led(0);
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
int fd, ret;
|
|
struct stat st;
|
|
struct sched_param param;
|
|
|
|
if (argc == 1)
|
|
ret = set_unload_heads_path("/dev/sda");
|
|
else if (argc == 2)
|
|
ret = set_unload_heads_path(argv[1]);
|
|
else
|
|
ret = -EINVAL;
|
|
|
|
if (ret || !valid_disk()) {
|
|
fprintf(stderr, "usage: %s <device> (default: /dev/sda)\n",
|
|
argv[0]);
|
|
exit(1);
|
|
}
|
|
|
|
fd = open("/dev/freefall", O_RDONLY);
|
|
if (fd < 0) {
|
|
perror("/dev/freefall");
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
if (stat("/sys/class/leds/hp::hddprotect/brightness", &st))
|
|
noled = 1;
|
|
|
|
if (daemon(0, 0) != 0) {
|
|
perror("daemon");
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
openlog(app_name, LOG_CONS | LOG_PID | LOG_NDELAY, LOG_LOCAL1);
|
|
|
|
param.sched_priority = sched_get_priority_max(SCHED_FIFO);
|
|
sched_setscheduler(0, SCHED_FIFO, ¶m);
|
|
mlockall(MCL_CURRENT|MCL_FUTURE);
|
|
|
|
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);
|
|
}
|
|
|
|
closelog();
|
|
close(fd);
|
|
return EXIT_SUCCESS;
|
|
}
|