static void DNS_init_unwind ( DNS_CONTROL_STRUCT *control_ptr, uint32_t sock ) { /* Body */ RTCS_setcfg(DNS, NULL); if (sock) shutdown(sock, 0); if (control_ptr) { if (control_ptr->CNAME_PARTID) RTCS_part_destroy(control_ptr->CNAME_PARTID); if (control_ptr->RR_LIST_PARTID) RTCS_part_destroy(control_ptr->RR_LIST_PARTID); if (control_ptr->NAME_RR_PARTID) RTCS_part_destroy(control_ptr->NAME_RR_PARTID); if (control_ptr->INT_QUERY_PARTID) RTCS_part_destroy(control_ptr->INT_QUERY_PARTID); if (control_ptr->ADDR_RR_PARTID) RTCS_part_destroy(control_ptr->ADDR_RR_PARTID); _mem_free(control_ptr); } return; } /* Endbody */
static uint_32 NAT_init_internal2 ( NAT_PARM_PTR parm_ptr /* [IN] Initialization parameters */ ) { /* Body */ NAT_CFG_STRUCT_PTR nat_cfg_ptr = RTCS_getcfg(NAT); uint_32 error; /* ** Make sure the netmask is valid. We use the fact that ** (x & x+1) == 0 <=> x = 2^n-1. */ if (~parm_ptr->IP_MASK & (~parm_ptr->IP_MASK + 1)) { return RTCSERR_INVALID_PARAMETER; } /* Endif */ /* Make sure other parameters have been supplied */ if (!parm_ptr->IP_PRV) { return RTCSERR_INVALID_PARAMETER; } /* Endif */ /* ** If the configuration structure has already been allocated, ** just add another private network and return */ if (nat_cfg_ptr) { error = NAT_add_private_network(&nat_cfg_ptr->PRIVATE_NETWORKS, parm_ptr->IP_PRV, parm_ptr->IP_MASK); if (error == RTCS_OK) { nat_cfg_ptr->NAT_EXEC = NAT_apply; } return error; } /* Endif */ // First time through the initialization, // Allocate the NAT configuration structure nat_cfg_ptr = _mem_alloc_system_zero(sizeof(NAT_CFG_STRUCT)); if (nat_cfg_ptr == NULL) { return RTCSERR_OUT_OF_MEMORY; } /* Endif */ /* Record the private network information */ error = NAT_add_private_network(&nat_cfg_ptr->PRIVATE_NETWORKS, parm_ptr->IP_PRV, parm_ptr->IP_MASK); if (error == RTCS_OK) { nat_cfg_ptr->NAT_EXEC = NAT_apply; nat_cfg_ptr->INITIAL_PRV_NET = parm_ptr->IP_PRV; nat_cfg_ptr->INITIAL_PRV_MASK = parm_ptr->IP_MASK; nat_cfg_ptr->TCP_TOUT.PRIVATE = &nat_cfg_ptr->TCP_HEAD; nat_cfg_ptr->UDP_TOUT.PRIVATE = &nat_cfg_ptr->UDP_HEAD; nat_cfg_ptr->FIN_TOUT.PRIVATE = &nat_cfg_ptr->FIN_HEAD; nat_cfg_ptr->ICMP_TOUT.PRIVATE = &nat_cfg_ptr->ICMP_HEAD; /* Initialize radix trees */ IPRADIX_init(&nat_cfg_ptr->ROOT_IN); IPRADIX_init(&nat_cfg_ptr->ROOT_OUT); /* Initialize timeout functions */ nat_cfg_ptr->UDP_TOUT.EVENT = NAT_expire; nat_cfg_ptr->TCP_TOUT.EVENT = NAT_expire; nat_cfg_ptr->FIN_TOUT.EVENT = NAT_expire; nat_cfg_ptr->ICMP_TOUT.EVENT = NAT_expire; nat_cfg_ptr->TIMEOUT_TCP = NAT_DEFAULT_TIMEOUT_TCP; nat_cfg_ptr->TIMEOUT_UDP = NAT_DEFAULT_TIMEOUT_UDP; nat_cfg_ptr->TIMEOUT_FIN = NAT_DEFAULT_TIMEOUT_FIN; nat_cfg_ptr->TIMEOUT_ICMP = NAT_DEFAULT_TIMEOUT_ICMP; /* Initialize last used port numbers */ nat_cfg_ptr->TCP_PORT = NAT_DEFAULT_PORT_MIN; nat_cfg_ptr->UDP_PORT = NAT_DEFAULT_PORT_MIN; nat_cfg_ptr->ICMP_ID = NAT_DEFAULT_PORT_MIN; nat_cfg_ptr->PORT_MAX = NAT_DEFAULT_PORT_MAX; nat_cfg_ptr->PORT_MIN = NAT_DEFAULT_PORT_MIN; /* Inititalize ALGs */ error = NAT_init_algs(nat_cfg_ptr); } /* Allocate partitions */ if (error == RTCS_OK) { nat_cfg_ptr->SESSION_PART = RTCS_part_create(sizeof(NAT_SESSION_STRUCT), NAT_SESSION_INITIAL_COUNT, NAT_SESSION_GROW_COUNT, NAT_SESSION_MAX, NULL, NULL); if (nat_cfg_ptr->SESSION_PART == NULL) { error = RTCSERR_OUT_OF_MEMORY; } /* Endif */ } /* Endif */ if (error == RTCS_OK) { nat_cfg_ptr->RADIX_IN = RTCS_part_create(sizeof(IPRADIX_NODE), NAT_SESSION_INITIAL_COUNT, NAT_SESSION_GROW_COUNT, NAT_SESSION_MAX, NULL, NULL); if (nat_cfg_ptr->RADIX_IN == NULL) { error = RTCSERR_OUT_OF_MEMORY; } /* Endif */ } /* Endif */ if (error == RTCS_OK) { nat_cfg_ptr->RADIX_OUT = RTCS_part_create(sizeof(IPRADIX_NODE), NAT_SESSION_INITIAL_COUNT, NAT_SESSION_GROW_COUNT, NAT_SESSION_MAX, NULL, NULL); if (nat_cfg_ptr->RADIX_OUT == NULL) { error = RTCSERR_OUT_OF_MEMORY; } /* Endif */ } /* Endif */ if (error == RTCS_OK) { // Initialize queue of DNAT Rules _queue_init( &nat_cfg_ptr->RULE_QUEUE, 0 ); RTCS_setcfg(NAT, nat_cfg_ptr); } else { if (nat_cfg_ptr != NULL) { if (nat_cfg_ptr->RADIX_IN != NULL) RTCS_part_destroy(nat_cfg_ptr->RADIX_IN); if (nat_cfg_ptr->RADIX_OUT != NULL) RTCS_part_destroy(nat_cfg_ptr->RADIX_OUT); if (nat_cfg_ptr->SESSION_PART != NULL) RTCS_part_destroy(nat_cfg_ptr->SESSION_PART); _mem_free(nat_cfg_ptr); } } /* Endif */ return error; } /* Endbody */
uint_32 RTCS_create ( void ) { /* Body */ RTCS_DATA_PTR RTCS_data_ptr; SOCKET_CONFIG_STRUCT_PTR socket_cfg_ptr; uint_32 error; /* ** Check and see if this is the first time we have initialized, */ if (RTCS_get_data() != NULL) { return RTCSERR_INITIALIZED; } // RTCS_data_ptr = RTCS_mem_alloc_system_zero(sizeof(RTCS_DATA)); if (RTCS_data_ptr == NULL) { error = RTCSERR_OUT_OF_MEMORY; } else { _mem_set_type(RTCS_data_ptr, MEM_TYPE_RTCS_DATA); RTCS_set_data(RTCS_data_ptr); /* ** Initialize socket state */ socket_cfg_ptr = RTCS_mem_alloc_system_zero(sizeof(SOCKET_CONFIG_STRUCT)); if (socket_cfg_ptr == NULL) { error = RTCSERR_OUT_OF_MEMORY; } else { _mem_set_type(socket_cfg_ptr, MEM_TYPE_SOCKET_CONFIG_STRUCT); socket_cfg_ptr->INITIALIZED = TRUE; socket_cfg_ptr->SOCKET_HEAD = NULL; socket_cfg_ptr->SOCKET_TAIL = NULL; RTCS_mutex_init(&socket_cfg_ptr->SOCK_MUTEX); RTCS_setcfg(SOCKET, socket_cfg_ptr); /* ** Initialize global data */ _IP_forward = FALSE; _TCP_bypass_rx = FALSE; _TCP_bypass_tx = FALSE; RTCS_data_ptr->VERSION = RTCS_VERSION; #if RTCSCFG_LOG_SOCKET_API||RTCSCFG_LOG_PCB RTCS_data_ptr->RTCS_LOG_PROTCOUNT = RTCSLOG_PROT_MAX; RTCSLOG_disable(RTCS_LOGCTRL_ALL); #endif RTCS_data_ptr->TCPIP_qid = RTCS_msgq_get_id(0,TCPIP_QUEUE); /* ** Create a pool of buffers for use in communicating to the TCP/IP task. */ RTCS_data_ptr->TCPIP_msg_pool = RTCS_msgpool_create(sizeof(TCPIP_MESSAGE), _RTCS_msgpool_init, _RTCS_msgpool_grow, _RTCS_msgpool_max); if (RTCS_data_ptr->TCPIP_msg_pool == MSGPOOL_NULL_POOL_ID) { RTCS_log_error( ERROR_TCPIP, RTCSERR_CREATE_POOL_FAILED, 0, 0, 0); error = RTCSERR_CREATE_POOL_FAILED; } else { /* ** Create the socket partition */ RTCS_data_ptr->RTCS_socket_partition = RTCS_part_create(sizeof(SOCKET_STRUCT), _RTCS_socket_part_init, _RTCS_socket_part_grow, _RTCS_socket_part_max, NULL, NULL); if (RTCS_data_ptr->RTCS_socket_partition == 0) { RTCS_log_error(ERROR_RTCS, RTCSERR_CREATE_PARTITION_FAILED, 0, 0, 0); error = RTCSERR_CREATE_PARTITION_FAILED; } else { /* ** Create TCPIP task */ error = RTCS_task_create("TCP/IP", _RTCSTASK_priority, _RTCSTASK_stacksize, TCPIP_task, NULL); if (error) { RTCS_part_destroy(RTCS_data_ptr->RTCS_socket_partition); } } if (error) { RTCS_msgpool_destroy( RTCS_data_ptr->TCPIP_msg_pool ); } } if (error) { RTCS_setcfg(SOCKET, NULL); _mem_free(socket_cfg_ptr); } } if (error) { RTCS_set_data(NULL); _mem_free(RTCS_data_ptr); } } return error; } /* Endbody */