• Home
  • Features
  • Pricing
  • Docs
  • Announcements
  • Sign In

taosdata / TDengine / #4867

26 Nov 2025 05:46AM UTC coverage: 64.473% (-0.03%) from 64.504%
#4867

push

travis-ci

guanshengliang
Merge branch '3.0' into cover/3.0

765 of 945 new or added lines in 33 files covered. (80.95%)

3147 existing lines in 126 files now uncovered.

158042 of 245129 relevant lines covered (64.47%)

111495961.54 hits per line

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

92.83
/source/libs/parser/src/parAuthenticator.c
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
#include "catalog.h"
17
#include "cmdnodes.h"
18
#include "parInt.h"
19

20
typedef struct SAuthCxt {
21
  SParseContext*   pParseCxt;
22
  SParseMetaCache* pMetaCache;
23
  int32_t          errCode;
24
} SAuthCxt;
25

26
typedef struct SSelectAuthCxt {
27
  SAuthCxt*    pAuthCxt;
28
  SSelectStmt* pSelect;
29
} SSelectAuthCxt;
30

31
typedef struct SAuthRewriteCxt {
32
  STableNode* pTarget;
33
} SAuthRewriteCxt;
34

35
static int32_t authQuery(SAuthCxt* pCxt, SNode* pStmt);
36

37
static int32_t setUserAuthInfo(SParseContext* pCxt, const char* pDbName, const char* pTabName, AUTH_TYPE type,
1,066,337✔
38
                               bool isView, bool effective, SUserAuthInfo* pAuth) {
39
  if (effective) {
1,066,337✔
40
    snprintf(pAuth->user, sizeof(pAuth->user), "%s", pCxt->pEffectiveUser ? pCxt->pEffectiveUser : "");
31,513✔
41
  } else {
42
    snprintf(pAuth->user, sizeof(pAuth->user), "%s", pCxt->pUser);
1,034,824✔
43
  }
44

45
  if (NULL == pTabName) {
1,066,337✔
46
    int32_t code = tNameSetDbName(&pAuth->tbName, pCxt->acctId, pDbName, strlen(pDbName));
106,334✔
47
    if (TSDB_CODE_SUCCESS != code) return code;
106,334✔
48
  } else {
49
    toName(pCxt->acctId, pDbName, pTabName, &pAuth->tbName);
960,003✔
50
  }
51
  pAuth->type = type;
1,066,337✔
52
  pAuth->isView = isView;
1,066,337✔
53
  return TSDB_CODE_SUCCESS;
1,066,337✔
54
}
55

56
static int32_t checkAuthImpl(SAuthCxt* pCxt, const char* pDbName, const char* pTabName, AUTH_TYPE type, SNode** pCond,
173,956,032✔
57
                             bool isView, bool effective) {
58
  SParseContext* pParseCxt = pCxt->pParseCxt;
173,956,032✔
59
  if (pParseCxt->isSuperUser) {
173,956,921✔
60
    return TSDB_CODE_SUCCESS;
172,893,891✔
61
  }
62

63
  AUTH_RES_TYPE auth_res_type = isView ? AUTH_RES_VIEW : AUTH_RES_BASIC;
1,066,337✔
64
  SUserAuthInfo authInfo = {0};
1,066,337✔
65
  int32_t       code = setUserAuthInfo(pCxt->pParseCxt, pDbName, pTabName, type, isView, effective, &authInfo);
1,066,337✔
66
  if (TSDB_CODE_SUCCESS != code) return code;
1,066,337✔
67
  SUserAuthRes authRes = {0};
1,066,337✔
68
  if (NULL != pCxt->pMetaCache) {
1,066,337✔
69
    code = getUserAuthFromCache(pCxt->pMetaCache, &authInfo, &authRes);
1,066,337✔
70
#ifdef TD_ENTERPRISE
71
    if (isView && TSDB_CODE_PAR_INTERNAL_ERROR == code) {
1,066,337✔
72
      authInfo.isView = false;
48,919✔
73
      code = getUserAuthFromCache(pCxt->pMetaCache, &authInfo, &authRes);
48,919✔
74
    }
75
#endif
76
  } else {
77
    SRequestConnInfo conn = {.pTrans = pParseCxt->pTransporter,
×
78
                             .requestId = pParseCxt->requestId,
×
79
                             .requestObjRefId = pParseCxt->requestRid,
×
80
                             .mgmtEps = pParseCxt->mgmtEpSet};
81
    code = catalogChkAuth(pParseCxt->pCatalog, &conn, &authInfo, &authRes);
×
82
  }
83
  if (TSDB_CODE_SUCCESS == code && NULL != pCond) {
1,066,337✔
84
    *pCond = authRes.pCond[auth_res_type];
290,051✔
85
  }
86
  return TSDB_CODE_SUCCESS == code ? (authRes.pass[auth_res_type] ? TSDB_CODE_SUCCESS : TSDB_CODE_PAR_PERMISSION_DENIED)
1,066,337✔
87
                                   : code;
2,132,674✔
88
}
89

