Home | History | Annotate | Download | only in tests 
      1 /*
      2  * This file is part of attach-p-cmd strace test.
      3  *
      4  * Copyright (c) 2016-2017 Dmitry V. Levin <ldv (at) 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