hfs-556.100.11.tar.gz

[apple/hfs.git] / tests / cases / test-set-protection-class.c

diff --git a/tests/cases/test-set-protection-class.c b/tests/cases/test-set-protection-class.c
index 164de50260370a6b4adbfce4be7881762a914508..55751f3e7eda89521361ff23877c90de0bcdf1c7 100644 (file)
--- a/tests/cases/test-set-protection-class.c
+++ b/tests/cases/test-set-protection-class.c
@@ -27,7 +27,7 @@ int run_set_protection_class(__unused test_ctx_t *ctx)
 {
        const char *tstdir;
        
-#if TARGET_OS_EMBEDDED
+#if (TARGET_OS_IPHONE && !TARGET_OS_SIMULATOR)
        // The root file system needs to be HFS
        struct statfs sfs;
        
@@ -38,7 +38,7 @@ int run_set_protection_class(__unused test_ctx_t *ctx)
        }
        
        tstdir = "/tmp";
-#else // !TARGET_OS_EMBEDDED
+#else // !(TARGET_OS_IPHONE && !TARGET_OS_SIMULATOR)
        disk_image_t *di = disk_image_get();
        tstdir = di->mount_point;
 #endif
@@ -139,7 +139,7 @@ int run_set_protection_class(__unused test_ctx_t *ctx)
 
                assert_with_errno((fd = open(path, O_RDWR)) >= 0);
 
-#if TARGET_OS_EMBEDDED
+#if (TARGET_OS_IPHONE && !TARGET_OS_SIMULATOR)
                assert_no_err(fcntl(fd, F_SETPROTECTIONCLASS, 2));
 #endif
 
@@ -158,9 +158,13 @@ int run_set_protection_class(__unused test_ctx_t *ctx)
                assert_no_err(close(fd));
                if (fd2 != -1)
                        assert_no_err(close(fd2));
+
+               assert_no_err(munmap(p, size));
        }
 
        unlink(path);
+       free(buf);
+       free(buf2);
 
        return 0;
 }