1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
// Copyright (C) 2022-2024 Internet Systems Consortium, Inc. ("ISC")
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#include <config.h>

#include <asiolink/asio_wrapper.h>
#include <tcp/tcp_connection_pool.h>
#include <util/multi_threading_mgr.h>

using namespace isc::asiolink;

namespace isc {
namespace tcp {

std::atomic<uint64_t>
TcpConnectionPool::started_counter_(0);

std::atomic<uint64_t>
TcpConnectionPool::stopped_counter_(0);

std::atomic<uint64_t>
TcpConnectionPool::rejected_counter_(0);

void
TcpConnectionPool::start(const TcpConnectionPtr& connection) {
    if (util::MultiThreadingMgr::instance().getMode()) {
        std::lock_guard<std::mutex> lk(mutex_);
        connections_.insert(connections_.end(), connection);
        started_counter_ += 1;
    } else {
        connections_.insert(connections_.end(), connection);
        started_counter_ += 1;
    }

    connection->asyncAccept();
}

void
TcpConnectionPool::stop(const TcpConnectionPtr& connection) {
    if (util::MultiThreadingMgr::instance().getMode()) {
        std::lock_guard<std::mutex> lk(mutex_);
        size_t before = connections_.size();
        connections_.remove(connection);
        size_t after = connections_.size();
        stopped_counter_ += before - after;
    } else {
        size_t before = connections_.size();
        connections_.remove(connection);
        size_t after = connections_.size();
        stopped_counter_ += before - after;
    }

    connection->close();
}

void
TcpConnectionPool::shutdown(const TcpConnectionPtr& connection) {
    if (util::MultiThreadingMgr::instance().getMode()) {
        std::lock_guard<std::mutex> lk(mutex_);
        size_t before = connections_.size();
        connections_.remove(connection);
        size_t after = connections_.size();
        stopped_counter_ += before - after;
    } else {
        size_t before = connections_.size();
        connections_.remove(connection);
        size_t after = connections_.size();
        stopped_counter_ += before - after;
    }

    connection->shutdown();
}

void
TcpConnectionPool::stopAll() {
    if (util::MultiThreadingMgr::instance().getMode()) {
        std::lock_guard<std::mutex> lk(mutex_);
        stopAllInternal();
    } else {
        stopAllInternal();
    }
}

void
TcpConnectionPool::stopAllInternal() {
    for (auto const& connection : connections_) {
        connection->close();
    }

    size_t cnt = connections_.size();
    connections_.clear();
    stopped_counter_ += cnt;
}

size_t
TcpConnectionPool::usedByRemoteIp(const IOAddress& remote_ip,
                                  size_t& total_connections) {
    if (util::MultiThreadingMgr::instance().getMode()) {
        std::lock_guard<std::mutex> lk(mutex_);
        return (usedByRemoteIpInternal(remote_ip, total_connections));
    } else {
        return (usedByRemoteIpInternal(remote_ip, total_connections));
    }
}

size_t
TcpConnectionPool::usedByRemoteIpInternal(const IOAddress& remote_ip,
                                          size_t& total_connections) {
    total_connections = connections_.size();
    size_t cnt = 0;
    for (auto const& conn : connections_) {
        auto const& ep = conn->getRemoteEndpoint();
        if ((ep != TcpConnection::NO_ENDPOINT()) &&
            (IOAddress(ep.address()) == remote_ip)) {
            ++cnt;
        }
    }
    return (cnt);
}

}
}