#4476

Committed 08 Jul 2025 09:38AM UTC coverage: 62.948% (+0.03%) from 62.922%

Build # #4476

Build Type

push

travis-ci

Committed by

web-flow

Commit Message

Merge pull request #31712 from taosdata/merge/mainto3.0

merge: from main to 3.0 branch

Run Details

158623 of 321496 branches covered (49.34%)

Branch coverage included in aggregate %.

56 of 60 new or added lines in 13 files covered. (93.33%)

1176 existing lines in 42 files now uncovered.

245593 of 320647 relevant lines covered (76.59%)

23441503.11 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

63.71

/source/common/src/tmisce.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 "tdatablock.h"
#include "tglobal.h"
#include "tjson.h"
#include "tmisce.h"
#include "tcompare.h"

int32_t taosGetIpv4FromEp(const char* ep, SEp* pEp) {
  pEp->port = 0;
  memset(pEp->fqdn, 0, TSDB_FQDN_LEN);
  tstrncpy(pEp->fqdn, ep, TSDB_FQDN_LEN);

  char* temp = strrchr(pEp->fqdn, ':');
  if (temp) {
    *temp = 0;
    pEp->port = taosStr2UInt16(temp + 1, NULL, 10);
    if (pEp->port < 0) {
      return TSDB_CODE_INVALID_PARA;
    }
  }

  if (pEp->port == 0) {
    pEp->port = tsServerPort;
  }

  if (pEp->port <= 0) {
    return TSDB_CODE_INVALID_PARA;
  }
  return 0;
}

static int8_t isValidPort(const char* str) {
  if (!str || !*str) return 0;
  for (const char* p = str; *p; ++p) {
    if (*p < '0' || *p > '9') {
      return 0;
    }
  }
  return 1;
}

static int32_t taosGetDualIpFromEp(const char* ep, SEp* pEp) {
  memset(pEp->fqdn, 0, TSDB_FQDN_LEN);
  pEp->port = 0;
  char buf[TSDB_FQDN_LEN] = {0};

  if (ep[0] == '[') {
    // [IPv6]:port format
    const char* end = strchr(ep, ']');
    if (!end) return TSDB_CODE_INVALID_PARA;

    int ipLen = end - ep - 1;
    if (ipLen >= TSDB_FQDN_LEN) ipLen = TSDB_FQDN_LEN - 1;

    strncpy(pEp->fqdn, ep + 1, ipLen);
    pEp->fqdn[ipLen] = '\0';

    if (*(end + 1) == ':' && *(end + 2)) {
      pEp->port = taosStr2UInt16(end + 2, NULL, 10);
    }
  } else {
    // Compatible with ::1:6030, ::1, IPv4:port, hostname:port, etc.
    strncpy(buf, ep, TSDB_FQDN_LEN - 1);
    buf[TSDB_FQDN_LEN - 1] = 0;

    char* lastColon = strrchr(buf, ':');
    char* firstColon = strchr(buf, ':');

    if (lastColon && firstColon != lastColon) {
      // Multiple colons, possibly IPv6:port or pure IPv6
      char* portStr = lastColon + 1;
      if (isValidPort(portStr) && lastColon != buf && (*(lastColon - 1) != ':')) {
        *lastColon = 0;
        tstrncpy(pEp->fqdn, buf, TSDB_FQDN_LEN);
        pEp->port = taosStr2UInt16(portStr, NULL, 10);
      } else {
        tstrncpy(pEp->fqdn, ep, TSDB_FQDN_LEN);
      }
    } else if (lastColon) {
      // Only one colon, IPv4:port or hostname:port
      char* portStr = lastColon + 1;
      if (isValidPort(portStr) && lastColon != buf) {
        *lastColon = 0;
        tstrncpy(pEp->fqdn, buf, TSDB_FQDN_LEN);
        pEp->port = taosStr2UInt16(portStr, NULL, 10);
      } else {
        tstrncpy(pEp->fqdn, ep, TSDB_FQDN_LEN);
      }
    } else {
      // No colon, pure hostname or IPv6 without port
      tstrncpy(pEp->fqdn, ep, TSDB_FQDN_LEN);
    }
  }

  if (pEp->port == 0) {
    pEp->port = tsServerPort;
  }

  if (pEp->port <= 0) {
    return TSDB_CODE_INVALID_PARA;
  }

  return 0;
}
int32_t taosGetFqdnPortFromEp(const char* ep, SEp* pEp) {
  if (tsEnableIpv6) {
    return taosGetDualIpFromEp(ep, pEp);
  } else {
    return taosGetIpv4FromEp(ep, pEp);
  }
}

int32_t addEpIntoEpSet(SEpSet* pEpSet, const char* fqdn, uint16_t port) {
  if (pEpSet == NULL || fqdn == NULL || strlen(fqdn) == 0) {
    return TSDB_CODE_INVALID_PARA;
  }

  int32_t index = pEpSet->numOfEps;
  if (index >= sizeof(pEpSet->eps) / sizeof(pEpSet->eps[0])) {
    return TSDB_CODE_OUT_OF_RANGE;
  }
  tstrncpy(pEpSet->eps[index].fqdn, fqdn, tListLen(pEpSet->eps[index].fqdn));
  pEpSet->eps[index].port = port;
  pEpSet->numOfEps += 1;
  return 0;
}

bool isEpsetEqual(const SEpSet* s1, const SEpSet* s2) {
  if (s1->numOfEps != s2->numOfEps || s1->inUse != s2->inUse) {
    return false;
  }

  for (int32_t i = 0; i < s1->numOfEps; i++) {
    if (s1->eps[i].port != s2->eps[i].port || strncmp(s1->eps[i].fqdn, s2->eps[i].fqdn, TSDB_FQDN_LEN) != 0)
      return false;
  }
  return true;
}

