本文共 5343 字，大约阅读时间需要 17 分钟。

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，目前不知道原因，先备注下。

转载地址：https://loongembedded.blog.csdn.net/article/details/29797783 如侵犯您的版权，请留言回复原文章的地址，我们会给您删除此文章，给您带来不便请您谅解！