1 /*
2  * This file is part of attach-p-cmd strace test.
3  *
4  * Copyright (c) 2016-2017 Dmitry V. Levin <ldv@altlinux.org>
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  * 3. The name of the author may not be used to endorse or promote products
16  *    derived from this software without specific prior written permission.
17  *
18  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
19  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
20  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
21  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
22  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
23  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
27  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28  */
29 
30 #include "tests.h"
31 #include <errno.h>
32 #include <signal.h>
33 #include <stdio.h>
34 #include <stdlib.h>
35 #include <sys/stat.h>
36 #include <time.h>
37 #include <unistd.h>
38 #include "attach-p-cmd.h"
39 
40 static void
41 wait_for_peer_invocation(void)
42 {
43 	/* create the lock directory */
44 	if (mkdir(lockdir, 0700))
45 		perror_msg_and_fail("mkdir: %s", lockdir);
46 
47 	/* wait for the lock directory to be removed by peer */
48 	while (mkdir(lockdir, 0700)) {
49 		if (EEXIST != errno)
50 			perror_msg_and_fail("mkdir: %s", lockdir);
51 	}
52 
53 	/* cleanup the lock directory */
54 	if (rmdir(lockdir))
55 		perror_msg_and_fail("rmdir: %s", lockdir);
56 }
57 
58 static void
59 wait_for_peer_termination(void)
60 {
61 	FILE *fp = fopen(pidfile, "r");
62 	if (!fp)
63 		perror_msg_and_fail("fopen: %s", pidfile);
64 
65 	pid_t pid;
66 	if (fscanf(fp, "%d", &pid) < 0)
67 		perror_msg_and_fail("fscanf: %s", pidfile);
68 	if (pid < 0)
69 		error_msg_and_fail("pid = %d", pid);
70 
71 	if (fclose(fp))
72 		perror_msg_and_fail("fclose: %s", pidfile);
73 
74 	if (unlink(pidfile))
75 		perror_msg_and_fail("unlink: %s", pidfile);
76 
77 	while (kill(pid, 0) == 0)
78 		;
79 }
80 
81 int
82 main(void)
83 {
84 	wait_for_peer_invocation();
85 	wait_for_peer_termination();
86 
87 	static const struct timespec ts = { .tv_nsec = 123456789 };
88 	if (nanosleep(&ts, NULL))
89 		perror_msg_and_fail("nanosleep");
90 
91 	static const char dir[] = "attach-p-cmd.test -p";
92 	pid_t pid = getpid();
93 	int rc = chdir(dir);
94 
95 	printf("%-5d chdir(\"%s\") = %s\n"
96 	       "%-5d +++ exited with 0 +++\n",
97 	       pid, dir, sprintrc(rc), pid);
98 
99 	return 0;
100 }
101