1. 前言

如果需要在程序內(nèi)部獲取當(dāng)前運(yùn)行程序的名稱，可以使用如下 4 種方式：

1. argv[0]；
2. /proc/self/comm、/proc/self/cmdline；
3. __progname；
4. readlink("/proc/self/exe")；

下面說一下這幾種方式適用的場景。

2. 適用場景

2.1 僅獲取程序名稱

如果僅僅想得到可執(zhí)行程序的名稱，不需要目錄信息，可以使用 __progname 或 /proc/self/comm。

//__progname
extern char *__progname;
printf("program name:%sn", __progname);

// /proc/slef/comm
char filename[64] = {0};
int fd = -1;
fd = open("/proc/self/comm", O_RDONLY);
read(fd, filename, sizeof(filename));
printf("program name:%sn", filename);

2.2 獲取終端輸入

如果想得到程序執(zhí)行時(shí)在 終端輸入的命令行 ，可以使用 argv[0] 或 /proc/self/cmdline。

//argv[0]
printf("terminal input:%sn", argv[0]);

// /proc/self/cmdline
char filename[128] = {0};
int fd = -1;
fd = open("/proc/self/cmdline", O_RDONLY);
read(fd, filename, sizeof(filename));
printf("terminal input:%sn", filename);

2.3 獲取程序絕對路徑

如果想得到可執(zhí)行程序所在的具體位置，那么就使用 /proc/self/exe。而且，即使在終端運(yùn)行的是可執(zhí)行程序的 軟鏈接 ，/proc/self/exe 返回的也是可執(zhí)行程序自身的位置。

// /proc/self/exe
char filename[128] = {0};
readlink("/proc/self/exe", filename, sizeof(filename));
printf("program absolute position:%sn", filename);

3. 源碼

main.c：

#include < stdio.h >
#include < stdlib.h >
#include < stdint.h >
#include < string.h >
#include < sys/types.h >
#include < sys/stat.h >
#include < fcntl.h >
#include < unistd.h >

extern char *__progname;

static void file_content_read(const char *filename, char *buf, uint32_t buf_size)
{
    int32_t fd = -1;

    fd = open(filename, O_RDONLY);
    if(-1 == fd)
    {
        printf("open %s failedn", filename);
        return;
    }

    read(fd, buf, buf_size);
}


int main(int argc, char *argv[])
{
    char file_info[64] = {0};

    printf("program name:%sn", __progname);

    memset(file_info, 0, sizeof(file_info));
    file_content_read("/proc/self/comm", file_info, sizeof(file_info));
    printf("program name:%sn", file_info);


    printf("terminal input:%sn", argv[0]);

    memset(file_info, 0, sizeof(file_info));
    file_content_read("/proc/self/cmdline", file_info, sizeof(file_info));
    printf("terminal input:%sn", file_info);


    memset(file_info, 0, sizeof(file_info));
    readlink("/proc/self/exe", file_info, sizeof(file_info));
    printf("program absolute position:%sn", file_info);

    return 0;
}

編譯：

gcc main.c -o proc-name

新建鏈接文件：

ln -s proc-name sy-proc-name

4. 運(yùn)行結(jié)果

4.1 運(yùn)行程序本身

4.2 運(yùn)行鏈接程序

5. 總結(jié)

多數(shù)情況下只需獲取程序的名稱，__progname，足矣。

tips

得到程序的名稱有什么用呢？

可以根據(jù)程序中獲取到的程序名稱的不同執(zhí)行不同的功能 ，大名鼎鼎的 busybox 就是這么干的！其原理就是針對可執(zhí)行程序創(chuàng)建不同名稱的鏈接文件，鏈接文件執(zhí)行時(shí)在程序中可以得到鏈接文件的名稱，以此名稱為判斷條件執(zhí)行不同的功能。

busybox

busybox --help ：

自實(shí)現(xiàn)

main.c ：

#include < stdio.h >
#include < stdlib.h >
#include < string.h >

extern char *__progname;

int main(int argc, char *argv[])
{
    char file_info[64] = {0};

    printf("program name:%sn", __progname);

    if(0 == strcmp(__progname, "self-ping"))
    {
        printf("this is self ping commandn");
    }
    else if(0 == strcmp(__progname, "self-gzip"))
    {
        printf("this is self gzip commandn");
    }
    else
    {
        printf("should create a linkn");
    }

    return 0;
}

編譯：

gcc main.c -o simu-bb

新建鏈接 ：

ln -s simu-bb self-ping
ln -s simu-bb self-gzip

運(yùn)行結(jié)果 ：