3 files changed, 217 insertions, 60 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
@@ -16,52 +16,187 @@
 #include <sys/types.h>
 #include <fcntl.h>
 #include <unistd.h>
 #include <errno.h>
 #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 )
 {
         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 );
 
-        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 ))
                 error_msg_and_die ( 1, "/etc/resumeat" );
 
         fclose ( fp );
         
@@ -73,18 +208,18 @@ void suspend_on_rtc ( )
         /* subtract 5 sec from event time... */
         t -= 5;
 
                 if ( log )
                 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" );   
                 // enable alarm irq
         if ( ioctl ( fd, RTC_AIE_ON, 0 ) < 0 )
                 error_msg_and_die ( 1, "ioctl RTC_AIE_ON" );
         
@@ -96,72 +231,79 @@ void suspend_on_rtc ( )
         
         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 );
-}
-
-
-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;
 }
diff --git a/core/opiealarm/opieatd b/core/opiealarm/opieatd
index 3b9dc0e..fc2d421 100755
--- a/core/opiealarm/opieatd
+++ b/core/opiealarm/opieatd
@@ -1,22 +1,37 @@
 #!/bin/sh
 
 timefile=/etc/resumeat
 
+trap exit 0 SIGTERM SIGINT SIGQUIT
+
+echo $$ >/var/run/opieatd.pid
+
 mkdir -p /var/spool/at
 [ -p /var/spool/at/trigger ] || mkfifo /var/spool/at/trigger
 
 while true; do
-        cat /var/spool/at/trigger | while read line; do
-                FILE=`ls -1 /var/spool/at/[0-9]* | head -n1`
-                echo "File = $FILE"
-                if [ -z "$FILE" ]; then
+        while read </var/spool/at/trigger dummy; do
+                FILES=`ls /var/spool/at/[0-9]* 2>/dev/null`
+        
+                if [ -z "$FILES" ]; then
                         echo "clear resume at"
                         echo "" >$timefile
                 else
-                        unixtime=`basename $FILE | cut -c1-10`
-                        echo "Datestring = $unixtime"
-                        echo "$unixtime" >$timefile
+                        for i in "$FILES"; do
+                                echo "File = $i"
+
+                                unixtime=`basename $i | cut -c1-10`
+                                pid=`basename $i | cut -c12-` 
+                                
+                                        if [ -d /proc/$pid ]; then                      
+                                        echo "Datestring = $unixtime"
+                                        echo "$unixtime" >$timefile
+                                else
+                                        rm -f $i
+                                fi
+                        done
                 fi
         done
 done
 
+exit 0
diff --git a/root/etc/init.d/opieatd b/root/etc/init.d/opieatd
index 83f6035..765426c 100755
--- a/root/etc/init.d/opieatd
+++ b/root/etc/init.d/opieatd
@@ -3,13 +3,13 @@
 PATH=/usr/bin:/bin
 HOME=/root
 
 case "$1" in
   start)
     /opt/QtPalmtop/bin/opieatd &
     ;;
   stop)
-    killall opieatd
+    kill `cat /var/run/opieatd.pid`
     ;;
 esac
 
 exit 0