FreePortFinder.cxx

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
#include "Framework/FreePortFinder.h"
#include "Framework/Logger.h"
 
#include <sys/socket.h>
#include <netinet/in.h>
#include <unistd.h>
#include <cstring>
 
namespace o2
{
namespace framework
{
 
FreePortFinder::FreePortFinder(unsigned short initialPort, unsigned short finalPort, unsigned short step)
  : mInitialPort{initialPort},
    mFinalPort{finalPort},
    mStep{step},
    mSocket{socket(AF_INET, SOCK_STREAM, 0)}
{
}
 
bool FreePortFinder::testPort(int port) const
{
  struct sockaddr_in addr;
  memset(&addr, 0, sizeof(addr));
  addr.sin_family = AF_INET;
  addr.sin_addr.s_addr = INADDR_ANY;
  addr.sin_port = htons(port);
  return bind(mSocket, (struct sockaddr*)&addr, sizeof(addr)) == 0;
}
 
void FreePortFinder::scan()
{
  for (mPort = mInitialPort; mPort < mFinalPort; mPort += mStep) {
    if (!testPort(mPort)) {
      if (mVerbose) {
        LOG(warn) << "Port range [" << mPort << ", " << mPort + mStep
                  << "] already taken. Skipping";
      }
      continue;
    }
    if (mVerbose) {
      LOG(info) << "Using port range [" << mPort << ", " << mPort + mStep << "]";
    }
    break;
  }
}
 
void FreePortFinder::setVerbose(bool v)
{
  mVerbose = v;
}
 
FreePortFinder::~FreePortFinder()
{
  close(mSocket);
}
 
unsigned short FreePortFinder::port() const { return mPort + 1; }
unsigned short FreePortFinder::range() const { return mStep; }
 
} // namespace framework
} // namespace o2