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 "spdk/env.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 static uint32_t last_core = UINT32_MAX; 62 63 /* Already know the last_core, just return */ 64 if (last_core != UINT32_MAX) { 65 return last_core; 66 } 67 68 SPDK_ENV_FOREACH_CORE(i) { 69 last_core = i; 70 } 71 72 return last_core; 73 } 74 75 uint32_t 76 spdk_env_get_next_core(uint32_t prev_core) 77 { 78 unsigned lcore; 79 80 lcore = rte_get_next_lcore(prev_core, 0, 0); 81 if (lcore == RTE_MAX_LCORE) { 82 return UINT32_MAX; 83 } 84 return lcore; 85 } 86 87 uint32_t 88 spdk_env_get_socket_id(uint32_t core) 89 { 90 return rte_lcore_to_socket_id(core); 91 } 92 93 int 94 spdk_env_thread_launch_pinned(uint32_t core, thread_start_fn fn, void *arg) 95 { 96 int rc; 97 98 rc = rte_eal_remote_launch(fn, arg, core); 99 100 return rc; 101 } 102 103 void 104 spdk_env_thread_wait_all(void) 105 { 106 rte_eal_mp_wait_lcore(); 107 } 108