void epsetAssign(SEpSet* pDst, const SEpSet* pSrc) {
  if (pSrc == NULL || pDst == NULL) {
    return;
  }

  pDst->inUse = pSrc->inUse;
  pDst->numOfEps = pSrc->numOfEps;
  for (int32_t i = 0; i < pSrc->numOfEps; ++i) {
    pDst->eps[i].port = pSrc->eps[i].port;
    tstrncpy(pDst->eps[i].fqdn, pSrc->eps[i].fqdn, tListLen(pSrc->eps[i].fqdn));
  }
}

void epAssign(SEp* pDst, SEp* pSrc) {
  if (pSrc == NULL || pDst == NULL) {
    return;
  }
  memset(pDst->fqdn, 0, tListLen(pSrc->fqdn));
  tstrncpy(pDst->fqdn, pSrc->fqdn, tListLen(pSrc->fqdn));
  pDst->port = pSrc->port;
}

void epsetSort(SEpSet* pDst) {
  if (pDst->numOfEps <= 1) {
    return;
  }
  int validIdx = false;
  SEp ep = {0};
  if (pDst->inUse >= 0 && pDst->inUse < pDst->numOfEps) {
    validIdx = true;
    epAssign(&ep, &pDst->eps[pDst->inUse]);
  }

  for (int i = 0; i < pDst->numOfEps - 1; i++) {
    for (int j = 0; j < pDst->numOfEps - 1 - i; j++) {
      SEp* f = &pDst->eps[j];
      SEp* s = &pDst->eps[j + 1];
      int  cmp = strncmp(f->fqdn, s->fqdn, sizeof(f->fqdn));
      if (cmp > 0 || (cmp == 0 && f->port > s->port)) {
        SEp ep1 = {0};
        epAssign(&ep1, f);
        epAssign(f, s);
        epAssign(s, &ep1);
      }
    }
  }
  if (validIdx == true)
    for (int i = 0; i < pDst->numOfEps; i++) {
      int cmp = strncmp(ep.fqdn, pDst->eps[i].fqdn, sizeof(ep.fqdn));
      if (cmp == 0 && ep.port == pDst->eps[i].port) {
        pDst->inUse = i;
        break;
      }
    }
}

void updateEpSet_s(SCorEpSet* pEpSet, SEpSet* pNewEpSet) {
  taosCorBeginWrite(&pEpSet->version);
  pEpSet->epSet = *pNewEpSet;
  taosCorEndWrite(&pEpSet->version);
}

SEpSet getEpSet_s(SCorEpSet* pEpSet) {
  SEpSet ep = {0};
  taosCorBeginRead(&pEpSet->version);
  ep = pEpSet->epSet;
  taosCorEndRead(&pEpSet->version);

  return ep;
}

int32_t epsetToStr(const SEpSet* pEpSet, char* pBuf, int32_t cap) {
  int32_t ret = 0;
  int32_t nwrite = 0;

  nwrite = snprintf(pBuf + nwrite, cap, "epset:{");
  if (nwrite <= 0 || nwrite >= cap) {
    return TSDB_CODE_OUT_OF_BUFFER;
  }
  cap -= nwrite;

  for (int _i = 0; (_i < pEpSet->numOfEps) && (cap > 0); _i++) {
    if (_i == pEpSet->numOfEps - 1) {
      ret = snprintf(pBuf + nwrite, cap, "%d. %s:%d", _i, pEpSet->eps[_i].fqdn, pEpSet->eps[_i].port);
    } else {
      ret = snprintf(pBuf + nwrite, cap, "%d. %s:%d, ", _i, pEpSet->eps[_i].fqdn, pEpSet->eps[_i].port);
    }

    if (ret <= 0 || ret >= cap) {
      return TSDB_CODE_OUT_OF_BUFFER;
    }

    nwrite += ret;
    cap -= ret;
  }

  if (cap <= 0) {
    return TSDB_CODE_OUT_OF_BUFFER;
  }

  ret = snprintf(pBuf + nwrite, cap, "}, inUse:%d", pEpSet->inUse);
  if (ret <= 0 || ret >= cap) {
    return TSDB_CODE_OUT_OF_BUFFER;
  } else {
    return TSDB_CODE_SUCCESS;
  }
}

