Telnet.cpp

   1 #include <sys/types.h>
   2 #include <sys/socket.h>
   3 #include <string.h>
   4 #include <errno.h>
   5 #include "Telnet.hpp"
   6 #include "Exceptions.hpp"
   7
   8 const char CRLF[] = { 13, 10, 0 };
   9
  10 Telnet::Telnet( int sock )
  11         : m_sock( sock )
  12 {
  13 }
  14
  15 Telnet::~Telnet()
  16 {
  17 }
  18
  19 bool Telnet::Read()
  20 {
  21         std::string buf;
  22
  23         // Read all that's waiting on the socket
  24         for(;;)
  25         {
  26                 char tmpBuf[1024];
  27                 int size = recv( m_sock, tmpBuf, 1024, 0 );
  28
  29                 if( size == -1 )
  30                 {
  31                         if( errno == EAGAIN )
  32                         {
  33                                 // No error, just nothing to be read from socket
  34                                 break;
  35                         }
  36
  37                         throw strerror( errno );
  38                 }
  39                 else if( size == 0 )
  40                 {
  41                         throw ConnectionTerminatedException;
  42                 }
  43
  44                 buf.append( tmpBuf, size );
  45         }
  46
  47         // Parse according to telnet protocol
  48         /*
  49         for( unsigned int i=0; i<buf.size(); i++ )
  50         {
  51
  52         }
  53         */
  54         // temporary
  55         m_readBuf.append( buf );
  56
  57         return m_readBuf.find( CRLF ) != std::string::npos;
  58 }
  59
  60 void Telnet::Write( const std::string& msg )
  61 {
  62         std::string buf( msg );
  63         buf.append( CRLF );
  64
  65         unsigned int pos = 0;
  66         char *ptr = (char*)buf.c_str();
  67
  68         while( pos != buf.size() )
  69         {
  70                 int size = send( m_sock, ptr, buf.size() - pos, 0 );
  71
  72                 if( size == -1 )
  73                 {
  74                         throw strerror( errno );
  75                 }
  76                 else if( size == 0 )
  77                 {
  78                         throw ConnectionTerminatedException;
  79                 }
  80
  81                 pos += size;
  82                 ptr += size;
  83         }
  84 }
  85
  86 std::string Telnet::GetBuf()
  87 {
  88         std::string ret;
  89
  90         unsigned int pos = m_readBuf.find( CRLF );
  91         if( pos == std::string::npos )
  92         {
  93                 throw "Trying to get telnet buffer when buffer not ready";
  94         }
  95
  96         ret = m_readBuf.substr( 0, pos );
  97
  98         m_readBuf.erase( 0, pos + 2 );
  99
 100         return ret;
 101 }