]> rtime.felk.cvut.cz Git - canping.git/blobdiff - src/vca_canping.c
Added mlockall() when using real-time priority
[canping.git] / src / vca_canping.c
index 21239768bae5f50ab7295f984ca4e69f68de5c3a..fc5ed82b8701342a36de6bfe01d37247e654bf06 100644 (file)
@@ -31,6 +31,8 @@
 #include <ul_list.h>
 #include <errno.h>
 #include <semaphore.h>
+#include <stdbool.h>
+#include <error.h>
 
 /* TODO: Handle the case where there are more canping slaves running
  * on one CAN bus. */
@@ -43,6 +45,7 @@
 
 #ifdef WITH_RTPRIO
 #include <sched.h>
+#include <sys/mman.h>
 
 int sched_policy = SCHED_OTHER;
 int sched_rtprio;
@@ -98,12 +101,20 @@ int option_wait_ms = 1000;
 int option_timeout = 4;
 int option_open_once = 0;
 int option_synch_start = 0;
+bool option_background = false;
+char *option_histogram = NULL;
 
 /* Lists */
 typedef struct threads {
        ul_list_head_t head;
 } threads_t;
 
+struct histogram {
+       unsigned *data;
+       unsigned allocated;
+       unsigned resolution;
+};
+
 typedef struct thread_data {
        pthread_t tid;
        long int canid;
@@ -115,6 +126,8 @@ typedef struct thread_data {
        int min, max;           /* min/max response times */
        int timeout;            /* number of timeouts */
 
+       struct histogram hist;
+
        ul_list_node_t node;
 } thread_data_t;
 
@@ -123,6 +136,44 @@ UL_LIST_CUST_DEC(thread_list, threads_t, thread_data_t, head, node);
 threads_t master_threads;
 threads_t slave_threads;
 
+int histogram_init(struct histogram *h,
+                  unsigned max_value,
+                  unsigned resolution)
+{
+       size_t mem;
+       h->allocated = max_value/resolution + 1;
+       h->resolution = resolution;
+       mem = h->allocated*sizeof(*h->data);
+       h->data = malloc(mem);
+       if (h->data) {
+               memset(h->data, 0, mem);
+               return 0;
+       } else
+               return -1;
+}
+
+void histogram_add(struct histogram *h, unsigned value)
+{
+       unsigned index = value / h->resolution;
+       if (index >= h->allocated)
+               index = h->allocated - 1;
+       h->data[index]++;
+}
+
+void histogram_fprint(struct histogram *h, FILE *f)
+{
+       unsigned long long sum = 0, cum;
+       unsigned i;
+       for (i = 0; i < h->allocated; i++)
+               sum += h->data[i];
+       cum = sum;
+       for (i = 0; i < h->allocated; i++) {
+               if (h->data[i] != 0)
+                       fprintf(f, "%d %lld\n", i*h->resolution, cum);
+               cum -= h->data[i];
+       }
+}
+
 /* Subtract the `struct timeval' values X and Y, storing the result in
    RESULT.  Return 1 if the difference is negative, otherwise 0.  */
      
@@ -215,6 +266,9 @@ void *master_thread(void *arg)
                perror("ioctl CANQUE_QUEUE_FLUSH");
                exit(EXIT_FLUSH_ERROR);
        }
+
+       if (option_histogram)
+               histogram_init(&td->hist, 1000*1000/*max [us]*/, 5/*resolution [us]*/);
        
        /* Prepare data structures for the select syscall. */
        FD_ZERO (&rdset);
@@ -295,6 +349,8 @@ void *master_thread(void *arg)
                                        (double)time*time  / td->count;
                                if (time > td->max) td->max = time;
                                if (time < td->min) td->min = time;
+                               if (option_histogram)
+                                       histogram_add(&td->hist, time);
                                total_count++;
                        } /* read */
                } 
@@ -319,6 +375,15 @@ void *master_thread(void *arg)
                vca_close_handle(vcah);
        }
 