90
static int32_t checkAuth(SAuthCxt* pCxt, const char* pDbName, const char* pTabName, AUTH_TYPE type, SNode** pCond) {
173,530,219✔
91
  return checkAuthImpl(pCxt, pDbName, pTabName, type, pCond, false, false);
173,530,219✔
92
}
93

94
static int32_t checkEffectiveAuth(SAuthCxt* pCxt, const char* pDbName, const char* pTabName, AUTH_TYPE type,
20,545✔
95
                                  SNode** pCond) {
96
  return checkAuthImpl(pCxt, pDbName, pTabName, type, NULL, false, true);
20,545✔
97
}
98

99
static int32_t checkViewAuth(SAuthCxt* pCxt, const char* pDbName, const char* pTabName, AUTH_TYPE type, SNode** pCond) {
417,920✔
100
  return checkAuthImpl(pCxt, pDbName, pTabName, type, NULL, true, false);
417,920✔
101
}
102

103
static int32_t checkViewEffectiveAuth(SAuthCxt* pCxt, const char* pDbName, const char* pTabName, AUTH_TYPE type,
10,968✔
104
                                      SNode** pCond) {
105
  return checkAuthImpl(pCxt, pDbName, pTabName, type, NULL, true, true);
10,968✔
106
}
107

108
static EDealRes authSubquery(SAuthCxt* pCxt, SNode* pStmt) {
9,008,378✔
109
  return TSDB_CODE_SUCCESS == authQuery(pCxt, pStmt) ? DEAL_RES_CONTINUE : DEAL_RES_ERROR;
9,008,378✔
110
}
111

112
static int32_t mergeStableTagCond(SNode** pWhere, SNode* pTagCond) {
338✔
113
  SLogicConditionNode* pLogicCond = NULL;
338✔
114
  int32_t              code = nodesMakeNode(QUERY_NODE_LOGIC_CONDITION, (SNode**)&pLogicCond);
338✔
115
  if (NULL == pLogicCond) {
338✔
116
    return code;
×
117
  }
118
  pLogicCond->node.resType.type = TSDB_DATA_TYPE_BOOL;
338✔
119
  pLogicCond->node.resType.bytes = tDataTypes[TSDB_DATA_TYPE_BOOL].bytes;
338✔
120
  pLogicCond->condType = LOGIC_COND_TYPE_AND;
338✔
121
  code = nodesListMakeStrictAppend(&pLogicCond->pParameterList, pTagCond);
338✔
122
  if (TSDB_CODE_SUCCESS == code) {
338✔
123
    code = nodesListMakeAppend(&pLogicCond->pParameterList, *pWhere);
338✔
124
  }
125
  if (TSDB_CODE_SUCCESS == code) {
338✔
126
    *pWhere = (SNode*)pLogicCond;
338✔
127
  } else {
128
    nodesDestroyNode((SNode*)pLogicCond);
×
129
  }
130
  return code;
338✔
131
}
132

133
EDealRes rewriteAuthTable(SNode* pNode, void* pContext) {
56,262✔
134
  if (QUERY_NODE_COLUMN == nodeType(pNode)) {
56,262✔
135
    SColumnNode*     pCol = (SColumnNode*)pNode;
18,416✔
136
    SAuthRewriteCxt* pCxt = (SAuthRewriteCxt*)pContext;
18,416✔
137
    tstrncpy(pCol->tableName, pCxt->pTarget->tableName, TSDB_TABLE_NAME_LEN);
18,416✔
138
    tstrncpy(pCol->tableAlias, pCxt->pTarget->tableAlias, TSDB_TABLE_NAME_LEN);
18,416✔
139
  }
140

141
  return DEAL_RES_CONTINUE;
56,262✔
142
}
143

