linux-imx

/* Copyright (c) 2016 Facebook

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of version 2 of the GNU General Public

 * License as published by the Free Software Foundation.

 */

#define _GNU_SOURCE

#include <sched.h>

#include <errno.h>

#include <stdio.h>

#include <sys/types.h>

#include <asm/unistd.h>

#include <fcntl.h>

#include <unistd.h>

#include <assert.h>

#include <sys/wait.h>

#include <stdlib.h>

#include <signal.h>

#include <linux/bpf.h>

#include <string.h>

#include <time.h>

#include <sys/resource.h>

#include <bpf/bpf.h>

#include "bpf_load.h"

​

#define MAX_CNT 1000000

​

static __u64 time_get_ns(void)

{

    struct timespec ts;

​

    clock_gettime(CLOCK_MONOTONIC, &ts);

    return ts.tv_sec * 1000000000ull + ts.tv_nsec;

}

​

static void test_task_rename(int cpu)

{

    __u64 start_time;

    char buf[] = "test\n";

    int i, fd;

​

    fd = open("/proc/self/comm", O_WRONLY|O_TRUNC);

    if (fd < 0) {

        printf("couldn't open /proc\n");

        exit(1);

    }

    start_time = time_get_ns();

    for (i = 0; i < MAX_CNT; i++) {

        if (write(fd, buf, sizeof(buf)) < 0) {

            printf("task rename failed: %s\n", strerror(errno));

            close(fd);

            return;

        }

    }

    printf("task_rename:%d: %lld events per sec\n",

           cpu, MAX_CNT * 1000000000ll / (time_get_ns() - start_time));

    close(fd);

}

​

static void test_urandom_read(int cpu)

{

    __u64 start_time;

    char buf[4];

    int i, fd;

​

    fd = open("/dev/urandom", O_RDONLY);

    if (fd < 0) {

        printf("couldn't open /dev/urandom\n");

        exit(1);

    }

    start_time = time_get_ns();

    for (i = 0; i < MAX_CNT; i++) {

        if (read(fd, buf, sizeof(buf)) < 0) {

            printf("failed to read from /dev/urandom: %s\n", strerror(errno));

            close(fd);

            return;

        }

    }

    printf("urandom_read:%d: %lld events per sec\n",

           cpu, MAX_CNT * 1000000000ll / (time_get_ns() - start_time));

    close(fd);

}

​

static void loop(int cpu, int flags)

{

    cpu_set_t cpuset;

​

    CPU_ZERO(&cpuset);

    CPU_SET(cpu, &cpuset);

    sched_setaffinity(0, sizeof(cpuset), &cpuset);

​

    if (flags & 1)