Telnet.cpp

   1 #include <iostream>
   2 #include <sys/types.h>
   3 #include <sys/socket.h>
   4 #include <string.h>
   5 #include <errno.h>
   6 #include "Telnet.hpp"
   7 #include "Exceptions.hpp"
   8
   9 static const char CRLF[] = { 13, 10, 0 };
  10 static const char IAC[] = { 255, 0 };
  11
  12 Telnet::Telnet( int sock )
  13         : m_sock( sock )
  14 {
  15 }
  16
  17 Telnet::~Telnet()
  18 {
  19 }
  20
  21 TelnetPtr Telnet::Create( int sock )
  22 {
  23         TelnetPtr ret( new Telnet( sock ) );
  24         ret->m_this = ret;
  25         ret->m_cmd.reset( new TelnetCommand( ret ) );
  26
  27         return ret;
  28 }
  29
  30 bool Telnet::Read()
  31 {
  32         std::string buf;
  33
  34         // Read all that's waiting on the socket
  35         for(;;)
  36         {
  37                 char tmpBuf[BufSize];
  38                 int size = recv( m_sock, tmpBuf, BufSize, 0 );
  39
  40                 if( size == -1 )
  41                 {
  42                         if( errno == EAGAIN )
  43                         {
  44                                 // No error, just nothing to be read from socket
  45                                 break;
  46                         }
  47
  48                         throw strerror( errno );
  49                 }
  50                 else if( size == 0 )
  51                 {
  52                         throw ConnectionTerminatedException;
  53                 }
  54
  55                 buf.append( tmpBuf, size );
  56         }
  57
  58         // Parse telnet commands
  59         for( unsigned int i=0; i<buf.size(); i++ )
  60         {
  61                 if( m_cmd->ParsingCommand() || (unsigned char)buf[i] == 255 )
  62                 {
  63                         if( !m_cmd->Parse( buf[i] ) )
  64                         {
  65                                 throw "Telnet parse error";
  66                         }
  67                 }
  68                 else
  69                 {
  70                         m_readBuf += buf[i];
  71                 }
  72         }
  73
  74         return m_readBuf.find( CRLF ) != std::string::npos;
  75 }
  76
  77 void Telnet::Write( const std::string& msg )
  78 {
  79         std::string buf;
  80         for( unsigned int i=0; i<msg.size(); i++ )
  81         {
  82                 buf += msg[i];
  83
  84                 if( (unsigned char)msg[i] == 255 )
  85                 {
  86                         buf.append( IAC );
  87                 }
  88         }
  89         buf.append( CRLF );
  90
  91         unsigned int pos = 0;
  92         char *ptr = (char*)buf.c_str();
  93
  94         while( pos != buf.size() )
  95         {
  96                 int size = send( m_sock, ptr, buf.size() - pos, 0 );
  97
  98                 if( size == -1 )
  99                 {
 100                         throw strerror( errno );
 101                 }
 102                 else if( size == 0 )
 103                 {
 104                         throw ConnectionTerminatedException;
 105                 }
 106
 107                 pos += size;
 108                 ptr += size;
 109         }
 110 }
 111
 112 std::string Telnet::GetBuf()
 113 {
 114         std::string ret;
 115
 116         unsigned int pos = m_readBuf.find( CRLF );
 117         if( pos == std::string::npos )
 118         {
 119                 throw "Trying to get telnet buffer when buffer not ready";
 120         }
 121
 122         ret = m_readBuf.substr( 0, pos );
 123
 124         m_readBuf.erase( 0, pos + 2 );
 125
 126         return ret;
 127 }
 128
 129 void Telnet::EraseCharacter()
 130 {
 131         if( m_readBuf.size() > 0 )
 132         {
 133                 m_readBuf.erase( m_readBuf.end() - 1 );
 134         }
 135 }
 136
 137 void Telnet::EraseLine()
 138 {
 139         throw "Unimplemented";
 140 }