lib/librte_eal/common/eal_common_thread.c

   1 /*-
   2  *   BSD LICENSE
   3  *
   4  *   Copyright(c) 2010-2014 Intel Corporation. All rights reserved.
   5  *   All rights reserved.
   6  *
   7  *   Redistribution and use in source and binary forms, with or without
   8  *   modification, are permitted provided that the following conditions
   9  *   are met:
  10  *
  11  *     * Redistributions of source code must retain the above copyright
  12  *       notice, this list of conditions and the following disclaimer.
  13  *     * Redistributions in binary form must reproduce the above copyright
  14  *       notice, this list of conditions and the following disclaimer in
  15  *       the documentation and/or other materials provided with the
  16  *       distribution.
  17  *     * Neither the name of Intel Corporation nor the names of its
  18  *       contributors may be used to endorse or promote products derived
  19  *       from this software without specific prior written permission.
  20  *
  21  *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  22  *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  23  *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  24  *   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  25  *   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  26  *   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  27  *   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  28  *   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  29  *   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  30  *   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  31  *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  32  */
  33
  34 #include <stdio.h>
  35 #include <stdlib.h>
  36 #include <stdint.h>
  37 #include <unistd.h>
  38 #include <pthread.h>
  39 #include <sched.h>
  40 #include <assert.h>
  41 #include <string.h>
  42
  43 #include <rte_lcore.h>
  44 #include <rte_memory.h>
  45 #include <rte_log.h>
  46
  47 #include "eal_thread.h"
  48
  49 RTE_DECLARE_PER_LCORE(unsigned , _socket_id);
  50
  51 unsigned rte_socket_id(void)
  52 {
  53         return RTE_PER_LCORE(_socket_id);
  54 }
  55
  56 int eal_cpuset_socket_id(rte_cpuset_t *cpusetp)
  57 {
  58         unsigned cpu = 0;
  59         int socket_id = SOCKET_ID_ANY;
  60         int sid;
  61
  62         if (cpusetp == NULL)
  63                 return SOCKET_ID_ANY;
  64
  65         do {
  66                 if (!CPU_ISSET(cpu, cpusetp))
  67                         continue;
  68
  69                 if (socket_id == SOCKET_ID_ANY)
  70                         socket_id = eal_cpu_socket_id(cpu);
  71
  72                 sid = eal_cpu_socket_id(cpu);
  73                 if (socket_id != sid) {
  74                         socket_id = SOCKET_ID_ANY;
  75                         break;
  76                 }
  77
  78         } while (++cpu < RTE_MAX_LCORE);
  79
  80         return socket_id;
  81 }
  82
  83 int
  84 rte_thread_set_affinity(rte_cpuset_t *cpusetp)
  85 {
  86         int s;
  87         unsigned lcore_id;
  88         pthread_t tid;
  89
  90         tid = pthread_self();
  91
  92         s = pthread_setaffinity_np(tid, sizeof(rte_cpuset_t), cpusetp);
  93         if (s != 0) {
  94                 RTE_LOG(ERR, EAL, "pthread_setaffinity_np failed\n");
  95                 return -1;
  96         }
  97
  98         /* store socket_id in TLS for quick access */
  99         RTE_PER_LCORE(_socket_id) =
 100                 eal_cpuset_socket_id(cpusetp);
 101
 102         /* store cpuset in TLS for quick access */
 103         memmove(&RTE_PER_LCORE(_cpuset), cpusetp,
 104                 sizeof(rte_cpuset_t));
 105
 106         lcore_id = rte_lcore_id();
 107         if (lcore_id != (unsigned)LCORE_ID_ANY) {
 108                 /* EAL thread will update lcore_config */
 109                 lcore_config[lcore_id].socket_id = RTE_PER_LCORE(_socket_id);
 110                 memmove(&lcore_config[lcore_id].cpuset, cpusetp,
 111                         sizeof(rte_cpuset_t));
 112         }
 113
 114         return 0;
 115 }
 116
 117 void
 118 rte_thread_get_affinity(rte_cpuset_t *cpusetp)
 119 {
 120         assert(cpusetp);
 121         memmove(cpusetp, &RTE_PER_LCORE(_cpuset),
 122                 sizeof(rte_cpuset_t));
 123 }
 124
 125 int
 126 eal_thread_dump_affinity(char *str, unsigned size)
 127 {
 128         rte_cpuset_t cpuset;
 129         unsigned cpu;
 130         int ret;
 131         unsigned int out = 0;
 132
 133         rte_thread_get_affinity(&cpuset);
 134
 135         for (cpu = 0; cpu < RTE_MAX_LCORE; cpu++) {
 136                 if (!CPU_ISSET(cpu, &cpuset))
 137                         continue;
 138
 139                 ret = snprintf(str + out,
 140                                size - out, "%u,", cpu);
 141                 if (ret < 0 || (unsigned)ret >= size - out) {
 142                         /* string will be truncated */
 143                         ret = -1;
 144                         goto exit;
 145                 }
 146
 147                 out += ret;
 148         }
 149
 150         ret = 0;
 151 exit:
 152         /* remove the last separator */
 153         if (out > 0)
 154                 str[out - 1] = '\0';
 155
 156         return ret;
 157 }