Line data Source code
1 : /**
2 : * Copyright (c) 2025 Huawei Technologies Co., Ltd.
3 : * This program is free software, you can redistribute it and/or modify it under the terms and conditions of
4 : * CANN Open Software License Agreement Version 2.0 (the "License").
5 : * Please refer to the License for details. You may not use this file except in compliance with the License.
6 : * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED,
7 : * INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE.
8 : * See LICENSE in the root of the software repository for the full text of the License.
9 : */
10 :
11 : #define _GNU_SOURCE
12 : #include "rs.h"
13 : #include "ra_rs_err.h"
14 : #include <unistd.h>
15 : #include <string.h>
16 : #include <stdlib.h>
17 : #include <sys/types.h>
18 : #include <ifaddrs.h>
19 : #include <netinet/in.h>
20 : #include <sys/epoll.h>
21 : #include <sys/eventfd.h>
22 : #include <sys/socket.h>
23 : #include <sys/fcntl.h>
24 : #include <arpa/inet.h>
25 : #include <dlfcn.h>
26 : #include <fnmatch.h>
27 : #include "securec.h"
28 : #include "rs_common_inner.h"
29 : #include "rs_inner.h"
30 : #include "rs_rdma_inner.h"
31 : #include "rs_nda.h"
32 : #include "rs_epoll.h"
33 : #include "rs_tls.h"
34 : #include "ssl_adp.h"
35 : #include "rs_socket.h"
36 : #include "dl_ibverbs_function.h"
37 : #include "dl_hal_function.h"
38 : #include "rs_drv_rdma.h"
39 : #include "file_opt.h"
40 : #ifdef CONFIG_TLV
41 : #include "rs_tlv.h"
42 : #endif
43 : #include "ra_rs_ctx.h"
44 : #include "rs_ctx.h"
45 : #include "rs_esched.h"
46 : #include "dl_net_function.h"
47 : #include "rs_ub.h"
48 : #include "rs_ctx_inner.h"
49 :
50 : __thread struct rs_cb *gRsCb = NULL; // lint !e17
51 : struct rs_cb *gRsCbList[RS_MAX_DEV_NUM] = {0}; // lint !e17
52 : int gInitCounter[RS_MAX_DEV_NUM] = {0};
53 :
54 : /* set current phyId g_rs_cb */
55 1 : void RsSetCtx(unsigned int phyId)
56 : {
57 1 : gRsCb = gRsCbList[phyId];
58 1 : }
59 :
60 : /* get current g_rs_cb */
61 3 : static struct rs_cb *RsGetCurRsCb(void)
62 : {
63 195 : for (int i = 0; i < RS_MAX_DEV_NUM; i++) {
64 192 : if (gRsCbList[i] != NULL) {
65 0 : hccp_info("[rs_get_cur_rs_cb], phyId[%u], rsCb[%p]", i, gRsCbList[i]);
66 0 : return gRsCbList[i];
67 : }
68 : }
69 3 : return NULL;
70 : }
71 :
72 365 : RS_ATTRI_VISI_DEF void RsGetCurTime(struct timeval *time)
73 : {
74 : int ret;
75 :
76 365 : RS_CHECK_POINTER_NULL_RETURN_VOID(time);
77 365 : ret = gettimeofday(time, NULL);
78 365 : if (ret) {
79 1 : hccp_warn("gettimeofday unsuccessful, ret[%d] expect 0", ret);
80 1 : ret = memset_s(time, sizeof(struct timeval), 0, sizeof(struct timeval));
81 1 : if (ret) {
82 1 : hccp_warn("memset_s unsuccessful, ret[%d] expect 0", ret);
83 : }
84 : }
85 :
86 365 : return;
87 : }
88 :
89 166 : RS_ATTRI_VISI_DEF void HccpTimeInterval(struct timeval *endTime, struct timeval *startTime, float *msec)
90 : {
91 166 : RS_CHECK_POINTER_NULL_RETURN_VOID(endTime);
92 166 : RS_CHECK_POINTER_NULL_RETURN_VOID(startTime);
93 166 : RS_CHECK_POINTER_NULL_RETURN_VOID(msec);
94 :
95 : /* if low position is sufficient, then borrow one from the high position */
96 166 : if (endTime->tv_usec < startTime->tv_usec) {
97 2 : endTime->tv_sec -= 1;
98 2 : endTime->tv_usec += MS_PER_SECOND_I * MS_PER_SECOND_I;
99 : }
100 :
101 166 : *msec = (float)((endTime->tv_sec - startTime->tv_sec) * MS_PER_SECOND_F +
102 166 : (endTime->tv_usec - startTime->tv_usec) / US_PER_MS_F);
103 :
104 166 : return;
105 : }
106 :
107 77 : RS_ATTRI_VISI_DEF void RsHeartbeatAlivePrint(struct RsPthreadInfo *pthreadInfo)
108 : {
109 77 : float timeCost = 0.0;
110 : struct timeval now;
111 :
112 77 : if (pthreadInfo == NULL) {
113 0 : hccp_err("pthread_info is NULL!");
114 0 : return;
115 : }
116 :
117 77 : RsGetCurTime(&now);
118 77 : HccpTimeInterval(&now, &pthreadInfo->lastCheckTime, &timeCost);
119 77 : if (timeCost >= RS_HEARTBEAT_TIME || timeCost <= 0) {
120 0 : hccp_info("pthread[%s] is alive!", pthreadInfo->pthreadName);
121 0 : RsGetCurTime(&pthreadInfo->lastCheckTime);
122 : }
123 :
124 77 : return;
125 : }
126 :
127 483 : int RsDev2rscb(uint32_t chipId, struct rs_cb **rsCb, bool initFlag)
128 : {
129 483 : if (gRsCb == NULL) {
130 44 : if (initFlag == false) {
131 2 : hccp_warn("No device initialized !");
132 : }
133 44 : return -ENODEV;
134 : }
135 :
136 439 : if (chipId == gRsCb->chipId) {
137 436 : *rsCb = gRsCb;
138 436 : return 0;
139 : }
140 :
141 3 : hccp_warn("get rs cb unsuccessful for dev %u !", chipId);
142 3 : *rsCb = NULL;
143 :
144 3 : return -ENODEV;
145 : }
146 :
147 32 : int RsGetHccpMode(unsigned int chipId)
148 : {
149 32 : struct rs_cb *rsCb = NULL;
150 : int ret;
151 :
152 32 : ret = RsDev2rscb(chipId, &rsCb, false);
153 32 : CHK_PRT_RETURN(ret, hccp_err("get rs_cb failed(%d)", ret), ret);
154 31 : return (int)rsCb->hccpMode;
155 : }
156 :
157 93 : int RsDev2conncb(uint32_t chipId, struct RsConnCb **connCb)
158 : {
159 : int ret;
160 93 : struct rs_cb *rsCb = NULL;
161 :
162 93 : ret = RsDev2rscb(chipId, &rsCb, false);
163 93 : CHK_PRT_RETURN(ret, hccp_err("get rs_cb failed(%d)", ret), ret);
164 :
165 93 : *connCb = &(rsCb->connCb);
166 :
167 93 : return 0;
168 : }
169 :
170 262 : int RsGetRdevCb(struct rs_cb *rsCb, unsigned int rdevIndex, struct RsRdevCb **rdevCb)
171 : {
172 262 : struct RsRdevCb *rdevCbTmp = NULL;
173 262 : struct RsRdevCb *rdevCbTmp2 = NULL;
174 :
175 262 : RS_LIST_GET_HEAD_ENTRY(rdevCbTmp, rdevCbTmp2, &rsCb->rdevList, list, struct RsRdevCb);
176 263 : for (; (&rdevCbTmp->list) != &rsCb->rdevList;
177 1 : rdevCbTmp = rdevCbTmp2, rdevCbTmp2 = list_entry(rdevCbTmp2->list.next, struct RsRdevCb, list)) {
178 262 : if (rdevCbTmp->rdevIndex == rdevIndex) {
179 261 : *rdevCb = rdevCbTmp;
180 261 : return 0;
181 : }
182 : }
183 :
184 1 : *rdevCb = NULL;
185 1 : hccp_err("rdev_cb for rdev_index[%u] do not available!", rdevIndex);
186 :
187 1 : return -ENODEV;
188 : }
189 :
190 47 : int RsRdev2rdevCb(unsigned int chipId, unsigned int rdevIndex, struct RsRdevCb **rdevCb)
191 : {
192 : int ret;
193 47 : struct rs_cb *rsCb = NULL;
194 :
195 47 : ret = RsDev2rscb(chipId, &rsCb, false);
196 47 : CHK_PRT_RETURN(ret, hccp_err("get rs_cb failed for chipId:%u, ret:%d", chipId, ret), -ENODEV);
197 :
198 47 : ret = RsGetRdevCb(rsCb, rdevIndex, rdevCb);
199 47 : CHK_PRT_RETURN(ret, hccp_err("rs_get_rdev_cb failed!, ret %d, rdevIndex %u", ret, rdevIndex), ret);
200 :
201 46 : return 0;
202 : }
203 :
204 44 : STATIC int RsPthreadMutexInit(struct rs_cb *rscb, struct RsInitConfig *cfg)
205 : {
206 : int ret;
207 : int err;
208 :
209 44 : RS_CHECK_POINTER_NULL_RETURN_INT(cfg);
210 44 : RS_CHECK_POINTER_NULL_RETURN_INT(rscb);
211 44 : rscb->chipId = cfg->chipId;
212 44 : rscb->hccpMode = cfg->hccpMode;
213 44 : rscb->connCb.rscb = rscb;
214 :
215 44 : ret = pthread_mutex_init(&rscb->mutex, NULL);
216 44 : CHK_PRT_RETURN(ret, hccp_err("rscb mutex_init failed ret %d!, normal ret 0", ret), -ESYSFUNC);
217 43 : ret = pthread_mutex_init(&rscb->connCb.connMutex, NULL);
218 43 : if (ret) {
219 0 : hccp_err("conn_cb mutex_init failed ret %d, normal ret 0!", ret);
220 0 : err = pthread_mutex_destroy(&rscb->mutex);
221 0 : hccp_dbg("pthread destroy ret %d", err);
222 0 : return -ESYSFUNC;
223 : }
224 :
225 43 : hccp_info("mutex init ok");
226 :
227 43 : RS_INIT_LIST_HEAD(&rscb->connCb.listenList);
228 43 : RS_INIT_LIST_HEAD(&rscb->connCb.serverAcceptList);
229 43 : RS_INIT_LIST_HEAD(&rscb->connCb.clientConnList);
230 43 : RS_INIT_LIST_HEAD(&rscb->connCb.serverConnList);
231 43 : RS_INIT_LIST_HEAD(&rscb->connCb.whiteList);
232 43 : RS_INIT_LIST_HEAD(&rscb->rdevList);
233 43 : RS_INIT_LIST_HEAD(&rscb->udevList);
234 43 : RS_INIT_LIST_HEAD(&rscb->heterogTcpFdList);
235 43 : rscb->connCb.wlistEnable = cfg->whiteListStatus;
236 43 : return 0;
237 : }
238 :
239 44 : STATIC int RsGetChipLogicId(unsigned int chipId, enum NetworkMode hccpMode, unsigned int *logicId)
240 : {
241 : (void)hccpMode;
242 44 : int ret = 0;
243 :
244 44 : ret = DlDrvDeviceGetIndexByPhyId(chipId, logicId);
245 44 : CHK_PRT_RETURN(ret != 0, hccp_err("hal get logicId failed, chipId[%u], ret[%d]", chipId, ret), -ENODEV);
246 :
247 44 : return 0;
248 : }
249 :
250 : #ifdef CUSTOM_INTERFACE
251 44 : STATIC int RsInitNetAdapt(struct rs_cb *rscb)
252 : {
253 44 : int ret = 0;
254 :
255 44 : if (rscb->protocol != PROTOCOL_UDMA) {
256 44 : return 0;
257 : }
258 :
259 0 : ret = RsNetAdaptInit();
260 0 : CHK_PRT_RETURN(ret != 0,
261 : hccp_err("rs_net_adapt_init chipId[%u] logic_devid[%u] failed, ret=%d", rscb->chipId, rscb->logicId, ret), ret);
262 :
263 0 : return ret;
264 : }
265 : #endif
266 :
267 : #ifdef CUSTOM_INTERFACE
268 0 : STATIC void RsDeInitNetAdapt(struct rs_cb *rscb)
269 : {
270 0 : if (rscb->protocol != PROTOCOL_UDMA) {
271 0 : return;
272 : }
273 :
274 0 : RsNetAdaptUninit();
275 : }
276 : #endif
277 :
278 44 : STATIC int RsInitRscbCfg(struct rs_cb *rscb)
279 : {
280 : enum ProductType productType;
281 : struct timeval start, end;
282 44 : float timeCost = 0.0;
283 : int ret;
284 :
285 44 : ret = RsGetChipLogicId(rscb->chipId, rscb->hccpMode, &rscb->logicId);
286 44 : CHK_PRT_RETURN(ret != 0, hccp_err("rs_get_chip_logic_id failed, ret[%d]", ret), ret);
287 :
288 44 : productType = RsGetProductType(rscb->logicId);
289 44 : CHK_PRT_RETURN(productType == PRODUCT_TYPE_INVALID, hccp_err("RsGetProductType failed, logicId:%u", rscb->logicId),
290 : -EINVAL);
291 : #ifdef CUSTOM_INTERFACE
292 44 : if (RsIsUdmaSupported() || RsIsRdmaSupported()) {
293 0 : ret = RsGetChipProtocol(rscb->chipId, rscb->hccpMode, &rscb->protocol, rscb->logicId);
294 0 : CHK_PRT_RETURN(ret != 0, hccp_err("rs_get_chip_protocol failed, ret[%d]", ret), ret);
295 : // make sure RsRoceGetApiVersion and RsNetGetApiVersion are valid
296 0 : ret = RsCtxApiInit(rscb->hccpMode, rscb->protocol);
297 0 : CHK_PRT_RETURN(ret != 0, hccp_err("rs_ctx_api_init failed, ret[%d]", ret), ret);
298 0 : ret = RsEschedInit(rscb);
299 0 : if (ret != 0) {
300 0 : hccp_err("rs_esched_init chipId[%u] logic_devid[%u] failed, ret=%d productType=%d", rscb->chipId,
301 : rscb->logicId, ret, productType);
302 0 : goto esched_init_err;
303 : }
304 : }
305 :
306 44 : ret = RsInitNetAdapt(rscb);
307 44 : if (ret != 0) {
308 0 : goto net_adapt_init_err;
309 : }
310 :
311 : #endif
312 :
313 44 : ret = rs_ssl_init(rscb);
314 44 : if (ret != 0) {
315 0 : hccp_err("init ssl failed, ret[%d]", ret);
316 0 : goto ssl_init_err;
317 : }
318 :
319 44 : RsGetCurTime(&start);
320 44 : ret = RsEpollConnectHandleInit(rscb);
321 44 : if (ret != 0) {
322 3 : hccp_err("create pthread failed, ret[%d]", ret);
323 3 : goto create_pthread_err;
324 : }
325 :
326 41 : RsGetCurTime(&end);
327 41 : HccpTimeInterval(&end, &start, &timeCost);
328 41 : hccp_info("rs_epoll_connect_handle_init ok cost [%f] ms", timeCost);
329 41 : return 0;
330 :
331 3 : create_pthread_err:
332 3 : rs_ssl_deinit(rscb);
333 3 : ssl_init_err:
334 : #ifdef CUSTOM_INTERFACE
335 3 : if (RsIsUdmaSupported() || RsIsRdmaSupported()) {
336 0 : RsDeInitNetAdapt(rscb);
337 0 : net_adapt_init_err:
338 0 : RsEschedDeinit(rscb->protocol);
339 0 : esched_init_err:
340 0 : (void)RsCtxApiDeinit(rscb->hccpMode, rscb->protocol);
341 : }
342 : #endif
343 3 : return ret;
344 : }
345 :
346 1 : STATIC void RsDeinitRscbCfg(struct rs_cb *rscb)
347 : {
348 1 : int tryAgain = RS_TRY_TIME;
349 1 : eventfd_t event = 1;
350 : int ret;
351 :
352 : #ifdef CUSTOM_INTERFACE
353 1 : if (RsIsUdmaSupported() || RsIsRdmaSupported()) {
354 0 : RsDeInitNetAdapt(rscb);
355 0 : RsEschedDeinit(rscb->protocol);
356 0 : (void)RsCtxApiDeinit(rscb->hccpMode, rscb->protocol);
357 : }
358 : #endif
359 1 : rs_ssl_deinit(rscb);
360 : // deinit resources in rs_epoll_connect_handle_init
361 : // deinit epoll thread, send event to eventfd to waking up epoll handle thread
362 1 : ret = (int)write(rscb->connCb.eventfd, &event, sizeof(eventfd_t));
363 1 : if (ret != sizeof(eventfd_t)) {
364 0 : hccp_warn("eventfd_write unsuccessful(0x%x), chipId:%u, errno:%d", ret, rscb->chipId, errno);
365 : }
366 2 : while (((rscb->state & RS_STATE_HALT) == 0) && (tryAgain != 0)) {
367 1 : usleep(RS_USLEEP_TIME);
368 1 : tryAgain--;
369 : };
370 1 : if (tryAgain == 0) {
371 0 : hccp_warn("try_again exhausted, epoll thread quit unsuccessful, rscb state:%u", rscb->state);
372 : }
373 1 : rscb->state &= ~RS_STATE_HALT;
374 :
375 : // deinit connect thread, already been RS_CONN_EXIT_FLAG, no need to change conn_flag
376 1 : if (rscb->connFlag != RS_CONN_EXIT_FLAG) {
377 1 : rscb->connFlag = 0;
378 : }
379 1 : tryAgain = RS_TRY_TIME;
380 200 : while ((rscb->connFlag != RS_CONN_EXIT_FLAG) && (tryAgain != 0)) {
381 199 : usleep(RS_USLEEP_TIME);
382 199 : tryAgain--;
383 : }
384 1 : if (tryAgain == 0) {
385 0 : hccp_warn("try_again exhausted, connect thread quit unsuccessful, rscb connFlag:%d", rscb->connFlag);
386 : }
387 :
388 1 : RsDestroyEpoll(rscb);
389 1 : }
390 :
391 45 : RS_ATTRI_VISI_DEF int RsInit(struct RsInitConfig *cfg)
392 : {
393 45 : struct rs_cb *rscb = NULL;
394 : int ret;
395 :
396 45 : RS_CHECK_POINTER_NULL_RETURN_INT(cfg);
397 45 : ret = DlHalInit();
398 45 : if (ret != 0) {
399 0 : hccp_err("[init][rs_init]dl_hal_init failed, ret = %d", ret);
400 0 : return ret;
401 : }
402 :
403 45 : int counter = __sync_fetch_and_add(&(gInitCounter[cfg->chipId]), 1);
404 45 : if (counter > 0) {
405 0 : hccp_warn("rs has been init for device %u!", cfg->chipId);
406 0 : return 0;
407 : }
408 45 : ret = RsDev2rscb(cfg->chipId, &rscb, true);
409 45 : CHK_PRT_RETURN(ret == 0, hccp_err("rs_cb exist for device %u! do NOT init it again!", cfg->chipId), -EEXIST);
410 :
411 45 : rscb = calloc(1, sizeof(struct rs_cb));
412 45 : CHK_PRT_RETURN(rscb == NULL, hccp_err("calloc rscb failed"), -ENOMEM);
413 :
414 44 : ret = RsPthreadMutexInit(rscb, cfg);
415 44 : if (ret != 0) {
416 1 : hccp_err("Init mutex failed, ret[%d]", ret);
417 1 : goto pthread_mutex_err;
418 : }
419 :
420 43 : ret = RsInitRscbCfg(rscb);
421 43 : if (ret != 0) {
422 3 : hccp_err("rs init rscb configure failed,ret:%d", ret);
423 3 : pthread_mutex_destroy(&rscb->mutex);
424 3 : pthread_mutex_destroy(&rscb->connCb.connMutex);
425 3 : goto pthread_mutex_err;
426 : }
427 :
428 40 : rscb->fdMap = calloc(1, sizeof(void *) * RS_MAX_FD_NUM);
429 40 : if (rscb->fdMap == NULL) {
430 0 : hccp_err("no memory for fd_map");
431 0 : ret = -ENOMEM;
432 0 : goto fd_map_err;
433 : }
434 :
435 40 : ret = getifaddrs(&rscb->ifaddrList);
436 40 : if (ret != 0) {
437 0 : hccp_err("getifaddrs failed, ret:%d", ret);
438 0 : goto getifaddrs_err;
439 : }
440 :
441 40 : gRsCbList[cfg->chipId] = gRsCb;
442 :
443 40 : hccp_run_info("rs init success, chipId[%u]", cfg->chipId);
444 40 : return 0;
445 :
446 0 : getifaddrs_err:
447 0 : free(rscb->fdMap);
448 0 : rscb->fdMap = NULL;
449 :
450 0 : fd_map_err:
451 0 : pthread_mutex_destroy(&rscb->mutex);
452 0 : pthread_mutex_destroy(&rscb->connCb.connMutex);
453 0 : RsDeinitRscbCfg(rscb);
454 :
455 4 : pthread_mutex_err:
456 4 : free(rscb);
457 4 : rscb = NULL;
458 4 : return ret;
459 : }
460 :
461 2 : RS_ATTRI_VISI_DEF int RsGetTlsEnable(unsigned int phyId, bool *tlsEnable)
462 : {
463 2 : struct rs_cb *rsCb = NULL;
464 : int ret;
465 :
466 2 : CHK_PRT_RETURN(tlsEnable == NULL, hccp_err("param err, tlsEnable is NULL"), -EINVAL);
467 1 : ret = RsGetRsCb(phyId, &rsCb);
468 1 : CHK_PRT_RETURN(ret != 0, hccp_err("RsGetRsCb failed, phyId(%u) invalid, ret(%d)", phyId, ret), ret);
469 :
470 0 : *tlsEnable = (rsCb->sslEnable == 0) ? false : true;
471 0 : return 0;
472 : }
473 :
474 1 : RS_ATTRI_VISI_DEF int RsGetHccnCfg(unsigned int phyId, enum HccnCfgKey key, char *value, unsigned int *valueLen)
475 : {
476 : #define HCCN_CFGFILE_PATH "/etc/hccl.cfg"
477 1 : const char *keyName[HCCN_CFG_KEY_INVALID] = {
478 : "udp_port_mode", "multi_qp_count", "multi_qp_udp_ports", "resv_mem", "qos_dscp"};
479 1 : unsigned int cfg_key = (unsigned int)key;
480 1 : unsigned int valLen = 0;
481 : unsigned int bufLen;
482 1 : int ret = 0;
483 :
484 1 : CHK_PRT_RETURN(value == NULL || valueLen == NULL, hccp_err("param err, value or valueLen is NULL"), -EINVAL);
485 1 : CHK_PRT_RETURN(cfg_key >= HCCN_CFG_KEY_INVALID,
486 : hccp_err("param err, key[%u] should < [%d]", cfg_key, HCCN_CFG_KEY_INVALID), -EINVAL);
487 :
488 1 : bufLen = *valueLen;
489 1 : CHK_PRT_RETURN(bufLen < HCCN_CFG_MSG_DATA_LEN,
490 : hccp_err("param err, bufLen[%u] should >= [%u]", bufLen, HCCN_CFG_MSG_DATA_LEN), -EINVAL);
491 :
492 1 : *valueLen = 0;
493 1 : ret = FileReadCfg(HCCN_CFGFILE_PATH, (int)phyId, keyName[cfg_key], value, bufLen);
494 1 : CHK_PRT_RETURN(ret == FILE_OPT_INNER_PARAM_ERR || ret == FILE_OPT_SYS_READ_FILE_ERR,
495 : hccp_run_warn("get hccn cfg file unsuccessful, ret(%d)", ret), 0);
496 1 : CHK_PRT_RETURN(ret == FILE_OPT_NO_MEM_ERR, hccp_err("value_len > buf_len[%d], ret(%d)", bufLen, ret), -ENOMEM);
497 1 : CHK_PRT_RETURN(ret != 0, hccp_run_warn("get hccn cfg [%s] unsuccessful, ret(%d)", keyName[cfg_key], ret), 0);
498 :
499 1 : valLen = (unsigned int)strlen(value);
500 1 : *valueLen = (valLen == 0) ? valLen : (valLen + 1);
501 1 : return 0;
502 : }
503 :
504 1 : RS_ATTRI_VISI_DEF int RsBindHostpid(unsigned int chipId, pid_t pid)
505 : {
506 : #define QUERY_BIND_HOST_PID_TIME_US 10000
507 : #define QUERY_BIND_HOST_PID_CNT 12000
508 1 : struct rs_cb *rsCb = NULL;
509 : unsigned int hostPid;
510 : pid_t devPid;
511 : int ret;
512 : int i;
513 :
514 : // get current hccp pid on device
515 1 : devPid = getpid();
516 1 : CHK_PRT_RETURN(devPid < 0, hccp_err("getpid failed, ret:%d errno:%d", devPid, errno), -EINVAL);
517 :
518 : // query corresponding host_pid every 10ms, total timeout cost 120s
519 1 : for (i = 0; i < QUERY_BIND_HOST_PID_CNT; i++) {
520 1 : ret = DlDrvQueryProcessHostPid(devPid, NULL, NULL, &hostPid, NULL);
521 1 : if (ret == DRV_ERROR_NONE) {
522 1 : break;
523 : }
524 :
525 0 : usleep(QUERY_BIND_HOST_PID_TIME_US);
526 : }
527 :
528 1 : if (i >= QUERY_BIND_HOST_PID_CNT) {
529 0 : hccp_err("query process host_pid failed, i:%d >= %d ret:%d", i, QUERY_BIND_HOST_PID_CNT, ret);
530 0 : return -EINVAL;
531 : }
532 :
533 1 : if (pid != (pid_t)hostPid) {
534 0 : hccp_err("check process failed, pid from tsd: %d, process hostPid: %u", pid, hostPid);
535 0 : return -EINVAL;
536 : }
537 :
538 1 : hccp_dbg("dl_drv_query_process_host_pid success, total retry cnt:%d", i);
539 :
540 : // save host_pid for later setup sharemem
541 1 : ret = RsDev2rscb(chipId, &rsCb, false);
542 1 : CHK_PRT_RETURN(ret, hccp_err("get rs_cb failed, ret:%d, chipId:%u", ret, chipId), -ENODEV);
543 0 : rsCb->hostPid = pid;
544 :
545 0 : return 0;
546 : }
547 :
548 : #ifdef CUSTOM_INTERFACE
549 1 : STATIC int RsSetRscbGrpId(struct rs_cb *rsCb, unsigned int devId)
550 : {
551 1 : GrpQueryGroupIdInfo grpQueryOut = {0};
552 1 : unsigned int chipId = rsCb->chipId;
553 1 : GrpQueryGroupId grpQueryIn = {0};
554 1 : struct MemInfo memInfo = {0};
555 : unsigned int outLen;
556 : unsigned int grpId;
557 : int ret;
558 :
559 : // query grp_name
560 1 : ret = DlHalMemGetInfoEx(devId, MEM_INFO_TYPE_SVM_GRP_INFO, &memInfo);
561 1 : CHK_PRT_RETURN(ret, hccp_err("dl_hal_mem_get_info_ex failed, ret:%d chipId:%u devId:%u", ret, chipId, devId), ret);
562 :
563 1 : hccp_dbg("query group name success, chipId:%u devId:%u grp_name:%s", chipId, devId, memInfo.grp_info.name);
564 :
565 : // query grp_id
566 1 : ret = memcpy_s(&grpQueryIn.grpName, BUFF_GRP_NAME_LEN, &memInfo.grp_info.name, SVM_GRP_NAME_LEN);
567 1 : CHK_PRT_RETURN(ret, hccp_err("memcpy_s failed, ret:%d chipId:%u devId:%u", ret, chipId, devId), ret);
568 1 : outLen = (unsigned int)sizeof(grpQueryOut);
569 1 : ret = DlHalGrpQuery(GRP_QUERY_GROUP_ID, &grpQueryIn, sizeof(grpQueryIn), &grpQueryOut, &outLen);
570 1 : CHK_PRT_RETURN(ret, hccp_err("dl_hal_grp_query failed, ret:%d chipId:%u devId:%u", ret, chipId, devId), ret);
571 1 : grpId = (unsigned int)grpQueryOut.groupId;
572 :
573 : // set grp_id
574 1 : rsCb->grpId = grpId;
575 :
576 1 : hccp_dbg("query group id success, chipId:%u devId:%u grpId:%u grp_name:%s", chipId, devId, grpId,
577 : grpQueryIn.grpName);
578 1 : return 0;
579 : }
580 :
581 3 : STATIC int RsBindSibling(struct rs_cb *rsCb, int hostPid, unsigned int vfId, unsigned int devId)
582 : {
583 : #define QUERY_BIND_SIBLING_TIME_US 10000
584 : #define QUERY_BIND_SIBLING_CNT 12000
585 3 : struct halQueryDevpidInfo pidInfo = {0};
586 : pid_t aicpuPid;
587 : int ret;
588 : int i;
589 :
590 : // query aicpu pid
591 3 : pidInfo.hostpid = hostPid;
592 3 : pidInfo.devid = devId;
593 3 : pidInfo.proc_type = DEVDRV_PROCESS_CP1;
594 3 : ret = DlHalQueryDevPid(pidInfo, &aicpuPid);
595 3 : CHK_PRT_RETURN(ret != 0, hccp_err("dl_hal_query_dev_pid failed, ret:%d devId:%u", ret, devId), ret);
596 :
597 : // try to bind sibling every 10ms, total timeout cost 120s
598 1 : for (i = 0; i < QUERY_BIND_SIBLING_CNT; i++) {
599 1 : ret = DlHalMemBindSibling(hostPid, aicpuPid, vfId, devId, SVM_MEM_BIND_SP_GRP);
600 1 : if (ret == DRV_ERROR_NONE) {
601 1 : break;
602 : }
603 :
604 0 : usleep(QUERY_BIND_SIBLING_TIME_US);
605 : }
606 :
607 1 : if (i >= QUERY_BIND_SIBLING_CNT) {
608 0 : hccp_err("bind sibling to setup sharemem failed, i:%d >= %d ret:%d", i, QUERY_BIND_SIBLING_CNT, ret);
609 0 : return -EINVAL;
610 : }
611 :
612 1 : rsCb->aicpuPid = aicpuPid;
613 1 : hccp_dbg("dl_hal_mem_bind_sibling success, total retry cnt:%d", i);
614 :
615 1 : return 0;
616 : }
617 :
618 36 : int RsSetupSharemem(struct rs_cb *rsCb, bool backupFlag, unsigned int backupPhyid)
619 : {
620 36 : unsigned int chipId = rsCb->chipId;
621 36 : pid_t pid = rsCb->hostPid;
622 36 : int64_t deviceInfo = 0;
623 : unsigned int logicId;
624 : int ret;
625 :
626 : // setup sharemem or skipped already, no need to setup again
627 36 : if (rsCb->grpSetupFlag) {
628 4 : hccp_dbg("grp_setup_flag:%d grp_id:%u chipId:%u", rsCb->grpSetupFlag, rsCb->grpId, chipId);
629 4 : return 0;
630 : }
631 :
632 32 : ret = DlDrvDeviceGetIndexByPhyId(chipId, &logicId);
633 32 : CHK_PRT_RETURN(ret, hccp_err("dl_drv_device_get_index_by_phy_id failed, ret:%d chipId:%u", ret, chipId), ret);
634 32 : ret = DlHalGetDeviceInfo(logicId, MODULE_TYPE_SYSTEM, INFO_TYPE_VERSION, &deviceInfo);
635 32 : CHK_PRT_RETURN(ret != 0,
636 : hccp_err("dl_hal_get_device_info failed, ret:%d logicId:%u chipId:%u", ret, logicId, chipId), ret);
637 : // not 910b/910_93 and not protocol udma, skip to setup share mem
638 32 : if (DlHalPlatGetChip((uint64_t)deviceInfo) != CHIP_TYPE_910B_910_93 && rsCb->protocol != PROTOCOL_UDMA) {
639 29 : hccp_info("logicId:%u chipId:%u protocol:%d skip to setup share mem", logicId, chipId, rsCb->protocol);
640 29 : rsCb->grpSetupFlag = true;
641 29 : return 0;
642 : }
643 :
644 : // use backup info to setup share mem
645 3 : if (backupFlag) {
646 1 : ret = DlDrvGetLocalDevIdByHostDevId(backupPhyid, &logicId);
647 1 : CHK_PRT_RETURN(ret != 0, hccp_err("DlDrvGetLocalDevIdByHostDevId failed, phyId(%u), ret(%d)", backupPhyid, ret),
648 : ret);
649 1 : hccp_dbg("setup sharemem with backup, phyId:%u logicId:%u", backupPhyid, logicId);
650 : }
651 :
652 : // bind sibling, default vfid is 0; query & save grp_id on rs_cb
653 3 : ret = RsBindSibling(rsCb, pid, 0, logicId);
654 3 : CHK_PRT_RETURN(ret != 0, hccp_err("rs_bind_sibling failed, ret:%d logicId:%u chipId:%u", ret, logicId, chipId),
655 : ret);
656 :
657 : // query & save grp_id on rs_cb
658 1 : ret = RsSetRscbGrpId(rsCb, logicId);
659 1 : CHK_PRT_RETURN(ret, hccp_err("rs_set_rscb_grp_id failed, ret:%d logicId:%u chipId:%u", ret, logicId, chipId), ret);
660 :
661 1 : rsCb->grpSetupFlag = true;
662 1 : return 0;
663 : }
664 : #endif
665 :
666 34 : STATIC int RsCompareIpGid(struct rdev rdevInfo, union ibv_gid *gid)
667 : {
668 34 : return RsDrvCompareIpGid(rdevInfo.family, rdevInfo.localIp, gid);
669 : }
670 :
671 34 : int RsQueryGid(struct rdev rdevInfo, struct ibv_context *ibCtxTmp, uint8_t ibPort, int *gidIdx)
672 : {
673 : static const char *portStates[] = {"Nop", "Down", "Init", "Armed", "", "Active Defer"};
674 34 : struct ibv_port_attr attr = {0};
675 : enum ibv_gid_type_sysfs type;
676 : union ibv_gid gidTmp;
677 : int ret;
678 : int i;
679 :
680 34 : CHK_PRT_RETURN(gidIdx == NULL, hccp_err("gid_idx is NULL"), -EINVAL);
681 :
682 34 : ret = RsIbvQueryPort(ibCtxTmp, ibPort, &attr);
683 34 : CHK_PRT_RETURN(ret, hccp_err("ibv_query_port failed, ret %d ibPort %u", ret, ibPort), -EOPENSRC);
684 :
685 72 : for (i = 0; i < attr.gid_tbl_len; i++) {
686 70 : ret = RsIbvQueryGidType(ibCtxTmp, ibPort, (unsigned int)i, &type);
687 70 : CHK_PRT_RETURN(ret, hccp_err("query gid type failed i %d, ret %d", i, ret), -EOPENSRC);
688 70 : if (type != IBV_GID_TYPE_SYSFS_ROCE_V2) {
689 36 : continue;
690 : }
691 34 : ret = RsIbvQueryGid(ibCtxTmp, ibPort, i, &gidTmp);
692 34 : CHK_PRT_RETURN(ret, hccp_err("query gid failed i %d, ret %d", i, ret), -EOPENSRC);
693 34 : ret = RsCompareIpGid(rdevInfo, &gidTmp);
694 34 : if (ret == 0) {
695 32 : CHK_PRT_RETURN(attr.state != IBV_PORT_ACTIVE,
696 : hccp_err("port number %u state is %s", ibPort, portStates[attr.state]), -ENOLINK);
697 30 : *gidIdx = i;
698 30 : return 0;
699 : }
700 : }
701 :
702 2 : if (i == attr.gid_tbl_len) {
703 2 : return -EEXIST;
704 : }
705 0 : return 0;
706 : }
707 :
708 26 : STATIC int RsGetDevRdevIndex(struct RsRdevCb *rdevCb, unsigned int *rdevIndex, int index)
709 : {
710 : (void)rdevCb;
711 : (void)rdevIndex;
712 : (void)index;
713 : #ifdef CUSTOM_INTERFACE
714 26 : struct roce_dev_data rdevData = {0}; // lint !e565
715 : int retVal;
716 :
717 26 : if (RsIsCustomInterfaceSupported()) {
718 26 : RS_PTHREAD_MUTEX_LOCK(&rdevCb->rsCb->mutex);
719 : /*lint -e132*/
720 26 : rdevCb->devName = RsIbvGetDeviceName(rdevCb->devList[index]); // lint !e101
721 26 : retVal = RsRoceGetRoceDevData(rdevCb->devName, &rdevData); // lint !e101
722 : /*lint +e132*/
723 26 : if (retVal) {
724 0 : hccp_err("rs_roce_get_roce_dev_data failed, retVal:%d, devName:%s", retVal, rdevCb->devName);
725 0 : RS_PTHREAD_MUTEX_ULOCK(&rdevCb->rsCb->mutex);
726 0 : return retVal;
727 : }
728 26 : *rdevIndex = rdevData.rdev_index; // rdev_index is same to port_id
729 26 : rdevCb->rdevIndex = *rdevIndex;
730 26 : RS_PTHREAD_MUTEX_ULOCK(&rdevCb->rsCb->mutex);
731 : }
732 : #endif
733 26 : return 0;
734 : }
735 :
736 4 : STATIC int RsGetHostRdevIndex(struct rdev rdevInfo, struct RsRdevCb *rdevCb, unsigned int *rdevIndex, int index)
737 : {
738 4 : struct RsRdevCb *rdevCbTmp2 = NULL;
739 4 : struct RsRdevCb *rdevCbTmp = NULL;
740 4 : unsigned int tmpRdevIndex = 0;
741 :
742 4 : RS_PTHREAD_MUTEX_LOCK(&rdevCb->rsCb->mutex);
743 4 : rdevCb->devName = RsIbvGetDeviceName(rdevCb->devList[index]);
744 4 : if (rdevCb->devName == NULL) {
745 0 : hccp_err("rs_ibv_get_device_name failed, errno:%d", errno);
746 0 : RS_PTHREAD_MUTEX_ULOCK(&rdevCb->rsCb->mutex);
747 0 : return -EINVAL;
748 : }
749 :
750 : struct RsIpAddrInfo localIp;
751 4 : int ret = RsConvertIpAddr(rdevInfo.family, &rdevInfo.localIp, &localIp);
752 4 : if (ret != 0) {
753 1 : hccp_err("convert(ntop) ip failed, ret:%d", ret);
754 1 : RS_PTHREAD_MUTEX_ULOCK(&rdevCb->rsCb->mutex);
755 1 : return ret;
756 : }
757 :
758 3 : RS_LIST_GET_HEAD_ENTRY(rdevCbTmp, rdevCbTmp2, &rdevCb->rsCb->rdevList, list, struct RsRdevCb);
759 3 : for (; (&rdevCbTmp->list) != &rdevCb->rsCb->rdevList;
760 0 : rdevCbTmp = rdevCbTmp2, rdevCbTmp2 = list_entry(rdevCbTmp2->list.next, struct RsRdevCb, list)) {
761 0 : tmpRdevIndex = rdevCbTmp->rdevIndex;
762 0 : if (!RsCompareIpAddr(&rdevCbTmp->localIp, &localIp)) {
763 0 : *rdevIndex = tmpRdevIndex;
764 0 : rdevCb->rdevIndex = *rdevIndex;
765 0 : rdevCb->localIp = localIp;
766 0 : RS_PTHREAD_MUTEX_ULOCK(&rdevCb->rsCb->mutex);
767 0 : return 0;
768 : }
769 : }
770 :
771 3 : *rdevIndex = tmpRdevIndex + 1;
772 3 : rdevCb->rdevIndex = *rdevIndex;
773 3 : rdevCb->localIp = localIp;
774 3 : RS_PTHREAD_MUTEX_ULOCK(&rdevCb->rsCb->mutex);
775 3 : return 0;
776 : }
777 :
778 31 : STATIC int RsGetIbCtxAndRdevIndex(struct rdev rdevInfo, struct RsRdevCb *rdevCb, unsigned int *rdevIndex)
779 : {
780 31 : struct ibv_context *ibCtxTmp = NULL;
781 31 : int gidIndex = -1;
782 : int ret;
783 : int i;
784 :
785 31 : for (i = 0; (i < rdevCb->devNum) && (rdevCb->devList[i] != NULL); ++i) { // lint !e101
786 31 : ibCtxTmp = RsIbvOpenDevice(rdevCb->devList[i]);
787 31 : CHK_PRT_RETURN(ibCtxTmp == NULL, hccp_err("ibv_open_device failed !"), -ENODEV);
788 31 : ret = RsQueryGid(rdevInfo, ibCtxTmp, rdevCb->ibPort, &gidIndex);
789 31 : if (ret == 0) {
790 29 : if (rdevCb->rsCb->hccpMode == NETWORK_PEER_ONLINE) {
791 3 : ret = RsGetHostRdevIndex(rdevInfo, rdevCb, rdevIndex, i);
792 : } else {
793 26 : ret = RsGetDevRdevIndex(rdevCb, rdevIndex, i);
794 : }
795 29 : if (ret != 0) {
796 0 : hccp_err("get index failed, ret:%d", ret);
797 0 : RsIbvCloseDevice(ibCtxTmp);
798 0 : return ret;
799 : }
800 29 : ret = RsIbvQueryDevice(ibCtxTmp, &rdevCb->deviceAttr);
801 29 : if (ret != 0) {
802 0 : hccp_err("query device failed, ret:%d", ret);
803 0 : RsIbvCloseDevice(ibCtxTmp);
804 0 : return ret;
805 : }
806 29 : rdevCb->ibCtx = ibCtxTmp;
807 29 : return 0;
808 2 : } else if (ret == -EEXIST) {
809 0 : RsIbvCloseDevice(ibCtxTmp);
810 : } else {
811 2 : RsIbvCloseDevice(ibCtxTmp);
812 2 : hccp_err("rs_query_gid failed, ret:%d", ret);
813 2 : return ret;
814 : }
815 : }
816 :
817 0 : CHK_PRT_RETURN(i == rdevCb->devNum,
818 : hccp_err("can not find ib_ctx for phyId[%u] local_ip[0x%x] in dev_list!", rdevInfo.phyId,
819 : rdevInfo.localIp.addr.s_addr),
820 : -EEXIST);
821 0 : return 0;
822 : }
823 :
824 41 : int RsGetRsCb(unsigned int phyId, struct rs_cb **rsCb)
825 : {
826 : unsigned int chipId;
827 : int ret;
828 :
829 41 : CHK_PRT_RETURN(phyId >= RS_MAX_DEV_NUM, hccp_err("rs set param error ! phyId:%u", phyId), -EINVAL);
830 41 : ret = rsGetLocalDevIDByHostDevID(phyId, &chipId);
831 41 : CHK_PRT_RETURN(ret, hccp_err("phyId[%u] invalid, ret %d", phyId, ret), ret);
832 :
833 33 : ret = RsDev2rscb(chipId, rsCb, false);
834 33 : CHK_PRT_RETURN(ret, hccp_err("get rs_cb failed, ret:%d", ret), -ENODEV);
835 33 : return 0;
836 : }
837 :
838 31 : STATIC int RsGetSqDepthAndQpMaxNum(struct RsRdevCb *rdevCb, unsigned int rdevIndex)
839 : {
840 : (void)rdevCb;
841 : (void)rdevIndex;
842 : #ifdef CUSTOM_INTERFACE
843 31 : unsigned int tempDepth = 0;
844 31 : unsigned int qpMaxNum = 0;
845 31 : unsigned int sqDepth = 0;
846 : int ret;
847 :
848 31 : if (RsIsCustomInterfaceSupported()) {
849 31 : ret = RsRoceGetTsqpDepth(rdevCb->devName, rdevIndex, &tempDepth, &qpMaxNum, &sqDepth);
850 31 : CHK_PRT_RETURN(ret,
851 : hccp_err("rs_roce_get_tsqp_depth failed, ret:%d, devName:%s, rdevIndex:%u", ret, rdevCb->devName,
852 : rdevIndex),
853 : ret);
854 :
855 31 : rdevCb->txDepth = sqDepth;
856 31 : rdevCb->rxDepth = sqDepth;
857 31 : rdevCb->qpMaxNum = qpMaxNum;
858 31 : hccp_run_info("qp_max_num:%u, sqDepth:%u", qpMaxNum, sqDepth);
859 : }
860 : #endif
861 31 : return 0;
862 : }
863 :
864 31 : STATIC int RsSetupPdAndNotify(struct RsRdevCb *rdevCb)
865 : {
866 : int ret;
867 :
868 31 : ret = RsDrvQueryNotifyAndAllocPd(rdevCb);
869 31 : CHK_PRT_RETURN(ret, hccp_err("rs_drv_query_notify_and_alloc_pd failed, ret[%d]", ret), ret);
870 :
871 30 : ret = RsDrvRegNotifyMr(rdevCb);
872 30 : if (ret) {
873 1 : hccp_err("reg notify mr failed, ret[%d]", ret);
874 1 : goto dealloc_pd;
875 : }
876 :
877 29 : return 0;
878 1 : dealloc_pd:
879 1 : RsIbvDeallocPd(rdevCb->ibPd);
880 1 : return ret;
881 : }
882 :
883 37 : STATIC int RsRdevCbInfoInit(struct rdev rdevInfo, struct rs_cb *rsCb, struct RsRdevCb *rdevCb)
884 : {
885 : int ret;
886 :
887 37 : rdevCb->ibPort = RS_PORT_DEF;
888 37 : rdevCb->rsCb = rsCb;
889 37 : rdevCb->notifyVaBase = rsCb->notifyVaBase;
890 37 : rdevCb->notifySize = rsCb->notifySize;
891 :
892 37 : rdevCb->localIp.family = (uint32_t)rdevInfo.family;
893 37 : rdevCb->localIp.binAddr = rdevInfo.localIp;
894 37 : ret = RsInetNtop(rdevInfo.family, &(rdevInfo.localIp), rdevCb->localIp.readAddr, RS_MAX_IP_LEN);
895 37 : CHK_PRT_RETURN(ret, hccp_err("rs_inet_ntop failed, ret %d", ret), -EINVAL);
896 :
897 37 : return 0;
898 : }
899 :
900 37 : STATIC int RsRdevCbInit(struct rdev rdevInfo, struct RsRdevCb *rdevCb, struct rs_cb *rsCb, unsigned int *rdevIndex)
901 : {
902 : int ret;
903 :
904 37 : ret = RsRdevCbInfoInit(rdevInfo, rsCb, rdevCb);
905 37 : CHK_PRT_RETURN(ret, hccp_err("rs_rdev_cb_info_init failed, ret %d", ret), ret);
906 :
907 37 : ret = pthread_mutex_init(&rdevCb->rdevMutex, NULL);
908 37 : CHK_PRT_RETURN(ret, hccp_err("rdev_cb mutex_init failed ret %d!, normal ret 0", ret), -ESYSFUNC);
909 :
910 36 : ret = pthread_mutex_init(&rdevCb->cqeErrCntMutex, NULL);
911 36 : if (ret) {
912 0 : hccp_err("rdev_cb cqe_err_cnt_mutex init failed ret %d!, normal ret 0", ret);
913 0 : goto destroy_rdev_mutex;
914 : }
915 :
916 36 : RS_PTHREAD_MUTEX_LOCK(&rdevCb->rdevMutex);
917 36 : RS_INIT_LIST_HEAD(&rdevCb->qpList);
918 36 : RS_INIT_LIST_HEAD(&rdevCb->typicalMrList);
919 36 : RS_PTHREAD_MUTEX_ULOCK(&rdevCb->rdevMutex);
920 :
921 36 : ret = RsGetIbCtxAndRdevIndex(rdevInfo, rdevCb, rdevIndex);
922 36 : if (ret) {
923 3 : hccp_err("rs_get_ib_ctx_and_rdev_index failed, ret:%d", ret);
924 3 : goto destroy_cqe_mutex;
925 : }
926 :
927 33 : ret = RsGetSqDepthAndQpMaxNum(rdevCb, *rdevIndex);
928 33 : if (ret) {
929 1 : hccp_err("rs_get_sq_depth_and_qp_max_num failed, ret[%d], rdevIndex[%u]", ret, *rdevIndex);
930 1 : goto close_dev;
931 : }
932 :
933 : #ifdef CUSTOM_INTERFACE
934 32 : if (RsIsCustomInterfaceSupported()) {
935 32 : ret = RsRoceMmapAiDbReg(rdevCb->ibCtx, (unsigned int)rdevCb->rsCb->aicpuPid);
936 32 : if (ret) {
937 0 : hccp_err("rs_roce_mmap_ai_db_reg failed, ret[%d], rdevIndex[%u]", ret, *rdevIndex);
938 0 : goto close_dev;
939 : }
940 : }
941 : #endif
942 :
943 32 : ret = RsInitNdaCb(rdevCb);
944 32 : if (ret != 0) {
945 0 : hccp_err("RsInitNdaCb failed, ret[%d], rdevIndex[%u]", ret, *rdevIndex);
946 0 : goto unmmap_ai_db;
947 : }
948 :
949 32 : ret = RsSetupPdAndNotify(rdevCb);
950 32 : if (ret != 0) {
951 3 : hccp_err("RsSetupPdAndNotify failed, ret[%d], rdevIndex[%u]", ret, *rdevIndex);
952 3 : goto free_nda_cb;
953 : }
954 :
955 29 : return 0;
956 :
957 3 : free_nda_cb:
958 3 : RsDeinitNdaCb(rdevCb);
959 3 : unmmap_ai_db:
960 : #ifdef CUSTOM_INTERFACE
961 3 : if (RsIsCustomInterfaceSupported()) {
962 3 : (void)RsRoceUnmmapAiDbReg(rdevCb->ibCtx);
963 : }
964 : #endif
965 0 : close_dev:
966 4 : RsIbvCloseDevice(rdevCb->ibCtx);
967 7 : destroy_cqe_mutex:
968 7 : pthread_mutex_destroy(&rdevCb->cqeErrCntMutex);
969 7 : destroy_rdev_mutex:
970 7 : pthread_mutex_destroy(&rdevCb->rdevMutex);
971 7 : return ret;
972 : }
973 :
974 4 : int RsRetryTimeoutExceptionCheck(struct SensorNode *sensorNode)
975 : {
976 4 : int ret = 0;
977 :
978 : /* sensor may not support, handle is 0 */
979 4 : if (sensorNode->sensorHandle == 0) {
980 2 : return 0;
981 : }
982 :
983 : /*
984 : * The notification alarm framework does not filter alarms. In this example, only one notification
985 : * alarm is reported by a single process, which does not need to be accurate. Therefore, no lock is used.
986 : */
987 2 : if (sensorNode->sensorUpdateCnt == 0) {
988 2 : ret = DlHalSensorNodeUpdateState(sensorNode->logicDevid, sensorNode->sensorHandle,
989 : RDMA_CQE_ERR_RETRY_TIMEOUT_EVENT_TYPE, GENERAL_EVENT_TYPE_ONE_TIME);
990 2 : if (ret == 0) {
991 1 : sensorNode->sensorUpdateCnt++;
992 : }
993 : }
994 :
995 2 : return ret;
996 : }
997 :
998 39 : STATIC int RsRdevInitWithBackupInfo(struct rdev rdevInfo, struct RsBackupInfo backupInfo, unsigned int notifyType,
999 : unsigned int *rdevIndex)
1000 : {
1001 39 : unsigned int phyId = rdevInfo.phyId;
1002 39 : struct RsRdevCb *rdevCb = NULL;
1003 39 : struct rs_cb *rsCb = NULL;
1004 : int ret;
1005 :
1006 39 : RS_CHECK_POINTER_NULL_RETURN_INT(rdevIndex);
1007 :
1008 39 : ret = RsApiInit();
1009 39 : CHK_PRT_RETURN(ret, hccp_err("RsApiInit failed! ret[%d]", ret), ret);
1010 :
1011 39 : ret = RsGetRsCb(phyId, &rsCb);
1012 39 : if (ret) {
1013 8 : hccp_err("RsGetRsCb failed, phyId[%u] invalid, ret %d", phyId, ret);
1014 8 : goto get_rs_cb_fail;
1015 : }
1016 :
1017 31 : rdevCb = calloc(1, sizeof(struct RsRdevCb));
1018 31 : if (rdevCb == NULL) {
1019 0 : hccp_err("calloc for rdev_cb failed");
1020 0 : ret = -ENOMEM;
1021 0 : goto get_rs_cb_fail;
1022 : }
1023 :
1024 31 : rdevCb->backupInfo.backupFlag = backupInfo.backupFlag;
1025 31 : (void)memcpy_s(&rdevCb->backupInfo.rdevInfo, sizeof(struct rdev), &backupInfo.rdevInfo, sizeof(struct rdev));
1026 : #ifdef CUSTOM_INTERFACE
1027 31 : if (RsIsCustomInterfaceSupported()) {
1028 : // setup sharemem for aicpu rdma unfold
1029 31 : ret = RsSetupSharemem(rsCb, rdevCb->backupInfo.backupFlag, rdevCb->backupInfo.rdevInfo.phyId);
1030 31 : if (ret != 0) {
1031 0 : hccp_err("[init][rs_rdev]RsSetupSharemem failed, phyId(%u), ret(%d)", phyId, ret);
1032 0 : goto free_rs_cb;
1033 : }
1034 : }
1035 : #endif
1036 :
1037 31 : rdevCb->notifyType = notifyType;
1038 31 : rdevCb->devList = RsIbvGetDeviceList(&(rdevCb->devNum));
1039 31 : if (rdevCb->devList == NULL || rdevCb->devNum == 0) {
1040 0 : hccp_err("dev_list is NULL, or devNum[%d] is 0", rdevCb->devNum);
1041 0 : ret = -EINVAL;
1042 0 : goto free_rs_cb;
1043 : }
1044 :
1045 31 : ret = RsSensorNodeRegister(phyId, rsCb);
1046 31 : if (ret != 0) {
1047 0 : hccp_err("[init][rs_rdev]rs_sensor_node_register failed, phyId(%u), ret(%d)", phyId, ret);
1048 0 : goto free_dev_list;
1049 : }
1050 :
1051 31 : hccp_info("ibv_get_device_list phyId[%d] dev_num[%d]", phyId, rdevCb->devNum);
1052 :
1053 31 : ret = RsRdevCbInit(rdevInfo, rdevCb, rsCb, rdevIndex);
1054 31 : if (ret != 0) {
1055 2 : RsSensorNodeUnregister(rdevCb->rsCb);
1056 2 : hccp_err("rs_rdev_cb_init failed ret %d!, normal ret 0", ret);
1057 2 : goto free_dev_list;
1058 : }
1059 :
1060 29 : RS_PTHREAD_MUTEX_LOCK(&rsCb->mutex);
1061 29 : RsListAddTail(&rdevCb->list, &rsCb->rdevList);
1062 29 : RS_PTHREAD_MUTEX_ULOCK(&rsCb->mutex);
1063 :
1064 29 : hccp_run_info("rdev init success, phyId:%u, localIp:0x%x, rdevIndex:%u", phyId, rdevInfo.localIp.addr.s_addr,
1065 : *rdevIndex);
1066 29 : return 0;
1067 :
1068 2 : free_dev_list:
1069 2 : RsIbvFreeDeviceList(rdevCb->devList);
1070 2 : free_rs_cb:
1071 2 : free(rdevCb);
1072 2 : rdevCb = NULL;
1073 10 : get_rs_cb_fail:
1074 10 : RsApiDeinit();
1075 10 : return ret;
1076 : }
1077 :
1078 1 : RS_ATTRI_VISI_DEF int RsRdevInitWithBackup(struct rdev rdevInfo, struct rdev backupRdevInfo, unsigned int notifyType,
1079 : unsigned int *rdevIndex)
1080 : {
1081 1 : struct RsBackupInfo backupInfo = {0};
1082 :
1083 1 : backupInfo.backupFlag = true;
1084 1 : (void)memcpy_s(&backupInfo.rdevInfo, sizeof(struct rdev), &backupRdevInfo, sizeof(struct rdev));
1085 :
1086 1 : return RsRdevInitWithBackupInfo(rdevInfo, backupInfo, notifyType, rdevIndex);
1087 : }
1088 :
1089 38 : RS_ATTRI_VISI_DEF int RsRdevInit(struct rdev rdevInfo, unsigned int notifyType, unsigned int *rdevIndex)
1090 : {
1091 38 : struct RsBackupInfo backupInfo = {0};
1092 :
1093 38 : return RsRdevInitWithBackupInfo(rdevInfo, backupInfo, notifyType, rdevIndex);
1094 : }
1095 :
1096 29 : STATIC void RsDestroyQpList(unsigned int phyId, unsigned int rdevIndex, struct RsRdevCb *rdevCb, struct RsQpCb *qpCb,
1097 : struct RsQpCb *qpCb2)
1098 : {
1099 : int ret;
1100 :
1101 29 : if (!RsListEmpty(&rdevCb->qpList)) {
1102 0 : hccp_warn("qp list do not empty!");
1103 0 : RS_LIST_GET_HEAD_ENTRY(qpCb, qpCb2, &rdevCb->qpList, list, struct RsQpCb);
1104 0 : for (; (&qpCb->list) != &rdevCb->qpList;
1105 0 : qpCb = qpCb2, qpCb2 = list_entry(qpCb2->list.next, struct RsQpCb, list)) {
1106 0 : hccp_info("qpn[%u] will be destroyed", qpCb->ibQp->qp_num);
1107 0 : if (qpCb->ibQpEx != NULL) {
1108 0 : ret = RsNdaQpDestroy(phyId, rdevIndex, qpCb->ibQp->qp_num);
1109 : } else {
1110 0 : ret = RsQpDestroy(phyId, rdevIndex, qpCb->ibQp->qp_num);
1111 : }
1112 0 : if (ret != 0) {
1113 0 : hccp_err("rs_qp_destroy failed, ret:%d", ret);
1114 : }
1115 : }
1116 : }
1117 :
1118 29 : return;
1119 : }
1120 :
1121 29 : STATIC void RsFreeTypicalMrCb(struct RsRdevCb *devCb)
1122 : {
1123 29 : struct RsListHead *typicalMrList = &devCb->typicalMrList;
1124 29 : struct RsMrCb *mrCurr = NULL;
1125 29 : struct RsMrCb *mrNext = NULL;
1126 :
1127 29 : RS_PTHREAD_MUTEX_LOCK(&devCb->rdevMutex);
1128 29 : RS_LIST_GET_HEAD_ENTRY(mrCurr, mrNext, typicalMrList, list, struct RsMrCb);
1129 29 : for (; (&mrCurr->list) != typicalMrList;
1130 0 : mrCurr = mrNext, mrNext = list_entry(mrNext->list.next, struct RsMrCb, list)) {
1131 0 : (void)RsDrvMrDereg(mrCurr->ibMr);
1132 0 : RsListDel(&mrCurr->list);
1133 0 : free(mrCurr);
1134 0 : mrCurr = NULL;
1135 : }
1136 29 : RS_PTHREAD_MUTEX_ULOCK(&devCb->rdevMutex);
1137 :
1138 29 : hccp_info("rs_free_typical_mr_cb is succ");
1139 29 : }
1140 :
1141 31 : RS_ATTRI_VISI_DEF int RsRdevDeinit(unsigned int phyId, unsigned int notifyType, unsigned int rdevIndex)
1142 : {
1143 : (void)notifyType;
1144 31 : struct RsRdevCb *rdevCb = NULL;
1145 31 : struct RsQpCb *qpCb2 = NULL;
1146 31 : struct RsQpCb *qpCb = NULL;
1147 : unsigned int chipId;
1148 : int ret;
1149 :
1150 31 : hccp_info("rdev deinit start, phyId:%u, rdevIndex:%u", phyId, rdevIndex);
1151 31 : CHK_PRT_RETURN(phyId >= RS_MAX_DEV_NUM, hccp_err("rs set param error ! phyId:%u", phyId), -EINVAL);
1152 31 : ret = rsGetLocalDevIDByHostDevID(phyId, &chipId);
1153 31 : CHK_PRT_RETURN(ret, hccp_err("phyId[%u] invalid, ret %d", phyId, ret), ret);
1154 :
1155 29 : ret = RsRdev2rdevCb(chipId, rdevIndex, &rdevCb);
1156 29 : CHK_PRT_RETURN(ret || rdevCb == NULL, hccp_err("rs_rdev2rdev_cb for chipId[%u] failed, ret %d", chipId, ret), ret);
1157 :
1158 29 : if (rdevCb->notifyType != NO_USE && rdevCb->notifyMr != NULL) {
1159 27 : ret = RsDrvMrDereg(rdevCb->notifyMr);
1160 27 : if (ret) {
1161 0 : hccp_err("rs_drv_mr_dereg failed, ret %d", ret);
1162 : }
1163 : }
1164 :
1165 29 : hccp_info("poll_cqe_num[%d]", rdevCb->pollCqeNum);
1166 :
1167 29 : RsDestroyQpList(phyId, rdevIndex, rdevCb, qpCb, qpCb2);
1168 :
1169 29 : RsFreeTypicalMrCb(rdevCb);
1170 :
1171 : #ifdef CUSTOM_INTERFACE
1172 29 : if (RsIsCustomInterfaceSupported()) {
1173 29 : (void)RsRoceUnmmapAiDbReg(rdevCb->ibCtx);
1174 : }
1175 : #endif
1176 :
1177 29 : RsIbvDeallocPd(rdevCb->ibPd);
1178 :
1179 29 : RsDeinitNdaCb(rdevCb);
1180 :
1181 29 : RsIbvCloseDevice(rdevCb->ibCtx);
1182 :
1183 : #ifdef CUSTOM_INTERFACE
1184 29 : if (RsIsCustomInterfaceSupported()) {
1185 29 : RsCloseBackupIbCtx(rdevCb);
1186 : }
1187 : #endif
1188 :
1189 29 : pthread_mutex_destroy(&rdevCb->cqeErrCntMutex);
1190 :
1191 29 : pthread_mutex_destroy(&rdevCb->rdevMutex);
1192 :
1193 29 : RsIbvFreeDeviceList(rdevCb->devList);
1194 :
1195 29 : RS_PTHREAD_MUTEX_LOCK(&gRsCb->mutex);
1196 29 : RsListDel(&rdevCb->list);
1197 29 : RS_PTHREAD_MUTEX_ULOCK(&gRsCb->mutex);
1198 29 : RsSensorNodeUnregister(rdevCb->rsCb);
1199 29 : RsApiDeinit();
1200 29 : hccp_run_info("rdev deinit success, phyId:%u, rdevIndex:%u", phyId, rdevIndex);
1201 29 : free(rdevCb);
1202 29 : rdevCb = NULL;
1203 29 : return 0;
1204 : }
1205 :
1206 0 : STATIC void RsHeterogTcpFreeFdNode(struct RsHeterogTcpFdInfo *fdNode)
1207 : {
1208 : int fd;
1209 :
1210 0 : RS_PTHREAD_MUTEX_LOCK(&gRsCb->mutex);
1211 0 : fd = fdNode->fd;
1212 0 : RsListDel(&fdNode->list);
1213 0 : free(fdNode);
1214 0 : fdNode = NULL;
1215 0 : gRsCb->fdMap[fd] = NULL;
1216 0 : RS_PTHREAD_MUTEX_ULOCK(&gRsCb->mutex);
1217 0 : }
1218 :
1219 1 : RS_ATTRI_VISI_DEF int RsEpollCtlAdd(const void *fdHandle, enum RaEpollEvent event)
1220 : {
1221 1 : struct RsHeterogTcpFdInfo *fdNode = NULL;
1222 1 : unsigned int tmpEvent = event;
1223 1 : int fd = RS_FD_INVALID;
1224 : int ret;
1225 :
1226 1 : if (event == RA_EPOLLONESHOT) {
1227 0 : tmpEvent = EPOLLIN | EPOLLET | EPOLLONESHOT;
1228 1 : } else if (event == RA_EPOLLIN) {
1229 1 : tmpEvent = EPOLLIN;
1230 : } else {
1231 0 : hccp_err("unknown event[%u]", tmpEvent);
1232 0 : return -EINVAL;
1233 : }
1234 :
1235 1 : if (gRsCb == NULL) {
1236 1 : gRsCb = RsGetCurRsCb();
1237 1 : if (gRsCb == NULL) {
1238 1 : hccp_err("[rs_epoll_ctl_add]rs_get_cur_rs_cb failed rs_cb(NULL)");
1239 1 : return -EINVAL;
1240 : }
1241 : }
1242 0 : tmpEvent = tmpEvent | EPOLLRDHUP;
1243 0 : fdNode = calloc(1, sizeof(struct RsHeterogTcpFdInfo));
1244 0 : CHK_PRT_RETURN(fdNode == NULL, hccp_err("no memory for fd_node"), -ENOMEM);
1245 :
1246 0 : fd = ((const struct SocketPeerInfo *)fdHandle)->fd;
1247 0 : fdNode->fd = fd;
1248 0 : RS_PTHREAD_MUTEX_LOCK(&gRsCb->mutex);
1249 0 : RsListAddTail(&fdNode->list, &gRsCb->heterogTcpFdList);
1250 0 : gRsCb->fdMap[fd] = fdHandle;
1251 0 : RS_PTHREAD_MUTEX_ULOCK(&gRsCb->mutex);
1252 0 : ret = RsEpollCtl(gRsCb->connCb.epollfd, EPOLL_CTL_ADD, fd, tmpEvent);
1253 0 : if (ret != 0) {
1254 0 : hccp_err("[rs_epoll_ctl_add]RsEpollCtl failed ret(%d), fd:%d, event:%u", ret, fd, event);
1255 0 : goto out;
1256 : }
1257 0 : return 0;
1258 0 : out:
1259 0 : RsHeterogTcpFreeFdNode(fdNode);
1260 0 : fdNode = NULL;
1261 0 : return ret;
1262 : }
1263 :
1264 5 : RS_ATTRI_VISI_DEF int RsEpollCtlMod(const void *fdHandle, enum RaEpollEvent event)
1265 : {
1266 5 : unsigned int tmpEvent = event;
1267 5 : int fd = RS_FD_INVALID;
1268 : int ret;
1269 :
1270 5 : if (event == RA_EPOLLONESHOT) {
1271 1 : tmpEvent = EPOLLIN | EPOLLET | EPOLLONESHOT;
1272 4 : } else if (event == RA_EPOLLIN) {
1273 3 : tmpEvent = EPOLLIN;
1274 : } else {
1275 1 : hccp_err("unknown event[%u]", event);
1276 1 : return -EINVAL;
1277 : }
1278 :
1279 4 : tmpEvent = tmpEvent | EPOLLRDHUP;
1280 4 : fd = ((const struct SocketPeerInfo *)fdHandle)->fd;
1281 :
1282 4 : if (gRsCb == NULL) {
1283 1 : gRsCb = RsGetCurRsCb();
1284 1 : if (gRsCb == NULL) {
1285 1 : hccp_err("[rs_epoll_ctl_mod]rs_get_cur_rs_cb failed rs_cb(NULL)");
1286 1 : return -EINVAL;
1287 : }
1288 : }
1289 :
1290 3 : ret = RsEpollCtl(gRsCb->connCb.epollfd, EPOLL_CTL_MOD, fd, tmpEvent);
1291 3 : CHK_PRT_RETURN(ret, hccp_err("[rs_epoll_ctl_mod]RsEpollCtl failed ret(%d), fd:%d, event:%u", ret, fd, event), ret);
1292 1 : return 0;
1293 : }
1294 :
1295 3 : RS_ATTRI_VISI_DEF int RsEpollCtlDel(int fd)
1296 : {
1297 : int ret;
1298 3 : struct RsHeterogTcpFdInfo *fdNode = NULL;
1299 3 : struct RsHeterogTcpFdInfo *fdNode1 = NULL;
1300 :
1301 3 : if (gRsCb == NULL) {
1302 1 : gRsCb = RsGetCurRsCb();
1303 1 : if (gRsCb == NULL) {
1304 1 : hccp_err("[rs_epoll_ctl_del]rs_get_cur_rs_cb failed rs_cb(NULL)");
1305 1 : return -EINVAL;
1306 : }
1307 : }
1308 2 : RS_LIST_GET_HEAD_ENTRY(fdNode, fdNode1, &gRsCb->heterogTcpFdList, list, struct RsHeterogTcpFdInfo);
1309 2 : for (; (&fdNode->list) != &gRsCb->heterogTcpFdList;
1310 0 : fdNode = fdNode1, fdNode1 = list_entry(fdNode1->list.next, struct RsHeterogTcpFdInfo, list)) {
1311 0 : if (fdNode->fd == fd) {
1312 : // 删除节点
1313 0 : RsHeterogTcpFreeFdNode(fdNode);
1314 0 : fdNode = NULL;
1315 0 : break; // lint !e108
1316 : }
1317 : }
1318 :
1319 : // 为了兼容epoll不同版本，这里加EPOLLIN参数
1320 2 : ret = RsEpollCtl(gRsCb->connCb.epollfd, EPOLL_CTL_DEL, fd, EPOLLIN);
1321 2 : CHK_PRT_RETURN(ret, hccp_err("[rs_epoll_ctl_del]RsEpollCtl failed ret(%d), fd:%d", ret, fd), ret);
1322 1 : return 0;
1323 : }
1324 :
1325 3 : RS_ATTRI_VISI_DEF void RsSetTcpRecvCallback(const void *callback)
1326 : {
1327 3 : if (gRsCb == NULL) {
1328 1 : hccp_err("param error, gRsCb is NULL");
1329 1 : return;
1330 : }
1331 2 : gRsCb->tcpRecvCallback = (void (*)(const void *))callback;
1332 : }
1333 :
1334 1 : STATIC void RsFreeAcceptOneNode(struct rs_cb *rscb, struct RsAcceptInfo *accept)
1335 : {
1336 : int ret;
1337 :
1338 1 : ret = RsEpollCtl(rscb->connCb.epollfd, EPOLL_CTL_DEL, accept->connFd, EPOLLIN);
1339 1 : if (ret) {
1340 1 : hccp_err("epoll ctl del fd %d failed, ret:%d", accept->connFd, ret);
1341 : }
1342 :
1343 1 : RS_PTHREAD_MUTEX_LOCK(&rscb->connCb.connMutex);
1344 1 : RsListDel(&accept->list);
1345 1 : RS_PTHREAD_MUTEX_ULOCK(&rscb->connCb.connMutex);
1346 :
1347 1 : if (rscb->sslEnable == RS_SSL_ENABLE) {
1348 1 : if (accept->ssl == NULL) {
1349 0 : hccp_warn("[Server] accept->ssl is NULL, it maybe has not establish tls link");
1350 : } else {
1351 1 : ssl_adp_shutdown(accept->ssl);
1352 1 : ssl_adp_free(accept->ssl);
1353 1 : accept->ssl = NULL;
1354 : }
1355 : }
1356 :
1357 1 : RS_CLOSE_RETRY_FOR_EINTR(ret, accept->connFd);
1358 :
1359 1 : hccp_info("free accept_server IP:%s, port:%d, connFd:%d", accept->serverIpAddr.readAddr, accept->sockPort,
1360 : accept->connFd);
1361 1 : accept->connFd = RS_FD_INVALID;
1362 :
1363 1 : free(accept);
1364 1 : accept = NULL;
1365 1 : }
1366 :
1367 40 : STATIC void RsFreeAccpetList(struct rs_cb *rscb)
1368 : {
1369 40 : struct RsAcceptInfo *accept = NULL;
1370 40 : struct RsAcceptInfo *accept2 = NULL;
1371 :
1372 40 : if (!RsListEmpty(&rscb->connCb.serverAcceptList)) {
1373 0 : hccp_warn("Server accept list do not empty!");
1374 0 : RS_LIST_GET_HEAD_ENTRY(accept, accept2, &rscb->connCb.serverAcceptList, list, struct RsAcceptInfo);
1375 0 : for (; (&accept->list) != &rscb->connCb.serverAcceptList;
1376 0 : accept = accept2, accept2 = list_entry(accept2->list.next, struct RsAcceptInfo, list)) {
1377 0 : RsFreeAcceptOneNode(rscb, accept);
1378 0 : accept = NULL;
1379 : }
1380 : }
1381 :
1382 40 : return;
1383 : }
1384 :
1385 12 : STATIC void RsFreeDesignatedAccpetNode(struct rs_cb *rscb, struct RsIpAddrInfo *localIp)
1386 : {
1387 12 : struct RsAcceptInfo *accept = NULL;
1388 12 : struct RsAcceptInfo *accept2 = NULL;
1389 :
1390 12 : if (!RsListEmpty(&rscb->connCb.serverAcceptList)) {
1391 0 : RS_LIST_GET_HEAD_ENTRY(accept, accept2, &rscb->connCb.serverAcceptList, list, struct RsAcceptInfo);
1392 0 : for (; (&accept->list) != &rscb->connCb.serverAcceptList;
1393 0 : accept = accept2, accept2 = list_entry(accept2->list.next, struct RsAcceptInfo, list)) {
1394 0 : if (!RsCompareIpAddr(&accept->serverIpAddr, localIp)) {
1395 0 : RsFreeAcceptOneNode(rscb, accept);
1396 0 : accept = NULL;
1397 : }
1398 : }
1399 : }
1400 :
1401 12 : return;
1402 : }
1403 :
1404 2 : STATIC void RsFreeConnOneNode(struct rs_cb *rscb, struct RsConnInfo *conn)
1405 : {
1406 : int ret;
1407 :
1408 2 : RS_PTHREAD_MUTEX_LOCK(&rscb->connCb.connMutex);
1409 2 : RsListDel(&conn->list);
1410 2 : RS_PTHREAD_MUTEX_ULOCK(&rscb->connCb.connMutex);
1411 :
1412 2 : if (rscb->sslEnable == RS_SSL_ENABLE) {
1413 0 : if (conn->ssl == NULL) {
1414 0 : hccp_warn("[Client] conn->ssl is NULL, it maybe has not establish tls link");
1415 : } else {
1416 0 : ssl_adp_shutdown(conn->ssl);
1417 0 : ssl_adp_free(conn->ssl);
1418 0 : conn->ssl = NULL;
1419 : }
1420 : }
1421 :
1422 2 : RS_CLOSE_RETRY_FOR_EINTR(ret, conn->connfd);
1423 :
1424 2 : hccp_info("free for conn IP:%s, port:%d, connfd:%d, state:%u", conn->clientIp.readAddr, conn->port, conn->connfd,
1425 : conn->state);
1426 :
1427 2 : conn->connfd = RS_FD_INVALID;
1428 2 : conn->state = RS_CONN_STATE_RESET;
1429 :
1430 2 : free(conn);
1431 2 : conn = NULL;
1432 2 : }
1433 :
1434 40 : STATIC void RsFreeClientConnList(struct rs_cb *rscb)
1435 : {
1436 40 : struct RsConnInfo *conn = NULL;
1437 40 : struct RsConnInfo *conn2 = NULL;
1438 :
1439 40 : if (!RsListEmpty(&rscb->connCb.clientConnList)) {
1440 0 : hccp_warn("Client conn node do not empty!");
1441 0 : RS_LIST_GET_HEAD_ENTRY(conn, conn2, &rscb->connCb.clientConnList, list, struct RsConnInfo);
1442 0 : for (; (&conn->list) != &rscb->connCb.clientConnList;
1443 0 : conn = conn2, conn2 = list_entry(conn2->list.next, struct RsConnInfo, list)) {
1444 0 : RsFreeConnOneNode(rscb, conn);
1445 0 : conn = NULL;
1446 : }
1447 : }
1448 :
1449 40 : return;
1450 : }
1451 :
1452 12 : STATIC void RsFreeDesignatedClientConnNode(struct rs_cb *rscb, struct RsIpAddrInfo *localIp)
1453 : {
1454 12 : struct RsConnInfo *conn = NULL;
1455 12 : struct RsConnInfo *conn2 = NULL;
1456 :
1457 12 : if (!RsListEmpty(&rscb->connCb.clientConnList)) {
1458 0 : RS_LIST_GET_HEAD_ENTRY(conn, conn2, &rscb->connCb.clientConnList, list, struct RsConnInfo);
1459 0 : for (; (&conn->list) != &rscb->connCb.clientConnList;
1460 0 : conn = conn2, conn2 = list_entry(conn2->list.next, struct RsConnInfo, list)) {
1461 0 : if (!RsCompareIpAddr(&conn->clientIp, localIp)) {
1462 0 : hccp_warn("Client conn node for IP[%s] do not empty!", localIp->readAddr);
1463 0 : RsFreeConnOneNode(rscb, conn);
1464 0 : conn = NULL;
1465 : }
1466 : }
1467 : }
1468 :
1469 12 : return;
1470 : }
1471 :
1472 40 : STATIC void RsFreeServerConnList(struct rs_cb *rscb)
1473 : {
1474 40 : struct RsConnInfo *conn = NULL;
1475 40 : struct RsConnInfo *conn2 = NULL;
1476 :
1477 40 : if (!RsListEmpty(&rscb->connCb.serverConnList)) {
1478 1 : hccp_warn("Server conn node do not empty!");
1479 1 : RS_LIST_GET_HEAD_ENTRY(conn, conn2, &rscb->connCb.serverConnList, list, struct RsConnInfo);
1480 2 : for (; (&conn->list) != &rscb->connCb.serverConnList;
1481 1 : conn = conn2, conn2 = list_entry(conn2->list.next, struct RsConnInfo, list)) {
1482 1 : RsFreeConnOneNode(rscb, conn);
1483 1 : conn = NULL;
1484 : }
1485 : }
1486 :
1487 40 : return;
1488 : }
1489 :
1490 12 : STATIC void RsFreeDesignatedServerConnNode(struct rs_cb *rscb, struct RsIpAddrInfo *localIp)
1491 : {
1492 12 : struct RsConnInfo *conn = NULL;
1493 12 : struct RsConnInfo *conn2 = NULL;
1494 :
1495 12 : if (!RsListEmpty(&rscb->connCb.serverConnList)) {
1496 1 : RS_LIST_GET_HEAD_ENTRY(conn, conn2, &rscb->connCb.serverConnList, list, struct RsConnInfo);
1497 2 : for (; (&conn->list) != &rscb->connCb.serverConnList;
1498 1 : conn = conn2, conn2 = list_entry(conn2->list.next, struct RsConnInfo, list)) {
1499 1 : if (!RsCompareIpAddr(&conn->serverIp, localIp)) {
1500 1 : hccp_warn("Server conn node for IP[%s] do not empty!", localIp->readAddr);
1501 1 : RsFreeConnOneNode(rscb, conn);
1502 1 : conn = NULL;
1503 : }
1504 : }
1505 : }
1506 12 : return;
1507 : }
1508 :
1509 1 : STATIC void RsFreeListenOneNode(struct rs_cb *rscb, struct RsListenInfo *listen)
1510 : {
1511 : int ret;
1512 :
1513 1 : ret = RsEpollCtl(rscb->connCb.epollfd, EPOLL_CTL_DEL, listen->listenFd, EPOLLIN);
1514 1 : if (ret) {
1515 0 : hccp_err("delete from epoll failed, ret:%d, epollfd:%d, listenFd:%d", ret, rscb->connCb.epollfd,
1516 : listen->listenFd);
1517 : }
1518 :
1519 1 : RS_PTHREAD_MUTEX_LOCK(&rscb->connCb.connMutex);
1520 1 : RsListDel(&listen->list);
1521 1 : RS_PTHREAD_MUTEX_ULOCK(&rscb->connCb.connMutex);
1522 :
1523 1 : RS_CLOSE_RETRY_FOR_EINTR(ret, listen->listenFd);
1524 :
1525 1 : hccp_info("free Listen IP:%s, port:%d, listenFd:%d, state:%u", listen->serverIpAddr.readAddr,
1526 : ntohs(listen->sockPort), listen->listenFd, listen->state);
1527 :
1528 1 : listen->listenFd = RS_FD_INVALID;
1529 1 : listen->state = RS_CONN_STATE_RESET;
1530 :
1531 1 : free(listen);
1532 1 : }
1533 :
1534 40 : STATIC void RsFreeListenList(struct rs_cb *rscb)
1535 : {
1536 40 : struct RsListenInfo *listen = NULL;
1537 40 : struct RsListenInfo *listen2 = NULL;
1538 :
1539 40 : if (!RsListEmpty(&rscb->connCb.listenList)) {
1540 1 : hccp_warn("Server listen node do not empty!");
1541 1 : RS_LIST_GET_HEAD_ENTRY(listen, listen2, &rscb->connCb.listenList, list, struct RsListenInfo);
1542 2 : for (; (&listen->list) != &rscb->connCb.listenList;
1543 1 : listen = listen2, listen2 = list_entry(listen2->list.next, struct RsListenInfo, list)) {
1544 1 : RsFreeListenOneNode(rscb, listen);
1545 1 : listen = NULL;
1546 : }
1547 : }
1548 :
1549 40 : return;
1550 : }
1551 :
1552 12 : STATIC void RsFreeDesignatedListenNode(struct rs_cb *rscb, struct RsIpAddrInfo *localIp)
1553 : {
1554 12 : struct RsListenInfo *listen = NULL;
1555 12 : struct RsListenInfo *listen2 = NULL;
1556 :
1557 12 : if (!RsListEmpty(&rscb->connCb.listenList)) {
1558 0 : RS_LIST_GET_HEAD_ENTRY(listen, listen2, &rscb->connCb.listenList, list, struct RsListenInfo);
1559 0 : for (; (&listen->list) != &rscb->connCb.listenList;
1560 0 : listen = listen2, listen2 = list_entry(listen2->list.next, struct RsListenInfo, list)) {
1561 0 : if (!RsCompareIpAddr(&listen->serverIpAddr, localIp)) {
1562 0 : RsFreeListenOneNode(rscb, listen);
1563 0 : listen = NULL;
1564 : }
1565 : }
1566 : }
1567 :
1568 12 : return;
1569 : }
1570 :
1571 11 : STATIC void RsWhiteListNodeFree(struct rs_cb *rscb, struct RsWhiteList *wlist)
1572 : {
1573 11 : struct RsWhiteListInfo *wlistNode = NULL;
1574 11 : struct RsWhiteListInfo *wlistNode1 = NULL;
1575 :
1576 11 : if (!RsListEmpty(&wlist->whiteList)) {
1577 10 : RS_LIST_GET_HEAD_ENTRY(wlistNode, wlistNode1, &wlist->whiteList, list, struct RsWhiteListInfo);
1578 21 : for (; (&wlistNode->list) != &wlist->whiteList;
1579 11 : wlistNode = wlistNode1, wlistNode1 = list_entry(wlistNode1->list.next, struct RsWhiteListInfo, list)) {
1580 11 : RS_PTHREAD_MUTEX_LOCK(&rscb->connCb.connMutex);
1581 11 : RsListDel(&wlistNode->list);
1582 11 : RS_PTHREAD_MUTEX_ULOCK(&rscb->connCb.connMutex);
1583 :
1584 11 : hccp_info("free White list client IP:%s, tag:%s", wlistNode->clientIp.readAddr, wlistNode->tag);
1585 11 : free(wlistNode);
1586 11 : wlistNode = NULL;
1587 : }
1588 : }
1589 11 : }
1590 :
1591 11 : STATIC void RsFreeWhiteOneNode(struct rs_cb *rscb, struct RsWhiteList *wlist)
1592 : {
1593 11 : RsWhiteListNodeFree(rscb, wlist);
1594 :
1595 11 : RS_PTHREAD_MUTEX_LOCK(&rscb->connCb.connMutex);
1596 11 : RsListDel(&wlist->list);
1597 11 : RS_PTHREAD_MUTEX_ULOCK(&rscb->connCb.connMutex);
1598 :
1599 11 : hccp_info("White list server IP:%s", wlist->serverIp.readAddr);
1600 11 : free(wlist);
1601 11 : wlist = NULL;
1602 11 : }
1603 :
1604 40 : STATIC void RsFreeWhiteList(struct rs_cb *rscb)
1605 : {
1606 40 : struct RsWhiteList *wlist = NULL;
1607 40 : struct RsWhiteList *wlist2 = NULL;
1608 :
1609 40 : if (!RsListEmpty(&rscb->connCb.whiteList)) {
1610 1 : hccp_warn("Server white list do not empty!");
1611 1 : RS_LIST_GET_HEAD_ENTRY(wlist, wlist2, &rscb->connCb.whiteList, list, struct RsWhiteList);
1612 2 : for (; (&wlist->list) != &rscb->connCb.whiteList;
1613 1 : wlist = wlist2, wlist2 = list_entry(wlist2->list.next, struct RsWhiteList, list)) {
1614 1 : RsFreeWhiteOneNode(rscb, wlist);
1615 1 : wlist = NULL;
1616 : }
1617 : }
1618 :
1619 40 : return;
1620 : }
1621 :
1622 12 : STATIC void RsFreeDesignatedWhiteNode(struct rs_cb *rscb, struct RsIpAddrInfo *localIp)
1623 : {
1624 12 : struct RsWhiteList *wlist = NULL;
1625 12 : struct RsWhiteList *wlist2 = NULL;
1626 :
1627 12 : if (!RsListEmpty(&rscb->connCb.whiteList)) {
1628 10 : RS_LIST_GET_HEAD_ENTRY(wlist, wlist2, &rscb->connCb.whiteList, list, struct RsWhiteList);
1629 20 : for (; (&wlist->list) != &rscb->connCb.whiteList;
1630 10 : wlist = wlist2, wlist2 = list_entry(wlist2->list.next, struct RsWhiteList, list)) {
1631 10 : if (!RsCompareIpAddr(&wlist->serverIp, localIp)) {
1632 10 : RsFreeWhiteOneNode(rscb, wlist);
1633 10 : wlist = NULL;
1634 : }
1635 : }
1636 : }
1637 :
1638 12 : return;
1639 : }
1640 :
1641 12 : STATIC void RsFreeSocketList(struct rs_cb *rscb, struct RsIpAddrInfo *localIp)
1642 : {
1643 12 : RsFreeDesignatedAccpetNode(rscb, localIp);
1644 12 : RsFreeDesignatedClientConnNode(rscb, localIp);
1645 :
1646 12 : RsFreeDesignatedServerConnNode(rscb, localIp);
1647 :
1648 12 : RsFreeDesignatedListenNode(rscb, localIp);
1649 :
1650 12 : RsFreeDesignatedWhiteNode(rscb, localIp);
1651 :
1652 12 : return;
1653 : }
1654 :
1655 15 : RS_ATTRI_VISI_DEF int RsSocketDeinit(struct rdev rdevInfo)
1656 : {
1657 : int ret;
1658 15 : unsigned int phyId = rdevInfo.phyId;
1659 : unsigned int chipId;
1660 15 : struct rs_cb *rscb = NULL;
1661 :
1662 15 : hccp_info("rs socket deinit start, phyId:%u", phyId);
1663 15 : CHK_PRT_RETURN(phyId >= RS_MAX_DEV_NUM, hccp_err("rs set param error ! phyId:%u", phyId), -EINVAL);
1664 15 : ret = rsGetLocalDevIDByHostDevID(phyId, &chipId);
1665 15 : CHK_PRT_RETURN(ret, hccp_err("phyId[%u] invalid, ret %d", phyId, ret), ret);
1666 :
1667 15 : CHK_PRT_RETURN((rdevInfo.family != AF_INET) && (rdevInfo.family != AF_INET6),
1668 : hccp_err("family[%d] invalid", rdevInfo.family), -EPROTONOSUPPORT);
1669 :
1670 13 : if (rdevInfo.family == AF_INET) {
1671 13 : unsigned int *localIp = NULL;
1672 13 : localIp = &(rdevInfo.localIp.addr.s_addr);
1673 13 : ret = RsSocketNodeid2vnic(*localIp, localIp);
1674 13 : hccp_info("socket deinit local IP is 0x%llx, ret:%d", *localIp, ret);
1675 : }
1676 :
1677 : struct RsIpAddrInfo localIp;
1678 13 : RsConvertIpAddr(rdevInfo.family, &rdevInfo.localIp, &localIp);
1679 :
1680 13 : ret = RsDev2rscb(chipId, &rscb, false);
1681 13 : CHK_PRT_RETURN(ret, hccp_err("get rscb failed for chipId:%u, ret:%d", chipId, ret), -ENODEV);
1682 :
1683 12 : RS_PTHREAD_MUTEX_LOCK(&rscb->mutex);
1684 12 : RsFreeSocketList(rscb, &localIp);
1685 12 : RS_PTHREAD_MUTEX_ULOCK(&rscb->mutex);
1686 12 : hccp_run_info("socket deinit success, phyId:%u, localIp:%s", phyId, localIp.readAddr);
1687 12 : return 0;
1688 : }
1689 :
1690 44 : STATIC void RsFreeRdevList(struct rs_cb *rsCb)
1691 : {
1692 44 : struct RsRdevCb *rdevCbCurr = NULL;
1693 44 : struct RsRdevCb *rdevCbNext = NULL;
1694 44 : unsigned int phyId = 0;
1695 : int ret;
1696 :
1697 44 : if (RsListEmpty(&rsCb->rdevList)) {
1698 41 : return;
1699 : }
1700 :
1701 3 : hccp_run_warn("Rdev list is not empty!");
1702 3 : ret = rsGetDevIDByLocalDevID(rsCb->chipId, &phyId);
1703 3 : if (ret != 0) {
1704 1 : hccp_err("chipId[%u] invalid, ret %d", rsCb->chipId, ret);
1705 1 : return;
1706 : }
1707 :
1708 2 : RS_LIST_GET_HEAD_ENTRY(rdevCbCurr, rdevCbNext, &rsCb->rdevList, list, struct RsRdevCb);
1709 4 : for (; (&rdevCbCurr->list) != &rsCb->rdevList;
1710 2 : rdevCbCurr = rdevCbNext, rdevCbNext = list_entry(rdevCbNext->list.next, struct RsRdevCb, list)) {
1711 2 : ret = RsRdevDeinit(phyId, rdevCbCurr->notifyType, rdevCbCurr->rdevIndex);
1712 2 : if (ret != 0) {
1713 1 : hccp_err("rs_rdev_deinit failed, ret:%d, phyId:%u", ret, phyId);
1714 : }
1715 : }
1716 :
1717 2 : return;
1718 : }
1719 :
1720 43 : STATIC void RsFreeUdevList(struct rs_cb *rsCb)
1721 : {
1722 43 : struct RsUbDevCb *udevCbCurr = NULL;
1723 43 : struct RsUbDevCb *udevCbNext = NULL;
1724 : int ret;
1725 :
1726 43 : if (RsListEmpty(&rsCb->udevList)) {
1727 41 : return;
1728 : }
1729 :
1730 2 : hccp_run_warn("Udev list is not empty!");
1731 2 : RS_LIST_GET_HEAD_ENTRY(udevCbCurr, udevCbNext, &rsCb->udevList, list, struct RsUbDevCb);
1732 4 : for (; (&udevCbCurr->list) != &rsCb->udevList;
1733 2 : udevCbCurr = udevCbNext, udevCbNext = list_entry(udevCbNext->list.next, struct RsUbDevCb, list)) {
1734 2 : ret = RsUbCtxDeinit(udevCbCurr);
1735 2 : if (ret != 0) {
1736 1 : hccp_err("rs_ub_ctx_deinit failed, ret:%d", ret);
1737 : }
1738 : }
1739 :
1740 2 : return;
1741 : }
1742 :
1743 40 : STATIC void RsFreeHeterogTcpFdList(struct rs_cb *rsCb)
1744 : {
1745 40 : struct RsHeterogTcpFdInfo *fdNode = NULL;
1746 40 : struct RsHeterogTcpFdInfo *fdNode1 = NULL;
1747 :
1748 40 : if (!RsListEmpty(&rsCb->heterogTcpFdList)) {
1749 0 : hccp_warn("heterog_tcp_fd_list do not empty!");
1750 0 : RS_LIST_GET_HEAD_ENTRY(fdNode, fdNode1, &rsCb->heterogTcpFdList, list, struct RsHeterogTcpFdInfo);
1751 0 : for (; (&fdNode->list) != &rsCb->heterogTcpFdList;
1752 0 : fdNode = fdNode1, fdNode1 = list_entry(fdNode1->list.next, struct RsHeterogTcpFdInfo, list)) {
1753 0 : hccp_info(">>>>>fd_node->fd:%d", fdNode->fd);
1754 : // 删除节点
1755 0 : RS_PTHREAD_MUTEX_LOCK(&rsCb->mutex);
1756 0 : RsListDel(&fdNode->list);
1757 0 : free(fdNode);
1758 0 : fdNode = NULL;
1759 0 : RS_PTHREAD_MUTEX_ULOCK(&rsCb->mutex);
1760 : }
1761 : }
1762 :
1763 40 : return;
1764 : }
1765 :
1766 40 : STATIC void RsListFree(struct rs_cb *rscb)
1767 : {
1768 40 : RsFreeAccpetList(rscb);
1769 40 : RsFreeClientConnList(rscb);
1770 :
1771 40 : RsFreeServerConnList(rscb);
1772 :
1773 40 : RsFreeListenList(rscb);
1774 :
1775 40 : RsFreeWhiteList(rscb);
1776 :
1777 40 : return;
1778 : }
1779 :
1780 41 : STATIC void RsSslFree(struct rs_cb *rscb)
1781 : {
1782 41 : if (rscb->sslEnable == RS_SSL_ENABLE) {
1783 2 : if (rscb->skidSubjectCb != NULL) {
1784 1 : if (memset_s(rscb->skidSubjectCb, sizeof(struct RsCertSkidSubjectCb), 0,
1785 : sizeof(struct RsCertSkidSubjectCb))) {
1786 1 : hccp_warn("memset_s for skid_subject_cb unsuccessful");
1787 : }
1788 1 : free(rscb->skidSubjectCb);
1789 1 : rscb->skidSubjectCb = NULL;
1790 : }
1791 2 : ssl_adp_ctx_free(rscb->serverSslCtx);
1792 2 : rscb->serverSslCtx = NULL;
1793 2 : ssl_adp_ctx_free(rscb->clientSslCtx);
1794 2 : rscb->clientSslCtx = NULL;
1795 : }
1796 41 : }
1797 :
1798 40 : STATIC void RsDeinitFreeRscb(struct rs_cb *rscb)
1799 : {
1800 40 : RS_PTHREAD_MUTEX_LOCK(&rscb->mutex);
1801 40 : RsListFree(rscb);
1802 :
1803 40 : free(rscb->fdMap);
1804 40 : rscb->fdMap = NULL;
1805 40 : freeifaddrs(rscb->ifaddrList);
1806 40 : rscb->ifaddrList = NULL;
1807 40 : RS_PTHREAD_MUTEX_ULOCK(&rscb->mutex);
1808 40 : RsFreeRdevList(rscb);
1809 40 : RsFreeUdevList(rscb);
1810 40 : RsSslFree(rscb);
1811 40 : RsFreeHeterogTcpFdList(rscb);
1812 : #ifdef CONFIG_TLV
1813 : if (RsIsTlvSupported()) {
1814 : if (rscb->tlvCb.initFlag) {
1815 : RsTlvDeinit(rscb->tlvCb.phyId);
1816 : }
1817 : }
1818 : #endif
1819 40 : pthread_mutex_destroy(&rscb->mutex);
1820 40 : pthread_mutex_destroy(&rscb->connCb.connMutex);
1821 40 : RsDestroyEpoll(rscb);
1822 :
1823 : #ifdef CUSTOM_INTERFACE
1824 40 : if (RsIsUdmaSupported() || RsIsRdmaSupported()) {
1825 0 : RsDeInitNetAdapt(rscb);
1826 0 : RsEschedDeinit(rscb->protocol);
1827 0 : (void)RsCtxApiDeinit(rscb->hccpMode, rscb->protocol);
1828 : }
1829 : #endif
1830 :
1831 40 : free(rscb);
1832 40 : rscb = NULL;
1833 40 : gRsCb = NULL;
1834 40 : }
1835 :
1836 44 : RS_ATTRI_VISI_DEF int RsDeinit(struct RsInitConfig *cfg)
1837 : {
1838 44 : struct rs_cb *rscb = gRsCb;
1839 : unsigned int chipId;
1840 : eventfd_t event;
1841 : int ret;
1842 :
1843 44 : CHK_PRT_RETURN(cfg == NULL, hccp_err("param error, cfg is NULL"), -EINVAL);
1844 :
1845 44 : chipId = cfg->chipId;
1846 44 : if (__sync_fetch_and_sub(&(gInitCounter[chipId]), 1) > 1) {
1847 0 : return 0;
1848 : }
1849 44 : if (rscb && (chipId == rscb->chipId)) {
1850 40 : event = 1;
1851 : /* send event to eventfd to waking up epoll handle thread */
1852 40 : ret = (int)write(rscb->connCb.eventfd, &event, sizeof(eventfd_t));
1853 40 : CHK_PRT_RETURN(ret != sizeof(eventfd_t),
1854 : hccp_err("eventfd_write failed(0x%x), chipId:%u, errno:%d", ret, chipId, errno), -EFILEOPER);
1855 :
1856 40 : hccp_info("epoll wait up ok, rscb->connFlag:%d", rscb->connFlag);
1857 : // already been RS_CONN_EXIT_FLAG, no need to change conn_flag
1858 40 : if (rscb->connFlag != RS_CONN_EXIT_FLAG) {
1859 40 : rscb->connFlag = 0;
1860 : }
1861 40 : int tryAgain = RS_TRY_TIME;
1862 80 : while (((rscb->state & RS_STATE_HALT) == 0) && tryAgain > 0) {
1863 40 : usleep(RS_USLEEP_TIME);
1864 40 : tryAgain--;
1865 : };
1866 :
1867 40 : if (tryAgain == 0) {
1868 0 : hccp_warn("try_again exhausted, rscb state:%u", rscb->state);
1869 : }
1870 :
1871 40 : tryAgain = RS_TRY_TIME;
1872 6366 : while ((rscb->connFlag != RS_CONN_EXIT_FLAG) && tryAgain > 0) {
1873 6326 : usleep(RS_USLEEP_TIME);
1874 6326 : tryAgain--;
1875 : }
1876 :
1877 40 : CHK_PRT_RETURN(tryAgain == 0, hccp_warn("connect thread quit unsuccessful"), -EAGAIN);
1878 40 : rscb->state &= ~RS_STATE_HALT;
1879 40 : RsDeinitFreeRscb(rscb);
1880 40 : gRsCbList[chipId] = NULL;
1881 40 : DlHalDeinit();
1882 :
1883 40 : hccp_run_info("rs_deinit chipId[%u] ok", chipId);
1884 :
1885 40 : return 0;
1886 : }
1887 :
1888 4 : DlHalDeinit();
1889 4 : return -ENODEV;
1890 : }
1891 :
1892 0 : RS_ATTRI_VISI_DEF int RsGetVnicIp(unsigned int phyId, unsigned int *vnicIp)
1893 : {
1894 0 : int64_t deviceInfo = 0;
1895 : int ret;
1896 :
1897 0 : CHK_PRT_RETURN(phyId >= RS_MAX_DEV_NUM, hccp_err("phyId:%u >= [%d], is invalid", phyId, RS_MAX_DEV_NUM), -EINVAL);
1898 0 : CHK_PRT_RETURN(vnicIp == NULL, hccp_err("vnic_ip is null!"), -EINVAL);
1899 :
1900 0 : ret = DlHalGetDeviceInfo(phyId, MODULE_TYPE_SYSTEM, INFO_TYPE_VNIC_IP, &deviceInfo);
1901 0 : CHK_PRT_RETURN(ret != 0, hccp_err("phyId:%u dl_hal_get_device_info failed! ret:%d", phyId, ret), ret);
1902 :
1903 0 : *vnicIp = (unsigned int)deviceInfo;
1904 0 : return 0;
1905 : }
1906 :
1907 4 : STATIC int RsGetVnicIpInfo(unsigned int phyId, unsigned int id, enum IdType type, struct IpInfo *info)
1908 : {
1909 4 : int64_t deviceInfo = 0;
1910 : unsigned int vnicIp;
1911 : int ret;
1912 :
1913 : // get vnic ip by id with different type
1914 4 : if (type == PHY_ID_VNIC_IP) {
1915 1 : ret = DlHalGetDeviceInfo(id, MODULE_TYPE_SYSTEM, INFO_TYPE_VNIC_IP, &deviceInfo);
1916 1 : CHK_PRT_RETURN(ret != 0,
1917 : hccp_err("cur_phy_id:%u dl_hal_get_device_info failed! phyId:%u ret:%d", phyId, id, ret), ret);
1918 3 : } else if (type == SDID_VNIC_IP) {
1919 1 : ret = DlHalGetDeviceInfo(id, MODULE_TYPE_SYSTEM, INFO_TYPE_SPOD_VNIC_IP, &deviceInfo);
1920 1 : CHK_PRT_RETURN(ret != 0, hccp_err("phyId:%u dl_hal_get_device_info failed! sdid:0x%x ret:%d", phyId, id, ret),
1921 : ret);
1922 : } else {
1923 2 : hccp_err("phyId:%u get vnic ip failed! id:0x%x, invalid type:%u", phyId, id, type);
1924 2 : return -EINVAL;
1925 : }
1926 :
1927 : // prepare ip info, only support IPv4
1928 2 : vnicIp = (unsigned int)deviceInfo;
1929 2 : info->family = AF_INET;
1930 2 : info->ip.addr.s_addr = vnicIp;
1931 :
1932 2 : hccp_dbg("phyId:%u query id:%u type:%u got vnic_ip:%u", phyId, id, type, vnicIp);
1933 :
1934 2 : return 0;
1935 : }
1936 :
1937 3 : RS_ATTRI_VISI_DEF int RsGetVnicIpInfos(unsigned int phyId, enum IdType type, unsigned int ids[], unsigned int num,
1938 : struct IpInfo infos[])
1939 : {
1940 : unsigned int i;
1941 : int ret;
1942 :
1943 3 : CHK_PRT_RETURN(ids == NULL, hccp_err("phyId:%u, ids is null!", phyId), -EINVAL);
1944 2 : CHK_PRT_RETURN(infos == NULL, hccp_err("phyId:%u, infos is null!", phyId), -EINVAL);
1945 :
1946 1 : for (i = 0; i < num; i++) {
1947 1 : ret = RsGetVnicIpInfo(phyId, ids[i], type, &infos[i]);
1948 1 : if (ret != 0) {
1949 1 : hccp_err("phyId:%u get vnic ip info failed! ids[%u]:0x%x type:%u", phyId, i, ids[i], type);
1950 1 : return ret;
1951 : }
1952 : }
1953 :
1954 0 : return 0;
1955 : }
1956 :
1957 464 : int rsGetLocalDevIDByHostDevID(unsigned int phyId, unsigned int *chipId)
1958 : {
1959 464 : CHK_PRT_RETURN(gRsCb == NULL, hccp_warn("No device initialized !"), -ENODEV);
1960 :
1961 452 : if (gRsCb->hccpMode == NETWORK_PEER_ONLINE) {
1962 49 : *chipId = phyId;
1963 49 : return 0;
1964 : } else {
1965 : // to be compatible with mdev scenario: ignore phyId to get current devId(chipId)
1966 403 : return DlDrvQueryProcessHostPid(getpid(), chipId, NULL, NULL, NULL);
1967 : }
1968 : }
1969 :
1970 15 : int rsGetDevIDByLocalDevID(unsigned int chipId, unsigned int *phyId)
1971 : {
1972 15 : CHK_PRT_RETURN(gRsCb == NULL, hccp_warn("No device initialized !"), -ENODEV);
1973 :
1974 15 : if (gRsCb->hccpMode == NETWORK_PEER_ONLINE) {
1975 0 : *phyId = chipId;
1976 0 : return 0;
1977 : } else {
1978 15 : return DlDrvGetDevIdByLocalDevId(chipId, phyId);
1979 : }
1980 : }
1981 :
1982 2 : RS_ATTRI_VISI_DEF int RsSetQpAttrQos(unsigned int phyId, unsigned int rdevIndex, unsigned int qpn, struct QosAttr *attr)
1983 : {
1984 2 : struct RsQpCb *qpCb = NULL;
1985 2 : int ret = 0;
1986 :
1987 2 : RS_QP_PARA_CHECK(phyId);
1988 2 : ret = RsQpn2qpcb(phyId, rdevIndex, qpn, &qpCb);
1989 2 : CHK_PRT_RETURN(ret != 0 || qpCb == NULL, hccp_err("get qp cb failed qpn %u, ret %d", qpn, ret), ret);
1990 :
1991 2 : qpCb->qosAttr.tc = attr->tc;
1992 2 : qpCb->qosAttr.sl = attr->sl;
1993 :
1994 2 : hccp_info("set qp qos attr: qpn[%u] tc[%u] sl[%u]", qpn, attr->tc, attr->sl);
1995 2 : return 0;
1996 : }
1997 :
1998 2 : RS_ATTRI_VISI_DEF int RsSetQpAttrTimeout(unsigned int phyId, unsigned int rdevIndex, unsigned int qpn,
1999 : unsigned int *timeout)
2000 : {
2001 2 : struct RsQpCb *qpCb = NULL;
2002 2 : int ret = 0;
2003 :
2004 2 : RS_QP_PARA_CHECK(phyId);
2005 2 : ret = RsQpn2qpcb(phyId, rdevIndex, qpn, &qpCb);
2006 2 : CHK_PRT_RETURN(ret != 0 || qpCb == NULL, hccp_err("get qp cb failed qpn %u, ret %d", qpn, ret), ret);
2007 :
2008 2 : qpCb->timeout = *timeout;
2009 :
2010 2 : hccp_info("set qp qos attr: qpn[%u] timeout[%u]", qpn, *timeout);
2011 2 : return 0;
2012 : }
2013 :
2014 2 : RS_ATTRI_VISI_DEF int RsSetQpAttrRetryCnt(unsigned int phyId, unsigned int rdevIndex, unsigned int qpn,
2015 : unsigned int *retryCnt)
2016 : {
2017 2 : struct RsQpCb *qpCb = NULL;
2018 2 : int ret = 0;
2019 :
2020 2 : RS_QP_PARA_CHECK(phyId);
2021 2 : ret = RsQpn2qpcb(phyId, rdevIndex, qpn, &qpCb);
2022 2 : CHK_PRT_RETURN(ret != 0 || qpCb == NULL, hccp_err("get qp cb failed qpn %u, ret %d", qpn, ret), ret);
2023 :
2024 2 : qpCb->retryCnt = *retryCnt;
2025 :
2026 2 : hccp_info("set qp qos attr: qpn[%u] retry_cnt[%u]", qpn, *retryCnt);
2027 2 : return 0;
2028 : }
2029 :
2030 1 : RS_ATTRI_VISI_DEF int RsGetCqeErrInfo(struct CqeErrInfo *info)
2031 : {
2032 : int ret;
2033 :
2034 1 : ret = RsDrvGetCqeErrInfo(info);
2035 1 : CHK_PRT_RETURN(ret, hccp_err("get failed! ret:%d", ret), ret);
2036 1 : return 0;
2037 : }
2038 :
2039 2 : RS_ATTRI_VISI_DEF int RsGetCqeErrInfoNum(unsigned int phyId, unsigned int rdevIdx, unsigned int *num)
2040 : {
2041 2 : struct RsRdevCb *rdevCb = NULL;
2042 : unsigned int chipId;
2043 : int ret;
2044 :
2045 2 : CHK_PRT_RETURN(phyId >= RS_MAX_DEV_NUM, hccp_err("rs get cqe err param error, phyId[%u]", phyId), -EINVAL);
2046 1 : ret = rsGetLocalDevIDByHostDevID(phyId, &chipId);
2047 1 : CHK_PRT_RETURN(ret, hccp_err("phyId[%u] invalid, ret %d", phyId, ret), ret);
2048 :
2049 1 : ret = RsRdev2rdevCb(chipId, rdevIdx, &rdevCb);
2050 1 : CHK_PRT_RETURN(ret != 0 || rdevCb == NULL, hccp_err("rs_rdev2rdev_cb for chipId[%u] failed, ret %d", chipId, ret),
2051 : ret);
2052 :
2053 0 : *num = rdevCb->cqeErrCnt;
2054 :
2055 0 : return 0;
2056 : }
2057 :
2058 3 : RS_ATTRI_VISI_DEF int RsGetCqeErrInfoList(unsigned int phyId, unsigned int rdevIdx, struct CqeErrInfo *info,
2059 : unsigned int *num)
2060 : {
2061 3 : struct RsQpCb *qpCbCurr = NULL;
2062 3 : struct RsQpCb *qpCbNext = NULL;
2063 3 : struct RsRdevCb *rdevCb = NULL;
2064 3 : unsigned int cqeErrIdx = 0;
2065 3 : unsigned int numTmp = *num;
2066 : unsigned int chipId;
2067 : int ret;
2068 :
2069 3 : CHK_PRT_RETURN(phyId >= RS_MAX_DEV_NUM, hccp_err("rs get cqe err param error, phyId[%u]", phyId), -EINVAL);
2070 2 : ret = rsGetLocalDevIDByHostDevID(phyId, &chipId);
2071 2 : CHK_PRT_RETURN(ret, hccp_err("phyId[%u] invalid, ret %d", phyId, ret), ret);
2072 :
2073 2 : ret = RsRdev2rdevCb(chipId, rdevIdx, &rdevCb);
2074 2 : CHK_PRT_RETURN(ret != 0 || rdevCb == NULL, hccp_err("rs_rdev2rdev_cb for chipId[%u] failed, ret %d", chipId, ret),
2075 : ret);
2076 :
2077 2 : if (RsListEmpty(&rdevCb->qpList)) {
2078 1 : *num = 0;
2079 1 : return 0;
2080 : }
2081 :
2082 1 : RS_LIST_GET_HEAD_ENTRY(qpCbCurr, qpCbNext, &rdevCb->qpList, list, struct RsQpCb);
2083 2 : for (; (&qpCbCurr->list) != &rdevCb->qpList;
2084 1 : qpCbCurr = qpCbNext, qpCbNext = list_entry(qpCbNext->list.next, struct RsQpCb, list)) {
2085 1 : if (qpCbCurr->cqeErrInfo.info.status != 0) {
2086 1 : RS_PTHREAD_MUTEX_LOCK(&qpCbCurr->cqeErrInfo.mutex);
2087 1 : info[cqeErrIdx].status = qpCbCurr->cqeErrInfo.info.status;
2088 1 : info[cqeErrIdx].qpn = qpCbCurr->cqeErrInfo.info.qpn;
2089 1 : info[cqeErrIdx].time = qpCbCurr->cqeErrInfo.info.time;
2090 1 : qpCbCurr->cqeErrInfo.info.status = 0;
2091 1 : RS_PTHREAD_MUTEX_ULOCK(&qpCbCurr->cqeErrInfo.mutex);
2092 1 : RS_PTHREAD_MUTEX_LOCK(&qpCbCurr->rdevCb->cqeErrCntMutex);
2093 1 : qpCbCurr->rdevCb->cqeErrCnt--;
2094 1 : RS_PTHREAD_MUTEX_ULOCK(&qpCbCurr->rdevCb->cqeErrCntMutex);
2095 1 : cqeErrIdx++;
2096 1 : if (cqeErrIdx == numTmp) {
2097 0 : break;
2098 : }
2099 : }
2100 : }
2101 :
2102 1 : *num = cqeErrIdx;
2103 :
2104 1 : return 0;
2105 : }
2106 :
2107 3 : int RsQueryMrCb(struct RsRdevCb *devCb, uint64_t addr, struct RsMrCb **mrCb, struct RsListHead *mrList)
2108 : {
2109 3 : struct RsMrCb *mrCurr = NULL;
2110 3 : struct RsMrCb *mrNext = NULL;
2111 :
2112 3 : RS_PTHREAD_MUTEX_LOCK(&devCb->rdevMutex);
2113 3 : RS_LIST_GET_HEAD_ENTRY(mrCurr, mrNext, mrList, list, struct RsMrCb);
2114 3 : for (; (&mrCurr->list) != mrList; mrCurr = mrNext, mrNext = list_entry(mrNext->list.next, struct RsMrCb, list)) {
2115 2 : if ((mrCurr->mrInfo.addr <= addr) && (addr < mrCurr->mrInfo.addr + mrCurr->mrInfo.len)) {
2116 2 : *mrCb = mrCurr;
2117 2 : RS_PTHREAD_MUTEX_ULOCK(&devCb->rdevMutex);
2118 2 : return 0;
2119 : }
2120 : }
2121 :
2122 1 : *mrCb = NULL;
2123 1 : RS_PTHREAD_MUTEX_ULOCK(&devCb->rdevMutex);
2124 :
2125 1 : hccp_info("cannot find mrcb for addr@0x%lx !", addr);
2126 :
2127 1 : return -ENODEV;
2128 : }
2129 :
2130 4 : STATIC int RsGetLinuxVersion(struct RsLinuxVersionInfo *verInfo)
2131 : {
2132 : #define LINUX_VERSION_MAX_CHAR 1024
2133 : #define LINUX_VERSION_TYPE_NUM 3
2134 : #define LINUX_VERSION_STR "Linux version "
2135 4 : char buffer[LINUX_VERSION_MAX_CHAR] = {0};
2136 : char *versionStr;
2137 4 : int retClose = 0;
2138 4 : int ret = 0;
2139 : int fd;
2140 :
2141 4 : fd = open("/proc/version", O_RDONLY);
2142 4 : CHK_PRT_RETURN(fd < 0, hccp_run_warn("open proc/version unsuccessful, errno[%d] fd[%d]", errno, fd), -EFILEOPER);
2143 :
2144 : do {
2145 3 : ret = (int)read(fd, buffer, sizeof(buffer) - 1);
2146 3 : } while ((ret < 0) && (errno == EINTR));
2147 :
2148 3 : if (ret < 0) {
2149 1 : hccp_run_warn("read fd unsuccessful[%d]", ret);
2150 1 : RS_CLOSE_RETRY_FOR_EINTR(retClose, fd);
2151 1 : return -EFILEOPER;
2152 : }
2153 :
2154 2 : versionStr = strstr(buffer, LINUX_VERSION_STR);
2155 2 : if (versionStr == NULL) {
2156 1 : hccp_run_warn("can't get Linux version");
2157 1 : RS_CLOSE_RETRY_FOR_EINTR(retClose, fd);
2158 1 : return -EFILEOPER;
2159 : }
2160 1 : versionStr += strlen(LINUX_VERSION_STR);
2161 1 : if (sscanf_s(versionStr, "%d.%d.%d", &verInfo->major, &verInfo->minor, &verInfo->patch) != LINUX_VERSION_TYPE_NUM) {
2162 1 : hccp_run_warn("can't extract Linux version");
2163 1 : RS_CLOSE_RETRY_FOR_EINTR(retClose, fd);
2164 1 : return -EFILEOPER;
2165 : }
2166 :
2167 0 : RS_CLOSE_RETRY_FOR_EINTR(retClose, fd);
2168 0 : return retClose;
2169 : }
2170 :
2171 5 : RS_ATTRI_VISI_DEF int RsGetSecRandom(unsigned int *value)
2172 : {
2173 : #define SEC_LINUX_VERSION_MAJOR 5
2174 : #define SEC_LINUX_VERSION_MINOR 18
2175 : #define SEC_LINUX_VERSION_PATCH 0
2176 5 : struct RsLinuxVersionInfo verInfo = {0};
2177 : int ret;
2178 :
2179 5 : ret = RsGetLinuxVersion(&verInfo);
2180 5 : CHK_PRT_RETURN(ret, hccp_run_warn("[rs_get_random]get_linux_version unsuccessful ret(%d)", ret), ret);
2181 :
2182 : // linux_version > 5.18, urandom is secure
2183 1 : if (verInfo.major > SEC_LINUX_VERSION_MAJOR ||
2184 1 : (verInfo.major == SEC_LINUX_VERSION_MAJOR && verInfo.minor > SEC_LINUX_VERSION_MINOR) ||
2185 0 : (verInfo.major == SEC_LINUX_VERSION_MAJOR && verInfo.minor == SEC_LINUX_VERSION_MINOR &&
2186 0 : verInfo.patch > SEC_LINUX_VERSION_PATCH)) {
2187 1 : ret = RsDrvGetRandomNum((int *)value);
2188 : } else {
2189 0 : hccp_run_warn("[rs_get_random]linux_version is not secure version");
2190 0 : return -ENOTSUPP;
2191 : }
2192 :
2193 1 : if (ret != 0) {
2194 1 : hccp_run_warn("[get][get_random]rs_get_sec_random unsuccessful, ret(%d)", ret);
2195 : }
2196 1 : return ret;
2197 : }
2198 :
2199 452 : RS_ATTRI_VISI_DEF enum ProductType RsGetProductType(int devId)
2200 : {
2201 : static enum ProductType type = PRODUCT_TYPE_NO_VALUE;
2202 : static halChipInfo chipInfo = {0};
2203 : int ret;
2204 :
2205 452 : if (type != PRODUCT_TYPE_NO_VALUE) { // Cache result after first query
2206 450 : hccp_info("[Get][ChipInfo]chip name is %s, type:%d", chipInfo.name, type);
2207 450 : return type;
2208 : }
2209 :
2210 2 : DlHalInit();
2211 2 : ret = DlHalGetChipInfo(devId, &chipInfo);
2212 2 : DlHalDeinit();
2213 :
2214 2 : CHK_PRT_RETURN(ret != 0, hccp_err("[Get][ChipInfo]DlHalGetChipInfo failed ret:%d", ret), PRODUCT_TYPE_INVALID);
2215 :
2216 2 : if (fnmatch("910_93[a-zA-Z1-9_]*", (const char *)chipInfo.name, 0) == 0) {
2217 0 : type = PRODUCT_TYPE_910_93;
2218 2 : } else if (fnmatch("910B[a-zA-Z1-9_]*", (const char *)chipInfo.name, 0) == 0) {
2219 0 : type = PRODUCT_TYPE_910B;
2220 2 : } else if (fnmatch("910_96[a-zA-Z1-9_]*", (const char *)chipInfo.name, 0) == 0) {
2221 0 : type = PRODUCT_TYPE_910_96;
2222 2 : } else if (fnmatch("910[a-zA-Z1-9]*", (const char *)chipInfo.name, 0) == 0) {
2223 0 : type = PRODUCT_TYPE_910;
2224 2 : } else if (fnmatch("310p[a-zA-Z1-9]*", (const char *)chipInfo.name, 0) == 0) {
2225 0 : type = PRODUCT_TYPE_310p;
2226 2 : } else if (fnmatch("950[a-zA-Z1-9]*", (const char *)chipInfo.name, 0) == 0) {
2227 0 : type = PRODUCT_TYPE_950;
2228 2 : } else if (fnmatch("350[a-zA-Z1-9]*", (const char *)chipInfo.name, 0) == 0) {
2229 0 : type = PRODUCT_TYPE_350;
2230 : } else {
2231 2 : type = PRODUCT_TYPE_OTHERS;
2232 : }
2233 :
2234 2 : hccp_run_info("[Get][ChipInfo]chip name is %s, type:%d", chipInfo.name, type);
2235 2 : return type;
2236 : }
|
