Linux下GetModuleFileName的四種寫法
問題的起因是要把一個東東從Windows移植到基于Linux的嵌入式系統(tǒng)上。移植過程中，遇到了GetModuleFileName的問題。為了解決這個問題，花了不少的時間，也走了不少彎路。下面是整理的結(jié)果。
首先摘錄一段文字，來源《UNIX Programming FAQ 中文版》
?
1.14. 我怎樣找到進(jìn)程的相應(yīng)可執(zhí)行文件？
這個問題可以作為常見未回答問題(Frequently Unanswered Questions)的一個好候選，因?yàn)槭聦?shí)上提出這個問題經(jīng)常意味著程序的設(shè)計(jì)有缺陷。:)
你能作的最佳猜測(best guess)是通過審視argv[0]的值而獲得。如果它包括一個/，那么它可能是可執(zhí)行程序的絕對或相對(對于在程序開始時的當(dāng)前目錄而言)路徑。如果不包括，那么你可以仿效shell對于PATH變量的查詢來查找這個程序。但是，不能保證成功，因?yàn)橛锌赡軋?zhí)行程序時argv[0]是一些任意值，也不排除這個可執(zhí)行文件在執(zhí)行后可能已經(jīng)被更名或刪除的情況。
如果所有你想做的只是能打印一個和錯誤消息一起出現(xiàn)的合適的名字，那么最好的方法在main()函數(shù)中將argv[0]的值保存在全局變量中以供整個程序使用。雖然沒有保證說argv[0]的值總是有意義，但在大多數(shù)情況下它是最好的選擇。
人們詢問這個問題的最普通原因是意圖定位他們程序的配置文件。這被認(rèn)為是不好的形式；包含可執(zhí)行文件的目錄應(yīng)當(dāng)*只*包含可執(zhí)行文件，而且基于管理的要求經(jīng)常試圖將配置文件放置在和可執(zhí)行文件不同的文件系統(tǒng)。
試圖做這個的一個比較不普通但更正規(guī)的理由是允許程序調(diào)用exec()執(zhí)行它自己；這是一種用來完全重新初始化進(jìn)程(比如被用于一些sendmail的版本)的辦法(比如當(dāng)一個守護(hù)程序捕獲一個SIGHUP信號)。
完全同意上面的觀點(diǎn)的！所以并不建議在Linux下去實(shí)現(xiàn)GetModuleFileName，不過出于技術(shù)的角度，討論一下這個問題也是可以的。
?
好，下面說說茴字的四種寫法。哦，不，是GetModuleFileName的四種寫法。
GetModuleFileName的四種寫法
方法一：從PATH入手
說明：上文提供的思路
int GetModuleFileName1( char* sModuleName, char* sFileName, int nSize)
{
?int ret = -1;
?if( strchr( sModuleName,/ ) != NULL )
??strcpy( sFileName, sModuleName );
?else
?{
??char* sPath = getenv("PATH");
??char* pHead = sPath;
??char* pTail = NULL;
??while( pHead != NULL && *pHead != \x0 )
??{
???pTail = strchr( pHead, : );
???if( pTail != NULL )
???{
????strncpy( sFileName, pHead, pTail-pHead );
????sFileName[pTail-pHead] = \x0;
????pHead = pTail+1;
???}
???else
???{
????strcpy( sFileName, pHead );
????pHead = NULL;
???}
???
???int nLen = strlen(sFileName);
???if( sFileName[nLen] != / )sFileName[nLen] = /;
???strcpy( sFileName+nLen+1,sModuleName);
???if( 0 == access( sFileName, F_OK ) )
???{
????ret = 0;
????break;
???}
??}
?}
?return ret;
}
方法二：利用which命令
說明：? 與方法一相比，完全是換湯不換藥，之所以放在這里，是因?yàn)槠渲杏玫搅艘粋€小技巧：即利用popen()實(shí)現(xiàn)在代碼中執(zhí)行一段command，并得到其執(zhí)行的結(jié)果。
???????
int GetModuleFileName2( char* sModuleName, char* sFileName, int nSize)
{
?int ret = 0;
?if( strchr( sModuleName,/ ) != NULL )
??strcpy( sFileName, sModuleName );
?else
?{
??char sBuffer[256] = { 0, };
??char sCommand[256] = { 0, };
??FILE* fp = NULL;
??sprintf( sCommand, "which %s", sModuleName );
??if((fp = popen(sCommand,"r")) == NULL){
???ret = -1;
??}???
??else
??{
???sFileName[0] = \x0;
???while(!feof(fp)){
????if(fgets(sBuffer,nSize-1,fp) == NULL){
?????continue;
????}
????strcat( sFileName, sBuffer );
???}
??}
???? int nLen = strlen( sFileName );
??if( 0 == nLen )
???ret = -1;
??else
???if( sFileName[nLen-1] = \n ) sFileName[nLen-1] = \x0;
?}
?return ret;
}
方法二：獲取環(huán)境變量"_"
int GetModuleFileName3( char* sModuleName, char* sFileName, int nSize)
{
?int ret = -1;
??? char* p = getenv("_");
?if( p != NULL && strstr( p, sModuleName ) != NULL )
?{
??ret = 0;
??strcpy( sFileName, p );
?}
?return ret;
}
方法四：讀取/proc/self/maps
說明：? 細(xì)節(jié)請參閱 http://autopackage.org/docs/binreloc/
int GetModuleFileName4( char* sModuleName, char* sFileName, int nSize)
{
?int ret = -1;
?char sLine[1024] = { 0 };
?void* pSymbol = (void*)"";
?FILE *fp;
?char *pPath;
?fp = fopen ("/proc/self/maps", "r");
?if ( fp != NULL )
?{
??while (!feof (fp))
??{
???unsigned long start, end;
???if ( !fgets (sLine, sizeof (sLine), fp))
????continue;
???if ( !strstr (sLine, " r-xp ") || !strchr (sLine, /))
????continue;
???sscanf (sLine, "%lx-%lx ", &start, &end);
???if (pSymbol >= (void *) start && pSymbol < (void *) end)
???{
????char *tmp;
????size_t len;
????/* Extract the filename; it is always an absolute path */
????pPath = strchr (sLine, /);
????/* Get rid of the newline */
????tmp = strrchr (pPath, \n);
????if (tmp) *tmp = 0;
????/* Get rid of "(deleted)" */
????//len = strlen (pPath);
????//if (len > 10 && strcmp (pPath + len - 10, " (deleted)") == 0)
????//{
????//?tmp = pPath + len - 10;
????//?*tmp = 0;
????//}
????ret = 0;
????strcpy( sFileName, pPath );
???}
??}
??fclose (fp);
??
?}
?return ret;
}
測試代碼：
int main ( int argc, char** argv ) {
?char buffer[256]={0};
?getchar();
?printf ("ModuleFileName1 is: %s\n", GetModuleFileName1( argv[0], buffer, 256)==-1?"Not found!":buffer);
?printf ("ModuleFileName2 is: %s\n", GetModuleFileName2( argv[0], buffer, 256)==-1?"Not found!":buffer);
?printf ("ModuleFileName3 is: %s\n", GetModuleFileName3( argv[0], buffer, 256)==-1?"Not found!":buffer);
?printf ("ModuleFileName4 is: %s\n", GetModuleFileName4( argv[0], buffer, 256)==-1?"Not found!":buffer);
?return 0;
}
測試結(jié)果：
正常輸出結(jié)果：
ModuleFileName1 is: /home/coldcrane/bin/hello
ModuleFileName2 is: /home/coldcrane/bin/hello
ModuleFileName3 is: /home/coldcrane/bin/hello
ModuleFileName4 is: /home/coldcrane/bin/hello
程序執(zhí)行時目錄被移動
$ mv ~/bin ~/nib
輸出結(jié)果：
ModuleFileName1 is: Not found!
ModuleFileName2 is: Not found!
ModuleFil