1 /*- 2 * BSD LICENSE 3 * 4 * Copyright (c) Intel Corporation. 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 "env_internal.h" 35 36 #include <rte_config.h> 37 #include <rte_lcore.h> 38 39 uint32_t 40 spdk_env_get_core_count(void) 41 { 42 return rte_lcore_count(); 43 } 44 45 uint32_t 46 spdk_env_get_current_core(void) 47 { 48 return rte_lcore_id(); 49 } 50 51 uint32_t 52 spdk_env_get_first_core(void) 53 { 54 return rte_get_next_lcore(-1, 0, 0); 55 } 56 57 uint32_t 58 spdk_env_get_last_core(void) 59 { 60 uint32_t i; 61 uint32_t last_core = UINT32_MAX; 62 63 SPDK_ENV_FOREACH_CORE(i) { 64 last_core = i; 65 } 66 67 assert(last_core != UINT32_MAX); 68 69 return last_core; 70 } 71 72 uint32_t 73 spdk_env_get_next_core(uint32_t prev_core) 74 { 75 unsigned lcore; 76 77 lcore = rte_get_next_lcore(prev_core, 0, 0); 78 if (lcore == RTE_MAX_LCORE) { 79 return UINT32_MAX; 80 } 81 return lcore; 82 } 83 84 uint32_t 85 spdk_env_get_socket_id(uint32_t core) 86 { 87 if (core >= RTE_MAX_LCORE) { 88 return SPDK_ENV_SOCKET_ID_ANY; 89 } 90 91 return rte_lcore_to_socket_id(core); 92 } 93 94 int 95 spdk_env_thread_launch_pinned(uint32_t core, thread_start_fn fn, void *arg) 96 { 97 int rc; 98 99 rc = rte_eal_remote_launch(fn, arg, core); 100 101 return rc; 102 } 103 104 void 105 spdk_env_thread_wait_all(void) 106 { 107 rte_eal_mp_wait_lcore(); 108 } 109