| /* |
| * FreeRTOS Platform V1.1.1 |
| * Copyright (C) 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved. |
| * |
| * Permission is hereby granted, free of charge, to any person obtaining a copy of |
| * this software and associated documentation files (the "Software"), to deal in |
| * the Software without restriction, including without limitation the rights to |
| * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of |
| * the Software, and to permit persons to whom the Software is furnished to do so, |
| * subject to the following conditions: |
| * |
| * The above copyright notice and this permission notice shall be included in all |
| * copies or substantial portions of the Software. |
| * |
| * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
| * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS |
| * FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR |
| * COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER |
| * IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN |
| * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. |
| * |
| * http://aws.amazon.com/freertos |
| * http://www.FreeRTOS.org |
| */ |
| |
| /* |
| * The implementation is FreeRTOS specific. It depends on aws_secure_sockets library. |
| */ |
| |
| /* Define _SECURE_SOCKETS_WRAPPER_NOT_REDEFINE to prevent secure sockets functions |
| * from redefining in iot_secure_sockets_wrapper_metrics.h */ |
| #define _SECURE_SOCKETS_WRAPPER_NOT_REDEFINE |
| |
| /* The config header is always included first. */ |
| #include "iot_config.h" |
| |
| /* Standard includes. */ |
| #include <stdio.h> |
| #include <string.h> |
| |
| /* Metrics include. */ |
| #include "platform/iot_metrics.h" |
| |
| /* Platform threads include. */ |
| #include "platform/iot_threads.h" |
| |
| /* Secure sockets include. */ |
| #include "iot_secure_sockets.h" |
| |
| #undef _SECURE_SOCKETS_WRAPPER_NOT_REDEFINE |
| |
| #if AWS_IOT_SECURE_SOCKETS_METRICS_ENABLED == 1 |
| |
| /** |
| * @brief Used to match metrics connection records by network connection. |
| * |
| * @param[in] pConnectionLink Pointer to a metrics connection record's link element. |
| * @param[in] pContext The network connection to match. |
| * |
| * @return `true` if the given metrics connection record matches the given |
| * network connection; `false` otherwise. |
| */ |
| static bool _connectionMatch( const IotLink_t * pConnectionLink, |
| void * pContext ); |
| |
| static void _metricsAddTcpConnection( Socket_t xSocket, |
| SocketsSockaddr_t * pxAddress ); |
| |
| /*------------------- Global Variables ------------------------*/ |
| |
| /** |
| * @brief Holds a list of active TCP connections. |
| */ |
| static IotListDouble_t _connectionList = IOT_LIST_DOUBLE_INITIALIZER; |
| |
| /** |
| * @brief Protects #_connectionList from concurrent access. |
| */ |
| static IotMutex_t _connectionListMutex; |
| |
| /*-----------------------------------------------------------*/ |
| |
| static bool _connectionMatch( const IotLink_t * pConnectionLink, |
| void * pContext ) |
| { |
| /* Retrieve the pointer the the TCP connection record. The given link |
| * pointer is never NULL. */ |
| IotMetricsTcpConnection_t * pTcpConnection = IotLink_Container( IotMetricsTcpConnection_t, |
| pConnectionLink, |
| link ); |
| |
| return( pTcpConnection->pNetworkContext == pContext ); |
| } |
| |
| /*-----------------------------------------------------------*/ |
| |
| bool IotMetrics_Init( void ) |
| { |
| IotListDouble_Create( &_connectionList ); |
| |
| return IotMutex_Create( &_connectionListMutex, false ); |
| } |
| |
| /*-----------------------------------------------------------*/ |
| |
| void IotMetrics_Cleanup( void ) |
| { |
| IotMutex_Destroy( &_connectionListMutex ); |
| } |
| |
| /*-----------------------------------------------------------*/ |
| |
| void IotMetrics_GetTcpConnections( void * pContext, |
| void ( * metricsCallback )( void *, const IotListDouble_t * ) ) |
| { |
| /* Provide the connection list. Ensure that it is not modified elsewhere by |
| * locking the connection list mutex. */ |
| IotMutex_Lock( &_connectionListMutex ); |
| metricsCallback( pContext, &_connectionList ); |
| IotMutex_Unlock( &_connectionListMutex ); |
| } |
| |
| /*-----------------------------------------------------------*/ |
| |
| static void _metricsAddTcpConnection( Socket_t xSocket, |
| SocketsSockaddr_t * pxAddress ) |
| { |
| IotMetricsTcpConnection_t * pTcpConnection = NULL; |
| void * pSocketContext = ( void * ) xSocket; |
| |
| IotMutex_Lock( &_connectionListMutex ); |
| |
| /* Only add if it doesn't exist in the _connectionList. */ |
| if( IotListDouble_FindFirstMatch( &_connectionList, |
| NULL, |
| _connectionMatch, |
| pSocketContext ) == NULL ) |
| { |
| /* Allocate memory for a new metrics connection. */ |
| pTcpConnection = pvPortMalloc( sizeof( IotMetricsTcpConnection_t ) ); |
| |
| if( pTcpConnection != NULL ) |
| { |
| ( void ) memset( pTcpConnection, 0x00, sizeof( IotMetricsTcpConnection_t ) ); |
| |
| pTcpConnection->pNetworkContext = pSocketContext; |
| |
| /* Convert IP to string. */ |
| SOCKETS_inet_ntoa( pxAddress->ulAddress, pTcpConnection->pRemoteAddress ); |
| |
| /* Formatting. */ |
| sprintf( pTcpConnection->pRemoteAddress + strlen( pTcpConnection->pRemoteAddress ), ":%d", SOCKETS_ntohs( pxAddress->usPort ) ); |
| |
| pTcpConnection->addressLength = strlen( pTcpConnection->pRemoteAddress ); |
| |
| /* Insert to the list. */ |
| IotListDouble_InsertTail( &_connectionList, &( pTcpConnection->link ) ); |
| } |
| } |
| |
| IotMutex_Unlock( &_connectionListMutex ); |
| } |
| |
| /*-----------------------------------------------------------*/ |
| |
| static void _metricsRemoveTcpConnection( Socket_t xSocket ) |
| { |
| IotMutex_Lock( &_connectionListMutex ); |
| |
| IotLink_t * pFoundConnectionLink = IotListDouble_RemoveFirstMatch( &_connectionList, |
| NULL, |
| _connectionMatch, |
| ( void * ) xSocket ); |
| |
| if( pFoundConnectionLink != NULL ) |
| { |
| IotMetricsTcpConnection_t * pFoundTcpConnection = IotLink_Container( IotMetricsTcpConnection_t, pFoundConnectionLink, link ); |
| |
| IotMetrics_FreeTcpConnection( pFoundTcpConnection ); |
| } |
| |
| IotMutex_Unlock( &_connectionListMutex ); |
| } |
| |
| /*-----------------------------------------------------------*/ |
| |
| BaseType_t Sockets_MetricsInit( void ) |
| { |
| BaseType_t result = SOCKETS_Init(); |
| |
| if( result == pdTRUE ) |
| { |
| result = IotMetrics_Init() ? pdTRUE : pdFALSE; |
| } |
| |
| return result; |
| } |
| |
| /*-----------------------------------------------------------*/ |
| |
| int32_t Sockets_MetricsConnect( Socket_t xSocket, |
| SocketsSockaddr_t * pxAddress, |
| Socklen_t xAddressLength ) |
| { |
| int32_t result = SOCKETS_Connect( xSocket, pxAddress, xAddressLength ); |
| |
| if( result == SOCKETS_ERROR_NONE ) |
| { |
| _metricsAddTcpConnection( xSocket, pxAddress ); |
| } |
| |
| return result; |
| } |
| |
| /*-----------------------------------------------------------*/ |
| |
| int32_t Sockets_MetricsShutdown( Socket_t xSocket, |
| uint32_t ulHow ) |
| { |
| int32_t result = SOCKETS_Shutdown( xSocket, ulHow ); |
| |
| if( result == SOCKETS_ERROR_NONE ) |
| { |
| _metricsRemoveTcpConnection( xSocket ); |
| } |
| |
| return result; |
| } |
| |
| #endif /* ifdef AWS_IOT_SECURE_SOCKETS_METRICS_ENABLED */ |