lib/eggdrop/socket.h File Reference

Go to the source code of this file.

Defines
#define	SOCKET_CLIENT   1
#define	SOCKET_SERVER   2
#define	SOCKET_BIND   4
#define	SOCKET_NONBLOCK   8
#define	SOCKET_TCP   16
#define	SOCKET_UDP   32
Functions
int	socket_create (const char *dest_ip, int dest_port, const char *src_ip, int src_port, int flags)
int	socket_close (int sock)
int	socket_set_nonblock (int desc, int value)
int	socket_get_name (int sock, char **ip, int *port)
int	socket_get_peer_name (int sock, char **peer_ip, int *peer_port)
int	socket_get_error (int sock)
int	socket_accept (int sock, char **peer_ip, int *peer_port)
int	socket_valid_ip (const char *ip)
int	socket_ip_to_uint (const char *ip, unsigned int *longip)
int	socket_ipv6_to_dots (const char *ip, char *dots)

---

Define Documentation

#define SOCKET_BIND   4

Definition at line 27 of file socket.h.

Referenced by socket_create().

#define SOCKET_CLIENT   1

Definition at line 25 of file socket.h.

Referenced by connect_host_resolved(), get_dns_idx(), and socket_create().

#define SOCKET_NONBLOCK   8

Definition at line 28 of file socket.h.

Referenced by connect_host_resolved(), egg_server(), get_dns_idx(), and socket_create().

#define SOCKET_SERVER   2

Definition at line 26 of file socket.h.

Referenced by egg_server(), and socket_create().

#define SOCKET_TCP   16

Definition at line 29 of file socket.h.

Referenced by connect_host_resolved(), and egg_server().

#define SOCKET_UDP   32

Definition at line 30 of file socket.h.

Referenced by get_dns_idx(), and socket_create().

---

Function Documentation

int socket_accept	(	int	sock,
		char **	peer_ip,
		int *	peer_port
	)

Definition at line 177 of file socket.c.

References sockname_t::addr, sockname_t::ipv4, NULL, and sockname_t::u.

Referenced by sockbuf_got_readable_server().

{
  int newsock;
  sockname_t name;
  socklen_t len;

  *peer_ip = NULL;
  *peer_port = 0;
  memset(&name, 0, sizeof(name));
  len = sizeof(name.u);
  newsock = accept(sock, &name.u.addr, &len);
#ifdef DO_IPV4
  if (len == sizeof(name.u.ipv4)) {
    *peer_ip = malloc(32);
    *peer_port = ntohs(name.u.ipv4.sin_port);
    inet_ntop(AF_INET, &name.u.ipv4.sin_addr, *peer_ip, 32);
  }
#endif
#ifdef DO_IPV6
  if (len == sizeof(name.u.ipv6)) {
    *peer_ip = malloc(128);
    *peer_port = ntohs(name.u.ipv6.sin6_port);
    if (IN6_IS_ADDR_V4MAPPED((&name.u.ipv6.sin6_addr))) {
      unsigned int ipv4part = 0;
      memcpy(&ipv4part, &name.u.ipv6.sin6_addr.s6_addr[12], 4);
      inet_ntop(AF_INET, &ipv4part, *peer_ip, 128);
    }
    else {
      inet_ntop(AF_INET6, &name.u.ipv6.sin6_addr, *peer_ip, 128);
    }
  }
#endif
  return(newsock);
}

int socket_close

(

int

sock

)

Definition at line 276 of file socket.c.

References SHUT_RDWR.

Referenced by sockbuf_close(), and sockbuf_delete().

{
  shutdown(sock, SHUT_RDWR);
  close(sock);
  return(0);
}

int socket_create	(	const char *	dest_ip,
		int	dest_port,
		const char *	src_ip,
		int	src_port,
		int	flags
	)

Definition at line 217 of file socket.c.

References sockname_t::addr, sockname_t::family, sockname_t::len, SOCKET_BIND, SOCKET_CLIENT, socket_name(), SOCKET_NONBLOCK, SOCKET_SERVER, socket_set_nonblock(), SOCKET_UDP, and sockname_t::u.

Referenced by connect_host_resolved(), egg_server(), and get_dns_idx().

