public

Arch Prctl

linux/arch_prctl.md

arch_prctl 是一个用于在 x86-64 架构上设置和获取线程特定寄存器 (FS 和 GS 寄存器) 值的系统调用。它通常在实现线程本地存储（TLS）或在操作系统内部进行某些低级别任务时使用。

`arch_prctl` 系统调用

系统调用原型如下：

#include <sys/prctl.h>

int arch_prctl(int code, unsigned long addr);

code: 控制操作的类型。可以是设置或获取寄存器的值，常用的值包括 ARCH_SET_FS, ARCH_SET_GS, ARCH_GET_FS, ARCH_GET_GS。
addr: 要设置的寄存器值或获取寄存器值的指针。

返回值:

成功时返回 0，失败时返回 -1，并设置 errno 以指示错误类型。

示例 1：设置 FS 寄存器

以下示例展示如何使用 ARCH_SET_FS 将 FS 寄存器设置为某个特定值。

#include <stdio.h>
#include <stdlib.h>
#include <sys/prctl.h>

int main() {
    unsigned long fs_value = 0x12345678; // 需要设置的值

    // 设置 FS 寄存器
    if (arch_prctl(ARCH_SET_FS, fs_value) != 0) {
        perror("arch_prctl ARCH_SET_FS");
        exit(EXIT_FAILURE);
    }

    printf("FS register set to 0x%lx\n", fs_value);
    return 0;
}

示例 2：获取 FS 寄存器的当前值

下面的示例展示如何使用 ARCH_GET_FS 获取当前 FS 寄存器的值。

#include <stdio.h>
#include <stdlib.h>
#include <sys/prctl.h>

int main() {
    unsigned long fs_value;

    // 获取 FS 寄存器的当前值
    if (arch_prctl(ARCH_GET_FS, (unsigned long)&fs_value) != 0) {
        perror("arch_prctl ARCH_GET_FS");
        exit(EXIT_FAILURE);
    }

    printf("Current FS register value: 0x%lx\n", fs_value);
    return 0;
}

说明

FS 和 GS 寄存器：FS 和 GS 寄存器在 x86-64 架构中通常用于指向线程本地存储（TLS）。这些寄存器可以保存指针，指向每个线程独有的数据段。通过 arch_prctl，可以直接读取或设置这些寄存器的值。
线程本地存储（TLS）：TLS 是每个线程独有的数据存储区域。在多线程编程中，TLS 允许每个线程都有独立的数据，以避免多个线程之间的数据冲突。

典型应用场景

线程本地存储（TLS）：在多线程编程中，操作系统会将某些线程本地数据存储在由 FS 或 GS 寄存器指向的内存段中。应用程序可以使用 arch_prctl 来设置或获取这些寄存器，以便在不同的线程之间管理 TLS。
低级别操作系统开发：在操作系统内核或者某些底层应用中，有时候需要直接操作这些寄存器以实现更精细的内存管理或进程间数据隔离。

通过以上例子，你可以看到 arch_prctl 提供了一种直接操作 FS 和 GS 寄存器的方法，使得程序员能够以低级方式管理线程本地数据。

在多线程环境中，每个线程都有自己独立的线程局部存储（Thread-Local Storage，TLS）。GCC 和操作系统协作，通过使用专门的寄存器（如 x86-64 架构上的 FS 或 GS 寄存器）来实现 TLS 的访问。以下是大致的工作原理：

1. 线程局部存储（TLS）的原理

TLS 数据：在多线程程序中，有些数据需要每个线程独有，这些数据被称为线程局部数据。TLS 允许每个线程有自己的一份变量副本，这样不同的线程不会干扰彼此的局部数据。
TLS 访问机制：操作系统和编译器（如 GCC）通过特定的寄存器（如 FS 或 GS）来指向 TLS 数据区域，从而每个线程都能访问自己的 TLS 数据，而无需显式地传递线程 ID 或者其他信息。

2. TLS 模型

GCC 支持多种 TLS 模型，常用的有：

Global Dynamic：适用于所有类型的线程局部变量，能处理最复杂的场景。
Local Dynamic：适用于大多数场景，开销相对较小。
Initial Exec：适用于在程序启动时能够完全解析的 TLS 变量，访问开销最低。
Local Exec：用于在程序加载时解析且不改变的 TLS 变量。

TLS 模型定义了线程局部变量的内存布局和访问方式。GCC 在编译时选择合适的模型，根据变量的属性（如 __thread 关键字）和目标平台的支持情况。

3. 使用寄存器指向 TLS 区域

在 x86-64 架构上，操作系统会为每个线程分配一个 TLS 区域，并使用 FS 或 GS 寄存器指向该区域。这些寄存器的值在每个线程的上下文中都是不同的。当一个线程执行时，FS 或 GS 寄存器会指向该线程的 TLS 数据区。

访问 TLS 数据的步骤：

GCC 生成的代码会使用 FS 或 GS 寄存器的值来计算 TLS 变量的地址。
例如，一个 TLS 变量可能在 FS:[offset] 或 GS:[offset] 的位置，具体偏移量由编译器在编译时决定。
当线程访问该 TLS 变量时，寄存器的值加上偏移量得到具体的内存地址，从而访问到正确的线程局部数据。

4. 操作系统的角色

线程切换：当操作系统调度器切换线程时，它会切换线程上下文，包括 FS 或 GS 寄存器的值。这样，每个线程在运行时总是会有正确的 FS 或 GS 寄存器指向它自己的 TLS 区域。
TLS 初始化：操作系统负责在线程创建时为新线程分配 TLS 区域，并正确设置 FS 或 GS 寄存器。

5. 代码示例

假设我们有一个使用 TLS 变量的简单程序：

#include <stdio.h>
#include <pthread.h>

__thread int tls_var = 0;

void* thread_function(void* arg) {
    tls_var = (int)(size_t)arg;
    printf("Thread %d: tls_var = %d\n", tls_var, tls_var);
    return NULL;
}

int main() {
    pthread_t threads[2];
    pthread_create(&threads[0], NULL, thread_function, (void*)(size_t)1);
    pthread_create(&threads[1], NULL, thread_function, (void*)(size_t)2);
    pthread_join(threads[0], NULL);
    pthread_join(threads[1], NULL);
    return 0;
}

__thread：定义了一个线程局部变量 tls_var。每个线程在访问 tls_var 时都会访问自己的一份副本。
GCC 生成的代码：在每个线程中，GCC 会生成代码来使用 FS 或 GS 寄存器加上偏移量来访问 tls_var，从而确保每个线程访问自己的副本。

总结

GCC 知道哪个线程在访问 TLS 是通过特定的硬件寄存器（如 FS 或 GS）以及编译器和操作系统协作来实现的。在不同的线程中，这些寄存器指向不同的内存区域，使得线程局部数据在各个线程中独立存在，而不需要显式地传递线程标识符或进行额外的线程管理。