datahideinbmpview.cpp

信息隐藏!!!这个程序的功能主要是在一副位图里面嵌套信息,从而达到信息隐藏的目的.

// DataHideInBMPView.cpp : implementation of the CDataHideInBMPView class
//

#include "stdafx.h"
#include "DataHideInBMP.h"

#include "DataHideInBMPDoc.h"
#include "DataHideInBMPView.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CDataHideInBMPView

IMPLEMENT_DYNCREATE(CDataHideInBMPView, CView)

BEGIN_MESSAGE_MAP(CDataHideInBMPView, CView)
	//{{AFX_MSG_MAP(CDataHideInBMPView)
	ON_COMMAND(ID_NORMAL, OnNormal)
	ON_COMMAND(ID_STRETCH, OnStretch)
	ON_COMMAND(ID_FILE_OPEN, OnFileOpen)
	ON_COMMAND(ID_EMBED, OnEmbed)
	ON_UPDATE_COMMAND_UI(ID_EMBED, OnUpdateEmbed)
	ON_COMMAND(ID_PICK, OnPick)
	ON_UPDATE_COMMAND_UI(ID_PICK, OnUpdatePick)
	//}}AFX_MSG_MAP
	// Standard printing commands
	ON_COMMAND(ID_FILE_PRINT, CView::OnFilePrint)
	ON_COMMAND(ID_FILE_PRINT_DIRECT, CView::OnFilePrint)
	ON_COMMAND(ID_FILE_PRINT_PREVIEW, CView::OnFilePrintPreview)
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CDataHideInBMPView construction/destruction

CDataHideInBMPView::CDataHideInBMPView()
{
	// TODO: add construction code here
	ImageStyle = 0;
    have_open_a_file = false;

	embed = false;
	pick = false;
	show_contrast = false;
}

CDataHideInBMPView::~CDataHideInBMPView()
{
}

BOOL CDataHideInBMPView::PreCreateWindow(CREATESTRUCT& cs)
{
	// TODO: Modify the Window class or styles here by modifying
	//  the CREATESTRUCT cs
	WNDCLASSEX wndcls;
	wndcls.cbSize=sizeof(WNDCLASSEX);
	HINSTANCE hinst=AfxGetInstanceHandle();
	if(CView::PreCreateWindow(cs)&&cs.lpszClass!=NULL)
	{
		HBRUSH hbkbrush=CreateSolidBrush(RGB(0,0,0));
		if(!GetClassInfoEx(hinst,cs.lpszClass,&wndcls))
			return FALSE;
		UnregisterClass(cs.lpszClass,hinst);
		wndcls.hbrBackground=hbkbrush;
		RegisterClassEx(&wndcls);  
		return TRUE;
	}
	else
		return FALSE;
	return CView::PreCreateWindow(cs);
}

/////////////////////////////////////////////////////////////////////////////
// CDataHideInBMPView drawing

void CDataHideInBMPView::OnDraw(CDC* pDC)
{
	CDataHideInBMPDoc* pDoc = GetDocument();
	ASSERT_VALID(pDoc);
	// TODO: add draw code for native data here
	RECT Rect;
	GetClientRect( &Rect );

	if (!show_contrast)
		m_BMPHider.Draw( pDC, 0, 0, Rect.right, Rect.bottom, ImageStyle );
	else
	{
		m_BMPHider.DrawContrast(pDC, Rect.right, Rect.bottom);
		pDC->SetBkColor((0,0,0));
		pDC->SetTextColor((255,255,255));
		pDC->TextOut(0, Rect.bottom-30, "左图为原始图，右图为嵌入了隐藏信息的图");
	}
}

/////////////////////////////////////////////////////////////////////////////
// CDataHideInBMPView printing

BOOL CDataHideInBMPView::OnPreparePrinting(CPrintInfo* pInfo)
{
	// default preparation
	return DoPreparePrinting(pInfo);
}

void CDataHideInBMPView::OnBeginPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
{
	// TODO: add extra initialization before printing
}

void CDataHideInBMPView::OnEndPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
{
	// TODO: add cleanup after printing
}

/////////////////////////////////////////////////////////////////////////////
// CDataHideInBMPView diagnostics

#ifdef _DEBUG
void CDataHideInBMPView::AssertValid() const
{
	CView::AssertValid();
}

void CDataHideInBMPView::Dump(CDumpContext& dc) const
{
	CView::Dump(dc);
}

