From 9938ea772d372ba955b7bc1f1c48a062c3a36f8a Mon Sep 17 00:00:00 2001
From: Michael Krelin <hacker@klever.net>
Date: Mon, 05 Jul 2004 01:53:09 +0000
Subject: initial commit into svn repository

git-svn-id: http://svn.klever.net/kin/pumpkin/trunk@1 fe716a7a-6dde-0310-88d9-d003556173a8
---
(limited to 'PumpKINDlg.h')

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()
+};
--
cgit v0.9.0.2