Do not close RRQ socket until the last ACK is received.

git-svn-id: http://svn.klever.net/kin/pumpkin/trunk@152 fe716a7a-6dde-0310-88d9-d003556173a8

1 files changed, 2 insertions, 1 deletions

diff --git a/PumpKINDlg.h b/PumpKINDlg.h
index c5469f1..23c2657 100644
--- a/PumpKINDlg.h
+++ b/PumpKINDlg.h
@@ -291,77 +291,78 @@ public:
         virtual void Abort();

         void ResetTimeout();

                 enum{

                 stateNone, stateDeny, stateInit, stateOACK, stateXfer, stateFinish, stateClosing

         };

         int state;

         void Deny(UINT errCode,UINT errID);

         void Deny(CFileException* e);

         CString ApplyRoot(LPCTSTR fileName);

         void TurnSlashes(CString& fn,BOOL bBack=TRUE);

         virtual void Destroy(BOOL success=TRUE);

         void PostError(CFileException* e);

         void PostError(UINT errCode,UINT errID);

         void PostTFTP(tftp* p,BOOL retryable=FALSE);

         CXferSocket(CPumpKINDlg *daddy,LPCTSTR fileName,LPCTSTR type,SOCKADDR_IN* sin);

         CFile m_File;

         virtual ULONG GetACK();

         CString m_Type;

         CString m_FileName;

         CXferSocket();

         void UpdateList();

         void SetPeer(SOCKADDR_IN *sin);

         virtual BOOL OnTFTP(tftp* p) = 0;

         virtual void OnReceive(int nErrorCode);

-        void DoSelect();

+        void DoSelect(BOOL do_select=FALSE);

         SOCKADDR_IN m_Peer;

         virtual void OnSend(int nErrorCode);

         CPumpKINDlg* m_Daddy;

         CTFTPList m_Queue;

 

         DECLARE_DYNAMIC(CXferSocket)

 };

 class CWRQSocket : public CXferSocket {

 public:

         BOOL m_bResume;

         void OnHostKnown();

         virtual void OnResolved();

         UINT m_LastSlack;

         ULONG GetACK();

         void DoXfer();

         UINT m_ACK;

         BOOL SaveAs(CString& fn);

         BOOL RenameFile(CString& fn);

         BOOL m_Rename;

         BOOL ConfirmRequest();

         BOOL Create(LPCTSTR localFile=NULL,LPCTSTR hostName=NULL);

         CWRQSocket(CPumpKINDlg* daddy,LPCTSTR fileName,LPCTSTR type,SOCKADDR_IN *sin);

         BOOL OnTFTP(tftp* p);

 

         DECLARE_DYNAMIC(CWRQSocket)

 };

 class CRRQSocket : public CXferSocket {

 public:

+        UINT m_ACKtoClose;

         void OnHostKnown();

         virtual void OnResolved();

         BOOL ConfirmRequest();

         WORD m_LastSlack;

         UINT m_ACK;

         BOOL OnTFTP(tftp* p);

         ULONG GetACK(void);

         void DoXfer();

         CRRQSocket(CPumpKINDlg *daddy,LPCTSTR fileName,LPCTSTR type,SOCKADDR_IN *sin);

         BOOL Create(LPCTSTR localFile=NULL,LPCTSTR hostName=NULL);

 

         DECLARE_DYNAMIC(CRRQSocket)

 };

 

         typedef         CMap<SOCKET,SOCKET,CXferSocket*,CXferSocket*>CTIDMap;

 typedef CMap<CTime*,CTime*,CTime*,CTime*> CTimeMap;

 

 /////////////////////////////////////////////////////////////////////////////

 // CPumpKINDlg dialog

 

 class CTrayer;

 class CRetrier;

 class CPumpKINDlg : public CDialog

 {