Fix poll/ppoll fortify test to avoid hanging in failed fortify clang test.

Bug: 19220800
Change-Id: Ie75c640183c4a41a499556fefb4f824a134a5fb1
This commit is contained in:
Yabin Cui 2015-02-03 17:52:32 -08:00
parent d80ec66121
commit f4fe6937aa

View File

@ -26,6 +26,7 @@
#include <sys/socket.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <time.h>
#if __BIONIC__
#define ASSERT_FORTIFY(expr) ASSERT_EXIT(expr, testing::KilledBySignal(SIGABRT), "FORTIFY")
@ -938,11 +939,15 @@ TEST(TEST_NAME, s_n_printf_macro_expansion) {
TEST_F(DEATHTEST, poll_fortified) {
nfds_t fd_count = atoi("2"); // suppress compiler optimizations
pollfd buf[1] = {{0, POLLIN, 0}};
ASSERT_FORTIFY(poll(buf, fd_count, -1));
// Set timeout to zero to prevent waiting in poll when fortify test fails.
ASSERT_FORTIFY(poll(buf, fd_count, 0));
}
TEST_F(DEATHTEST, ppoll_fortified) {
nfds_t fd_count = atoi("2"); // suppress compiler optimizations
pollfd buf[1] = {{0, POLLIN, 0}};
ASSERT_FORTIFY(ppoll(buf, fd_count, NULL, NULL));
// Set timeout to zero to prevent waiting in ppoll when fortify test fails.
timespec timeout;
timeout.tv_sec = timeout.tv_nsec = 0;
ASSERT_FORTIFY(ppoll(buf, fd_count, &timeout, NULL));
}