/* 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 (void) { // Crée le file descriptor du socket socket_ = socket (PF_INET, SOCK_STREAM, 0); if (socket_ < 0) throw errno_exception ("Socket: La chaussette n'a pas pu être tricoté", errno); } SocketClient::SocketClient (SocketServer & socketServer) { socket_ = socketServer.accept (); if (socket_ < 0) throw errno_exception ("Socket: 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 ("Socket: Impossible de mettre la chaussette", errno); } void SocketClient::connect (const std::string & address, int port) { connect (Address(address, port)); } bool SocketClient::read (std::string & strReaded, bool bloquant) { int charReaded; char buffer[BUFFER_SIZE]; // Vérifie si il y a quelque chose à lire et retourne si non if (!bloquant) { fd_set fds; timeval tv = {0, 0}; FD_ZERO(&fds); FD_SET(socket_, &fds); if(select(socket_ + 1, &fds, NULL, NULL, &tv) == 0) return false; } strReaded.clear(); 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; strReaded.append(buffer, charReaded); }while (charReaded == BUFFER_SIZE - 1); return true; } void SocketClient::write (const std::string & str) { write(reinterpret_cast(str.data()), str.size()); } void SocketClient::write (const uint8_t * buffer, unsigned size) { int retval; retval = ::write (socket_, buffer, size); if (retval < 0) throw errno_exception ("SocketClient: erreur d'écriture (write()) ", errno); } //int //SocketClient::getFD(void) //{ // return socket_; //}