diff options
Diffstat (limited to 'common/src/nx_udp_socket_bind.c')
| -rw-r--r-- | common/src/nx_udp_socket_bind.c | 309 |
1 files changed, 309 insertions, 0 deletions
diff --git a/common/src/nx_udp_socket_bind.c b/common/src/nx_udp_socket_bind.c new file mode 100644 index 00000000..8d4713e8 --- /dev/null +++ b/common/src/nx_udp_socket_bind.c @@ -0,0 +1,309 @@ +/**************************************************************************/ +/* */ +/* Copyright (c) Microsoft Corporation. All rights reserved. */ +/* */ +/* This software is licensed under the Microsoft Software License */ +/* Terms for Microsoft Azure RTOS. Full text of the license can be */ +/* found in the LICENSE file at https://aka.ms/AzureRTOS_EULA */ +/* and in the root directory of this software. */ +/* */ +/**************************************************************************/ + + +/**************************************************************************/ +/**************************************************************************/ +/** */ +/** NetX Component */ +/** */ +/** User Datagram Protocol (UDP) */ +/** */ +/**************************************************************************/ +/**************************************************************************/ + +#define NX_SOURCE_CODE + + +/* Include necessary system files. */ + +#include "nx_api.h" +#include "nx_udp.h" +#include "tx_thread.h" + + +/**************************************************************************/ +/* */ +/* FUNCTION RELEASE */ +/* */ +/* _nx_udp_socket_bind PORTABLE C */ +/* 6.0 */ +/* AUTHOR */ +/* */ +/* Yuxin Zhou, Microsoft Corporation */ +/* */ +/* DESCRIPTION */ +/* */ +/* This function binds the UDP socket a specific UDP port. If the */ +/* requested port number is currently bound to another socket, */ +/* this function waits for specified period of time (wait_option) */ +/* for the other socket to unbind the port number. */ +/* */ +/* INPUT */ +/* */ +/* socket_ptr Pointer to UDP socket */ +/* port 16-bit UDP port number */ +/* wait_option Suspension option */ +/* */ +/* OUTPUT */ +/* */ +/* status Completion status */ +/* */ +/* CALLS */ +/* */ +/* _nx_udp_free_port_find Find a free UDP port */ +/* tx_mutex_get Obtain protection mutex */ +/* tx_mutex_put Release protection mutex */ +/* _tx_thread_system_suspend Suspend thread */ +/* */ +/* CALLED BY */ +/* */ +/* Application Code */ +/* */ +/* RELEASE HISTORY */ +/* */ +/* DATE NAME DESCRIPTION */ +/* */ +/* 05-19-2020 Yuxin Zhou Initial Version 6.0 */ +/* */ +/**************************************************************************/ +UINT _nx_udp_socket_bind(NX_UDP_SOCKET *socket_ptr, UINT port, ULONG wait_option) +{ + +TX_INTERRUPT_SAVE_AREA +UINT index; +#ifdef NX_NAT_ENABLE +UINT bound; +#endif /* NX_NAT_ENABLE */ +NX_IP *ip_ptr; +TX_THREAD *thread_ptr; +NX_UDP_SOCKET *search_ptr; +NX_UDP_SOCKET *end_ptr; + + + /* Setup the pointer to the associated IP instance. */ + ip_ptr = socket_ptr -> nx_udp_socket_ip_ptr; + + /* If trace is enabled, insert this event into the trace buffer. */ + NX_TRACE_IN_LINE_INSERT(NX_TRACE_UDP_SOCKET_BIND, ip_ptr, socket_ptr, port, wait_option, NX_TRACE_UDP_EVENTS, 0, 0); + + /* Obtain the IP mutex so we can figure out whether or not the port has already + been bound to. */ + tx_mutex_get(&(ip_ptr -> nx_ip_protection), TX_WAIT_FOREVER); + + /* Determine if the socket has already been bound to port or if a socket bind is + already pending from another thread. */ + if ((socket_ptr -> nx_udp_socket_bound_next) || + (socket_ptr -> nx_udp_socket_bind_in_progress)) + { + + /* Release the protection mutex. */ + tx_mutex_put(&(ip_ptr -> nx_ip_protection)); + + /* Return an already bound error code. */ + return(NX_ALREADY_BOUND); + } + + /* Determine if the port needs to be allocated. */ + if (port == NX_ANY_PORT) + { + + /* Call the find routine to allocate a UDP port. */ + port = NX_SEARCH_PORT_START + (UINT)(NX_RAND() % ((NX_MAX_PORT + 1) - NX_SEARCH_PORT_START)); + if (_nx_udp_free_port_find(ip_ptr, port, &port) != NX_SUCCESS) + { + + /* Release the protection mutex. */ + tx_mutex_put(&(ip_ptr -> nx_ip_protection)); + + /* There was no free port, return an error code. */ + return(NX_NO_FREE_PORTS); + } + } +#ifdef NX_NAT_ENABLE + else + { + + /* Check if this IP interface has a NAT service. */ + if (ip_ptr -> nx_ip_nat_port_verify) + { + + /* Yes, so check the port by NAT handler. */ + bound = (ip_ptr -> nx_ip_nat_port_verify)(ip_ptr, NX_PROTOCOL_UDP, port); + + /* Check to see if the port has been bound by NAT. */ + if (bound == NX_TRUE) + { + + /* Release the protection. */ + tx_mutex_put(&(ip_ptr -> nx_ip_protection)); + + /* Return the port unavailable error. */ + return(NX_PORT_UNAVAILABLE); + } + } + } +#endif + + /* Save the port number in the UDP socket structure. */ + socket_ptr -> nx_udp_socket_port = port; + + /* Calculate the hash index in the UDP port array of the associated IP instance. */ + index = (UINT)((port + (port >> 8)) & NX_UDP_PORT_TABLE_MASK); + + /* Pickup the head of the UDP ports bound list. */ + search_ptr = ip_ptr -> nx_ip_udp_port_table[index]; + + /* Determine if we need to perform a list search. */ + if (search_ptr) + { + + /* Walk through the circular list of UDP sockets that are already + bound. */ + end_ptr = search_ptr; + do + { + + /* Determine if this entry is the same as the requested port. */ + if (search_ptr -> nx_udp_socket_port == port) + { + + /* Yes, the port has already been allocated. */ + break; + } + + /* Move to the next entry in the list. */ + search_ptr = search_ptr -> nx_udp_socket_bound_next; + } while (search_ptr != end_ptr); + } + + /* Now determine if the port is available. */ + if ((search_ptr == NX_NULL) || (search_ptr -> nx_udp_socket_port != port)) + { + + /* Place this UDP socket structure on the list of bound ports. */ + + /* Disable interrupts. */ + TX_DISABLE + + /* Determine if the list is NULL. */ + if (search_ptr) + { + + /* There are already sockets on this list... just add this one + to the end. */ + socket_ptr -> nx_udp_socket_bound_next = ip_ptr -> nx_ip_udp_port_table[index]; + socket_ptr -> nx_udp_socket_bound_previous = (ip_ptr -> nx_ip_udp_port_table[index]) -> nx_udp_socket_bound_previous; + ((ip_ptr -> nx_ip_udp_port_table[index]) -> nx_udp_socket_bound_previous) -> nx_udp_socket_bound_next = socket_ptr; + (ip_ptr -> nx_ip_udp_port_table[index]) -> nx_udp_socket_bound_previous = socket_ptr; + } + else + { + + /* Nothing is on the UDP port list. Add this UDP socket to an + empty list. */ + socket_ptr -> nx_udp_socket_bound_next = socket_ptr; + socket_ptr -> nx_udp_socket_bound_previous = socket_ptr; + ip_ptr -> nx_ip_udp_port_table[index] = socket_ptr; + } + + /* Restore interrupts. */ + TX_RESTORE + + /* Release the mutex protection. */ + tx_mutex_put(&(ip_ptr -> nx_ip_protection)); + + /* Return success to the caller. */ + return(NX_SUCCESS); + } + else if (wait_option) + { + + /* Prepare for suspension of this thread. */ + + /* Disable interrupts. */ + TX_DISABLE + + /* Pickup thread pointer. */ + thread_ptr = _tx_thread_current_ptr; + + /* Setup cleanup routine pointer. */ + thread_ptr -> tx_thread_suspend_cleanup = _nx_udp_bind_cleanup; + + /* Setup cleanup information, i.e. this socket control + block. */ + thread_ptr -> tx_thread_suspend_control_block = (void *)socket_ptr; + + /* Also remember the socket that has bound to the port, since the thread + is going to be suspended on that socket. */ + socket_ptr -> nx_udp_socket_bound_previous = search_ptr; + + /* Set the socket bind in progress flag (thread pointer). */ + socket_ptr -> nx_udp_socket_bind_in_progress = thread_ptr; + + /* Setup suspension list. */ + if (search_ptr -> nx_udp_socket_bind_suspension_list) + { + + /* This list is not NULL, add current thread to the end. */ + thread_ptr -> tx_thread_suspended_next = search_ptr -> nx_udp_socket_bind_suspension_list; + thread_ptr -> tx_thread_suspended_previous = (search_ptr -> nx_udp_socket_bind_suspension_list) -> tx_thread_suspended_previous; + ((search_ptr -> nx_udp_socket_bind_suspension_list) -> tx_thread_suspended_previous) -> tx_thread_suspended_next = thread_ptr; + (search_ptr -> nx_udp_socket_bind_suspension_list) -> tx_thread_suspended_previous = thread_ptr; + } + else + { + + /* No other threads are suspended. Setup the head pointer and + just setup this threads pointers to itself. */ + search_ptr -> nx_udp_socket_bind_suspension_list = thread_ptr; + thread_ptr -> tx_thread_suspended_next = thread_ptr; + thread_ptr -> tx_thread_suspended_previous = thread_ptr; + } + + /* Increment the suspended thread count. */ + search_ptr -> nx_udp_socket_bind_suspended_count++; + + /* Set the state to suspended. */ + thread_ptr -> tx_thread_state = TX_TCP_IP; + + /* Set the suspending flag. */ + thread_ptr -> tx_thread_suspending = TX_TRUE; + + /* Temporarily disable preemption. */ + _tx_thread_preempt_disable++; + + /* Save the timeout value. */ + thread_ptr -> tx_thread_timer.tx_timer_internal_remaining_ticks = wait_option; + + /* Restore interrupts. */ + TX_RESTORE + + /* Release the mutex protection. */ + tx_mutex_put(&(ip_ptr -> nx_ip_protection)); + + /* Call actual thread suspension routine. */ + _tx_thread_system_suspend(thread_ptr); + + /* Return the completion status. */ + return(thread_ptr -> tx_thread_suspend_status); + } + else + { + + /* Release the IP protection. */ + tx_mutex_put(&(ip_ptr -> nx_ip_protection)); + + /* Return the port unavailable error. */ + return(NX_PORT_UNAVAILABLE); + } +} + |
