vlInetFileTransport.h

Go to the documentation of this file.

/*********************************************************************
** Copyright (c) 1992-2010 VT MAK
** All rights reserved.
*********************************************************************/
#pragma once

#ifndef VXWORKS

#include "vlFilename.h"
#include "vlString.h"
#include "vlUtil.h"
#include "vlInetAddr.h"

#include <fstream>
#include <map>
#include <vector>

#ifdef DtUSE_UTILITIES_NAMESPACE
namespace DtUSE_UTILITIES_NAMESPACE
{
#endif


class DtInetSocket;
class DtInetTcpSocket;
class DtInetSocketMgr;

class DT_DLL_VLUTIL DtInetFileTransportObject
{
public:
   DtInetFileTransportObject();
   DtInetFileTransportObject(const DtInetFileTransportObject&);

   DtInetFileTransportObject& operator=(const DtInetFileTransportObject&);

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

class DT_DLL_VLUTIL DtInetFileTransporter
{
public:

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

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

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

public:

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

   virtual ~DtInetFileTransporter();

   virtual void tick();

   virtual unsigned int port() const;

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

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

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

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

public:

//   static void receiveFileCb(DtTcpSocket* sock, void* usr);
   static bool receiveFileCb(DtInetSocket* sock, void* usr);

protected:

//   virtual void receiveFile(DtTcpSocket* sock);
   virtual void receiveFile(DtInetTcpSocket* sock);

   virtual int nextFileIdentifier();

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

   virtual void finishFile(DtInetFileTransportObject& object);

   virtual DtInetFileTransportObject& transportObject(int iFileIdentifier);

protected:

   DtInetFileTransporter(const DtInetFileTransporter&);

   DtInetFileTransporter& operator=(const DtInetFileTransporter&);

protected:

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

protected:

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

   DtFileObjectMap myActiveFiles;
};

#ifdef DtUSE_UTILITIES_NAMESPACE
} 
#endif

#endif