我正在为非常小的静态链接程序实现一小部分 libc,我认为添加 TLS 支持将是一个很好的学习体验。
真棒的想法!我不得不在项目中实现自己的 TLS,因为我无法使用任何常见的线程库,例如 pthread。对于您的问题,我没有完整的解决方案,但分享我的经验可能会很有用。
我使用 arch_prctl(2) 设置 FS 寄存器。我需要以某种方式使用 set_thread_area(2) 吗?
答案取决于您实际使用的架构。如果您使用的是 x86-64 位,则应仅使用arch_prctl将 FS 寄存器设置为要用作 TLS 的内存区域(它允许您寻址大于 4GB 的内存区域)。而对于 x86-32,您必须使用set_thread_area,因为它是内核支持的唯一系统调用。
我的实现背后的想法是为每个线程分配一个私有内存区域并将其地址保存到%GS
寄存器中。这是一个相当简单的方法,但在我的情况下,它工作得很好。每次您想访问线程的私有区域时,您只需将保存的值%GS
和标识内存位置的偏移量用作基地址。我通常为每个线程分配一个内存页面(4096),并将其划分为 8 个字节的块。因此,我为每个线程设置了 512 个私有内存插槽,可以像索引从 0 到 511 的数组一样访问它们。
这是我使用的代码:
#define _GNU_SOURCE 1
#include "tls.h"
#include <asm/ldt.h>
#include <stdlib.h>
#include <sys/mman.h>
#include <sys/prctl.h>
#include <asm/prctl.h>
#include <sys/syscall.h>
#include <unistd.h>
void * install_tls() {
void *addr = mmap(0, 4096, PROT_READ|PROT_WRITE,
MAP_PRIVATE|MAP_ANONYMOUS, -1, 0);
if (syscall(SYS_arch_prctl,ARCH_SET_GS, addr) < 0)
return NULL;
return addr;
}
void freeTLS() {
void *addr;
syscall(SYS_arch_prctl,ARCH_GET_GS, &addr);
munmap(addr, 4096);
}
bool set_tls_value(int idx, unsigned long val) {
if (idx < 0 || idx >= 4096/8) {
return false;
}
asm volatile(
"movq %0, %%gs:(%1)\n"
:
: "q"((void *)val), "q"(8ll * idx));
return true;
}
unsigned long get_tls_value(int idx) {
long long rc;
if (idx < 0 || idx >= 4096/8) {
return 0;
}
asm volatile(
"movq %%gs:(%1), %0\n"
: "=q"(rc)
: "q"(8ll * idx));
return rc;
}
这是带有一些宏的标题:
#ifndef TLS_H
#define TLS_H
#include <stdbool.h>
void *install_tls();
void freeTLS();
bool set_tls_value (int, unsigned long);
unsigned long get_tls_value(int );
/*
*macros used to set and retrieve the values
from the tls area
*/
#define TLS_TID 0x0
#define TLS_FD 0x8
#define TLS_MONITORED 0x10
#define set_local_tid(_x) \
set_tls_value(TLS_TID, (unsigned long)_x)
#define set_local_fd(_x) \
set_tls_value(TLS_FD, (unsigned long)_x)
#define set_local_monitored(_x) \
set_tls_value(TLS_MONITORED, (unsigned long)_x)
#define get_local_tid() \
get_tls_value(TLS_TID)
#define get_local_fd() \
get_tls_value(TLS_FD)
#define get_local_monitored() \
get_tls_value(TLS_MONITORED)
#endif /* end of include guard: TLS_H */
每个线程要完成的第一个动作是安装 TLS 内存区域。一旦 TLS 被初始化,每个线程都可以开始使用这个区域作为私有 TLS。