Linux如何獲取進(jìn)程的基地信息

1. 問(wèn)題

執(zhí)行程序時(shí)，與程序同目錄下存在一 config/ 目錄，其中存放了 json 格式的配置文件。進(jìn)行部署時(shí)需要將程序和 config/ 目錄放在同一位置。那么，問(wèn)題來(lái)了，如何保證不管這個(gè)程序部署到什么位置，不管以什么方式運(yùn)行，程序自己都能訪問(wèn)到配置文件呢？

解決問(wèn)題之前，先說(shuō)一下進(jìn)程的工作目錄——Current Working Directory。

2. 什么是進(jìn)程的工作目錄

進(jìn)程的工作目錄是指進(jìn)程的調(diào)用者在啟動(dòng)進(jìn)程對(duì)應(yīng)的程序時(shí)， 調(diào)用者所在的目錄 。比如說(shuō)，在 /home/sdc/ 目錄下有一個(gè)可執(zhí)行程序 hello-world。如果在根目錄下啟動(dòng) hello-world，那么 hello-world 進(jìn)程的工作目錄為根目錄 / 。如果在 /home/sdc 目錄下啟動(dòng) hello-world，那么其工作目錄為 /home/sdc。

Linux C 中，有兩個(gè)函數(shù)可以獲取進(jìn)程工作目錄：getcwd() 和 readlink()。

2.1 getcwd()

man 3 getcwd()：https://man7.org/linux/man-pages/man2/getcwd.2.html

#include < unistd.h >
char *getcwd(char *buf, size_t size);

該函數(shù)將獲取到的進(jìn)程工作目錄的絕對(duì)路徑存儲(chǔ)在形參 buf 中。同時(shí)也會(huì)返回一個(gè)指針，指針指向的內(nèi)容和 buf 存儲(chǔ)的內(nèi)容相同。如果路徑的長(zhǎng)度比 size 大，那么該函數(shù)返回為 NULL。

2.2 readlink()

man 2 readlink：https://www.man7.org/linux/man-pages/man2/readlink.2.html

#include < unistd.h >
ssize_t readlink(const char *pathname, char *buf, size_t bufsiz);

該函數(shù)用于獲取符號(hào)鏈接所指的文件。借助 /proc/self/cwd，即 pathname = "/proc/self/cwd"，readlink() 即可獲得當(dāng)前進(jìn)程的工作目錄。

該函數(shù)同樣將進(jìn)程工作目錄的絕對(duì)路徑存儲(chǔ)在形參 buf 中，并返回存儲(chǔ)的路徑長(zhǎng)度。但是，buf 中存儲(chǔ)路徑不帶字符串結(jié)束符 '?'。而且，如果返回的長(zhǎng)度和 bufsiz 相同，那么路徑信息很可能被截取了。

2.3 簡(jiǎn)單例程

hello-world.c:

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

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

    getcwd(work_dir, sizeof(work_dir)))
    printf("getcwd:%sn", work_dir);

    memset(work_dir, 0, sizeof(work_dir));

    dir_size = readlink("/proc/self/cwd", work_dir, sizeof(work_dir));
    work_dir[dir_size] = '?';
    printf("readlink:%sn", work_dir);

    return 0;
}

執(zhí)行結(jié)果：

3. 獲取進(jìn)程對(duì)應(yīng)程序的絕對(duì)路徑

如果想要解決文章開(kāi)頭提出的問(wèn)題，那么進(jìn)程需要獲取程序所在的絕對(duì)路徑。該功能通過(guò) readlink(const char *pathname, char *buf, size_t bufsiz)實(shí)現(xiàn)。不過(guò) pathname 不再是 /proc/self/cwd，而是 /proc/self/exe 。

開(kāi)箱即用代碼 ：

#include < stdio.h >
#include < stdlib.h >
#include < string.h >
#include < unistd.h >
#include < stdint.h >
#include < fcntl.h >

#define CONFIG_FILENAME    "./config/psl.json"

//get current process elf absolute path
static int32_t curr_elf_abs_path_get(int8_t *path_buf, int32_t buf_size)
{
    ssize_t len = 0;
    int32_t i = 0;

    len = readlink("/proc/self/exe", path_buf, buf_size);
    if(-1 == len)
    {
        perror("readlinkn");
        return -1;
    }

    if(len == buf_size)
    {
        printf("Warn:path may be truncatedn");
    }

    //from last to head, find first '/'
    for(i = len; i > 0; i--)
    {
        if(path_buf[i] == '/')
        {
            path_buf[i + 1] = '?';
            break;
        }
    }

    return 0;
}


int main(int argc, char *argv[])
{
    int8_t elf_path[128] = {0};
    int8_t config_file_path[256] = {0};
    int fd = -1;
    int32_t i = 0;
    int32_t ret = 0;

    ret = curr_elf_abs_path_get(elf_path, sizeof(elf_path));
    if(0 != ret)
    {
        printf("get exe path failedn");
        return -1;
    }
    printf("current process exe absolute path:%sn", elf_path);
    sprintf(config_file_path, "%s%s", elf_path, CONFIG_FILENAME);
    fd = open(config_file_path, O_RDWR);
    if(-1 == fd)
    {
        perror("open");
        return -1;
    }

    printf("open %s successn", config_file_path);

    close(fd);

    return 0;
}

CMakeLists.txt:

cmake_minimum_required(VERSION 3.12)

project(exe-abs-path C)

set(CMAKE_VERBOSE_MAKEFILE ON)

add_executable(${PROJECT_NAME} main.c)

執(zhí)行結(jié)果：

4. 總結(jié)

我當(dāng)然可以借助 shell 腳本解決該問(wèn)題?？墒?，我還是喜歡讓程序自己”實(shí)現(xiàn)“該功能，這樣我在部署和使用時(shí)會(huì)方便很多。

5. 說(shuō)明

/proc/self/ 目錄中存放了當(dāng)前進(jìn)程的很多有用信息。

/proc/self/fd/：進(jìn)程打開(kāi)的文件描述符；

/proc/self/cmdline：進(jìn)程被執(zhí)行時(shí)，命令行中輸入的命令；

/proc/self/task：進(jìn)程中包含的線程 id；

/proc/self/environ：進(jìn)程運(yùn)行時(shí)的環(huán)境變量信息。