144
static int32_t rewriteAppendStableTagCond(SNode** pWhere, SNode* pTagCond, STableNode* pTable) {
17,402✔
145
  SNode*  pTagCondCopy = NULL;
17,402✔
146
  int32_t code = nodesCloneNode(pTagCond, &pTagCondCopy);
17,402✔
147
  if (NULL == pTagCondCopy) {
17,402✔
148
    return code;
×
149
  }
150

151
  SAuthRewriteCxt cxt = {.pTarget = pTable};
17,402✔
152
  nodesWalkExpr(pTagCondCopy, rewriteAuthTable, &cxt);
17,402✔
153

154
  if (NULL == *pWhere) {
17,402✔
155
    *pWhere = pTagCondCopy;
17,064✔
156
    return TSDB_CODE_SUCCESS;
17,064✔
157
  }
158

159
  if (QUERY_NODE_LOGIC_CONDITION == nodeType(*pWhere) &&
338✔
160
      LOGIC_COND_TYPE_AND == ((SLogicConditionNode*)*pWhere)->condType) {
×
161
    return nodesListStrictAppend(((SLogicConditionNode*)*pWhere)->pParameterList, pTagCondCopy);
×
162
  }
163

164
  return mergeStableTagCond(pWhere, pTagCondCopy);
338✔
165
}
166

167
static EDealRes authSelectImpl(SNode* pNode, void* pContext) {
1,081,278,763✔
168
  SSelectAuthCxt* pCxt = pContext;
1,081,278,763✔
169
  SAuthCxt*       pAuthCxt = pCxt->pAuthCxt;
1,081,278,763✔
170
  bool            isView = false;
1,081,279,243✔
171
  if (QUERY_NODE_REAL_TABLE == nodeType(pNode)) {
1,081,279,243✔
172
    SNode*      pTagCond = NULL;
100,763,273✔
173
    STableNode* pTable = (STableNode*)pNode;
100,763,478✔
174
    if ((pAuthCxt->pParseCxt->enableSysInfo == 0) && IS_INFORMATION_SCHEMA_DB(pTable->dbName) &&
100,763,478✔
175
        (strcmp(pTable->tableName, TSDB_INS_TABLE_VGROUPS) == 0)) {
40,454✔
176
      pAuthCxt->errCode = TSDB_CODE_PAR_PERMISSION_DENIED;
1,432✔
177
      return DEAL_RES_ERROR;
1,432✔
178
    }
179
#ifdef TD_ENTERPRISE
180
    SName name = {0};
100,761,871✔
181
    toName(pAuthCxt->pParseCxt->acctId, pTable->dbName, pTable->tableName, &name);
100,762,348✔
182
    STableMeta* pTableMeta = NULL;
100,762,523✔
183
    toName(pAuthCxt->pParseCxt->acctId, pTable->dbName, pTable->tableName, &name);
100,762,523✔
184
    int32_t code = getTargetMetaImpl(pAuthCxt->pParseCxt, pAuthCxt->pMetaCache, &name, &pTableMeta, true);
100,762,071✔
185
    if (TSDB_CODE_SUCCESS == code && TSDB_VIEW_TABLE == pTableMeta->tableType) {
100,761,556✔
186
      isView = true;
232,005✔
187
    }
188
    taosMemoryFree(pTableMeta);
100,761,810✔
189
#endif
190
    if (!isView) {
100,761,872✔
191
      pAuthCxt->errCode = checkAuth(pAuthCxt, pTable->dbName, pTable->tableName, AUTH_TYPE_READ, &pTagCond);
100,529,867✔
192
      if (TSDB_CODE_SUCCESS != pAuthCxt->errCode && NULL != pAuthCxt->pParseCxt->pEffectiveUser) {
100,530,035✔
193
        pAuthCxt->errCode = checkEffectiveAuth(pAuthCxt, pTable->dbName, pTable->tableName, AUTH_TYPE_READ, NULL);
20,545✔
194
      }
195
      if (TSDB_CODE_SUCCESS == pAuthCxt->errCode && NULL != pTagCond) {
100,529,857✔
196
        pAuthCxt->errCode = rewriteAppendStableTagCond(&pCxt->pSelect->pWhere, pTagCond, pTable);
17,402✔
197
      }
198
    } else {
199
      pAuthCxt->errCode = checkViewAuth(pAuthCxt, pTable->dbName, pTable->tableName, AUTH_TYPE_READ, NULL);
232,005✔
200
      if (TSDB_CODE_SUCCESS != pAuthCxt->errCode && NULL != pAuthCxt->pParseCxt->pEffectiveUser) {
232,005✔
201
        pAuthCxt->errCode = checkViewEffectiveAuth(pAuthCxt, pTable->dbName, pTable->tableName, AUTH_TYPE_READ, NULL);
10,968✔
202
      }
203
    }
204
    return TSDB_CODE_SUCCESS == pAuthCxt->errCode ? DEAL_RES_CONTINUE : DEAL_RES_ERROR;
100,763,064✔
205
  } else if (QUERY_NODE_TEMP_TABLE == nodeType(pNode)) {
980,515,952✔
206
    return authSubquery(pAuthCxt, ((STempTableNode*)pNode)->pSubquery);
9,007,898✔
207
  }
208
  return DEAL_RES_CONTINUE;
971,508,054✔
209
}
210

