ext/kgio/sock_for_fd.h

   1 #ifndef SOCK_FOR_FD_H
   2 #define SOCK_FOR_FD_H
   3 #include <ruby.h>
   4 #ifdef HAVE_RUBY_IO_H
   5 #  include <ruby/io.h>
   6 #else
   7 #  include <stdio.h>
   8 #  include <rubyio.h>
   9 #endif
  10
  11 #if defined(MakeOpenFile) && \
  12     defined(HAVE_RB_IO_T) && (HAVE_RB_IO_T == 1) && \
  13     defined(HAVE_RB_IO_ASCII8BIT_BINMODE) && \
  14     defined(HAVE_ST_FD) && \
  15     defined(HAVE_ST_MODE)
  16 #  define SOCK_FOR_FD (19)
  17 #  define FMODE_NOREVLOOKUP 0x100
  18 #elif defined(MakeOpenFile) && \
  19       (defined(OpenFile) || defined(HAVE_RB_IO_T)) && \
  20       defined(HAVE_RB_FDOPEN) && \
  21       defined(HAVE_ST_F) && \
  22       defined(HAVE_ST_F2) && \
  23       defined(HAVE_ST_MODE)
  24 #  define SOCK_FOR_FD (18)
  25 #else
  26 #  define SOCK_FOR_FD (-1)
  27 #endif
  28
  29 #if SOCK_FOR_FD == 19  /* modeled after ext/socket/init.c */
  30 static VALUE sock_for_fd(VALUE klass, int fd)
  31 {
  32         VALUE sock = rb_obj_alloc(klass);
  33         rb_io_t *fp;
  34
  35         MakeOpenFile(sock, fp);
  36         fp->fd = fd;
  37         fp->mode = FMODE_READWRITE|FMODE_DUPLEX|FMODE_NOREVLOOKUP;
  38         rb_io_ascii8bit_binmode(sock);
  39         rb_io_synchronized(fp);
  40         return sock;
  41 }
  42 #elif SOCK_FOR_FD == 18 /* modeled after init_sock() in ext/socket/socket.c */
  43 static VALUE sock_for_fd(VALUE klass, int fd)
  44 {
  45         VALUE sock = rb_obj_alloc(klass);
  46         OpenFile *fp;
  47
  48         MakeOpenFile(sock, fp);
  49         fp->f = rb_fdopen(fd, "r");
  50         fp->f2 = rb_fdopen(fd, "w");
  51         fp->mode = FMODE_READWRITE;
  52         rb_io_synchronized(fp);
  53         return sock;
  54 }
  55 #else /* Rubinius, et al. */
  56 static ID id_for_fd;
  57 static VALUE sock_for_fd(VALUE klass, int fd)
  58 {
  59         return rb_funcall(klass, id_for_fd, 1, INT2NUM(fd));
  60 }
  61 static void init_sock_for_fd(void)
  62 {
  63         id_for_fd = rb_intern("for_fd");
  64 }
  65 #endif /* sock_for_fd */
  66 #if SOCK_FOR_FD > 0
  67 #  define init_sock_for_fd() for (;0;)
  68 #endif
  69 #endif /* SOCK_FOR_FD_H */