100cc81f701e9c068a80a798138d5d3b5916132c
[folly.git] / folly / test / SubprocessTestParentDeathHelper.cpp
1 /*
2  * Copyright 2017 Facebook, Inc.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *   http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 // This is a helper for the parentDeathSignal test in SubprocessTest.cpp.
18 //
19 // Basically, we create two processes, a parent and a child, and set the
20 // child to receive SIGUSR1 when the parent exits.  We set the child to
21 // create a file when that happens.  The child then kills the parent; the test
22 // will verify that the file actually gets created, which means that everything
23 // worked as intended.
24
25 #include <fcntl.h>
26 #include <signal.h>
27 #include <sys/types.h>
28
29 #include <glog/logging.h>
30
31 #include <folly/Conv.h>
32 #include <folly/Subprocess.h>
33 #include <folly/portability/GFlags.h>
34 #include <folly/portability/Unistd.h>
35
36 using folly::Subprocess;
37
38 DEFINE_bool(child, false, "");
39
40 namespace {
41 constexpr int kSignal = SIGUSR1;
42 }  // namespace
43
44 void runChild(const char* file) {
45   // Block SIGUSR1 so it's queued
46   sigset_t sigs;
47   CHECK_ERR(sigemptyset(&sigs));
48   CHECK_ERR(sigaddset(&sigs, kSignal));
49   CHECK_ERR(sigprocmask(SIG_BLOCK, &sigs, nullptr));
50
51   // Kill the parent, wait for our signal.
52   CHECK_ERR(kill(getppid(), SIGKILL));
53
54   int sig = 0;
55   CHECK_ERR(sigwait(&sigs, &sig));
56   CHECK_EQ(sig, kSignal);
57
58   // Signal completion by creating the file
59   CHECK_ERR(creat(file, 0600));
60 }
61
62 [[noreturn]] void runParent(const char* file) {
63   std::vector<std::string> args {"/proc/self/exe", "--child", file};
64   Subprocess proc(
65       args,
66       Subprocess::Options().parentDeathSignal(kSignal));
67   CHECK(proc.poll().running());
68
69   // The child will kill us.
70   for (;;) {
71     pause();
72   }
73 }
74
75 int main(int argc, char *argv[]) {
76   gflags::ParseCommandLineFlags(&argc, &argv, true);
77   CHECK_EQ(argc, 2);
78   if (FLAGS_child) {
79     runChild(argv[1]);
80   } else {
81     runParent(argv[1]);
82   }
83   return 0;
84 }