LSB信息隐藏之BMP图像数据的读取

#include "stdafx.h"
#include "LSB_Coder.h"


//-------------------------------------------------------------------------------------------
//读图像的位图数据、宽、高、颜色表及每像素位数等数据进内存，存放在相应的全局变量中
bool LSB::readBmp(char *bmpName) 
{
	FILE *fp=fopen(bmpName,"rb");//二进制读方式打开指定的图像文件


	if(fp==0) return 0;


	//跳过位图文件头结构BITMAPFILEHEADER


	fseek(fp, sizeof(BITMAPFILEHEADER),0);


	//定义位图信息头结构变量，读取位图信息头进内存，存放在变量head中


	BITMAPINFOHEADER head;  


	fread(&head, sizeof(BITMAPINFOHEADER), 1,fp); //获取图像宽、高、每像素所占位数等信息


	bmpWidth = head.biWidth;


	bmpHeight = head.biHeight;


	biBitCount = head.biBitCount;//定义变量，计算图像每行像素所占的字节数（必须是4的倍数）


	int lineByte=(bmpWidth * biBitCount/8+3)/4*4;//灰度图像有颜色表，且颜色表表项为256


	if(biBitCount==8)
	{


		//申请颜色表所需要的空间，读颜色表进内存


		pColorTable=new RGBQUAD[256];


		fread(pColorTable,sizeof(RGBQUAD),256,fp);


	}


	//申请位图数据所需要的空间，读位图数据进内存


	pBmpBuf= new unsigned char[lineByte * bmpHeight];


	fread(pBmpBuf,1,lineByte * bmpHeight,fp);


	fclose(fp);//关闭文件


	return 1;//读取文件成功
}


//-----------------------------------------------------------------------------------------
//给定一个图像位图数据、宽、高、颜色表指针及每像素所占的位数等信息,将其写到指定文件中
//bool LSB::saveBmp(char *bmpName, unsigned char *imgBuf, int width, int height, int biBitCount, RGBQUAD *pColorTable)
bool LSB::saveBmp(char *bmpName)
{


	//如果位图数据指针为0，则没有数据传入，函数返回


	if(!pBmpBuf)


		return 0;


	//颜色表大小，以字节为单位，灰度图像颜色表为1024字节，彩色图像颜色表大小为0


	int colorTablesize=0;


	if(biBitCount==8)


		colorTablesize=1024;


	//待存储图像数据每行字节数为4的倍数


	int lineByte=(bmpWidth * biBitCount/8+3)/4*4;
	
	//以二进制写的方式打开文件


	FILE *fp=fopen(bmpName,"wb");


	if(fp==0) return 0;


	//申请位图文件头结构变量，填写文件头信息


	BITMAPFILEHEADER fileHead;


	fileHead.bfType = 0x4D42;//bmp类型


	//bfSize是图像文件4个组成部分之和


	fileHead.bfSize= sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER)


		+ colorTablesize + lineByte*bmpHeight;


	fileHead.bfReserved1 = 0;


	fileHead.bfReserved2 = 0;


	//bfOffBits是图像文件前3个部分所需空间之和


	fileHead.bfOffBits=54+colorTablesize;


	//写文件头进文件


	fwrite(&fileHead, sizeof(BITMAPFILEHEADER),1, fp);


	//申请位图信息头结构变量，填写信息头信息


	BITMAPINFOHEADER head; 


	head.biBitCount=biBitCount;


	head.biClrImportant=0;


	head.biClrUsed=0;


	head.biCompression=0;


	head.biHeight=bmpHeight;


	head.biPlanes=1;


	head.biSize=40;


	head.biSizeImage=lineByte*bmpHeight;


	head.biWidth=bmpWidth;


	head.biXPelsPerMeter=0;


	head.biYPelsPerMeter=0;


	//写位图信息头进内存


	fwrite(&head, sizeof(BITMAPINFOHEADER),1, fp);


	//如果灰度图像，有颜色表，写入文件 


	if(biBitCount==8)


		fwrite(pColorTable, sizeof(RGBQUAD),256, fp);


	//写位图数据进文件


	fwrite(pBmpBuf, bmpHeight*lineByte, 1, fp);


	//关闭文件


	fclose(fp);


    //清除缓冲区，pBmpBuf和pColorTable是全局变量，在文件读入时申请的空间
	//delete []pBmpBuf;
	//if(biBitCount==8)
	//delete []pColorTable;


	return 1;


}




//-------------------------------------------------------------------------------------------
//一下为信息隐藏部分
void LSB::LSBCoder(const char*textFileName)
{
	ifstream textFile;
	textFile.open(textFileName,ios::in | ios::binary);
	//_infoStream.open(_originalPicPath,ios::in | ios::binary);
	textFile.seekg(0,textFile.end);
	long textFileLength = textFile.tellg();


	BYTE* pTextFile = new BYTE[textFileLength+1];
	cout<<"隐藏时文件长度:"<<textFileLength<<endl;
	textFile.seekg(0,textFile.beg);
	textFile.read((char*)pTextFile,textFileLength);
	textFile.close();


	 BYTE textData;
	 for(int i=0,k=0; i< textFileLength; ++i)
	 {
		 for(int j=0; j<8; ++j)
		 {
			 textData = pTextFile[i]>>j;
			 textData = textData&0x01;
			 if(textData==0)
			 {
				  pBmpBuf[k+32] = pBmpBuf[k+32]&0xfe;
			 }
			 else
			 {
				 pBmpBuf[k+32] = pBmpBuf[k+32]|0x01;
			 }
			 ++k;
		 }
	 }
	cout<<"信息隐藏完毕"<<endl;
	long length;
	for(int i=0; i<32; ++i)
	{
		length = textFileLength>>i;
		length = length&0x00000001;
		if(length==0)
		{
			pBmpBuf[i] = pBmpBuf[i]&0x1e;
		}
		else
		{
			pBmpBuf[i] =pBmpBuf[i]|0x01;
		}


	}
}
//------------------------------------------------
//  解码基于LSB的信息隐藏
//-------------------------------------------------
void  LSB::LSBDecoder(const char* textFileName)
{
	long length = 0x00000000;
	BYTE bit;
	//获取txt文件长度
	for(int i=0; i<32; ++i)
	{
		 bit =pBmpBuf[i]&0x01;
		  if(bit==0)
		  {
			   length = length&0x7fffffff;
		  }
		  else
		  {
				  length = length|0x80000000;
		  }
		   if (i<31)    length = length>>1;
	}
	cout<<"解码时文件长度:"<<length<<endl;
	 //开始解码
	 BYTE* pTextFile = new BYTE[length];
	 BYTE textData;
	 for(int i=0,k=0; i<length*8; ++i)
	 {
		 if(i && i%8==0){++k;}
		 textData = pBmpBuf[i+32]&0x01;
		 if(textData==0)
		 {
			  pTextFile[k] = pTextFile[k]&0x7f;
		 }
		 else
		 {
			  pTextFile[k] = pTextFile[k]|0x80;
		 }
		 if (i%8 != 7) pTextFile[k] = pTextFile[k]>>1;
	 }
	 cout<<"解码完毕"<<endl;


	  ofstream textFile;
	  textFile.open(textFileName,ios::out | ios::binary);
	  textFile.write((char*)pTextFile,length);
	  textFile.close();
	  delete pTextFile;
	  //char savepicPath[]="Apple_Coder.bmp";
	  //saveBmp(savepicPath, pBmpBuf, bmpWidth, bmpHeight, biBitCount, pColorTable);
}