crashpad/test/multiprocess_exec_test_child.cc
Scott Graham 380a1e9eaf Make multiprocess_exec_test_child depend on base to get OS_ defines
Some doofus really wanted to shoot himself in the foot (
https://codereview.chromium.org/808493003/diff/80001/util/test/multiprocess_exec_test_child.cc
), and it worked.

Depend on base to get build_config.h instead, so that OS_POSIX is also
set on Fuchsia.

Also, don't call getrlimit() on Fuchsia.

Bug: crashpad:196
Change-Id: I102de22751fd90889bfb59b6c4a577e99357094b
Reviewed-on: https://chromium-review.googlesource.com/868742
Commit-Queue: Scott Graham <scottmg@chromium.org>
Reviewed-by: Mark Mentovai <mark@chromium.org>
2018-01-17 18:28:22 +00:00

94 lines
2.6 KiB
C++
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

// Copyright 2014 The Crashpad Authors. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include <errno.h>
#include <limits.h>
#include <stdio.h>
#include <sys/types.h>
#include <algorithm>
#include "base/logging.h"
#include "build/build_config.h"
#if defined(OS_POSIX)
#include <sys/resource.h>
#include <unistd.h>
#elif defined(OS_WIN)
#include <windows.h>
#endif
int main(int argc, char* argv[]) {
#if defined(OS_POSIX)
#if defined(OS_FUCHSIA)
// getrlimit() is not implemented on Fuchsia. By construction, the child only
// receieves specific fds that it's given, but check low values as mild
// verification.
int last_fd = 1024;
#else
rlimit rlimit_nofile;
if (getrlimit(RLIMIT_NOFILE, &rlimit_nofile) != 0) {
LOG(FATAL) << "getrlimit";
}
int last_fd = static_cast<int>(rlimit_nofile.rlim_cur);
#endif // OS_FUCHSIA
// Make sure that theres nothing open at any FD higher than 3. All FDs other
// than stdin, stdout, and stderr should have been closed prior to or at
// exec().
for (int fd = STDERR_FILENO + 1; fd < last_fd; ++fd) {
if (close(fd) == 0 || errno != EBADF) {
LOG(FATAL) << "close";
}
}
// Read a byte from stdin, expecting it to be a specific value.
char c;
ssize_t rv = read(STDIN_FILENO, &c, 1);
if (rv != 1 || c != 'z') {
LOG(FATAL) << "read";
}
// Write a byte to stdout.
c = 'Z';
rv = write(STDOUT_FILENO, &c, 1);
if (rv != 1) {
LOG(FATAL) << "write";
}
#elif defined(OS_WIN)
// TODO(scottmg): Verify that only the handles we expect to be open, are.
// Read a byte from stdin, expecting it to be a specific value.
char c;
DWORD bytes_read;
HANDLE stdin_handle = GetStdHandle(STD_INPUT_HANDLE);
if (!ReadFile(stdin_handle, &c, 1, &bytes_read, nullptr) ||
bytes_read != 1 || c != 'z') {
LOG(FATAL) << "ReadFile";
}
// Write a byte to stdout.
c = 'Z';
DWORD bytes_written;
if (!WriteFile(
GetStdHandle(STD_OUTPUT_HANDLE), &c, 1, &bytes_written, nullptr) ||
bytes_written != 1) {
LOG(FATAL) << "WriteFile";
}
#endif // OS_POSIX
return 0;
}