/src/opensips/socket_info.h

Source (jump to first uncovered line)
/*
 * Copyright (C) 2001-2003 FhG Fokus
 *
 * This file is part of opensips, a free SIP server.
 *
 * opensips is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version
 *
 * opensips is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to" the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301  USA
 */

/*!
 * \file
 * \brief Find & manage listen addresses.
 * Contains code that initializes and handles server listen addresses
 * lists (struct socket_info). It is used mainly on startup.
 */


#ifndef socket_info_h
#define socket_info_h

#include <stdlib.h>

#include "ip_addr.h"
#include "dprint.h"
#include "globals.h"
#include "net/trans.h"
#include "ut.h"

struct last_real_ports {
  unsigned short local;
  unsigned short remote;
};

struct socket_info {
  int socket;
  str name; /*!< name - eg.: foo.bar or 10.0.0.1 */
  str tag;  /* the tag of the interface, use only in OpenSIPS ecosystem */
  struct ip_addr address; /*!< ip address */
  str address_str;        /*!< ip address converted to string -- optimization*/
  unsigned short port_no;  /*!< port number */
  str port_no_str; /*!< port number converted to string -- optimization*/
  enum si_flags flags; /*!< SI_IS_IP | SI_IS_LO | SI_IS_MCAST | SI_IS_ANYCAST */
  union sockaddr_union su;
  int proto; /*!< tcp or udp*/
  str sock_str;
  str adv_sock_str;
  str tag_sock_str;
  str adv_name_str; /* Advertised name of this interface */
  str adv_port_str; /* Advertised port of this interface */
  struct ip_addr adv_address; /* Advertised address in ip_addr form (for find_si) */
  unsigned short adv_port;    /* optimization for grep_sock_info() */
  unsigned short workers;
  unsigned short tos;
  struct scaling_profile *s_profile;
  void *extra_data;
  enum sip_protos internal_proto;

  /* these are IP-level local/remote ports used during the last write op via
   * this sock (or a connection belonging to this sock). These values are 
   * optional (populated only by the TCP-based protocol, for ephemeral ports.
   * Note: they are populate ONLY by a write op and they are not ever reset,
   * they are simply overwritten by the next write op on this socket/conn.
   * IMPORTANT: when reading them, be sure you are just after a write ops,
   * otherwise you may read old data here */
  struct last_real_ports *last_real_ports;
};

struct socket_info_full {
  struct socket_info socket_info;
  struct last_real_ports last_real_ports;
  struct socket_info_full* next;
  struct socket_info_full* prev;
};

#define get_socket_real_name(_s) \
  (&(_s)->sock_str)

#define get_socket_sip_name(_s) \
  ((_s)->adv_sock_str.len?&(_s)->adv_sock_str:&(_s)->sock_str)

#define get_socket_internal_name(_s) \
  ((_s)->tag_sock_str.len?&(_s)->tag_sock_str:&(_s)->sock_str)

#define NUM_IP_OCTETS 4
#define PROTO_NAME_MAX_SIZE  8 /* CHANGEME if you define a bigger protocol name
               * currently hep_tcp - biggest proto */

int new_sock2list(struct socket_id *sid, struct socket_info_full** list);

int fix_socket_list(struct socket_info_full **);

/*
 * This function will retrieve a list of all ip addresses and ports that
 * OpenSIPS is listening on, with respect to the transport protocol specified
 * with 'protocol'.
 *
 * The first parameter, ipList, is a pointer to a pointer. It will be assigned
 * new block of memory holding the IP Addresses and ports being listened to
 * with respect to 'protocol'.  The array maps a 2D array into a 1 dimensional
 * space, and is layed out as follows:
 *
 * The first NUM_IP_OCTETS indices will be the IP address, and the next index
 * the port.  So if NUM_IP_OCTETS is equal to 4 and there are two IP addresses
 * found, then:
 *
 *  - ipList[0] will be the first octet of the first ip address
 *  - ipList[3] will be the last octet of the first ip address.
 *  - iplist[4] will be the port of the first ip address
 *  -
 *  - iplist[5] will be the first octet of the first ip address,
 *  - and so on.
 *
 * The function will return the number of sockets which were found.  This can
 * be used to index into ipList.
 *
 * NOTE: This function assigns a block of memory equal to:
 *
 *            returnedValue * (NUM_IP_OCTETS + 1) * sizeof(int);
 *
 *       Therefore it is CRUCIAL that you free ipList when you are done with
 *       its contents, to avoid a nasty memory leak.
 */
