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 366 : RS_ATTRI_VISI_DEF void RsGetCurTime(struct timeval *time)
73 : {
74 : int ret;
75 :
76 366 : RS_CHECK_POINTER_NULL_RETURN_VOID(time);
77 366 : ret = gettimeofday(time, NULL);
78 364 : 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 364 : return;
87 : }
88 :
89 165 : RS_ATTRI_VISI_DEF void HccpTimeInterval(struct timeval *endTime, struct timeval *startTime, float *msec)
90 : {
91 165 : RS_CHECK_POINTER_NULL_RETURN_VOID(endTime);
92 165 : RS_CHECK_POINTER_NULL_RETURN_VOID(startTime);
93 165 : RS_CHECK_POINTER_NULL_RETURN_VOID(msec);
94 :
95 : /* if low position is sufficient, then borrow one from the high position */
96 165 : 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 165 : *msec = (float)((endTime->tv_sec - startTime->tv_sec) * MS_PER_SECOND_F +
102 165 : (endTime->tv_usec - startTime->tv_usec) / US_PER_MS_F);
103 :
104 165 : 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 29 : hccp_run_info("rdev init success, phyId:%u, localIp:%s, rdevIndex:%u", phyId, rdevCb->localIp.readAddr, *rdevIndex);
1064 29 : return 0;
1065 :
1066 2 : free_dev_list:
1067 2 : RsIbvFreeDeviceList(rdevCb->devList);
1068 2 : free_rs_cb:
1069 2 : free(rdevCb);
1070 2 : rdevCb = NULL;
1071 10 : get_rs_cb_fail:
1072 10 : RsApiDeinit();
1073 10 : return ret;
1074 : }
1075 :
1076 1 : RS_ATTRI_VISI_DEF int RsRdevInitWithBackup(struct rdev rdevInfo, struct rdev backupRdevInfo, unsigned int notifyType,
1077 : unsigned int *rdevIndex)
1078 : {
1079 1 : struct RsBackupInfo backupInfo = {0};
1080 :
1081 1 : backupInfo.backupFlag = true;
1082 1 : (void)memcpy_s(&backupInfo.rdevInfo, sizeof(struct rdev), &backupRdevInfo, sizeof(struct rdev));
1083 :
1084 1 : return RsRdevInitWithBackupInfo(rdevInfo, backupInfo, notifyType, rdevIndex);
1085 : }
1086 :
1087 38 : RS_ATTRI_VISI_DEF int RsRdevInit(struct rdev rdevInfo, unsigned int notifyType, unsigned int *rdevIndex)
1088 : {
1089 38 : struct RsBackupInfo backupInfo = {0};
1090 :
1091 38 : return RsRdevInitWithBackupInfo(rdevInfo, backupInfo, notifyType, rdevIndex);
1092 : }
1093 :
1094 29 : STATIC void RsDestroyQpList(unsigned int phyId, unsigned int rdevIndex, struct RsRdevCb *rdevCb, struct RsQpCb *qpCb,
1095 : struct RsQpCb *qpCb2)
1096 : {
1097 : int ret;
1098 :
1099 29 : if (!RsListEmpty(&rdevCb->qpList)) {
1100 0 : hccp_warn("qp list do not empty!");
1101 0 : RS_LIST_GET_HEAD_ENTRY(qpCb, qpCb2, &rdevCb->qpList, list, struct RsQpCb);
1102 0 : for (; (&qpCb->list) != &rdevCb->qpList;
1103 0 : qpCb = qpCb2, qpCb2 = list_entry(qpCb2->list.next, struct RsQpCb, list)) {
1104 0 : hccp_info("qpn[%u] will be destroyed", qpCb->ibQp->qp_num);
1105 0 : if (qpCb->ibQpEx != NULL) {
1106 0 : ret = RsNdaQpDestroy(phyId, rdevIndex, qpCb->ibQp->qp_num);
1107 : } else {
1108 0 : ret = RsQpDestroy(phyId, rdevIndex, qpCb->ibQp->qp_num);
1109 : }
1110 0 : if (ret != 0) {
1111 0 : hccp_err("rs_qp_destroy failed, ret:%d", ret);
1112 : }
1113 : }
1114 : }
1115 :
1116 29 : return;
1117 : }
1118 :
1119 29 : STATIC void RsFreeTypicalMrCb(struct RsRdevCb *devCb)
1120 : {
1121 29 : struct RsListHead *typicalMrList = &devCb->typicalMrList;
1122 29 : struct RsMrCb *mrCurr = NULL;
1123 29 : struct RsMrCb *mrNext = NULL;
1124 :
1125 29 : RS_PTHREAD_MUTEX_LOCK(&devCb->rdevMutex);
1126 29 : RS_LIST_GET_HEAD_ENTRY(mrCurr, mrNext, typicalMrList, list, struct RsMrCb);
1127 29 : for (; (&mrCurr->list) != typicalMrList;
1128 0 : mrCurr = mrNext, mrNext = list_entry(mrNext->list.next, struct RsMrCb, list)) {
1129 0 : (void)RsDrvMrDereg(mrCurr->ibMr);
1130 0 : RsListDel(&mrCurr->list);
1131 0 : free(mrCurr);
1132 0 : mrCurr = NULL;
1133 : }
1134 29 : RS_PTHREAD_MUTEX_ULOCK(&devCb->rdevMutex);
1135 :
1136 29 : hccp_info("rs_free_typical_mr_cb is succ");
1137 29 : }
1138 :
1139 31 : RS_ATTRI_VISI_DEF int RsRdevDeinit(unsigned int phyId, unsigned int notifyType, unsigned int rdevIndex)
1140 : {
1141 : (void)notifyType;
1142 31 : struct RsRdevCb *rdevCb = NULL;
1143 31 : struct RsQpCb *qpCb2 = NULL;
1144 31 : struct RsQpCb *qpCb = NULL;
1145 : unsigned int chipId;
1146 : int ret;
1147 :
1148 31 : hccp_info("rdev deinit start, phyId:%u, rdevIndex:%u", phyId, rdevIndex);
1149 31 : CHK_PRT_RETURN(phyId >= RS_MAX_DEV_NUM, hccp_err("rs set param error ! phyId:%u", phyId), -EINVAL);
1150 31 : ret = rsGetLocalDevIDByHostDevID(phyId, &chipId);
1151 31 : CHK_PRT_RETURN(ret, hccp_err("phyId[%u] invalid, ret %d", phyId, ret), ret);
1152 :
1153 29 : ret = RsRdev2rdevCb(chipId, rdevIndex, &rdevCb);
1154 29 : CHK_PRT_RETURN(ret || rdevCb == NULL, hccp_err("rs_rdev2rdev_cb for chipId[%u] failed, ret %d", chipId, ret), ret);
1155 :
1156 29 : if (rdevCb->notifyType != NO_USE && rdevCb->notifyMr != NULL) {
1157 27 : ret = RsDrvMrDereg(rdevCb->notifyMr);
1158 27 : if (ret) {
1159 0 : hccp_err("rs_drv_mr_dereg failed, ret %d", ret);
1160 : }
1161 : }
1162 :
1163 29 : hccp_info("poll_cqe_num[%d]", rdevCb->pollCqeNum);
1164 :
1165 29 : RsDestroyQpList(phyId, rdevIndex, rdevCb, qpCb, qpCb2);
1166 :
1167 29 : RsFreeTypicalMrCb(rdevCb);
1168 :
1169 : #ifdef CUSTOM_INTERFACE
1170 29 : if (RsIsCustomInterfaceSupported()) {
1171 29 : (void)RsRoceUnmmapAiDbReg(rdevCb->ibCtx);
1172 : }
1173 : #endif
1174 :
1175 29 : RsIbvDeallocPd(rdevCb->ibPd);
1176 :
1177 29 : RsDeinitNdaCb(rdevCb);
1178 :
1179 29 : RsIbvCloseDevice(rdevCb->ibCtx);
1180 :
1181 : #ifdef CUSTOM_INTERFACE
1182 29 : if (RsIsCustomInterfaceSupported()) {
1183 29 : RsCloseBackupIbCtx(rdevCb);
1184 : }
1185 : #endif
1186 :
1187 29 : pthread_mutex_destroy(&rdevCb->cqeErrCntMutex);
1188 :
1189 29 : pthread_mutex_destroy(&rdevCb->rdevMutex);
1190 :
1191 29 : RsIbvFreeDeviceList(rdevCb->devList);
1192 :
1193 29 : RS_PTHREAD_MUTEX_LOCK(&gRsCb->mutex);
1194 29 : RsListDel(&rdevCb->list);
1195 29 : RS_PTHREAD_MUTEX_ULOCK(&gRsCb->mutex);
1196 29 : RsSensorNodeUnregister(rdevCb->rsCb);
1197 29 : RsApiDeinit();
1198 29 : hccp_run_info("rdev deinit success, phyId:%u, rdevIndex:%u", phyId, rdevIndex);
1199 29 : free(rdevCb);
1200 29 : rdevCb = NULL;
1201 29 : return 0;
1202 : }
1203 :
1204 0 : STATIC void RsHeterogTcpFreeFdNode(struct RsHeterogTcpFdInfo *fdNode)
1205 : {
1206 : int fd;
1207 :
1208 0 : RS_PTHREAD_MUTEX_LOCK(&gRsCb->mutex);
1209 0 : fd = fdNode->fd;
1210 0 : RsListDel(&fdNode->list);
1211 0 : free(fdNode);
1212 0 : fdNode = NULL;
1213 0 : gRsCb->fdMap[fd] = NULL;
1214 0 : RS_PTHREAD_MUTEX_ULOCK(&gRsCb->mutex);
1215 0 : }
1216 :
1217 1 : RS_ATTRI_VISI_DEF int RsEpollCtlAdd(const void *fdHandle, enum RaEpollEvent event)
1218 : {
1219 1 : struct RsHeterogTcpFdInfo *fdNode = NULL;
1220 1 : unsigned int tmpEvent = event;
1221 1 : int fd = RS_FD_INVALID;
1222 : int ret;
1223 :
1224 1 : if (event == RA_EPOLLONESHOT) {
1225 0 : tmpEvent = EPOLLIN | EPOLLET | EPOLLONESHOT;
1226 1 : } else if (event == RA_EPOLLIN) {
1227 1 : tmpEvent = EPOLLIN;
1228 : } else {
1229 0 : hccp_err("unknown event[%u]", tmpEvent);
1230 0 : return -EINVAL;
1231 : }
1232 :
1233 1 : if (gRsCb == NULL) {
1234 1 : gRsCb = RsGetCurRsCb();
1235 1 : if (gRsCb == NULL) {
1236 1 : hccp_err("[rs_epoll_ctl_add]rs_get_cur_rs_cb failed rs_cb(NULL)");
1237 1 : return -EINVAL;
1238 : }
1239 : }
1240 0 : tmpEvent = tmpEvent | EPOLLRDHUP;
1241 0 : fdNode = calloc(1, sizeof(struct RsHeterogTcpFdInfo));
1242 0 : CHK_PRT_RETURN(fdNode == NULL, hccp_err("no memory for fd_node"), -ENOMEM);
1243 :
1244 0 : fd = ((const struct SocketPeerInfo *)fdHandle)->fd;
1245 0 : fdNode->fd = fd;
1246 0 : RS_PTHREAD_MUTEX_LOCK(&gRsCb->mutex);
1247 0 : RsListAddTail(&fdNode->list, &gRsCb->heterogTcpFdList);
1248 0 : gRsCb->fdMap[fd] = fdHandle;
1249 0 : RS_PTHREAD_MUTEX_ULOCK(&gRsCb->mutex);
1250 0 : ret = RsEpollCtl(gRsCb->connCb.epollfd, EPOLL_CTL_ADD, fd, tmpEvent);
1251 0 : if (ret != 0) {
1252 0 : hccp_err("[rs_epoll_ctl_add]RsEpollCtl failed ret(%d), fd:%d, event:%u", ret, fd, event);
1253 0 : goto out;
1254 : }
1255 0 : return 0;
1256 0 : out:
1257 0 : RsHeterogTcpFreeFdNode(fdNode);
1258 0 : fdNode = NULL;
1259 0 : return ret;
1260 : }
1261 :
1262 5 : RS_ATTRI_VISI_DEF int RsEpollCtlMod(const void *fdHandle, enum RaEpollEvent event)
1263 : {
1264 5 : unsigned int tmpEvent = event;
1265 5 : int fd = RS_FD_INVALID;
1266 : int ret;
1267 :
1268 5 : if (event == RA_EPOLLONESHOT) {
1269 1 : tmpEvent = EPOLLIN | EPOLLET | EPOLLONESHOT;
1270 4 : } else if (event == RA_EPOLLIN) {
1271 3 : tmpEvent = EPOLLIN;
1272 : } else {
1273 1 : hccp_err("unknown event[%u]", event);
1274 1 : return -EINVAL;
1275 : }
1276 :
1277 4 : tmpEvent = tmpEvent | EPOLLRDHUP;
1278 4 : fd = ((const struct SocketPeerInfo *)fdHandle)->fd;
1279 :
1280 4 : if (gRsCb == NULL) {
1281 1 : gRsCb = RsGetCurRsCb();
1282 1 : if (gRsCb == NULL) {
1283 1 : hccp_err("[rs_epoll_ctl_mod]rs_get_cur_rs_cb failed rs_cb(NULL)");
1284 1 : return -EINVAL;
1285 : }
1286 : }
1287 :
1288 3 : ret = RsEpollCtl(gRsCb->connCb.epollfd, EPOLL_CTL_MOD, fd, tmpEvent);
1289 3 : CHK_PRT_RETURN(ret, hccp_err("[rs_epoll_ctl_mod]RsEpollCtl failed ret(%d), fd:%d, event:%u", ret, fd, event), ret);
1290 1 : return 0;
1291 : }
1292 :
1293 3 : RS_ATTRI_VISI_DEF int RsEpollCtlDel(int fd)
1294 : {
1295 : int ret;
1296 3 : struct RsHeterogTcpFdInfo *fdNode = NULL;
1297 3 : struct RsHeterogTcpFdInfo *fdNode1 = NULL;
1298 :
1299 3 : if (gRsCb == NULL) {
1300 1 : gRsCb = RsGetCurRsCb();
1301 1 : if (gRsCb == NULL) {
1302 1 : hccp_err("[rs_epoll_ctl_del]rs_get_cur_rs_cb failed rs_cb(NULL)");
1303 1 : return -EINVAL;
1304 : }
1305 : }
1306 2 : RS_LIST_GET_HEAD_ENTRY(fdNode, fdNode1, &gRsCb->heterogTcpFdList, list, struct RsHeterogTcpFdInfo);
1307 2 : for (; (&fdNode->list) != &gRsCb->heterogTcpFdList;
1308 0 : fdNode = fdNode1, fdNode1 = list_entry(fdNode1->list.next, struct RsHeterogTcpFdInfo, list)) {
1309 0 : if (fdNode->fd == fd) {
1310 : // 删除节点
1311 0 : RsHeterogTcpFreeFdNode(fdNode);
1312 0 : fdNode = NULL;
1313 0 : break; // lint !e108
1314 : }
1315 : }
1316 :
1317 : // 为了兼容epoll不同版本，这里加EPOLLIN参数
1318 2 : ret = RsEpollCtl(gRsCb->connCb.epollfd, EPOLL_CTL_DEL, fd, EPOLLIN);
1319 2 : CHK_PRT_RETURN(ret, hccp_err("[rs_epoll_ctl_del]RsEpollCtl failed ret(%d), fd:%d", ret, fd), ret);
1320 1 : return 0;
1321 : }
1322 :
1323 3 : RS_ATTRI_VISI_DEF void RsSetTcpRecvCallback(const void *callback)
1324 : {
1325 3 : if (gRsCb == NULL) {
1326 1 : hccp_err("param error, gRsCb is NULL");
1327 1 : return;
1328 : }
1329 2 : gRsCb->tcpRecvCallback = (void (*)(const void *))callback;
1330 : }
1331 :
1332 1 : STATIC void RsFreeAcceptOneNode(struct rs_cb *rscb, struct RsAcceptInfo *accept)
1333 : {
1334 : int ret;
1335 :
1336 1 : ret = RsEpollCtl(rscb->connCb.epollfd, EPOLL_CTL_DEL, accept->connFd, EPOLLIN);
1337 1 : if (ret) {
1338 1 : hccp_err("epoll ctl del fd %d failed, ret:%d", accept->connFd, ret);
1339 : }
1340 :
1341 1 : RS_PTHREAD_MUTEX_LOCK(&rscb->connCb.connMutex);
1342 1 : RsListDel(&accept->list);
1343 1 : RS_PTHREAD_MUTEX_ULOCK(&rscb->connCb.connMutex);
1344 :
1345 1 : if (rscb->sslEnable == RS_SSL_ENABLE) {
1346 1 : if (accept->ssl == NULL) {
1347 0 : hccp_warn("[Server] accept->ssl is NULL, it maybe has not establish tls link");
1348 : } else {
1349 1 : ssl_adp_shutdown(accept->ssl);
1350 1 : ssl_adp_free(accept->ssl);
1351 1 : accept->ssl = NULL;
1352 : }
1353 : }
1354 :
1355 1 : RS_CLOSE_RETRY_FOR_EINTR(ret, accept->connFd);
1356 :
1357 1 : hccp_info("free accept_server IP:%s, port:%d, connFd:%d", accept->serverIpAddr.readAddr, accept->sockPort,
1358 : accept->connFd);
1359 1 : accept->connFd = RS_FD_INVALID;
1360 :
1361 1 : free(accept);
1362 1 : accept = NULL;
1363 1 : }
1364 :
1365 40 : STATIC void RsFreeAccpetList(struct rs_cb *rscb)
1366 : {
1367 40 : struct RsAcceptInfo *accept = NULL;
1368 40 : struct RsAcceptInfo *accept2 = NULL;
1369 :
1370 40 : if (!RsListEmpty(&rscb->connCb.serverAcceptList)) {
1371 0 : hccp_warn("Server accept list do not empty!");
1372 0 : RS_LIST_GET_HEAD_ENTRY(accept, accept2, &rscb->connCb.serverAcceptList, list, struct RsAcceptInfo);
1373 0 : for (; (&accept->list) != &rscb->connCb.serverAcceptList;
1374 0 : accept = accept2, accept2 = list_entry(accept2->list.next, struct RsAcceptInfo, list)) {
1375 0 : RsFreeAcceptOneNode(rscb, accept);
1376 0 : accept = NULL;
1377 : }
1378 : }
1379 :
1380 40 : return;
1381 : }
1382 :
1383 12 : STATIC void RsFreeDesignatedAccpetNode(struct rs_cb *rscb, struct RsIpAddrInfo *localIp)
1384 : {
1385 12 : struct RsAcceptInfo *accept = NULL;
1386 12 : struct RsAcceptInfo *accept2 = NULL;
1387 :
1388 12 : if (!RsListEmpty(&rscb->connCb.serverAcceptList)) {
1389 0 : RS_LIST_GET_HEAD_ENTRY(accept, accept2, &rscb->connCb.serverAcceptList, list, struct RsAcceptInfo);
1390 0 : for (; (&accept->list) != &rscb->connCb.serverAcceptList;
1391 0 : accept = accept2, accept2 = list_entry(accept2->list.next, struct RsAcceptInfo, list)) {
1392 0 : if (!RsCompareIpAddr(&accept->serverIpAddr, localIp)) {
1393 0 : RsFreeAcceptOneNode(rscb, accept);
1394 0 : accept = NULL;
1395 : }
1396 : }
1397 : }
1398 :
1399 12 : return;
1400 : }
1401 :
1402 2 : STATIC void RsFreeConnOneNode(struct rs_cb *rscb, struct RsConnInfo *conn)
1403 : {
1404 : int ret;
1405 :
1406 2 : RS_PTHREAD_MUTEX_LOCK(&rscb->connCb.connMutex);
1407 2 : RsListDel(&conn->list);
1408 2 : RS_PTHREAD_MUTEX_ULOCK(&rscb->connCb.connMutex);
1409 :
1410 2 : if (rscb->sslEnable == RS_SSL_ENABLE) {
1411 0 : if (conn->ssl == NULL) {
1412 0 : hccp_warn("[Client] conn->ssl is NULL, it maybe has not establish tls link");
1413 : } else {
1414 0 : ssl_adp_shutdown(conn->ssl);
1415 0 : ssl_adp_free(conn->ssl);
1416 0 : conn->ssl = NULL;
1417 : }
1418 : }
1419 :
1420 2 : RS_CLOSE_RETRY_FOR_EINTR(ret, conn->connfd);
1421 :
1422 2 : hccp_info("free for conn IP:%s, port:%d, connfd:%d, state:%u", conn->clientIp.readAddr, conn->port, conn->connfd,
1423 : conn->state);
1424 :
1425 2 : conn->connfd = RS_FD_INVALID;
1426 2 : conn->state = RS_CONN_STATE_RESET;
1427 :
1428 2 : free(conn);
1429 2 : conn = NULL;
1430 2 : }
1431 :
1432 40 : STATIC void RsFreeClientConnList(struct rs_cb *rscb)
1433 : {
1434 40 : struct RsConnInfo *conn = NULL;
1435 40 : struct RsConnInfo *conn2 = NULL;
1436 :
1437 40 : if (!RsListEmpty(&rscb->connCb.clientConnList)) {
1438 0 : hccp_warn("Client conn node do not empty!");
1439 0 : RS_LIST_GET_HEAD_ENTRY(conn, conn2, &rscb->connCb.clientConnList, list, struct RsConnInfo);
1440 0 : for (; (&conn->list) != &rscb->connCb.clientConnList;
1441 0 : conn = conn2, conn2 = list_entry(conn2->list.next, struct RsConnInfo, list)) {
1442 0 : RsFreeConnOneNode(rscb, conn);
1443 0 : conn = NULL;
1444 : }
1445 : }
1446 :
1447 40 : return;
1448 : }
1449 :
1450 12 : STATIC void RsFreeDesignatedClientConnNode(struct rs_cb *rscb, struct RsIpAddrInfo *localIp)
1451 : {
1452 12 : struct RsConnInfo *conn = NULL;
1453 12 : struct RsConnInfo *conn2 = NULL;
1454 :
1455 12 : if (!RsListEmpty(&rscb->connCb.clientConnList)) {
1456 0 : RS_LIST_GET_HEAD_ENTRY(conn, conn2, &rscb->connCb.clientConnList, list, struct RsConnInfo);
1457 0 : for (; (&conn->list) != &rscb->connCb.clientConnList;
1458 0 : conn = conn2, conn2 = list_entry(conn2->list.next, struct RsConnInfo, list)) {
1459 0 : if (!RsCompareIpAddr(&conn->clientIp, localIp)) {
1460 0 : hccp_warn("Client conn node for IP[%s] do not empty!", localIp->readAddr);
1461 0 : RsFreeConnOneNode(rscb, conn);
1462 0 : conn = NULL;
1463 : }
1464 : }
1465 : }
1466 :
1467 12 : return;
1468 : }
1469 :
1470 40 : STATIC void RsFreeServerConnList(struct rs_cb *rscb)
1471 : {
1472 40 : struct RsConnInfo *conn = NULL;
1473 40 : struct RsConnInfo *conn2 = NULL;
1474 :
1475 40 : if (!RsListEmpty(&rscb->connCb.serverConnList)) {
1476 1 : hccp_warn("Server conn node do not empty!");
1477 1 : RS_LIST_GET_HEAD_ENTRY(conn, conn2, &rscb->connCb.serverConnList, list, struct RsConnInfo);
1478 2 : for (; (&conn->list) != &rscb->connCb.serverConnList;
1479 1 : conn = conn2, conn2 = list_entry(conn2->list.next, struct RsConnInfo, list)) {
1480 1 : RsFreeConnOneNode(rscb, conn);
1481 1 : conn = NULL;
1482 : }
1483 : }
1484 :
1485 40 : return;
1486 : }
1487 :
1488 12 : STATIC void RsFreeDesignatedServerConnNode(struct rs_cb *rscb, struct RsIpAddrInfo *localIp)
1489 : {
1490 12 : struct RsConnInfo *conn = NULL;
1491 12 : struct RsConnInfo *conn2 = NULL;
1492 :
1493 12 : if (!RsListEmpty(&rscb->connCb.serverConnList)) {
1494 1 : RS_LIST_GET_HEAD_ENTRY(conn, conn2, &rscb->connCb.serverConnList, list, struct RsConnInfo);
1495 2 : for (; (&conn->list) != &rscb->connCb.serverConnList;
1496 1 : conn = conn2, conn2 = list_entry(conn2->list.next, struct RsConnInfo, list)) {
1497 1 : if (!RsCompareIpAddr(&conn->serverIp, localIp)) {
1498 1 : hccp_warn("Server conn node for IP[%s] do not empty!", localIp->readAddr);
1499 1 : RsFreeConnOneNode(rscb, conn);
1500 1 : conn = NULL;
1501 : }
1502 : }
1503 : }
1504 12 : return;
1505 : }
1506 :
1507 1 : STATIC void RsFreeListenOneNode(struct rs_cb *rscb, struct RsListenInfo *listen)
1508 : {
1509 : int ret;
1510 :
1511 1 : ret = RsEpollCtl(rscb->connCb.epollfd, EPOLL_CTL_DEL, listen->listenFd, EPOLLIN);
1512 1 : if (ret) {
1513 0 : hccp_err("delete from epoll failed, ret:%d, epollfd:%d, listenFd:%d", ret, rscb->connCb.epollfd,
1514 : listen->listenFd);
1515 : }
1516 :
1517 1 : RS_PTHREAD_MUTEX_LOCK(&rscb->connCb.connMutex);
1518 1 : RsListDel(&listen->list);
1519 1 : RS_PTHREAD_MUTEX_ULOCK(&rscb->connCb.connMutex);
1520 :
1521 1 : RS_CLOSE_RETRY_FOR_EINTR(ret, listen->listenFd);
1522 :
1523 1 : hccp_info("free Listen IP:%s, port:%d, listenFd:%d, state:%u", listen->serverIpAddr.readAddr,
1524 : ntohs(listen->sockPort), listen->listenFd, listen->state);
1525 :
1526 1 : listen->listenFd = RS_FD_INVALID;
1527 1 : listen->state = RS_CONN_STATE_RESET;
1528 :
1529 1 : free(listen);
1530 1 : }
1531 :
1532 40 : STATIC void RsFreeListenList(struct rs_cb *rscb)
1533 : {
1534 40 : struct RsListenInfo *listen = NULL;
1535 40 : struct RsListenInfo *listen2 = NULL;
1536 :
1537 40 : if (!RsListEmpty(&rscb->connCb.listenList)) {
1538 1 : hccp_warn("Server listen node do not empty!");
1539 1 : RS_LIST_GET_HEAD_ENTRY(listen, listen2, &rscb->connCb.listenList, list, struct RsListenInfo);
1540 2 : for (; (&listen->list) != &rscb->connCb.listenList;
1541 1 : listen = listen2, listen2 = list_entry(listen2->list.next, struct RsListenInfo, list)) {
1542 1 : RsFreeListenOneNode(rscb, listen);
1543 1 : listen = NULL;
1544 : }
1545 : }
1546 :
1547 40 : return;
1548 : }
1549 :
1550 12 : STATIC void RsFreeDesignatedListenNode(struct rs_cb *rscb, struct RsIpAddrInfo *localIp)
1551 : {
1552 12 : struct RsListenInfo *listen = NULL;
1553 12 : struct RsListenInfo *listen2 = NULL;
1554 :
1555 12 : if (!RsListEmpty(&rscb->connCb.listenList)) {
1556 0 : RS_LIST_GET_HEAD_ENTRY(listen, listen2, &rscb->connCb.listenList, list, struct RsListenInfo);
1557 0 : for (; (&listen->list) != &rscb->connCb.listenList;
1558 0 : listen = listen2, listen2 = list_entry(listen2->list.next, struct RsListenInfo, list)) {
1559 0 : if (!RsCompareIpAddr(&listen->serverIpAddr, localIp)) {
1560 0 : RsFreeListenOneNode(rscb, listen);
1561 0 : listen = NULL;
1562 : }
1563 : }
1564 : }
1565 :
1566 12 : return;
1567 : }
1568 :
1569 11 : STATIC void RsWhiteListNodeFree(struct rs_cb *rscb, struct RsWhiteList *wlist)
1570 : {
1571 11 : struct RsWhiteListInfo *wlistNode = NULL;
1572 11 : struct RsWhiteListInfo *wlistNode1 = NULL;
1573 :
1574 11 : if (!RsListEmpty(&wlist->whiteList)) {
1575 10 : RS_LIST_GET_HEAD_ENTRY(wlistNode, wlistNode1, &wlist->whiteList, list, struct RsWhiteListInfo);
1576 21 : for (; (&wlistNode->list) != &wlist->whiteList;
1577 11 : wlistNode = wlistNode1, wlistNode1 = list_entry(wlistNode1->list.next, struct RsWhiteListInfo, list)) {
1578 11 : RS_PTHREAD_MUTEX_LOCK(&rscb->connCb.connMutex);
1579 11 : RsListDel(&wlistNode->list);
1580 11 : RS_PTHREAD_MUTEX_ULOCK(&rscb->connCb.connMutex);
1581 :
1582 11 : hccp_info("free White list client IP:%s, tag:%s", wlistNode->clientIp.readAddr, wlistNode->tag);
1583 11 : free(wlistNode);
1584 11 : wlistNode = NULL;
1585 : }
1586 : }
1587 11 : }
1588 :
1589 11 : STATIC void RsFreeWhiteOneNode(struct rs_cb *rscb, struct RsWhiteList *wlist)
1590 : {
1591 11 : RsWhiteListNodeFree(rscb, wlist);
1592 :
1593 11 : RS_PTHREAD_MUTEX_LOCK(&rscb->connCb.connMutex);
1594 11 : RsListDel(&wlist->list);
1595 11 : RS_PTHREAD_MUTEX_ULOCK(&rscb->connCb.connMutex);
1596 :
1597 11 : hccp_info("White list server IP:%s", wlist->serverIp.readAddr);
1598 11 : free(wlist);
1599 11 : wlist = NULL;
1600 11 : }
1601 :
1602 40 : STATIC void RsFreeWhiteList(struct rs_cb *rscb)
1603 : {
1604 40 : struct RsWhiteList *wlist = NULL;
1605 40 : struct RsWhiteList *wlist2 = NULL;
1606 :
1607 40 : if (!RsListEmpty(&rscb->connCb.whiteList)) {
1608 1 : hccp_warn("Server white list do not empty!");
1609 1 : RS_LIST_GET_HEAD_ENTRY(wlist, wlist2, &rscb->connCb.whiteList, list, struct RsWhiteList);
1610 2 : for (; (&wlist->list) != &rscb->connCb.whiteList;
1611 1 : wlist = wlist2, wlist2 = list_entry(wlist2->list.next, struct RsWhiteList, list)) {
1612 1 : RsFreeWhiteOneNode(rscb, wlist);
1613 1 : wlist = NULL;
1614 : }
1615 : }
1616 :
1617 40 : return;
1618 : }
1619 :
1620 12 : STATIC void RsFreeDesignatedWhiteNode(struct rs_cb *rscb, struct RsIpAddrInfo *localIp)
1621 : {
1622 12 : struct RsWhiteList *wlist = NULL;
1623 12 : struct RsWhiteList *wlist2 = NULL;
1624 :
1625 12 : if (!RsListEmpty(&rscb->connCb.whiteList)) {
1626 10 : RS_LIST_GET_HEAD_ENTRY(wlist, wlist2, &rscb->connCb.whiteList, list, struct RsWhiteList);
1627 20 : for (; (&wlist->list) != &rscb->connCb.whiteList;
1628 10 : wlist = wlist2, wlist2 = list_entry(wlist2->list.next, struct RsWhiteList, list)) {
1629 10 : if (!RsCompareIpAddr(&wlist->serverIp, localIp)) {
1630 10 : RsFreeWhiteOneNode(rscb, wlist);
1631 10 : wlist = NULL;
1632 : }
1633 : }
1634 : }
1635 :
1636 12 : return;
1637 : }
1638 :
1639 12 : STATIC void RsFreeSocketList(struct rs_cb *rscb, struct RsIpAddrInfo *localIp)
1640 : {
1641 12 : RsFreeDesignatedAccpetNode(rscb, localIp);
1642 12 : RsFreeDesignatedClientConnNode(rscb, localIp);
1643 :
1644 12 : RsFreeDesignatedServerConnNode(rscb, localIp);
1645 :
1646 12 : RsFreeDesignatedListenNode(rscb, localIp);
1647 :
1648 12 : RsFreeDesignatedWhiteNode(rscb, localIp);
1649 :
1650 12 : return;
1651 : }
1652 :
1653 15 : RS_ATTRI_VISI_DEF int RsSocketDeinit(struct rdev rdevInfo)
1654 : {
1655 : int ret;
1656 15 : unsigned int phyId = rdevInfo.phyId;
1657 : unsigned int chipId;
1658 15 : struct rs_cb *rscb = NULL;
1659 :
1660 15 : hccp_info("rs socket deinit start, phyId:%u", phyId);
1661 15 : CHK_PRT_RETURN(phyId >= RS_MAX_DEV_NUM, hccp_err("rs set param error ! phyId:%u", phyId), -EINVAL);
1662 15 : ret = rsGetLocalDevIDByHostDevID(phyId, &chipId);
1663 15 : CHK_PRT_RETURN(ret, hccp_err("phyId[%u] invalid, ret %d", phyId, ret), ret);
1664 :
1665 15 : CHK_PRT_RETURN((rdevInfo.family != AF_INET) && (rdevInfo.family != AF_INET6),
1666 : hccp_err("family[%d] invalid", rdevInfo.family), -EPROTONOSUPPORT);
1667 :
1668 13 : if (rdevInfo.family == AF_INET) {
1669 13 : unsigned int *localIp = NULL;
1670 13 : localIp = &(rdevInfo.localIp.addr.s_addr);
1671 13 : ret = RsSocketNodeid2vnic(*localIp, localIp);
1672 13 : hccp_info("socket deinit local IP is 0x%llx, ret:%d", *localIp, ret);
1673 : }
1674 :
1675 : struct RsIpAddrInfo localIp;
1676 13 : RsConvertIpAddr(rdevInfo.family, &rdevInfo.localIp, &localIp);
1677 :
1678 13 : ret = RsDev2rscb(chipId, &rscb, false);
1679 13 : CHK_PRT_RETURN(ret, hccp_err("get rscb failed for chipId:%u, ret:%d", chipId, ret), -ENODEV);
1680 :
1681 12 : RS_PTHREAD_MUTEX_LOCK(&rscb->mutex);
1682 12 : RsFreeSocketList(rscb, &localIp);
1683 12 : RS_PTHREAD_MUTEX_ULOCK(&rscb->mutex);
1684 12 : hccp_run_info("socket deinit success, phyId:%u, localIp:%s", phyId, localIp.readAddr);
1685 12 : return 0;
1686 : }
1687 :
1688 44 : STATIC void RsFreeRdevList(struct rs_cb *rsCb)
1689 : {
1690 44 : struct RsRdevCb *rdevCbCurr = NULL;
1691 44 : struct RsRdevCb *rdevCbNext = NULL;
1692 44 : unsigned int phyId = 0;
1693 : int ret;
1694 :
1695 44 : if (RsListEmpty(&rsCb->rdevList)) {
1696 41 : return;
1697 : }
1698 :
1699 3 : hccp_run_warn("Rdev list is not empty!");
1700 3 : ret = rsGetDevIDByLocalDevID(rsCb->chipId, &phyId);
1701 3 : if (ret != 0) {
1702 1 : hccp_err("chipId[%u] invalid, ret %d", rsCb->chipId, ret);
1703 1 : return;
1704 : }
1705 :
1706 2 : RS_LIST_GET_HEAD_ENTRY(rdevCbCurr, rdevCbNext, &rsCb->rdevList, list, struct RsRdevCb);
1707 4 : for (; (&rdevCbCurr->list) != &rsCb->rdevList;
1708 2 : rdevCbCurr = rdevCbNext, rdevCbNext = list_entry(rdevCbNext->list.next, struct RsRdevCb, list)) {
1709 2 : ret = RsRdevDeinit(phyId, rdevCbCurr->notifyType, rdevCbCurr->rdevIndex);
1710 2 : if (ret != 0) {
1711 1 : hccp_err("rs_rdev_deinit failed, ret:%d, phyId:%u", ret, phyId);
1712 : }
1713 : }
1714 :
1715 2 : return;
1716 : }
1717 :
1718 43 : STATIC void RsFreeUdevList(struct rs_cb *rsCb)
1719 : {
1720 43 : struct RsUbDevCb *udevCbCurr = NULL;
1721 43 : struct RsUbDevCb *udevCbNext = NULL;
1722 : int ret;
1723 :
1724 43 : if (RsListEmpty(&rsCb->udevList)) {
1725 41 : return;
1726 : }
1727 :
1728 2 : hccp_run_warn("Udev list is not empty!");
1729 2 : RS_LIST_GET_HEAD_ENTRY(udevCbCurr, udevCbNext, &rsCb->udevList, list, struct RsUbDevCb);
1730 4 : for (; (&udevCbCurr->list) != &rsCb->udevList;
1731 2 : udevCbCurr = udevCbNext, udevCbNext = list_entry(udevCbNext->list.next, struct RsUbDevCb, list)) {
1732 2 : ret = RsUbCtxDeinit(udevCbCurr);
1733 2 : if (ret != 0) {
1734 1 : hccp_err("rs_ub_ctx_deinit failed, ret:%d", ret);
1735 : }
1736 : }
1737 :
1738 2 : return;
1739 : }
1740 :
1741 40 : STATIC void RsFreeHeterogTcpFdList(struct rs_cb *rsCb)
1742 : {
1743 40 : struct RsHeterogTcpFdInfo *fdNode = NULL;
1744 40 : struct RsHeterogTcpFdInfo *fdNode1 = NULL;
1745 :
1746 40 : if (!RsListEmpty(&rsCb->heterogTcpFdList)) {
1747 0 : hccp_warn("heterog_tcp_fd_list do not empty!");
1748 0 : RS_LIST_GET_HEAD_ENTRY(fdNode, fdNode1, &rsCb->heterogTcpFdList, list, struct RsHeterogTcpFdInfo);
1749 0 : for (; (&fdNode->list) != &rsCb->heterogTcpFdList;
1750 0 : fdNode = fdNode1, fdNode1 = list_entry(fdNode1->list.next, struct RsHeterogTcpFdInfo, list)) {
1751 0 : hccp_info(">>>>>fd_node->fd:%d", fdNode->fd);
1752 : // 删除节点
1753 0 : RS_PTHREAD_MUTEX_LOCK(&rsCb->mutex);
1754 0 : RsListDel(&fdNode->list);
1755 0 : free(fdNode);
1756 0 : fdNode = NULL;
1757 0 : RS_PTHREAD_MUTEX_ULOCK(&rsCb->mutex);
1758 : }
1759 : }
1760 :
1761 40 : return;
1762 : }
1763 :
1764 40 : STATIC void RsListFree(struct rs_cb *rscb)
1765 : {
1766 40 : RsFreeAccpetList(rscb);
1767 40 : RsFreeClientConnList(rscb);
1768 :
1769 40 : RsFreeServerConnList(rscb);
1770 :
1771 40 : RsFreeListenList(rscb);
1772 :
1773 40 : RsFreeWhiteList(rscb);
1774 :
1775 40 : return;
1776 : }
1777 :
1778 41 : STATIC void RsSslFree(struct rs_cb *rscb)
1779 : {
1780 41 : if (rscb->sslEnable == RS_SSL_ENABLE) {
1781 2 : if (rscb->skidSubjectCb != NULL) {
1782 1 : if (memset_s(rscb->skidSubjectCb, sizeof(struct RsCertSkidSubjectCb), 0,
1783 : sizeof(struct RsCertSkidSubjectCb))) {
1784 1 : hccp_warn("memset_s for skid_subject_cb unsuccessful");
1785 : }
1786 1 : free(rscb->skidSubjectCb);
1787 1 : rscb->skidSubjectCb = NULL;
1788 : }
1789 2 : ssl_adp_ctx_free(rscb->serverSslCtx);
1790 2 : rscb->serverSslCtx = NULL;
1791 2 : ssl_adp_ctx_free(rscb->clientSslCtx);
1792 2 : rscb->clientSslCtx = NULL;
1793 : }
1794 41 : }
1795 :
1796 40 : STATIC void RsDeinitFreeRscb(struct rs_cb *rscb)
1797 : {
1798 40 : RS_PTHREAD_MUTEX_LOCK(&rscb->mutex);
1799 40 : RsListFree(rscb);
1800 :
1801 40 : free(rscb->fdMap);
1802 40 : rscb->fdMap = NULL;
1803 40 : freeifaddrs(rscb->ifaddrList);
1804 40 : rscb->ifaddrList = NULL;
1805 40 : RS_PTHREAD_MUTEX_ULOCK(&rscb->mutex);
1806 40 : RsFreeRdevList(rscb);
1807 40 : RsFreeUdevList(rscb);
1808 40 : RsSslFree(rscb);
1809 40 : RsFreeHeterogTcpFdList(rscb);
1810 : #ifdef CONFIG_TLV
1811 : if (RsIsTlvSupported()) {
1812 : if (rscb->tlvCb.initFlag) {
1813 : RsTlvDeinit(rscb->tlvCb.phyId);
1814 : }
1815 : }
1816 : #endif
1817 40 : pthread_mutex_destroy(&rscb->mutex);
1818 40 : pthread_mutex_destroy(&rscb->connCb.connMutex);
1819 40 : RsDestroyEpoll(rscb);
1820 :
1821 : #ifdef CUSTOM_INTERFACE
1822 40 : if (RsIsUdmaSupported() || RsIsRdmaSupported()) {
1823 0 : RsDeInitNetAdapt(rscb);
1824 0 : RsEschedDeinit(rscb->protocol);
1825 0 : (void)RsCtxApiDeinit(rscb->hccpMode, rscb->protocol);
1826 : }
1827 : #endif
1828 :
1829 40 : free(rscb);
1830 40 : rscb = NULL;
1831 40 : gRsCb = NULL;
1832 40 : }
1833 :
1834 44 : RS_ATTRI_VISI_DEF int RsDeinit(struct RsInitConfig *cfg)
1835 : {
1836 44 : struct rs_cb *rscb = gRsCb;
1837 : unsigned int chipId;
1838 : eventfd_t event;
1839 : int ret;
1840 :
1841 44 : CHK_PRT_RETURN(cfg == NULL, hccp_err("param error, cfg is NULL"), -EINVAL);
1842 :
1843 44 : chipId = cfg->chipId;
1844 44 : if (__sync_fetch_and_sub(&(gInitCounter[chipId]), 1) > 1) {
1845 0 : return 0;
1846 : }
1847 44 : if (rscb && (chipId == rscb->chipId)) {
1848 40 : event = 1;
1849 : /* send event to eventfd to waking up epoll handle thread */
1850 40 : ret = (int)write(rscb->connCb.eventfd, &event, sizeof(eventfd_t));
1851 40 : CHK_PRT_RETURN(ret != sizeof(eventfd_t),
1852 : hccp_err("eventfd_write failed(0x%x), chipId:%u, errno:%d", ret, chipId, errno), -EFILEOPER);
1853 :
1854 40 : hccp_info("epoll wait up ok, rscb->connFlag:%d", rscb->connFlag);
1855 : // already been RS_CONN_EXIT_FLAG, no need to change conn_flag
1856 40 : if (rscb->connFlag != RS_CONN_EXIT_FLAG) {
1857 40 : rscb->connFlag = 0;
1858 : }
1859 40 : int tryAgain = RS_TRY_TIME;
1860 80 : while (((rscb->state & RS_STATE_HALT) == 0) && tryAgain > 0) {
1861 40 : usleep(RS_USLEEP_TIME);
1862 40 : tryAgain--;
1863 : };
1864 :
1865 40 : if (tryAgain == 0) {
1866 0 : hccp_warn("try_again exhausted, rscb state:%u", rscb->state);
1867 : }
1868 :
1869 40 : tryAgain = RS_TRY_TIME;
1870 6359 : while ((rscb->connFlag != RS_CONN_EXIT_FLAG) && tryAgain > 0) {
1871 6319 : usleep(RS_USLEEP_TIME);
1872 6319 : tryAgain--;
1873 : }
1874 :
1875 40 : CHK_PRT_RETURN(tryAgain == 0, hccp_warn("connect thread quit unsuccessful"), -EAGAIN);
1876 40 : rscb->state &= ~RS_STATE_HALT;
1877 40 : RsDeinitFreeRscb(rscb);
1878 40 : gRsCbList[chipId] = NULL;
1879 40 : DlHalDeinit();
1880 :
1881 40 : hccp_run_info("rs_deinit chipId[%u] ok", chipId);
1882 :
1883 40 : return 0;
1884 : }
1885 :
1886 4 : DlHalDeinit();
1887 4 : return -ENODEV;
1888 : }
1889 :
1890 0 : RS_ATTRI_VISI_DEF int RsGetVnicIp(unsigned int phyId, unsigned int *vnicIp)
1891 : {
1892 0 : int64_t deviceInfo = 0;
1893 : int ret;
1894 :
1895 0 : CHK_PRT_RETURN(phyId >= RS_MAX_DEV_NUM, hccp_err("phyId:%u >= [%d], is invalid", phyId, RS_MAX_DEV_NUM), -EINVAL);
1896 0 : CHK_PRT_RETURN(vnicIp == NULL, hccp_err("vnic_ip is null!"), -EINVAL);
1897 :
1898 0 : ret = DlHalGetDeviceInfo(phyId, MODULE_TYPE_SYSTEM, INFO_TYPE_VNIC_IP, &deviceInfo);
1899 0 : CHK_PRT_RETURN(ret != 0, hccp_err("phyId:%u dl_hal_get_device_info failed! ret:%d", phyId, ret), ret);
1900 :
1901 0 : *vnicIp = (unsigned int)deviceInfo;
1902 0 : return 0;
1903 : }
1904 :
1905 4 : STATIC int RsGetVnicIpInfo(unsigned int phyId, unsigned int id, enum IdType type, struct IpInfo *info)
1906 : {
1907 4 : int64_t deviceInfo = 0;
1908 : unsigned int vnicIp;
1909 : int ret;
1910 :
1911 : // get vnic ip by id with different type
1912 4 : if (type == PHY_ID_VNIC_IP) {
1913 1 : ret = DlHalGetDeviceInfo(id, MODULE_TYPE_SYSTEM, INFO_TYPE_VNIC_IP, &deviceInfo);
1914 1 : CHK_PRT_RETURN(ret != 0,
1915 : hccp_err("cur_phy_id:%u dl_hal_get_device_info failed! phyId:%u ret:%d", phyId, id, ret), ret);
1916 3 : } else if (type == SDID_VNIC_IP) {
1917 1 : ret = DlHalGetDeviceInfo(id, MODULE_TYPE_SYSTEM, INFO_TYPE_SPOD_VNIC_IP, &deviceInfo);
1918 1 : CHK_PRT_RETURN(ret != 0, hccp_err("phyId:%u dl_hal_get_device_info failed! sdid:0x%x ret:%d", phyId, id, ret),
1919 : ret);
1920 : } else {
1921 2 : hccp_err("phyId:%u get vnic ip failed! id:0x%x, invalid type:%u", phyId, id, type);
1922 2 : return -EINVAL;
1923 : }
1924 :
1925 : // prepare ip info, only support IPv4
1926 2 : vnicIp = (unsigned int)deviceInfo;
1927 2 : info->family = AF_INET;
1928 2 : info->ip.addr.s_addr = vnicIp;
1929 :
1930 2 : hccp_dbg("phyId:%u query id:%u type:%u got vnic_ip:%u", phyId, id, type, vnicIp);
1931 :
1932 2 : return 0;
1933 : }
1934 :
1935 3 : RS_ATTRI_VISI_DEF int RsGetVnicIpInfos(unsigned int phyId, enum IdType type, unsigned int ids[], unsigned int num,
1936 : struct IpInfo infos[])
1937 : {
1938 : unsigned int i;
1939 : int ret;
1940 :
1941 3 : CHK_PRT_RETURN(ids == NULL, hccp_err("phyId:%u, ids is null!", phyId), -EINVAL);
1942 2 : CHK_PRT_RETURN(infos == NULL, hccp_err("phyId:%u, infos is null!", phyId), -EINVAL);
1943 :
1944 1 : for (i = 0; i < num; i++) {
1945 1 : ret = RsGetVnicIpInfo(phyId, ids[i], type, &infos[i]);
1946 1 : if (ret != 0) {
1947 1 : hccp_err("phyId:%u get vnic ip info failed! ids[%u]:0x%x type:%u", phyId, i, ids[i], type);
1948 1 : return ret;
1949 : }
1950 : }
1951 :
1952 0 : return 0;
1953 : }
1954 :
1955 464 : int rsGetLocalDevIDByHostDevID(unsigned int phyId, unsigned int *chipId)
1956 : {
1957 464 : CHK_PRT_RETURN(gRsCb == NULL, hccp_warn("No device initialized !"), -ENODEV);
1958 :
1959 452 : if (gRsCb->hccpMode == NETWORK_PEER_ONLINE) {
1960 49 : *chipId = phyId;
1961 49 : return 0;
1962 : } else {
1963 : // to be compatible with mdev scenario: ignore phyId to get current devId(chipId)
1964 403 : return DlDrvQueryProcessHostPid(getpid(), chipId, NULL, NULL, NULL);
1965 : }
1966 : }
1967 :
1968 15 : int rsGetDevIDByLocalDevID(unsigned int chipId, unsigned int *phyId)
1969 : {
1970 15 : CHK_PRT_RETURN(gRsCb == NULL, hccp_warn("No device initialized !"), -ENODEV);
1971 :
1972 15 : if (gRsCb->hccpMode == NETWORK_PEER_ONLINE) {
1973 0 : *phyId = chipId;
1974 0 : return 0;
1975 : } else {
1976 15 : return DlDrvGetDevIdByLocalDevId(chipId, phyId);
1977 : }
1978 : }
1979 :
1980 2 : RS_ATTRI_VISI_DEF int RsSetQpAttrQos(unsigned int phyId, unsigned int rdevIndex, unsigned int qpn, struct QosAttr *attr)
1981 : {
1982 2 : struct RsQpCb *qpCb = NULL;
1983 2 : int ret = 0;
1984 :
1985 2 : RS_QP_PARA_CHECK(phyId);
1986 2 : ret = RsQpn2qpcb(phyId, rdevIndex, qpn, &qpCb);
1987 2 : CHK_PRT_RETURN(ret != 0 || qpCb == NULL, hccp_err("get qp cb failed qpn %u, ret %d", qpn, ret), ret);
1988 :
1989 2 : qpCb->qosAttr.tc = attr->tc;
1990 2 : qpCb->qosAttr.sl = attr->sl;
1991 :
1992 2 : hccp_info("set qp qos attr: qpn[%u] tc[%u] sl[%u]", qpn, attr->tc, attr->sl);
1993 2 : return 0;
1994 : }
1995 :
1996 2 : RS_ATTRI_VISI_DEF int RsSetQpAttrTimeout(unsigned int phyId, unsigned int rdevIndex, unsigned int qpn,
1997 : unsigned int *timeout)
1998 : {
1999 2 : struct RsQpCb *qpCb = NULL;
2000 2 : int ret = 0;
2001 :
2002 2 : RS_QP_PARA_CHECK(phyId);
2003 2 : ret = RsQpn2qpcb(phyId, rdevIndex, qpn, &qpCb);
2004 2 : CHK_PRT_RETURN(ret != 0 || qpCb == NULL, hccp_err("get qp cb failed qpn %u, ret %d", qpn, ret), ret);
2005 :
2006 2 : qpCb->timeout = *timeout;
2007 :
2008 2 : hccp_info("set qp qos attr: qpn[%u] timeout[%u]", qpn, *timeout);
2009 2 : return 0;
2010 : }
2011 :
2012 2 : RS_ATTRI_VISI_DEF int RsSetQpAttrRetryCnt(unsigned int phyId, unsigned int rdevIndex, unsigned int qpn,
2013 : unsigned int *retryCnt)
2014 : {
2015 2 : struct RsQpCb *qpCb = NULL;
2016 2 : int ret = 0;
2017 :
2018 2 : RS_QP_PARA_CHECK(phyId);
2019 2 : ret = RsQpn2qpcb(phyId, rdevIndex, qpn, &qpCb);
2020 2 : CHK_PRT_RETURN(ret != 0 || qpCb == NULL, hccp_err("get qp cb failed qpn %u, ret %d", qpn, ret), ret);
2021 :
2022 2 : qpCb->retryCnt = *retryCnt;
2023 :
2024 2 : hccp_info("set qp qos attr: qpn[%u] retry_cnt[%u]", qpn, *retryCnt);
2025 2 : return 0;
2026 : }
2027 :
2028 1 : RS_ATTRI_VISI_DEF int RsGetCqeErrInfo(struct CqeErrInfo *info)
2029 : {
2030 : int ret;
2031 :
2032 1 : ret = RsDrvGetCqeErrInfo(info);
2033 1 : CHK_PRT_RETURN(ret, hccp_err("get failed! ret:%d", ret), ret);
2034 1 : return 0;
2035 : }
2036 :
2037 2 : RS_ATTRI_VISI_DEF int RsGetCqeErrInfoNum(unsigned int phyId, unsigned int rdevIdx, unsigned int *num)
2038 : {
2039 2 : struct RsRdevCb *rdevCb = NULL;
2040 : unsigned int chipId;
2041 : int ret;
2042 :
2043 2 : CHK_PRT_RETURN(phyId >= RS_MAX_DEV_NUM, hccp_err("rs get cqe err param error, phyId[%u]", phyId), -EINVAL);
2044 1 : ret = rsGetLocalDevIDByHostDevID(phyId, &chipId);
2045 1 : CHK_PRT_RETURN(ret, hccp_err("phyId[%u] invalid, ret %d", phyId, ret), ret);
2046 :
2047 1 : ret = RsRdev2rdevCb(chipId, rdevIdx, &rdevCb);
2048 1 : CHK_PRT_RETURN(ret != 0 || rdevCb == NULL, hccp_err("rs_rdev2rdev_cb for chipId[%u] failed, ret %d", chipId, ret),
2049 : ret);
2050 :
2051 0 : *num = rdevCb->cqeErrCnt;
2052 :
2053 0 : return 0;
2054 : }
2055 :
2056 3 : RS_ATTRI_VISI_DEF int RsGetCqeErrInfoList(unsigned int phyId, unsigned int rdevIdx, struct CqeErrInfo *info,
2057 : unsigned int *num)
2058 : {
2059 3 : struct RsQpCb *qpCbCurr = NULL;
2060 3 : struct RsQpCb *qpCbNext = NULL;
2061 3 : struct RsRdevCb *rdevCb = NULL;
2062 3 : unsigned int cqeErrIdx = 0;
2063 3 : unsigned int numTmp = *num;
2064 : unsigned int chipId;
2065 : int ret;
2066 :
2067 3 : CHK_PRT_RETURN(phyId >= RS_MAX_DEV_NUM, hccp_err("rs get cqe err param error, phyId[%u]", phyId), -EINVAL);
2068 2 : ret = rsGetLocalDevIDByHostDevID(phyId, &chipId);
2069 2 : CHK_PRT_RETURN(ret, hccp_err("phyId[%u] invalid, ret %d", phyId, ret), ret);
2070 :
2071 2 : ret = RsRdev2rdevCb(chipId, rdevIdx, &rdevCb);
2072 2 : CHK_PRT_RETURN(ret != 0 || rdevCb == NULL, hccp_err("rs_rdev2rdev_cb for chipId[%u] failed, ret %d", chipId, ret),
2073 : ret);
2074 :
2075 2 : if (RsListEmpty(&rdevCb->qpList)) {
2076 1 : *num = 0;
2077 1 : return 0;
2078 : }
2079 :
2080 1 : RS_LIST_GET_HEAD_ENTRY(qpCbCurr, qpCbNext, &rdevCb->qpList, list, struct RsQpCb);
2081 2 : for (; (&qpCbCurr->list) != &rdevCb->qpList;
2082 1 : qpCbCurr = qpCbNext, qpCbNext = list_entry(qpCbNext->list.next, struct RsQpCb, list)) {
2083 1 : if (qpCbCurr->cqeErrInfo.info.status != 0) {
2084 1 : RS_PTHREAD_MUTEX_LOCK(&qpCbCurr->cqeErrInfo.mutex);
2085 1 : info[cqeErrIdx].status = qpCbCurr->cqeErrInfo.info.status;
2086 1 : info[cqeErrIdx].qpn = qpCbCurr->cqeErrInfo.info.qpn;
2087 1 : info[cqeErrIdx].time = qpCbCurr->cqeErrInfo.info.time;
2088 1 : qpCbCurr->cqeErrInfo.info.status = 0;
2089 1 : RS_PTHREAD_MUTEX_ULOCK(&qpCbCurr->cqeErrInfo.mutex);
2090 1 : RS_PTHREAD_MUTEX_LOCK(&qpCbCurr->rdevCb->cqeErrCntMutex);
2091 1 : qpCbCurr->rdevCb->cqeErrCnt--;
2092 1 : RS_PTHREAD_MUTEX_ULOCK(&qpCbCurr->rdevCb->cqeErrCntMutex);
2093 1 : cqeErrIdx++;
2094 1 : if (cqeErrIdx == numTmp) {
2095 0 : break;
2096 : }
2097 : }
2098 : }
2099 :
2100 1 : *num = cqeErrIdx;
2101 :
2102 1 : return 0;
2103 : }
2104 :
2105 3 : int RsQueryMrCb(struct RsRdevCb *devCb, uint64_t addr, struct RsMrCb **mrCb, struct RsListHead *mrList)
2106 : {
2107 3 : struct RsMrCb *mrCurr = NULL;
2108 3 : struct RsMrCb *mrNext = NULL;
2109 :
2110 3 : RS_PTHREAD_MUTEX_LOCK(&devCb->rdevMutex);
2111 3 : RS_LIST_GET_HEAD_ENTRY(mrCurr, mrNext, mrList, list, struct RsMrCb);
2112 3 : for (; (&mrCurr->list) != mrList; mrCurr = mrNext, mrNext = list_entry(mrNext->list.next, struct RsMrCb, list)) {
2113 2 : if ((mrCurr->mrInfo.addr <= addr) && (addr < mrCurr->mrInfo.addr + mrCurr->mrInfo.len)) {
2114 2 : *mrCb = mrCurr;
2115 2 : RS_PTHREAD_MUTEX_ULOCK(&devCb->rdevMutex);
2116 2 : return 0;
2117 : }
2118 : }
2119 :
2120 1 : *mrCb = NULL;
2121 1 : RS_PTHREAD_MUTEX_ULOCK(&devCb->rdevMutex);
2122 :
2123 1 : hccp_info("cannot find mrcb for addr@0x%lx !", addr);
2124 :
2125 1 : return -ENODEV;
2126 : }
2127 :
2128 4 : STATIC int RsGetLinuxVersion(struct RsLinuxVersionInfo *verInfo)
2129 : {
2130 : #define LINUX_VERSION_MAX_CHAR 1024
2131 : #define LINUX_VERSION_TYPE_NUM 3
2132 : #define LINUX_VERSION_STR "Linux version "
2133 4 : char buffer[LINUX_VERSION_MAX_CHAR] = {0};
2134 : char *versionStr;
2135 4 : int retClose = 0;
2136 4 : int ret = 0;
2137 : int fd;
2138 :
2139 4 : fd = open("/proc/version", O_RDONLY);
2140 4 : CHK_PRT_RETURN(fd < 0, hccp_run_warn("open proc/version unsuccessful, errno[%d] fd[%d]", errno, fd), -EFILEOPER);
2141 :
2142 : do {
2143 3 : ret = (int)read(fd, buffer, sizeof(buffer) - 1);
2144 3 : } while ((ret < 0) && (errno == EINTR));
2145 :
2146 3 : if (ret < 0) {
2147 1 : hccp_run_warn("read fd unsuccessful[%d]", ret);
2148 1 : RS_CLOSE_RETRY_FOR_EINTR(retClose, fd);
2149 1 : return -EFILEOPER;
2150 : }
2151 :
2152 2 : versionStr = strstr(buffer, LINUX_VERSION_STR);
2153 2 : if (versionStr == NULL) {
2154 1 : hccp_run_warn("can't get Linux version");
2155 1 : RS_CLOSE_RETRY_FOR_EINTR(retClose, fd);
2156 1 : return -EFILEOPER;
2157 : }
2158 1 : versionStr += strlen(LINUX_VERSION_STR);
2159 1 : if (sscanf_s(versionStr, "%d.%d.%d", &verInfo->major, &verInfo->minor, &verInfo->patch) != LINUX_VERSION_TYPE_NUM) {
2160 1 : hccp_run_warn("can't extract Linux version");
2161 1 : RS_CLOSE_RETRY_FOR_EINTR(retClose, fd);
2162 1 : return -EFILEOPER;
2163 : }
2164 :
2165 0 : RS_CLOSE_RETRY_FOR_EINTR(retClose, fd);
2166 0 : return retClose;
2167 : }
2168 :
2169 5 : RS_ATTRI_VISI_DEF int RsGetSecRandom(unsigned int *value)
2170 : {
2171 : #define SEC_LINUX_VERSION_MAJOR 5
2172 : #define SEC_LINUX_VERSION_MINOR 18
2173 : #define SEC_LINUX_VERSION_PATCH 0
2174 5 : struct RsLinuxVersionInfo verInfo = {0};
2175 : int ret;
2176 :
2177 5 : ret = RsGetLinuxVersion(&verInfo);
2178 5 : CHK_PRT_RETURN(ret, hccp_run_warn("[rs_get_random]get_linux_version unsuccessful ret(%d)", ret), ret);
2179 :
2180 : // linux_version > 5.18, urandom is secure
2181 1 : if (verInfo.major > SEC_LINUX_VERSION_MAJOR ||
2182 1 : (verInfo.major == SEC_LINUX_VERSION_MAJOR && verInfo.minor > SEC_LINUX_VERSION_MINOR) ||
2183 0 : (verInfo.major == SEC_LINUX_VERSION_MAJOR && verInfo.minor == SEC_LINUX_VERSION_MINOR &&
2184 0 : verInfo.patch > SEC_LINUX_VERSION_PATCH)) {
2185 1 : ret = RsDrvGetRandomNum((int *)value);
2186 : } else {
2187 0 : hccp_run_warn("[rs_get_random]linux_version is not secure version");
2188 0 : return -ENOTSUPP;
2189 : }
2190 :
2191 1 : if (ret != 0) {
2192 1 : hccp_run_warn("[get][get_random]rs_get_sec_random unsuccessful, ret(%d)", ret);
2193 : }
2194 1 : return ret;
2195 : }
2196 :
2197 452 : RS_ATTRI_VISI_DEF enum ProductType RsGetProductType(int devId)
2198 : {
2199 : static enum ProductType type = PRODUCT_TYPE_NO_VALUE;
2200 : static halChipInfo chipInfo = {0};
2201 : int ret;
2202 :
2203 452 : if (type != PRODUCT_TYPE_NO_VALUE) { // Cache result after first query
2204 450 : hccp_info("[Get][ChipInfo]chip name is %s, type:%d", chipInfo.name, type);
2205 450 : return type;
2206 : }
2207 :
2208 2 : DlHalInit();
2209 2 : ret = DlHalGetChipInfo(devId, &chipInfo);
2210 2 : DlHalDeinit();
2211 :
2212 2 : CHK_PRT_RETURN(ret != 0, hccp_err("[Get][ChipInfo]DlHalGetChipInfo failed ret:%d", ret), PRODUCT_TYPE_INVALID);
2213 :
2214 2 : if (fnmatch("910_93[a-zA-Z1-9_]*", (const char *)chipInfo.name, 0) == 0) {
2215 0 : type = PRODUCT_TYPE_910_93;
2216 2 : } else if (fnmatch("910B[a-zA-Z1-9_]*", (const char *)chipInfo.name, 0) == 0) {
2217 0 : type = PRODUCT_TYPE_910B;
2218 2 : } else if (fnmatch("910_96[a-zA-Z1-9_]*", (const char *)chipInfo.name, 0) == 0) {
2219 0 : type = PRODUCT_TYPE_910_96;
2220 2 : } else if (fnmatch("910[a-zA-Z1-9]*", (const char *)chipInfo.name, 0) == 0) {
2221 0 : type = PRODUCT_TYPE_910;
2222 2 : } else if (fnmatch("310p[a-zA-Z1-9]*", (const char *)chipInfo.name, 0) == 0) {
2223 0 : type = PRODUCT_TYPE_310p;
2224 2 : } else if (fnmatch("950[a-zA-Z1-9]*", (const char *)chipInfo.name, 0) == 0) {
2225 0 : type = PRODUCT_TYPE_950;
2226 2 : } else if (fnmatch("350[a-zA-Z1-9]*", (const char *)chipInfo.name, 0) == 0) {
2227 0 : type = PRODUCT_TYPE_350;
2228 : } else {
2229 2 : type = PRODUCT_TYPE_OTHERS;
2230 : }
2231 :
2232 2 : hccp_run_info("[Get][ChipInfo]chip name is %s, type:%d", chipInfo.name, type);
2233 2 : return type;
2234 : }
|
