rstpclient.h

/* libradar_stp
 * <http://www.k9nl.net/radar/> <radar@k9nl.net>
 * 
 * rstpclient.h - RSTP client class
 *
 * THIS FILE IS NOT MEANT FOR LIBRARY CLIENT USE
 *
 * Copyright (C) 2007 Nicholas J. Luther
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License,
 * version 2, as published by the Free Software Foundation.
 * 
 * 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.
 * You may contact the author by writing to Nick Luther, 1655 South
 * Westhaven Drive, Oshkosh, Wisconsin 54904, USA.
 * 
 */
 
 
#ifndef _LIBRSTP_INC_RSTPCLIENT_H
#define _LIBRSTP_INC_RSTPCLIENT_H
 
 
#include <string>
#include <queue>
#include <set>

#include "lplproj.h"
#include "thread/thread.h"
#include "thread/mutex.h"
#include "socket/tcpsocket.h"

namespace libradar_stp {

// CAUTION: be careful copying RSTPClients
class RSTPClient : public LPORTLIB_NAMESPACE::thread::Thread {
public:
  
  RSTPClient();
  ~RSTPClient();
  
  bool connect( const std::string& addr, unsigned short port, const std::string& login = "",
                const std::string& pass = "" );
  void quit();
  
  void addStream( const std::string& stream );
  void dropStream( const std::string& stream );
  
  
  typedef void (*CBFUNC)( void*, const std::string&, const std::string& );
  typedef void (*CBFUNC2)( void* );

  void setDataCallback( CBFUNC func, void* arg = 0 );
  void setConnLossCallback( CBFUNC2 func, void* arg = 0 );
  
  std::string getError();
  unsigned int getLastDataTime() const;
  
protected:

  void beginThread();
  
private:

  LPORTLIB_NAMESPACE::socket::TCPSocket m_sock;
  bool m_quit_request;
  unsigned int m_lh;
  unsigned int m_last_data;
  bool m_nop;
  
  CBFUNC m_data_cb;
  void* m_data_arg;
  CBFUNC2 m_cl_cb;
  void* m_cl_arg;  
  
  std::string m_error;
  
  std::queue< std::string > m_sadd_queue;
  LPORTLIB_NAMESPACE::thread::Mutex m_sadd_queue_mutex;
  std::queue< std::string > m_sdrp_queue;
  LPORTLIB_NAMESPACE::thread::Mutex m_sdrp_queue_mutex;
  
}; // class

class RedundantRSTPClient : public LPORTLIB_NAMESPACE::thread::Thread {
public:

  RedundantRSTPClient();
  ~RedundantRSTPClient();
  
  void addStream( const std::string& stream );
  void addServer( const std::string& addr, unsigned short port, unsigned int timeout = 1000,
                  const std::string& login = "", const std::string& pass = "" );
  
  void start();               
  void quit();
  
  typedef void (*CBFUNC)( void*, const std::string&, const std::string& );
  typedef void (*CBFUNC3)( void*, const std::string& );
  void setDataCallback( CBFUNC func, void* arg = 0 );
  void setErrorCallback( CBFUNC3 func, void* arg = 0 );

   
protected:
  
  void beginThread();
  
private:

  bool m_quit_request;
  bool m_started_thread;

  struct Server {
    RSTPClient* rstp_client;
    std::string addr;
    unsigned short port;
    unsigned int timeout;
    unsigned int last_attempt;
    std::string login;
    std::string pass;
    bool join_now;
  }; // struct
  
  std::vector< Server* > m_servers;
  std::set< std::string > m_streams;
  LPORTLIB_NAMESPACE::thread::Mutex m_streams_mutex;
  CBFUNC m_data_cb;
  CBFUNC3 m_error_cb;
  void* m_data_cb_arg;
  void* m_error_cb_arg;
  
  bool m_running;
  
public:

  static void _cl_cb ( void* arg );
  
}; // class

};// namespace

#endif

---