int get_socket_list_from_proto(unsigned int **ipList, int protocol);

/*
 * Returns the sum of the number of bytes waiting to be consumed on all network
 * interfaces and transports that OpenSIPS is listening on.
 *
 * Note: This currently only works on systems supporting the
 *       /proc/net/[tcp|udp] interface.  On other systems, zero will always
 *       be returned.  Details of why this is so can be found in
 *       network_stats.c
 */
int get_total_bytes_waiting(int only_proto);

void print_aliases();

#define grep_sock_info(_host, _port, _proto) \
  grep_sock_info_ext(_host, _port, _proto, 0)

#define grep_internal_sock_info(_host, _port, _proto) \
  grep_sock_info_ext(_host, _port, _proto, 1)

const struct socket_info* grep_sock_info_ext(str* host, unsigned short port,
                    unsigned short proto, int check_tag);

const struct socket_info* parse_sock_info(str *spec);

const struct socket_info* find_si(const struct ip_addr* ip, unsigned short port,
                        unsigned short proto);

#define set_sip_defaults( _port, _proto) \
  do { \
    if ((_proto)==PROTO_NONE) (_proto) = PROTO_UDP; \
    if ((_port)==0) { \
      if ((_proto)==PROTO_TLS) (_port) = SIPS_PORT; else\
      (_port) = SIP_PORT; \
    } \
  } while(0)



/*! \brief helper function:
 * \return next protocol, if the last one is reached return 0
 * \note useful for cycling on the supported protocols */
static inline int next_proto(unsigned short proto)
{
  for( proto++ ; proto<PROTO_LAST ; proto++ )
    if (protos[proto].id!=PROTO_NONE)
      return proto;
  return PROTO_NONE;
}



/*! \brief gets first non-null socket_info structure
 * (useful if for. e.g we are not listening on any udp sockets )
 */
inline static const struct socket_info* get_first_socket(void)
{
  int p;

  for( p=0 ; p<PROTO_LAST ; p++ )
    if (protos[p].listeners)
      return &protos[p].listeners->socket_info;

  return NULL;
}


/*! \brief Sets protocol
 * \return -1 on error, 0 on success
 */