int32_t taosGenCrashJsonMsg(int signum, char** pMsg, int64_t clusterId, int64_t startTime) {
  int32_t code = 0;
  SJson*  pJson = tjsonCreateObject();
  if (pJson == NULL) return terrno;

  char tmp[4096] = {0};

  TAOS_CHECK_GOTO(tjsonAddDoubleToObject(pJson, "reportVersion", 1), NULL, _exit);

  TAOS_CHECK_GOTO(tjsonAddIntegerToObject(pJson, "clusterId", clusterId), NULL, _exit);
  TAOS_CHECK_GOTO(tjsonAddIntegerToObject(pJson, "startTime", startTime), NULL, _exit);

  // Do NOT invoke the taosGetFqdn here.
  // this function may be invoked when memory exception occurs,so we should assume that it is running in a memory locked
  // environment. The lock operation by taosGetFqdn may cause this program deadlock.
  TAOS_CHECK_GOTO(tjsonAddStringToObject(pJson, "fqdn", tsLocalFqdn), NULL, _exit);

  TAOS_CHECK_GOTO(tjsonAddIntegerToObject(pJson, "pid", taosGetPId()), NULL, _exit);

  code = taosGetAppName(tmp, NULL);
  if (code != 0) {
    TAOS_CHECK_GOTO(code, NULL, _exit);
  }
  TAOS_CHECK_GOTO(tjsonAddStringToObject(pJson, "appName", tmp), NULL, _exit);

  if (taosGetOsReleaseName(tmp, NULL, NULL, sizeof(tmp)) == 0) {
    TAOS_CHECK_GOTO(tjsonAddStringToObject(pJson, "os", tmp), NULL, _exit);
  } else {
    // do nothing
  }

  float numOfCores = 0;
  if (taosGetCpuInfo(tmp, sizeof(tmp), &numOfCores) == 0) {
    TAOS_CHECK_GOTO(tjsonAddStringToObject(pJson, "cpuModel", tmp), NULL, _exit);
    TAOS_CHECK_GOTO(tjsonAddDoubleToObject(pJson, "numOfCpu", numOfCores), NULL, _exit);
  } else {
    TAOS_CHECK_GOTO(tjsonAddDoubleToObject(pJson, "numOfCpu", tsNumOfCores), NULL, _exit);
  }

  int32_t nBytes = snprintf(tmp, sizeof(tmp), "%" PRId64 " kB", tsTotalMemoryKB);
  if (nBytes <= 9 || nBytes >= sizeof(tmp)) {
    TAOS_CHECK_GOTO(TSDB_CODE_OUT_OF_RANGE, NULL, _exit);
  }
  TAOS_CHECK_GOTO(tjsonAddStringToObject(pJson, "memory", tmp), NULL, _exit);

  TAOS_CHECK_GOTO(tjsonAddStringToObject(pJson, "version", td_version), NULL, _exit);
  TAOS_CHECK_GOTO(tjsonAddStringToObject(pJson, "buildInfo", td_buildinfo), NULL, _exit);
  TAOS_CHECK_GOTO(tjsonAddStringToObject(pJson, "gitInfo", td_gitinfo), NULL, _exit);

  TAOS_CHECK_GOTO(tjsonAddIntegerToObject(pJson, "crashSig", signum), NULL, _exit);
  TAOS_CHECK_GOTO(tjsonAddIntegerToObject(pJson, "crashTs", taosGetTimestampUs()), NULL, _exit);

#if 0
#ifdef _TD_DARWIN_64
  taosLogTraceToBuf(tmp, sizeof(tmp), 4);
#elif !defined(WINDOWS)
  taosLogTraceToBuf(tmp, sizeof(tmp), 3);
#else
  taosLogTraceToBuf(tmp, sizeof(tmp), 8);
#endif

  TAOS_CHECK_GOTO(tjsonAddStringToObject(pJson, "stackInfo", tmp), NULL, _exit);
#endif
  char* pCont = tjsonToString(pJson);
  if (pCont == NULL) {
    code = terrno;
    TAOS_CHECK_GOTO(code, NULL, _exit);
    goto _exit;
  }

  tjsonDelete(pJson);
  *pMsg = pCont;
  pJson = NULL;
_exit:
  tjsonDelete(pJson);
  TAOS_RETURN(code);
}

