166 lines
		
	
	
		
			5.2 KiB
		
	
	
	
		
			C
		
	
	
	
			
		
		
	
	
			166 lines
		
	
	
		
			5.2 KiB
		
	
	
	
		
			C
		
	
	
	
| /*
 | |
|  * Check decoding of successful PERF_EVENT_IOC_{ID,QUERY_BPF} ioctls.
 | |
|  *
 | |
|  * Copyright (c) 2018 The strace developers.
 | |
|  * All rights reserved.
 | |
|  *
 | |
|  * Redistribution and use in source and binary forms, with or without
 | |
|  * modification, are permitted provided that the following conditions
 | |
|  * are met:
 | |
|  * 1. Redistributions of source code must retain the above copyright
 | |
|  *    notice, this list of conditions and the following disclaimer.
 | |
|  * 2. Redistributions in binary form must reproduce the above copyright
 | |
|  *    notice, this list of conditions and the following disclaimer in the
 | |
|  *    documentation and/or other materials provided with the distribution.
 | |
|  * 3. The name of the author may not be used to endorse or promote products
 | |
|  *    derived from this software without specific prior written permission.
 | |
|  *
 | |
|  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 | |
|  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 | |
|  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 | |
|  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 | |
|  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 | |
|  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 | |
|  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 | |
|  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 | |
|  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 | |
|  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 | |
|  */
 | |
| 
 | |
| #include "tests.h"
 | |
| 
 | |
| #ifdef HAVE_LINUX_PERF_EVENT_H
 | |
| 
 | |
| # include <assert.h>
 | |
| # include <inttypes.h>
 | |
| # include <stdio.h>
 | |
| # include <stdlib.h>
 | |
| # include <string.h>
 | |
| # include <sys/ioctl.h>
 | |
| # include <linux/perf_event.h>
 | |
| 
 | |
| # ifndef PERF_EVENT_IOC_ID
 | |
| #  define PERF_EVENT_IOC_ID			_IOR('$', 7, void *)
 | |
| # endif
 | |
| 
 | |
| # ifndef PERF_EVENT_IOC_QUERY_BPF
 | |
| #  define PERF_EVENT_IOC_QUERY_BPF		_IOWR('$', 10, void *)
 | |
| 
 | |
| struct perf_event_query_bpf {
 | |
|         uint32_t ids_len;
 | |
|         uint32_t prog_cnt;
 | |
|         uint32_t ids[0];
 | |
| };
 | |
| # endif
 | |
| 
 | |
| int
 | |
| main(int argc, char **argv)
 | |