inline static int parse_proto(unsigned char* s, long len, int* proto)
{
#define PROTO2UINT(a, b, c) ((  (((unsigned int)(a))<<16)+ \
                (((unsigned int)(b))<<8)+  \
                ((unsigned int)(c)) ) | 0x20202020)
  unsigned int i;
  unsigned int j;

  /* must support 2-char arrays for ws
   * must support 3-char arrays for udp, tcp, tls, wss
   * must support 4-char arrays for sctp
   * must support 5-char arrays for ipsec
   * must support 7-char arrays for hep_tcp and hep_udp */
  *proto=PROTO_NONE;
  if ((len < 2 || len > 5) && len != 7) return -1;

  i=PROTO2UINT(s[0], s[1], s[2]);
  switch(i){
    case PROTO2UINT('u', 'd', 'p'):
      if(len==3) { *proto=PROTO_UDP; return 0; }
      break;
    case PROTO2UINT('t', 'c', 'p'):
      if(len==3) { *proto=PROTO_TCP; return 0; }
      break;
    case PROTO2UINT('t', 'l', 's'):
      if(len==3) { *proto=PROTO_TLS; return 0; }
      break;
    case PROTO2UINT('s', 'c', 't'):
      if(len==4 && (s[3]=='p' || s[3]=='P')) {
        *proto=PROTO_SCTP; return 0;
      }
      break;
    case PROTO2UINT('w', 's', 's'):
      if(len==3) { *proto=PROTO_WSS; return 0; }
      break;
    case PROTO2UINT('b', 'i', 'n'):
      if(len==3) { *proto=PROTO_BIN; return 0; }
      else if(len==4 && (s[3]=='s' || s[3]=='S')) {
        *proto=PROTO_BINS; return 0;
      }
      break;
    case PROTO2UINT('i', 'p', 's'):
      if(len==5 && (s[3]|0x20)=='e' && (s[4]|0x20)=='c') {
        *proto=PROTO_IPSEC; return 0;
      }
      return -1;

    case PROTO2UINT('h', 'e', 'p'):
      if (len != 7 || s[3] != '_') return -1;

      j=PROTO2UINT(s[4], s[5], s[6]);
      switch (j) {
        case PROTO2UINT('u','d', 'p'):
          *proto=PROTO_HEP_UDP;
          return 0;
        case PROTO2UINT('t','c', 'p'):
          *proto=PROTO_HEP_TCP;
          return 0;
        case PROTO2UINT('t','l', 's'):
          *proto=PROTO_HEP_TLS;
          return 0;
        default:
          return -1;
      }
      break;
    case PROTO2UINT('s', 'm', 'p'):
      if(len==4 && (s[3]=='p' || s[3]=='P')) {
        *proto=PROTO_SMPP; return 0;
      }
      break;
    case PROTO2UINT('m', 's', 'r'):
      if(len==4) {
        if (s[3]=='p' || s[3]=='P') {
          *proto=PROTO_MSRP; return 0;
        }
      } else
      if(len==5) {
        if ((s[3]=='p' || s[3]=='P') && (s[4]=='s' || s[4]=='S')) {
          *proto=PROTO_MSRPS; return 0;
        }
      }
      break;
    default:
      if(len==2 && (s[0]|0x20)=='w' && (s[1]|0x20)=='s') {
        *proto=PROTO_WS; return 0;
      }
      return -1;
  }
  return -1;
}



/*! \brief
 * parses [proto:]host[:port] where proto= udp|tcp|tls
 * \return 0 on success and -1 on failure
 */
inline static int parse_phostport(char* s, int slen, char** host, int* hlen,
                          int* port, int* proto)
{
  char* first; /* first ':' occurrence */
  char* second; /* second ':' occurrence */
  char* p;
  int   bracket;
  str   tmp;
  char* end;

  first=second=0;
  bracket=0;
  end = s + slen;

  /* find the first 2 ':', ignoring possible ipv6 addresses
   * (substrings between [])
   */
  for(p=s; p<end ; p++){
    switch(*p){
      case '[':
        bracket++;
        if (bracket>1) goto error_brackets;
        break;
      case ']':
        bracket--;
        if (bracket<0) goto error_brackets;
        break;
      case ':':
        if (bracket==0){
          if (first==0) first=p;
          else if( second==0) second=p;
          else goto error_colons;
        }
        break;
    }
  }
  if (p==s) return -1;
  if (*(p-1)==':') goto error_colons;

  if (first==0){ /* no ':' => only host */
    *host=s;
    *hlen=(int)(p-s);
    *port=0;
    *proto=0;
    return 0;
  }
  if (second){ /* 2 ':' found => check if valid */
    if (parse_proto((unsigned char*)s, first-s, proto)<0)
      goto error_proto;
    tmp.s = second+1;
    tmp.len = end - tmp.s;
    if (str2int( &tmp, (unsigned int*)port )==-1) goto error_port;
    *host=first+1;
    *hlen=(int)(second-*host);
    return 0;
  }
  /* only 1 ':' found => it's either proto:host or host:port */
  tmp.s = first+1;
  tmp.len = end - tmp.s;
  if (str2int( &tmp, (unsigned int*)port )==-1) {
    /* invalid port => it's proto:host */
    if (parse_proto((unsigned char*)s, first-s, proto)<0) goto error_proto;
    *port=0;
    *host=first+1;
    *hlen=(int)(p-*host);
  }else{
    /* valid port => its host:port */
    *proto=0;
    *host=s;
    *hlen=(int)(first-*host);
  }
  return 0;
error_brackets:
  LM_ERR("too many brackets in %s\n", s);
  return -1;
error_colons:
  LM_ERR(" too many colons in %s\n", s);
  return -1;
error_proto:
  LM_ERR("bad protocol in %s\n", s);
  return -1;
error_port:
  LM_ERR("bad port number in %s\n", s);
  return -1;
}


