[apple/libpthread.git] / tests / pthread_cwd.c

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <fcntl.h>
#include <string.h>
#include <sys/syscall.h>
#include <sys/errno.h>
#include <sys/stat.h>
#include <sys/param.h>
#include <pthread/private.h>

#include "darwintest_defaults.h"

// /tmp is a symlink, so use full path for strict compare
#define WORKDIR		"/private/var/tmp/ptwork"
#define	WORKDIR1	WORKDIR "/one"
#define	WORKDIR2	WORKDIR "/two"

/*
 * This is a slow routine, just like getcwd(); people should remember that
 * they set something, instead of asking us what they told us.
 */
static char *
pthread_getcwd_np(char *buf, size_t size)
{
	int fd_cwd;

	if (buf == NULL)
		return (NULL);

	/*
	 * Open the "current working directory"; if we are running on a per
	 * thread working directory, that's the one we will get.
	 */
	if ((fd_cwd = open(".", O_RDONLY)) == -1)
		return (NULL);

	/*
	 * Switch off the per thread current working directory, in case we
	 * were on one; this fails if we aren't running with one.
	 */
	if (pthread_fchdir_np( -1) == -1) {
		/* We aren't runniing with one... alll done. */
		close (fd_cwd);
		return (NULL);
	}

	/*
	 * If we successfully switched off, then we switch back...
	 * this may fail catastrophically, if we no longer have rights;
	 * this should never happen, but threads may clobber our fd out
	 * from under us, etc..
	 */
	if (pthread_fchdir_np(fd_cwd) == -1) {
		close(fd_cwd);
		errno = EBADF;	/* sigil for catastrophic failure */
		return (NULL);
	}

	/* Close our directory handle */
	close(fd_cwd);

	/*
	 * And call the regular getcwd(), which will return the per thread
	 * current working directory instead of the process one.
	 */
	return getcwd(buf, size);
}

T_DECL(pthread_cwd, "per-thread working directory")
{
	char buf[MAXPATHLEN];

	T_SETUPBEGIN;

	T_ASSERT_EQ(pthread_fchdir_np(-1), -1, "test should not start with per-thread cwd");

	/* Blow the umask to avoid shooting our foot */
	umask(0);		/* "always successful" */

	/* Now set us up the test directories... */
	T_WITH_ERRNO; T_ASSERT_TRUE(mkdir(WORKDIR, 0777) != -1 || errno == EEXIST, NULL);
	T_WITH_ERRNO; T_ASSERT_TRUE(mkdir(WORKDIR1, 0777) != -1 || errno == EEXIST, NULL);
	T_WITH_ERRNO; T_ASSERT_TRUE(mkdir(WORKDIR2, 0777) != -1 || errno == EEXIST, NULL);

	T_SETUPEND;

	T_LOG("start in " WORKDIR1);
	T_ASSERT_POSIX_SUCCESS(chdir(WORKDIR1), NULL);
	T_ASSERT_EQ_STR(WORKDIR1, getcwd(buf, MAXPATHLEN), NULL);
	T_ASSERT_NULL(pthread_getcwd_np(buf, MAXPATHLEN), "pthread_getcwd_np should return NULL without per-thread cwd, got: %s", buf);

	T_LOG("move per-thread CWD to " WORKDIR2);
	T_ASSERT_POSIX_SUCCESS(pthread_chdir_np(WORKDIR2), NULL);
	T_ASSERT_EQ_STR(WORKDIR2, getcwd(buf, MAXPATHLEN), NULL);
	T_ASSERT_EQ_STR(WORKDIR2, pthread_getcwd_np(buf, MAXPATHLEN), NULL);

	T_LOG("unset per-thread CWD and confirm things go back");
	T_ASSERT_POSIX_SUCCESS(pthread_fchdir_np(-1), NULL);
	T_ASSERT_NULL(pthread_getcwd_np(buf, MAXPATHLEN), "pthread_getcwd_np should return NULL after reseting per-thread cwd, got: %s", buf);
	T_ASSERT_EQ_STR(WORKDIR1, getcwd(buf, MAXPATHLEN), NULL);
}
Commit	Line	Data
2546420a A	1	#include <stdio.h>
	2	#include <stdlib.h>
	3	#include <unistd.h>
	4	#include <fcntl.h>
	5	#include <string.h>
	6	#include <sys/syscall.h>
	7	#include <sys/errno.h>
	8	#include <sys/stat.h>
	9	#include <sys/param.h>
	10	#include <pthread/private.h>
	11
