VS2005控制台程序修改nb0文件
VS2005控制台程序修改nb0文件
我们要实现的功能就是通过CMD传递进来的值来在nb0文件末尾增加版本信息，新建控制台程序，自动生成的main函数如下，默认的代码非常简单：
int _tmain(int argc, _TCHAR * argv[])
{
return0;
}为了实现此功能，我们发现有两种实现的办法，差别在于使用TCHAR还是char，下面分别给出这两种方式的实现代码，这两种实现方式CMD端传递的参数都一样，如下：
图1
1. 采用TCHAR
(1) main函数
int _tmain(int argc, _TCHAR * argv[])
{
TCHAR*pImageVersion;
TCHAR*pFileDir;
TCHAR*pFileName;
pFileDir= argv[1];
pFileName= argv[2];
pImageVersion= argv[3];
PIMG_CHECK_VERSION pImg_check_version;
printf("argc= %d\r\n",argc);
for(inti = 0; i < argc; i++)
{
printf("argv[%d]:%s\r\n",i,argv[i]);
}
wcscpy(pImg_check_version->ImageVersion,pImageVersion);
AddDataToNb0File(pFileDir,pFileName,pImg_check_version);
return0;
}(2) AddDataToNb0File
BOOL AddDataToNb0File(TCHAR *pFileDir,TCHAR *pFileName, PIMG_CHECK_VERSION pCheckVersion)
{
BOOLret = FALSE;
TCHARtFilePath[FILE_NAME_LEN];
DWORDdwSize;
memset((PBYTE)tFilePath,0,sizeof(tFilePath));
wcscpy(tFilePath,pFileDir);
wcscat(tFilePath,pFileName);
HANDLEhfile = CreateFile(tFilePath, GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ, 0,OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);
if(hfile== INVALID_HANDLE_VALUE)
{
printf("AddDataToNb0File()GetLastError=0x%x\r\n",GetLastError());
gotoExit;
}
DWORDFileSize = GetFileSize (hfile, NULL) ;
if(FileSize == 0xFFFFFFFF)
{
gotoExit;
}
SetFilePointer(hfile,(FileSize- sizeof(IMG_CHECK_VERSION)),NULL,FILE_BEGIN);
WriteFile(hfile,(PBYTE)pCheckVersion,sizeof(IMG_CHECK_VERSION),&dwSize,NULL);
SetFilePointer(hfile,FileSize,NULL,FILE_BEGIN);
SetEndOfFile(hfile);
ret= TRUE;
Exit:
if(hfile!= INVALID_HANDLE_VALUE)
{
CloseHandle(hfile);
}
returnret;
}(3) IMG_CHECK_VERSION结构体的定义
typedef struct _IMG_CHECK_VERSION_
{
TCHARImageVersion[VERSION_LEN];
}IMG_CHECK_VERSION,*PIMG_CHECK_VERSION;见main函数的printf函数，我们的本意是输出传递进来参数的所有字符串内容的，但自己的输出却是：
图2
可见只输出对应参数的第一个字符，比如argv[2]是传递“stepldr.nb0”进来的，但只输出s，虽然我们想要的功能能够达到，但是调试起来如果想看这些信息比较不方便。
修改之前stepldr.nb0文件最后一段全部为0，我们用UltraEdit文件打开如下:
图3
修改之后如下：
图4
由图4可知stepldr.nb0从0x1ff8到0x1fff这8个字节的数据由之前的0x00改为0x31、0x00、0x32、0x00、0x33、0x00、0x34、0x00，这对应于图1中传递进来的1234，为什么1个字符会占用两个字节呢？是因为我们的IMG_CHECK_VERSION结构体成员ImageVersion类型是TCHAR(相当于wchar_t)，是宽字符，也就是说一个字符占用2个字节。
2. 采用char
(1) main函数
int main(int argc, char * argv[])
{
char*pImageVersion;
char*pFileDir;
char*pFileName;
pFileDir= argv[1];
pFileName= argv[2];
pImageVersion= argv[3];
PIMG_CHECK_VERSION pImg_check_version;
printf("argc= %d\r\n",argc);
for(inti = 0; i < argc; i++)
{
printf("argv[%d]:%s\r\n",i,argv[i]);
}
strcpy(pImg_check_version->ImageVersion,pImageVersion);
AddDataToNb0File(pFileDir,pFileName,pImg_check_version);
return0;
}(2) AddDataToNb0File函数
BOOL AddDataToNb0File(char *pFileDir, char*pFileName, PIMG_CHECK_VERSION pCheckVersion)
{
BOOLret = FALSE;
chartFilePath[FILE_NAME_LEN];
DWORDdwSize;
memset((PBYTE)tFilePath,0,sizeof(tFilePath));
strcpy(tFilePath,pFileDir);
strcat(tFilePath,pFileName);
DWORDdwFileDirNum = MultiByteToWideChar (CP_ACP, 0, tFilePath, -1, NULL, 0);
wchar_t*pFileDirAndName = new wchar_t[dwFileDirNum];
MultiByteToWideChar(CP_ACP, 0, tFilePath, -1, pFileDirAndName, dwFileDirNum);
printf("filedir and name : %s,dwFileDirNum=%d \r\n",pFileDirAndName,dwFileDirNum);
HANDLEhfile = CreateFile(pFileDirAndName, GENERIC_READ | GENERIC_WRITE,FILE_SHARE_READ, 0, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);
//HANDLEhfile = CreateFile((LPCTSTR)tFilePath, GENERIC_READ |GENERIC_WRITE, FILE_SHARE_READ, 0, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);
if(hfile== INVALID_HANDLE_VALUE)
{
printf("AddDataToNb0File()GetLastError=0x%x\r\n",GetLastError());
gotoExit;
}
DWORDFileSize = GetFileSize (hfile, NULL) ;
if(FileSize == 0xFFFFFFFF)
{
gotoExit;
}
SetFilePointer(hfile,(FileSize- sizeof(IMG_CHECK_VERSION)),NULL,FILE_BEGIN);
WriteFile(hfile,(PBYTE)pCheckVersion,sizeof(IMG_CHECK_VERSION),&dwSize,NULL);
SetFilePointer(hfile,FileSize,NULL,FILE_BEGIN);
SetEndOfFile(hfile);
ret= TRUE;
Exit:
if(hfile!= INVALID_HANDLE_VALUE)
{
CloseHandle(hfile);
}
returnret;
}这里要注意的是CreateFile函数的第1个参数类型要求是LPCTSTR，如果我们直接采用(LPCTSTR)tFilePath强制类型转换，GetLastError()返回值是2(ERROR_FILE_NOT_FOUND)，表示找不到文件，采用MultiByteToWideChar函数转换为宽字符后就OK了。
(3) IMG_CHECK_VERSION结构体
typedef struct _IMG_CHECK_VERSION_
{
charImageVersion[VERSION_LEN];
}IMG_CHECK_VERSION,*PIMG_CHECK_VERSION;采用这样的方式修改stepldr.nb0后，如下：
图5
可见char占用一个字节，综合比较具体是采用哪种，根据需要来选择。
遇到一个很奇怪的问题，我在A电脑上用VS2005创建的这个控制台程序，编译出来后在B电脑上运行是提示不能运行这个程序，然后我在B电脑上编译A电脑的这个工程，重新生成控制台程序就可以，然后我用B电脑上编译的在A电脑上也OK，目前不知道原因，先备注下。
浙公网安备 33010602011771号