1 files changed, 184 insertions, 0 deletions

diff --git a/tsquerythread.h b/tsquerythread.h
new file mode 100644
index 0000000..c41c535
--- /dev/null
+++ b/tsquerythread.h
@@ -0,0 +1,184 @@
+/*
+ * 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; version 2 of the License.
+ *
+ * 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., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+ *
+ * Authors: Manuel Mausz (manuel@mausz.at)
+ *          Christian Raschko (c.raschko@netcore.at)
+ */
+
+//Header guard
+#ifndef TSQUERY_H
+#define TSQUERY_H
+
+//Libraries
+#include <wx/wx.h>
+#include <wx/socket.h>
+#include <wx/xml2.h>
+
+#include "tsheaders.h"
+
+//! TSQuery
+/*! TSQuery
+ */
+class TSQueryThread : public wxThread
+{
+  DECLARE_CLASS(TSQueryThread)
+  public:
+
+    /*! thread execution starts here
+    *  \return Returns ExitCode.
+    */
+    virtual ExitCode Entry();
+
+    /*! and stops here
+    */
+    virtual void OnExit();
+
+   /*! Default CTor, Initializes the object.
+    *  Creates all necessary member objects.
+    */
+    TSQueryThread(TSClient *client);
+
+   /*! Default DTor.
+    */
+    ~TSQueryThread();
+
+   /*! Sets the password for incomming connections.
+    *  \param str Password string.
+    *  \return Returns false if fails, check GetLastError for details.
+    *  \sa GetLastError()
+    */
+    bool SetPassword(wxString const &str);
+
+   /*! Sets the allowed Addresses.
+    *  You can use either an IP or an URL, seperate with ','
+    *  \param str Server address string.
+    *  \return Returns false if fails, check GetLastError for details.
+    *  \sa GetLastError()
+    */
+    bool SetAllowedAddresses(wxString const &str);
+
+    /*! Start thread.
+    *  \return Returns false if fails, check GetLastError for details.
+    *  \sa GetLastError()
+    */
+    bool Start();
+
+    /*! Stop thread.
+    *  \return Returns false if fails, check GetLastError for details.
+    *  \sa GetLastError()
+    */
+    bool Stop();
+
+   /*! Sets the server address.
+    *  You can use either an IP or an URL.
+    *  \param str Server address string.
+    *  \return Returns false if fails, check GetLastError for details.
+    *  \sa GetLastError()
+    */
+    bool SetServerAddress(wxString const &str);
+
+   /*! Sets the connection port.
+    *  \param port Connection port.
+    *  \return Returns false if fails, check GetLastError for details.
+    *  \sa GetLastError()
+    */
+    bool SetPort(wxUint16  const &port);
+
+   /*! Listen for incomming connections.
+    *  \return Returns false if fails, check GetLastError for details.
+    *  \sa GetLastError()
+    */
+    bool Accept();
+
+   /*! Dumps object.
+    *  \param ostrm Stream to write.
+    */
+    void Dump(wxOutputStream &ostrm) const;
+
+   /*! Gets the server Address.
+    *  \return Returns the current password.
+    */
+    wxString const &GetServerAddress() const
+    {
+      return m_ServerAddress;
+    }
+   /*! Gets the password for incomming connections.
+    *  \return Returns the current password.
+    */
+    wxString const &GetPassword() const
+    {
+      return m_Password;
+    }
+
+   /*! Gets the allowed addresses for incomming connections.
+    *  \return Returns the allowed addresses.
+    */
+    wxString const &GetAllowedAddresses() const
+    {
+      return m_AllowedAddresses;
+    }
+
+   /*! Gets the port for incomming connections.
+    *  \return Returns the current port.
+    */
+    wxUint16 const &GetPort() const
+    {
+      return m_Port;
+    }
+   /*! Gets the LastError message, call this method
+    *  to get more information for an error.
+    *  \return LastError message.
+    */
+    wxString const &GetLastError() const
+    {
+      return m_LastError;
+    }
+
+  private:
+    bool CmdGetStatus(wxSocketBase *sock);
+    bool CmdGetChannels(wxSocketBase *sock);
+    bool CmdGetUsers(wxSocketBase *sock);
+    bool CmdCreateChannel(wxSocketBase *sock);
+    bool CmdDeleteChannel(wxSocketBase *sock);
+    bool CmdModifyChannel(wxSocketBase *sock);
+    bool CmdMoveUser(wxSocketBase *sock);
+    bool CmdKickUser(wxSocketBase *sock);
+    bool CmdKill(wxSocketBase *sock);
+
+    bool SendDOM(wxSocketBase *sock);
+    bool ProcessCommand(wxSocketBase *sock);
+    bool SendError(wxString const &cmd, wxString const &str, wxSocketBase *sock);
+    bool IsIPAllowed(wxString const &str);
+    bool IsValidFormat();
+    wxString GetNodeValue(wxXml2Node &pElemRoot, wxString const &str);
+
+    //Sets the LastError message.
+    void SetLastError(wxString const &str)
+    {
+      m_LastError = str;
+    }
+
+    //Members
+    wxString         m_LastError;
+    wxString         m_AllowedAddresses;
+    wxString         m_Password;
+    wxSocketServer  *m_pSocket;
+    wxIPV4address    m_ServerAddr;
+    TSClient        *m_pClient;
+    wxXml2Document   m_pXml;
+    wxUint16         m_Port;
+    wxString         m_ServerAddress;
+};
+
+#endif