summaryrefslogtreecommitdiff
path: root/i/marvin/src/socket/socket_client.cc
diff options
context:
space:
mode:
Diffstat (limited to 'i/marvin/src/socket/socket_client.cc')
-rw-r--r--i/marvin/src/socket/socket_client.cc117
1 files changed, 117 insertions, 0 deletions
diff --git a/i/marvin/src/socket/socket_client.cc b/i/marvin/src/socket/socket_client.cc
new file mode 100644
index 0000000..9dd19fa
--- /dev/null
+++ b/i/marvin/src/socket/socket_client.cc
@@ -0,0 +1,117 @@
+/* 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 <sys/socket.h>
+#include <cerrno>
+
+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)
+{
+ 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;
+ retval = ::write (socket_, buffer, size);
+ if (retval < 0)
+ throw errno_exception ("SocketClient: erreur d'écriture (write()) ", errno);
+ }
+}
+
+//int
+//SocketClient::getFD(void)
+//{
+// return socket_;
+//}