guiclient/echo.cpp

Go to the documentation of this file.

//this is a wxWidgets based server that just echo's any infomation recieved

#include <wx/wx.h>
#include <wx/socket.h>
#include "../trucknet/trucknet.h"

class echoApp : public wxApp
{
        public:
                virtual bool OnInit();
};

class echoFrame : public wxFrame
{
        public:
                echoFrame();
                ~echoFrame();

                void OnServer(wxSocketEvent& event);
                void OnClient(wxSocketEvent& event);

        private:
                wxSocketServer  *server;
                wxTextCtrl              *text;

                DECLARE_EVENT_TABLE()
};

 
enum
{       SERVER_SOCKET,
        CLIENT_SOCKET,
};

BEGIN_EVENT_TABLE(echoFrame, wxFrame)
        EVT_SOCKET(SERVER_SOCKET,       echoFrame::OnServer)
        EVT_SOCKET(CLIENT_SOCKET,       echoFrame::OnClient)
END_EVENT_TABLE()

        

IMPLEMENT_APP(echoApp)

bool echoApp::OnInit()
{
        echoFrame *frame = new echoFrame();
        frame->Show();
        SetTopWindow(frame);
        return true;
}

echoFrame::echoFrame() :
        wxFrame((wxFrame *)NULL, wxID_ANY, wxT("Echo Server"), wxDefaultPosition,
                                wxSize(300, 200))
{       CreateStatusBar();
        text = new wxTextCtrl(this, wxID_ANY, wxT(""), \
                        wxDefaultPosition, wxDefaultSize, \
                        wxTE_MULTILINE | wxTE_READONLY);
        
        wxIPV4address addr; //defalts to localhost
        addr.Service(GUI_PORT.PORT);
        
        server = new wxSocketServer(addr,wxSOCKET_REUSEADDR);

        if (! server->Ok())
        {       text->AppendText(wxT("Problem creating socket...\n"));
                return;
        }
        else
                SetStatusText(wxT("Server listening..."));

        server->SetEventHandler(*this, SERVER_SOCKET);
        server->SetNotify(wxSOCKET_CONNECTION_FLAG | wxSOCKET_LOST_FLAG);
        server->Notify(true);
}

echoFrame::~echoFrame()
{       delete server;
}

void echoFrame::OnServer(wxSocketEvent& event)
{       switch(event.GetSocketEvent())
        {       case wxSOCKET_CONNECTION:
                        wxSocketBase *client;
                        client = server->Accept(false); //false = nonblocking
                        if (client)
                                SetStatusText(wxT("Connected."));
                        else
                                SetStatusText(wxT("Attempted connection failed..."));
                        client->SetEventHandler(*this, CLIENT_SOCKET); 
                        client->SetNotify(wxSOCKET_INPUT_FLAG | wxSOCKET_LOST_FLAG);
                        client->Notify(true);
                        client->SetFlags(wxSOCKET_NOWAIT);
                        break;
                case wxSOCKET_LOST:
                        SetStatusText(wxT("Server socket lost."));
                        break;
                default:
                        SetStatusText(wxT("Unknown event on server socket!"));
        }
}



void echoFrame::OnClient(wxSocketEvent& event)
{       wxSocketBase *sock = event.GetSocket();
        int count;
        const int buflen = 32;//make multiple of 4 for unicode
        char buf[buflen + 1];

        switch(event.GetSocketEvent())
        {       case wxSOCKET_INPUT:
                        *text << wxT("Data from client:\n");
                        do
                        {       count = sock->Read(buf, buflen).LastCount();
                                if(count > 0)
                                {       *text << wxString(buf, *wxConvCurrent, count);
                                        sock->Write(buf, count);
                                }
                        } while(count == buflen);
                        *text << wxT("\n\n");
                        break;
                case wxSOCKET_LOST:
                        SetStatusText(wxT("Client connection lost."));
                        break;
                default:
                        SetStatusText(wxT("Unknown event on client socket!"));
        }
}

---