int32_t dumpConfToDataBlock(SSDataBlock* pBlock, int32_t startCol, char* likePattern) {
  int32_t  code = 0;
  SConfig* pConf = taosGetCfg();
  if (pConf == NULL) {
    return TSDB_CODE_INVALID_CFG;
  }

  int32_t      numOfRows = 0;
  int32_t      col = startCol;
  SConfigItem* pItem = NULL;
  SConfigIter* pIter = NULL;

  int8_t locked = 0;

  size_t       exSize = 0;
  size_t       index = 0;
  SConfigItem* pDataDirItem = cfgGetItem(pConf, "dataDir");
  if (pDataDirItem) {
    exSize = TMAX(taosArrayGetSize(pDataDirItem->array), 1) - 1;
  }

  TAOS_CHECK_GOTO(blockDataEnsureCapacity(pBlock, cfgGetSize(pConf) + exSize), NULL, _exit);

  TAOS_CHECK_GOTO(cfgCreateIter(pConf, &pIter), NULL, _exit);

  cfgLock(pConf);
  locked = 1;

  while ((pItem = cfgNextIter(pIter)) != NULL) {
  _start:
    col = startCol;

    // GRANT_CFG_SKIP;
    char name[TSDB_CONFIG_OPTION_LEN + VARSTR_HEADER_SIZE] = {0};
    if (likePattern && rawStrPatternMatch(pItem->name, likePattern) != TSDB_PATTERN_MATCH) {
      continue;
    }
    STR_WITH_MAXSIZE_TO_VARSTR(name, pItem->name, TSDB_CONFIG_OPTION_LEN + VARSTR_HEADER_SIZE);

    SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, col++);
    if (pColInfo == NULL) {
      code = terrno;
      TAOS_CHECK_GOTO(code, NULL, _exit);
    }

    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, numOfRows, name, false), NULL, _exit);

    char      value[TSDB_CONFIG_PATH_LEN + VARSTR_HEADER_SIZE] = {0};
    int32_t   valueLen = 0;
    SDiskCfg* pDiskCfg = NULL;
    if (strcasecmp(pItem->name, "dataDir") == 0 && exSize > 0) {
      char* buf = &value[VARSTR_HEADER_SIZE];
      pDiskCfg = taosArrayGet(pItem->array, index);
      valueLen = tsnprintf(buf, TSDB_CONFIG_PATH_LEN, "%s", pDiskCfg->dir);
      index++;
    } else {
      TAOS_CHECK_GOTO(cfgDumpItemValue(pItem, &value[VARSTR_HEADER_SIZE], TSDB_CONFIG_PATH_LEN, &valueLen), NULL,
                      _exit);
    }
    varDataSetLen(value, valueLen);

    pColInfo = taosArrayGet(pBlock->pDataBlock, col++);
    if (pColInfo == NULL) {
      code = terrno;
      TAOS_CHECK_GOTO(code, NULL, _exit);
    }

    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, numOfRows, value, false), NULL, _exit);

    char scope[TSDB_CONFIG_SCOPE_LEN + VARSTR_HEADER_SIZE] = {0};
    TAOS_CHECK_GOTO(cfgDumpItemScope(pItem, &scope[VARSTR_HEADER_SIZE], TSDB_CONFIG_SCOPE_LEN, &valueLen), NULL, _exit);
    varDataSetLen(scope, valueLen);

    pColInfo = taosArrayGet(pBlock->pDataBlock, col++);
    if (pColInfo == NULL) {
      code = terrno;
      TAOS_CHECK_GOTO(code, NULL, _exit);
    }
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, numOfRows, scope, false), NULL, _exit);

    char category[TSDB_CONFIG_CATEGORY_LEN + VARSTR_HEADER_SIZE] = {0};
    TAOS_CHECK_GOTO(cfgDumpItemCategory(pItem, &category[VARSTR_HEADER_SIZE], TSDB_CONFIG_CATEGORY_LEN, &valueLen),
                    NULL, _exit);
    varDataSetLen(category, valueLen);

    pColInfo = taosArrayGet(pBlock->pDataBlock, col++);
    if (pColInfo == NULL) {
      code = terrno;
      TAOS_CHECK_GOTO(code, NULL, _exit);
    }
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, numOfRows, category, false), NULL, _exit);

    char info[TSDB_CONFIG_INFO_LEN + VARSTR_HEADER_SIZE] = {0};
    if (strcasecmp(pItem->name, "dataDir") == 0 && pDiskCfg) {
      char* buf = &info[VARSTR_HEADER_SIZE];
      valueLen = tsnprintf(buf, TSDB_CONFIG_INFO_LEN, "level %d primary %d disabled %" PRIi8, pDiskCfg->level,
                           pDiskCfg->primary, pDiskCfg->disable);
    } else {
      valueLen = 0;
    }
    varDataSetLen(info, valueLen);

    pColInfo = taosArrayGet(pBlock->pDataBlock, col++);
    if (pColInfo == NULL) {
      code = terrno;
      TAOS_CHECK_GOTO(code, NULL, _exit);
    }
    TAOS_CHECK_GOTO(colDataSetVal(pColInfo, numOfRows, info, false), NULL, _exit);

    numOfRows++;
    if (index > 0 && index <= exSize) {
      goto _start;
    }
  }
  pBlock->info.rows = numOfRows;
_exit:
  if (locked) cfgUnLock(pConf);
  cfgDestroyIter(pIter);
  TAOS_RETURN(code);
}

