/* SPDX-License-Identifier: Apache-2.0
*
* The OpenSearch Contributors require contributions made to
* this file be licensed under the Apache-2.0 license or a
* compatible open source license.
*/
/*
* Modifications Copyright OpenSearch Contributors. See
* GitHub history for details.
*
* Licensed to Elasticsearch B.V. under one or more contributor
* license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright
* ownership. Elasticsearch B.V. licenses this file to you under
* the Apache License, Version 2.0 (the "License"); you may
* not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
using System;
using System.Collections.Generic;
using System.Collections.ObjectModel;
using System.Net.NetworkInformation;
namespace OpenSearch.Net.Diagnostics
{
///
/// Gets statistics about TCP connections
///
public static class TcpStats
{
private static readonly int StateLength = Enum.GetNames(typeof(TcpState)).Length;
private static readonly ReadOnlyDictionary Empty
= new ReadOnlyDictionary(new Dictionary());
///
/// Gets the active TCP connections
///
/// TcpConnectionInformation[]
/// Can return `null` when there is a permissions issue retrieving TCP connections.
public static TcpConnectionInformation[] GetActiveTcpConnections()
{
try
{
return IPGlobalProperties.GetIPGlobalProperties().GetActiveTcpConnections();
}
catch (NetworkInformationException) // host might not allow this information to be fetched.
{
// ignored
}
return null;
}
///
/// Gets the sum for each state of the active TCP connections
///
public static ReadOnlyDictionary GetStates()
{
var connections = GetActiveTcpConnections();
if (connections == null)
{
return Empty;
}
var states = new Dictionary(StateLength);
for (var index = 0; index < connections.Length; index++)
{
var connection = connections[index];
if (states.TryGetValue(connection.State, out var count))
states[connection.State] = ++count;
else
states.Add(connection.State, 1);
}
return new ReadOnlyDictionary(states);
}
///
/// Gets the TCP statistics for a given network interface component
///
public static TcpStatistics GetTcpStatistics(NetworkInterfaceComponent version)
{
var properties = IPGlobalProperties.GetIPGlobalProperties();
switch (version)
{
case NetworkInterfaceComponent.IPv4:
return properties.GetTcpIPv4Statistics();
case NetworkInterfaceComponent.IPv6:
return properties.GetTcpIPv6Statistics();
default:
throw new ArgumentException("version");
}
}
}
}