root/honeybow/trunk/mwwatcher/src/MwWatcherDlg.h

// MwWatcherDlg.h
//
// Header file
// Copyright (C) 2006 Artemis: Chinese Honeynet Project
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
//
/////////////////////////////////////////////////////////////////////////////
//

#if !defined(AFX_MWWATCHERDLG_H__2639544C_D040_493C_A9AC_80413B2861E6__INCLUDED_)
#define AFX_MWWATCHERDLG_H__2639544C_D040_493C_A9AC_80413B2861E6__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

#define WM_SHOWTASK             (WM_USER + 201)
#define MAX_DIR_WATCH   10

#define IDT_RUN_TIMER           1
#define IDT_GLOBAL_TIMER        2

/////////////////////////////////////////////////////////////////////////////
// 
#include "DirectoryChanges.h"

class CDirectoryChangeHandler_ListBox : public CDirectoryChangeHandler
{
public:
        CDirectoryChangeHandler_ListBox(CListBox & list_box);
        /*      : CDirectoryChangeHandler(),
                m_listBox( list_box ){}*/
        ~CDirectoryChangeHandler_ListBox();
protected:
        //These functions are called when the directory to watch has had a change made to it
        void On_FileAdded(const CString & strFileName)
        {
                CString strTemp(_T("File Added: ") + strFileName);
                AddNotification(strTemp);

                //m_listBox.AddString((LPCTSTR)strTemp/*_T("File Added: ") + strFileName*/);
                //SetHorizontalBar((LPCTSTR)strTemp);

        }
        void On_FileRemoved(const CString & strFileName)
        {
                CString strTemp(_T("File Removed: ") + strFileName);
                AddNotification(strTemp);

                //m_listBox.AddString(_T("File Removed: ") + strFileName);
                //SetHorizontalBar(_T("File Removed: ") + strFileName);
        }
        void On_FileModified(const CString & strFileName)
        {
                CString strTemp(_T("File Modified: ") + strFileName);
                AddNotification(strTemp);

                //m_listBox.AddString(_T("File Modified: ") + strFileName);
                //SetHorizontalBar(_T("File Modified: ") + strFileName);
        }
        void On_FileNameChanged(const CString & strOldFileName, const CString & strNewFileName)
        {
                CString strTemp(_T("File name changed from: ") + strOldFileName + _T(" to: ") + strNewFileName);
                AddNotification(strTemp);

                //m_listBox.AddString(_T("File name changed from: ") + strOldFileName + _T(" to: ") + strNewFileName);
                //SetHorizontalBar(_T("File name changed from: ") + strOldFileName + _T(" to: ") + strNewFileName);
        }

        void On_WatchStarted(DWORD dwError, const CString & strDirectoryName)
        {
                if( dwError == 0 )
                {
                        CString strTemp( _T("Directory Watch Started: ") + strDirectoryName);
                        AddNotification(strTemp);

                        //m_listBox.AddString( _T("Directory Watch Started: ") + strDirectoryName);
                        //SetHorizontalBar(_T("Directory Watch Started: ") + strDirectoryName);
                }
                else
                {
                        CString strTemp( _T("Directory watch failed: ") + strDirectoryName);
                        AddNotification(strTemp);

                        //m_listBox.AddString( _T("Directory watch failed: ") + strDirectoryName);
                        //SetHorizontalBar(_T("Directory watch failed: ") + strDirectoryName);
                }
        }
        void On_WatchStopped(const CString & strDirectoryName)
        {
                //
                //      NOTE:  Be careful ! On_WatchStopped can be called AFTER this list box has
                //      been destoroyed.  Be carefull whenever interacting w/ windows with this function
                //      because the window may not be valid when this function is executed for the last time.
                //
                //
                if( ::IsWindow( m_listBox.GetSafeHwnd() ) )
                {
                        CString strTemp(_T("Stopped watching directory: ") + strDirectoryName);
                        AddNotification(strTemp);

                        //m_listBox.AddString(_T("Stopped watching directory: ") + strDirectoryName);
                        //SetHorizontalBar(_T("Stopped watching directory: ") + strDirectoryName);
                }
                else
                {
                        TRACE(_T("NOTE: m_listBox is not a valid window! This has probably been called as the result of program shutdown sometime after CDirWatcherDlg::OnBtnMonitor2() was used to watch a direcotry.\n"));
                        TRACE(_T("NOTE: Be sure that you are careful in your implementation of this function to watch out for the same condition if you are dealing w/ windows.\n"));
//                      MessageBeep(static_cast<UINT>(-1));
//                      MessageBeep(static_cast<UINT>(-1));
//                      MessageBeep(static_cast<UINT>(-1));
                        
                        //
                        //      See the comments in DirectoryChanges.h for the CDirectoryChangeHandler::On_WatchStopped() function.
                        //
                        //
                }
        }
        void On_FileUploaded(const CString & strFileName)
        {
                CString strTemp(_T("File Uploaded: ") + strFileName);
                AddNotification(strTemp);

                //m_listBox.AddString(_T("File Uploaded: ") + strFileName);
                //SetHorizontalBar(_T("File Uploaded: ") + strFileName);
        }

