socket_guard.h

Go to the documentation of this file.

/**********************************************************************
 
  Audacity: A Digital Audio Editor
 
  @file socket_guard.h
 
  @author Vitaly Sverchinsky
 
  Part of lib-ipc library
 
**********************************************************************/
 
#pragma once
 
#include <cassert>
#include "ipc-types.h"
 
class socket_guard
{
   SOCKET mSocket{INVALID_SOCKET};
public:
   explicit socket_guard(const socket_guard&) = delete;
   socket_guard& operator=(const socket_guard&) = delete;
 
   explicit socket_guard(SOCKET sock = INVALID_SOCKET) noexcept : mSocket(sock) { }
 
   explicit socket_guard(socket_guard&& rhs) noexcept
      : mSocket(rhs.mSocket)
   {
      rhs.mSocket = INVALID_SOCKET;
   }
 
   explicit operator bool() const noexcept
   {
      return mSocket != INVALID_SOCKET;
   }
 
   socket_guard& operator=(socket_guard&& rhs) noexcept
   {
      if(this == &rhs)
         return * this;
 
      assert(mSocket == INVALID_SOCKET || rhs.mSocket != mSocket);
 
      std::swap(mSocket, rhs.mSocket);
      rhs.reset();
 
      return *this;
   }
 
   SOCKET operator*() const noexcept { return mSocket; }
 
   SOCKET get() const noexcept { return mSocket; }
 
   SOCKET release() noexcept
   {
      SOCKET sock {INVALID_SOCKET};
      std::swap(sock, mSocket);
      return sock;
   }
 
   void reset() noexcept
   {
      if(mSocket != INVALID_SOCKET)
      {
         CLOSE_SOCKET(mSocket);
         mSocket = INVALID_SOCKET;
      }
   }
 
   ~socket_guard() noexcept
   {
      reset();
   }
};