211
static int32_t authSelect(SAuthCxt* pCxt, SSelectStmt* pSelect) {
104,568,944✔
212
  SSelectAuthCxt cxt = {.pAuthCxt = pCxt, .pSelect = pSelect};
104,568,944✔
213
  nodesWalkSelectStmt(pSelect, SQL_CLAUSE_FROM, authSelectImpl, &cxt);
104,568,944✔
214
  return pCxt->errCode;
104,569,379✔
215
}
216

217
static int32_t authSetOperator(SAuthCxt* pCxt, SSetOperator* pSetOper) {
2,938,541✔
218
  int32_t code = authQuery(pCxt, pSetOper->pLeft);
2,938,541✔
219
  if (TSDB_CODE_SUCCESS == code) {
2,938,541✔
220
    code = authQuery(pCxt, pSetOper->pRight);
2,938,541✔
221
  }
222
  return code;
2,938,541✔
223
}
224

225
static int32_t authDropUser(SAuthCxt* pCxt, SDropUserStmt* pStmt) {
76,251✔
226
  if (!pCxt->pParseCxt->isSuperUser || 0 == strcmp(pStmt->userName, TSDB_DEFAULT_USER)) {
76,251✔
227
    return TSDB_CODE_PAR_PERMISSION_DENIED;
1,420✔
228
  }
229
  return TSDB_CODE_SUCCESS;
74,831✔
230
}
231

232
static int32_t authDelete(SAuthCxt* pCxt, SDeleteStmt* pDelete) {
1,788,510✔
233
  SNode*      pTagCond = NULL;
1,788,510✔
234
  STableNode* pTable = (STableNode*)pDelete->pFromTable;
1,788,510✔
235
  int32_t     code = checkAuth(pCxt, pTable->dbName, pTable->tableName, AUTH_TYPE_WRITE, &pTagCond);
1,788,510✔
236
  if (TSDB_CODE_SUCCESS == code && NULL != pTagCond) {
1,788,510✔
UNCOV
237
    code = rewriteAppendStableTagCond(&pDelete->pWhere, pTagCond, pTable);
×
238
  }
239
  return code;
1,788,510✔
240
}
241

242
static int32_t authInsert(SAuthCxt* pCxt, SInsertStmt* pInsert) {
71,188✔
243
  SNode*      pTagCond = NULL;
71,188✔
244
  STableNode* pTable = (STableNode*)pInsert->pTable;
71,188✔
245
  // todo check tag condition for subtable
246
  int32_t code = checkAuth(pCxt, pTable->dbName, pTable->tableName, AUTH_TYPE_WRITE, &pTagCond);
71,188✔
247
  if (TSDB_CODE_SUCCESS == code) {
71,188✔
248
    code = authQuery(pCxt, pInsert->pQuery);
71,188✔
249
  }
250
  return code;
71,188✔
251
}
252

253
static int32_t authShowTables(SAuthCxt* pCxt, SShowStmt* pStmt) {
662,376✔
254
  return checkAuth(pCxt, ((SValueNode*)pStmt->pDbName)->literal, NULL, AUTH_TYPE_READ_OR_WRITE, NULL);
662,376✔
255
}
256

257
static int32_t authShowVtables(SAuthCxt* pCxt, SShowStmt* pStmt) { return authShowTables(pCxt, pStmt); }
60,681✔
258

UNCOV
259
static int32_t authShowUsage(SAuthCxt* pCxt, SShowStmt* pStmt) {
×
UNCOV
260
  return checkAuth(pCxt, ((SValueNode*)pStmt->pDbName)->literal, NULL, AUTH_TYPE_READ_OR_WRITE, NULL);
×
261
}
262

263
static int32_t authShowCreateTable(SAuthCxt* pCxt, SShowCreateTableStmt* pStmt) {
125,271✔
264
  SNode* pTagCond = NULL;
125,271✔
265
  // todo check tag condition for subtable
266
  return checkAuth(pCxt, pStmt->dbName, pStmt->tableName, AUTH_TYPE_READ, &pTagCond);
125,271✔
267
}
268

UNCOV
269
static int32_t authShowCreateView(SAuthCxt* pCxt, SShowCreateViewStmt* pStmt) {
×
270
#ifndef TD_ENTERPRISE
271
  return TSDB_CODE_OPS_NOT_SUPPORT;
272
#endif
273

274
  return TSDB_CODE_SUCCESS;
×
275
}
276

