/ext/socket/socket.c
C | 2030 lines | 1031 code | 174 blank | 825 comment | 178 complexity | 43a9ba5bcc555b1b15ae4a500151057e MD5 | raw file
Possible License(s): LGPL-2.1, AGPL-3.0, 0BSD, Unlicense, GPL-2.0, BSD-3-Clause
Large files files are truncated, but you can click here to view the full file
- /************************************************
- socket.c -
- created at: Thu Mar 31 12:21:29 JST 1994
- Copyright (C) 1993-2007 Yukihiro Matsumoto
- ************************************************/
- #include "rubysocket.h"
- static VALUE sym_wait_writable;
- static VALUE sock_s_unpack_sockaddr_in(VALUE, VALUE);
- void
- rsock_sys_fail_host_port(const char *mesg, VALUE host, VALUE port)
- {
- rsock_syserr_fail_host_port(errno, mesg, host, port);
- }
- void
- rsock_syserr_fail_host_port(int err, const char *mesg, VALUE host, VALUE port)
- {
- VALUE message;
- message = rb_sprintf("%s for %+"PRIsVALUE" port % "PRIsVALUE"",
- mesg, host, port);
- rb_syserr_fail_str(err, message);
- }
- void
- rsock_sys_fail_path(const char *mesg, VALUE path)
- {
- rsock_syserr_fail_path(errno, mesg, path);
- }
- void
- rsock_syserr_fail_path(int err, const char *mesg, VALUE path)
- {
- VALUE message;
- if (RB_TYPE_P(path, T_STRING)) {
- message = rb_sprintf("%s for % "PRIsVALUE"", mesg, path);
- rb_syserr_fail_str(err, message);
- }
- else {
- rb_syserr_fail(err, mesg);
- }
- }
- void
- rsock_sys_fail_sockaddr(const char *mesg, struct sockaddr *addr, socklen_t len)
- {
- rsock_syserr_fail_sockaddr(errno, mesg, addr, len);
- }
- void
- rsock_syserr_fail_sockaddr(int err, const char *mesg, struct sockaddr *addr, socklen_t len)
- {
- VALUE rai;
- rai = rsock_addrinfo_new(addr, len, PF_UNSPEC, 0, 0, Qnil, Qnil);
- rsock_syserr_fail_raddrinfo(err, mesg, rai);
- }
- void
- rsock_sys_fail_raddrinfo(const char *mesg, VALUE rai)
- {
- rsock_syserr_fail_raddrinfo(errno, mesg, rai);
- }
- void
- rsock_syserr_fail_raddrinfo(int err, const char *mesg, VALUE rai)
- {
- VALUE str, message;
- str = rsock_addrinfo_inspect_sockaddr(rai);
- message = rb_sprintf("%s for %"PRIsVALUE"", mesg, str);
- rb_syserr_fail_str(err, message);
- }
- void
- rsock_sys_fail_raddrinfo_or_sockaddr(const char *mesg, VALUE addr, VALUE rai)
- {
- rsock_syserr_fail_raddrinfo_or_sockaddr(errno, mesg, addr, rai);
- }
- void
- rsock_syserr_fail_raddrinfo_or_sockaddr(int err, const char *mesg, VALUE addr, VALUE rai)
- {
- if (NIL_P(rai)) {
- StringValue(addr);
- rsock_syserr_fail_sockaddr(err, mesg,
- (struct sockaddr *)RSTRING_PTR(addr),
- (socklen_t)RSTRING_LEN(addr)); /* overflow should be checked already */
- }
- else
- rsock_syserr_fail_raddrinfo(err, mesg, rai);
- }
- static void
- setup_domain_and_type(VALUE domain, int *dv, VALUE type, int *tv)
- {
- *dv = rsock_family_arg(domain);
- *tv = rsock_socktype_arg(type);
- }
- /*
- * call-seq:
- * Socket.new(domain, socktype [, protocol]) => socket
- *
- * Creates a new socket object.
- *
- * _domain_ should be a communications domain such as: :INET, :INET6, :UNIX, etc.
- *
- * _socktype_ should be a socket type such as: :STREAM, :DGRAM, :RAW, etc.
- *
- * _protocol_ is optional and should be a protocol defined in the domain.
- * If protocol is not given, 0 is used internally.
- *
- * Socket.new(:INET, :STREAM) # TCP socket
- * Socket.new(:INET, :DGRAM) # UDP socket
- * Socket.new(:UNIX, :STREAM) # UNIX stream socket
- * Socket.new(:UNIX, :DGRAM) # UNIX datagram socket
- */
- static VALUE
- sock_initialize(int argc, VALUE *argv, VALUE sock)
- {
- VALUE domain, type, protocol;
- int fd;
- int d, t;
- rb_scan_args(argc, argv, "21", &domain, &type, &protocol);
- if (NIL_P(protocol))
- protocol = INT2FIX(0);
- setup_domain_and_type(domain, &d, type, &t);
- fd = rsock_socket(d, t, NUM2INT(protocol));
- if (fd < 0) rb_sys_fail("socket(2)");
- return rsock_init_sock(sock, fd);
- }
- #if defined HAVE_SOCKETPAIR
- static VALUE
- io_call_close(VALUE io)
- {
- return rb_funcallv(io, rb_intern("close"), 0, 0);
- }
- static VALUE
- io_close(VALUE io)
- {
- return rb_rescue(io_call_close, io, 0, 0);
- }
- static VALUE
- pair_yield(VALUE pair)
- {
- return rb_ensure(rb_yield, pair, io_close, rb_ary_entry(pair, 1));
- }
- #endif
- #if defined HAVE_SOCKETPAIR
- static int
- rsock_socketpair0(int domain, int type, int protocol, int descriptors[2])
- {
- #ifdef SOCK_CLOEXEC
- type |= SOCK_CLOEXEC;
- #endif
- #ifdef SOCK_NONBLOCK
- type |= SOCK_NONBLOCK;
- #endif
- int result = socketpair(domain, type, protocol, descriptors);
- if (result == -1)
- return -1;
- #ifndef SOCK_CLOEXEC
- rb_fd_fix_cloexec(descriptors[0]);
- rb_fd_fix_cloexec(descriptors[1]);
- #endif
- #ifndef SOCK_NONBLOCK
- rsock_make_fd_nonblock(descriptors[0]);
- rsock_make_fd_nonblock(descriptors[1]);
- #endif
- return result;
- }
- static int
- rsock_socketpair(int domain, int type, int protocol, int descriptors[2])
- {
- int result;
- result = rsock_socketpair0(domain, type, protocol, descriptors);
- if (result < 0 && rb_gc_for_fd(errno)) {
- result = rsock_socketpair0(domain, type, protocol, descriptors);
- }
- return result;
- }
- /*
- * call-seq:
- * Socket.pair(domain, type, protocol) => [socket1, socket2]
- * Socket.socketpair(domain, type, protocol) => [socket1, socket2]
- *
- * Creates a pair of sockets connected each other.
- *
- * _domain_ should be a communications domain such as: :INET, :INET6, :UNIX, etc.
- *
- * _socktype_ should be a socket type such as: :STREAM, :DGRAM, :RAW, etc.
- *
- * _protocol_ should be a protocol defined in the domain,
- * defaults to 0 for the domain.
- *
- * s1, s2 = Socket.pair(:UNIX, :STREAM, 0)
- * s1.send "a", 0
- * s1.send "b", 0
- * s1.close
- * p s2.recv(10) #=> "ab"
- * p s2.recv(10) #=> ""
- * p s2.recv(10) #=> ""
- *
- * s1, s2 = Socket.pair(:UNIX, :DGRAM, 0)
- * s1.send "a", 0
- * s1.send "b", 0
- * p s2.recv(10) #=> "a"
- * p s2.recv(10) #=> "b"
- *
- */
- VALUE
- rsock_sock_s_socketpair(int argc, VALUE *argv, VALUE klass)
- {
- VALUE domain, type, protocol;
- int d, t, p, sp[2];
- int ret;
- VALUE s1, s2, r;
- rb_scan_args(argc, argv, "21", &domain, &type, &protocol);
- if (NIL_P(protocol))
- protocol = INT2FIX(0);
- setup_domain_and_type(domain, &d, type, &t);
- p = NUM2INT(protocol);
- ret = rsock_socketpair(d, t, p, sp);
- if (ret < 0) {
- rb_sys_fail("socketpair(2)");
- }
- s1 = rsock_init_sock(rb_obj_alloc(klass), sp[0]);
- s2 = rsock_init_sock(rb_obj_alloc(klass), sp[1]);
- r = rb_assoc_new(s1, s2);
- if (rb_block_given_p()) {
- return rb_ensure(pair_yield, r, io_close, s1);
- }
- return r;
- }
- #else
- #define rsock_sock_s_socketpair rb_f_notimplement
- #endif
- /*
- * call-seq:
- * socket.connect(remote_sockaddr) => 0
- *
- * Requests a connection to be made on the given +remote_sockaddr+. Returns 0 if
- * successful, otherwise an exception is raised.
- *
- * === Parameter
- * * +remote_sockaddr+ - the +struct+ sockaddr contained in a string or Addrinfo object
- *
- * === Example:
- * # Pull down Google's web page
- * require 'socket'
- * include Socket::Constants
- * socket = Socket.new( AF_INET, SOCK_STREAM, 0 )
- * sockaddr = Socket.pack_sockaddr_in( 80, 'www.google.com' )
- * socket.connect( sockaddr )
- * socket.write( "GET / HTTP/1.0\r\n\r\n" )
- * results = socket.read
- *
- * === Unix-based Exceptions
- * On unix-based systems the following system exceptions may be raised if
- * the call to _connect_ fails:
- * * Errno::EACCES - search permission is denied for a component of the prefix
- * path or write access to the +socket+ is denied
- * * Errno::EADDRINUSE - the _sockaddr_ is already in use
- * * Errno::EADDRNOTAVAIL - the specified _sockaddr_ is not available from the
- * local machine
- * * Errno::EAFNOSUPPORT - the specified _sockaddr_ is not a valid address for
- * the address family of the specified +socket+
- * * Errno::EALREADY - a connection is already in progress for the specified
- * socket
- * * Errno::EBADF - the +socket+ is not a valid file descriptor
- * * Errno::ECONNREFUSED - the target _sockaddr_ was not listening for connections
- * refused the connection request
- * * Errno::ECONNRESET - the remote host reset the connection request
- * * Errno::EFAULT - the _sockaddr_ cannot be accessed
- * * Errno::EHOSTUNREACH - the destination host cannot be reached (probably
- * because the host is down or a remote router cannot reach it)
- * * Errno::EINPROGRESS - the O_NONBLOCK is set for the +socket+ and the
- * connection cannot be immediately established; the connection will be
- * established asynchronously
- * * Errno::EINTR - the attempt to establish the connection was interrupted by
- * delivery of a signal that was caught; the connection will be established
- * asynchronously
- * * Errno::EISCONN - the specified +socket+ is already connected
- * * Errno::EINVAL - the address length used for the _sockaddr_ is not a valid
- * length for the address family or there is an invalid family in _sockaddr_
- * * Errno::ENAMETOOLONG - the pathname resolved had a length which exceeded
- * PATH_MAX
- * * Errno::ENETDOWN - the local interface used to reach the destination is down
- * * Errno::ENETUNREACH - no route to the network is present
- * * Errno::ENOBUFS - no buffer space is available
- * * Errno::ENOSR - there were insufficient STREAMS resources available to
- * complete the operation
- * * Errno::ENOTSOCK - the +socket+ argument does not refer to a socket
- * * Errno::EOPNOTSUPP - the calling +socket+ is listening and cannot be connected
- * * Errno::EPROTOTYPE - the _sockaddr_ has a different type than the socket
- * bound to the specified peer address
- * * Errno::ETIMEDOUT - the attempt to connect timed out before a connection
- * was made.
- *
- * On unix-based systems if the address family of the calling +socket+ is
- * AF_UNIX the follow exceptions may be raised if the call to _connect_
- * fails:
- * * Errno::EIO - an i/o error occurred while reading from or writing to the
- * file system
- * * Errno::ELOOP - too many symbolic links were encountered in translating
- * the pathname in _sockaddr_
- * * Errno::ENAMETOOLLONG - a component of a pathname exceeded NAME_MAX
- * characters, or an entire pathname exceeded PATH_MAX characters
- * * Errno::ENOENT - a component of the pathname does not name an existing file
- * or the pathname is an empty string
- * * Errno::ENOTDIR - a component of the path prefix of the pathname in _sockaddr_
- * is not a directory
- *
- * === Windows Exceptions
- * On Windows systems the following system exceptions may be raised if
- * the call to _connect_ fails:
- * * Errno::ENETDOWN - the network is down
- * * Errno::EADDRINUSE - the socket's local address is already in use
- * * Errno::EINTR - the socket was cancelled
- * * Errno::EINPROGRESS - a blocking socket is in progress or the service provider
- * is still processing a callback function. Or a nonblocking connect call is
- * in progress on the +socket+.
- * * Errno::EALREADY - see Errno::EINVAL
- * * Errno::EADDRNOTAVAIL - the remote address is not a valid address, such as
- * ADDR_ANY TODO check ADDRANY TO INADDR_ANY
- * * Errno::EAFNOSUPPORT - addresses in the specified family cannot be used with
- * with this +socket+
- * * Errno::ECONNREFUSED - the target _sockaddr_ was not listening for connections
- * refused the connection request
- * * Errno::EFAULT - the socket's internal address or address length parameter
- * is too small or is not a valid part of the user space address
- * * Errno::EINVAL - the +socket+ is a listening socket
- * * Errno::EISCONN - the +socket+ is already connected
- * * Errno::ENETUNREACH - the network cannot be reached from this host at this time
- * * Errno::EHOSTUNREACH - no route to the network is present
- * * Errno::ENOBUFS - no buffer space is available
- * * Errno::ENOTSOCK - the +socket+ argument does not refer to a socket
- * * Errno::ETIMEDOUT - the attempt to connect timed out before a connection
- * was made.
- * * Errno::EWOULDBLOCK - the socket is marked as nonblocking and the
- * connection cannot be completed immediately
- * * Errno::EACCES - the attempt to connect the datagram socket to the
- * broadcast address failed
- *
- * === See
- * * connect manual pages on unix-based systems
- * * connect function in Microsoft's Winsock functions reference
- */
- static VALUE
- sock_connect(VALUE sock, VALUE addr)
- {
- VALUE rai;
- rb_io_t *fptr;
- int fd, n;
- SockAddrStringValueWithAddrinfo(addr, rai);
- addr = rb_str_new4(addr);
- GetOpenFile(sock, fptr);
- fd = fptr->fd;
- n = rsock_connect(fd, (struct sockaddr*)RSTRING_PTR(addr), RSTRING_SOCKLEN(addr), 0, NULL);
- if (n < 0) {
- rsock_sys_fail_raddrinfo_or_sockaddr("connect(2)", addr, rai);
- }
- return INT2FIX(n);
- }
- /* :nodoc: */
- static VALUE
- sock_connect_nonblock(VALUE sock, VALUE addr, VALUE ex)
- {
- VALUE rai;
- rb_io_t *fptr;
- int n;
- SockAddrStringValueWithAddrinfo(addr, rai);
- addr = rb_str_new4(addr);
- GetOpenFile(sock, fptr);
- rb_io_set_nonblock(fptr);
- n = connect(fptr->fd, (struct sockaddr*)RSTRING_PTR(addr), RSTRING_SOCKLEN(addr));
- if (n < 0) {
- int e = errno;
- if (e == EINPROGRESS) {
- if (ex == Qfalse) {
- return sym_wait_writable;
- }
- rb_readwrite_syserr_fail(RB_IO_WAIT_WRITABLE, e, "connect(2) would block");
- }
- if (e == EISCONN) {
- if (ex == Qfalse) {
- return INT2FIX(0);
- }
- }
- rsock_syserr_fail_raddrinfo_or_sockaddr(e, "connect(2)", addr, rai);
- }
- return INT2FIX(n);
- }
- /*
- * call-seq:
- * socket.bind(local_sockaddr) => 0
- *
- * Binds to the given local address.
- *
- * === Parameter
- * * +local_sockaddr+ - the +struct+ sockaddr contained in a string or an Addrinfo object
- *
- * === Example
- * require 'socket'
- *
- * # use Addrinfo
- * socket = Socket.new(:INET, :STREAM, 0)
- * socket.bind(Addrinfo.tcp("127.0.0.1", 2222))
- * p socket.local_address #=> #<Addrinfo: 127.0.0.1:2222 TCP>
- *
- * # use struct sockaddr
- * include Socket::Constants
- * socket = Socket.new( AF_INET, SOCK_STREAM, 0 )
- * sockaddr = Socket.pack_sockaddr_in( 2200, 'localhost' )
- * socket.bind( sockaddr )
- *
- * === Unix-based Exceptions
- * On unix-based based systems the following system exceptions may be raised if
- * the call to _bind_ fails:
- * * Errno::EACCES - the specified _sockaddr_ is protected and the current
- * user does not have permission to bind to it
- * * Errno::EADDRINUSE - the specified _sockaddr_ is already in use
- * * Errno::EADDRNOTAVAIL - the specified _sockaddr_ is not available from the
- * local machine
- * * Errno::EAFNOSUPPORT - the specified _sockaddr_ is not a valid address for
- * the family of the calling +socket+
- * * Errno::EBADF - the _sockaddr_ specified is not a valid file descriptor
- * * Errno::EFAULT - the _sockaddr_ argument cannot be accessed
- * * Errno::EINVAL - the +socket+ is already bound to an address, and the
- * protocol does not support binding to the new _sockaddr_ or the +socket+
- * has been shut down.
- * * Errno::EINVAL - the address length is not a valid length for the address
- * family
- * * Errno::ENAMETOOLONG - the pathname resolved had a length which exceeded
- * PATH_MAX
- * * Errno::ENOBUFS - no buffer space is available
- * * Errno::ENOSR - there were insufficient STREAMS resources available to
- * complete the operation
- * * Errno::ENOTSOCK - the +socket+ does not refer to a socket
- * * Errno::EOPNOTSUPP - the socket type of the +socket+ does not support
- * binding to an address
- *
- * On unix-based based systems if the address family of the calling +socket+ is
- * Socket::AF_UNIX the follow exceptions may be raised if the call to _bind_
- * fails:
- * * Errno::EACCES - search permission is denied for a component of the prefix
- * path or write access to the +socket+ is denied
- * * Errno::EDESTADDRREQ - the _sockaddr_ argument is a null pointer
- * * Errno::EISDIR - same as Errno::EDESTADDRREQ
- * * Errno::EIO - an i/o error occurred
- * * Errno::ELOOP - too many symbolic links were encountered in translating
- * the pathname in _sockaddr_
- * * Errno::ENAMETOOLLONG - a component of a pathname exceeded NAME_MAX
- * characters, or an entire pathname exceeded PATH_MAX characters
- * * Errno::ENOENT - a component of the pathname does not name an existing file
- * or the pathname is an empty string
- * * Errno::ENOTDIR - a component of the path prefix of the pathname in _sockaddr_
- * is not a directory
- * * Errno::EROFS - the name would reside on a read only filesystem
- *
- * === Windows Exceptions
- * On Windows systems the following system exceptions may be raised if
- * the call to _bind_ fails:
- * * Errno::ENETDOWN-- the network is down
- * * Errno::EACCES - the attempt to connect the datagram socket to the
- * broadcast address failed
- * * Errno::EADDRINUSE - the socket's local address is already in use
- * * Errno::EADDRNOTAVAIL - the specified address is not a valid address for this
- * computer
- * * Errno::EFAULT - the socket's internal address or address length parameter
- * is too small or is not a valid part of the user space addressed
- * * Errno::EINVAL - the +socket+ is already bound to an address
- * * Errno::ENOBUFS - no buffer space is available
- * * Errno::ENOTSOCK - the +socket+ argument does not refer to a socket
- *
- * === See
- * * bind manual pages on unix-based systems
- * * bind function in Microsoft's Winsock functions reference
- */
- static VALUE
- sock_bind(VALUE sock, VALUE addr)
- {
- VALUE rai;
- rb_io_t *fptr;
- SockAddrStringValueWithAddrinfo(addr, rai);
- GetOpenFile(sock, fptr);
- if (bind(fptr->fd, (struct sockaddr*)RSTRING_PTR(addr), RSTRING_SOCKLEN(addr)) < 0)
- rsock_sys_fail_raddrinfo_or_sockaddr("bind(2)", addr, rai);
- return INT2FIX(0);
- }
- /*
- * call-seq:
- * socket.listen( int ) => 0
- *
- * Listens for connections, using the specified +int+ as the backlog. A call
- * to _listen_ only applies if the +socket+ is of type SOCK_STREAM or
- * SOCK_SEQPACKET.
- *
- * === Parameter
- * * +backlog+ - the maximum length of the queue for pending connections.
- *
- * === Example 1
- * require 'socket'
- * include Socket::Constants
- * socket = Socket.new( AF_INET, SOCK_STREAM, 0 )
- * sockaddr = Socket.pack_sockaddr_in( 2200, 'localhost' )
- * socket.bind( sockaddr )
- * socket.listen( 5 )
- *
- * === Example 2 (listening on an arbitrary port, unix-based systems only):
- * require 'socket'
- * include Socket::Constants
- * socket = Socket.new( AF_INET, SOCK_STREAM, 0 )
- * socket.listen( 1 )
- *
- * === Unix-based Exceptions
- * On unix based systems the above will work because a new +sockaddr+ struct
- * is created on the address ADDR_ANY, for an arbitrary port number as handed
- * off by the kernel. It will not work on Windows, because Windows requires that
- * the +socket+ is bound by calling _bind_ before it can _listen_.
- *
- * If the _backlog_ amount exceeds the implementation-dependent maximum
- * queue length, the implementation's maximum queue length will be used.
- *
- * On unix-based based systems the following system exceptions may be raised if the
- * call to _listen_ fails:
- * * Errno::EBADF - the _socket_ argument is not a valid file descriptor
- * * Errno::EDESTADDRREQ - the _socket_ is not bound to a local address, and
- * the protocol does not support listening on an unbound socket
- * * Errno::EINVAL - the _socket_ is already connected
- * * Errno::ENOTSOCK - the _socket_ argument does not refer to a socket
- * * Errno::EOPNOTSUPP - the _socket_ protocol does not support listen
- * * Errno::EACCES - the calling process does not have appropriate privileges
- * * Errno::EINVAL - the _socket_ has been shut down
- * * Errno::ENOBUFS - insufficient resources are available in the system to
- * complete the call
- *
- * === Windows Exceptions
- * On Windows systems the following system exceptions may be raised if
- * the call to _listen_ fails:
- * * Errno::ENETDOWN - the network is down
- * * Errno::EADDRINUSE - the socket's local address is already in use. This
- * usually occurs during the execution of _bind_ but could be delayed
- * if the call to _bind_ was to a partially wildcard address (involving
- * ADDR_ANY) and if a specific address needs to be committed at the
- * time of the call to _listen_
- * * Errno::EINPROGRESS - a Windows Sockets 1.1 call is in progress or the
- * service provider is still processing a callback function
- * * Errno::EINVAL - the +socket+ has not been bound with a call to _bind_.
- * * Errno::EISCONN - the +socket+ is already connected
- * * Errno::EMFILE - no more socket descriptors are available
- * * Errno::ENOBUFS - no buffer space is available
- * * Errno::ENOTSOC - +socket+ is not a socket
- * * Errno::EOPNOTSUPP - the referenced +socket+ is not a type that supports
- * the _listen_ method
- *
- * === See
- * * listen manual pages on unix-based systems
- * * listen function in Microsoft's Winsock functions reference
- */
- VALUE
- rsock_sock_listen(VALUE sock, VALUE log)
- {
- rb_io_t *fptr;
- int backlog;
- backlog = NUM2INT(log);
- GetOpenFile(sock, fptr);
- if (listen(fptr->fd, backlog) < 0)
- rb_sys_fail("listen(2)");
- return INT2FIX(0);
- }
- /*
- * call-seq:
- * socket.recvfrom(maxlen) => [mesg, sender_addrinfo]
- * socket.recvfrom(maxlen, flags) => [mesg, sender_addrinfo]
- *
- * Receives up to _maxlen_ bytes from +socket+. _flags_ is zero or more
- * of the +MSG_+ options. The first element of the results, _mesg_, is the data
- * received. The second element, _sender_addrinfo_, contains protocol-specific
- * address information of the sender.
- *
- * === Parameters
- * * +maxlen+ - the maximum number of bytes to receive from the socket
- * * +flags+ - zero or more of the +MSG_+ options
- *
- * === Example
- * # In one file, start this first
- * require 'socket'
- * include Socket::Constants
- * socket = Socket.new( AF_INET, SOCK_STREAM, 0 )
- * sockaddr = Socket.pack_sockaddr_in( 2200, 'localhost' )
- * socket.bind( sockaddr )
- * socket.listen( 5 )
- * client, client_addrinfo = socket.accept
- * data = client.recvfrom( 20 )[0].chomp
- * puts "I only received 20 bytes '#{data}'"
- * sleep 1
- * socket.close
- *
- * # In another file, start this second
- * require 'socket'
- * include Socket::Constants
- * socket = Socket.new( AF_INET, SOCK_STREAM, 0 )
- * sockaddr = Socket.pack_sockaddr_in( 2200, 'localhost' )
- * socket.connect( sockaddr )
- * socket.puts "Watch this get cut short!"
- * socket.close
- *
- * === Unix-based Exceptions
- * On unix-based based systems the following system exceptions may be raised if the
- * call to _recvfrom_ fails:
- * * Errno::EAGAIN - the +socket+ file descriptor is marked as O_NONBLOCK and no
- * data is waiting to be received; or MSG_OOB is set and no out-of-band data
- * is available and either the +socket+ file descriptor is marked as
- * O_NONBLOCK or the +socket+ does not support blocking to wait for
- * out-of-band-data
- * * Errno::EWOULDBLOCK - see Errno::EAGAIN
- * * Errno::EBADF - the +socket+ is not a valid file descriptor
- * * Errno::ECONNRESET - a connection was forcibly closed by a peer
- * * Errno::EFAULT - the socket's internal buffer, address or address length
- * cannot be accessed or written
- * * Errno::EINTR - a signal interrupted _recvfrom_ before any data was available
- * * Errno::EINVAL - the MSG_OOB flag is set and no out-of-band data is available
- * * Errno::EIO - an i/o error occurred while reading from or writing to the
- * filesystem
- * * Errno::ENOBUFS - insufficient resources were available in the system to
- * perform the operation
- * * Errno::ENOMEM - insufficient memory was available to fulfill the request
- * * Errno::ENOSR - there were insufficient STREAMS resources available to
- * complete the operation
- * * Errno::ENOTCONN - a receive is attempted on a connection-mode socket that
- * is not connected
- * * Errno::ENOTSOCK - the +socket+ does not refer to a socket
- * * Errno::EOPNOTSUPP - the specified flags are not supported for this socket type
- * * Errno::ETIMEDOUT - the connection timed out during connection establishment
- * or due to a transmission timeout on an active connection
- *
- * === Windows Exceptions
- * On Windows systems the following system exceptions may be raised if
- * the call to _recvfrom_ fails:
- * * Errno::ENETDOWN - the network is down
- * * Errno::EFAULT - the internal buffer and from parameters on +socket+ are not
- * part of the user address space, or the internal fromlen parameter is
- * too small to accommodate the peer address
- * * Errno::EINTR - the (blocking) call was cancelled by an internal call to
- * the WinSock function WSACancelBlockingCall
- * * Errno::EINPROGRESS - a blocking Windows Sockets 1.1 call is in progress or
- * the service provider is still processing a callback function
- * * Errno::EINVAL - +socket+ has not been bound with a call to _bind_, or an
- * unknown flag was specified, or MSG_OOB was specified for a socket with
- * SO_OOBINLINE enabled, or (for byte stream-style sockets only) the internal
- * len parameter on +socket+ was zero or negative
- * * Errno::EISCONN - +socket+ is already connected. The call to _recvfrom_ is
- * not permitted with a connected socket on a socket that is connection
- * oriented or connectionless.
- * * Errno::ENETRESET - the connection has been broken due to the keep-alive
- * activity detecting a failure while the operation was in progress.
- * * Errno::EOPNOTSUPP - MSG_OOB was specified, but +socket+ is not stream-style
- * such as type SOCK_STREAM. OOB data is not supported in the communication
- * domain associated with +socket+, or +socket+ is unidirectional and
- * supports only send operations
- * * Errno::ESHUTDOWN - +socket+ has been shutdown. It is not possible to
- * call _recvfrom_ on a socket after _shutdown_ has been invoked.
- * * Errno::EWOULDBLOCK - +socket+ is marked as nonblocking and a call to
- * _recvfrom_ would block.
- * * Errno::EMSGSIZE - the message was too large to fit into the specified buffer
- * and was truncated.
- * * Errno::ETIMEDOUT - the connection has been dropped, because of a network
- * failure or because the system on the other end went down without
- * notice
- * * Errno::ECONNRESET - the virtual circuit was reset by the remote side
- * executing a hard or abortive close. The application should close the
- * socket; it is no longer usable. On a UDP-datagram socket this error
- * indicates a previous send operation resulted in an ICMP Port Unreachable
- * message.
- */
- static VALUE
- sock_recvfrom(int argc, VALUE *argv, VALUE sock)
- {
- return rsock_s_recvfrom(sock, argc, argv, RECV_SOCKET);
- }
- /* :nodoc: */
- static VALUE
- sock_recvfrom_nonblock(VALUE sock, VALUE len, VALUE flg, VALUE str, VALUE ex)
- {
- return rsock_s_recvfrom_nonblock(sock, len, flg, str, ex, RECV_SOCKET);
- }
- /*
- * call-seq:
- * socket.accept => [client_socket, client_addrinfo]
- *
- * Accepts a next connection.
- * Returns a new Socket object and Addrinfo object.
- *
- * serv = Socket.new(:INET, :STREAM, 0)
- * serv.listen(5)
- * c = Socket.new(:INET, :STREAM, 0)
- * c.connect(serv.connect_address)
- * p serv.accept #=> [#<Socket:fd 6>, #<Addrinfo: 127.0.0.1:48555 TCP>]
- *
- */
- static VALUE
- sock_accept(VALUE server)
- {
- union_sockaddr buffer;
- socklen_t length = (socklen_t)sizeof(buffer);
- VALUE peer = rsock_s_accept(rb_cSocket, server, &buffer.addr, &length);
- return rb_assoc_new(peer, rsock_io_socket_addrinfo(peer, &buffer.addr, length));
- }
- /* :nodoc: */
- static VALUE
- sock_accept_nonblock(VALUE sock, VALUE ex)
- {
- rb_io_t *fptr;
- VALUE sock2;
- union_sockaddr buf;
- struct sockaddr *addr = &buf.addr;
- socklen_t len = (socklen_t)sizeof buf;
- GetOpenFile(sock, fptr);
- sock2 = rsock_s_accept_nonblock(rb_cSocket, ex, fptr, addr, &len);
- if (SYMBOL_P(sock2)) /* :wait_readable */
- return sock2;
- return rb_assoc_new(sock2, rsock_io_socket_addrinfo(sock2, &buf.addr, len));
- }
- /*
- * call-seq:
- * socket.sysaccept => [client_socket_fd, client_addrinfo]
- *
- * Accepts an incoming connection returning an array containing the (integer)
- * file descriptor for the incoming connection, _client_socket_fd_,
- * and an Addrinfo, _client_addrinfo_.
- *
- * === Example
- * # In one script, start this first
- * require 'socket'
- * include Socket::Constants
- * socket = Socket.new( AF_INET, SOCK_STREAM, 0 )
- * sockaddr = Socket.pack_sockaddr_in( 2200, 'localhost' )
- * socket.bind( sockaddr )
- * socket.listen( 5 )
- * client_fd, client_addrinfo = socket.sysaccept
- * client_socket = Socket.for_fd( client_fd )
- * puts "The client said, '#{client_socket.readline.chomp}'"
- * client_socket.puts "Hello from script one!"
- * socket.close
- *
- * # In another script, start this second
- * require 'socket'
- * include Socket::Constants
- * socket = Socket.new( AF_INET, SOCK_STREAM, 0 )
- * sockaddr = Socket.pack_sockaddr_in( 2200, 'localhost' )
- * socket.connect( sockaddr )
- * socket.puts "Hello from script 2."
- * puts "The server said, '#{socket.readline.chomp}'"
- * socket.close
- *
- * Refer to Socket#accept for the exceptions that may be thrown if the call
- * to _sysaccept_ fails.
- *
- * === See
- * * Socket#accept
- */
- static VALUE
- sock_sysaccept(VALUE server)
- {
- union_sockaddr buffer;
- socklen_t length = (socklen_t)sizeof(buffer);
- VALUE peer = rsock_s_accept(0, server, &buffer.addr, &length);
- return rb_assoc_new(peer, rsock_io_socket_addrinfo(peer, &buffer.addr, length));
- }
- #ifdef HAVE_GETHOSTNAME
- /*
- * call-seq:
- * Socket.gethostname => hostname
- *
- * Returns the hostname.
- *
- * p Socket.gethostname #=> "hal"
- *
- * Note that it is not guaranteed to be able to convert to IP address using gethostbyname, getaddrinfo, etc.
- * If you need local IP address, use Socket.ip_address_list.
- */
- static VALUE
- sock_gethostname(VALUE obj)
- {
- #if defined(NI_MAXHOST)
- # define RUBY_MAX_HOST_NAME_LEN NI_MAXHOST
- #elif defined(HOST_NAME_MAX)
- # define RUBY_MAX_HOST_NAME_LEN HOST_NAME_MAX
- #else
- # define RUBY_MAX_HOST_NAME_LEN 1024
- #endif
- long len = RUBY_MAX_HOST_NAME_LEN;
- VALUE name;
- name = rb_str_new(0, len);
- while (gethostname(RSTRING_PTR(name), len) < 0) {
- int e = errno;
- switch (e) {
- case ENAMETOOLONG:
- #ifdef __linux__
- case EINVAL:
- /* glibc before version 2.1 uses EINVAL instead of ENAMETOOLONG */
- #endif
- break;
- default:
- rb_syserr_fail(e, "gethostname(3)");
- }
- rb_str_modify_expand(name, len);
- len += len;
- }
- rb_str_resize(name, strlen(RSTRING_PTR(name)));
- return name;
- }
- #else
- #ifdef HAVE_UNAME
- #include <sys/utsname.h>
- static VALUE
- sock_gethostname(VALUE obj)
- {
- struct utsname un;
- uname(&un);
- return rb_str_new2(un.nodename);
- }
- #else
- #define sock_gethostname rb_f_notimplement
- #endif
- #endif
- static VALUE
- make_addrinfo(struct rb_addrinfo *res0, int norevlookup)
- {
- VALUE base, ary;
- struct addrinfo *res;
- if (res0 == NULL) {
- rb_raise(rb_eSocket, "host not found");
- }
- base = rb_ary_new();
- for (res = res0->ai; res; res = res->ai_next) {
- ary = rsock_ipaddr(res->ai_addr, res->ai_addrlen, norevlookup);
- if (res->ai_canonname) {
- RARRAY_ASET(ary, 2, rb_str_new2(res->ai_canonname));
- }
- rb_ary_push(ary, INT2FIX(res->ai_family));
- rb_ary_push(ary, INT2FIX(res->ai_socktype));
- rb_ary_push(ary, INT2FIX(res->ai_protocol));
- rb_ary_push(base, ary);
- }
- return base;
- }
- static VALUE
- sock_sockaddr(struct sockaddr *addr, socklen_t len)
- {
- char *ptr;
- switch (addr->sa_family) {
- case AF_INET:
- ptr = (char*)&((struct sockaddr_in*)addr)->sin_addr.s_addr;
- len = (socklen_t)sizeof(((struct sockaddr_in*)addr)->sin_addr.s_addr);
- break;
- #ifdef AF_INET6
- case AF_INET6:
- ptr = (char*)&((struct sockaddr_in6*)addr)->sin6_addr.s6_addr;
- len = (socklen_t)sizeof(((struct sockaddr_in6*)addr)->sin6_addr.s6_addr);
- break;
- #endif
- default:
- rb_raise(rb_eSocket, "unknown socket family:%d", addr->sa_family);
- break;
- }
- return rb_str_new(ptr, len);
- }
- /*
- * call-seq:
- * Socket.gethostbyname(hostname) => [official_hostname, alias_hostnames, address_family, *address_list]
- *
- * Use Addrinfo.getaddrinfo instead.
- * This method is deprecated for the following reasons:
- *
- * - The 3rd element of the result is the address family of the first address.
- * The address families of the rest of the addresses are not returned.
- * - Uncommon address representation:
- * 4/16-bytes binary string to represent IPv4/IPv6 address.
- * - gethostbyname() may take a long time and it may block other threads.
- * (GVL cannot be released since gethostbyname() is not thread safe.)
- * - This method uses gethostbyname() function already removed from POSIX.
- *
- * This method obtains the host information for _hostname_.
- *
- * p Socket.gethostbyname("hal") #=> ["localhost", ["hal"], 2, "\x7F\x00\x00\x01"]
- *
- */
- static VALUE
- sock_s_gethostbyname(VALUE obj, VALUE host)
- {
- rb_warn("Socket.gethostbyname is deprecated; use Addrinfo.getaddrinfo instead.");
- struct rb_addrinfo *res =
- rsock_addrinfo(host, Qnil, AF_UNSPEC, SOCK_STREAM, AI_CANONNAME);
- return rsock_make_hostent(host, res, sock_sockaddr);
- }
- /*
- * call-seq:
- * Socket.gethostbyaddr(address_string [, address_family]) => hostent
- *
- * Use Addrinfo#getnameinfo instead.
- * This method is deprecated for the following reasons:
- *
- * - Uncommon address representation:
- * 4/16-bytes binary string to represent IPv4/IPv6 address.
- * - gethostbyaddr() may take a long time and it may block other threads.
- * (GVL cannot be released since gethostbyname() is not thread safe.)
- * - This method uses gethostbyname() function already removed from POSIX.
- *
- * This method obtains the host information for _address_.
- *
- * p Socket.gethostbyaddr([221,186,184,68].pack("CCCC"))
- * #=> ["carbon.ruby-lang.org", [], 2, "\xDD\xBA\xB8D"]
- *
- * p Socket.gethostbyaddr([127,0,0,1].pack("CCCC"))
- * ["localhost", [], 2, "\x7F\x00\x00\x01"]
- * p Socket.gethostbyaddr(([0]*15+[1]).pack("C"*16))
- * #=> ["localhost", ["ip6-localhost", "ip6-loopback"], 10,
- * "\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x01"]
- *
- */
- static VALUE
- sock_s_gethostbyaddr(int argc, VALUE *argv, VALUE _)
- {
- VALUE addr, family;
- struct hostent *h;
- char **pch;
- VALUE ary, names;
- int t = AF_INET;
- rb_warn("Socket.gethostbyaddr is deprecated; use Addrinfo#getnameinfo instead.");
- rb_scan_args(argc, argv, "11", &addr, &family);
- StringValue(addr);
- if (!NIL_P(family)) {
- t = rsock_family_arg(family);
- }
- #ifdef AF_INET6
- else if (RSTRING_LEN(addr) == 16) {
- t = AF_INET6;
- }
- #endif
- h = gethostbyaddr(RSTRING_PTR(addr), RSTRING_SOCKLEN(addr), t);
- if (h == NULL) {
- #ifdef HAVE_HSTRERROR
- extern int h_errno;
- rb_raise(rb_eSocket, "%s", (char*)hstrerror(h_errno));
- #else
- rb_raise(rb_eSocket, "host not found");
- #endif
- }
- ary = rb_ary_new();
- rb_ary_push(ary, rb_str_new2(h->h_name));
- names = rb_ary_new();
- rb_ary_push(ary, names);
- if (h->h_aliases != NULL) {
- for (pch = h->h_aliases; *pch; pch++) {
- rb_ary_push(names, rb_str_new2(*pch));
- }
- }
- rb_ary_push(ary, INT2NUM(h->h_addrtype));
- #ifdef h_addr
- for (pch = h->h_addr_list; *pch; pch++) {
- rb_ary_push(ary, rb_str_new(*pch, h->h_length));
- }
- #else
- rb_ary_push(ary, rb_str_new(h->h_addr, h->h_length));
- #endif
- return ary;
- }
- /*
- * call-seq:
- * Socket.getservbyname(service_name) => port_number
- * Socket.getservbyname(service_name, protocol_name) => port_number
- *
- * Obtains the port number for _service_name_.
- *
- * If _protocol_name_ is not given, "tcp" is assumed.
- *
- * Socket.getservbyname("smtp") #=> 25
- * Socket.getservbyname("shell") #=> 514
- * Socket.getservbyname("syslog", "udp") #=> 514
- */
- static VALUE
- sock_s_getservbyname(int argc, VALUE *argv, VALUE _)
- {
- VALUE service, proto;
- struct servent *sp;
- long port;
- const char *servicename, *protoname = "tcp";
- rb_scan_args(argc, argv, "11", &service, &proto);
- StringValue(service);
- if (!NIL_P(proto)) StringValue(proto);
- servicename = StringValueCStr(service);
- if (!NIL_P(proto)) protoname = StringValueCStr(proto);
- sp = getservbyname(servicename, protoname);
- if (sp) {
- port = ntohs(sp->s_port);
- }
- else {
- char *end;
- port = STRTOUL(servicename, &end, 0);
- if (*end != '\0') {
- rb_raise(rb_eSocket, "no such service %s/%s", servicename, protoname);
- }
- }
- return INT2FIX(port);
- }
- /*
- * call-seq:
- * Socket.getservbyport(port [, protocol_name]) => service
- *
- * Obtains the port number for _port_.
- *
- * If _protocol_name_ is not given, "tcp" is assumed.
- *
- * Socket.getservbyport(80) #=> "www"
- * Socket.getservbyport(514, "tcp") #=> "shell"
- * Socket.getservbyport(514, "udp") #=> "syslog"
- *
- */
- static VALUE
- sock_s_getservbyport(int argc, VALUE *argv, VALUE _)
- {
- VALUE port, proto;
- struct servent *sp;
- long portnum;
- const char *protoname = "tcp";
- rb_scan_args(argc, argv, "11", &port, &proto);
- portnum = NUM2LONG(port);
- if (portnum != (uint16_t)portnum) {
- const char *s = portnum > 0 ? "big" : "small";
- rb_raise(rb_eRangeError, "integer %ld too %s to convert into `int16_t'", portnum, s);
- }
- if (!NIL_P(proto)) protoname = StringValueCStr(proto);
- sp = getservbyport((int)htons((uint16_t)portnum), protoname);
- if (!sp) {
- rb_raise(rb_eSocket, "no such service for port %d/%s", (int)portnum, protoname);
- }
- return rb_str_new2(sp->s_name);
- }
- /*
- * call-seq:
- * Socket.getaddrinfo(nodename, servname[, family[, socktype[, protocol[, flags[, reverse_lookup]]]]]) => array
- *
- * Obtains address information for _nodename_:_servname_.
- *
- * Note that Addrinfo.getaddrinfo provides the same functionality in
- * an object oriented style.
- *
- * _family_ should be an address family such as: :INET, :INET6, etc.
- *
- * _socktype_ should be a socket type such as: :STREAM, :DGRAM, :RAW, etc.
- *
- * _protocol_ should be a protocol defined in the family,
- * and defaults to 0 for the family.
- *
- * _flags_ should be bitwise OR of Socket::AI_* constants.
- *
- * Socket.getaddrinfo("www.ruby-lang.org", "http", nil, :STREAM)
- * #=> [["AF_INET", 80, "carbon.ruby-lang.org", "221.186.184.68", 2, 1, 6]] # PF_INET/SOCK_STREAM/IPPROTO_TCP
- *
- * Socket.getaddrinfo("localhost", nil)
- * #=> [["AF_INET", 0, "localhost", "127.0.0.1", 2, 1, 6], # PF_INET/SOCK_STREAM/IPPROTO_TCP
- * # ["AF_INET", 0, "localhost", "127.0.0.1", 2, 2, 17], # PF_INET/SOCK_DGRAM/IPPROTO_UDP
- * # ["AF_INET", 0, "localhost", "127.0.0.1", 2, 3, 0]] # PF_INET/SOCK_RAW/IPPROTO_IP
- *
- * _reverse_lookup_ directs the form of the third element, and has to
- * be one of below. If _reverse_lookup_ is omitted, the default value is +nil+.
- *
- * +true+, +:hostname+: hostname is obtained from numeric address using reverse lookup, which may take a time.
- * +false+, +:numeric+: hostname is the same as numeric address.
- * +nil+: obey to the current +do_not_reverse_lookup+ flag.
- *
- * If Addrinfo object is preferred, use Addrinfo.getaddrinfo.
- */
- static VALUE
- sock_s_getaddrinfo(int argc, VALUE *argv, VALUE _)
- {
- VALUE host, port, family, socktype, protocol, flags, ret, revlookup;
- struct addrinfo hints;
- struct rb_addrinfo *res;
- int norevlookup;
- rb_scan_args(argc, argv, "25", &host, &port, &family, &socktype, &protocol, &flags, &revlookup);
- MEMZERO(&hints, struct addrinfo, 1);
- hints.ai_family = NIL_P(family) ? PF_UNSPEC : rsock_family_arg(family);
- if (!NIL_P(socktype)) {
- hints.ai_socktype = rsock_socktype_arg(socktype);
- }
- if (!NIL_P(protocol)) {
- hints.ai_protocol = NUM2INT(protocol);
- }
- if (!NIL_P(flags)) {
- hints.ai_flags = NUM2INT(flags);
- }
- if (NIL_P(revlookup) || !rsock_revlookup_flag(revlookup, &norevlookup)) {
- norevlookup = rsock_do_not_reverse_lookup;
- }
- res = rsock_getaddrinfo(host, port, &hints, 0);
- ret = make_addrinfo(res, norevlookup);
- rb_freeaddrinfo(res);
- return ret;
- }
- /*
- * call-seq:
- * Socket.getnameinfo(sockaddr [, flags]) => [hostname, servicename]
- *
- * Obtains name information for _sockaddr_.
- *
- * _sockaddr_ should be one of follows.
- * - packed sockaddr string such as Socket.sockaddr_in(80, "127.0.0.1")
- * - 3-elements array such as ["AF_INET", 80, "127.0.0.1"]
- * - 4-elements array such as ["AF_INET", 80, ignored, "127.0.0.1"]
- *
- * _flags_ should be bitwise OR of Socket::NI_* constants.
- *
- * Note:
- * The last form is compatible with IPSocket#addr and IPSocket#peeraddr.
- *
- * Socket.getnameinfo(Socket.sockaddr_in(80, "127.0.0.1")) #=> ["localhost", "www"]
- * Socket.getnameinfo(["AF_INET", 80, "127.0.0.1"]) #=> ["localhost", "www"]
- * Socket.getnameinfo(["AF_INET", 80, "localhost", "127.0.0.1"]) #=> ["localhost", "www"]
- *
- * If Addrinfo object is preferred, use Addrinfo#getnameinfo.
- */
- static VALUE
- sock_s_getnameinfo(int argc, VALUE *argv, VALUE _)
- {
- VALUE sa, af = Qnil, host = Qnil, port = Qnil, flags, tmp;
- char hbuf[1024], pbuf[1024];
- int fl;
- struct rb_addrinfo *res = NULL;
- struct addrinfo hints, *r;
- int error, saved_errno;
- union_sockaddr ss;
- struct sockaddr *sap;
- socklen_t salen;
- sa = flags = Qnil;
- rb_scan_args(argc, argv, "11", &sa, &flags);
- fl = 0;
- if (!NIL_P(flags)) {
- fl = NUM2INT(flags);
- }
- tmp = rb_check_sockaddr_string_type(sa);
- if (!NIL_P(tmp)) {
- sa = tmp;
- if (sizeof(ss) < (size_t)RSTRING_LEN(sa)) {
- rb_raise(rb_eTypeError, "sockaddr length too big");
- }
- memcpy(&ss, RSTRING_PTR(sa), RSTRING_LEN(sa));
- if (!VALIDATE_SOCKLEN(&ss.addr, RSTRING_LEN(sa))) {
- rb_raise(rb_eTypeError, "sockaddr size differs - should not happen");
- }
- sap = &ss.addr;
- salen = RSTRING_SOCKLEN(sa);
- goto call_nameinfo;
- }
- tmp = rb_check_array_type(sa);
- if (!NIL_P(tmp)) {
- sa = tmp;
- MEMZERO(&hints, struct addrinfo, 1);
- if (RARRAY_LEN(sa) == 3) {
- af = RARRAY_AREF(sa, 0);
- port = RARRAY_AREF(sa, 1);
- host = RARRAY_AREF(sa, 2);
- }
- else if (RARRAY_LEN(sa) >= 4) {
- af = RARRAY_AREF(sa, 0);
- port = RARRAY_AREF(sa, 1);
- host = RARRAY_AREF(sa, 3);
- if (NIL_P(host)) {
- host = RARRAY_AREF(sa, 2);
- }
- else {
- /*
- * 4th element holds numeric form, don't resolve.
- * see rsock_ipaddr().
- */
- #ifdef AI_NUMERICHOST /* AIX 4.3.3 doesn't have AI_NUMERICHOST. */
- hints.ai_flags |= AI_NUMERICHOST;
- #endif
- }
- }
- else {
- rb_raise(rb_eArgError, "array size should be 3 or 4, %ld given",
- RARRAY_LEN(sa));
- }
- hints.ai_socktype = (fl & NI_DGRAM) ? SOCK_DGRAM : SOCK_STREAM;
- /* af */
- hints.ai_family = NIL_P(af) ? PF_UNSPEC : rsock_family_arg(af);
- res = rsock_getaddrinfo(host, port, &hints, 0);
- sap = res->ai->ai_addr;
- salen = res->ai->ai_addrlen;
- }
- else {
- rb_raise(rb_eTypeError, "expecting String or Array");
- }
- call_nameinfo:
- error = rb_getnameinfo(sap, salen, hbuf, sizeof(hbuf),
- pbuf, sizeof(pbuf), fl);
- if (error) goto error_exit_name;
- if (res) {
- for (r = res->ai->ai_next; r; r = r->ai_next) {
- char hbuf2[1024], pbuf2[1024];
- sap = r->ai_addr;
- salen = r->ai_addrlen;
- error = rb_getnameinfo(sap, salen, hbuf2, sizeof(hbuf2),
- pbuf2, sizeof(pbuf2), fl);
- if (error) goto error_exit_name;
- if (strcmp(hbuf, hbuf2) != 0|| strcmp(pbuf, pbuf2) != 0) {
- rb_freeaddrinfo(res);
- rb_raise(rb_eSocket, "sockaddr resolved to multiple nodename");
- }
- }
- rb_freeaddrinfo(res);
- }
- return rb_assoc_new(rb_str_new2(hbuf), rb_str_new2(pbuf));
- error_exit_name:
- saved_errno = errno;
- if (res) rb_freeaddrinfo(res);
- errno = saved_errno;
- rsock_raise_socket_error("getnameinfo", error);
- UNREACHABLE_RETURN(Qnil);
- }
- /*
- * call-seq:
- * Socket.sockaddr_in(port, host) => sockaddr
- * Socket.pack_sockaddr_in(port, host) => sockaddr
- *
- * Packs _port_ and _host_ as an AF_INET/AF_INET6 sockaddr string.
- *
- * Socket.sockaddr_in(80, "127.0.0.1")
- * #=> "\x02\x00\x00P\x7F\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00"
- *
- * Socket.sockaddr_in(80, "::1")
- * #=> "\n\x00\x00P\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x01\x00\x00\x00\x00"
- *
- */
- static VALUE
- sock_s_pack_sockaddr_in(VALUE self, VALUE port, VALUE host)
- {
- struct rb_addrinfo *res = rsock_addrinfo(host, port, AF_UNSPEC, 0, 0);
- VALUE addr = rb_str_new((char*)res->ai->ai_addr, res->ai->ai_addrlen);
- rb_freeaddrinfo(res);
- return addr;
- }
- /*
- * call-seq:
- * Socket.unpack_sockaddr_in(sockaddr) => [port, ip_address]
- *
- * Unpacks _sockaddr_ into port and ip_address.
- *
- * _sockaddr_ should be a string or an addrinfo for AF_INET/AF_INET6.
- *
- * sockaddr = Socket.sockaddr_in(80, "127.0.0.1")
- * p sockaddr #=> "\x02\x00\x00P\x7F\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00"
- * p Socket.unpack_sockaddr_in(sockaddr) #=> [80, "127.0.0.1"]
- *
- */
- static VALUE
- sock_s_unpack_sockaddr_in(VALUE self, VALUE addr)
- {
- struct sockaddr_in * sockaddr;
- VALUE host;
- sockaddr = (struct sockaddr_in*)SockAddrStringValuePtr(addr);
- if (RSTRING_LEN(addr) <
- (char*)&((struct sockaddr *)sockaddr)->sa_family +
- sizeof(((struct sockaddr *)sockaddr)->sa_family) -
- (char*)sockaddr)
- rb_raise(rb_eArgError, "too short sockaddr");
- if (((struct sockaddr *)sockaddr)->sa_family != AF_INET
- #ifdef INET6
- && ((struct sockaddr *)sockaddr)->sa_family != AF_INET6
- #endif
- ) {
- #ifdef INET6
- rb_raise(rb_eArgError, "not an AF_INET/AF_INET6 sockaddr");
- #else
- rb_raise(rb_eArgError, "not an AF_INET sockaddr");
- #endif
- }
- host = rsock_make_ipaddr((struct sockaddr*)sockaddr, RSTRING_SOCKLEN(addr));
- return rb_assoc_new(INT2NUM(ntohs(sockaddr->sin_port)), host);
- }
- #ifdef HAVE_SYS_UN_H
- /*
- * call-seq:
- * Socket.sockaddr_un(path) => sockaddr
- * Socket.pack_sockaddr_un(path) => sockaddr
- *
- * Packs _path_ as an AF_UNIX sockaddr string.
- *
- * Socket.sockaddr_un("/tmp/sock") #=> "\x01\x00/tmp/sock\x00\x00..."
- *
- */
- static VALUE
- sock_s_pack_sockaddr_un(VALUE self, VALUE path)
- {
- struct sockaddr_un sockaddr;
- VALUE addr;
- StringValue(path);
- INIT_SOCKADDR_UN(&sockaddr, sizeof(struct sockaddr_un));
- if (sizeof(sockaddr.sun_path) < (size_t)RSTRING_LEN(path)) {
- rb_raise(rb_eArgError, "too long unix socket path (%"PRIuSIZE" bytes given but %"PRIuSIZE" bytes max)",
- (size_t)RSTRING_LEN(path), sizeof(sockaddr.sun_path));
- }
- memcpy(sockaddr.sun_path, RSTRING_PTR(path), RSTRING_LEN(path));
- addr = rb_str_new((char*)&sockaddr, rsock_unix_sockaddr_len(path));
- return addr;
- }
- /*
- * call-seq:
- * Socket.unpack_sockaddr_un(sockaddr) => path
- *
- * Unpacks _sockaddr_ into path.
- *
- * _sockaddr_ should be a string or an addrinfo for AF_UNIX.
- *
- * sockaddr = Socket.sockaddr_un("/tmp/sock")
- * p Socket.unpack_sockaddr_un(sockaddr) #=> "/tmp/sock"
- *
- */
- static VALUE
- sock_s_unpack_sockaddr_un(VALUE self, VALUE addr)
- {
- struct sockaddr_un * sockaddr;
- VALUE path;
- sockaddr = (struct sockaddr_un*)SockAddrStringValuePtr(addr);
- if (RSTRING_LEN(addr) <
- (char*)&((struct sockaddr *)sockaddr)->sa_family +
- sizeof(((struct sockaddr *)sockaddr)->sa_family) -
- (char*)sockaddr)
- rb_raise(rb_eArgError, "too short sockaddr");
- if (((struct sockaddr *)sockaddr)->sa_family != AF_UNIX) {
- rb_raise(rb_eArgError, "not an AF_UNIX sockaddr");
- }
- if (sizeof(struct sockaddr_un) < (size_t)RSTRING_LEN(addr)) {
- rb_raise(rb_eTypeError, "too long sockaddr_un - %ld longer than %d",
- RSTRING_LEN(addr), (int)sizeof(struct sockaddr_un));
- }
- path = rsock_unixpath_str(sockaddr, RSTRING_SOCKLEN(addr));
- return path;
- }
- #endif
- #if defined(HAVE_GETIFADDRS) || defined(SIOCGLIFCONF) || defined(SIOCGIFCONF) || defined(_WIN32)
- static socklen_t
- sockaddr_len(struct sockaddr *addr)
- {
- if (addr == NULL)
- return 0;
- #ifdef HAVE_STRUCT_SOCKADDR_SA_LEN
- if (addr->sa_len != 0)
- return addr->sa_len;
- #endif
- switch (addr->sa_family) {
- case AF_INET:
- return (socklen_t)sizeof(struct sockaddr_in);
- #ifdef AF_INET6
- case AF_INET6:
- return (socklen_t)sizeof(struct sockaddr_in6);
- #endif
- #ifdef HAVE_SYS_UN_H
- case AF_UNIX:
- return (socklen_t)sizeof(struct sockaddr_un);
- #endif
- #ifdef AF_PACKET
- case AF_PACKET:
- return (socklen_t)(offsetof(struct sockaddr_ll, sll_addr) + ((struct sockaddr_ll *)addr)->sll_halen);
- #endif
- default:
- return (socklen_t)(offsetof(struct sockaddr, sa_family) + sizeof(addr->sa_family));
- }
- }
- socklen_t
- rsock_sockaddr_len(struct sockaddr *addr)
- {
- return sockaddr_len(addr);
- }
- static VALUE
- sockaddr_obj(struct sockaddr *addr, socklen_t len)
- {
- #if defined(AF_INET6) && defined(__KAME__)
- struct sockaddr_in6 addr6;
- #endif
- if (addr == NULL)
- return Qnil;
- len = sockaddr_len(addr);
- #if defined(__KAME__) && defined(AF_INET6)
- if (addr->sa_family == AF_INET6) {
- /* KAME uses the 2nd 16bit word of link local IPv6 address as interface index internally */
- /* http://orange.kame.net/dev/cvsweb.cgi/kame/IMPLEMENTATION */
- /* convert fe80:1::1 to fe80::1%1 */
- len = (socklen_t)sizeof(struct sockaddr_in6);
- memcpy(&addr6, addr, len);
- addr = (struct sockaddr *)&addr6;
- if (IN6_IS_ADDR_LINKLOCAL(&addr6.sin6_addr) &&
- addr6.sin6_scope_id == 0 &&
- (addr6.sin6_addr.s6_addr[2] || addr6.sin6_addr.s6_addr[3])) {
- addr6.sin6_scope_id = (addr6.sin6_addr.s6_addr[2] << 8) | addr6.sin6_addr.s6_addr[3];
- addr6.sin6_addr.s6_addr[2] = 0;
- addr6.sin6_addr.s6_addr[3] = 0;
- }
- }
- #endif
- return rsock_addrinfo_new(addr, len, addr->sa_family, 0, 0, Qnil, Qnil);
- }
- VALUE
- rsock_sockaddr_obj(struct sockaddr *addr, socklen_t len)
- {
- return sockaddr_obj(addr, len);
- }
- #endif
- #if defined(HAVE_GETIFADDRS) || (defined(SIOCGLIFCONF) && defined(SIOCGLIFNUM) && !defined(__hpux)) || defined(SIOCGIFCONF) || defined(_WIN32)
- /*
- * call-seq:
- * Socket.ip_address_list => array
- *
- * Returns local IP addresses as an array.
- *
- * The array contains Addrinfo objects.
- *
- * pp Socket.ip_address_list
- * #=> [#<Addrinfo: 127.0.0.1>,
- * #<Addrinfo: 192…
Large files files are truncated, but you can click here to view the full file