From 8ec42ced8063d739ec6c3d656b0556c3aafbc395 Mon Sep 17 00:00:00 2001
From: Raymond Lu <songyulu@hdfgroup.org>
Date: Mon, 6 May 2013 15:51:26 -0500
Subject: [svn-r23675] Issue 8380 - H5Zunregister caused seg fault.  This is
 the followup commit.  Quincey reviewed my previous commit and suggested me to
 use H5I_iterate to iterate through all opened objects in H5Z_unregister.  I
 changed it in this commit.

Tested with h5committest.
---
 src/H5Z.c           | 200 ++++++++++++++++++++++++++++++----------------------
 test/CMakeLists.txt |   4 ++
 test/unregister.c   |   2 -
 3 files changed, 120 insertions(+), 86 deletions(-)

diff --git a/src/H5Z.c b/src/H5Z.c
index 5d68c25..1652a7d 100644
--- a/src/H5Z.c
+++ b/src/H5Z.c
@@ -46,6 +46,11 @@ typedef struct H5Z_stats_t {
 } H5Z_stats_t;
 #endif /* H5Z_DEBUG */
 
+typedef struct H5Z_object_t {
+    H5Z_filter_t filter_id;     /* ID of the filter we're looking for         */
+    htri_t       found;         /* Whether we find an object using the filter */
+} H5Z_object_t;
+
 /* Enumerated type for dataset creation prelude callbacks */
 typedef enum {
     H5Z_PRELUDE_CAN_APPLY,      /* Call "can apply" callback */
@@ -65,6 +70,7 @@ static H5Z_stats_t	*H5Z_stat_table_g = NULL;
 
 /* Local functions */
 static int H5Z_find_idx(H5Z_filter_t id);
+static int H5Z_get_object_cb(void *obj_ptr, hid_t obj_id, void *key);
 
 
 /*-------------------------------------------------------------------------
@@ -409,18 +415,11 @@ done:
 herr_t
 H5Z_unregister (H5Z_filter_t filter_id)
 {
-    hid_t  *file_list = NULL;
-    hid_t  *obj_id_list = NULL;
-    size_t num_obj_id = 0;
-    size_t num_file_id = 0;
-    H5F_t    *f = NULL;         /* File to query */
-    H5I_type_t id_type;
-    hid_t ocpl_id;
-    H5P_genplist_t  *plist;                 /* Property list */
-    size_t filter_index;                   /* Local index variable for filter */
-    int    i;
-    htri_t filter_in_pline = FALSE;
-    herr_t ret_value=SUCCEED;   /* Return value */
+    size_t       num_obj_id = 0;
+    size_t       num_file_id = 0;
+    size_t       filter_index;        /* Local index variable for filter */
+    H5Z_object_t object;
+    herr_t       ret_value=SUCCEED;   /* Return value */
 
     FUNC_ENTER_NOAPI(FAIL)
 
@@ -440,86 +439,33 @@ H5Z_unregister (H5Z_filter_t filter_id)
         HGOTO_ERROR(H5E_FILE, H5E_CANTGET, FAIL, "can't get object number")
 
     if(num_obj_id) {
-        if(NULL == (obj_id_list = (hid_t *)H5MM_malloc(num_obj_id*sizeof(hid_t))))
-            HGOTO_ERROR(H5E_RESOURCE, H5E_CANTALLOC, FAIL, "failed to allocate space")
-
-        /* Find all opened objects that may use the filter (datasets and groups).  Passing NULL as a pointer to 
-         * file structure indicates searching among all opened files */
-        if(H5F_get_obj_ids(NULL, H5F_OBJ_DATASET | H5F_OBJ_GROUP, num_obj_id, obj_id_list, FALSE, &num_obj_id) < 0)
-            HGOTO_ERROR(H5E_FILE, H5E_CANTGET, FAIL, "can't get object IDs")
-    }
-
-    /* Check if any opened object (dataset or group) uses the filter.  If so, fail with a message */
-    for(i=0; i<num_obj_id; i++) {
-        if((id_type = H5I_get_type(obj_id_list[i])) < 0)
-            HGOTO_ERROR(H5E_ARGS, H5E_BADTYPE, FAIL, "bad object id");
-
-        switch(id_type) {
-            case H5I_GROUP:
-            {
-                H5G_t *group = NULL;
-
-                if(NULL == (group = (H5G_t *)H5I_object_verify(obj_id_list[i], H5I_GROUP)))
-	            HGOTO_ERROR(H5E_ARGS, H5E_BADTYPE, FAIL, "not a group")
+        /* Initialize it here because it isn't needed when the object is file */
+        object.filter_id = filter_id;
+        object.found = FALSE;
 
-                if((ocpl_id = H5G_get_create_plist(group)) < 0) 
-                    HGOTO_ERROR(H5E_ARGS, H5E_CANTGET, FAIL, "can't get group creation property list")
-
-            }
-                break;
-
-            case H5I_DATASET:
-            {
-                H5D_t *dataset = NULL;
-
-                if(NULL == (dataset = (H5D_t *)H5I_object_verify(obj_id_list[i], H5I_DATASET)))
-	            HGOTO_ERROR(H5E_ARGS, H5E_BADTYPE, FAIL, "not a dataset")
-
-                if((ocpl_id = H5D_get_create_plist(dataset)) < 0) 
-                    HGOTO_ERROR(H5E_ARGS, H5E_CANTGET, FAIL, "can't get dataset creation property list")
-            }    
-                break;
+        /* Iterate through all opened datasets, returns a failure if any of them uses the filter */
+        if(H5I_iterate(H5I_DATASET, H5Z_get_object_cb, &object, FALSE) < 0)
+            HGOTO_ERROR(H5E_FILE, H5E_BADITER, FAIL, "iteration failed")
 
-            default:
-                HGOTO_ERROR(H5E_ARGS, H5E_BADVALUE, FAIL, "not a valid object")
-        } /* end switch */
+        if(object.found)
+            HGOTO_ERROR(H5E_PLINE, H5E_CANTRELEASE, FAIL, "can't unregister filter because a dataset is still using it")
 
-	/* Get the plist structure of object creation */
-	if(NULL == (plist = H5P_object_verify(ocpl_id, H5P_OBJECT_CREATE)))
-	    HGOTO_ERROR(H5E_ATOM, H5E_BADATOM, FAIL, "can't find object for ID")
+        /* Iterate through all opened groups, returns a failure if any of them uses the filter */
+        if(H5I_iterate(H5I_GROUP, H5Z_get_object_cb, &object, FALSE) < 0)
+            HGOTO_ERROR(H5E_FILE, H5E_BADITER, FAIL, "iteration failed")
 
-	/* Check if the object creation property list uses the filter */
-	if((filter_in_pline = H5P_filter_in_pline(plist, filter_id)) < 0)
-	    HGOTO_ERROR(H5E_PLIST, H5E_CANTGET, FAIL, "can't check filter in pipeline")
-	
-        if(filter_in_pline)
-	    HGOTO_ERROR(H5E_PLIST, H5E_CANTGET, FAIL, "can't unregister filter because some object is still using it")
+        if(object.found)
+            HGOTO_ERROR(H5E_PLINE, H5E_CANTRELEASE, FAIL, "can't unregister filter because a group is still using it")
     }
 
     /* Count the number of opened files */
     if(H5F_get_obj_count(NULL, H5F_OBJ_FILE, FALSE, &num_file_id) < 0)
         HGOTO_ERROR(H5E_FILE, H5E_CANTGET, FAIL, "can't get file number")
    
-    /* Get the list of IDs for all opened files */ 
+    /* Iterate through all opened files and flush them */
     if(num_file_id) {
-        if(NULL == (file_list = (hid_t *)H5MM_malloc(num_file_id*sizeof(hid_t))))
-            HGOTO_ERROR(H5E_RESOURCE, H5E_CANTALLOC, FAIL, "failed to allocate space")
-
-        if(H5F_get_obj_ids(NULL, H5F_OBJ_FILE, num_file_id, file_list, FALSE, &num_file_id) < 0)
-            HGOTO_ERROR(H5E_FILE, H5E_CANTGET, FAIL, "can't get file IDs")
-    }
-
-    /* Flush all opened files in case any file uses the filter */
-    for(i=0; i<num_file_id; i++) {
-        if(NULL == (f = (H5F_t *)H5I_object_verify(file_list[i], H5I_FILE)))
-            HGOTO_ERROR(H5E_ARGS, H5E_BADTYPE, FAIL, "not a file")
-
-        /* Call the flush routine for mounted file hierarchies. Do a global flush 
-         * if the file is opened for write */
-        if(H5F_ACC_RDWR & H5F_INTENT(f)) {
-            if(H5F_flush_mounts(f, H5AC_dxpl_id) < 0)
-                HGOTO_ERROR(H5E_FILE, H5E_CANTFLUSH, FAIL, "unable to flush file hierarchy")
-        } /* end if */
+        if(H5I_iterate(H5I_FILE, H5Z_get_object_cb, &object, FALSE) < 0)
+            HGOTO_ERROR(H5E_FILE, H5E_BADITER, FAIL, "iteration failed")
     }
 
     /* Remove filter from table */
@@ -531,14 +477,100 @@ H5Z_unregister (H5Z_filter_t filter_id)
     H5Z_table_used_g--;
 
 done:
-    if(file_list) H5MM_free(file_list);
-    if(obj_id_list) H5MM_free(obj_id_list);
-
     FUNC_LEAVE_NOAPI(ret_value)
 } /* end H5Z_unregister() */
 
 
 /*-------------------------------------------------------------------------
+ * Function:	H5Z_get_object_cb
+ *
+ * Purpose:	The callback function for H5Z_unregister. It iterates 
+ *              through all opened objects.  If the object is a dataset
+ *              or a group and it uses the filter to be unregistered, the 
+ *              function returns TRUE. If the object is a file, the function
+ *              flushes it to the disk. 
+ *
+ * Return:      TRUE if the object uses the filter.
+ *              FALSE otherwise.
+ *
+ * Programmer:  Raymond Lu
+ *              6 May 2013
+ *
+ *-------------------------------------------------------------------------
+ */
+static int
+H5Z_get_object_cb(void *obj_ptr, hid_t obj_id, void *key)
+{
+    H5I_type_t      id_type;
+    hid_t           ocpl_id;
+    H5P_genplist_t  *plist;                 /* Property list */
+    H5Z_object_t    *object = (H5Z_object_t *)key;
+    htri_t          filter_in_pline = FALSE;
+    int             ret_value = FALSE;    /* Return value */
+
+    FUNC_ENTER_NOAPI(FAIL)
+
+    HDassert(obj_ptr);
+
+    if((id_type = H5I_get_type(obj_id)) < 0)
+        HGOTO_ERROR(H5E_ARGS, H5E_BADTYPE, FAIL, "bad object id");
+
+    switch(id_type) {
+        case H5I_FILE:
+        {
+            H5F_t    *f = NULL;         /* File to query */
+
+            if(NULL == (f = (H5F_t *)H5I_object_verify(obj_id, H5I_FILE)))
+                HGOTO_ERROR(H5E_ARGS, H5E_BADTYPE, FAIL, "not a file")
+
+            /* Call the flush routine for mounted file hierarchies. Do a global flush 
+             * if the file is opened for write */
+            if(H5F_ACC_RDWR & H5F_INTENT(f)) {
+                if(H5F_flush_mounts(f, H5AC_dxpl_id) < 0)
+                    HGOTO_ERROR(H5E_FILE, H5E_CANTFLUSH, FAIL, "unable to flush file hierarchy")
+            } /* end if */
+
+            /* H5I_iterate expects FALSE to continue to other files */
+            HGOTO_DONE(FALSE) 
+        } 
+        case H5I_GROUP:
+            if((ocpl_id = H5G_get_create_plist(obj_ptr)) < 0) 
+                HGOTO_ERROR(H5E_ARGS, H5E_CANTGET, FAIL, "can't get group creation property list")
+
+            break;
+
+        case H5I_DATASET:
+            if((ocpl_id = H5D_get_create_plist(obj_ptr)) < 0) 
+                HGOTO_ERROR(H5E_ARGS, H5E_CANTGET, FAIL, "can't get dataset creation property list")
+                
+            break;
+
+        default:
+            HGOTO_ERROR(H5E_ARGS, H5E_BADVALUE, FAIL, "not a valid object")
+    } /* end switch */
+
+    /* Get the plist structure of object creation */
+    if(NULL == (plist = H5P_object_verify(ocpl_id, H5P_OBJECT_CREATE)))
+        HGOTO_ERROR(H5E_ATOM, H5E_BADATOM, FAIL, "can't find object for ID")
+
+    /* Check if the object creation property list uses the filter */
+    if((filter_in_pline = H5P_filter_in_pline(plist, object->filter_id)) < 0)
+        HGOTO_ERROR(H5E_PLIST, H5E_CANTGET, FAIL, "can't check filter in pipeline")
+
+    /* H5I_iterate expects TRUE to stop the loop over objects. Stop the loop and let 
+     * H5Z_unregister return failure */ 
+    if(filter_in_pline) {
+        object->found = TRUE;
+        ret_value = TRUE;
+    }
+
+done:
+    FUNC_LEAVE_NOAPI(ret_value)
+} /* end H5F_get_objects_cb() */
+
+
+
+/*-------------------------------------------------------------------------
  * Function:	H5Zfilter_avail
  *
  * Purpose:	Check if a filter is available
diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt
index 8904fee..766ed7b 100644
--- a/test/CMakeLists.txt
+++ b/test/CMakeLists.txt
@@ -397,6 +397,8 @@ ADD_TEST (
         testmeta.h5
         tstint1.h5
         tstint2.h5
+        unregister_filter_1.h5
+        unregister_filter_2.h5
 )
 
 SET (H5_TESTS
@@ -453,6 +455,7 @@ SET (H5_TESTS
     file_image
     enc_dec_plist
     enc_dec_plist_with_endianess
+    unregister
 )
 
 FOREACH (test ${H5_TESTS})
@@ -728,6 +731,7 @@ IF (HDF5_TEST_VFD)
       tcheck_version
       testmeta
       links_env
+      unregister
 )
   
   IF (DIRECT_VFD)
diff --git a/test/unregister.c b/test/unregister.c
index 88f4d1b..297035d 100644
--- a/test/unregister.c
+++ b/test/unregister.c
@@ -145,7 +145,6 @@ test_unregister_filters(hid_t my_fapl)
         goto error;
     } /* end if */
 
-
     /* Close the group */
     if(H5Gclose(gid) < 0) goto error;
 
@@ -176,7 +175,6 @@ test_unregister_filters(hid_t my_fapl)
         goto error;
 
     /* Unregister the filter before closing the dataset.  It should fail */
-    /*if(H5Zunregister(H5Z_FILTER_DUMMY) < 0) goto error;*/
     H5E_BEGIN_TRY {
         ret = H5Zunregister(H5Z_FILTER_DUMMY);
     } H5E_END_TRY;
-- 
cgit v0.12