        CListBox & m_listBox;
        HANDLE  hLogFile;
        HANDLE  hMutex;
        BOOL    bIsLog;
        

private:
        void SetHorizontalBar();
        void AddNotification(const CString &strMessage);
};

/////////////////////////////////////////////////////////////////////////////
// CMwWatcherDlg dialog

class CMwWatcherDlg : public CDialog
{
// Construction
public:
        CMwWatcherDlg(CWnd* pParent = NULL);    // standard constructor
        
// Dialog Data
        //{{AFX_DATA(CMwWatcherDlg)
        enum { IDD = IDD_MWWATCHER_DIALOG };
        CListBox        m_SettingBox;
        CListBox        m_NotifyBox;
                // NOTE: the ClassWizard will add data members here
        //}}AFX_DATA

        // ClassWizard generated virtual function overrides
        //{{AFX_VIRTUAL(CMwWatcherDlg)
        protected:
        virtual void DoDataExchange(CDataExchange* pDX);        // DDX/DDV support
        //}}AFX_VIRTUAL

// Implementation
protected:
        HICON   m_hIcon;
        LONG    m_old_cx;
        LONG    m_old_cy;

        // Generated message map functions
        //{{AFX_MSG(CMwWatcherDlg)
        virtual BOOL OnInitDialog();
        afx_msg void OnSysCommand(UINT nID, LPARAM lParam);
        afx_msg void OnPaint();
        afx_msg HCURSOR OnQueryDragIcon();
        virtual void OnCancel();
        virtual void OnOK();
        afx_msg void OnHide();
        afx_msg LRESULT OnBtnShowTask(WPARAM wParam,LPARAM lParam);
        afx_msg void OnRestart();
        afx_msg void OnChangeSettings();
        afx_msg void OnTimer(UINT nIDEvent);
        afx_msg void OnClear();
        afx_msg void OnSize(UINT nType, int cx, int cy);
        //}}AFX_MSG
        DECLARE_MESSAGE_MAP()

        CDirectoryChangeWatcher                 m_DirWatcher;
        //CDirectoryChangeHandler_ListBox m_DirChangeHandler;
        
        DWORD   GetChangeFlags(int n);
        BOOL    GetWatchSubDirFlag(int n);
        void    GetDirectoryToMonitor(int n);
        BOOL    GetFilterFlag(void);
        void    GetRebootSetting(void);
        BOOL    GetUploadMethod(void);
        void    GetDelayTime(void);

        BOOL    InitService(int n);
        void    InitFtp();

        void    FtpInit();//initialize the ftp commond file
        void    FtpPut();//upload file to ftp

        void    CopyPut();//copy sample to a directory

        void    Reboot();

        BOOL    isInWhiteList(const CString &strFileName);

        CString strInitFilePath;

private:
        CString m_strDirectoryToMonitor[MAX_DIR_WATCH];
        CString strInitCommand;
        CString strGFtpAddress;
        CString strGFtpUserName;
        CString strGFtpUserPasswd;
        CString strGFtpPath;
        DWORD   dwGDelay;
        LONG    lRebootTime;
        BOOL    bSetReboot;
        BOOL    bFtpPut;
        HANDLE  hLogFile;
        CHAR    HomePath[MAX_PATH+1];
};

//{{AFX_INSERT_LOCATION}}
// Microsoft Visual C++ will insert additional declarations immediately before the previous line.

#endif // !defined(AFX_MWWATCHERDLG_H__2639544C_D040_493C_A9AC_80413B2861E6__INCLUDED_)