{
  char *passive[] = {"::", "0.0.0.0"};
  int sock = -1, pfamily, try;
  sockname_t dest_name, src_name;

  /* If no source ip address is given, try :: and 0.0.0.0 (passive). */
  for (try = 0; try < 2; try++) {
    /* Resolve the ip addresses. */
    socket_name(&dest_name, dest_ip ? dest_ip : passive[try], dest_port);
    socket_name(&src_name, src_ip ? src_ip : passive[try], src_port);

    if (src_ip || src_port) flags |= SOCKET_BIND;

    if (flags & SOCKET_CLIENT) pfamily = dest_name.family;
    else if (flags & SOCKET_SERVER) pfamily = src_name.family;
    else {
      errno = EADDRNOTAVAIL;
      return(-1);
    }

    /* Create the socket. */
    if (flags & SOCKET_UDP) sock = socket(pfamily, SOCK_DGRAM, 0);
    else sock = socket(pfamily, SOCK_STREAM, 0);

    if (sock >= 0) break;
  }

  if (sock < 0) return(-2);

  if (flags & SOCKET_NONBLOCK) socket_set_nonblock(sock, 1);

  /* Do the bind if necessary. */
  if (flags & (SOCKET_SERVER|SOCKET_BIND)) {
    int yes = 1;

    setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int));
    if (bind(sock, &src_name.u.addr, src_name.len) != 0) {
      close(sock);
      return(-3);
    }
    if (flags & SOCKET_SERVER) listen(sock, 50);
  }

  if (flags & SOCKET_CLIENT) {
    if (connect(sock, &dest_name.u.addr, dest_name.len) != 0) {
      if (errno != EINPROGRESS) {
        close(sock);
        return(-4);
      }
    }
  }

  errno = 0;

  /* Yay, we're done. */
  return(sock);
}

int socket_get_error

(

int

sock

)

Definition at line 164 of file socket.c.

Referenced by sockbuf_got_eof(), and sockbuf_got_writable_client().

{
  int err;
  socklen_t size;

  if (sock < 0) return(0);

  size = sizeof(int);
  err = 0;
  getsockopt(sock, SOL_SOCKET, SO_ERROR, &err, &size);
  return(err);
}

int socket_get_name	(	int	sock,
		char **	ip,
		int *	port
	)

Definition at line 89 of file socket.c.

References sockname_t::addr, sockname_t::ipv4, NULL, and sockname_t::u.

Referenced by dcc_on_connect(), egg_server(), server_on_connect(), sockbuf_on_connect(), sockbuf_on_newclient(), and sockbuf_set_sock().

{
  sockname_t name;
  socklen_t namelen;

  if (ip) *ip = NULL;
  if (port) *port = 0;

#ifdef DO_IPV4
  namelen = sizeof(name.u.ipv4);
  if (!getsockname(sock, &name.u.addr, &namelen) && namelen == sizeof(name.u.ipv4)) {
    if (ip) {
      *ip = malloc(32);
      inet_ntop(AF_INET, &name.u.ipv4.sin_addr, *ip, 32);
    }
    if (port) *port = ntohs(name.u.ipv4.sin_port);
    return(0);
  }
#endif
#ifdef DO_IPV6
  namelen = sizeof(name.u.ipv6);
  if (!getsockname(sock, &name.u.addr, &namelen) && namelen == sizeof(name.u.ipv6)) {
    if (ip) {
      *ip = malloc(128);
      if (IN6_IS_ADDR_V4MAPPED((&name.u.ipv6.sin6_addr))) {
        unsigned int ipv4part = 0;
        memcpy(&ipv4part, &name.u.ipv6.sin6_addr.s6_addr[12], 4);
        inet_ntop(AF_INET, &ipv4part, *ip, 128);
      }
      else {
        inet_ntop(AF_INET6, &name.u.ipv6.sin6_addr, *ip, 128);
      }
    }
    if (port) *port = ntohs(name.u.ipv6.sin6_port);
    return(0);
  }
#endif

  return(-1);
}

int socket_get_peer_name	(	int	sock,
		char **	peer_ip,
		int *	peer_port
	)

Definition at line 130 of file socket.c.

References sockname_t::addr, sockname_t::ipv4, NULL, and sockname_t::u.

Referenced by sockbuf_got_writable_client().