CDataHideInBMPDoc* CDataHideInBMPView::GetDocument() // non-debug version is inline
{
	ASSERT(m_pDocument->IsKindOf(RUNTIME_CLASS(CDataHideInBMPDoc)));
	return (CDataHideInBMPDoc*)m_pDocument;
}
#endif //_DEBUG

/////////////////////////////////////////////////////////////////////////////
// CDataHideInBMPView message handlers

void CDataHideInBMPView::OnNormal() 
{
	// TODO: Add your command handler code here
	ImageStyle = 0;

	CMenu *pMnu = AfxGetMainWnd()->GetMenu( );
	pMnu->CheckMenuItem(ID_STRETCH,MF_UNCHECKED);
	pMnu->CheckMenuItem(ID_NORMAL,MF_CHECKED);
	
	InvalidateRect( NULL, TRUE );
	UpdateWindow();	
}

void CDataHideInBMPView::OnStretch() 
{
	// TODO: Add your command handler code here
	ImageStyle = 1;
	
	CMenu *pMnu = AfxGetMainWnd()->GetMenu( );
	pMnu->CheckMenuItem(ID_STRETCH,MF_CHECKED);
	pMnu->CheckMenuItem(ID_NORMAL,MF_UNCHECKED);
	
	InvalidateRect( NULL, TRUE );
	UpdateWindow();	
}

void CDataHideInBMPView::OnFileOpen() 
{
	// TODO: Add your command handler code here
	show_contrast = false;   //非对比显示
	static char szFilter[] = "BMP Files(*.BMP)|*.BMP||";

	CFileDialog FileDlg( TRUE, NULL, NULL,
		OFN_HIDEREADONLY, szFilter );

	if( FileDlg.DoModal() == IDOK &&
		m_BMPHider.Load( FileDlg.GetPathName() ) )
	{
		InvalidateRect( NULL, TRUE );
		UpdateWindow();
		have_open_a_file=true;
		if (m_BMPHider.tag == 0)//该图无隐藏信息
		{
			embed = true;
			pick = false;  //菜单亮暗控制
		}//
		else
		{
			embed = false;
			pick = true;  //菜单亮暗控制
		}//
	}	
}

void CDataHideInBMPView::OnEmbed() 
{
	// TODO: Add your command handler code here
	//弹出打开对话框，用户选择嵌入的文件

	static char szFilter[] = "All Files(*.*)|*.*||";

	CFileDialog FileDlg( TRUE, NULL, NULL,
		OFN_HIDEREADONLY, szFilter );

	if( FileDlg.DoModal() == IDOK &&
		m_BMPHider.LoadEmbFile( FileDlg.GetPathName() ) )
	{ 
		if (m_BMPHider.bitmap_size / m_BMPHider.embfile_size < 8)
		{
			AfxMessageBox("文件太大，无法嵌入！");
		}
		else
		{
			show_contrast = true;
			m_BMPHider.BackUpDib(); //图片原始数据备份供对比显示
			m_BMPHider.Embed();   //嵌入
			
			Invalidate(true);//刷屏，显示加入隐藏信息后的图像
			
			static char szFilter[] = "BMP Files(*.BMP)|*.BMP||";
			CFileDialog FileDlg( FALSE, "bmp", NULL,                //保存
				OFN_HIDEREADONLY|OFN_OVERWRITEPROMPT, szFilter );
			
			if( FileDlg.DoModal() == IDOK )
			{
				m_BMPHider.Save( FileDlg.GetPathName() );
			}
			embed = false;
		}
		
	}	
}

void CDataHideInBMPView::OnUpdateEmbed(CCmdUI* pCmdUI) 
{
	// TODO: Add your command update UI handler code here
	pCmdUI->Enable(embed);	
}

void CDataHideInBMPView::OnPick() 
{
	// TODO: Add your command handler code here
	//提取完毕，弹出保存对话框，用户进行保存
	static char szFilter[] = "All Files(*.*)|*.*||";
	CFileDialog FileDlg( FALSE, "", NULL,                //保存
		OFN_HIDEREADONLY|OFN_OVERWRITEPROMPT, szFilter );
				
	if (FileDlg.DoModal() == IDOK)
	{	
		m_BMPHider.Pick();
		m_BMPHider.SavePicked(FileDlg.GetPathName());
		pick = false;
	}	
}

void CDataHideInBMPView::OnUpdatePick(CCmdUI* pCmdUI) 
{
	// TODO: Add your command update UI handler code here
	pCmdUI->Enable(pick);	
}