IOHandlerData.h

Go to the documentation of this file.

/*
 * Copyright (C) 2007-2015 Hypertable, Inc.
 *
 * This file is part of Hypertable.
 *
 * Hypertable is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 3
 * of the License, or any later version.
 *
 * Hypertable is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
 * 02110-1301, USA.
 */

#ifndef AsyncComm_IOHandlerData_h
#define AsyncComm_IOHandlerData_h

#include "CommBuf.h"
#include "IOHandler.h"

#include <Common/Error.h>

#include <list>

extern "C" {
#include <netdb.h>
#include <string.h>
}

namespace Hypertable {

  class IOHandlerData : public IOHandler {

  public:

    IOHandlerData(int sd, const InetAddr &addr,
                  const DispatchHandlerPtr &dhp, bool connected=false)
      : IOHandler(sd, dhp) {
      memcpy(&m_addr, &addr, sizeof(InetAddr));
      m_connected = connected;
      reset_incoming_message_state();
    }

    virtual ~IOHandlerData() { }

    void disconnect() override {
      EventPtr event = std::make_shared<Event>(Event::DISCONNECT, m_addr, m_proxy, m_error);
      deliver_event(event);
    }

    void reset_incoming_message_state() {
      m_got_header = false;
      m_event.reset();
      m_message_header_ptr = m_message_header;
      m_message_header_remaining = CommHeader::FIXED_LENGTH;
      m_message = 0;
      m_message_ptr = 0;
      m_message_remaining = 0;
      m_message_aligned = false;
    }

    void free_message_buffer() {
      if (m_message_aligned)
        free(m_message);
      else
        delete [] m_message;
      m_message = 0;
    }

    int send_message(CommBufPtr &cbp, uint32_t timeout_ms=0,
                     DispatchHandler *disp_handler=nullptr);

    int flush_send_queue();

    bool handle_event(struct pollfd *event,
                      ClockT::time_point arrival_time) override;

#if defined(__APPLE__) || defined(__FreeBSD__)

    bool handle_event(struct kevent *event,
                      ClockT::time_point arrival_time) override;
#elif defined(__linux__)

    bool handle_event(struct epoll_event *event,
                      ClockT::time_point arrival_time) override;
#elif defined(__sun__)

    bool handle_event(port_event_t *event,
                      ClockT::time_point arrival_time) override;
#else
    ImplementMe;
#endif

    bool handle_write_readiness();

  private:

    void handle_message_header(ClockT::time_point arrival_time);

    void handle_message_body();

    void handle_disconnect();

    bool m_connected {};

    bool m_got_header {};

    bool m_message_aligned {};

    EventPtr m_event;

    uint8_t m_message_header[64];

    uint8_t *m_message_header_ptr {};

    size_t m_message_header_remaining;

    uint8_t *m_message {};

    uint8_t *m_message_ptr {};

    size_t m_message_remaining {};

    std::list<CommBufPtr> m_send_queue;
  };
}

#endif // AsyncComm_IOHandlerData_h