/* function will write the proto as string, starting from the p pointer. The
   new resulting pointer will be returned (where writing ended) */
static inline char* proto2str(int proto, char *p)
{
  switch (proto) {
    case PROTO_UDP:
      *(p++) = 'u';
      *(p++) = 'd';
      *(p++) = 'p';
      break;
    case PROTO_TCP:
      *(p++) = 't';
      *(p++) = 'c';
      *(p++) = 'p';
      break;
    case PROTO_TLS:
      *(p++) = 't';
      *(p++) = 'l';
      *(p++) = 's';
      break;
    case PROTO_SCTP:
      *(p++) = 's';
      *(p++) = 'c';
      *(p++) = 't';
      *(p++) = 'p';
      break;
    case PROTO_WS:
      *(p++) = 'w';
      *(p++) = 's';
      break;
    case PROTO_WSS:
      *(p++) = 'w';
      *(p++) = 's';
      *(p++) = 's';
      break;
    case PROTO_IPSEC:
      *(p++) = 'i';
      *(p++) = 'p';
      *(p++) = 's';
      *(p++) = 'e';
      *(p++) = 'c';
      break;
    case PROTO_BIN:
      *(p++) = 'b';
      *(p++) = 'i';
      *(p++) = 'n';
      break;
    case PROTO_BINS:
      *(p++) = 'b';
      *(p++) = 'i';
      *(p++) = 'n';
      *(p++) = 's';
      break;
    case PROTO_HEP_UDP:
      *(p++) = 'h';
      *(p++) = 'e';
      *(p++) = 'p';
      *(p++) = '_';
      *(p++) = 'u';
      *(p++) = 'd';
      *(p++) = 'p';
      break;
    case PROTO_HEP_TCP:
      *(p++) = 'h';
      *(p++) = 'e';
      *(p++) = 'p';
      *(p++) = '_';
      *(p++) = 't';
      *(p++) = 'c';
      *(p++) = 'p';
      break;
    case PROTO_HEP_TLS:
      *(p++) = 'h';
      *(p++) = 'e';
      *(p++) = 'p';
      *(p++) = '_';
      *(p++) = 't';
      *(p++) = 'l';
      *(p++) = 's';
      break;
    case PROTO_SMPP:
      *(p++) = 's';
      *(p++) = 'm';
      *(p++) = 'p';
      *(p++) = 'p';
      break;
    case PROTO_MSRP:
      *(p++) = 'm';
      *(p++) = 's';
      *(p++) = 'r';
      *(p++) = 'p';
      break;
    case PROTO_MSRPS:
      *(p++) = 'm';
      *(p++) = 's';
      *(p++) = 'r';
      *(p++) = 'p';
      *(p++) = 's';
      break;

    default:
      LM_CRIT("unsupported proto %d\n", proto);
  }

  return p;
}

/* Similar to proto2str(), but proto is written in uppercase. The
   new resulting pointer will be returned (where writing ended) */
static inline char* proto2upper(int proto, char *p)
{
  switch (proto) {
  case PROTO_UDP:
    p = memcpy(p, STR_L("UDP")) + sizeof("UDP")-1;
    break;
  case PROTO_TCP:
    p = memcpy(p, STR_L("TCP")) + sizeof("TCP")-1;
    break;
  case PROTO_TLS:
    p = memcpy(p, STR_L("TLS")) + sizeof("TLS")-1;
    break;
  case PROTO_SCTP:
    p = memcpy(p, STR_L("SCTP")) + sizeof("SCTP")-1;
    break;
  case PROTO_WS:
    p = memcpy(p, STR_L("WS")) + sizeof("WS")-1;
    break;
  case PROTO_WSS:
    p = memcpy(p, STR_L("WSS")) + sizeof("WSS")-1;
    break;
  case PROTO_BIN:
    p = memcpy(p, STR_L("BIN")) + sizeof("BIN")-1;
    break;
  case PROTO_BINS:
    p = memcpy(p, STR_L("BINS")) + sizeof("BINS")-1;
    break;
  case PROTO_HEP_UDP:
    p = memcpy(p, STR_L("HEP_UDP")) + sizeof("HEP_UDP")-1;
    break;
  case PROTO_HEP_TCP:
    p = memcpy(p, STR_L("HEP_TCP")) + sizeof("HEP_TCP")-1;
    break;
  case PROTO_SMPP:
    p = memcpy(p, STR_L("SMPP")) + sizeof("SMPP")-1;
    break;
  case PROTO_MSRP:
    p = memcpy(p, STR_L("MSRP")) + sizeof("MSRP")-1;
    break;
  case PROTO_MSRPS:
    p = memcpy(p, STR_L("MSRPS")) + sizeof("MSRPS")-1;
    break;
  default:
    LM_CRIT("unsupported proto %d\n", proto);
  }

  return p;
}