277
static int32_t authCreateTable(SAuthCxt* pCxt, SCreateTableStmt* pStmt) {
4,466,174✔
278
  SNode* pTagCond = NULL;
4,466,174✔
279
  // todo check tag condition for subtable
280
  return checkAuth(pCxt, pStmt->dbName, NULL, AUTH_TYPE_WRITE, &pTagCond);
4,466,174✔
281
}
282

283
static int32_t authCreateVTable(SAuthCxt* pCxt, SCreateVTableStmt* pStmt) {
144,259✔
284
  PAR_ERR_RET(checkAuth(pCxt, pStmt->dbName, NULL, AUTH_TYPE_WRITE, NULL));
144,259✔
285
  SNode* pCol = NULL;
133,667✔
286
  FOREACH(pCol, pStmt->pCols) {
1,605,747✔
287
    SColumnDefNode* pColDef = (SColumnDefNode*)pCol;
1,480,024✔
288
    if (NULL == pColDef) {
1,480,024✔
UNCOV
289
      PAR_ERR_RET(TSDB_CODE_PAR_INVALID_COLUMN);
×
290
    }
291
    SColumnOptions* pOptions = (SColumnOptions*)pColDef->pOptions;
1,480,024✔
292
    if (pOptions && pOptions->hasRef) {
1,480,024✔
293
      PAR_ERR_RET(checkAuth(pCxt, pOptions->refDb, pOptions->refTable, AUTH_TYPE_READ, NULL));
797,943✔
294
    }
295
  }
296
  return TSDB_CODE_SUCCESS;
125,723✔
297
}
298

299
static int32_t authCreateVSubTable(SAuthCxt* pCxt, SCreateVSubTableStmt* pStmt) {
349,195✔
300
  int32_t    code = TSDB_CODE_SUCCESS;
349,195✔
301
  SNode*     pNode = NULL;
349,195✔
302
  SNodeList* pTmpList = pStmt->pSpecificColRefs ? pStmt->pSpecificColRefs : pStmt->pColRefs;
349,195✔
303
  PAR_ERR_RET(checkAuth(pCxt, pStmt->dbName, NULL, AUTH_TYPE_WRITE, NULL));
349,195✔
304
  if (NULL == pTmpList) {
338,603✔
305
    // no column reference
306
    return TSDB_CODE_SUCCESS;
6,874✔
307
  }
308

309
  FOREACH(pNode, pTmpList) {
2,371,055✔
310
    SColumnRefNode* pColRef = (SColumnRefNode*)pNode;
2,047,270✔
311
    if (NULL == pColRef) {
2,047,270✔
UNCOV
312
      PAR_ERR_RET(TSDB_CODE_PAR_INVALID_COLUMN);
×
313
    }
314
    PAR_ERR_RET(checkAuth(pCxt, pColRef->refDbName, pColRef->refTableName, AUTH_TYPE_READ, NULL));
2,047,270✔
315
  }
316
  return code;
323,785✔
317
}
318

319
static int32_t authCreateStream(SAuthCxt* pCxt, SCreateStreamStmt* pStmt) {
689,840✔
320
  int32_t   code = TSDB_CODE_SUCCESS;
689,840✔
321

322
  if (IS_SYS_DBNAME(pStmt->streamDbName)) {
689,840✔
UNCOV
323
    return TSDB_CODE_PAR_PERMISSION_DENIED;
×
324
  }
325
  if (IS_SYS_DBNAME(pStmt->targetDbName)) {
689,840✔
326
    return TSDB_CODE_PAR_PERMISSION_DENIED;
302✔
327
  }
328
  if (pStmt->pTrigger) {
689,538✔
329
    SStreamTriggerNode *pTrigger = (SStreamTriggerNode*)pStmt->pTrigger;
689,538✔
330
    STableNode* pTriggerTable = (STableNode*)pTrigger->pTrigerTable;
689,538✔
331
    if (pTriggerTable && IS_SYS_DBNAME(pTriggerTable->dbName)) {
689,538✔
332
      return TSDB_CODE_PAR_PERMISSION_DENIED;
302✔
333
    }
334
  }
335
  return code;
689,236✔
336
}
337

