dicom_server_v2.h

Go to the documentation of this file.

// BSD 3-Clause License
// Copyright (c) 2021-2025, 🍀☀🌕🌥 🌊
// See the LICENSE file in the project root for full license information.
 
#ifndef PACS_NETWORK_V2_DICOM_SERVER_V2_HPP
#define PACS_NETWORK_V2_DICOM_SERVER_V2_HPP
 
#include "kcenon/pacs/network/v2/dicom_association_handler.h"
#include "kcenon/pacs/network/server_config.h"
#include "kcenon/pacs/security/access_control_manager.h"
#include "kcenon/pacs/services/scp_service.h"
 
#include <atomic>
#include <chrono>
#include <condition_variable>
#include <functional>
#include <map>
#include <memory>
#include <mutex>
#include <string>
#include <string_view>
#include <unordered_map>
#include <vector>
 
// Forward declarations for public network_system interfaces
namespace kcenon::network::interfaces {
class i_protocol_server;
class i_session;
}  // namespace kcenon::network::interfaces
 
namespace kcenon::pacs::network::v2 {
 
// =============================================================================
// DICOM Server V2 Class
// =============================================================================
 
class dicom_server_v2 {
public:
    // =========================================================================
    // Type Aliases
    // =========================================================================
 
    using clock = std::chrono::steady_clock;
    using duration = std::chrono::milliseconds;
    using time_point = clock::time_point;
 
    using association_established_callback =
        std::function<void(const std::string& session_id,
                           const std::string& calling_ae,
                           const std::string& called_ae)>;
 
    using association_closed_callback =
        std::function<void(const std::string& session_id, bool graceful)>;
 
    using error_callback = std::function<void(const std::string& error)>;
 
    // =========================================================================
    // Construction / Destruction
    // =========================================================================
 
    explicit dicom_server_v2(const server_config& config);
 
    ~dicom_server_v2();
 
    // Non-copyable, non-movable (owns network resources)
    dicom_server_v2(const dicom_server_v2&) = delete;
    dicom_server_v2& operator=(const dicom_server_v2&) = delete;
    dicom_server_v2(dicom_server_v2&&) = delete;
    dicom_server_v2& operator=(dicom_server_v2&&) = delete;
 
    // =========================================================================
    // Service Registration
    // =========================================================================
 
    void register_service(services::scp_service_ptr service);
 
    [[nodiscard]] std::vector<std::string> supported_sop_classes() const;
 
    // =========================================================================
    // Lifecycle Management
    // =========================================================================
 
    [[nodiscard]] Result<std::monostate> start();
 
    void stop(duration timeout = std::chrono::seconds{30});
 
    void wait_for_shutdown();
 
    // =========================================================================
    // Status Queries
    // =========================================================================
 
    [[nodiscard]] bool is_running() const noexcept;
 
    [[nodiscard]] size_t active_associations() const noexcept;
 
    [[nodiscard]] server_statistics get_statistics() const;
 
    [[nodiscard]] const server_config& config() const noexcept;
 
    // =========================================================================
    // Callbacks
    // =========================================================================
 
    void on_association_established(association_established_callback callback);
 
    void on_association_closed(association_closed_callback callback);
 
    void on_error(error_callback callback);
 
    // =========================================================================
    // Security / Access Control
    // =========================================================================
 
    void set_access_control(std::shared_ptr<security::access_control_manager> acm);
 
    [[nodiscard]] std::shared_ptr<security::access_control_manager>
    get_access_control() const noexcept;
 
    void set_access_control_enabled(bool enabled);
 
    [[nodiscard]] bool is_access_control_enabled() const noexcept;
 
private:
    // =========================================================================
    // Network System Callbacks
    // =========================================================================
 
    void on_connection(std::shared_ptr<kcenon::network::interfaces::i_session> session);
 
    void on_disconnection(std::string_view session_id);
 
    void on_receive(std::string_view session_id,
                    const std::vector<uint8_t>& data);
 
    void on_network_error(std::string_view session_id,
                          std::error_code ec);
 
    // =========================================================================
    // Handler Management
    // =========================================================================
 
    void create_handler(std::shared_ptr<kcenon::network::interfaces::i_session> session);
 
    void remove_handler(const std::string& session_id);
 
    [[nodiscard]] std::shared_ptr<dicom_association_handler>
    find_handler(const std::string& session_id) const;
 
    void check_idle_timeouts();
 
    // =========================================================================
    // Internal Helpers
    // =========================================================================
 
    [[nodiscard]] dicom_association_handler::service_map build_service_map() const;
 
    void report_error(const std::string& error);
 
    // =========================================================================
    // Member Variables
    // =========================================================================
 
    server_config config_;
 
    std::shared_ptr<kcenon::network::interfaces::i_protocol_server> server_;
 
    std::vector<services::scp_service_ptr> services_;
 
    std::map<std::string, services::scp_service*> sop_class_to_service_;
 
    std::unordered_map<std::string, std::shared_ptr<dicom_association_handler>> handlers_;
 
    mutable server_statistics stats_;
 
    std::atomic<bool> running_{false};
 
    mutable std::mutex handlers_mutex_;
 
    mutable std::mutex services_mutex_;
 
    mutable std::mutex stats_mutex_;
 
    std::condition_variable shutdown_cv_;
 
    std::mutex shutdown_mutex_;
 
    association_established_callback on_established_cb_;
 
    association_closed_callback on_closed_cb_;
 
    error_callback on_error_cb_;
 
    mutable std::mutex callback_mutex_;
 
    std::shared_ptr<security::access_control_manager> access_control_;
 
    std::atomic<bool> access_control_enabled_{false};
 
    mutable std::mutex acl_mutex_;
};
 
}  // namespace kcenon::pacs::network::v2
 
#endif  // PACS_NETWORK_V2_DICOM_SERVER_V2_HPP