devel/libmdw/trunk/libmdw/src/TcpServer.h

/***************************************************************************
 *   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.             *
 ***************************************************************************/
#ifndef MDWTCPSERVER_H
#define MDWTCPSERVER_H
#include <vector>

#include "ObjectPool.h"
#include "Stream.h"
#include "StreamServerApi.h"

namespace mdw
{
class TcpServer : public StreamServer
{
public:
    TcpServer (short port);
    TcpServer (short port, ObjectPool<std::vector<char> > &pool);

    virtual void accept (StreamServerVisitor &visitor, int timeOutSec, int timeOutMicro);
    virtual void write (const char * buffer, int size, int streamId);
    virtual Stream &outStream (int streamId);
    virtual void close (int sockedId);

    ~TcpServer();
private:
    int _servSock;
    int _maxSelectFd;
    void init (short port);

class TcpStream : public Stream
    {
    public:
        void reset (TcpServer &t, int streamId);
        virtual void output (const char *buffer, int size);
    private:
        TcpServer *_t;
        int _id;
    };

    struct TcpSoc
    {
        TcpStream stream;
        StreamInfo info;
        bool connected;
    };

    std::vector<TcpSoc> _connectedSocket;
    ObjectPool<std::vector<char> > *_pool;
};

}

#endif

---