src/lib/dcpomatic_socket.h

   1 /*
   2     Copyright (C) 2012-2015 Carl Hetherington <cth@carlh.net>
   3
   4     This program is free software; you can redistribute it and/or modify
   5     it under the terms of the GNU General Public License as published by
   6     the Free Software Foundation; either version 2 of the License, or
   7     (at your option) any later version.
   8
   9     This program is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12     GNU General Public License for more details.
  13
  14     You should have received a copy of the GNU General Public License
  15     along with this program; if not, write to the Free Software
  16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17
  18 */
  19
  20 #include <boost/asio.hpp>
  21
  22 /** @class Socket
  23  *  @brief A class to wrap a boost::asio::ip::tcp::socket with some things
  24  *  that are useful for DCP-o-matic.
  25  *
  26  *  This class wraps some things that I could not work out how to do easily with boost;
  27  *  most notably, sync read/write calls with timeouts.
  28  */
  29 class Socket : public boost::noncopyable
  30 {
  31 public:
  32         Socket (int timeout = 30);
  33
  34         /** @return Our underlying socket */
  35         boost::asio::ip::tcp::socket& socket () {
  36                 return _socket;
  37         }
  38
  39         void connect (boost::asio::ip::tcp::endpoint);
  40
  41         void write (uint32_t n);
  42         void write (uint8_t const * data, int size);
  43
  44         void read (uint8_t* data, int size);
  45         uint32_t read_uint32 ();
  46
  47 private:
  48         void check ();
  49
  50         Socket (Socket const &);
  51
  52         boost::asio::io_service _io_service;
  53         boost::asio::deadline_timer _deadline;
  54         boost::asio::ip::tcp::socket _socket;
  55         int _timeout;
  56 };