dyld-353.2.1.tar.gz

[apple/dyld.git] / unit-tests / test-cases / restrict-environ / main.c

diff --git a/unit-tests/test-cases/restrict-environ/main.c b/unit-tests/test-cases/restrict-environ/main.c
index c4c0da8ffd629b120ae7a4a32a60d19a8548b2bb..e68d0386d7dff185937255115a15c9c4f4e66668 100644 (file)
--- a/unit-tests/test-cases/restrict-environ/main.c
+++ b/unit-tests/test-cases/restrict-environ/main.c
@@ -23,6 +23,7 @@
 #include <stdio.h>  // fprintf(), NULL
 #include <stdlib.h> // exit(), EXIT_SUCCESS
 #include <string.h> // strcmp(), strncmp()
+#include <mach-o/dyld_priv.h> // dyld_process_is_restricted()
 
 #include "test.h" // PASS(), FAIL(), XPASS(), XFAIL()
 
@@ -58,6 +59,13 @@ int main(int argc, const char* argv[], const char* envp[], const char* apple[])
                return EXIT_SUCCESS;
        }
        
+       // verify SPI says process is restricted
+       if ( !dyld_process_is_restricted() ) {
+               FAIL("restrict-environ: dyld_process_is_restrictet() returns false");
+               return EXIT_SUCCESS;
+       }
+       
+       
        PASS("restrict-environ");
        return EXIT_SUCCESS;
 }