static inline char *proto2a(int proto)
{
  static char b[PROTO_NAME_MAX_SIZE+1];
  char *p;

  /* print the proto name */
  p = proto2str( proto, b);

  /* make it null terminated */
  *p = '\0';

  return  b;
}


#define MAX_SOCKET_STR ( 4 + 1 + IP_ADDR_MAX_STR_SIZE+1+INT2STR_MAX_LEN+1)
#define sock_str_len(_sock,_type) (3 + 1*((_sock)->proto==PROTO_SCTP) + 1 + \
    (((_type)==0) ? (_sock)->address_str.len + (_sock)->port_no_str.len + 1 : \
        (((_type)==1) ? (_sock)->adv_name_str.len + (_sock)->adv_port_str.len + 1 : \
            (_sock)->tag.len)))

/* builds the full name of the socket ( proto:name[:port] ), using different
   naming for it, depending on the "type" parameter :
      0 - real name
      1 - advertised name
      2 - tagged name
*/
static inline char* socket2str(const struct socket_info *sock, char *s, int *len, int type)
{
  static char buf[MAX_SOCKET_STR];
  char *p,*p1;

  if (s) {
    /* buffer provided -> check lenght */
    if ( sock_str_len(sock,type) > *len ) {
      LM_ERR("buffer too short\n");
      return 0;
    }
    p = p1 = s;
  } else {
    p = p1 = buf;
  }

  p = proto2str( sock->proto, p);
  if (p==NULL) return 0;

  *(p++) = ':';
  switch (type) {
  case 0:
    memcpy( p, sock->address_str.s, sock->address_str.len);
    p += sock->address_str.len;
    *(p++) = ':';
    memcpy( p, sock->port_no_str.s, sock->port_no_str.len);
    p += sock->port_no_str.len;
    break;
  case 1:
    memcpy( p, sock->adv_name_str.s, sock->adv_name_str.len);
    p += sock->adv_name_str.len;
    *(p++) = ':';
    memcpy( p, sock->adv_port_str.s, sock->adv_port_str.len);
    p += sock->adv_port_str.len;
    break;
  case 2:
    memcpy( p, sock->tag.s, sock->tag.len);
    p += sock->tag.len;
    break;
  default:
    LM_BUG("unsupported type %d in printing socket name <%.*s>\n",
      type, sock->name.len, sock->name.s);
  }
  *len = (int)(long)(p-p1);
  LM_DBG("<%.*s>\n",*len,p1);
  return p1;
}


#define get_sock_info_list(_proto) \
  (((_proto)>=PROTO_FIRST && (_proto)<PROTO_LAST)?(&protos[_proto].listeners):0)

int is_localhost(struct socket_info *si);
int fix_socket(struct socket_info_full *sif, int add_aliases);


int probe_max_sock_buff( int sock, int buff_choice, int buff_max,
    int buff_increment);

struct socket_id *socket_info2id(struct socket_info *si);
struct socket_info_full* new_sock_info( struct socket_id *sid);
void push_sock2list(struct socket_info_full *si);
void pop_sock2list(struct socket_info_full *si);
void free_sock_info(struct socket_info_full* si);
int update_default_socket_info(struct socket_info *si);
void remove_default_socket_info(struct socket_info *si);

#endif

Coverage Report

Created: 2025-07-18 06:32