NetSendStr.h

Go to the documentation of this file.

#ifndef NETSENDSTR_H
#define NETSENDSTR_H

#include "carma/szaarrayutils/netbuf.h"
#include "carma/szaarrayutils/netobj.h"

#define NET_SEND_HANDLER(fn) void (fn)(void* arg)

#ifdef NET_ERROR_HANDLER
#undef NET_ERROR_HANDLER
#endif

#define NET_ERROR_HANDLER(fn) void (fn)(void* arg)

namespace sza {
  namespace util {
    
    class NetSendStr {
    public:
      
      enum NetSendId {   
        
        // The message body is being sent 
        
        NET_SEND_DATA   = sza::array::NetSendStr::NET_SEND_DATA,
        
        // The message has been completely sent 
        
        NET_SEND_DONE   = sza::array::NetSendStr::NET_SEND_DONE,
        
        // Closed connection detected at start of message 
        
        NET_SEND_CLOSED = sza::array::NetSendStr::NET_SEND_CLOSED,
        
        // I/O error detected while sending 
        
        NET_SEND_ERROR  = sza::array::NetSendStr::NET_SEND_ERROR
      };
      
      NetSendStr();
      
      NetSendStr(int fd, unsigned long size);
      
      virtual ~NetSendStr();
      
      void attach(int fd);
      
      void setBuffer(void* buffer, unsigned int size);
      
      NetSendId send();
      
      NetSendId send(int fd);
      
      NetSendId state();
      
      int getFd();
      
      void startPut(int opcode);
      
      void endPut();
      
      void putChar(long ndata, unsigned char *data);
      
      void putShort(long ndata, unsigned short *data);
      
      void putLong(long ndata, unsigned long *data);
      
      void putFloat(long ndata, float *data);   
      
      void putDouble(long ndata, double *data);
      
      void putObj(const NetObjTable* types, int id, void *obj);
      
      void incNput(long nbytes);
      
      void installSendHandler(NET_SEND_HANDLER(*handler), void* arg);

      void installErrorHandler(NET_ERROR_HANDLER(*handler), void* arg);

    private:
      
      void privateConstructor(int fd, unsigned long size);
      
      sza::array::NetSendStr* netStream_;
      
      bool netBufAllocated_;
      
      bool attached_;
      
      NetSendId privateSend(int fd);
      
      NetSendId privateState(sza::array::NetSendStr::NetSendId id);
      
      // Handler functions

      NET_SEND_HANDLER(*sendHandler_);

      void* sendArg_;

      NET_SEND_HANDLER(*errorHandler_);

      void* errorArg_;

    }; // End class NetSendStr
    
  }; // End namespace util
}; // End namespace sza

#endif // End #ifndef