IOTestReadService.cpp

/*
Copyright 2007, 2008, 2009, 2010, 2011 Instituto de Sistemas e Robotica, Instituto Superior Tecnico

This file is part of MeRMaID.

MeRMaID is free software: you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

MeRMaID 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 Lesser General Public License for more details.

You should have received a copy of the GNU Lesser General Public License
along with MeRMaID.  If not, see <http://www.gnu.org/licenses/>.
*/



#include "config.h"

#include "IOTestReadService.hpp"
#include <iostream>
#include <cassert>
#include <IOManager.hpp>
#include <DeviceConfigurator.hpp>
#include <StreamOperationHandlerMethod.hpp>
#include <Exception.hpp>

using namespace components::ioservice;

using mermaid::support::io::IOManager;
using mermaid::support::io::DeviceConfigurator;
using mermaid::support::io::StreamOperationHandlerMethod;
using mermaid::support::io::StreamOperationHandlerMethodBase;
using mermaid::support::errorhandling::Exception;

#define PIPE_NAME "/tmp/test_read_pipe"
#define FILE_NAME  "test_file.txt"

IOTestReadService::IOTestReadService (shared_ptr<ActiveObject> ao, shared_ptr<Entity> entity, shared_ptr<ServiceInstanceDescription> serviceInstanceDescription, shared_ptr<ServiceConfiguration> serviceConfiguration) : Service (ao, entity, serviceInstanceDescription, serviceConfiguration),
    device_ (IOManager::getInstance()->createDevice (PIPE_NAME, getActiveObject(), shared_ptr<DeviceConfigurator> (new DeviceConfigurator), 100)), file_ (FILE_NAME, ifstream::in)
{
  std::cerr << "IOTestRead constructor" << std::endl;
  assert (file_.good());
  file_.seekg (0, ios::end);
  fsize_ = file_.tellg();
  file_.seekg (0, ios::beg);
  
};

void IOTestReadService::initialize()
{
  std::cerr << "IOTestRead initialized" << std::endl;
  device_->enableRead (shared_ptr<StreamOperationHandlerMethodBase> (new StreamOperationHandlerMethod<IOTestReadService> (this, &IOTestReadService::handler)));
};

void IOTestReadService::update()
{
};

void IOTestReadService::handler (size_t bytesRead, size_t requestedBytes, const char* data, bool success)
{
  if (bytesRead == 0)
    croak();
    
  char* buf = new char[bytesRead];
  assert (buf);
  
  file_.read (buf, bytesRead);
  fsize_ -= bytesRead;
  
  buf[bytesRead - 1] = 0;
  if (file_.gcount() != (int) bytesRead || !std::equal (buf, buf + bytesRead - 1, data)) {
    //std::cerr << data << " " << file_.gcount() << " " << bytesRead << " " << file_.fail() << " " << file_.eof() << std::endl << buf << std::endl;
    croak();
  }
  
  if (fsize_ == 0) endTest();
  delete[] buf;
  // puts(data);
}

void IOTestReadService::croak()
{
  // there was an error somewhere
  assert (0);
}

IOTestReadService::~IOTestReadService()
{
  file_.close();
}

void IOTestReadService::endTest()
{
  device_->close();
  getActiveObject()->stop();
  IOManager::getInstance()->getIOThread()->getActiveObject()->stop();
}