-rw-r--r-- | core/opiealarm/opiealarm.c | 246 |
1 files changed, 194 insertions, 52 deletions
diff --git a/core/opiealarm/opiealarm.c b/core/opiealarm/opiealarm.c index 33d631d..3a06236 100644 --- a/core/opiealarm/opiealarm.c +++ b/core/opiealarm/opiealarm.c @@ -20,13 +20,29 @@ #include <time.h> #include <stdlib.h> #include <syslog.h> #include <signal.h> +#include <errno.h> +#include <string.h> #define PIDFILE "/var/run/opiealarm.pid" +#define APMFILE "/proc/apm" + +FILE *log; // debug only -FILE *log; +int resume ( void ); +int suspend ( void ); +int main ( int argc, char **argv ); +int fork_with_pidfile ( void ); +int kill_with_pidfile ( void ); +void remove_pidfile ( void ); +void usage ( void ); +void sig_handler ( int sig ); +void error_msg_and_die ( int perr, const char *msg ); +int onac ( void ); + +static int opiealarm_was_running; void log_msg ( const char *msg ) { @@ -41,23 +57,142 @@ void error_msg_and_die ( int perr, const char *msg ) if ( perr ) log_msg ( strerror ( errno )); log_msg ( msg ); - unlink ( PIDFILE ); + while ( 1 ) // pretend we are waiting on RTC, so opiealarm -r can kill us + sleep ( 1 ); +} + + +void sig_handler ( int sig ) +{ + log_msg ( "GOT SIGNAL -> EXITING\n" ); + fclose ( log ); + remove_pidfile ( ); + exit ( 0 ); +} + +void usage ( void ) +{ + fprintf ( stderr, "Usage: opiealarm -r|-s\n" ); exit ( 1 ); } +int fork_with_pidfile ( void ) +{ + FILE *fp; + pid_t pid; + + pid = fork ( ); + + if ( pid > 0 ) + exit ( 0 ); + else if ( pid < 0 ) { + perror ( "forking failed" ); + return 0; + } + + signal ( SIGTERM, sig_handler ); + signal ( SIGINT, sig_handler ); + + // save pid + if (( fp = fopen ( PIDFILE, "w" ))) { + fprintf ( fp, "%d", getpid ( )); + fclose ( fp ); + + // detach + close ( 0 ); + close ( 1 ); + close ( 2 ); + + setpgid ( 0, 0 ); + + return 1; + } + else { + perror ( PIDFILE ); + return 0; + } +} + +int kill_with_pidfile ( void ) +{ + FILE *fp; + pid_t pid; + + if (( fp = fopen ( PIDFILE, "r" ))) { + if ( fscanf ( fp, "%d", &pid ) == 1 ) + return ( kill ( pid, SIGTERM ) == 0 ) ? 1 : 0; + fclose ( fp ); + } + return 0; +} + +void remove_pidfile ( void ) +{ + unlink ( PIDFILE ); + + signal ( SIGTERM, SIG_DFL ); + signal ( SIGINT, SIG_DFL ); +} + + +int main ( int argc, char **argv ) +{ + int mode = 0; + int opt; + + while (( opt = getopt ( argc, argv, "rs" )) != EOF ) { + switch ( opt ) { + case 's': + mode = 's'; + break; + case 'r': + mode = 'r'; + break; + default: + usage ( ); + } + } + + if ( geteuid ( ) != 0 ) { + fprintf ( stderr, "You need root priviledges to run opiealarm." ); + return 2; + } + + if ( !mode ) + usage ( ); + + // kill running opiealarm + opiealarm_was_running = kill_with_pidfile ( ); + remove_pidfile ( ); + + if ( mode == 'r' ) + return resume ( ); + else + return suspend ( ); + return 0; +} + -void suspend_on_rtc ( ) +int suspend ( void ) { FILE *fp; char buf [64]; time_t t; struct tm *tm; int fd; + + + if ( !fork_with_pidfile ( )) + return 3; + + log = fopen ( "/tmp/opiealarm.log", "w" ); + log_msg ( "STARTING\n" ); + if (!( fp = fopen ( "/etc/resumeat", "r" ))) error_msg_and_die ( 1, "/etc/resumeat" ); if ( !fgets ( buf, sizeof( buf ) - 1, fp )) @@ -77,10 +212,10 @@ void suspend_on_rtc ( ) fprintf ( log, "Setting RTC alarm to %d\n", t ); tm = gmtime ( &t ); - // Write alarm time to RTC - if (( fd = open ( "/dev/misc/rtc", O_RDWR ) < 0 ); + // Write alarm time to RTC + if (( fd = open ( "/dev/misc/rtc", O_RDWR )) < 0 ) error_msg_and_die ( 1, "/dev/misc/rtc" ); // set alarm time if ( ioctl ( fd, RTC_ALM_SET, tm ) < 0 ) error_msg_and_die ( 1, "ioctl RTC_ALM_SET" ); @@ -100,68 +235,75 @@ void suspend_on_rtc ( ) if ( ioctl ( fd, RTC_AIE_OFF, 0 ) < 0 ) error_msg_and_die ( 1, "ioctl RTC_AIE_OFF" ); close ( fd ); -} - - -void sig_handler ( int sig ) -{ - log_msg ( "GOT SIGNAL -> EXITING\n" ); + + log_msg ( "EXITING\n" ); + fclose ( log ); - unlink ( PIDFILE ); - exit ( 0 ); + remove_pidfile ( ); + + return 0; } -int main ( ) +static int onac ( void ) { FILE *fp; - pid_t pid; + int on = 0; - if ( geteuid ( ) != 0 ) { - fprintf ( stderr, "You need root priviledges to run opiealarm." ); - return 1; - } + if (( fp = fopen ( APMFILE, "r" ))) { + int ac = 0; - pid = fork ( ); - - if ( pid > 0 ) - return 0; - else if ( pid < 0 ) { - perror ( "Could not fork." ); - return 2; - } - - // save pid - if (!( fp = fopen ( PIDFILE, "w" ))) { - perror ( PIDFILE ); - return 3; + if ( fscanf ( fp, "%*[^ ] %*d.%*d 0x%*x 0x%x 0x%*x 0x%*x %*d%% %*i %*c", &ac ) == 1 ) + on = ( ac == 0x01 ) ? 1 : 0; + + fclose ( fp ); } - - fprintf ( fp, "%d", getpid ( )); - fclose ( fp ); + return on; +} - // detach - close ( 0 ); - close ( 1 ); - close ( 2 ); +int resume ( void ) +{ + FILE *fp; - setpgid ( 0, 0 ); + // re-suspend when on AC (optional) when woken up via RTC - log = fopen ( "/tmp/opiealarm.log", "w" ); - log_msg ( "STARTING\n" ); - - signal ( SIGTERM, sig_handler ); - signal ( SIGINT, sig_handler ); + if ( !opiealarm_was_running ) { // opiealarm -s got it's RTC signal -> wake up by RTC + if ( onac ( )) { + time_t start, now; + char *argv [4]; + + if ( !fork_with_pidfile ( )) + return 4; - extractevent ( ); - - signal ( SIGTERM, SIG_DFL ); - signal ( SIGINT, SIG_DFL ); + // sleep 120sec (not less!) + time ( &start ); + do { + sleep ( 1 ); + time ( &now ); + } while (( now - start ) < 120 ); - log_msg ( "EXITING\n" ); + if ( onac ( )) { // still on ac + // system() without fork + argv[0] = "qcop"; + argv[1] = "QPE/Desktop"; + argv[2] = "suspend()"; + argv[3] = 0; - fclose ( log ); - unlink ( PIDFILE ); + // hard coded for now ...but needed + setenv ( "LOGNAME", "root", 1 ); + setenv ( "HOME", "/root", 1 ); + setenv ( "LD_LIBRARY_PATH", "/opt/QtPalmtop/lib", 1 ); + setenv ( "QTDIR", "/opt/QtPalmtop", 1 ); + + remove_pidfile ( ); + + execv ( "/opt/QtPalmtop/bin/qcop", argv ); + + perror ( "exec for qcop failed" ); + return 5; + } + } + } return 0; } |