hb_system_sleep: refactoring.

Also:

- release power	assertions before freeing them

- actually free	the power assertions in	  hb_close() (previously unused)



git-svn-id: svn://svn.handbrake.fr/HandBrake/trunk@5340 b64f7644-9d1e-0410-96f1-a4d463321fa5

4 files changed, 78 insertions, 39 deletions

diff --git a/libhb/hb.c b/libhb/hb.c
index 56df9a9be..a7df7a220 100644
--- a/libhb/hb.c
+++ b/libhb/hb.c
@@ -64,9 +64,8 @@ struct hb_handle_s
        on multi-pass encodes where frames get dropped.     */
     hb_interjob_t * interjob;
 
-    // Power Management opaque pointer
-    // For OSX, it's an IOPMAssertionID*
-    void          * hb_system_sleep_opaque;
+    // power management opaque pointer
+    void *system_sleep_opaque;
 } ;
 
 hb_work_object_t * hb_objects = NULL;
@@ -411,7 +410,7 @@ hb_handle_t * hb_init( int verbose, int update_check )
     h->build = -1;
 
     /* Initialize opaque for PowerManagement purposes */
-    h->hb_system_sleep_opaque = hb_system_sleep_opaque_init();
+    h->system_sleep_opaque = hb_system_sleep_opaque_init();
 
     if( update_check )
     {
@@ -518,7 +517,7 @@ hb_handle_t * hb_init_dl( int verbose, int update_check )
     h->build = -1;
 
     /* Initialize opaque for PowerManagement purposes */
-    h->hb_system_sleep_opaque = hb_system_sleep_opaque_init();
+    h->system_sleep_opaque = hb_system_sleep_opaque_init();
 
     if( update_check )
     {
@@ -1679,6 +1678,8 @@ void hb_close( hb_handle_t ** _h )
     hb_lock_close( &h->state_lock );
     hb_lock_close( &h->pause_lock );
 
+    hb_system_sleep_opaque_close(&h->system_sleep_opaque);
+
     free( h->interjob );
 
     free( h );
@@ -1886,14 +1887,14 @@ void hb_set_state( hb_handle_t * h, hb_state_t * s )
     hb_unlock( h->pause_lock );
 }
 
-void hb_prevent_sleep( hb_handle_t * h )
+void hb_system_sleep_allow(hb_handle_t *h)
 {
-    hb_system_sleep_prevent( h->hb_system_sleep_opaque );
+    hb_system_sleep_private_enable(h->system_sleep_opaque);
 }
 
-void hb_allow_sleep( hb_handle_t * h )
+void hb_system_sleep_prevent(hb_handle_t *h)
 {
-    hb_system_sleep_allow( h->hb_system_sleep_opaque );
+    hb_system_sleep_private_disable(h->system_sleep_opaque);
 }
 
 /* Passes a pointer to persistent data */
diff --git a/libhb/hb.h b/libhb/hb.h
index 07100c946..e4c81a4e4 100644
--- a/libhb/hb.h
+++ b/libhb/hb.h
@@ -95,8 +95,8 @@ void          hb_pause( hb_handle_t * );
 void          hb_resume( hb_handle_t * );
 void          hb_stop( hb_handle_t * );
 
-void          hb_prevent_sleep( hb_handle_t * );
-void          hb_allow_sleep( hb_handle_t * );
+void          hb_system_sleep_allow(hb_handle_t*);
+void          hb_system_sleep_prevent(hb_handle_t*);
 
 /* Persistent data between jobs. */
 typedef struct hb_interjob_s
diff --git a/libhb/ports.c b/libhb/ports.c
index 7097f2ee9..98945fec0 100644
--- a/libhb/ports.c
+++ b/libhb/ports.c
@@ -792,63 +792,101 @@ char *strtok_r(char *s, const char *delim, char **save_ptr)
 
 #ifdef __APPLE__
 // 128 chars limit for IOPMAssertionCreateWithName
-static CFStringRef reasonForActivity= CFSTR("HandBrake is currently scanning and/or encoding");
+static CFStringRef reasonForActivity =
+    CFSTR("HandBrake is currently scanning and/or encoding");
 #endif
 
 void* hb_system_sleep_opaque_init()
 {
-    void* opaque;
+    void *opaque = NULL;
 #ifdef __APPLE__
-    opaque = calloc( sizeof( IOPMAssertionID ), 1);
-    IOPMAssertionID * assertionID = (IOPMAssertionID *)opaque;
+    opaque = calloc(sizeof(IOPMAssertionID), 1);
+    if (opaque == NULL)
+    {
+        hb_error("hb_system_sleep: failed to allocate opaque");
+        return NULL;
+    }
+
+    IOPMAssertionID *assertionID = (IOPMAssertionID*)opaque;
     *assertionID = -1;
 #endif
-
     return opaque;
 }
 
-void hb_system_sleep_opaque_close(void **_opaque)
+void hb_system_sleep_opaque_close(void **opaque)
 {
+    if (*opaque != NULL)
+    {
+        hb_system_sleep_private_enable(*opaque);
+    }
 #ifdef __APPLE__
-    IOPMAssertionID * assertionID = (IOPMAssertionID *) *_opaque;
-    free( assertionID );
+    if (*opaque != NULL)
+    {
+        IOPMAssertionID *assertionID = (IOPMAssertionID*)*opaque;
+        free(assertionID);
+    }
 #endif
-    *_opaque = NULL;
+    *opaque = NULL;
 }
 
-void hb_system_sleep_allow(void *opaque)
+void hb_system_sleep_private_enable(void *opaque)
 {
 #ifdef __APPLE__
-    IOPMAssertionID * assertionID = (IOPMAssertionID *)opaque;
+    if (opaque == NULL)
+    {
+        hb_error("hb_system_sleep: opaque is NULL");
+    }
 
+    IOPMAssertionID *assertionID = (IOPMAssertionID*)opaque;
     if (*assertionID == -1)
+    {
+        // nothing to do
         return;
+    }
 
     IOReturn success = IOPMAssertionRelease(*assertionID);
-
-    if (success == kIOReturnSuccess) {
-        hb_deep_log( 3, "osxsleep: IOPM assertion %d successfully released, sleep allowed", *assertionID );
+    if (success == kIOReturnSuccess)
+    {
+        hb_deep_log(3,
+                    "hb_system_sleep: assertion %d released, sleep allowed",
+                    *assertionID);
         *assertionID = -1;
-    } else {
-        hb_log( "osxsleep: error while trying to unset power management assertion" );
+    }
+    else
+    {
+        hb_log("hb_system_sleep: failed to allow system sleep");
     }
 #endif
 }
 
-void hb_system_sleep_prevent(void *opaque)
+void hb_system_sleep_private_disable(void *opaque)
 {
 #ifdef __APPLE__
-    IOPMAssertionID * assertionID = (IOPMAssertionID *)opaque;
-
+    if (opaque == NULL)
+    {
+        hb_error("hb_system_sleep: opaque is NULL");
+    }
+    
+    IOPMAssertionID *assertionID = (IOPMAssertionID*)opaque;
     if (*assertionID != -1)
+    {
+        // nothing to do
         return;
+    }
 
     IOReturn success = IOPMAssertionCreateWithName(kIOPMAssertionTypeNoIdleSleep,
-                                                   kIOPMAssertionLevelOn, reasonForActivity, assertionID);
-    if (success == kIOReturnSuccess) {
-        hb_deep_log( 3, "IOPM assertion %d successfully created, prevent sleep", *assertionID);
-    } else {
-        hb_log( "osxsleep: error while trying to set power management assertion" );
+                                                   kIOPMAssertionLevelOn,
+                                                   reasonForActivity,
+                                                   assertionID);
+    if (success == kIOReturnSuccess)
+    {
+        hb_deep_log(3,
+                    "hb_system_sleep: assertion %d created, sleep prevented",
+                    *assertionID);
+    }
+    else
+    {
+        hb_log("hb_system_sleep: failed to prevent system sleep");
     }
 #endif
 }
diff --git a/libhb/ports.h b/libhb/ports.h
index 162781347..2c0aeb1ee 100644
--- a/libhb/ports.h
+++ b/libhb/ports.h
@@ -105,10 +105,10 @@ void       hb_net_close( hb_net_t ** );
 /************************************************************************
 * OS Sleep Allow / Prevent
 ***********************************************************************/
-void     * hb_system_sleep_opaque_init();
-void       hb_system_sleep_opaque_close( void ** opaque );
-void       hb_system_sleep_allow( void * opaque );
-void       hb_system_sleep_prevent( void * opaque );
+void* hb_system_sleep_opaque_init();
+void  hb_system_sleep_opaque_close(void **opaque);
+void  hb_system_sleep_private_enable(void *opaque);
+void  hb_system_sleep_private_disable(void *opaque);
 
 #endif /* __LIBHB__ */