+       if (option_histogram) {
+               FILE *f;
+               char buf[100];
+               snprintf(buf, sizeof(buf), "%s-%d.dat", option_histogram, ping_id);
+               f = fopen(buf, "w");
+               histogram_fprint(&td->hist, f);
+               fclose(f);
+       }
+
        sem_post(&finish_sem);
        return (void *)ret;
 }
@@ -404,6 +469,9 @@ void *slave_thread(void *arg)
        pongmsg.length = option_length;
        for (i=0; i < option_length; i++) pongmsg.data[i] = i;
 
+       /* Signal that I'm ready */
+       sem_post(&ready_sem);
+
        while (!IS_FINISH_FLAG()) {
                /* Receive a ping message */
                pingmsg.flags=0;
@@ -472,6 +540,9 @@ void start_slaves(int slaves, int first_id)
                pthread_create(&td->tid, NULL, slave_thread, (void *)td);
                dbg(2, "Slave thread: %p\n", (void *)td->tid);
        }
+
+       /* Wait for all threads beeing ready */
+       for (i = 0; i < slaves; i++) sem_wait(&ready_sem);
 }
 
 #ifdef WITH_RTPRIO
@@ -501,8 +572,10 @@ void print_help(void)
        printf("Usage: canping -m <master threads> [other options]\n"
               "       canping -s <slave threads> [other options]\n\n"
               "Other options:\n"
+              "  -b            go to background (fork) after initialization, prints child PID\n"
               "  -c count      how many messages each master sends\n"
               "  -d dev        device (e.g. /dev/can1)\n"
+              "  -g prefix     store cumulative histogram in file <prefix>-<id>.dat\n"
               "  -h            print this help\n"
               "  -i id         id of first master message\n"
               "  -l length     length of the messages (0..8)\n"
@@ -526,15 +599,21 @@ int parse_options(int argc, char *argv[])
        int c;
        
        opterr = 0;
-       while ((c = getopt (argc, argv, "c:d:hi:l:m:os:t:vw:yrR:")) != -1)
+       while ((c = getopt (argc, argv, "bc:d:g:h:i:l:m:os:t:vw:yrR:")) != -1)
                switch (c)
                {
+               case 'b':
+                       option_background = true;
+                       break;
                case 'c':
                        option_count = atoi(optarg);
                        break;
                case 'd':
                        option_device = optarg;
                        break;
+               case 'g':
+                       option_histogram = optarg;
+                       break;
                case 'h':
                        print_help();
                        exit(EXIT_OK);
@@ -684,9 +763,11 @@ int main(int argc, char *argv[])
        parse_options(argc, argv);
 
 #ifdef WITH_RTPRIO
-       if(sched_policy != SCHED_OTHER)
+       if(sched_policy != SCHED_OTHER) {
                if(set_sched_policy_and_prio(sched_policy, sched_rtprio) <0)
                        exit(EXIT_BAD_PARAM);
+               mlockall(MCL_CURRENT | MCL_FUTURE);
+       }
 #endif
 
 
@@ -712,9 +793,33 @@ int main(int argc, char *argv[])
                }
        }
 
-       if (option_masters) start_masters(option_masters, option_first_id);
-       if (option_slaves) start_slaves(option_slaves, option_first_id);
+       sem_t *child_ready;
+       int fork_ret = 0;
+       if ((child_ready = sem_open("canping", O_CREAT)) == NULL)
+               error(1, errno, "sem_open");
+       if (option_background) {
+               /* Go to background when everything is ready */
+               fork_ret = fork();
+               if (fork_ret < 0)
+                       error(1, errno, "Cannot go to background");
+               if (fork_ret == 0) {
+                       int devnull = open("/dev/null", O_WRONLY);
+                       dup2(devnull, 1);
+               }
+       }
 
+       if (fork_ret == 0) {
+               /* Child */
+               if (option_masters) start_masters(option_masters, option_first_id);
+               if (option_slaves) start_slaves(option_slaves, option_first_id);
+               sem_post(child_ready);
+       } else {
+               /* Parent */
+               sem_wait(child_ready);
+               printf("%d\n", fork_ret);
+               exit(0);
+       }
+       
        wait_for_threads();
 
        if (option_open_once) {