| {
 | |
| 	static const uint64_t magic64 = 0xfacefeeddeadc0deULL;
 | |
| 
 | |
| 	TAIL_ALLOC_OBJECT_CONST_PTR(uint64_t, u64_ptr);
 | |
| 	uint64_t *const u64_efault = u64_ptr + 1;
 | |
| 	uint32_t *const u32_arr = tail_alloc(sizeof(uint32_t) * 4);
 | |
| 	uint32_t *const u32_efault = u32_arr + 4;
 | |
| 
 | |
| 	unsigned long num_skip;
 | |
| 	long inject_retval;
 | |
| 	bool locked = false;
 | |
| 
 | |
| 	*u64_ptr = magic64;
 | |
| 
 | |
| 	if (argc == 1)
 | |
| 		return 0;
 | |
| 
 | |
| 	if (argc < 3)
 | |
| 		error_msg_and_fail("Usage: %s NUM_SKIP INJECT_RETVAL", argv[0]);
 | |
| 
 | |
| 	num_skip = strtoul(argv[1], NULL, 0);
 | |
| 	inject_retval = strtol(argv[2], NULL, 0);
 | |
| 
 | |
| 	if (inject_retval < 0)
 | |
| 		error_msg_and_fail("Expected non-negative INJECT_RETVAL, "
 | |
| 				   "but got %ld", inject_retval);
 | |
| 
 | |
| 	for (unsigned long i = 0; i < num_skip; i++) {
 | |
| 		long ret = ioctl(-1, PERF_EVENT_IOC_ID, NULL);
 | |
| 
 | |
| 		printf("ioctl(-1, PERF_EVENT_IOC_ID, NULL) = %s%s\n",
 | |
| 		       sprintrc(ret),
 | |
| 		       ret == inject_retval ? " (INJECTED)" : "");
 | |
| 
 | |
| 		if (ret != inject_retval)
 | |
| 			continue;
 | |
| 
 | |
| 		locked = true;
 | |
| 		break;
 | |
| 	}
 | |
| 
 | |
| 	if (!locked)
 | |
| 		error_msg_and_fail("Hasn't locked on ioctl(-1"
 | |
| 				   ", PERF_EVENT_IOC_ID, NULL) returning %lu",
 | |
| 				   inject_retval);
 | |
| 
 | |
| 	/* PERF_EVENT_IOC_ID */
 | |
| 	assert(ioctl(-1, PERF_EVENT_IOC_ID, NULL) == inject_retval);
 | |
| 	printf("ioctl(-1, PERF_EVENT_IOC_ID, NULL) = %ld (INJECTED)\n",
 | |
| 	       inject_retval);
 | |
| 
 | |
| 	assert(ioctl(-1, PERF_EVENT_IOC_ID, u64_efault) == inject_retval);
 | |
| 	printf("ioctl(-1, PERF_EVENT_IOC_ID, %p) = %ld (INJECTED)\n",
 | |
| 	       u64_efault, inject_retval);
 | |
| 
 | |
| 	assert(ioctl(-1, PERF_EVENT_IOC_ID, u64_ptr) == inject_retval);
 | |
| 	printf("ioctl(-1, PERF_EVENT_IOC_ID, [%" PRIu64 "]) = %ld (INJECTED)\n",
 | |
| 	       magic64, inject_retval);
 | |
| 
 | |
| 	/* PERF_EVENT_IOC_QUERY_BPF */
 | |
| 	assert(ioctl(-1, PERF_EVENT_IOC_QUERY_BPF, NULL) == inject_retval);
 | |
| 	printf("ioctl(-1, PERF_EVENT_IOC_QUERY_BPF, NULL) = %ld (INJECTED)\n",
 | |
| 	       inject_retval);
 | |
| 
 | |
| 	assert(ioctl(-1, PERF_EVENT_IOC_QUERY_BPF, u32_efault)
 | |
| 	       == inject_retval);
 | |
| 	printf("ioctl(-1, PERF_EVENT_IOC_QUERY_BPF, %p) = %ld (INJECTED)\n",
 | |
| 	       u32_efault, inject_retval);
 | |
| 
 | |
| 	u32_arr[3] = 0xdeadbeef;
 | |
| 	assert(ioctl(-1, PERF_EVENT_IOC_QUERY_BPF, u32_arr + 3)
 | |
| 	       == inject_retval);
 | |
| 	printf("ioctl(-1, PERF_EVENT_IOC_QUERY_BPF, {ids_len=3735928559, ...}) "
 | |
| 	       "= %ld (INJECTED)\n",
 | |
| 	       inject_retval);
 | |
| 
 | |
| 	u32_arr[2] = 0xdecaffed;
 | |
| 	assert(ioctl(-1, PERF_EVENT_IOC_QUERY_BPF, u32_arr + 2)
 | |
| 	       == inject_retval);
 | |
| 	printf("ioctl(-1, PERF_EVENT_IOC_QUERY_BPF, {ids_len=3737845741"
 | |
| 	       ", prog_cnt=3735928559, ids=%p})"
 | |
| 	       " = %ld (INJECTED)\n",
 | |
| 	       u32_efault, inject_retval);
 | |
| 
 | |
| 	u32_arr[0] = 0xbadc0ded;
 | |
| 	u32_arr[1] = 5;
 | |
| 	assert(ioctl(-1, PERF_EVENT_IOC_QUERY_BPF, u32_arr) == inject_retval);
 | |
| 	printf("ioctl(-1, PERF_EVENT_IOC_QUERY_BPF, {ids_len=3134983661"
 | |
| 	       ", prog_cnt=5, ids=[3737845741, 3735928559, ... /* %p */]})"
 | |
| 	       " = %ld (INJECTED)\n",
 | |
| 	       u32_efault, inject_retval);
 | |
| 
 | |
| 	u32_arr[1] = 2;
 | |
| 	assert(ioctl(-1, PERF_EVENT_IOC_QUERY_BPF, u32_arr) == inject_retval);
 | |
| 	printf("ioctl(-1, PERF_EVENT_IOC_QUERY_BPF, {ids_len=3134983661"
 | |
| 	       ", prog_cnt=2, ids=[3737845741, 3735928559]})"
 | |
| 	       " = %ld (INJECTED)\n",
 | |
| 	       inject_retval);
 | |
| 
 | |
| 	puts("+++ exited with 0 +++");
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| #else
 | |
| 
 | |
| SKIP_MAIN_UNDEFINED("HAVE_LINUX_PERF_EVENT_H");
 | |
| 
 | |
| #endif
 |