1/*
2 * Check decoding of prctl PR_GET_TID_ADDRESS operation.
3 *
4 * Copyright (c) 2016 Eugene Syromyatnikov <evgsyr@gmail.com>
5 * Copyright (c) 2016-2017 The strace developers.
6 * All rights reserved.
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
10 * are met:
11 * 1. Redistributions of source code must retain the above copyright
12 *    notice, this list of conditions and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 *    notice, this list of conditions and the following disclaimer in the
15 *    documentation and/or other materials provided with the distribution.
16 * 3. The name of the author may not be used to endorse or promote products
17 *    derived from this software without specific prior written permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
20 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
21 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
22 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
23 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
24 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
28 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29 */
30
31#include "tests.h"
32#include <asm/unistd.h>
33#include <linux/prctl.h>
34
35#if defined __NR_prctl && defined __NR_set_tid_address && \
36	defined PR_GET_TID_ADDRESS
37
38# include <inttypes.h>
39# include <stdio.h>
40# include <unistd.h>
41
42static const char *
43sprintaddr(kernel_ulong_t addr)
44{
45	static char buf[sizeof("0x") + sizeof(addr) * 2];
46
47	if (addr) {
48		snprintf(buf, sizeof(buf), "%#llx", (unsigned long long) addr);
49
50		return buf;
51	}
52
53	return "NULL";
54}
55
56int
57main(void)
58{
59	static const kernel_ulong_t bogus_addr =
60		(kernel_ulong_t) 0xfffffffffffffffdULL;
61
62	/* Note that kernel puts kernel-sized pointer even on x32 */
63	TAIL_ALLOC_OBJECT_CONST_PTR(kernel_ulong_t, ptr);
64	long rc;
65	long set_ok;
66
67	*ptr = (kernel_ulong_t) 0xbadc0dedda7a1057ULL;
68
69	rc = syscall(__NR_prctl, PR_GET_TID_ADDRESS, NULL);
70	printf("prctl(PR_GET_TID_ADDRESS, NULL) = %s\n", sprintrc(rc));
71
72	rc = syscall(__NR_prctl, PR_GET_TID_ADDRESS, bogus_addr);
73	printf("prctl(PR_GET_TID_ADDRESS, %#llx) = %s\n",
74	       (unsigned long long) bogus_addr, sprintrc(rc));
75
76	rc = syscall(__NR_prctl, PR_GET_TID_ADDRESS, ptr);
77	if (rc) {
78		printf("prctl(PR_GET_TID_ADDRESS, %p) = %s\n",
79		       ptr, sprintrc(rc));
80	} else {
81		printf("prctl(PR_GET_TID_ADDRESS, [%s]) = %s\n",
82		       sprintaddr(*ptr), sprintrc(rc));
83	}
84
85	set_ok = syscall(__NR_set_tid_address, bogus_addr);
86
87	rc = syscall(__NR_prctl, PR_GET_TID_ADDRESS, ptr);
88	if (rc) {
89		printf("prctl(PR_GET_TID_ADDRESS, %p) = %s\n",
90		       ptr, sprintrc(rc));
91	} else {
92		printf("prctl(PR_GET_TID_ADDRESS, [%s]) = %s\n",
93		       sprintaddr(set_ok ? bogus_addr : *ptr), sprintrc(rc));
94	}
95
96	puts("+++ exited with 0 +++");
97	return 0;
98}
99
100#else
101
102SKIP_MAIN_UNDEFINED("__NR_prctl && __NR_set_tid_address && PR_GET_TID_ADDRESS")
103
104#endif
105