1	/*
2	* Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
3	*
4	* This program is free software: you can use, redistribute, and/or modify
5	* it under the terms of the GNU Affero General Public License, version 3
6	* or later ("AGPL"), as published by the Free Software Foundation.
7	*
8	* This program is distributed in the hope that it will be useful, but WITHOUT
9	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
10	* FITNESS FOR A PARTICULAR PURPOSE.
11	*
12	* You should have received a copy of the GNU Affero General Public License
13	* along with this program. If not, see <http://www.gnu.org/licenses/>.
14	*/
15
16	#define _DEFAULT_SOURCE
17	#include "tdatablock.h"
18	#include "tglobal.h"
19	#include "tjson.h"
20	#include "tmisce.h"
21	#include "tcompare.h"
22
23	int32_t taosGetIpv4FromEp(const char* ep, SEp* pEp) {	122,393✔
24	pEp->port = 0;	122,393✔
25	memset(pEp->fqdn, 0, TSDB_FQDN_LEN);	122,393✔
26	tstrncpy(pEp->fqdn, ep, TSDB_FQDN_LEN);	122,393✔
27
28	char* temp = strrchr(pEp->fqdn, ':');	122,393✔
29	if (temp) {	122,393✔
30	*temp = 0;	106,951✔
31	pEp->port = taosStr2UInt16(temp + 1, NULL, 10);	106,951✔
32	if (pEp->port < 0) {
33	return TSDB_CODE_INVALID_PARA;
34	}
35	}
36
37	if (pEp->port == 0) {	122,401✔
38	pEp->port = tsServerPort;	15,409✔
39	}
40
41	if (pEp->port <= 0) {	122,401!
42	return TSDB_CODE_INVALID_PARA;	×
43	}
44	return 0;	122,401✔
45	}
46
47	static int8_t isValidPort(const char* str) {	4✔
48	if (!str \|\| !*str) return 0;	4!
49	for (const char* p = str; *p; ++p) {	17✔
50	if (p < '0' \|\| p > '9') {	13!
51	return 0;	×
52	}
53	}
54	return 1;	4✔
55	}
56
57	static int32_t taosGetDualIpFromEp(const char* ep, SEp* pEp) {	8✔
58	memset(pEp->fqdn, 0, TSDB_FQDN_LEN);	8✔
59	pEp->port = 0;	8✔
60	char buf[TSDB_FQDN_LEN] = {0};	8✔
61
62	if (ep[0] == '[') {	8✔
63	// [IPv6]:port format
64	const char* end = strchr(ep, ']');	2✔
65	if (!end) return TSDB_CODE_INVALID_PARA;	2!
66
67	int ipLen = end - ep - 1;	2✔
68	if (ipLen >= TSDB_FQDN_LEN) ipLen = TSDB_FQDN_LEN - 1;	2!
69
70	strncpy(pEp->fqdn, ep + 1, ipLen);	2✔
71	pEp->fqdn[ipLen] = '\0';	2✔
72
73	if ((end + 1) == ':' && (end + 2)) {	2!
74	pEp->port = taosStr2UInt16(end + 2, NULL, 10);	1✔
75	}
76	} else {
77	// Compatible with ::1:6030, ::1, IPv4:port, hostname:port, etc.
78	strncpy(buf, ep, TSDB_FQDN_LEN - 1);	6✔
79	buf[TSDB_FQDN_LEN - 1] = 0;	6✔
80
81	char* lastColon = strrchr(buf, ':');	6✔
82	char* firstColon = strchr(buf, ':');	6✔
83
84	if (lastColon && firstColon != lastColon) {	9✔
85	// Multiple colons, possibly IPv6:port or pure IPv6
86	char* portStr = lastColon + 1;	3✔
87	if (isValidPort(portStr) && lastColon != buf && (*(lastColon - 1) != ':')) {	3!
88	*lastColon = 0;	2✔
89	tstrncpy(pEp->fqdn, buf, TSDB_FQDN_LEN);	2✔
90	pEp->port = taosStr2UInt16(portStr, NULL, 10);	2✔
91	} else {
92	tstrncpy(pEp->fqdn, ep, TSDB_FQDN_LEN);	1✔
93	}
94	} else if (lastColon) {	3✔
95	// Only one colon, IPv4:port or hostname:port
96	char* portStr = lastColon + 1;	1✔
97	if (isValidPort(portStr) && lastColon != buf) {	1!
98	*lastColon = 0;	1✔
99	tstrncpy(pEp->fqdn, buf, TSDB_FQDN_LEN);	1✔
100	pEp->port = taosStr2UInt16(portStr, NULL, 10);	1✔
101	} else {
102	tstrncpy(pEp->fqdn, ep, TSDB_FQDN_LEN);	×
103	}
104	} else {
105	// No colon, pure hostname or IPv6 without port
106	tstrncpy(pEp->fqdn, ep, TSDB_FQDN_LEN);	2✔
107	}
108	}
109
110	if (pEp->port == 0) {	8✔
111	pEp->port = tsServerPort;	4✔
112	}
113
114	if (pEp->port <= 0) {	8!
115	return TSDB_CODE_INVALID_PARA;	×
116	}
117
118	return 0;	8✔
119	}
120	int32_t taosGetFqdnPortFromEp(const char* ep, SEp* pEp) {	122,438✔
121	if (tsEnableIpv6) {	122,438✔
122	return taosGetDualIpFromEp(ep, pEp);	8✔
123	} else {
124	return taosGetIpv4FromEp(ep, pEp);	122,430✔
125	}
126	}
127
128	int32_t addEpIntoEpSet(SEpSet* pEpSet, const char* fqdn, uint16_t port) {	351,511✔
129	if (pEpSet == NULL \|\| fqdn == NULL \|\| strlen(fqdn) == 0) {	351,511!
130	return TSDB_CODE_INVALID_PARA;	×
131	}
132
133	int32_t index = pEpSet->numOfEps;	351,511✔
134	if (index >= sizeof(pEpSet->eps) / sizeof(pEpSet->eps[0])) {	351,511!
135	return TSDB_CODE_OUT_OF_RANGE;	×
136	}
137	tstrncpy(pEpSet->eps[index].fqdn, fqdn, tListLen(pEpSet->eps[index].fqdn));	351,511✔
138	pEpSet->eps[index].port = port;	351,511✔
139	pEpSet->numOfEps += 1;	351,511✔
140	return 0;	351,511✔
141	}
142
143	bool isEpsetEqual(const SEpSet* s1, const SEpSet* s2) {	200,610✔
144	if (s1->numOfEps != s2->numOfEps \|\| s1->inUse != s2->inUse) {	200,610✔
145	return false;	16,244✔
146	}
147
148	for (int32_t i = 0; i < s1->numOfEps; i++) {	375,550✔
149	if (s1->eps[i].port != s2->eps[i].port \|\| strncmp(s1->eps[i].fqdn, s2->eps[i].fqdn, TSDB_FQDN_LEN) != 0)	191,441✔
150	return false;	257✔
151	}
152	return true;	184,109✔
153	}
154
155	void epsetAssign(SEpSet* pDst, const SEpSet* pSrc) {	157,699✔
156	if (pSrc == NULL \|\| pDst == NULL) {	157,699!
UNCOV 157	return;	×
158	}
159
160	pDst->inUse = pSrc->inUse;	157,703✔
161	pDst->numOfEps = pSrc->numOfEps;	157,703✔
162	for (int32_t i = 0; i < pSrc->numOfEps; ++i) {	319,285✔
163	pDst->eps[i].port = pSrc->eps[i].port;	161,582✔
164	tstrncpy(pDst->eps[i].fqdn, pSrc->eps[i].fqdn, tListLen(pSrc->eps[i].fqdn));	161,582✔
165	}
166	}
167
168	void epAssign(SEp* pDst, SEp* pSrc) {	221,581✔
169	if (pSrc == NULL \|\| pDst == NULL) {	221,581!
170	return;	×
171	}
172	memset(pDst->fqdn, 0, tListLen(pSrc->fqdn));	221,584✔
173	tstrncpy(pDst->fqdn, pSrc->fqdn, tListLen(pSrc->fqdn));	221,584✔
174	pDst->port = pSrc->port;	221,584✔
175	}
176
177	void epsetSort(SEpSet* pDst) {	992,254✔
178	if (pDst->numOfEps <= 1) {	992,254✔
179	return;	904,561✔
180	}
181	int validIdx = false;	87,693✔
182	SEp ep = {0};	87,693✔
183	if (pDst->inUse >= 0 && pDst->inUse < pDst->numOfEps) {	87,693!
184	validIdx = true;	87,735✔
185	epAssign(&ep, &pDst->eps[pDst->inUse]);	87,735✔
186	}
187
188	for (int i = 0; i < pDst->numOfEps - 1; i++) {	251,921✔
189	for (int j = 0; j < pDst->numOfEps - 1 - i; j++) {	413,380✔
190	SEp* f = &pDst->eps[j];	249,199✔
191	SEp* s = &pDst->eps[j + 1];	249,199✔
192	int cmp = strncmp(f->fqdn, s->fqdn, sizeof(f->fqdn));	249,199✔
193	if (cmp > 0 \|\| (cmp == 0 && f->port > s->port)) {	249,199!
194	SEp ep1 = {0};	44,601✔
195	epAssign(&ep1, f);	44,601✔
196	epAssign(f, s);	44,617✔
197	epAssign(s, &ep1);	44,617✔
198	}
199	}
200	}
201	if (validIdx == true)	87,756✔
202	for (int i = 0; i < pDst->numOfEps; i++) {	166,877!
203	int cmp = strncmp(ep.fqdn, pDst->eps[i].fqdn, sizeof(ep.fqdn));	166,880✔
204	if (cmp == 0 && ep.port == pDst->eps[i].port) {	166,880✔
205	pDst->inUse = i;	87,742✔
206	break;	87,742✔
207	}
208	}
209	}
210
211	void updateEpSet_s(SCorEpSet* pEpSet, SEpSet* pNewEpSet) {	16,267✔
212	taosCorBeginWrite(&pEpSet->version);	16,276!
213	pEpSet->epSet = *pNewEpSet;	16,266✔
214	taosCorEndWrite(&pEpSet->version);	16,266✔
215	}	16,267✔
216
217	SEpSet getEpSet_s(SCorEpSet* pEpSet) {	56,439,607✔
218	SEpSet ep = {0};	56,439,607✔
219	taosCorBeginRead(&pEpSet->version);	56,433,763!
220	ep = pEpSet->epSet;	57,093,263✔
221	taosCorEndRead(&pEpSet->version);	57,093,263!
222
223	return ep;	57,109,371✔
224	}
225
226	int32_t epsetToStr(const SEpSet* pEpSet, char* pBuf, int32_t cap) {	208,237✔
227	int32_t ret = 0;	208,237✔
228	int32_t nwrite = 0;	208,237✔
229
230	nwrite = snprintf(pBuf + nwrite, cap, "epset:{");	208,237✔
231	if (nwrite <= 0 \|\| nwrite >= cap) {	208,237!
232	return TSDB_CODE_OUT_OF_BUFFER;	×
233	}
234	cap -= nwrite;	208,237✔
235
236	for (int _i = 0; (_i < pEpSet->numOfEps) && (cap > 0); _i++) {	582,793!
237	if (_i == pEpSet->numOfEps - 1) {	374,557✔
238	ret = snprintf(pBuf + nwrite, cap, "%d. %s:%d", _i, pEpSet->eps[_i].fqdn, pEpSet->eps[_i].port);	208,237✔
239	} else {
240	ret = snprintf(pBuf + nwrite, cap, "%d. %s:%d, ", _i, pEpSet->eps[_i].fqdn, pEpSet->eps[_i].port);	166,320✔
241	}
242
243	if (ret <= 0 \|\| ret >= cap) {	374,557!
244	return TSDB_CODE_OUT_OF_BUFFER;	1✔
245	}
246
247	nwrite += ret;	374,556✔
248	cap -= ret;	374,556✔
249	}
250
251	if (cap <= 0) {	208,236!
252	return TSDB_CODE_OUT_OF_BUFFER;	×
253	}
254
255	ret = snprintf(pBuf + nwrite, cap, "}, inUse:%d", pEpSet->inUse);	208,236✔
256	if (ret <= 0 \|\| ret >= cap) {	208,236!
257	return TSDB_CODE_OUT_OF_BUFFER;	×
258	} else {
259	return TSDB_CODE_SUCCESS;	208,237✔
260	}
261	}
262
263	int32_t taosGenCrashJsonMsg(int signum, char** pMsg, int64_t clusterId, int64_t startTime) {	×
264	int32_t code = 0;	×
265	SJson* pJson = tjsonCreateObject();	×
266	if (pJson == NULL) return terrno;	×
267
268	char tmp[4096] = {0};	×
269
270	TAOS_CHECK_GOTO(tjsonAddDoubleToObject(pJson, "reportVersion", 1), NULL, _exit);	×
271
272	TAOS_CHECK_GOTO(tjsonAddIntegerToObject(pJson, "clusterId", clusterId), NULL, _exit);	×
273	TAOS_CHECK_GOTO(tjsonAddIntegerToObject(pJson, "startTime", startTime), NULL, _exit);	×
274
275	// Do NOT invoke the taosGetFqdn here.
276	// this function may be invoked when memory exception occurs,so we should assume that it is running in a memory locked
277	// environment. The lock operation by taosGetFqdn may cause this program deadlock.
278	TAOS_CHECK_GOTO(tjsonAddStringToObject(pJson, "fqdn", tsLocalFqdn), NULL, _exit);	×
279
280	TAOS_CHECK_GOTO(tjsonAddIntegerToObject(pJson, "pid", taosGetPId()), NULL, _exit);	×
281
282	code = taosGetAppName(tmp, NULL);	×
283	if (code != 0) {	×
284	TAOS_CHECK_GOTO(code, NULL, _exit);	×
285	}
286	TAOS_CHECK_GOTO(tjsonAddStringToObject(pJson, "appName", tmp), NULL, _exit);	×
287
288	if (taosGetOsReleaseName(tmp, NULL, NULL, sizeof(tmp)) == 0) {	×
289	TAOS_CHECK_GOTO(tjsonAddStringToObject(pJson, "os", tmp), NULL, _exit);	×
290	} else {
291	// do nothing
292	}
293
294	float numOfCores = 0;	×
295	if (taosGetCpuInfo(tmp, sizeof(tmp), &numOfCores) == 0) {	×
296	TAOS_CHECK_GOTO(tjsonAddStringToObject(pJson, "cpuModel", tmp), NULL, _exit);	×
297	TAOS_CHECK_GOTO(tjsonAddDoubleToObject(pJson, "numOfCpu", numOfCores), NULL, _exit);	×
298	} else {
299	TAOS_CHECK_GOTO(tjsonAddDoubleToObject(pJson, "numOfCpu", tsNumOfCores), NULL, _exit);	×
300	}
301
302	int32_t nBytes = snprintf(tmp, sizeof(tmp), "%" PRId64 " kB", tsTotalMemoryKB);	×
303	if (nBytes <= 9 \|\| nBytes >= sizeof(tmp)) {	×
304	TAOS_CHECK_GOTO(TSDB_CODE_OUT_OF_RANGE, NULL, _exit);	×
305	}
306	TAOS_CHECK_GOTO(tjsonAddStringToObject(pJson, "memory", tmp), NULL, _exit);	×
307
308	TAOS_CHECK_GOTO(tjsonAddStringToObject(pJson, "version", td_version), NULL, _exit);	×
309	TAOS_CHECK_GOTO(tjsonAddStringToObject(pJson, "buildInfo", td_buildinfo), NULL, _exit);	×
310	TAOS_CHECK_GOTO(tjsonAddStringToObject(pJson, "gitInfo", td_gitinfo), NULL, _exit);	×
311
312	TAOS_CHECK_GOTO(tjsonAddIntegerToObject(pJson, "crashSig", signum), NULL, _exit);	×
313	TAOS_CHECK_GOTO(tjsonAddIntegerToObject(pJson, "crashTs", taosGetTimestampUs()), NULL, _exit);	×
314
315	#if 0
316	#ifdef _TD_DARWIN_64
317	taosLogTraceToBuf(tmp, sizeof(tmp), 4);
318	#elif !defined(WINDOWS)
319	taosLogTraceToBuf(tmp, sizeof(tmp), 3);
320	#else
321	taosLogTraceToBuf(tmp, sizeof(tmp), 8);
322	#endif
323
324	TAOS_CHECK_GOTO(tjsonAddStringToObject(pJson, "stackInfo", tmp), NULL, _exit);
325	#endif
326	char* pCont = tjsonToString(pJson);	×
327	if (pCont == NULL) {	×
328	code = terrno;	×
329	TAOS_CHECK_GOTO(code, NULL, _exit);	×
330	goto _exit;	×
331	}
332
333	tjsonDelete(pJson);	×
334	*pMsg = pCont;	×
335	pJson = NULL;	×
336	_exit:	×
337	tjsonDelete(pJson);	×
338	TAOS_RETURN(code);	×
339	}
340
341	int32_t dumpConfToDataBlock(SSDataBlock* pBlock, int32_t startCol, char* likePattern) {	9,412✔
342	int32_t code = 0;	9,412✔
343	SConfig* pConf = taosGetCfg();	9,412✔
344	if (pConf == NULL) {	9,412!
345	return TSDB_CODE_INVALID_CFG;	×
346	}
347
348	int32_t numOfRows = 0;	9,412✔
349	int32_t col = startCol;	9,412✔
350	SConfigItem* pItem = NULL;	9,412✔
351	SConfigIter* pIter = NULL;	9,412✔
352
353	int8_t locked = 0;	9,412✔
354
355	size_t exSize = 0;	9,412✔
356	size_t index = 0;	9,412✔
357	SConfigItem* pDataDirItem = cfgGetItem(pConf, "dataDir");	9,412✔
358	if (pDataDirItem) {	9,412✔
359	exSize = TMAX(taosArrayGetSize(pDataDirItem->array), 1) - 1;	8,840✔
360	}
361
362	TAOS_CHECK_GOTO(blockDataEnsureCapacity(pBlock, cfgGetSize(pConf) + exSize), NULL, _exit);	9,412!
363
364	TAOS_CHECK_GOTO(cfgCreateIter(pConf, &pIter), NULL, _exit);	9,412!
365
366	cfgLock(pConf);	9,412✔
367	locked = 1;	9,412✔
368
369	while ((pItem = cfgNextIter(pIter)) != NULL) {	2,092,948✔
370	_start:	2,083,536✔
371	col = startCol;	2,083,536✔
372
373	// GRANT_CFG_SKIP;
374	char name[TSDB_CONFIG_OPTION_LEN + VARSTR_HEADER_SIZE] = {0};	2,083,536✔
375	if (likePattern && rawStrPatternMatch(pItem->name, likePattern) != TSDB_PATTERN_MATCH) {	2,083,536✔
376	continue;	600✔
377	}
378	STR_WITH_MAXSIZE_TO_VARSTR(name, pItem->name, TSDB_CONFIG_OPTION_LEN + VARSTR_HEADER_SIZE);	2,082,936✔
379
380	SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, col++);	2,082,936✔
381	if (pColInfo == NULL) {	2,082,936!
382	code = terrno;	×
383	TAOS_CHECK_GOTO(code, NULL, _exit);	×
384	}
385
386	TAOS_CHECK_GOTO(colDataSetVal(pColInfo, numOfRows, name, false), NULL, _exit);	2,082,936!
387
388	char value[TSDB_CONFIG_PATH_LEN + VARSTR_HEADER_SIZE] = {0};	2,082,936✔
389	int32_t valueLen = 0;	2,082,936✔
390	SDiskCfg* pDiskCfg = NULL;	2,082,936✔
391	if (strcasecmp(pItem->name, "dataDir") == 0 && exSize > 0) {	2,082,936!
392	char* buf = &value[VARSTR_HEADER_SIZE];	×
393	pDiskCfg = taosArrayGet(pItem->array, index);	×
394	valueLen = tsnprintf(buf, TSDB_CONFIG_PATH_LEN, "%s", pDiskCfg->dir);	×
395	index++;	×
396	} else {
397	TAOS_CHECK_GOTO(cfgDumpItemValue(pItem, &value[VARSTR_HEADER_SIZE], TSDB_CONFIG_PATH_LEN, &valueLen), NULL,	2,082,936!
398	_exit);
399	}
400	varDataSetLen(value, valueLen);	2,082,936✔
401
402	pColInfo = taosArrayGet(pBlock->pDataBlock, col++);	2,082,936✔
403	if (pColInfo == NULL) {	2,082,936!
404	code = terrno;	×
405	TAOS_CHECK_GOTO(code, NULL, _exit);	×
406	}
407
408	TAOS_CHECK_GOTO(colDataSetVal(pColInfo, numOfRows, value, false), NULL, _exit);	2,082,936!
409
410	char scope[TSDB_CONFIG_SCOPE_LEN + VARSTR_HEADER_SIZE] = {0};	2,082,936✔
411	TAOS_CHECK_GOTO(cfgDumpItemScope(pItem, &scope[VARSTR_HEADER_SIZE], TSDB_CONFIG_SCOPE_LEN, &valueLen), NULL, _exit);	2,082,936!
412	varDataSetLen(scope, valueLen);	2,082,936✔
413
414	pColInfo = taosArrayGet(pBlock->pDataBlock, col++);	2,082,936✔
415	if (pColInfo == NULL) {	2,082,936!
416	code = terrno;	×
417	TAOS_CHECK_GOTO(code, NULL, _exit);	×
418	}
419	TAOS_CHECK_GOTO(colDataSetVal(pColInfo, numOfRows, scope, false), NULL, _exit);	2,082,936!
420
421	char category[TSDB_CONFIG_CATEGORY_LEN + VARSTR_HEADER_SIZE] = {0};	2,082,936✔
422	TAOS_CHECK_GOTO(cfgDumpItemCategory(pItem, &category[VARSTR_HEADER_SIZE], TSDB_CONFIG_CATEGORY_LEN, &valueLen),	2,082,936!
423	NULL, _exit);
424	varDataSetLen(category, valueLen);	2,082,936✔
425
426	pColInfo = taosArrayGet(pBlock->pDataBlock, col++);	2,082,936✔
427	if (pColInfo == NULL) {	2,082,936!
428	code = terrno;	×
429	TAOS_CHECK_GOTO(code, NULL, _exit);	×
430	}
431	TAOS_CHECK_GOTO(colDataSetVal(pColInfo, numOfRows, category, false), NULL, _exit);	2,082,936!
432
433	char info[TSDB_CONFIG_INFO_LEN + VARSTR_HEADER_SIZE] = {0};	2,082,936✔
434	if (strcasecmp(pItem->name, "dataDir") == 0 && pDiskCfg) {	2,082,936!
435	char* buf = &info[VARSTR_HEADER_SIZE];	×
436	valueLen = tsnprintf(buf, TSDB_CONFIG_INFO_LEN, "level %d primary %d disabled %" PRIi8, pDiskCfg->level,	×
437	pDiskCfg->primary, pDiskCfg->disable);	×
438	} else {
439	valueLen = 0;	2,082,936✔
440	}
441	varDataSetLen(info, valueLen);	2,082,936✔
442
443	pColInfo = taosArrayGet(pBlock->pDataBlock, col++);	2,082,936✔
444	if (pColInfo == NULL) {	2,082,936!
445	code = terrno;	×
446	TAOS_CHECK_GOTO(code, NULL, _exit);	×
447	}
448	TAOS_CHECK_GOTO(colDataSetVal(pColInfo, numOfRows, info, false), NULL, _exit);	2,082,936!
449
450	numOfRows++;	2,082,936✔
451	if (index > 0 && index <= exSize) {	2,082,936!
452	goto _start;	×
453	}
454	}
455	pBlock->info.rows = numOfRows;	9,412✔
456	_exit:	9,412✔
457	if (locked) cfgUnLock(pConf);	9,412!
458	cfgDestroyIter(pIter);	9,412✔
459	TAOS_RETURN(code);	9,412✔
460	}

taosdata / TDengine / #4476

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous