allwpilib/src/main/native/cpp/support/raw_socket_ostream.cpp

/*----------------------------------------------------------------------------*/
/* Copyright (c) FIRST 2015. All Rights Reserved.                             */
/* Open Source Software - may be modified and shared by FRC teams. The code   */
/* must be accompanied by the FIRST BSD license file in the root directory of */
/* the project.                                                               */
/*----------------------------------------------------------------------------*/

#include "support/raw_socket_ostream.h"
#include "tcpsockets/NetworkStream.h"

using namespace wpi;

raw_socket_ostream::~raw_socket_ostream() {
  flush();
  if (m_shouldClose) close();
}

void raw_socket_ostream::write_impl(const char* data, std::size_t len) {
  std::size_t pos = 0;

  while (pos < len) {
    NetworkStream::Error err;
    std::size_t count =
        m_stream.send(&data[pos], len - pos, &err);
    if (count == 0) {
      error_detected();
      return;
    }
    pos += count;
  }
}

uint64_t raw_socket_ostream::current_pos() const { return 0; }

void raw_socket_ostream::close() {
  if (!m_shouldClose) return;
  flush();
  m_stream.close();
}