Home | History | Annotate | Download | only in tests
      1 /*
      2  * Check decoding of prctl PR_GET_TID_ADDRESS operation.
      3  *
      4  * Copyright (c) 2016 Eugene Syromyatnikov <evgsyr (at) 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 
     42 static const char *
     43 sprintaddr(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 
     56 int
     57 main(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 
    102 SKIP_MAIN_UNDEFINED("__NR_prctl && __NR_set_tid_address && PR_GET_TID_ADDRESS")
    103 
    104 #endif
    105