Home | History | Annotate | Line # | Download | only in test 
      1 /* Copyright Joyent, Inc. and other Node contributors. All rights reserved.
      2  *
      3  * Permission is hereby granted, free of charge, to any person obtaining a copy
      4  * of this software and associated documentation files (the "Software"), to
      5  * deal in the Software without restriction, including without limitation the
      6  * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
      7  * sell copies of the Software, and to permit persons to whom the Software is
      8  * furnished to do so, subject to the following conditions:
      9  *
     10  * The above copyright notice and this permission notice shall be included in
     11  * all copies or substantial portions of the Software.
     12  *
     13  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
     14  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
     15  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
     16  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
     17  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
     18  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
     19  * IN THE SOFTWARE.
     20  */
     21 
     22 #include "uv.h"
     23 #include "task.h"
     24 
     25 
     26 static int get_tty_fd(void) {
     27   /* Make sure we have an FD that refers to a tty */
     28 #ifdef _WIN32
     29   HANDLE handle;
     30   handle = CreateFileA("conin$",
     31                        GENERIC_READ | GENERIC_WRITE,
     32                        FILE_SHARE_READ | FILE_SHARE_WRITE,
     33                        NULL,
     34                        OPEN_EXISTING,
     35                        FILE_ATTRIBUTE_NORMAL,
     36                        NULL);
     37   if (handle == INVALID_HANDLE_VALUE)
     38     return -1;
     39   return _open_osfhandle((intptr_t) handle, 0);
     40 #else /* unix */
     41   return open("/dev/tty", O_RDONLY, 0);
     42 #endif
     43 }
     44 
     45 
     46 TEST_IMPL(handle_fileno) {
     47   int r;
     48   int tty_fd;
     49   struct sockaddr_in addr;
     50   uv_os_fd_t fd;
     51   uv_tcp_t tcp;
     52   uv_udp_t udp;
     53   uv_pipe_t pipe;
     54   uv_tty_t tty;
     55   uv_idle_t idle;
     56   uv_loop_t* loop;
     57 
     58   loop = uv_default_loop();
     59   ASSERT_OK(uv_ip4_addr("127.0.0.1", TEST_PORT, &addr));
     60 
     61   r = uv_idle_init(loop, &idle);
     62   ASSERT_OK(r);
     63   r = uv_fileno((uv_handle_t*) &idle, &fd);
     64   ASSERT_EQ(r, UV_EINVAL);
     65   uv_close((uv_handle_t*) &idle, NULL);
     66 
     67   r = uv_tcp_init(loop, &tcp);
     68   ASSERT_OK(r);
     69   r = uv_fileno((uv_handle_t*) &tcp, &fd);
     70   ASSERT_EQ(r, UV_EBADF);
     71   r = uv_tcp_bind(&tcp, (const struct sockaddr*) &addr, 0);
     72   ASSERT_OK(r);
     73   r = uv_fileno((uv_handle_t*) &tcp, &fd);
     74   ASSERT_OK(r);
     75   uv_close((uv_handle_t*) &tcp, NULL);
     76   r = uv_fileno((uv_handle_t*) &tcp, &fd);
     77   ASSERT_EQ(r, UV_EBADF);
     78 
     79   r = uv_udp_init(loop, &udp);
     80   ASSERT_OK(r);
     81   r = uv_fileno((uv_handle_t*) &udp, &fd);
     82   ASSERT_EQ(r, UV_EBADF);
     83   r = uv_udp_bind(&udp, (const struct sockaddr*) &addr, 0);
     84   ASSERT_OK(r);
     85   r = uv_fileno((uv_handle_t*) &udp, &fd);
     86   ASSERT_OK(r);
     87   uv_close((uv_handle_t*) &udp, NULL);
     88   r = uv_fileno((uv_handle_t*) &udp, &fd);
     89   ASSERT_EQ(r, UV_EBADF);
     90 
     91   r = uv_pipe_init(loop, &pipe, 0);
     92   ASSERT_OK(r);
     93   r = uv_fileno((uv_handle_t*) &pipe, &fd);
     94   ASSERT_EQ(r, UV_EBADF);
     95   r = uv_pipe_bind(&pipe, TEST_PIPENAME);
     96   ASSERT_OK(r);
     97   r = uv_fileno((uv_handle_t*) &pipe, &fd);
     98   ASSERT_OK(r);
     99   uv_close((uv_handle_t*) &pipe, NULL);
    100   r = uv_fileno((uv_handle_t*) &pipe, &fd);
    101   ASSERT_EQ(r, UV_EBADF);
    102 
    103   tty_fd = get_tty_fd();
    104   if (tty_fd < 0) {
    105     fprintf(stderr, "Cannot open a TTY fd");
    106     fflush(stderr);
    107   } else {
    108     r = uv_tty_init(loop, &tty, tty_fd, 0);
    109     ASSERT_OK(r);
    110     ASSERT(uv_is_readable((uv_stream_t*) &tty));
    111     ASSERT(!uv_is_writable((uv_stream_t*) &tty));
    112     r = uv_fileno((uv_handle_t*) &tty, &fd);
    113     ASSERT_OK(r);
    114     uv_close((uv_handle_t*) &tty, NULL);
    115     r = uv_fileno((uv_handle_t*) &tty, &fd);
    116     ASSERT_EQ(r, UV_EBADF);
    117     ASSERT(!uv_is_readable((uv_stream_t*) &tty));
    118     ASSERT(!uv_is_writable((uv_stream_t*) &tty));
    119   }
    120 
    121   uv_run(loop, UV_RUN_DEFAULT);
    122 
    123   MAKE_VALGRIND_HAPPY(loop);
    124   return 0;
    125 }
    126