| // Copyright 2016 The Fuchsia Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include <stdio.h> |
| #include <unistd.h> |
| |
| #include <zircon/syscalls.h> |
| #include <zircon/types.h> |
| #include <unittest/unittest.h> |
| |
| static const char* msg = "This is a test message, please discard."; |
| |
| bool cleanup_test(void) { |
| BEGIN_TEST; |
| zx_handle_t p0[2], p1[2]; |
| zx_signals_t pending; |
| zx_status_t r; |
| |
| // TEST1 |
| // Create a channel, close one end, try to wait on the other. |
| r = zx_channel_create(0, p1, p1 + 1); |
| ASSERT_EQ(r, ZX_OK, "cleanup-test: channel create 1 failed"); |
| |
| zx_handle_close(p1[1]); |
| unittest_printf("cleanup-test: about to wait, should return with PEER_CLOSED\n"); |
| r = zx_object_wait_one(p1[0], ZX_CHANNEL_READABLE | ZX_CHANNEL_PEER_CLOSED, |
| ZX_TIME_INFINITE, &pending); |
| ASSERT_EQ(r, ZX_OK, "cleanup-test: FAILED"); |
| |
| ASSERT_EQ(pending, ZX_CHANNEL_PEER_CLOSED, "cleanup-test: FAILED"); |
| unittest_printf("cleanup-test: SUCCESS, observed PEER_CLOSED signal\n\n"); |
| zx_handle_close(p1[0]); |
| |
| // TEST2 |
| // Create a channel, close one end. Then create an event and write a |
| // message on the channel sending the event along. See that the write |
| // fails (because the other end is closed) and that the event is |
| // consumed (because the write failed). |
| r = zx_channel_create(0, p1, p1 + 1); |
| ASSERT_EQ(r, ZX_OK, "cleanup-test: channel create 1 failed"); |
| zx_handle_close(p1[1]); |
| |
| zx_handle_t event = ZX_HANDLE_INVALID; |
| r = zx_event_create(0u, &event); |
| ASSERT_EQ(r, ZX_OK, ""); |
| ASSERT_NE(event, ZX_HANDLE_INVALID, "cleanup-test: event create failed"); |
| r = zx_channel_write(p1[0], 0, &msg, sizeof(msg), &event, 1); |
| ASSERT_EQ(r, ZX_ERR_PEER_CLOSED, "cleanup-test: unexpected message_write return code"); |
| |
| r = zx_object_signal(event, 0u, ZX_EVENT_SIGNALED); |
| ASSERT_EQ(r, ZX_ERR_BAD_HANDLE, "cleanup-test: able to signal event!"); |
| unittest_printf("cleanup-test: SUCCESS, event is closed\n\n"); |
| |
| zx_handle_close(p1[0]); |
| |
| // TEST3 |
| // Simulates the case where we prepare a message channel with a |
| // message+channelhandle already in it and the far end closed, |
| // like we pass to newly created processes, but then (say |
| // process creation fails), we delete the other end of the |
| // channel we were going to send. At this point we expect |
| // that the channel handle bundled with the message should |
| // be closed and waiting on the opposing handle should |
| // signal PEER_CLOSED. |
| r = zx_channel_create(0, p0, p0 + 1); |
| ASSERT_EQ(r, ZX_OK, "cleanup-test: channel create 0 failed"); |
| |
| r = zx_channel_create(0, p1, p1 + 1); |
| ASSERT_EQ(r, ZX_OK, "cleanup-test: channel create 1 failed"); |
| |
| r = zx_channel_write(p0[0], 0, &msg, sizeof(msg), &p1[1], 1); |
| ASSERT_EQ(r, ZX_OK, "cleanup-test: channel write failed"); |
| |
| zx_handle_close(p0[0]); |
| zx_handle_close(p0[1]); |
| |
| unittest_printf("cleanup-test: about to wait, should return with PEER_CLOSED\n"); |
| r = zx_object_wait_one(p1[0], ZX_CHANNEL_PEER_CLOSED, ZX_TIME_INFINITE, NULL); |
| ASSERT_EQ(r, ZX_OK, "cleanup-test: FAILED"); |
| |
| unittest_printf("cleanup-test: PASSED\n"); |
| zx_handle_close(p1[0]); |
| END_TEST; |
| } |
| |
| BEGIN_TEST_CASE(cleanup_tests) |
| RUN_TEST(cleanup_test) |
| END_TEST_CASE(cleanup_tests) |
| |
| int main(int argc, char** argv) { |
| return unittest_run_all_tests(argc, argv) ? 0 : -1; |
| } |