devel/libmdw/trunk/libmdw/src/TextAuthenticationProtocol.cpp

/***************************************************************************
 *   Copyright (C) 2010 by Jean-Baptiste Wons   *
 *   wonsjb@gmail.com   *
 *                                                                         *
 *   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 "TextAuthenticationProtocol.h"
#include "Stream.h"
#include "AuthenticationManager.h"

namespace mdw
{

TextAuthenticationProtocol::TextAuthenticationProtocol (StreamServer &server) : _server (server)
{
}


TextAuthenticationProtocol::~TextAuthenticationProtocol()
{
}

void TextAuthenticationProtocol::visitRead (std::vector<char> &buffer, StreamInfo &info)
{
    int streamId = info.getStreamId();
    StatusState &state = _statusMap[streamId];
    switch (state.state)
    {
        case WaitingForUsername_e:
            for (int i = 0; i < buffer.size(); ++i)
            {
                if (buffer[i] == '\r' || buffer[i] == '\n')
                {
                    state.state = WaitingForPassword_e;
                    Stream &stream = _server.outStream (streamId);
                    stream << "Password : ";
                    break;
                }
                state.username += buffer[i];
            }
            break;
        case WaitingForPassword_e:
            for (int i = 0; i < buffer.size(); ++i)
            {
                if (buffer[i] == '\r' || buffer[i] == '\n')
                {
                    state.state = WaitingForPassword_e;
                    AuthenticationManager &AM = AuthenticationManager::getInstance();
                    Authentified *auth = AM.authentify (state.username, state.password, streamId);

                    if (auth == 0)
                    {
                        Stream &stream = _server.outStream (streamId);
                        if (state.numberFailures < 3)
                        {
                            stream << "Wrong Username / Password" << endl;
                            stream << "Username : ";
                            state.numberFailures++;
                            state.state = WaitingForUsername_e;
                            state.username.clear();
                            state.password.clear();
                        }
                        else
                        {
                            stream << "To many tries" << endl;
                            close (streamId);
                        }
                    }
                    else
                    {
                        state.state = Connected_e;
                        state.auth = auth;
                    }
                    break;
                }
                state.password += buffer[i];
            }
            break;
        case Connected_e:
            _currentVisitor->visitRead (&*buffer.begin(), buffer.size(), info);
            break;
    }
}

void TextAuthenticationProtocol::visitConnect (StreamInfo &info)
{
    int streamId = info.getStreamId();
    Stream &stream = _server.outStream (streamId);
    stream << "*************************************" << endl;
    stream << "*     Welcome on this server       **" << endl;
    stream << "*************************************" << endl;
    stream << "You are connecting from " << info.getIp() << endl;
    stream << "Username : ";
    StatusState &state = _statusMap[streamId];
    state.username.clear();
    state.password.clear();
    state.state = WaitingForUsername_e;
    state.numberFailures = 0;
    state.auth = 0;
}
void TextAuthenticationProtocol::visitDisconnect (StreamInfo &info)
{
    _currentVisitor->visitDisconnect (info);
    _statusMap.erase (info.getStreamId());
}

void TextAuthenticationProtocol::accept (StreamServerVisitor &visitor, int timeOutSec, int timeOutMicro)
{
    _server.accept (visitor, timeOutSec, timeOutMicro);
}
void TextAuthenticationProtocol::write (const char *buffer, int size, int streamId)
{
}
Stream &TextAuthenticationProtocol::outStream (int streamId)
{
    return _server.outStream (streamId);
}
void TextAuthenticationProtocol::close (int streamId)
{
    _server.close (streamId);
    _statusMap.erase (streamId);
}

}

---