From 8f486613be58ced269db1d437e560c16558604e8 Mon Sep 17 00:00:00 2001 From: becquet Date: Thu, 10 May 2007 18:49:20 +0000 Subject: Création de chuck, le programme du robot 2007. --- i/chuck/src/socket/socket_client.cc | 115 ++++++++++++++++++++++++++++++++++++ 1 file changed, 115 insertions(+) create mode 100644 i/chuck/src/socket/socket_client.cc (limited to 'i/chuck/src/socket/socket_client.cc') diff --git a/i/chuck/src/socket/socket_client.cc b/i/chuck/src/socket/socket_client.cc new file mode 100644 index 0000000..951ea6a --- /dev/null +++ b/i/chuck/src/socket/socket_client.cc @@ -0,0 +1,115 @@ +/* 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_; +//} -- cgit v1.2.3