vlFileTransport.h

Go to the documentation of this file.

/*********************************************************************
** Copyright (c) 2004 MaK Technologies, Inc.
** All rights reserved.
*********************************************************************/
/*********************************************************************
** $RCSfile: vlFileTransport.h,v $ $Revision: 1.2.6.1 $ $State: Exp $
*********************************************************************/
#ifndef vlFileTransport_H_
#define vlFileTransport_H_


#ifndef VXWORKS

#include <vlutil/vlFilename.h>
#include <vlutil/vlString.h>
#include <vlutil/vlUtil.h>
#include "vlSocket.h"

#include <fstream>
#include <map>

#ifdef DtUSE_UTILITIES_NAMESPACE
namespace DtUSE_UTILITIES_NAMESPACE
{
#endif


class DtTcpSocket;
class DtTcpSocketMgr;

class DT_DLL_NETCOMPAT DtFileTransportObject
{
public:
   DtFileTransportObject();
   DtFileTransportObject(const DtFileTransportObject&);

   DtFileTransportObject& operator=(const DtFileTransportObject&);

   int               myFileIdentifier;
   DtFilename        myFilename;
   int               myFileLength;
   unsigned long     myBufferLength;
   unsigned long     myBytesReceived;
   std::vector<char> myBuffer;
   int               myFlags;
};

class DT_DLL_NETCOMPAT DtFileTransporter
{
public:

   typedef void (*DtFileTransferedCallbackFn)
      (const DtFilename& file, void* usr);

   typedef std::map<int, DtFileTransportObject> DtFileObjectMap;

   enum DtFileTransportFlags
   {
      DtFileNormal = 0,
      DtFileCompressed = 1,
      DtFileEncrypted = 2
   };

public:

   DtFileTransporter(
      int port = DtDEFAULT_PORT, 
      bool findFirstAvailable = false,
      int portRangeToTry = 10, 
      const DtFilename &tagetDirectory = ".",
      int *status = 0);

   virtual ~DtFileTransporter();

   virtual void tick();

   virtual unsigned int port() const;

   virtual void addFileTransferedCallback(
      DtFileTransferedCallbackFn fn, void* data);
   virtual void removeFileTransferedCallback(
      DtFileTransferedCallbackFn fn, void* data);

   virtual bool sendFile(
      const DtFilename& filename, DtIpv4Addr destAddr, int port);

   virtual bool sendFile(
      const DtFilename& filename, DtIpv4Addr destAddr);

   const DtFilename& targetDirectory() const;
   void setTargetDirectory(const DtFilename& f);

public:

   static void receiveFileCb(DtTcpSocket* sock, void* usr);

protected:

   virtual void receiveFile(DtTcpSocket* sock);

   virtual int nextFileIdentifier();

   virtual bool readFileToBuffer(std::ifstream& inFile,
      std::vector<char>& fileBuffer, int fileLengthInBytes, int& retBufLen,
      DtFileTransportFlags& flags);

   virtual void finishFile(DtFileTransportObject& object);

   virtual DtFileTransportObject& transportObject(int iFileIdentifier);

protected:

   DtFileTransporter(const DtFileTransporter&);

   DtFileTransporter& operator=(const DtFileTransporter&);

protected:

   typedef std::map<DtFileTransferedCallbackFn, void*> FileTransferedCallbackMapType;

protected:

   FileTransferedCallbackMapType myCallbacks;
   int myPort;
   DtTcpSocketMgr* mySocketMgr;
   DtFilename myTargetDirectory;

   DtFileObjectMap myActiveFiles;
};

#ifdef DtUSE_UTILITIES_NAMESPACE
} 
#endif

#endif   



#endif