65 lines · c
1/*2 * Copyright © 2018 Alexey Dobriyan <adobriyan@gmail.com>3 *4 * Permission to use, copy, modify, and distribute this software for any5 * purpose with or without fee is hereby granted, provided that the above6 * copyright notice and this permission notice appear in all copies.7 *8 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES9 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF10 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR11 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES12 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN13 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF14 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.15 */16// Test that /proc/thread-self gives correct TGID/PID.17#undef NDEBUG18#include <assert.h>19#include <sched.h>20#include <stdio.h>21#include <unistd.h>22#include <sys/mman.h>23#include <sys/wait.h>24 25#include "proc.h"26 27int f(void *arg)28{29 char buf1[64], buf2[64];30 pid_t pid, tid;31 ssize_t rv;32 33 pid = sys_getpid();34 tid = sys_gettid();35 snprintf(buf1, sizeof(buf1), "%u/task/%u", pid, tid);36 37 rv = readlink("/proc/thread-self", buf2, sizeof(buf2));38 assert(rv == strlen(buf1));39 buf2[rv] = '\0';40 assert(streq(buf1, buf2));41 42 if (arg)43 exit(0);44 return 0;45}46 47int main(void)48{49 const int PAGE_SIZE = sysconf(_SC_PAGESIZE);50 pid_t pid;51 void *stack;52 53 /* main thread */54 f((void *)0);55 56 stack = mmap(NULL, 2 * PAGE_SIZE, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANONYMOUS, -1, 0);57 assert(stack != MAP_FAILED);58 /* side thread */59 pid = clone(f, stack + PAGE_SIZE, CLONE_THREAD|CLONE_SIGHAND|CLONE_VM, (void *)1);60 assert(pid > 0);61 pause();62 63 return 0;64}65