File indexing completed on 2025-01-05 04:37:34

 /*
     SPDX-FileCopyrightText: 2009 Joris Guisson <joris.guisson@gmail.com>
 
     SPDX-License-Identifier: GPL-2.0-or-later
 */
 
 #ifndef UTP_UTPSERVER_P_H
 #define UTP_UTPSERVER_P_H
 
 #include "connection.h"
 #include "outputqueue.h"
 #include "pollpipe.h"
 #include "utpserver.h"
 #include "utpsocket.h"
 
 #include <QMap>
 #include <QRecursiveMutex>
 #include <QSocketNotifier>
 #include <QTimer>
 
 #include <ktorrent_export.h>
 #include <net/poll.h>
 #include <net/serversocket.h>
 #include <net/socket.h>
 #include <net/wakeuppipe.h>
 #include <util/ptrmap.h>
 
 namespace utp
 {
 class UTPServerThread;
 
 /**
     Utility class used by UTPServer to make sure that ServerInterface::newConnection is called
     from the main thread and not from UTP thread (which is dangerous).
 */
 class MainThreadCall : public QObject
 {
     Q_OBJECT
 public:
     MainThreadCall(UTPServer *server);
     ~MainThreadCall() override;
 
 public Q_SLOTS:
     /**
         Calls UTPServer::handlePendingConnections, this should be executed in
         the main thread.
     */
     void handlePendingConnections();
 
 private:
     UTPServer *server;
 };
 
 typedef bt::PtrMap<quint16, Connection>::iterator ConItr;
 
 struct PollPipePair {
     PollPipe::Ptr read_pipe;
     PollPipe::Ptr write_pipe;
 
     PollPipePair();
 
     bool testRead(ConItr b, ConItr e);
     bool testWrite(ConItr b, ConItr e);
 };
 
 typedef bt::PtrMap<net::Poll *, PollPipePair>::iterator PollPipePairItr;
 typedef QMap<quint16, Connection::Ptr>::iterator ConnectionMapItr;
 
 class UTPServer::Private : public net::ServerSocket::DataHandler
 {
 public:
     Private(UTPServer *p);
     ~Private() override;
 
     bool bind(const net::Address &addr);
     void syn(const PacketParser &parser, bt::Buffer::Ptr buffer, const net::Address &addr);
     void reset(const Header *hdr);
     void wakeUpPollPipes(Connection::Ptr conn, bool readable, bool writeable);
     Connection::Ptr find(quint16 conn_id);
     void stop();
     void dataReceived(bt::Buffer::Ptr buffer, const net::Address &addr) override;
     void readyToWrite(net::ServerSocket *sock) override;
 
 public:
     UTPServer *p;
     QList<net::ServerSocket::Ptr> sockets;
     bool running;
     QMap<quint16, Connection::Ptr> connections;
     UTPServerThread *utp_thread;
     QRecursiveMutex mutex;
     bt::PtrMap<net::Poll *, PollPipePair> poll_pipes;
     bool create_sockets;
     bt::Uint8 tos;
     OutputQueue output_queue;
     QList<mse::EncryptedPacketSocket::Ptr> pending;
     QMutex pending_mutex;
     MainThreadCall *mtc;
     QList<Connection::WPtr> last_accepted;
     QTimer *timer;
 };
 }
 
 #endif