cpp_utils/other/sslserver_main.cpp

//  gpp sslserver_main.cpp  -o sslserver -I../include ../src/Logger.cpp ../src/Socket.cpp -ggdb ../src/SocketServer.cpp ../src/Connection.cpp ../src/Poll.cpp  ../src/TcpConnection.cpp ../src/SslConnection.cpp  -lssl -lcrypto  ../src/Addrinfo.cpp

#include "Logger.hpp"
#include "Common.hpp"

#include "Message.hpp"
#include "SslConnection.hpp"
#include "SocketServer.hpp"


#include <iostream>
#include <string>

#include <signal.h>

class EchoMessage : public Message
{
public:

  EchoMessage( void *msgParam = 0)
    : Message(msgParam)
  {
    TRACE;
  }

  bool buildMessage( const void   *msgPart,
                     const size_t  msgLen )
  {
    TRACE;
    m_buffer = std::string( (const char*) msgPart, msgLen );
    onMessageReady();
    return true;
  }

  void onMessageReady()
  {
    TRACE;

    LOG( Logger::INFO, std::string("Got message: \"").
                        append(m_buffer).append("\" from: ").
                        append(m_connection->getHost().append(":").
                        append(TToStr(m_connection->getPort())) ).c_str() );

    std::string reply("Got your message, ");
    reply.append(m_connection->getHost()).append(":").
          append(TToStr(m_connection->getPort())).
          append(" \"").append(m_buffer).append("\"");

    m_connection->send( reply.c_str(), reply.length() );
  }

  Message* clone()
  {
    TRACE;
    return new EchoMessage(m_param);
  }

protected:

  size_t getExpectedLength()
  {
    TRACE;
    return 0;
  }
};


SocketServer *socketServer;

void signalHandler(int s)
{
  LOG_STATIC( Logger::INFO, std::string("Exiting after receiving signal: ").
                              append(TToStr(s)).c_str() );
  socketServer->stop();
  delete socketServer;
  SslConnection::destroy();
  Logger::destroy();
  exit(1);
}


int main(int argc, char* argv[] )
{
  if ( argc != 5 ) {
    std::cerr << "Usage: " << argv[0] <<  " <HOST> <PORT> <CERT> <PRIVKEY>" << std::endl;
    return 1;
  }

  struct sigaction sigIntHandler;
  sigIntHandler.sa_handler = signalHandler;
  sigemptyset(&sigIntHandler.sa_mask);
  sigIntHandler.sa_flags = 0;
  sigaction(SIGINT, &sigIntHandler, NULL);


  Logger::createInstance();
  Logger::init(std::cout);
  Logger::setLogLevel(Logger::FINEST);
  SslConnection::init();

  EchoMessage msg;
  SslConnection conn(argv[1], argv[2], &msg);
  if ( !conn.initServerContext(argv[3], argv[4]) ) {
    LOG_STATIC( Logger::ERR, "Failed to init SSL context, exiting...");
    SslConnection::destroy();
    Logger::destroy();
    return 1;
  }

  socketServer = new SocketServer(&conn);

  if ( !socketServer->start() ) {
    LOG_STATIC( Logger::ERR, "Failed to start TCP server, exiting...");
    delete socketServer;
    SslConnection::destroy();
    Logger::destroy();
    return 1;
  }

  // never reached

  return 0;
}