#pragma once #include <library/cpp/testing/common/env.h> #include <util/generic/noncopyable.h> #include <util/generic/ptr.h> #include <util/generic/string.h> #include <util/network/sock.h> class TInet6StreamSocket; // set two options: SO_REUSEADDR and SO_REUSEPORT, both are required for // correct implementation of TPortManager because of different operating systems // incompatibility: singe SO_REUSEADDR is enough for Linux, but not enough for Darwin template <class TSocketType> void SetReuseAddressAndPort(const TSocketType& sock) { const int retAddr = SetSockOpt(sock, SOL_SOCKET, SO_REUSEADDR, 1); if (retAddr < 0) { ythrow yexception() << "can't set SO_REUSEADDR: " << LastSystemErrorText(-retAddr); } #ifdef SO_REUSEPORT const int retPort = SetSockOpt(sock, SOL_SOCKET, SO_REUSEPORT, 1); if (retPort < 0) { ythrow yexception() << "can't set SO_REUSEPORT: " << LastSystemErrorText(-retPort); } #endif } class TPortManager: public TNonCopyable { public: TPortManager(bool reservePortsForCurrentTest = true); ~TPortManager(); // Gets free TCP port ui16 GetPort(ui16 port = 0); // Gets free TCP port ui16 GetTcpPort(ui16 port = 0); // Gets free UDP port ui16 GetUdpPort(ui16 port = 0); // Gets one free port for use in both TCP and UDP protocols ui16 GetTcpAndUdpPort(ui16 port = 0); ui16 GetPortsRange(const ui16 startPort, const ui16 range); private: class TPortManagerImpl; THolder<TPortManagerImpl> Impl_; }; ui16 GetRandomPort();