604 lines
20 KiB
C
604 lines
20 KiB
C
/*
|
|
* Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
|
|
*
|
|
* This program is free software: you can use, redistribute, and/or modify
|
|
* it under the terms of the GNU Affero General Public License, version 3
|
|
* or later ("AGPL"), as published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#define _DEFAULT_SOURCE
|
|
#include "dmImp.h"
|
|
|
|
#define INTERNAL_USER "_dnd"
|
|
#define INTERNAL_CKEY "_key"
|
|
#define INTERNAL_SECRET "_pwd"
|
|
|
|
static void dmGetMnodeEpSet(SDnode *pDnode, SEpSet *pEpSet) {
|
|
taosRLockLatch(&pDnode->data.latch);
|
|
*pEpSet = pDnode->data.mnodeEps;
|
|
taosRUnLockLatch(&pDnode->data.latch);
|
|
}
|
|
|
|
static void dmSetMnodeEpSet(SDnode *pDnode, SEpSet *pEpSet) {
|
|
dInfo("mnode is changed, num:%d use:%d", pEpSet->numOfEps, pEpSet->inUse);
|
|
|
|
taosWLockLatch(&pDnode->data.latch);
|
|
pDnode->data.mnodeEps = *pEpSet;
|
|
for (int32_t i = 0; i < pEpSet->numOfEps; ++i) {
|
|
dInfo("mnode index:%d %s:%u", i, pEpSet->eps[i].fqdn, pEpSet->eps[i].port);
|
|
}
|
|
|
|
taosWUnLockLatch(&pDnode->data.latch);
|
|
}
|
|
|
|
static inline NodeMsgFp dmGetMsgFp(SMgmtWrapper *pWrapper, SRpcMsg *pRpc) {
|
|
NodeMsgFp msgFp = pWrapper->msgFps[TMSG_INDEX(pRpc->msgType)];
|
|
if (msgFp == NULL) {
|
|
terrno = TSDB_CODE_MSG_NOT_PROCESSED;
|
|
}
|
|
|
|
return msgFp;
|
|
}
|
|
|
|
static inline int32_t dmBuildMsg(SNodeMsg *pMsg, SRpcMsg *pRpc) {
|
|
SRpcConnInfo connInfo = {0};
|
|
if ((pRpc->msgType & 1U) && rpcGetConnInfo(pRpc->handle, &connInfo) != 0) {
|
|
terrno = TSDB_CODE_MND_NO_USER_FROM_CONN;
|
|
dError("failed to build msg since %s, app:%p handle:%p", terrstr(), pRpc->ahandle, pRpc->handle);
|
|
return -1;
|
|
}
|
|
|
|
memcpy(pMsg->user, connInfo.user, TSDB_USER_LEN);
|
|
pMsg->clientIp = connInfo.clientIp;
|
|
pMsg->clientPort = connInfo.clientPort;
|
|
memcpy(&pMsg->rpcMsg, pRpc, sizeof(SRpcMsg));
|
|
if ((pRpc->msgType & 1u)) {
|
|
assert(pRpc->refId != 0);
|
|
}
|
|
// assert(pRpc->handle != NULL && pRpc->refId != 0 && pMsg->rpcMsg.refId != 0);
|
|
return 0;
|
|
}
|
|
|
|
static void dmProcessRpcMsg(SMgmtWrapper *pWrapper, SRpcMsg *pRpc, SEpSet *pEpSet) {
|
|
int32_t code = -1;
|
|
SNodeMsg *pMsg = NULL;
|
|
NodeMsgFp msgFp = NULL;
|
|
uint16_t msgType = pRpc->msgType;
|
|
bool needRelease = false;
|
|
bool isReq = msgType & 1U;
|
|
|
|
if (pEpSet && pEpSet->numOfEps > 0 && msgType == TDMT_MND_STATUS_RSP) {
|
|
dmSetMnodeEpSet(pWrapper->pDnode, pEpSet);
|
|
}
|
|
|
|
if (dmMarkWrapper(pWrapper) != 0) goto _OVER;
|
|
|
|
needRelease = true;
|
|
if ((msgFp = dmGetMsgFp(pWrapper, pRpc)) == NULL) goto _OVER;
|
|
if ((pMsg = taosAllocateQitem(sizeof(SNodeMsg))) == NULL) goto _OVER;
|
|
if (dmBuildMsg(pMsg, pRpc) != 0) goto _OVER;
|
|
|
|
if (pWrapper->procType == DND_PROC_SINGLE) {
|
|
dTrace("msg:%p, created, type:%s handle:%p user:%s", pMsg, TMSG_INFO(msgType), pRpc->handle, pMsg->user);
|
|
code = (*msgFp)(pWrapper, pMsg);
|
|
} else if (pWrapper->procType == DND_PROC_PARENT) {
|
|
dTrace("msg:%p, created and put into child queue, type:%s handle:%p code:0x%04x user:%s contLen:%d", pMsg,
|
|
TMSG_INFO(msgType), pRpc->handle, pMsg->rpcMsg.code & 0XFFFF, pMsg->user, pRpc->contLen);
|
|
code = taosProcPutToChildQ(pWrapper->procObj, pMsg, sizeof(SNodeMsg), pRpc->pCont, pRpc->contLen,
|
|
(isReq && (pMsg->rpcMsg.code == 0)) ? pRpc->handle : NULL, pRpc->refId, PROC_FUNC_REQ);
|
|
} else {
|
|
dTrace("msg:%p, should not processed in child process, handle:%p user:%s", pMsg, pRpc->handle, pMsg->user);
|
|
ASSERT(1);
|
|
}
|
|
|
|
_OVER:
|
|
if (code == 0) {
|
|
if (pWrapper->procType == DND_PROC_PARENT) {
|
|
dTrace("msg:%p, freed in parent process", pMsg);
|
|
taosFreeQitem(pMsg);
|
|
rpcFreeCont(pRpc->pCont);
|
|
}
|
|
} else {
|
|
dError("msg:%p, type:%s handle:%p failed to process since 0x%04x:%s", pMsg, TMSG_INFO(msgType), pRpc->handle,
|
|
code & 0XFFFF, terrstr());
|
|
if (msgType & 1U) {
|
|
if (terrno != 0) code = terrno;
|
|
if (code == TSDB_CODE_NODE_NOT_DEPLOYED || code == TSDB_CODE_NODE_OFFLINE) {
|
|
if (msgType > TDMT_MND_MSG && msgType < TDMT_VND_MSG) {
|
|
code = TSDB_CODE_NODE_REDIRECT;
|
|
}
|
|
}
|
|
|
|
SRpcMsg rsp = {.handle = pRpc->handle, .ahandle = pRpc->ahandle, .code = code, .refId = pRpc->refId};
|
|
tmsgSendRsp(&rsp);
|
|
}
|
|
dTrace("msg:%p, is freed", pMsg);
|
|
taosFreeQitem(pMsg);
|
|
rpcFreeCont(pRpc->pCont);
|
|
}
|
|
|
|
if (needRelease) {
|
|
dmReleaseWrapper(pWrapper);
|
|
}
|
|
}
|
|
|
|
static void dmProcessMsg(SDnode *pDnode, SRpcMsg *pMsg, SEpSet *pEpSet) {
|
|
SDnodeTrans * pTrans = &pDnode->trans;
|
|
tmsg_t msgType = pMsg->msgType;
|
|
bool isReq = msgType & 1u;
|
|
SMsgHandle * pHandle = &pTrans->msgHandles[TMSG_INDEX(msgType)];
|
|
SMgmtWrapper *pWrapper = pHandle->pNdWrapper;
|
|
|
|
if (msgType == TDMT_DND_SERVER_STATUS) {
|
|
dTrace("server status req will be processed, handle:%p, app:%p", pMsg->handle, pMsg->ahandle);
|
|
dmProcessServerStatusReq(pDnode, pMsg);
|
|
return;
|
|
}
|
|
|
|
if (msgType == TDMT_DND_NET_TEST) {
|
|
dTrace("net test req will be processed, handle:%p, app:%p", pMsg->handle, pMsg->ahandle);
|
|
dmProcessServerStatusReq(pDnode, pMsg);
|
|
return;
|
|
}
|
|
|
|
if (pDnode->status != DND_STAT_RUNNING) {
|
|
dError("msg:%s ignored since dnode not running, handle:%p app:%p", TMSG_INFO(msgType), pMsg->handle, pMsg->ahandle);
|
|
if (isReq) {
|
|
SRpcMsg rspMsg = {
|
|
.handle = pMsg->handle, .code = TSDB_CODE_APP_NOT_READY, .ahandle = pMsg->ahandle, .refId = pMsg->refId};
|
|
rpcSendResponse(&rspMsg);
|
|
}
|
|
rpcFreeCont(pMsg->pCont);
|
|
return;
|
|
}
|
|
|
|
if (isReq && pMsg->pCont == NULL) {
|
|
dError("req:%s not processed since its empty, handle:%p app:%p", TMSG_INFO(msgType), pMsg->handle, pMsg->ahandle);
|
|
SRpcMsg rspMsg = {
|
|
.handle = pMsg->handle, .code = TSDB_CODE_INVALID_MSG_LEN, .ahandle = pMsg->ahandle, .refId = pMsg->refId};
|
|
rpcSendResponse(&rspMsg);
|
|
return;
|
|
}
|
|
|
|
if (pWrapper == NULL) {
|
|
dError("msg:%s not processed since no handle, handle:%p app:%p", TMSG_INFO(msgType), pMsg->handle, pMsg->ahandle);
|
|
if (isReq) {
|
|
SRpcMsg rspMsg = {
|
|
.handle = pMsg->handle, .code = TSDB_CODE_MSG_NOT_PROCESSED, .ahandle = pMsg->ahandle, .refId = pMsg->refId};
|
|
rpcSendResponse(&rspMsg);
|
|
}
|
|
rpcFreeCont(pMsg->pCont);
|
|
return;
|
|
}
|
|
|
|
if (pHandle->pMndWrapper != NULL || pHandle->pQndWrapper != NULL) {
|
|
SMsgHead *pHead = pMsg->pCont;
|
|
int32_t vgId = ntohl(pHead->vgId);
|
|
if (vgId == QNODE_HANDLE) {
|
|
pWrapper = pHandle->pQndWrapper;
|
|
} else if (vgId == MNODE_HANDLE) {
|
|
pWrapper = pHandle->pMndWrapper;
|
|
} else {
|
|
}
|
|
}
|
|
|
|
dTrace("msg:%s will be processed by %s, app:%p", TMSG_INFO(msgType), pWrapper->name, pMsg->ahandle);
|
|
if (isReq) {
|
|
assert(pMsg->refId != 0);
|
|
}
|
|
dmProcessRpcMsg(pWrapper, pMsg, pEpSet);
|
|
}
|
|
|
|
int32_t dmInitMsgHandle(SDnode *pDnode) {
|
|
SDnodeTrans *pTrans = &pDnode->trans;
|
|
|
|
for (EDndNodeType n = DNODE; n < NODE_END; ++n) {
|
|
SMgmtWrapper *pWrapper = &pDnode->wrappers[n];
|
|
|
|
for (int32_t msgIndex = 0; msgIndex < TDMT_MAX; ++msgIndex) {
|
|
NodeMsgFp msgFp = pWrapper->msgFps[msgIndex];
|
|
int8_t vgId = pWrapper->msgVgIds[msgIndex];
|
|
if (msgFp == NULL) continue;
|
|
|
|
SMsgHandle *pHandle = &pTrans->msgHandles[msgIndex];
|
|
if (vgId == QNODE_HANDLE) {
|
|
if (pHandle->pQndWrapper != NULL) {
|
|
dError("msg:%s has multiple process nodes", tMsgInfo[msgIndex]);
|
|
return -1;
|
|
}
|
|
pHandle->pQndWrapper = pWrapper;
|
|
} else if (vgId == MNODE_HANDLE) {
|
|
if (pHandle->pMndWrapper != NULL) {
|
|
dError("msg:%s has multiple process nodes", tMsgInfo[msgIndex]);
|
|
return -1;
|
|
}
|
|
pHandle->pMndWrapper = pWrapper;
|
|
} else {
|
|
if (pHandle->pNdWrapper != NULL) {
|
|
dError("msg:%s has multiple process nodes", tMsgInfo[msgIndex]);
|
|
return -1;
|
|
}
|
|
pHandle->pNdWrapper = pWrapper;
|
|
}
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static inline int32_t dmSendRpcReq(SDnode *pDnode, const SEpSet *pEpSet, SRpcMsg *pReq) {
|
|
if (pDnode->trans.clientRpc == NULL) {
|
|
terrno = TSDB_CODE_NODE_OFFLINE;
|
|
return -1;
|
|
}
|
|
|
|
rpcSendRequest(pDnode->trans.clientRpc, pEpSet, pReq, NULL);
|
|
return 0;
|
|
}
|
|
|
|
static void dmSendRpcRedirectRsp(SDnode *pDnode, const SRpcMsg *pReq) {
|
|
SEpSet epSet = {0};
|
|
dmGetMnodeEpSet(pDnode, &epSet);
|
|
|
|
dDebug("RPC %p, req is redirected, num:%d use:%d", pReq->handle, epSet.numOfEps, epSet.inUse);
|
|
for (int32_t i = 0; i < epSet.numOfEps; ++i) {
|
|
dDebug("mnode index:%d %s:%u", i, epSet.eps[i].fqdn, epSet.eps[i].port);
|
|
if (strcmp(epSet.eps[i].fqdn, pDnode->data.localFqdn) == 0 && epSet.eps[i].port == pDnode->data.serverPort) {
|
|
epSet.inUse = (i + 1) % epSet.numOfEps;
|
|
}
|
|
|
|
epSet.eps[i].port = htons(epSet.eps[i].port);
|
|
}
|
|
SRpcMsg resp;
|
|
SMEpSet msg = {.epSet = epSet};
|
|
int32_t len = tSerializeSMEpSet(NULL, 0, &msg);
|
|
resp.pCont = rpcMallocCont(len);
|
|
resp.contLen = len;
|
|
tSerializeSMEpSet(resp.pCont, len, &msg);
|
|
|
|
resp.code = TSDB_CODE_RPC_REDIRECT;
|
|
resp.handle = pReq->handle;
|
|
resp.refId = pReq->refId;
|
|
rpcSendResponse(&resp);
|
|
}
|
|
|
|
static inline void dmSendRpcRsp(SDnode *pDnode, const SRpcMsg *pRsp) {
|
|
if (pRsp->code == TSDB_CODE_NODE_REDIRECT) {
|
|
dmSendRpcRedirectRsp(pDnode, pRsp);
|
|
} else {
|
|
rpcSendResponse(pRsp);
|
|
}
|
|
}
|
|
|
|
void dmSendRecv(SDnode *pDnode, SEpSet *pEpSet, SRpcMsg *pReq, SRpcMsg *pRsp) {
|
|
rpcSendRecv(pDnode->trans.clientRpc, pEpSet, pReq, pRsp);
|
|
}
|
|
|
|
void dmSendToMnodeRecv(SDnode *pDnode, SRpcMsg *pReq, SRpcMsg *pRsp) {
|
|
SEpSet epSet = {0};
|
|
dmGetMnodeEpSet(pDnode, &epSet);
|
|
rpcSendRecv(pDnode->trans.clientRpc, &epSet, pReq, pRsp);
|
|
}
|
|
|
|
static inline int32_t dmSendReq(SMgmtWrapper *pWrapper, const SEpSet *pEpSet, SRpcMsg *pReq) {
|
|
if (pWrapper->pDnode->status != DND_STAT_RUNNING) {
|
|
terrno = TSDB_CODE_NODE_OFFLINE;
|
|
dError("failed to send rpc msg since %s, handle:%p", terrstr(), pReq->handle);
|
|
return -1;
|
|
}
|
|
|
|
if (pWrapper->procType != DND_PROC_CHILD) {
|
|
return dmSendRpcReq(pWrapper->pDnode, pEpSet, pReq);
|
|
} else {
|
|
char *pHead = taosMemoryMalloc(sizeof(SRpcMsg) + sizeof(SEpSet));
|
|
if (pHead == NULL) {
|
|
terrno = TSDB_CODE_OUT_OF_MEMORY;
|
|
return -1;
|
|
}
|
|
|
|
memcpy(pHead, pReq, sizeof(SRpcMsg));
|
|
memcpy(pHead + sizeof(SRpcMsg), pEpSet, sizeof(SEpSet));
|
|
taosProcPutToParentQ(pWrapper->procObj, pHead, sizeof(SRpcMsg) + sizeof(SEpSet), pReq->pCont, pReq->contLen,
|
|
PROC_FUNC_REQ);
|
|
taosMemoryFree(pHead);
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
static inline void dmSendRsp(SMgmtWrapper *pWrapper, const SRpcMsg *pRsp) {
|
|
if (pWrapper->procType != DND_PROC_CHILD) {
|
|
dmSendRpcRsp(pWrapper->pDnode, pRsp);
|
|
} else {
|
|
taosProcPutToParentQ(pWrapper->procObj, pRsp, sizeof(SRpcMsg), pRsp->pCont, pRsp->contLen, PROC_FUNC_RSP);
|
|
}
|
|
}
|
|
|
|
static inline void dmSendRedirectRsp(SMgmtWrapper *pWrapper, const SRpcMsg *pRsp, const SEpSet *pNewEpSet) {
|
|
ASSERT(pRsp->code == TSDB_CODE_RPC_REDIRECT);
|
|
ASSERT(pRsp->pCont == NULL);
|
|
if (pWrapper->procType != DND_PROC_CHILD) {
|
|
SRpcMsg resp = {0};
|
|
SMEpSet msg = {.epSet = *pNewEpSet};
|
|
int32_t len = tSerializeSMEpSet(NULL, 0, &msg);
|
|
resp.pCont = rpcMallocCont(len);
|
|
resp.contLen = len;
|
|
tSerializeSMEpSet(resp.pCont, len, &msg);
|
|
|
|
resp.code = TSDB_CODE_RPC_REDIRECT;
|
|
resp.handle = pRsp->handle;
|
|
resp.refId = pRsp->refId;
|
|
rpcSendResponse(&resp);
|
|
} else {
|
|
taosProcPutToParentQ(pWrapper->procObj, pRsp, sizeof(SRpcMsg), pRsp->pCont, pRsp->contLen, PROC_FUNC_RSP);
|
|
}
|
|
}
|
|
|
|
#if 0
|
|
static inline void dmSendRedirectRsp(SMgmtWrapper *pWrapper, const SRpcMsg *pRsp, const SEpSet *pNewEpSet) {
|
|
ASSERT(pRsp->code == TSDB_CODE_RPC_REDIRECT);
|
|
if (pWrapper->procType != DND_PROC_CHILD) {
|
|
rpcSendRedirectRsp(pRsp->handle, pNewEpSet);
|
|
} else {
|
|
taosProcPutToParentQ(pWrapper->procObj, pRsp, sizeof(SRpcMsg), pRsp->pCont, pRsp->contLen, PROC_FUNC_RSP);
|
|
}
|
|
}
|
|
#endif
|
|
|
|
static inline void dmRegisterBrokenLinkArg(SMgmtWrapper *pWrapper, SRpcMsg *pMsg) {
|
|
if (pWrapper->procType != DND_PROC_CHILD) {
|
|
rpcRegisterBrokenLinkArg(pMsg);
|
|
} else {
|
|
taosProcPutToParentQ(pWrapper->procObj, pMsg, sizeof(SRpcMsg), pMsg->pCont, pMsg->contLen, PROC_FUNC_REGIST);
|
|
}
|
|
}
|
|
|
|
static inline void dmReleaseHandle(SMgmtWrapper *pWrapper, void *handle, int8_t type) {
|
|
if (pWrapper->procType != DND_PROC_CHILD) {
|
|
rpcReleaseHandle(handle, type);
|
|
} else {
|
|
SRpcMsg msg = {.handle = handle, .code = type};
|
|
taosProcPutToParentQ(pWrapper->procObj, &msg, sizeof(SRpcMsg), NULL, 0, PROC_FUNC_RELEASE);
|
|
}
|
|
}
|
|
|
|
static void dmConsumeChildQueue(SMgmtWrapper *pWrapper, SNodeMsg *pMsg, int16_t msgLen, void *pCont, int32_t contLen,
|
|
EProcFuncType ftype) {
|
|
SRpcMsg *pRpc = &pMsg->rpcMsg;
|
|
pRpc->pCont = pCont;
|
|
dTrace("msg:%p, get from child queue, handle:%p app:%p", pMsg, pRpc->handle, pRpc->ahandle);
|
|
|
|
NodeMsgFp msgFp = pWrapper->msgFps[TMSG_INDEX(pRpc->msgType)];
|
|
int32_t code = (*msgFp)(pWrapper, pMsg);
|
|
|
|
if (code != 0) {
|
|
dError("msg:%p, failed to process since code:0x%04x:%s", pMsg, code & 0XFFFF, tstrerror(code));
|
|
if (pRpc->msgType & 1U) {
|
|
SRpcMsg rsp = {.handle = pRpc->handle, .ahandle = pRpc->ahandle, .code = terrno, .refId = pRpc->refId};
|
|
dmSendRsp(pWrapper, &rsp);
|
|
}
|
|
|
|
dTrace("msg:%p, is freed", pMsg);
|
|
taosFreeQitem(pMsg);
|
|
rpcFreeCont(pCont);
|
|
}
|
|
}
|
|
|
|
static void dmConsumeParentQueue(SMgmtWrapper *pWrapper, SRpcMsg *pMsg, int16_t msgLen, void *pCont, int32_t contLen,
|
|
EProcFuncType ftype) {
|
|
int32_t code = pMsg->code & 0xFFFF;
|
|
pMsg->pCont = pCont;
|
|
|
|
if (ftype == PROC_FUNC_REQ) {
|
|
dTrace("msg:%p, get from parent queue, send req:%s handle:%p code:0x%04x, app:%p", pMsg, TMSG_INFO(pMsg->msgType),
|
|
pMsg->handle, code, pMsg->ahandle);
|
|
dmSendRpcReq(pWrapper->pDnode, (SEpSet *)((char *)pMsg + sizeof(SRpcMsg)), pMsg);
|
|
} else if (ftype == PROC_FUNC_RSP) {
|
|
dTrace("msg:%p, get from parent queue, rsp handle:%p code:0x%04x, app:%p", pMsg, pMsg->handle, code, pMsg->ahandle);
|
|
pMsg->refId = taosProcRemoveHandle(pWrapper->procObj, pMsg->handle);
|
|
dmSendRpcRsp(pWrapper->pDnode, pMsg);
|
|
} else if (ftype == PROC_FUNC_REGIST) {
|
|
dTrace("msg:%p, get from parent queue, regist handle:%p code:0x%04x, app:%p", pMsg, pMsg->handle, code,
|
|
pMsg->ahandle);
|
|
rpcRegisterBrokenLinkArg(pMsg);
|
|
} else if (ftype == PROC_FUNC_RELEASE) {
|
|
dTrace("msg:%p, get from parent queue, release handle:%p code:0x%04x, app:%p", pMsg, pMsg->handle, code,
|
|
pMsg->ahandle);
|
|
taosProcRemoveHandle(pWrapper->procObj, pMsg->handle);
|
|
rpcReleaseHandle(pMsg->handle, (int8_t)pMsg->code);
|
|
rpcFreeCont(pCont);
|
|
} else {
|
|
dError("msg:%p, invalid ftype:%d while get from parent queue, handle:%p", pMsg, ftype, pMsg->handle);
|
|
}
|
|
|
|
taosMemoryFree(pMsg);
|
|
}
|
|
|
|
SProcCfg dmGenProcCfg(SMgmtWrapper *pWrapper) {
|
|
SProcCfg cfg = {.childConsumeFp = (ProcConsumeFp)dmConsumeChildQueue,
|
|
.childMallocHeadFp = (ProcMallocFp)taosAllocateQitem,
|
|
.childFreeHeadFp = (ProcFreeFp)taosFreeQitem,
|
|
.childMallocBodyFp = (ProcMallocFp)rpcMallocCont,
|
|
.childFreeBodyFp = (ProcFreeFp)rpcFreeCont,
|
|
.parentConsumeFp = (ProcConsumeFp)dmConsumeParentQueue,
|
|
.parentMallocHeadFp = (ProcMallocFp)taosMemoryMalloc,
|
|
.parentFreeHeadFp = (ProcFreeFp)taosMemoryFree,
|
|
.parentMallocBodyFp = (ProcMallocFp)rpcMallocCont,
|
|
.parentFreeBodyFp = (ProcFreeFp)rpcFreeCont,
|
|
.shm = pWrapper->procShm,
|
|
.parent = pWrapper,
|
|
.name = pWrapper->name};
|
|
return cfg;
|
|
}
|
|
|
|
bool rpcRfp(int32_t code) {
|
|
if (code == TSDB_CODE_RPC_REDIRECT) {
|
|
return true;
|
|
} else {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
static int32_t dmInitClient(SDnode *pDnode) {
|
|
SDnodeTrans *pTrans = &pDnode->trans;
|
|
|
|
SRpcInit rpcInit = {0};
|
|
rpcInit.label = "DND";
|
|
rpcInit.numOfThreads = 1;
|
|
rpcInit.cfp = (RpcCfp)dmProcessMsg;
|
|
rpcInit.sessions = 1024;
|
|
rpcInit.connType = TAOS_CONN_CLIENT;
|
|
rpcInit.idleTime = tsShellActivityTimer * 1000;
|
|
rpcInit.user = INTERNAL_USER;
|
|
rpcInit.ckey = INTERNAL_CKEY;
|
|
rpcInit.spi = 1;
|
|
rpcInit.parent = pDnode;
|
|
rpcInit.rfp = rpcRfp;
|
|
|
|
char pass[TSDB_PASSWORD_LEN + 1] = {0};
|
|
taosEncryptPass_c((uint8_t *)(INTERNAL_SECRET), strlen(INTERNAL_SECRET), pass);
|
|
rpcInit.secret = pass;
|
|
|
|
pTrans->clientRpc = rpcOpen(&rpcInit);
|
|
if (pTrans->clientRpc == NULL) {
|
|
dError("failed to init dnode rpc client");
|
|
return -1;
|
|
}
|
|
|
|
dDebug("dnode rpc client is initialized");
|
|
return 0;
|
|
}
|
|
|
|
static void dmCleanupClient(SDnode *pDnode) {
|
|
SDnodeTrans *pTrans = &pDnode->trans;
|
|
if (pTrans->clientRpc) {
|
|
rpcClose(pTrans->clientRpc);
|
|
pTrans->clientRpc = NULL;
|
|
dDebug("dnode rpc client is closed");
|
|
}
|
|
}
|
|
|
|
static inline int32_t dmGetHideUserAuth(SDnode *pDnode, char *user, char *spi, char *encrypt, char *secret,
|
|
char *ckey) {
|
|
int32_t code = 0;
|
|
char pass[TSDB_PASSWORD_LEN + 1] = {0};
|
|
|
|
if (strcmp(user, INTERNAL_USER) == 0) {
|
|
taosEncryptPass_c((uint8_t *)(INTERNAL_SECRET), strlen(INTERNAL_SECRET), pass);
|
|
} else if (strcmp(user, TSDB_NETTEST_USER) == 0) {
|
|
taosEncryptPass_c((uint8_t *)(TSDB_NETTEST_USER), strlen(TSDB_NETTEST_USER), pass);
|
|
} else {
|
|
code = -1;
|
|
}
|
|
|
|
if (code == 0) {
|
|
memcpy(secret, pass, TSDB_PASSWORD_LEN);
|
|
*spi = 1;
|
|
*encrypt = 0;
|
|
*ckey = 0;
|
|
}
|
|
|
|
return code;
|
|
}
|
|
|
|
static inline int32_t dmRetrieveUserAuthInfo(SDnode *pDnode, char *user, char *spi, char *encrypt, char *secret,
|
|
char *ckey) {
|
|
if (dmGetHideUserAuth(pDnode, user, spi, encrypt, secret, ckey) == 0) {
|
|
dTrace("user:%s, get auth from mnode, spi:%d encrypt:%d", user, *spi, *encrypt);
|
|
return 0;
|
|
}
|
|
|
|
SAuthReq authReq = {0};
|
|
tstrncpy(authReq.user, user, TSDB_USER_LEN);
|
|
int32_t contLen = tSerializeSAuthReq(NULL, 0, &authReq);
|
|
void * pReq = rpcMallocCont(contLen);
|
|
tSerializeSAuthReq(pReq, contLen, &authReq);
|
|
|
|
SRpcMsg rpcMsg = {.pCont = pReq, .contLen = contLen, .msgType = TDMT_MND_AUTH, .ahandle = (void *)9528};
|
|
SRpcMsg rpcRsp = {0};
|
|
dTrace("user:%s, send user auth req to other mnodes, spi:%d encrypt:%d", user, authReq.spi, authReq.encrypt);
|
|
dmSendToMnodeRecv(pDnode, &rpcMsg, &rpcRsp);
|
|
|
|
if (rpcRsp.code != 0) {
|
|
terrno = rpcRsp.code;
|
|
dError("user:%s, failed to get user auth from other mnodes since %s", user, terrstr());
|
|
} else {
|
|
SAuthRsp authRsp = {0};
|
|
tDeserializeSAuthReq(rpcRsp.pCont, rpcRsp.contLen, &authRsp);
|
|
memcpy(secret, authRsp.secret, TSDB_PASSWORD_LEN);
|
|
memcpy(ckey, authRsp.ckey, TSDB_PASSWORD_LEN);
|
|
*spi = authRsp.spi;
|
|
*encrypt = authRsp.encrypt;
|
|
dTrace("user:%s, success to get user auth from other mnodes, spi:%d encrypt:%d", user, authRsp.spi,
|
|
authRsp.encrypt);
|
|
}
|
|
|
|
rpcFreeCont(rpcRsp.pCont);
|
|
return rpcRsp.code;
|
|
}
|
|
|
|
static int32_t dmInitServer(SDnode *pDnode) {
|
|
SDnodeTrans *pTrans = &pDnode->trans;
|
|
|
|
SRpcInit rpcInit = {0};
|
|
|
|
strncpy(rpcInit.localFqdn, pDnode->data.localFqdn, strlen(pDnode->data.localFqdn));
|
|
rpcInit.localPort = pDnode->data.serverPort;
|
|
rpcInit.label = "DND";
|
|
rpcInit.numOfThreads = tsNumOfRpcThreads;
|
|
rpcInit.cfp = (RpcCfp)dmProcessMsg;
|
|
rpcInit.sessions = tsMaxShellConns;
|
|
rpcInit.connType = TAOS_CONN_SERVER;
|
|
rpcInit.idleTime = tsShellActivityTimer * 1000;
|
|
rpcInit.afp = (RpcAfp)dmRetrieveUserAuthInfo;
|
|
rpcInit.parent = pDnode;
|
|
|
|
pTrans->serverRpc = rpcOpen(&rpcInit);
|
|
if (pTrans->serverRpc == NULL) {
|
|
dError("failed to init dnode rpc server");
|
|
return -1;
|
|
}
|
|
|
|
dDebug("dnode rpc server is initialized");
|
|
return 0;
|
|
}
|
|
|
|
static void dmCleanupServer(SDnode *pDnode) {
|
|
SDnodeTrans *pTrans = &pDnode->trans;
|
|
if (pTrans->serverRpc) {
|
|
rpcClose(pTrans->serverRpc);
|
|
pTrans->serverRpc = NULL;
|
|
dDebug("dnode rpc server is closed");
|
|
}
|
|
}
|
|
|
|
int32_t dmInitTrans(SDnode *pDnode) {
|
|
if (dmInitServer(pDnode) != 0) return -1;
|
|
if (dmInitClient(pDnode) != 0) return -1;
|
|
return 0;
|
|
}
|
|
|
|
void dmCleanupTrans(SDnode *pDnode) {
|
|
dmCleanupServer(pDnode);
|
|
dmCleanupClient(pDnode);
|
|
}
|
|
|
|
SMsgCb dmGetMsgcb(SMgmtWrapper *pWrapper) {
|
|
SMsgCb msgCb = {
|
|
.sendReqFp = dmSendReq,
|
|
.sendRspFp = dmSendRsp,
|
|
.sendRedirectRspFp = dmSendRedirectRsp,
|
|
.registerBrokenLinkArgFp = dmRegisterBrokenLinkArg,
|
|
.releaseHandleFp = dmReleaseHandle,
|
|
.reportStartupFp = dmReportStartupByWrapper,
|
|
.pWrapper = pWrapper,
|
|
};
|
|
return msgCb;
|
|
}
|