-rw-r--r-- | core/opiealarm/opiealarm.c | 34 |
1 files changed, 27 insertions, 7 deletions
diff --git a/core/opiealarm/opiealarm.c b/core/opiealarm/opiealarm.c index ce0103e..d96ffaf 100644 --- a/core/opiealarm/opiealarm.c +++ b/core/opiealarm/opiealarm.c @@ -34,71 +34,87 @@ int resume ( int resuspend ); int suspend ( int fix_rtc ); 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; +static pid_t startpid = 0; + void log_msg ( const char *msg ) { if ( log ) { fprintf ( log, msg ); fflush ( log ); } } void error_msg_and_die ( int perr, const char *msg ) { if ( perr ) log_msg ( strerror ( errno )); log_msg ( msg ); + if ( getpid ( ) != startpid ) + kill ( startpid, SIGUSR1 ); + 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 ); + if ( log ) { + log_msg ( "GOT SIGNAL -> EXITING\n" ); + fclose ( log ); + } remove_pidfile ( ); exit ( 0 ); } +void sig_handler_parent ( int sig ) +{ + exit ( 0 ); +} + void usage ( void ) { fprintf ( stderr, "Usage: opiealarm -s [-f] | -r [-a]\n\n" ); fprintf ( stderr, "\t-s\tSuspend mode: set RTC alarm\n" ); fprintf ( stderr, "\t-f \tFix RTC, if RTC and system have more than 5sec difference (suspend mode)\n" ); fprintf ( stderr, "\t-r\tResume mode: kill running opiealarm\n" ); fprintf ( stderr, "\t-a <x>\tResuspend in <x> seconds (resume mode)\n\n" ); exit ( 1 ); } int fork_with_pidfile ( void ) { FILE *fp; pid_t pid; pid = fork ( ); - if ( pid > 0 ) + if ( pid > 0 ) { + signal ( SIGUSR1, sig_handler_parent ); + while ( 1 ) + sleep ( 1000 ); 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 ); @@ -171,69 +187,70 @@ int main ( int argc, char **argv ) } } if ( geteuid ( ) != 0 ) { fprintf ( stderr, "You need root priviledges to run opiealarm." ); return 2; } if ( !mode ) usage ( ); // kill running opiealarm + startpid = getpid ( ); opiealarm_was_running = kill_with_pidfile ( ); remove_pidfile ( ); switch ( mode ) { case 'r': return resume ( ac_resusp ); case 's': default : return suspend ( fix_rtc ); } return 0; } + int suspend ( int fix_rtc ) { FILE *fp; char buf [64]; time_t alrt, syst, rtct; struct tm alr, sys, rtc; int fd; int rtc_sys_diff; 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 )) error_msg_and_die ( 1, "/etc/resumeat" ); fclose ( fp ); alrt = atoi ( buf ); if ( alrt == 0 ) error_msg_and_die ( 0, "/etc/resumeat contains an invalid time description" ); - /* subtract 5 sec from event time... */ - alrt -= 5; + alrt -= 5; // wake up 5 sec before the specified time + if ( log ) fprintf ( log, "Setting RTC alarm to %d\n", alrt ); tzset ( ); alr = *gmtime ( &alrt ); // get system time time ( &syst ); sys = *localtime ( &syst ); @@ -265,25 +282,27 @@ int suspend ( int fix_rtc ) if ( ioctl ( fd, RTC_SET_TIME, &set ) < 0 ) error_msg_and_die ( 1, "ioctl RTC_SET_TIME" ); } // set alarm time if ( ioctl ( fd, RTC_ALM_SET, &alr ) < 0 ) error_msg_and_die ( 1, "ioctl RTC_ALM_SET" ); // enable alarm irq if ( ioctl ( fd, RTC_AIE_ON, 0 ) < 0 ) error_msg_and_die ( 1, "ioctl RTC_AIE_ON" ); - log_msg ( "SLEEPING\n" ); + if ( log ) + fprintf( log, "SIGUSR: pid %d - SLEEPING: pid %d\n", startpid, getpid ( )); + kill ( startpid, SIGUSR1 ); // wait for alarm irq if ( read ( fd, buf, sizeof( unsigned long )) < 0 ) error_msg_and_die ( 1, "read rtc alarm" ); log_msg ( "WAKEUP\n" ); // disable alarm irq if ( ioctl ( fd, RTC_AIE_OFF, 0 ) < 0 ) error_msg_and_die ( 1, "ioctl RTC_AIE_OFF" ); close ( fd ); @@ -317,24 +336,25 @@ int resume ( int resuspend ) { FILE *fp; // re-suspend when on AC (optional) when woken up via RTC if ( !opiealarm_was_running ) { // opiealarm -s got it's RTC signal -> wake up by RTC if ( resuspend && onac ( )) { time_t start, now; char *argv [4]; if ( !fork_with_pidfile ( )) return 4; + kill ( startpid, SIGUSR1 ); // sleep <resuspend> sec (not less!) time ( &start ); do { sleep ( 1 ); time ( &now ); } while (( now - start ) < resuspend ); if ( onac ( )) { // still on ac // system() without fork argv[0] = "qcop"; argv[1] = "QPE/Desktop"; |