| -rw-r--r-- | PumpKINDlg.h | 327 |
1 files changed, 327 insertions, 0 deletions
diff --git a/PumpKINDlg.h b/PumpKINDlg.h new file mode 100644 index 0000000..42ae62d --- a/dev/null +++ b/PumpKINDlg.h @@ -0,0 +1,327 @@ +// PumpKINDlg.h : header file
+//
+
+//#define REVERSEBYTES(w) MAKEWORD(HIBYTE(w),LOBYTE(w));
+#define REVERSEBYTES(w) (htons((WORD)w))
+#define tftpoBSize "blksize"
+#define tftpoTSize "tsize"
+#define tftpoTOut "timeout"
+
+#define tftpoXResume "x-resume"
+
+struct tftp {
+ typedef CMapStringToString tftpOptions;
+ typedef WORD tftpLength;
+ typedef WORD tftpOpcode;
+ tftpLength length;
+ tftpOpcode opcode;
+ enum {
+ opRRQ = 1,
+ opWRQ = 2,
+ opDATA = 3,
+ opACK = 4,
+ opERROR = 5,
+ opOACK = 6,
+
+ errUndefined = 0,
+ errNotFound = 1,
+ errAccessViolation = 2,
+ errDiskFull = 3,
+ errIllegalOp = 4,
+ errUnknownTID = 5,
+ errFileExists = 6,
+ errNoUser = 7,
+ errOption = 8
+ };
+ struct tftpUNKNOWN {
+ BYTE data[1];
+ };
+ struct tftpRQ {
+ BYTE data[1];
+public:
+ void Set(LPCTSTR file,LPCTSTR type,tftpOptions* ops=NULL);
+ static UINT tftpSize(LPCTSTR file,LPCTSTR type,tftpOptions* ops=NULL);
+ };
+ struct tftpRRQ {
+ BYTE data[1];
+public:
+ void Set(LPCTSTR file,LPCTSTR type,tftpOptions* ops=NULL);
+ static UINT tftpSize(LPCTSTR file,LPCTSTR type,tftpOptions* ops=NULL);
+ };
+ struct tftpWRQ {
+ BYTE data[1];
+public:
+ void Set(LPCTSTR file,LPCTSTR type,tftpOptions* ops=NULL);
+ static UINT tftpSize(LPCTSTR file,LPCTSTR type,tftpOptions* ops=NULL);
+ };
+ struct tftpDATA {
+ WORD block;
+ BYTE data[1];
+ public:
+ WORD Block();
+ void SetBlock(WORD b);
+ static UINT tftpSize(UINT blkSize=512);
+ };
+ struct tftpACK {
+ WORD block;
+ public:
+ static UINT tftpSize();
+ void SetBlock(WORD b);
+ WORD Block();
+ };
+ struct tftpERROR {
+ typedef WORD tftpErrorCode;
+ tftpErrorCode code;
+ BYTE data[1];
+ public:
+ void SetCode(WORD c);
+ WORD Code();
+ static UINT tftpSize(LPCTSTR msg);
+ };
+ struct tftpOACK {
+ BYTE data[1];
+ public:
+ void Set(tftpOptions* ops);
+ static UINT tftpSize(tftpOptions* ops);
+ };
+ union tftpPacket {
+ tftpUNKNOWN m_UNKNOWN;
+ tftpRQ m_RQ;
+ tftpRRQ m_RRQ;
+ tftpWRQ m_WRQ;
+ tftpDATA m_DATA;
+ tftpACK m_ACK;
+ tftpERROR m_ERROR;
+ tftpOACK m_OACK;
+ } data;
+
+ void *udpBase(void) { return &opcode; }
+public:
+ static tftp* Copy(tftp *src);
+ CString errMessage();
+ void SetOpcode(WORD op);
+ void errSet(UINT code,LPCTSTR msg);
+ static tftp* Allocate(UINT tftpSize=0);
+ BOOL Send(CAsyncSocket *socket,SOCKADDR_IN* saddr);
+ CString rqType();
+ CString rqFileName();
+ UINT GetOptions(tftpOptions* ops);
+ tftp();
+ UINT Opcode();
+ BOOL Receive(CAsyncSocket* socket,UINT maxLength,SOCKADDR_IN *sin=NULL);
+};
+#define tftpHdrSize (sizeof(tftp)-sizeof(tftp::tftpPacket))
+#define tftpSlackSize sizeof(tftp::tftpLength)
+
+class CPumpKINDlg;
+class CListenSocket : public CAsyncSocket {
+public:
+ virtual void OnReceive(int nErrorCode);
+ CPumpKINDlg* m_Daddy;
+};
+
+typedef CList<tftp*,tftp*> CTFTPList;
+class CResolver;
+class CXferSocket : public CAsyncSocket {
+public:
+ UINT m__timeOut;
+ UINT m__blkSize;
+ tftp::tftpOptions m_Options;
+ LONG m_xferSize;
+ UINT m_timeOut;
+ UINT m_blkSize;
+ BOOL m_bRetry;
+ void SetTry(tftp *p=NULL);
+ tftp* m_Retry;
+ void OnRetry();
+ BOOL CheckBadRelativeness(LPCTSTR file);
+ CString ApplyRootGently(LPCTSTR fn);
+ CString m_HostName;
+ virtual void OnResolved();
+ virtual void OnFailedToResolve();
+ CResolver *m_wndResolver;
+ BYTE m_ResolveBuff[MAXGETHOSTSTRUCT];
+ 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();
+ 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:
+ 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
+{
+// Construction
+public:
+ CString m_bnwRequest;
+ CString m_bnwSuccess;
+ CString m_bnwAbort;
+ CBellsNWhistles m_bnw;
+ CTrayer *m_Trayer;
+ CTimeSpan m_RetryTimeOut;
+ virtual ~CPumpKINDlg();
+ CRetrier* m_Retrier;
+ virtual BOOL PreTranslateMessage(MSG* pMsg);
+ BOOL m_bShown;
+ BOOL m_bExiting;
+ void SaveSettings();
+ void LoadSettings();
+ void SetupButtons();
+ BITMAP m_bitmapBack;
+ CBitmap m_bmpBack;
+ UINT m_PromptTimeOut;
+ UINT m_SpeakPort;
+ void LogLine(UINT msgID);
+ CTimeMap m_LogTimes;
+ void LogLine(LPCTSTR str);
+ int m_LogLength;
+ enum {
+ subitemFile=0, subitemType, subitemPeer, subitemBytes, subitemTSize
+ };
+ int m_iWRQ;
+ int m_iRRQ;
+ CImageList m_Images;
+ CTIDMap m_Xfers;
+ CTimeSpan m_TFTPTimeOut;
+ enum {
+ rrqGiveAll=0,
+ rrqAlwaysConfirm,
+ rrqDenyAll
+ };
+ enum {
+ wrqTakeAll=0,
+ wrqConfirmIfExists,
+ wrqAlwaysConfirm,
+ wrqDenyAll
+ };
+ UINT m_RRQMode;
+ UINT m_WRQMode;
+ BOOL m_bTFTPSubdirs;
+ CString m_TFTPRoot;
+ UINT m_ListenPort;
+ UINT m_BlockSize;
+ CListenSocket m_Listener;
+ CPumpKINDlg(CWnd* pParent = NULL); // standard constructor
+
+// Dialog Data
+ //{{AFX_DATA(CPumpKINDlg)
+ enum { IDD = IDD_PUMPKIN_DIALOG };
+ CButton m_AbortCtl;
+ CButton m_OptionsCtl;
+ CListBox m_Log;
+ CListCtrl m_List;
+ //}}AFX_DATA
+
+ // ClassWizard generated virtual function overrides
+ //{{AFX_VIRTUAL(CPumpKINDlg)
+ protected:
+ virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV support
+ //}}AFX_VIRTUAL
+
+// Implementation
+protected:
+ CToolTipCtrl m_tooltip;
+ HICON m_hIcon;
+
+ // Generated message map functions
+ //{{AFX_MSG(CPumpKINDlg)
+ virtual BOOL OnInitDialog();
+ afx_msg void OnSysCommand(UINT nID, LPARAM lParam);
+ afx_msg void OnDestroy();
+ afx_msg void OnPaint();
+ afx_msg HCURSOR OnQueryDragIcon();
+ afx_msg int OnCreate(LPCREATESTRUCT lpCreateStruct);
+ afx_msg void OnOptions();
+ afx_msg void OnTimer(UINT nIDEvent);
+ afx_msg void OnExit();
+ afx_msg void OnPut();
+ afx_msg void OnGet();
+ afx_msg void OnDeleteallitemsConnections(NMHDR* pNMHDR, LRESULT* pResult);
+ afx_msg void OnDeleteitemConnections(NMHDR* pNMHDR, LRESULT* pResult);
+ afx_msg void OnInsertitemConnections(NMHDR* pNMHDR, LRESULT* pResult);
+ afx_msg void OnItemchangedConnections(NMHDR* pNMHDR, LRESULT* pResult);
+ afx_msg void OnAbort();
+ afx_msg void OnClose();
+ afx_msg void OnTrayShowpumpkinwindow();
+ afx_msg void OnTrayExit();
+ afx_msg void OnTrayAboutpumpkin();
+ afx_msg void OnTrayFetchfile();
+ afx_msg void OnTrayHelp();
+ afx_msg void OnTrayOptions();
+ afx_msg void OnTraySendfile();
+ afx_msg void OnWindowPosChanging(WINDOWPOS FAR* lpwndpos);
+ afx_msg void OnSelchangeLog();
+ afx_msg void OnTrayOpenfilesfolder();
+ afx_msg void OnDropFiles(HDROP hDropInfo);
+ virtual void OnCancel();
+ afx_msg void OnHelp();
+ //}}AFX_MSG
+ DECLARE_MESSAGE_MAP()
+};
|