338
static int32_t authCreateMultiTable(SAuthCxt* pCxt, SCreateMultiTablesStmt* pStmt) {
35,085,783✔
339
  int32_t code = TSDB_CODE_SUCCESS;
35,085,783✔
340
  SNode*  pNode = NULL;
35,085,783✔
341
  FOREACH(pNode, pStmt->pSubTables) {
74,707,280✔
342
    if (pNode->type == QUERY_NODE_CREATE_SUBTABLE_CLAUSE) {
39,626,224✔
343
      SCreateSubTableClause* pClause = (SCreateSubTableClause*)pNode;
39,630,319✔
344
      code = checkAuth(pCxt, pClause->dbName, NULL, AUTH_TYPE_WRITE, NULL);
39,630,319✔
345
      if (TSDB_CODE_SUCCESS != code) {
39,621,917✔
346
        break;
358✔
347
      }
348
    } else {
UNCOV
349
      SCreateSubTableFromFileClause* pClause = (SCreateSubTableFromFileClause*)pNode;
×
UNCOV
350
      code = checkAuth(pCxt, pClause->useDbName, NULL, AUTH_TYPE_WRITE, NULL);
×
351
      if (TSDB_CODE_SUCCESS != code) {
218✔
UNCOV
352
        break;
×
353
      }
354
    }
355
  }
356
  return code;
35,080,819✔
357
}
358

359
static int32_t authDropTable(SAuthCxt* pCxt, SDropTableStmt* pStmt) {
1,633,868✔
360
  int32_t code = TSDB_CODE_SUCCESS;
1,633,868✔
361
  if (pStmt->withOpt && !pCxt->pParseCxt->isSuperUser) {
1,633,868✔
362
    return TSDB_CODE_PAR_PERMISSION_DENIED;
660✔
363
  }
364
  SNode* pNode = NULL;
1,633,208✔
365
  FOREACH(pNode, pStmt->pTables) {
3,547,259✔
366
    SDropTableClause* pClause = (SDropTableClause*)pNode;
1,914,389✔
367
    code = checkAuth(pCxt, pClause->dbName, pClause->tableName, AUTH_TYPE_WRITE, NULL);
1,914,389✔
368
    if (TSDB_CODE_SUCCESS != code) {
1,914,389✔
369
      break;
338✔
370
    }
371
  }
372
  return code;
1,633,208✔
373
}
374

375
static int32_t authDropStable(SAuthCxt* pCxt, SDropSuperTableStmt* pStmt) {
182,486✔
376
  if (pStmt->withOpt && !pCxt->pParseCxt->isSuperUser) {
182,486✔
377
    return TSDB_CODE_PAR_PERMISSION_DENIED;
330✔
378
  }
379
  return checkAuth(pCxt, pStmt->dbName, pStmt->tableName, AUTH_TYPE_WRITE, NULL);
182,156✔
380
}
381

382
static int32_t authDropVtable(SAuthCxt* pCxt, SDropVirtualTableStmt* pStmt) {
74,997✔
383
  if (pStmt->withOpt && !pCxt->pParseCxt->isSuperUser) {
74,997✔
UNCOV
384
    return TSDB_CODE_PAR_PERMISSION_DENIED;
×
385
  }
386
  return checkAuth(pCxt, pStmt->dbName, pStmt->tableName, AUTH_TYPE_WRITE, NULL);
74,997✔
387
}
388

389
static int32_t authAlterTable(SAuthCxt* pCxt, SAlterTableStmt* pStmt) {
19,958,749✔
390
  SNode* pTagCond = NULL;
19,958,749✔
391
  // todo check tag condition for subtable
392
  return checkAuth(pCxt, pStmt->dbName, pStmt->tableName, AUTH_TYPE_WRITE, NULL);
19,958,749✔
393
}
394

395
static int32_t authAlterVTable(SAuthCxt* pCxt, SAlterTableStmt* pStmt) {
397,552✔
396
  PAR_ERR_RET(checkAuth(pCxt, pStmt->dbName, pStmt->tableName, AUTH_TYPE_WRITE, NULL));
397,552✔
397
  if (pStmt->alterType == TSDB_ALTER_TABLE_ADD_COLUMN_WITH_COLUMN_REF ||
339,144✔
398
      pStmt->alterType == TSDB_ALTER_TABLE_ALTER_COLUMN_REF) {
309,264✔
399
    PAR_ERR_RET(checkAuth(pCxt, pStmt->dbName, pStmt->refTableName, AUTH_TYPE_READ, NULL));
131,915✔
400
  }
401
  PAR_RET(TSDB_CODE_SUCCESS);
317,904✔
402
}
403

404
static int32_t authCreateView(SAuthCxt* pCxt, SCreateViewStmt* pStmt) {
260,548✔
405
#ifndef TD_ENTERPRISE
406
  return TSDB_CODE_OPS_NOT_SUPPORT;
407
#endif
408
  return checkAuth(pCxt, pStmt->dbName, NULL, AUTH_TYPE_WRITE, NULL);
260,548✔
409
}
410

