[libc] Add the POSIX waitpid function and the BSD wait4 function.

Reviewed By: lntue, michaelrj

Differential Revision: https://reviews.llvm.org/D135225

Author: Siva Chandra Reddy

libc/config/linux/api.td

@@ -272,7 +272,7 @@ def SysStatAPI : PublicAPI<"sys/stat.h"> {
 }
 
 def SysWaitAPI : PublicAPI<"sys/wait.h"> {
-  let Types = ["pid_t"];
+  let Types = ["pid_t", "struct rusage"];
 }
 
 def SysSendfileAPI : PublicAPI<"sys/sendfile.h"> {

libc/config/linux/x86_64/entrypoints.txt

@@ -133,6 +133,8 @@ set(TARGET_LIBC_ENTRYPOINTS
 
     # sys/wait.h entrypoints
     libc.src.sys.wait.wait
+    libc.src.sys.wait.wait4
+    libc.src.sys.wait.waitpid
 
     # unistd.h entrypoints
     libc.src.unistd.access

libc/include/CMakeLists.txt

@@ -284,6 +284,7 @@ add_gen_header(
     .llvm_libc_common_h
     .llvm-libc-macros.sys_wait_macros
     .llvm-libc-types.pid_t
+    .llvm-libc-types.struct_rusage
 )
 
 if(NOT LLVM_LIBC_FULL_BUILD)

libc/include/llvm-libc-macros/linux/sys-wait-macros.h

@@ -9,6 +9,11 @@
 #ifndef __LLVM_LIBC_MACROS_LINUX_SYS_WAIT_MACROS_H
 #define __LLVM_LIBC_MACROS_LINUX_SYS_WAIT_MACROS_H
 
+// Wait flags
+#define WNOHANG 1    // Do not block
+#define WUNTRACED 2  // Report is a child has stopped even if untraced
+#define WCONTINUED 8 // Report if a stopped child has been resumed by SIGCONT
+
 // Wait status info macros
 #define WTERMSIG(status) (((status)&0x7F))
 #define WIFEXITED(status) (WTERMSIG(status) == 0)

libc/include/llvm-libc-types/CMakeLists.txt

@@ -43,14 +43,17 @@ add_header(pthread_t HDR pthread_t.h DEPENDS .__thread_type)
 add_header(pthread_mutexattr_t HDR pthread_mutexattr_t.h)
 add_header(pthread_once_t HDR pthread_once_t.h DEPENDS .__futex_word)
 add_header(rlim_t HDR rlim_t.h)
+add_header(time_t HDR time_t.h)
+add_header(suseconds_t HDR suseconds_t.h)
+add_header(struct_timeval HDR struct_timeval.h DEPENDS .suseconds_t .time_t)
 add_header(struct_rlimit HDR struct_rlimit.h DEPENDS .rlim_t)
+add_header(struct_rusage HDR struct_rusage.h DEPENDS .struct_timeval)
 add_header(ssize_t HDR ssize_t.h)
 add_header(struct_dirent HDR struct_dirent.h DEPENDS .ino_t .off_t)
 add_header(union_sigval HDR union_sigval.h)
 add_header(siginfo_t HDR siginfo_t.h DEPENDS .union_sigval .pid_t .uid_t)
 add_header(sigset_t HDR sigset_t.h DEPENDS libc.include.llvm-libc-macros.signal_macros)
 add_header(struct_sigaction HDR struct_sigaction.h DEPENDS .sigset_t .siginfo_t)
-add_header(time_t HDR time_t.h)
 add_header(struct_timespec HDR struct_timespec.h DEPENDS .time_t)
 add_header(
   struct_stat

libc/include/llvm-libc-types/struct_rusage.h

@@ -0,0 +1,37 @@
+//===-- Definition of type struct rusage ----------------------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef __LLVM_LIBC_TYPES_STRUCT_RUSAGE_H__
+#define __LLVM_LIBC_TYPES_STRUCT_RUSAGE_H__
+
+#include <llvm-libc-types/struct_timeval.h>
+
+struct rusage {
+  struct timeval ru_utime;
+  struct timeval ru_stime;
+#ifdef __linux__
+  // Following fields are linux extensions as expected by the
+  // linux syscalls.
+  long ru_maxrss;   // Maximum resident set size
+  long ru_ixrss;    // Integral shared memory size
+  long ru_idrss;    // Integral unshared data size
+  long ru_isrss;    // Integral unshared stack size
+  long ru_minflt;   // Page reclaims
+  long ru_majflt;   // Page faults
+  long ru_nswap;    // Swaps
+  long ru_inblock;  // Block input operations
+  long ru_oublock;  // Block output operations
+  long ru_msgsnd;   // Messages sent
+  long ru_msgrcv;   // Messages received
+  long ru_nsignals; // Signals received
+  long ru_nvcsw;    // Voluntary context switches
+  long ru_nivcsw;   // Involuntary context switches
+#endif
+};
+
+#endif // __LLVM_LIBC_TYPES_STRUCT_RUSAGE_H__

libc/include/llvm-libc-types/struct_timeval.h

@@ -0,0 +1,20 @@
+//===-- Definition of struct timeval -------------------------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef __LLVM_LIBC_TYPES_TIMEVAL_H__
+#define __LLVM_LIBC_TYPES_TIMEVAL_H__
+
+#include <llvm-libc-types/suseconds_t.h>
+#include <llvm-libc-types/time_t.h>
+
+struct timeval {
+  time_t tv_sec;       // Seconds
+  suseconds_t tv_usec; // Micro seconds
+};
+
+#endif // __LLVM_LIBC_TYPES_TIMEVAL_H__

libc/include/llvm-libc-types/suseconds_t.h

@@ -0,0 +1,14 @@
+//===-- Definition of suseconds_t type ------------------------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef __LLVM_LIBC_TYPES_SUSECONDS_T_H__
+#define __LLVM_LIBC_TYPES_SUSECONDS_T_H__
+
+typedef __INT32_TYPE__ suseconds_t;
+
+#endif // __LLVM_LIBC_TYPES_SUSECONDS_T_H__

libc/spec/bsd_ext.td

@@ -18,7 +18,22 @@ def BsdExtensions : StandardSpec<"BSDExtensions"> {
       ]
   >;
 
+  HeaderSpec SysWait = HeaderSpec<
+      "sys/wait.h",
+      [], // Macros
+      [StructRUsage], // Types
+      [], // Enumerations
+      [
+        FunctionSpec<
+            "wait4",
+            RetValSpec<PidT>,
+            [ArgSpec<PidT>, ArgSpec<IntPtr>, ArgSpec<IntType>, ArgSpec<StructRUsagePtr>]
+        >
+      ]
+  >;
+
   let Headers = [
     String,
+    SysWait,
   ];
 }

libc/spec/posix.td

@@ -971,13 +971,18 @@ def POSIX : StandardSpec<"POSIX"> {
   HeaderSpec SysWait = HeaderSpec<
     "sys/wait.h",
     [], // Macros
-    [PidT],
+    [PidT, StructRUsage],
     [], // Enumerations
     [
       FunctionSpec<
         "wait",
         RetValSpec<PidT>,
         [ArgSpec<IntPtr>]
+      >,
+      FunctionSpec<
+        "waitpid",
+        RetValSpec<PidT>,
+        [ArgSpec<PidT>, ArgSpec<IntPtr>, ArgSpec<IntType>]
       >
     ]
   >;

libc/spec/spec.td

@@ -115,6 +115,8 @@ def FILERestrictedPtr : RestrictedPtrType<FILE>;
 def PThreadTType : NamedType<"pthread_t">;
 
 def PidT : NamedType<"pid_t">;
+def StructRUsage : NamedType<"struct rusage">;
+def StructRUsagePtr : PtrType<StructRUsage>;
 
 //added because __assert_fail needs it.
 def UnsignedType : NamedType<"unsigned">;

libc/src/sys/wait/CMakeLists.txt

@@ -8,3 +8,17 @@ add_entrypoint_object(
   DEPENDS
     .${LIBC_TARGET_OS}.wait
 )
+
+add_entrypoint_object(
+  wait4
+  ALIAS
+  DEPENDS
+    .${LIBC_TARGET_OS}.wait4
+)
+
+add_entrypoint_object(
+  waitpid
+  ALIAS
+  DEPENDS
+    .${LIBC_TARGET_OS}.waitpid
+)

libc/src/sys/wait/linux/CMakeLists.txt

@@ -11,3 +11,31 @@ add_entrypoint_object(
     libc.src.__support.OSUtil.osutil
     libc.src.errno.errno
 )
+
+add_entrypoint_object(
+  wait4
+  SRCS
+    wait4.cpp
+  HDRS
+    ../wait4.h
+  DEPENDS
+    libc.include.errno
+    libc.include.sys_wait
+    libc.include.sys_syscall
+    libc.src.__support.OSUtil.osutil
+    libc.src.errno.errno
+)
+
+add_entrypoint_object(
+  waitpid
+  SRCS
+    waitpid.cpp
+  HDRS
+    ../waitpid.h
+  DEPENDS
+    libc.include.errno
+    libc.include.sys_wait
+    libc.include.sys_syscall
+    libc.src.__support.OSUtil.osutil
+    libc.src.errno.errno
+)

libc/src/sys/wait/linux/wait4.cpp

@@ -0,0 +1,31 @@
+//===-- Linux implementation of wait4 -------------------------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#include "src/sys/wait/wait4.h"
+
+#include "src/__support/OSUtil/syscall.h" // For internal syscall function.
+#include "src/__support/common.h"
+
+#include <errno.h>
+#include <sys/syscall.h> // For syscall numbers.
+#include <sys/wait.h>
+
+namespace __llvm_libc {
+
+LLVM_LIBC_FUNCTION(pid_t, wait4,
+                   (pid_t pid, int *wait_status, int options,
+                    struct rusage *usage)) {
+  pid = __llvm_libc::syscall_impl(SYS_wait4, pid, wait_status, options, usage);
+  if (pid < 0) {
+    errno = -pid;
+    return -1;
+  }
+  return pid;
+}
+
+} // namespace __llvm_libc

libc/src/sys/wait/linux/waitpid.cpp

@@ -0,0 +1,29 @@
+//===-- Linux implementation of waitpid -----------------------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#include "src/sys/wait/waitpid.h"
+
+#include "src/__support/OSUtil/syscall.h" // For internal syscall function.
+#include "src/__support/common.h"
+
+#include <errno.h>
+#include <sys/syscall.h> // For syscall numbers.
+#include <sys/wait.h>
+
+namespace __llvm_libc {
+
+LLVM_LIBC_FUNCTION(pid_t, waitpid, (pid_t pid, int *wait_status, int options)) {
+  pid = __llvm_libc::syscall_impl(SYS_wait4, pid, wait_status, options, 0);
+  if (pid < 0) {
+    errno = -pid;
+    return -1;
+  }
+  return pid;
+}
+
+} // namespace __llvm_libc

libc/src/sys/wait/wait4.h

@@ -0,0 +1,20 @@
+//===-- Implementation header for wait4 -------------------------*- C++ -*-===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef LLVM_LIBC_SRC_SYS_WAIT_WAIT4_H
+#define LLVM_LIBC_SRC_SYS_WAIT_WAIT4_H
+
+#include <sys/wait.h>
+
+namespace __llvm_libc {
+
+pid_t wait4(pid_t pid, int *waitstatus, int options, struct rusage *usage);
+
+} // namespace __llvm_libc
+
+#endif // LLVM_LIBC_SRC_SYS_WAIT_WAIT4_H

libc/src/sys/wait/waitpid.h

@@ -0,0 +1,20 @@
+//===-- Implementation header for waitpid -----------------------*- C++ -*-===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef LLVM_LIBC_SRC_SYS_WAIT_WAITPID_H
+#define LLVM_LIBC_SRC_SYS_WAIT_WAITPID_H
+
+#include <sys/wait.h>
+
+namespace __llvm_libc {
+
+pid_t waitpid(pid_t pid, int *waitstatus, int options);
+
+} // namespace __llvm_libc
+
+#endif // LLVM_LIBC_SRC_SYS_WAIT_WAITPID_H

libc/test/integration/src/unistd/CMakeLists.txt

@@ -16,5 +16,7 @@ add_integration_test(
     libc.include.unistd
     libc.src.signal.raise
     libc.src.sys.wait.wait
+    libc.src.sys.wait.wait4
+    libc.src.sys.wait.waitpid
     libc.src.unistd.fork
 )