/* socket_client.cc - Définition de la classe des sockets client */ /* Simulotron - Programme de simulation de robot {{{ * * Copyright (C) 2005 Nicolas Haller * * Robot APB Team/Efrei 2005. * Web: http://assos.efrei.fr/robot/ * Email: robot AT efrei DOT fr * * This program 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 2 of the License, or * (at your option) any later version. * * This program 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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. * * }}} */ #include "socket/socket_client.hh" #include "socket/socket_server.hh" #include "socket/address.hh" #include "utils/errno_exception.hh" #include #include SocketClient::SocketClient (const Address & address) { // Crée le file descriptor du socket socket_ = socket (PF_INET, SOCK_STREAM, 0); if (socket_ < 0) throw errno_exception ("La chaussette n'a pas pu être tricoté", errno); // On bind le socket if (::bind (socket_, address.getSockaddr (), sizeof (*address.getSockaddr ())) < 0) throw errno_exception ("Impossible d'assigner la chausette", errno); } SocketClient::SocketClient (SocketServer & socketServer) { socket_ = socketServer.accept (); if (socket_ < 0) throw errno_exception ("Problème lors de la construction d'une chaussette cliente par accept", errno); } SocketClient::~SocketClient (void) { if (socket_ >= 0) ::close (socket_); } void SocketClient::connect (const Address & address) { if (::connect (socket_, address.getSockaddr (), sizeof (sockaddr_in)) < 0) throw errno_exception ("Impossible de mettre la chaussette", errno); } std::string SocketClient::read (void) { std::string str; int charReaded; char buffer[BUFFER_SIZE]; do { charReaded = ::read (socket_, buffer, BUFFER_SIZE - 1); if (charReaded < 0) throw errno_exception ("SocketClient: problème de lecture", errno); buffer[charReaded] = 0; str += buffer; }while (charReaded == BUFFER_SIZE - 1); return str; } void SocketClient::write (const std::string & str) { char buffer[BUFFER_SIZE]; unsigned int pos = 0; int retval = 0; while (str.size () != pos) { int size = str.size () - pos < BUFFER_SIZE - 1 ? str.size () - pos:BUFFER_SIZE - 1; str.copy (buffer, size, pos); pos += size; buffer[size] = 0; retval = ::write (socket_, buffer, size); if (retval < 0) throw errno_exception ("SocketClient: erreur d'écriture (write()) ", errno); } } char SocketClient::getChar (void) { char getted; if (::read (socket_, &getted, 1) < 0) throw errno_exception ("SocketClient: erreur d'écriture (write()) ", errno); return getted; } void SocketClient::putChar (char c) { if (::write (socket_, &c, 1) < 0) throw errno_exception ("SocketClient: erreur d'écriture (write()) ", errno); }