sockstream.h

Go to the documentation of this file.

/*
  Copyright (c) Mircea Neacsu (2014-2025) Licensed under MIT License.
  This file is part of MLIB project. See LICENSE file for full license terms.
*/

 
#pragma once
 
#if __has_include("defs.h")
#include "defs.h"
#endif
 
#include "safe_winsock.h"
#include <iostream>
 
 
#include <assert.h>
#include "sockbuf.h"
 
namespace mlib {
 
// clang-format on
 

 
template <typename T>
class generic_sockstream : public std::basic_iostream<T>
{
public:
  generic_sockstream ()
    : std::basic_iostream<T> (new sockbuf<T> ()){};

  explicit generic_sockstream (const sockbuf<T>& sb)
    : std::basic_iostream<T> (new sockbuf<T> (sb)){};

  explicit generic_sockstream (const sock& s)
    : std::basic_iostream<T> (new sockbuf<T> (s)){};

  explicit generic_sockstream (sock::type t, int domain = AF_INET, int proto = 0)
    : std::basic_iostream<T> (new sockbuf<T> (t, domain, proto)){};

  explicit generic_sockstream (const inaddr& remote, sock::type t = sock::stream);
 
  ~generic_sockstream ();

  sockbuf<T>* rdbuf ()
  {
    return (sockbuf<T>*)std::basic_ios<T>::rdbuf ();
  }

  sockbuf<T>* operator->()
  {
    return rdbuf ();
  }
};
 
template <typename T>
generic_sockstream<T>::generic_sockstream (const inaddr& remote, sock::type t)
  : std::basic_iostream<T> (new sockbuf<T> ())
{
  rdbuf ()->open (t);
  rdbuf ()->connect (remote);
}
 
template <typename T>
generic_sockstream< T >::~generic_sockstream ()
{
  delete std::basic_ios<T>::rdbuf ();
}

typedef generic_sockstream<char> isockstream;

typedef generic_sockstream<char> osockstream;

typedef generic_sockstream<char> sockstream;
 
 
} // namespace mlib