411
static int32_t authDropView(SAuthCxt* pCxt, SDropViewStmt* pStmt) {
185,915✔
412
#ifndef TD_ENTERPRISE
413
  return TSDB_CODE_OPS_NOT_SUPPORT;
414
#endif
415
  return checkViewAuth(pCxt, pStmt->dbName, pStmt->viewName, AUTH_TYPE_ALTER, NULL);
185,915✔
416
}
417

418
static int32_t authCreateRsma(SAuthCxt* pCxt, SCreateRsmaStmt* pStmt) {
232,142✔
419
  return TSDB_CODE_SUCCESS;
232,142✔
420
}
421

422
static int32_t authDropRsma(SAuthCxt* pCxt, SDropRsmaStmt* pStmt) {
4,674✔
423
  return TSDB_CODE_SUCCESS;
4,674✔
424
}
425

426
static int32_t authQuery(SAuthCxt* pCxt, SNode* pStmt) {
197,614,073✔
427
  switch (nodeType(pStmt)) {
197,614,073✔
428
    case QUERY_NODE_SET_OPERATOR:
2,938,541✔
429
      return authSetOperator(pCxt, (SSetOperator*)pStmt);
2,938,541✔
430
    case QUERY_NODE_SELECT_STMT:
104,568,227✔
431
      return authSelect(pCxt, (SSelectStmt*)pStmt);
104,568,227✔
432
    case QUERY_NODE_DROP_USER_STMT:
76,251✔
433
      return authDropUser(pCxt, (SDropUserStmt*)pStmt);
76,251✔
434
    case QUERY_NODE_DELETE_STMT:
1,788,510✔
435
      return authDelete(pCxt, (SDeleteStmt*)pStmt);
1,788,510✔
436
    case QUERY_NODE_INSERT_STMT:
71,188✔
437
      return authInsert(pCxt, (SInsertStmt*)pStmt);
71,188✔
438
    case QUERY_NODE_CREATE_TABLE_STMT:
4,466,174✔
439
      return authCreateTable(pCxt, (SCreateTableStmt*)pStmt);
4,466,174✔
440
    case QUERY_NODE_CREATE_VIRTUAL_TABLE_STMT:
144,259✔
441
      return authCreateVTable(pCxt, (SCreateVTableStmt*)pStmt);
144,259✔
442
    case QUERY_NODE_CREATE_VIRTUAL_SUBTABLE_STMT:
349,195✔
443
      return authCreateVSubTable(pCxt, (SCreateVSubTableStmt*)pStmt);
349,195✔
444
    case QUERY_NODE_CREATE_MULTI_TABLES_STMT:
35,072,983✔
445
      return authCreateMultiTable(pCxt, (SCreateMultiTablesStmt*)pStmt);
35,072,983✔
446
    case QUERY_NODE_CREATE_STREAM_STMT:
689,840✔
447
      return authCreateStream(pCxt, (SCreateStreamStmt*)pStmt);
689,840✔
448
    case QUERY_NODE_DROP_TABLE_STMT:
1,633,868✔
449
      return authDropTable(pCxt, (SDropTableStmt*)pStmt);
1,633,868✔
450
    case QUERY_NODE_DROP_SUPER_TABLE_STMT:
182,486✔
451
      return authDropStable(pCxt, (SDropSuperTableStmt*)pStmt);
182,486✔
452
    case QUERY_NODE_DROP_VIRTUAL_TABLE_STMT:
74,997✔
453
      return authDropVtable(pCxt, (SDropVirtualTableStmt*)pStmt);
74,997✔
454
    case QUERY_NODE_ALTER_TABLE_STMT:
19,958,749✔
455
    case QUERY_NODE_ALTER_SUPER_TABLE_STMT:
456
      return authAlterTable(pCxt, (SAlterTableStmt*)pStmt);
19,958,749✔
457
    case QUERY_NODE_ALTER_VIRTUAL_TABLE_STMT:
397,552✔
458
      return authAlterVTable(pCxt, (SAlterTableStmt*)pStmt);
397,552✔
459
    case QUERY_NODE_SHOW_DNODES_STMT:
1,094,703✔
460
    case QUERY_NODE_SHOW_MNODES_STMT:
461
    case QUERY_NODE_SHOW_MODULES_STMT:
462
    case QUERY_NODE_SHOW_QNODES_STMT:
463
    case QUERY_NODE_SHOW_SNODES_STMT:
464
    case QUERY_NODE_SHOW_BACKUP_NODES_STMT:
465
    case QUERY_NODE_SHOW_CLUSTER_STMT:
466
    case QUERY_NODE_SHOW_LICENCES_STMT:
467
    case QUERY_NODE_SHOW_VGROUPS_STMT:
468
    case QUERY_NODE_SHOW_DB_ALIVE_STMT:
469
    // case QUERY_NODE_SHOW_CLUSTER_ALIVE_STMT:
470
    case QUERY_NODE_SHOW_CREATE_DATABASE_STMT:
471
    case QUERY_NODE_SHOW_TABLE_DISTRIBUTED_STMT:
472
    case QUERY_NODE_SHOW_DNODE_VARIABLES_STMT:
473
    case QUERY_NODE_SHOW_VNODES_STMT:
474
    case QUERY_NODE_SHOW_SCORES_STMT:
475
    case QUERY_NODE_SHOW_USERS_STMT:
476
    case QUERY_NODE_SHOW_USERS_FULL_STMT:
477
    case QUERY_NODE_SHOW_USER_PRIVILEGES_STMT:
478
    case QUERY_NODE_SHOW_GRANTS_FULL_STMT:
479
    case QUERY_NODE_SHOW_GRANTS_LOGS_STMT:
480
    case QUERY_NODE_SHOW_CLUSTER_MACHINES_STMT:
481
    case QUERY_NODE_SHOW_ARBGROUPS_STMT:
482
    case QUERY_NODE_SHOW_ENCRYPTIONS_STMT:
483
    case QUERY_NODE_SHOW_MOUNTS_STMT:
484
      return !pCxt->pParseCxt->enableSysInfo ? TSDB_CODE_PAR_PERMISSION_DENIED : TSDB_CODE_SUCCESS;
1,094,703✔
485
    case QUERY_NODE_SHOW_USAGE_STMT:
1,054✔
486
    case QUERY_NODE_SHOW_ANODES_STMT:
487
    case QUERY_NODE_SHOW_ANODES_FULL_STMT:
488
      return TSDB_CODE_SUCCESS;
1,054✔
489
    case QUERY_NODE_SHOW_TABLES_STMT:
601,695✔
490
    case QUERY_NODE_SHOW_STABLES_STMT:
491
      return authShowTables(pCxt, (SShowStmt*)pStmt);
601,695✔
492
    case QUERY_NODE_SHOW_VTABLES_STMT:
60,681✔
493
      return authShowVtables(pCxt, (SShowStmt*)pStmt);
60,681✔
494
    case QUERY_NODE_SHOW_CREATE_TABLE_STMT:
125,271✔
495
    case QUERY_NODE_SHOW_CREATE_VTABLE_STMT:
496
    case QUERY_NODE_SHOW_CREATE_STABLE_STMT:
497
      return authShowCreateTable(pCxt, (SShowCreateTableStmt*)pStmt);
125,271✔
498
      //    case QUERY_NODE_SHOW_CREATE_VIEW_STMT:
499
      //      return authShowCreateView(pCxt, (SShowCreateViewStmt*)pStmt);
500
    case QUERY_NODE_CREATE_VIEW_STMT:
260,548✔
501
      return authCreateView(pCxt, (SCreateViewStmt*)pStmt);
260,548✔
502
    case QUERY_NODE_DROP_VIEW_STMT:
185,915✔
503
      return authDropView(pCxt, (SDropViewStmt*)pStmt);
185,915✔
504
    case QUERY_NODE_CREATE_RSMA_STMT:
232,142✔
505
      return authCreateRsma(pCxt, (SCreateRsmaStmt*)pStmt);
232,142✔
506
    case QUERY_NODE_DROP_RSMA_STMT:
22,972✔
507
      return authDropRsma(pCxt, (SDropRsmaStmt*)pStmt);
22,972✔
508
    default:
22,616,704✔
509
      break;
22,616,704✔
510
  }
511

512
  return TSDB_CODE_SUCCESS;
22,616,704✔
513
}
514

515
int32_t authenticate(SParseContext* pParseCxt, SQuery* pQuery, SParseMetaCache* pMetaCache) {
182,653,314✔
516
  SAuthCxt cxt = {.pParseCxt = pParseCxt, .pMetaCache = pMetaCache, .errCode = TSDB_CODE_SUCCESS};
182,653,314✔
517
  return authQuery(&cxt, pQuery->pRoot);
182,656,006✔
518
}
STATUS · Troubleshooting · Open an Issue · Sales · Support · CAREERS · ENTERPRISE · START FREE · SCHEDULE DEMO
ANNOUNCEMENTS · TWITTER · TOS & SLA · Supported CI Services · What's a CI service? · Automated Testing

© 2026 Coveralls, Inc