{
  sockname_t name;
  socklen_t namelen;

  if (peer_ip) *peer_ip = NULL;
  if (peer_port) *peer_port = 0;

#ifdef DO_IPV4
  namelen = sizeof(name.u.ipv4);
  if (!getpeername(sock, &name.u.addr, &namelen) && namelen == sizeof(name.u.ipv4)) {
    if (peer_ip) {
      *peer_ip = malloc(32);
      inet_ntop(AF_INET, &name.u.ipv4.sin_addr, *peer_ip, 32);
    }
    if (peer_port) *peer_port = ntohs(name.u.ipv4.sin_port);
    return(0);
  }
#endif
#ifdef DO_IPV6
  namelen = sizeof(name.u.ipv6);
  if (!getpeername(sock, &name.u.addr, &namelen) && namelen == sizeof(name.u.ipv6)) {
    if (peer_ip) {
      *peer_ip = malloc(128);
      inet_ntop(AF_INET6, &name.u.ipv6.sin6_addr, *peer_ip, 128);
    }
    if (peer_port) *peer_port = ntohs(name.u.ipv6.sin6_port);
    return(0);
  }
#endif

  return(-1);
}

int socket_ip_to_uint	(	const char *	ip,
		unsigned int *	longip
	)

Definition at line 306 of file socket.c.

Referenced by dcc_dns_callback().

{
  struct in_addr addr;

  inet_pton(AF_INET, ip, &addr);
  *longip = htonl(addr.s_addr);
  return(0);
}

int socket_ipv6_to_dots	(	const char *	ip,
		char *	dots
	)

Definition at line 317 of file socket.c.

Referenced by egg_dns_reverse().

{
#ifndef DO_IPV6
  dots[0] = 0;
  return(-1);
#else
  struct in6_addr buf;

  dots[0] = 0;
  if (inet_pton(AF_INET6, ip, &buf) <= 0) return(-1);
  sprintf(dots, "%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x.%x",
    buf.s6_addr[0] >> 4, buf.s6_addr[0] & 0xf,
    buf.s6_addr[1] >> 4, buf.s6_addr[1] & 0xf,
    buf.s6_addr[2] >> 4, buf.s6_addr[2] & 0xf,
    buf.s6_addr[3] >> 4, buf.s6_addr[3] & 0xf,
    buf.s6_addr[4] >> 4, buf.s6_addr[4] & 0xf,
    buf.s6_addr[5] >> 4, buf.s6_addr[5] & 0xf,
    buf.s6_addr[6] >> 4, buf.s6_addr[6] & 0xf,
    buf.s6_addr[7] >> 4, buf.s6_addr[7] & 0xf,
    buf.s6_addr[8] >> 4, buf.s6_addr[8] & 0xf,
    buf.s6_addr[9] >> 4, buf.s6_addr[9] & 0xf,
    buf.s6_addr[10] >> 4, buf.s6_addr[10] & 0xf,
    buf.s6_addr[11] >> 4, buf.s6_addr[11] & 0xf,
    buf.s6_addr[12] >> 4, buf.s6_addr[12] & 0xf,
    buf.s6_addr[13] >> 4, buf.s6_addr[13] & 0xf,
    buf.s6_addr[14] >> 4, buf.s6_addr[14] & 0xf,
    buf.s6_addr[15] >> 4, buf.s6_addr[15] & 0xf
  );
  return(0);
#endif
}

int socket_set_nonblock	(	int	desc,
		int	value
	)

Definition at line 283 of file socket.c.

Referenced by sockbuf_got_readable_server(), and socket_create().

{
  int oldflags = fcntl(sock, F_GETFL, 0);
  if (oldflags == -1) return -1;
  if (value != 0) oldflags |= O_NONBLOCK;
  else oldflags &= ~O_NONBLOCK;

  return fcntl(sock, F_SETFL, oldflags);
}

int socket_valid_ip

(

const char *

ip

)

Definition at line 293 of file socket.c.

Referenced by egg_dns_lookup(), and egg_dns_reverse().

{
  char buf[512];

#ifdef DO_IPV6
  if (inet_pton(AF_INET6, ip, buf) > 0) return(1);
#endif
#ifdef DO_IPV4
  if (inet_pton(AF_INET, ip, buf) > 0) return(1);
#endif
  return(0);
}