#include "SocketLib/SocketSet.h"
int main()
{    
    int err;                    // for getting errors    
    ListeningSocket lsock;
    lsock.Listen(4000);
    DataSocket dsock;    
    char buffer[128] = "Hello There!";     
    bool done = false;          // used for quitting
    vector<DataSocket>::iterator itr;
    vector<DataSocket> socketlist;     // list of sockets    
    while( !done ){        
        SocketSet sset;                
        // add listen sockets
        sset.AddSocket( lsock );        
        // add all of the data sockets
        for( itr = socketlist.begin(); itr != socketlist.end(); itr++ )
        {
            sset.AddSocket( *itr );
        }        
        int active = sset.Poll( 1000 );
        if(active)
        {
            if(sset.HasActivity(lsock))
            {
                DataSocket dsock;
                dsock = lsock.Accept();
                socketlist.push_back( dsock );
            }         
            // loop through each socket and see if it has any activity
            for( itr = socketlist.begin(); itr != socketlist.end(); itr++ )
            {
                if(sset.HasActivity(*itr))
                {
                    // incomming data
                    cout << "receiving data from socket " << itr->GetSock() << "..." << endl;
                    try
                    {                    
                        err = itr->Receive(buffer, 128);
                        if(err){
                            cout << "Data: " << buffer << endl;
                        }
                    }
                    catch(Exception& e)
                    {                
                        cout << "Exception Error: " << err << endl;
                        socketlist.erase( itr );
                        itr--;
                    }    
                    // if the message was "servquit", then quit the server.
                    if( strcmp( buffer, "servquit" ) == 0 )
                        done = true;
                }
            }
        }    
    }
    return 0;
}

    vector<DataSocket>::iterator itr;
    vector<DataSocket> socketlist;     // list of sockets