git.ozlabs.org Git - ccan/blob - ccan/io/test/run-08-read-after-hangup.c

   1 #include <ccan/io/io.h>
   2 /* Include the C files directly. */
   3 #include <ccan/io/poll.c>
   4 #include <ccan/io/io.c>
   5 #include <ccan/tap/tap.h>
   6 #include <sys/wait.h>
   7 #include <stdio.h>
   8 #include <signal.h>
   9
  10 static char inbuf[8];
  11
  12 static struct io_plan wake_it(struct io_conn *conn, struct io_conn *reader)
  13 {
  14         io_wake(inbuf);
  15         return io_close();
  16 }
  17
  18 static struct io_plan read_buf(struct io_conn *conn, void *unused)
  19 {
  20         return io_read(inbuf, 8, io_close_cb, NULL);
  21 }
  22
  23 int main(void)
  24 {
  25         int fds[2];
  26         struct io_conn *conn;
  27
  28         plan_tests(3);
  29
  30         ok1(pipe(fds) == 0);
  31         conn = io_new_conn(fds[0], io_wait(inbuf, read_buf, NULL));
  32         io_new_conn(fds[1], io_write("EASYTEST", 8, wake_it, conn));
  33
  34         ok1(io_loop() == NULL);
  35         ok1(memcmp(inbuf, "EASYTEST", sizeof(inbuf)) == 0);
  36
  37         /* This exits depending on whether all tests passed */
  38         return exit_status();
  39 }