基于UDP的传输(Base UDP File Transfer)

写了个UDP传输文件的东东.(经丢包测试)

先扔个头文件上来MARK下

  1#pragma once
  2#include "Net.h"
  3/**//********************************************************************
  4    created:    2009/04/07
  5    created:    7:4:2009   17:10
  6    filename:     f:\work\MyProject\myiu\IU\include\UDPFilesStrategy.h
  7    file path:    f:\work\MyProject\myiu\IU\include
  8    file base:    UDPFilesStrategy
  9    file ext:    h
 10    author:        kevin meng
 11    
 12    purpose:    UDP传输文件的上层策略
 13*********************************************************************/
 14
 15/**///////////////////////////////////////////////////////////////////////////
 16//加载Files.lib
 17#pragma comment (lib,"../bin/Files.lib")
 18#include "Files.h"
 19/**///////////////////////////////////////////////////////////////////////////
 20//加载临界锁
 21#include "CriticalSectionLock.h"
 22/**//********************************************************************
 23purpose:    滑动窗口 UDP 传送文件
 24只使用了简单的滑动窗口机制
 25实现:
 261.固定的窗口大小
 272.固定的传输速率
 28
 29没有实现:
 301.拥塞窗口机制 
 312.RTT计算
 323.等等
 33该类的App层:
 34*********************************************************************/
 35extern "C"{
 36
 37//上层APP控制,必须继承的类(以该类虚函数为接口)
 38class NET_API CUDPFilesStrategy_App{
 39public:
 40    CUDPFilesStrategy_App(){};
 41    virtual ~CUDPFilesStrategy_App(){};
 42public:
 43    //由CUDPFilesStrategy 调用继承CUDPFilesStrategy_App 并实现DoOnFileSize 的函数 传递文件大小
 44    virtual void DoOnFileSize(uint nFileSize) = 0;
 45    
 46    //由CUDPFilesStrategy 调用继承CUDPFilesStrategy_App 并实现DoOnTransferState 的函数 说明现在传输的状态
 47    //************************************
 48    // Method:    DoOnTransferState
 49    // FullName:  CUDPFilesStrategy_App::DoOnTransferState
 50    // Access:    virtual public 
 51    // Returns:   void
 52    // Qualifier:
 53    // Parameter: ushort nPercent        (范围0-100)该参数为100时说明传输完毕
 54    // Parameter: uint nSpeed            传输速度(kb/s)
 55    // Parameter: uint nComplete        传输完成数量
 56    // Parameter: uint nSpare            传输剩余数量
 57    // Parameter: uint nTagID            标示传输文件的ID
 58    //************************************
 59    virtual void DoOnTransferState(ushort nPercent,uint nSpeed,uint nComplete,uint nSpare,uint nTagID = 0) = 0;
 60    
 61    //由CUDPFilesStrategy 调用继承CUDPFilesStrategy_App 并实现DoOnSend 的函数 发送数据
 62    virtual void DoOnSend(const char* pBuf,uint nBufSize,uint nTagID = 0/**//*标示传输文件的ID*/) = 0;
 63protected:
 64private:
 65};
 66
 67class NET_API CUDPFilesStrategy
 68{
 69public:
 70    CUDPFilesStrategy(void);
 71    virtual ~CUDPFilesStrategy(void);
 72protected:
 73    //判断该段是否在窗口内
 74    inline bool _InWindows(uint nPosition);
 75    void _SetFileSize(uint nFileSize);
 76    
 77    
 78    //常量
 79    typedef enum _E{
 80        e_WindSize = 10,//窗口大小
 81        e_SendSpeed = 100,//发送速度PACK/S   (每秒100KB)
 82        e_OnceSendSize = 1024//一次发送最大字节数(数据)
 83    };
 84    //消息ID
 85    typedef enum _E_MSG_ID{
 86        e_SendData =    10000,//发送数据的ID
 87        e_RecvAck =        10001//发送ACK的ID
 88    };
 89    /**//////////////////////////////////////////////////////
 90    //消息结构体SR: SEND->RECV
 91    typedef struct  tag_MSG_SR_DATA{
 92        uint nMSGID;//_E_MSG_ID
 93        uint nSeq;//序号
 94        uint nSize;//数据区实际长度
 95        char szBuf[e_OnceSendSize];
 96        tag_MSG_SR_DATA(void){
 97            nMSGID = e_SendData;
 98        }
 99    };
    //#


    /**/////////////////////////////////////////////////////
    //消息结构体RS: RECV->SEND
    typedef struct  tag_MSG_RS_ACK{
        uint nMSGID;//_E_MSG_ID
        uint nSeq;//序号
        tag_MSG_RS_ACK(void){
            nMSGID = e_RecvAck;
        }
    };
    //#

    /**/////////////////////////////////////////////////////
    //线程相关
    HANDLE m_hThread;
    bool m_bThreadRun;
    //#
    
    /**////////////////////////////////////////////////////
    //当前窗口位置相关
    uint m_nCurrentPosition;//当前窗口起始位置1,2,3,4  从1开始
    CCriticalSectionLock m_iCurrentPositionLock;
    inline uint _GetCurrentPosition();
    inline void _SetCurrentPosition(uint val);
    //#

    /**////////////////////////////////////////////////////
    //文件操作,属性
    uint m_nFileBlock;//文件块数(文件按照一次发送大小最后计算出来的块数量)(不会改变)
    CCriticalSectionLock m_iFileBlockLock;
    inline uint _GetFileBlock();
    inline void _SetFileBlock(uint val);
    CFiles m_iCFiles;//操作文件的类
    uint m_nFileID;//传送或接收文件的ID;(不会改变)
    uint m_nFileSize;//文件长度(不会改变)
    
    //#

    
    CUDPFilesStrategy_App* m_piCUDPFilesStrategy_App;//上层APP的回调指针

    uint m_nSleekWindows;//滑动窗口大小
    uint m_nCongestionWindows;//拥塞窗口大小

private:    

};


class NET_API CUDPFilesStrategySend : public CUDPFilesStrategy
{
public:
    CUDPFilesStrategySend();
    virtual ~CUDPFilesStrategySend();
public:
    //************************************
    // Method:    BeginSendFile
    // FullName:  CFilesUdp::BeginSendFile
    // Access:    public 
    // Returns:   BOOL
    // Qualifier:
    // Parameter: const char * pszFilePath                    文件路径
    // Parameter: CFilesUdpApp * pCFilesUdpApp                上层调动层的指针
    //************************************
    BOOL BeginSendFile(const char* pszFilePath,CUDPFilesStrategy_App* pCUDPFilesStrategy_App,uint nID);
    void EndSendFile(void);

    //由APP层调用的RECV(),APP层收到
    inline void RecvAck(const char* pBuf,uint nBufSize);
protected:
private:
    /**//////////////////////////////////////////////////////////
    //发送线程相关
    static unsigned __stdcall __ThreadSend(void* pData);
    void __ThreadSendFn(void);
    //#

    //发送数据前组包
    //************************************
    // Method:    __CompositePacket
    // FullName:  CUDPFilesStrategySend::__CompositePacket
    // Access:    private 
    // Returns:   uint
    // Qualifier:
    // Parameter: IN const char * pBuf            传入数据
    // Parameter: uint nSize                    传入数据大小
    // Parameter: OUT char * pSendBuf            组包后的数据
    // Parameter: uint nCurrentPosition            当前块ID
    //************************************
    inline uint __CompositePacket(IN const char* pBuf,uint nSize,OUT char* pSendBuf,uint nCurrentPosition);
};


class NET_API CUDPFilesStrategyRecv : public CUDPFilesStrategy
{
public:
    CUDPFilesStrategyRecv();
    virtual ~CUDPFilesStrategyRecv();
public:
    BOOL BeginRecvFile(const char* pszFilePath,CUDPFilesStrategy_App* pCUDPFilesStrategy_App,uint nID);
    void EndRecvFile(void);
    //由APP层调用的RECV
    void RecvData(const char* pBuf,uint nBufSize);

protected:
private:
    static unsigned __stdcall __ThreadDelaySendACK(void* pData);//延迟发送ACK
    void __ThreadDelaySendACKFn(void);

    //解包
//    uint __ParsePacket(const char* pBuf,uint nSize,char* pSendBuf);
};
#include "UDPFilesStrategy.inl"
}//extern "C"