a0619f9c	12	#include "darwintest_defaults.h"
2546420a	13
2546420a A	14	// /tmp is a symlink, so use full path for strict compare
	15	#define WORKDIR "/private/var/tmp/ptwork"
	16	#define WORKDIR1 WORKDIR "/one"
	17	#define WORKDIR2 WORKDIR "/two"
	18
	19	/*
	20	* This is a slow routine, just like getcwd(); people should remember that
	21	* they set something, instead of asking us what they told us.
	22	*/
	23	static char *
	24	pthread_getcwd_np(char *buf, size_t size)
	25	{
	26	int fd_cwd;
	27
	28	if (buf == NULL)
	29	return (NULL);
	30
	31	/*
	32	* Open the "current working directory"; if we are running on a per
	33	* thread working directory, that's the one we will get.
	34	*/
	35	if ((fd_cwd = open(".", O_RDONLY)) == -1)
	36	return (NULL);
	37
	38	/*
	39	* Switch off the per thread current working directory, in case we
	40	* were on one; this fails if we aren't running with one.
	41	*/
	42	if (pthread_fchdir_np( -1) == -1) {
	43	/* We aren't runniing with one... alll done. */
	44	close (fd_cwd);
	45	return (NULL);
	46	}
	47
	48	/*
	49	* If we successfully switched off, then we switch back...
	50	* this may fail catastrophically, if we no longer have rights;
	51	* this should never happen, but threads may clobber our fd out
	52	* from under us, etc..
	53	*/
	54	if (pthread_fchdir_np(fd_cwd) == -1) {
	55	close(fd_cwd);
	56	errno = EBADF; /* sigil for catastrophic failure */
	57	return (NULL);
	58	}
	59
	60	/* Close our directory handle */
	61	close(fd_cwd);
	62
	63	/*
	64	* And call the regular getcwd(), which will return the per thread
	65	* current working directory instead of the process one.
	66	*/
	67	return getcwd(buf, size);
	68	}
	69
	70	T_DECL(pthread_cwd, "per-thread working directory")
	71	{
	72	char buf[MAXPATHLEN];
	73
	74	T_SETUPBEGIN;
	75
	76	T_ASSERT_EQ(pthread_fchdir_np(-1), -1, "test should not start with per-thread cwd");
	77
78	/* Blow the umask to avoid shooting our foot */
79	umask(0); /* "always successful" */
80
81	/* Now set us up the test directories... */
a0619f9c A	82	T_WITH_ERRNO; T_ASSERT_TRUE(mkdir(WORKDIR, 0777) != -1 \|\| errno == EEXIST, NULL);
	83	T_WITH_ERRNO; T_ASSERT_TRUE(mkdir(WORKDIR1, 0777) != -1 \|\| errno == EEXIST, NULL);
	84	T_WITH_ERRNO; T_ASSERT_TRUE(mkdir(WORKDIR2, 0777) != -1 \|\| errno == EEXIST, NULL);
2546420a A	85
	86	T_SETUPEND;
	87
	88	T_LOG("start in " WORKDIR1);
	89	T_ASSERT_POSIX_SUCCESS(chdir(WORKDIR1), NULL);
	90	T_ASSERT_EQ_STR(WORKDIR1, getcwd(buf, MAXPATHLEN), NULL);
	91	T_ASSERT_NULL(pthread_getcwd_np(buf, MAXPATHLEN), "pthread_getcwd_np should return NULL without per-thread cwd, got: %s", buf);
	92
	93	T_LOG("move per-thread CWD to " WORKDIR2);
	94	T_ASSERT_POSIX_SUCCESS(pthread_chdir_np(WORKDIR2), NULL);
	95	T_ASSERT_EQ_STR(WORKDIR2, getcwd(buf, MAXPATHLEN), NULL);
	96	T_ASSERT_EQ_STR(WORKDIR2, pthread_getcwd_np(buf, MAXPATHLEN), NULL);
	97
	98	T_LOG("unset per-thread CWD and confirm things go back");
	99	T_ASSERT_POSIX_SUCCESS(pthread_fchdir_np(-1), NULL);
	100	T_ASSERT_NULL(pthread_getcwd_np(buf, MAXPATHLEN), "pthread_getcwd_np should return NULL after reseting per-thread cwd, got: %s", buf);
	101	T_ASSERT_EQ_STR(WORKDIR1, getcwd(buf, MAXPATHLEN), NULL);
	102	}