]>
Commit | Line | Data |
---|---|---|
7c673cae FG |
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 | ||
11fdf7f2 TL |
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 | ||
7c673cae FG |
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 | { | |
11fdf7f2 TL |
87 | if (core >= RTE_MAX_LCORE) { |
88 | return SPDK_ENV_SOCKET_ID_ANY; | |
89 | } | |
90 | ||
7c673cae FG |
91 | return rte_lcore_to_socket_id(core); |
92 | } | |
11fdf7f2 TL |
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 | } |