mndUser.c 49.6 KB
Newer Older
H
refact  
Hongze Cheng 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/*
 * 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/>.
 */

S
Shengliang Guan 已提交
16
#define _DEFAULT_SOURCE
S
Shengliang Guan 已提交
17
#include "mndUser.h"
S
Shengliang Guan 已提交
18
#include "mndDb.h"
S
Shengliang Guan 已提交
19
#include "mndPrivilege.h"
S
Shengliang Guan 已提交
20
#include "mndShow.h"
X
Xiaoyu Wang 已提交
21
#include "mndStb.h"
22
#include "mndTopic.h"
S
Shengliang Guan 已提交
23
#include "mndTrans.h"
S
tbase64  
Shengliang Guan 已提交
24
#include "tbase64.h"
S
Shengliang Guan 已提交
25

C
table  
cademfly 已提交
26
#define USER_VER_NUMBER   3
27
#define USER_RESERVE_SIZE 64
S
Shengliang Guan 已提交
28

S
Shengliang Guan 已提交
29 30 31 32
static int32_t  mndCreateDefaultUsers(SMnode *pMnode);
static SSdbRow *mndUserActionDecode(SSdbRaw *pRaw);
static int32_t  mndUserActionInsert(SSdb *pSdb, SUserObj *pUser);
static int32_t  mndUserActionDelete(SSdb *pSdb, SUserObj *pUser);
S
Shengliang Guan 已提交
33
static int32_t  mndUserActionUpdate(SSdb *pSdb, SUserObj *pOld, SUserObj *pNew);
S
Shengliang Guan 已提交
34 35 36 37 38 39
static int32_t  mndCreateUser(SMnode *pMnode, char *acct, SCreateUserReq *pCreate, SRpcMsg *pReq);
static int32_t  mndProcessCreateUserReq(SRpcMsg *pReq);
static int32_t  mndProcessAlterUserReq(SRpcMsg *pReq);
static int32_t  mndProcessDropUserReq(SRpcMsg *pReq);
static int32_t  mndProcessGetUserAuthReq(SRpcMsg *pReq);
static int32_t  mndRetrieveUsers(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rows);
S
Shengliang Guan 已提交
40
static void     mndCancelGetNextUser(SMnode *pMnode, void *pIter);
41 42
static int32_t  mndRetrievePrivileges(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rows);
static void     mndCancelGetNextPrivileges(SMnode *pMnode, void *pIter);
S
Shengliang Guan 已提交
43 44

int32_t mndInitUser(SMnode *pMnode) {
S
Shengliang Guan 已提交
45 46 47 48 49 50 51 52 53 54
  SSdbTable table = {
      .sdbType = SDB_USER,
      .keyType = SDB_KEY_BINARY,
      .deployFp = (SdbDeployFp)mndCreateDefaultUsers,
      .encodeFp = (SdbEncodeFp)mndUserActionEncode,
      .decodeFp = (SdbDecodeFp)mndUserActionDecode,
      .insertFp = (SdbInsertFp)mndUserActionInsert,
      .updateFp = (SdbUpdateFp)mndUserActionUpdate,
      .deleteFp = (SdbDeleteFp)mndUserActionDelete,
  };
S
Shengliang Guan 已提交
55

S
Shengliang Guan 已提交
56 57 58
  mndSetMsgHandle(pMnode, TDMT_MND_CREATE_USER, mndProcessCreateUserReq);
  mndSetMsgHandle(pMnode, TDMT_MND_ALTER_USER, mndProcessAlterUserReq);
  mndSetMsgHandle(pMnode, TDMT_MND_DROP_USER, mndProcessDropUserReq);
S
Shengliang Guan 已提交
59
  mndSetMsgHandle(pMnode, TDMT_MND_GET_USER_AUTH, mndProcessGetUserAuthReq);
S
Shengliang Guan 已提交
60

S
Shengliang Guan 已提交
61 62
  mndAddShowRetrieveHandle(pMnode, TSDB_MGMT_TABLE_USER, mndRetrieveUsers);
  mndAddShowFreeIterHandle(pMnode, TSDB_MGMT_TABLE_USER, mndCancelGetNextUser);
S
Shengliang Guan 已提交
63 64
  mndAddShowRetrieveHandle(pMnode, TSDB_MGMT_TABLE_PRIVILEGES, mndRetrievePrivileges);
  mndAddShowFreeIterHandle(pMnode, TSDB_MGMT_TABLE_PRIVILEGES, mndCancelGetNextPrivileges);
S
Shengliang Guan 已提交
65 66 67 68 69 70 71
  return sdbSetTable(pMnode->pSdb, table);
}

void mndCleanupUser(SMnode *pMnode) {}

static int32_t mndCreateDefaultUser(SMnode *pMnode, char *acct, char *user, char *pass) {
  SUserObj userObj = {0};
S
Shengliang Guan 已提交
72
  taosEncryptPass_c((uint8_t *)pass, strlen(pass), userObj.pass);
S
Shengliang Guan 已提交
73 74 75 76
  tstrncpy(userObj.user, user, TSDB_USER_LEN);
  tstrncpy(userObj.acct, acct, TSDB_USER_LEN);
  userObj.createdTime = taosGetTimestampMs();
  userObj.updateTime = userObj.createdTime;
77 78
  userObj.sysInfo = 1;
  userObj.enable = 1;
S
Shengliang Guan 已提交
79 80

  if (strcmp(user, TSDB_DEFAULT_USER) == 0) {
81
    userObj.superUser = 1;
S
Shengliang Guan 已提交
82 83 84 85
  }

  SSdbRaw *pRaw = mndUserActionEncode(&userObj);
  if (pRaw == NULL) return -1;
S
Shengliang Guan 已提交
86
  (void)sdbSetRawStatus(pRaw, SDB_STATUS_READY);
S
Shengliang Guan 已提交
87

88
  mInfo("user:%s, will be created when deploying, raw:%p", userObj.user, pRaw);
89

90
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_RETRY, TRN_CONFLICT_NOTHING, NULL, "create-user");
91
  if (pTrans == NULL) {
S
Shengliang Guan 已提交
92
    sdbFreeRaw(pRaw);
93 94 95
    mError("user:%s, failed to create since %s", userObj.user, terrstr());
    return -1;
  }
96
  mInfo("trans:%d, used to create user:%s", pTrans->id, userObj.user);
97 98 99 100 101 102

  if (mndTransAppendCommitlog(pTrans, pRaw) != 0) {
    mError("trans:%d, failed to commit redo log since %s", pTrans->id, terrstr());
    mndTransDrop(pTrans);
    return -1;
  }
S
Shengliang Guan 已提交
103
  (void)sdbSetRawStatus(pRaw, SDB_STATUS_READY);
104 105 106 107 108 109 110 111 112

  if (mndTransPrepare(pMnode, pTrans) != 0) {
    mError("trans:%d, failed to prepare since %s", pTrans->id, terrstr());
    mndTransDrop(pTrans);
    return -1;
  }

  mndTransDrop(pTrans);
  return 0;
S
Shengliang Guan 已提交
113 114 115 116 117 118 119 120 121 122
}

static int32_t mndCreateDefaultUsers(SMnode *pMnode) {
  if (mndCreateDefaultUser(pMnode, TSDB_DEFAULT_USER, TSDB_DEFAULT_USER, TSDB_DEFAULT_PASS) != 0) {
    return -1;
  }

  return 0;
}

123
SSdbRaw *mndUserActionEncode(SUserObj *pUser) {
124 125
  terrno = TSDB_CODE_OUT_OF_MEMORY;

S
Shengliang Guan 已提交
126 127
  int32_t numOfReadDbs = taosHashGetSize(pUser->readDbs);
  int32_t numOfWriteDbs = taosHashGetSize(pUser->writeDbs);
C
cademfly 已提交
128 129
  int32_t numOfReadStbs = taosHashGetSize(pUser->readTbs);
  int32_t numOfWriteStbs = taosHashGetSize(pUser->writeTbs);
130
  int32_t numOfTopics = taosHashGetSize(pUser->topics);
X
Xiaoyu Wang 已提交
131 132 133
  int32_t numOfUseDbs = taosHashGetSize(pUser->useDbs);
  int32_t size = sizeof(SUserObj) + USER_RESERVE_SIZE +
                 (numOfReadDbs + numOfWriteDbs + numOfUseDbs) * TSDB_DB_FNAME_LEN + numOfTopics * TSDB_TOPIC_FNAME_LEN;
S
Shengliang Guan 已提交
134

C
cademfly 已提交
135
  char *stb = taosHashIterate(pUser->readTbs, NULL);
C
cademfly 已提交
136 137 138 139 140 141 142 143 144 145
  while (stb != NULL) {
    size_t keyLen = 0;
    void  *key = taosHashGetKey(stb, &keyLen);
    size += sizeof(int32_t);
    size += keyLen;

    size_t valueLen = 0;
    valueLen = strlen(stb);
    size += sizeof(int32_t);
    size += valueLen;
C
cademfly 已提交
146
    stb = taosHashIterate(pUser->readTbs, stb);
C
cademfly 已提交
147 148
  }

C
cademfly 已提交
149
  stb = taosHashIterate(pUser->writeTbs, NULL);
C
cademfly 已提交
150 151 152 153 154
  while (stb != NULL) {
    size_t keyLen = 0;
    void  *key = taosHashGetKey(stb, &keyLen);
    size += sizeof(int32_t);
    size += keyLen;
X
Xiaoyu Wang 已提交
155

C
cademfly 已提交
156 157 158
    size_t valueLen = 0;
    valueLen = strlen(stb);
    size += sizeof(int32_t);
X
Xiaoyu Wang 已提交
159
    size += valueLen;
C
cademfly 已提交
160
    stb = taosHashIterate(pUser->writeTbs, stb);
C
cademfly 已提交
161 162
  }

163
  SSdbRaw *pRaw = sdbAllocRaw(SDB_USER, USER_VER_NUMBER, size);
164
  if (pRaw == NULL) goto _OVER;
S
Shengliang Guan 已提交
165 166

  int32_t dataPos = 0;
167 168 169 170 171 172
  SDB_SET_BINARY(pRaw, dataPos, pUser->user, TSDB_USER_LEN, _OVER)
  SDB_SET_BINARY(pRaw, dataPos, pUser->pass, TSDB_PASSWORD_LEN, _OVER)
  SDB_SET_BINARY(pRaw, dataPos, pUser->acct, TSDB_USER_LEN, _OVER)
  SDB_SET_INT64(pRaw, dataPos, pUser->createdTime, _OVER)
  SDB_SET_INT64(pRaw, dataPos, pUser->updateTime, _OVER)
  SDB_SET_INT8(pRaw, dataPos, pUser->superUser, _OVER)
173 174 175
  SDB_SET_INT8(pRaw, dataPos, pUser->sysInfo, _OVER)
  SDB_SET_INT8(pRaw, dataPos, pUser->enable, _OVER)
  SDB_SET_INT8(pRaw, dataPos, pUser->reserve, _OVER)
176
  SDB_SET_INT32(pRaw, dataPos, pUser->authVersion, _OVER)
177 178
  SDB_SET_INT32(pRaw, dataPos, numOfReadDbs, _OVER)
  SDB_SET_INT32(pRaw, dataPos, numOfWriteDbs, _OVER)
179
  SDB_SET_INT32(pRaw, dataPos, numOfTopics, _OVER)
180 181 182

  char *db = taosHashIterate(pUser->readDbs, NULL);
  while (db != NULL) {
183
    SDB_SET_BINARY(pRaw, dataPos, db, TSDB_DB_FNAME_LEN, _OVER);
184 185 186 187 188
    db = taosHashIterate(pUser->readDbs, db);
  }

  db = taosHashIterate(pUser->writeDbs, NULL);
  while (db != NULL) {
189
    SDB_SET_BINARY(pRaw, dataPos, db, TSDB_DB_FNAME_LEN, _OVER);
190 191 192
    db = taosHashIterate(pUser->writeDbs, db);
  }

193 194 195
  char *topic = taosHashIterate(pUser->topics, NULL);
  while (topic != NULL) {
    SDB_SET_BINARY(pRaw, dataPos, topic, TSDB_TOPIC_FNAME_LEN, _OVER);
196
    topic = taosHashIterate(pUser->topics, topic);
197 198
  }

X
Xiaoyu Wang 已提交
199 200 201 202
  SDB_SET_INT32(pRaw, dataPos, numOfReadStbs, _OVER)
  SDB_SET_INT32(pRaw, dataPos, numOfWriteStbs, _OVER)
  SDB_SET_INT32(pRaw, dataPos, numOfUseDbs, _OVER)

C
cademfly 已提交
203
  stb = taosHashIterate(pUser->readTbs, NULL);
C
table  
cademfly 已提交
204
  while (stb != NULL) {
C
cademfly 已提交
205 206 207 208 209 210
    size_t keyLen = 0;
    void  *key = taosHashGetKey(stb, &keyLen);
    SDB_SET_INT32(pRaw, dataPos, keyLen, _OVER)
    SDB_SET_BINARY(pRaw, dataPos, key, keyLen, _OVER);

    size_t valueLen = 0;
X
Xiaoyu Wang 已提交
211
    valueLen = strlen(stb) + 1;
C
cademfly 已提交
212 213
    SDB_SET_INT32(pRaw, dataPos, valueLen, _OVER)
    SDB_SET_BINARY(pRaw, dataPos, stb, valueLen, _OVER);
C
cademfly 已提交
214
    stb = taosHashIterate(pUser->readTbs, stb);
C
table  
cademfly 已提交
215 216
  }

C
cademfly 已提交
217
  stb = taosHashIterate(pUser->writeTbs, NULL);
C
table  
cademfly 已提交
218
  while (stb != NULL) {
C
cademfly 已提交
219 220 221 222
    size_t keyLen = 0;
    void  *key = taosHashGetKey(stb, &keyLen);
    SDB_SET_INT32(pRaw, dataPos, keyLen, _OVER)
    SDB_SET_BINARY(pRaw, dataPos, key, keyLen, _OVER);
X
Xiaoyu Wang 已提交
223

C
cademfly 已提交
224
    size_t valueLen = 0;
X
Xiaoyu Wang 已提交
225
    valueLen = strlen(stb) + 1;
C
cademfly 已提交
226 227
    SDB_SET_INT32(pRaw, dataPos, valueLen, _OVER)
    SDB_SET_BINARY(pRaw, dataPos, stb, valueLen, _OVER);
C
cademfly 已提交
228
    stb = taosHashIterate(pUser->writeTbs, stb);
C
table  
cademfly 已提交
229 230
  }

X
Xiaoyu Wang 已提交
231 232 233 234 235 236 237 238
  int32_t *useDb = taosHashIterate(pUser->useDbs, NULL);
  while (useDb != NULL) {
    size_t keyLen = 0;
    void  *key = taosHashGetKey(useDb, &keyLen);
    SDB_SET_INT32(pRaw, dataPos, keyLen, _OVER)
    SDB_SET_BINARY(pRaw, dataPos, key, keyLen, _OVER);

    SDB_SET_INT32(pRaw, dataPos, *useDb, _OVER)
D
dmchen 已提交
239
    useDb = taosHashIterate(pUser->useDbs, useDb);
X
Xiaoyu Wang 已提交
240 241
  }

242 243
  SDB_SET_RESERVE(pRaw, dataPos, USER_RESERVE_SIZE, _OVER)
  SDB_SET_DATALEN(pRaw, dataPos, _OVER)
244 245 246

  terrno = 0;

247
_OVER:
248 249 250 251 252
  if (terrno != 0) {
    mError("user:%s, failed to encode to raw:%p since %s", pUser->user, pRaw, terrstr());
    sdbFreeRaw(pRaw);
    return NULL;
  }
S
Shengliang Guan 已提交
253

S
Shengliang Guan 已提交
254
  mTrace("user:%s, encode to raw:%p, row:%p", pUser->user, pRaw, pUser);
S
Shengliang Guan 已提交
255
  return pRaw;
S
Shengliang Guan 已提交
256 257
}

S
Shengliang Guan 已提交
258
static SSdbRow *mndUserActionDecode(SSdbRaw *pRaw) {
259
  terrno = TSDB_CODE_OUT_OF_MEMORY;
260 261
  SSdbRow  *pRow = NULL;
  SUserObj *pUser = NULL;
262

S
Shengliang Guan 已提交
263
  int8_t sver = 0;
264
  if (sdbGetRawSoftVer(pRaw, &sver) != 0) goto _OVER;
S
Shengliang Guan 已提交
265

C
table  
cademfly 已提交
266
  if (sver != 1 && sver != 2 && sver != 3) {
S
Shengliang Guan 已提交
267
    terrno = TSDB_CODE_SDB_INVALID_DATA_VER;
268
    goto _OVER;
S
Shengliang Guan 已提交
269
  }
S
Shengliang Guan 已提交
270

271
  pRow = sdbAllocRow(sizeof(SUserObj));
272
  if (pRow == NULL) goto _OVER;
273

274
  pUser = sdbGetRowObj(pRow);
275
  if (pUser == NULL) goto _OVER;
276

S
Shengliang Guan 已提交
277
  int32_t dataPos = 0;
278 279 280 281 282 283
  SDB_GET_BINARY(pRaw, dataPos, pUser->user, TSDB_USER_LEN, _OVER)
  SDB_GET_BINARY(pRaw, dataPos, pUser->pass, TSDB_PASSWORD_LEN, _OVER)
  SDB_GET_BINARY(pRaw, dataPos, pUser->acct, TSDB_USER_LEN, _OVER)
  SDB_GET_INT64(pRaw, dataPos, &pUser->createdTime, _OVER)
  SDB_GET_INT64(pRaw, dataPos, &pUser->updateTime, _OVER)
  SDB_GET_INT8(pRaw, dataPos, &pUser->superUser, _OVER)
284 285 286
  SDB_GET_INT8(pRaw, dataPos, &pUser->sysInfo, _OVER)
  SDB_GET_INT8(pRaw, dataPos, &pUser->enable, _OVER)
  SDB_GET_INT8(pRaw, dataPos, &pUser->reserve, _OVER)
287
  SDB_GET_INT32(pRaw, dataPos, &pUser->authVersion, _OVER)
288 289 290

  int32_t numOfReadDbs = 0;
  int32_t numOfWriteDbs = 0;
291
  int32_t numOfTopics = 0;
292 293
  SDB_GET_INT32(pRaw, dataPos, &numOfReadDbs, _OVER)
  SDB_GET_INT32(pRaw, dataPos, &numOfWriteDbs, _OVER)
294 295 296 297
  if (sver >= 2) {
    SDB_GET_INT32(pRaw, dataPos, &numOfTopics, _OVER)
  }

S
Shengliang Guan 已提交
298 299 300
  pUser->readDbs = taosHashInit(numOfReadDbs, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_ENTRY_LOCK);
  pUser->writeDbs =
      taosHashInit(numOfWriteDbs, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_ENTRY_LOCK);
301 302
  pUser->topics = taosHashInit(numOfTopics, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_ENTRY_LOCK);
  if (pUser->readDbs == NULL || pUser->writeDbs == NULL || pUser->topics == NULL) goto _OVER;
303 304 305

  for (int32_t i = 0; i < numOfReadDbs; ++i) {
    char db[TSDB_DB_FNAME_LEN] = {0};
306
    SDB_GET_BINARY(pRaw, dataPos, db, TSDB_DB_FNAME_LEN, _OVER)
307 308 309 310 311 312
    int32_t len = strlen(db) + 1;
    taosHashPut(pUser->readDbs, db, len, db, TSDB_DB_FNAME_LEN);
  }

  for (int32_t i = 0; i < numOfWriteDbs; ++i) {
    char db[TSDB_DB_FNAME_LEN] = {0};
313
    SDB_GET_BINARY(pRaw, dataPos, db, TSDB_DB_FNAME_LEN, _OVER)
314 315 316 317
    int32_t len = strlen(db) + 1;
    taosHashPut(pUser->writeDbs, db, len, db, TSDB_DB_FNAME_LEN);
  }

318 319 320 321 322 323 324 325 326
  if (sver >= 2) {
    for (int32_t i = 0; i < numOfTopics; ++i) {
      char topic[TSDB_TOPIC_FNAME_LEN] = {0};
      SDB_GET_BINARY(pRaw, dataPos, topic, TSDB_TOPIC_FNAME_LEN, _OVER)
      int32_t len = strlen(topic) + 1;
      taosHashPut(pUser->topics, topic, len, topic, TSDB_TOPIC_FNAME_LEN);
    }
  }

X
Xiaoyu Wang 已提交
327
  if (sver >= 3) {
X
Xiaoyu Wang 已提交
328 329 330 331 332 333 334 335 336 337 338 339 340
    int32_t numOfReadStbs = 0;
    int32_t numOfWriteStbs = 0;
    int32_t numOfUseDbs = 0;
    SDB_GET_INT32(pRaw, dataPos, &numOfReadStbs, _OVER)
    SDB_GET_INT32(pRaw, dataPos, &numOfWriteStbs, _OVER)
    SDB_GET_INT32(pRaw, dataPos, &numOfUseDbs, _OVER)

    pUser->readTbs =
        taosHashInit(numOfReadStbs, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_ENTRY_LOCK);
    pUser->writeTbs =
        taosHashInit(numOfWriteStbs, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_ENTRY_LOCK);
    pUser->useDbs = taosHashInit(numOfUseDbs, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_ENTRY_LOCK);

C
table  
cademfly 已提交
341
    for (int32_t i = 0; i < numOfReadStbs; ++i) {
C
cademfly 已提交
342 343 344
      int32_t keyLen = 0;
      SDB_GET_INT32(pRaw, dataPos, &keyLen, _OVER);

C
cademfly 已提交
345 346
      char *key = taosMemoryCalloc(keyLen, sizeof(char));
      memset(key, 0, keyLen);
C
cademfly 已提交
347 348 349 350
      SDB_GET_BINARY(pRaw, dataPos, key, keyLen, _OVER);

      int32_t valuelen = 0;
      SDB_GET_INT32(pRaw, dataPos, &valuelen, _OVER);
C
cademfly 已提交
351
      char *value = taosMemoryCalloc(valuelen, sizeof(char));
X
Xiaoyu Wang 已提交
352
      memset(value, 0, valuelen);
C
cademfly 已提交
353 354
      SDB_GET_BINARY(pRaw, dataPos, value, valuelen, _OVER)

C
cademfly 已提交
355
      taosHashPut(pUser->readTbs, key, keyLen, value, valuelen);
C
cademfly 已提交
356 357 358

      taosMemoryFree(key);
      taosMemoryFree(value);
C
table  
cademfly 已提交
359 360 361
    }

    for (int32_t i = 0; i < numOfWriteStbs; ++i) {
C
cademfly 已提交
362 363 364
      int32_t keyLen = 0;
      SDB_GET_INT32(pRaw, dataPos, &keyLen, _OVER);

C
cademfly 已提交
365 366
      char *key = taosMemoryCalloc(keyLen, sizeof(char));
      memset(key, 0, keyLen);
C
cademfly 已提交
367 368 369 370
      SDB_GET_BINARY(pRaw, dataPos, key, keyLen, _OVER);

      int32_t valuelen = 0;
      SDB_GET_INT32(pRaw, dataPos, &valuelen, _OVER);
C
cademfly 已提交
371
      char *value = taosMemoryCalloc(valuelen, sizeof(char));
X
Xiaoyu Wang 已提交
372
      memset(value, 0, valuelen);
C
cademfly 已提交
373 374
      SDB_GET_BINARY(pRaw, dataPos, value, valuelen, _OVER)

C
cademfly 已提交
375
      taosHashPut(pUser->writeTbs, key, keyLen, value, valuelen);
C
cademfly 已提交
376 377 378

      taosMemoryFree(key);
      taosMemoryFree(value);
C
table  
cademfly 已提交
379
    }
X
Xiaoyu Wang 已提交
380 381 382 383 384 385 386 387 388 389 390 391 392

    for (int32_t i = 0; i < numOfUseDbs; ++i) {
      int32_t keyLen = 0;
      SDB_GET_INT32(pRaw, dataPos, &keyLen, _OVER);

      char *key = taosMemoryCalloc(keyLen, sizeof(char));
      memset(key, 0, keyLen);
      SDB_GET_BINARY(pRaw, dataPos, key, keyLen, _OVER);

      int32_t ref = 0;
      SDB_GET_INT32(pRaw, dataPos, &ref, _OVER);

      taosHashPut(pUser->useDbs, key, keyLen, &ref, sizeof(ref));
X
Xiaoyu Wang 已提交
393
      taosMemoryFree(key);
X
Xiaoyu Wang 已提交
394
    }
C
table  
cademfly 已提交
395 396
  }

397
  SDB_GET_RESERVE(pRaw, dataPos, USER_RESERVE_SIZE, _OVER)
S
Shengliang Guan 已提交
398
  taosInitRWLatch(&pUser->lock);
399 400 401

  terrno = 0;

402
_OVER:
403
  if (terrno != 0) {
404 405 406 407 408
    mError("user:%s, failed to decode from raw:%p since %s", pUser == NULL ? "null" : pUser->user, pRaw, terrstr());
    if (pUser != NULL) {
      taosHashCleanup(pUser->readDbs);
      taosHashCleanup(pUser->writeDbs);
      taosHashCleanup(pUser->topics);
C
cademfly 已提交
409 410
      taosHashCleanup(pUser->readTbs);
      taosHashCleanup(pUser->writeTbs);
X
Xiaoyu Wang 已提交
411
      taosHashCleanup(pUser->useDbs);
412
    }
wafwerar's avatar
wafwerar 已提交
413
    taosMemoryFreeClear(pRow);
414 415
    return NULL;
  }
S
Shengliang Guan 已提交
416

S
Shengliang Guan 已提交
417
  mTrace("user:%s, decode from raw:%p, row:%p", pUser->user, pRaw, pUser);
S
Shengliang Guan 已提交
418
  return pRow;
S
Shengliang Guan 已提交
419
}
S
Shengliang Guan 已提交
420

S
Shengliang Guan 已提交
421
static int32_t mndUserActionInsert(SSdb *pSdb, SUserObj *pUser) {
S
Shengliang Guan 已提交
422
  mTrace("user:%s, perform insert action, row:%p", pUser->user, pUser);
S
Shengliang Guan 已提交
423

S
Shengliang Guan 已提交
424 425
  SAcctObj *pAcct = sdbAcquire(pSdb, SDB_ACCT, pUser->acct);
  if (pAcct == NULL) {
S
Shengliang Guan 已提交
426
    terrno = TSDB_CODE_MND_ACCT_NOT_EXIST;
S
Shengliang Guan 已提交
427
    mError("user:%s, failed to perform insert action since %s", pUser->user, terrstr());
S
Shengliang Guan 已提交
428
    return -1;
S
Shengliang Guan 已提交
429
  }
S
Shengliang Guan 已提交
430 431
  pUser->acctId = pAcct->acctId;
  sdbRelease(pSdb, pAcct);
S
Shengliang Guan 已提交
432

S
Shengliang Guan 已提交
433 434
  return 0;
}
S
Shengliang Guan 已提交
435

C
cademfly 已提交
436 437 438 439 440 441 442 443 444 445 446
SHashObj *mndDupTableHash(SHashObj *pOld) {
  SHashObj *pNew =
      taosHashInit(taosHashGetSize(pOld), taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_ENTRY_LOCK);
  if (pNew == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return NULL;
  }

  char *tb = taosHashIterate(pOld, NULL);
  while (tb != NULL) {
    size_t keyLen = 0;
X
Xiaoyu Wang 已提交
447
    char  *key = taosHashGetKey(tb, &keyLen);
C
cademfly 已提交
448 449 450 451 452 453 454 455 456 457 458 459 460 461

    int32_t valueLen = strlen(tb) + 1;
    if (taosHashPut(pNew, key, keyLen, tb, valueLen) != 0) {
      taosHashCancelIterate(pOld, tb);
      taosHashCleanup(pNew);
      terrno = TSDB_CODE_OUT_OF_MEMORY;
      return NULL;
    }
    tb = taosHashIterate(pOld, tb);
  }

  return pNew;
}

X
Xiaoyu Wang 已提交
462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486
SHashObj *mndDupUseDbHash(SHashObj *pOld) {
  SHashObj *pNew =
      taosHashInit(taosHashGetSize(pOld), taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_ENTRY_LOCK);
  if (pNew == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return NULL;
  }

  int32_t *db = taosHashIterate(pOld, NULL);
  while (db != NULL) {
    size_t keyLen = 0;
    char  *key = taosHashGetKey(db, &keyLen);

    if (taosHashPut(pNew, key, keyLen, db, sizeof(*db)) != 0) {
      taosHashCancelIterate(pOld, db);
      taosHashCleanup(pNew);
      terrno = TSDB_CODE_OUT_OF_MEMORY;
      return NULL;
    }
    db = taosHashIterate(pOld, db);
  }

  return pNew;
}

487 488 489 490 491 492 493 494
static int32_t mndUserDupObj(SUserObj *pUser, SUserObj *pNew) {
  memcpy(pNew, pUser, sizeof(SUserObj));
  pNew->authVersion++;
  pNew->updateTime = taosGetTimestampMs();

  taosRLockLatch(&pUser->lock);
  pNew->readDbs = mndDupDbHash(pUser->readDbs);
  pNew->writeDbs = mndDupDbHash(pUser->writeDbs);
C
cademfly 已提交
495 496
  pNew->readTbs = mndDupTableHash(pUser->readTbs);
  pNew->writeTbs = mndDupTableHash(pUser->writeTbs);
497
  pNew->topics = mndDupTopicHash(pUser->topics);
X
Xiaoyu Wang 已提交
498
  pNew->useDbs = mndDupUseDbHash(pUser->useDbs);
499 500 501 502 503 504 505 506 507
  taosRUnLockLatch(&pUser->lock);

  if (pNew->readDbs == NULL || pNew->writeDbs == NULL || pNew->topics == NULL) {
    return -1;
  }
  return 0;
}

static void mndUserFreeObj(SUserObj *pUser) {
508 509
  taosHashCleanup(pUser->readDbs);
  taosHashCleanup(pUser->writeDbs);
510
  taosHashCleanup(pUser->topics);
C
cademfly 已提交
511 512
  taosHashCleanup(pUser->readTbs);
  taosHashCleanup(pUser->writeTbs);
X
Xiaoyu Wang 已提交
513
  taosHashCleanup(pUser->useDbs);
514 515
  pUser->readDbs = NULL;
  pUser->writeDbs = NULL;
516
  pUser->topics = NULL;
C
cademfly 已提交
517 518
  pUser->readTbs = NULL;
  pUser->writeTbs = NULL;
X
Xiaoyu Wang 已提交
519
  pUser->useDbs = NULL;
520 521 522 523 524
}

static int32_t mndUserActionDelete(SSdb *pSdb, SUserObj *pUser) {
  mTrace("user:%s, perform delete action, row:%p", pUser->user, pUser);
  mndUserFreeObj(pUser);
S
Shengliang Guan 已提交
525 526 527
  return 0;
}

S
Shengliang Guan 已提交
528
static int32_t mndUserActionUpdate(SSdb *pSdb, SUserObj *pOld, SUserObj *pNew) {
S
Shengliang Guan 已提交
529
  mTrace("user:%s, perform update action, old row:%p new row:%p", pOld->user, pOld, pNew);
S
Shengliang Guan 已提交
530
  taosWLockLatch(&pOld->lock);
S
Shengliang Guan 已提交
531
  pOld->updateTime = pNew->updateTime;
D
dapan1121 已提交
532
  pOld->authVersion = pNew->authVersion;
533 534
  pOld->sysInfo = pNew->sysInfo;
  pOld->enable = pNew->enable;
S
Shengliang Guan 已提交
535
  memcpy(pOld->pass, pNew->pass, TSDB_PASSWORD_LEN);
wafwerar's avatar
wafwerar 已提交
536 537
  TSWAP(pOld->readDbs, pNew->readDbs);
  TSWAP(pOld->writeDbs, pNew->writeDbs);
538
  TSWAP(pOld->topics, pNew->topics);
C
cademfly 已提交
539 540
  TSWAP(pOld->readTbs, pNew->readTbs);
  TSWAP(pOld->writeTbs, pNew->writeTbs);
X
Xiaoyu Wang 已提交
541
  TSWAP(pOld->useDbs, pNew->useDbs);
S
Shengliang Guan 已提交
542
  taosWUnLockLatch(&pOld->lock);
543

S
Shengliang Guan 已提交
544 545 546
  return 0;
}

547
SUserObj *mndAcquireUser(SMnode *pMnode, const char *userName) {
S
Shengliang Guan 已提交
548 549 550
  SSdb     *pSdb = pMnode->pSdb;
  SUserObj *pUser = sdbAcquire(pSdb, SDB_USER, userName);
  if (pUser == NULL) {
dengyihao's avatar
dengyihao 已提交
551 552 553 554 555
    if (terrno == TSDB_CODE_SDB_OBJ_NOT_THERE) {
      terrno = TSDB_CODE_MND_USER_NOT_EXIST;
    } else {
      terrno = TSDB_CODE_MND_USER_NOT_AVAILABLE;
    }
S
Shengliang Guan 已提交
556 557
  }
  return pUser;
S
Shengliang Guan 已提交
558
}
S
Shengliang Guan 已提交
559

S
Shengliang Guan 已提交
560 561 562
void mndReleaseUser(SMnode *pMnode, SUserObj *pUser) {
  SSdb *pSdb = pMnode->pSdb;
  sdbRelease(pSdb, pUser);
S
Shengliang Guan 已提交
563 564
}

S
Shengliang Guan 已提交
565
static int32_t mndCreateUser(SMnode *pMnode, char *acct, SCreateUserReq *pCreate, SRpcMsg *pReq) {
S
Shengliang Guan 已提交
566
  SUserObj userObj = {0};
S
Shengliang Guan 已提交
567 568
  taosEncryptPass_c((uint8_t *)pCreate->pass, strlen(pCreate->pass), userObj.pass);
  tstrncpy(userObj.user, pCreate->user, TSDB_USER_LEN);
S
Shengliang Guan 已提交
569 570 571
  tstrncpy(userObj.acct, acct, TSDB_USER_LEN);
  userObj.createdTime = taosGetTimestampMs();
  userObj.updateTime = userObj.createdTime;
572
  userObj.superUser = 0;  // pCreate->superUser;
573 574
  userObj.sysInfo = pCreate->sysInfo;
  userObj.enable = pCreate->enable;
S
Shengliang Guan 已提交
575

576
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_ROLLBACK, TRN_CONFLICT_NOTHING, pReq, "create-user");
S
Shengliang Guan 已提交
577
  if (pTrans == NULL) {
S
Shengliang Guan 已提交
578
    mError("user:%s, failed to create since %s", pCreate->user, terrstr());
S
Shengliang Guan 已提交
579 580
    return -1;
  }
581
  mInfo("trans:%d, used to create user:%s", pTrans->id, pCreate->user);
S
Shengliang Guan 已提交
582

583 584 585
  SSdbRaw *pCommitRaw = mndUserActionEncode(&userObj);
  if (pCommitRaw == NULL || mndTransAppendCommitlog(pTrans, pCommitRaw) != 0) {
    mError("trans:%d, failed to commit redo log since %s", pTrans->id, terrstr());
S
Shengliang Guan 已提交
586
    mndTransDrop(pTrans);
S
Shengliang Guan 已提交
587
    return -1;
S
Shengliang Guan 已提交
588
  }
S
Shengliang Guan 已提交
589
  (void)sdbSetRawStatus(pCommitRaw, SDB_STATUS_READY);
S
Shengliang Guan 已提交
590

S
Shengliang Guan 已提交
591
  if (mndTransPrepare(pMnode, pTrans) != 0) {
S
Shengliang Guan 已提交
592
    mError("trans:%d, failed to prepare since %s", pTrans->id, terrstr());
S
Shengliang Guan 已提交
593
    mndTransDrop(pTrans);
S
Shengliang Guan 已提交
594
    return -1;
S
Shengliang Guan 已提交
595 596
  }

S
Shengliang Guan 已提交
597
  mndTransDrop(pTrans);
S
Shengliang Guan 已提交
598
  return 0;
S
Shengliang Guan 已提交
599 600
}

S
Shengliang Guan 已提交
601 602
static int32_t mndProcessCreateUserReq(SRpcMsg *pReq) {
  SMnode        *pMnode = pReq->info.node;
S
Shengliang Guan 已提交
603 604 605 606 607
  int32_t        code = -1;
  SUserObj      *pUser = NULL;
  SUserObj      *pOperUser = NULL;
  SCreateUserReq createReq = {0};

S
Shengliang Guan 已提交
608
  if (tDeserializeSCreateUserReq(pReq->pCont, pReq->contLen, &createReq) != 0) {
S
Shengliang Guan 已提交
609
    terrno = TSDB_CODE_INVALID_MSG;
610
    goto _OVER;
S
Shengliang Guan 已提交
611
  }
S
Shengliang Guan 已提交
612

613
  mInfo("user:%s, start to create", createReq.user);
614 615 616
  if (mndCheckOperPrivilege(pMnode, pReq->info.conn.user, MND_OPER_CREATE_USER) != 0) {
    goto _OVER;
  }
S
Shengliang Guan 已提交
617

S
Shengliang Guan 已提交
618
  if (createReq.user[0] == 0) {
S
Shengliang Guan 已提交
619
    terrno = TSDB_CODE_MND_INVALID_USER_FORMAT;
620
    goto _OVER;
S
Shengliang Guan 已提交
621 622
  }

S
Shengliang Guan 已提交
623
  if (createReq.pass[0] == 0) {
S
Shengliang Guan 已提交
624
    terrno = TSDB_CODE_MND_INVALID_PASS_FORMAT;
625
    goto _OVER;
S
Shengliang Guan 已提交
626 627
  }

S
Shengliang Guan 已提交
628
  pUser = mndAcquireUser(pMnode, createReq.user);
S
Shengliang Guan 已提交
629
  if (pUser != NULL) {
S
Shengliang Guan 已提交
630
    terrno = TSDB_CODE_MND_USER_ALREADY_EXIST;
631
    goto _OVER;
S
Shengliang Guan 已提交
632 633
  }

634
  pOperUser = mndAcquireUser(pMnode, pReq->info.conn.user);
S
Shengliang Guan 已提交
635
  if (pOperUser == NULL) {
S
Shengliang Guan 已提交
636
    terrno = TSDB_CODE_MND_NO_USER_FROM_CONN;
637
    goto _OVER;
S
Shengliang Guan 已提交
638 639
  }

C
Cary Xu 已提交
640 641 642 643
  if ((terrno = grantCheck(TSDB_GRANT_USER)) != 0) {
    code = terrno;
    goto _OVER;
  }
644

S
Shengliang Guan 已提交
645
  code = mndCreateUser(pMnode, pOperUser->acct, &createReq, pReq);
S
Shengliang Guan 已提交
646
  if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
S
Shengliang Guan 已提交
647

648
_OVER:
S
Shengliang Guan 已提交
649
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
S
Shengliang Guan 已提交
650
    mError("user:%s, failed to create since %s", createReq.user, terrstr());
S
Shengliang Guan 已提交
651 652
  }

S
Shengliang Guan 已提交
653 654 655 656
  mndReleaseUser(pMnode, pUser);
  mndReleaseUser(pMnode, pOperUser);

  return code;
S
Shengliang Guan 已提交
657 658
}

S
Shengliang Guan 已提交
659
static int32_t mndAlterUser(SMnode *pMnode, SUserObj *pOld, SUserObj *pNew, SRpcMsg *pReq) {
660
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_ROLLBACK, TRN_CONFLICT_NOTHING, pReq, "alter-user");
S
Shengliang Guan 已提交
661
  if (pTrans == NULL) {
S
Shengliang Guan 已提交
662
    mError("user:%s, failed to alter since %s", pOld->user, terrstr());
S
Shengliang Guan 已提交
663 664
    return -1;
  }
665
  mInfo("trans:%d, used to alter user:%s", pTrans->id, pOld->user);
S
Shengliang Guan 已提交
666

667 668 669
  SSdbRaw *pCommitRaw = mndUserActionEncode(pNew);
  if (pCommitRaw == NULL || mndTransAppendCommitlog(pTrans, pCommitRaw) != 0) {
    mError("trans:%d, failed to append commit log since %s", pTrans->id, terrstr());
S
Shengliang Guan 已提交
670 671 672
    mndTransDrop(pTrans);
    return -1;
  }
S
Shengliang Guan 已提交
673
  (void)sdbSetRawStatus(pCommitRaw, SDB_STATUS_READY);
S
Shengliang Guan 已提交
674 675 676 677 678 679 680 681 682 683 684

  if (mndTransPrepare(pMnode, pTrans) != 0) {
    mError("trans:%d, failed to prepare since %s", pTrans->id, terrstr());
    mndTransDrop(pTrans);
    return -1;
  }

  mndTransDrop(pTrans);
  return 0;
}

685
SHashObj *mndDupObjHash(SHashObj *pOld, int32_t dataLen) {
S
Shengliang Guan 已提交
686 687
  SHashObj *pNew =
      taosHashInit(taosHashGetSize(pOld), taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_ENTRY_LOCK);
S
Shengliang Guan 已提交
688 689 690 691 692 693 694 695
  if (pNew == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return NULL;
  }

  char *db = taosHashIterate(pOld, NULL);
  while (db != NULL) {
    int32_t len = strlen(db) + 1;
696
    if (taosHashPut(pNew, db, len, db, dataLen) != 0) {
S
Shengliang Guan 已提交
697 698
      taosHashCancelIterate(pOld, db);
      taosHashCleanup(pNew);
S
Shengliang Guan 已提交
699
      terrno = TSDB_CODE_OUT_OF_MEMORY;
S
Shengliang Guan 已提交
700 701 702 703 704 705 706 707
      return NULL;
    }
    db = taosHashIterate(pOld, db);
  }

  return pNew;
}

708 709 710 711
SHashObj *mndDupDbHash(SHashObj *pOld) { return mndDupObjHash(pOld, TSDB_DB_FNAME_LEN); }

SHashObj *mndDupTopicHash(SHashObj *pOld) { return mndDupObjHash(pOld, TSDB_TOPIC_FNAME_LEN); }

X
Xiaoyu Wang 已提交
712 713
static int32_t mndTablePriviledge(SMnode *pMnode, SHashObj *hash, SHashObj *useDbHash, SAlterUserReq *alterReq,
                                  SSdb *pSdb) {
X
Xiaoyu Wang 已提交
714 715
  void *pIter = NULL;
  char  tbFName[TSDB_TABLE_FNAME_LEN] = {0};
C
cademfly 已提交
716

C
cademfly 已提交
717 718
  snprintf(tbFName, sizeof(tbFName), "%s.%s", alterReq->objname, alterReq->tabName);
  int32_t len = strlen(tbFName) + 1;
C
cademfly 已提交
719

X
Xiaoyu Wang 已提交
720
  if (alterReq->tagCond != NULL && alterReq->tagCondLen != 0) {
721
    char *value = taosHashGet(hash, tbFName, len);
X
Xiaoyu Wang 已提交
722
    if (value != NULL) {
723 724 725 726
      terrno = TSDB_CODE_MND_PRIVILEDGE_EXIST;
      return -1;
    }

X
Xiaoyu Wang 已提交
727
    int32_t condLen = alterReq->tagCondLen;
728 729 730
    if (taosHashPut(hash, tbFName, len, alterReq->tagCond, condLen) != 0) {
      return -1;
    }
X
Xiaoyu Wang 已提交
731
  } else {
732 733 734
    if (taosHashPut(hash, tbFName, len, "t", 2) != 0) {
      return -1;
    }
C
cademfly 已提交
735
  }
C
cademfly 已提交
736

X
Xiaoyu Wang 已提交
737 738 739 740 741 742 743 744 745 746
  int32_t  dbKeyLen = strlen(alterReq->objname) + 1;
  int32_t  ref = 1;
  int32_t *currRef = taosHashGet(useDbHash, alterReq->objname, dbKeyLen);
  if (NULL != currRef) {
    ref = (*currRef) + 1;
  }
  if (taosHashPut(useDbHash, alterReq->objname, dbKeyLen, &ref, sizeof(ref)) != 0) {
    return -1;
  }

C
cademfly 已提交
747 748 749
  return 0;
}

X
Xiaoyu Wang 已提交
750 751
static int32_t mndRemoveTablePriviledge(SMnode *pMnode, SHashObj *hash, SHashObj *useDbHash, SAlterUserReq *alterReq,
                                        SSdb *pSdb) {
X
Xiaoyu Wang 已提交
752 753
  void *pIter = NULL;
  char  tbFName[TSDB_TABLE_FNAME_LEN] = {0};
C
cademfly 已提交
754 755
  snprintf(tbFName, sizeof(tbFName), "%s.%s", alterReq->objname, alterReq->tabName);
  int32_t len = strlen(tbFName) + 1;
C
cademfly 已提交
756

C
cademfly 已提交
757 758 759
  if (taosHashRemove(hash, tbFName, len) != 0) {
    return -1;
  }
C
cademfly 已提交
760

X
Xiaoyu Wang 已提交
761 762 763 764 765 766 767 768 769 770 771 772 773
  int32_t  dbKeyLen = strlen(alterReq->objname) + 1;
  int32_t *currRef = taosHashGet(useDbHash, alterReq->objname, dbKeyLen);
  if (NULL == currRef || 1 == *currRef) {
    if (taosHashRemove(useDbHash, alterReq->objname, dbKeyLen) != 0) {
      return -1;
    }
    return 0;
  }
  int32_t ref = (*currRef) - 1;
  if (taosHashPut(useDbHash, alterReq->objname, dbKeyLen, &ref, sizeof(ref)) != 0) {
    return -1;
  }

C
cademfly 已提交
774 775 776
  return 0;
}

S
Shengliang Guan 已提交
777 778
static int32_t mndProcessAlterUserReq(SRpcMsg *pReq) {
  SMnode       *pMnode = pReq->info.node;
S
Shengliang Guan 已提交
779 780
  SSdb         *pSdb = pMnode->pSdb;
  void         *pIter = NULL;
S
Shengliang Guan 已提交
781 782 783
  int32_t       code = -1;
  SUserObj     *pUser = NULL;
  SUserObj     *pOperUser = NULL;
S
Shengliang Guan 已提交
784
  SUserObj      newUser = {0};
S
Shengliang Guan 已提交
785 786
  SAlterUserReq alterReq = {0};

S
Shengliang Guan 已提交
787
  if (tDeserializeSAlterUserReq(pReq->pCont, pReq->contLen, &alterReq) != 0) {
S
Shengliang Guan 已提交
788
    terrno = TSDB_CODE_INVALID_MSG;
789
    goto _OVER;
S
Shengliang Guan 已提交
790
  }
S
Shengliang Guan 已提交
791

792
  mInfo("user:%s, start to alter", alterReq.user);
S
Shengliang Guan 已提交
793

S
Shengliang Guan 已提交
794
  if (alterReq.user[0] == 0) {
S
Shengliang Guan 已提交
795
    terrno = TSDB_CODE_MND_INVALID_USER_FORMAT;
796 797 798 799 800
    goto _OVER;
  }

  if (TSDB_ALTER_USER_PASSWD == alterReq.alterType && alterReq.pass[0] == 0) {
    terrno = TSDB_CODE_MND_INVALID_PASS_FORMAT;
801
    goto _OVER;
S
Shengliang Guan 已提交
802 803
  }

S
Shengliang Guan 已提交
804
  pUser = mndAcquireUser(pMnode, alterReq.user);
S
Shengliang Guan 已提交
805 806
  if (pUser == NULL) {
    terrno = TSDB_CODE_MND_USER_NOT_EXIST;
807
    goto _OVER;
S
Shengliang Guan 已提交
808 809
  }

810
  pOperUser = mndAcquireUser(pMnode, pReq->info.conn.user);
S
Shengliang Guan 已提交
811 812
  if (pOperUser == NULL) {
    terrno = TSDB_CODE_MND_NO_USER_FROM_CONN;
813
    goto _OVER;
S
Shengliang Guan 已提交
814 815
  }

816
  if (mndCheckAlterUserPrivilege(pOperUser, pUser, &alterReq) != 0) {
S
Shengliang Guan 已提交
817 818 819
    goto _OVER;
  }

820
  if (mndUserDupObj(pUser, &newUser) != 0) goto _OVER;
S
Shengliang Guan 已提交
821 822 823 824

  if (alterReq.alterType == TSDB_ALTER_USER_PASSWD) {
    char pass[TSDB_PASSWORD_LEN + 1] = {0};
    taosEncryptPass_c((uint8_t *)alterReq.pass, strlen(alterReq.pass), pass);
825
    memcpy(newUser.pass, pass, TSDB_PASSWORD_LEN);
S
Shengliang Guan 已提交
826 827 828
  }

  if (alterReq.alterType == TSDB_ALTER_USER_SUPERUSER) {
S
Shengliang Guan 已提交
829
    newUser.superUser = alterReq.superUser;
S
Shengliang Guan 已提交
830 831
  }

832 833 834 835 836 837 838 839
  if (alterReq.alterType == TSDB_ALTER_USER_ENABLE) {
    newUser.enable = alterReq.enable;
  }

  if (alterReq.alterType == TSDB_ALTER_USER_SYSINFO) {
    newUser.sysInfo = alterReq.sysInfo;
  }

S
Shengliang Guan 已提交
840
  if (alterReq.alterType == TSDB_ALTER_USER_ADD_READ_DB || alterReq.alterType == TSDB_ALTER_USER_ADD_ALL_DB) {
841 842 843
    if (strcmp(alterReq.objname, "1.*") != 0) {
      int32_t len = strlen(alterReq.objname) + 1;
      SDbObj *pDb = mndAcquireDb(pMnode, alterReq.objname);
S
Shengliang Guan 已提交
844 845 846 847
      if (pDb == NULL) {
        mndReleaseDb(pMnode, pDb);
        goto _OVER;
      }
848
      if (taosHashPut(newUser.readDbs, alterReq.objname, len, alterReq.objname, TSDB_DB_FNAME_LEN) != 0) {
S
Shengliang Guan 已提交
849 850 851 852 853 854 855 856 857 858 859 860
        mndReleaseDb(pMnode, pDb);
        goto _OVER;
      }
    } else {
      while (1) {
        SDbObj *pDb = NULL;
        pIter = sdbFetch(pSdb, SDB_DB, pIter, (void **)&pDb);
        if (pIter == NULL) break;
        int32_t len = strlen(pDb->name) + 1;
        taosHashPut(newUser.readDbs, pDb->name, len, pDb->name, TSDB_DB_FNAME_LEN);
        sdbRelease(pSdb, pDb);
      }
S
Shengliang Guan 已提交
861
    }
S
Shengliang Guan 已提交
862 863 864
  }

  if (alterReq.alterType == TSDB_ALTER_USER_ADD_WRITE_DB || alterReq.alterType == TSDB_ALTER_USER_ADD_ALL_DB) {
865 866 867
    if (strcmp(alterReq.objname, "1.*") != 0) {
      int32_t len = strlen(alterReq.objname) + 1;
      SDbObj *pDb = mndAcquireDb(pMnode, alterReq.objname);
S
Shengliang Guan 已提交
868 869 870 871
      if (pDb == NULL) {
        mndReleaseDb(pMnode, pDb);
        goto _OVER;
      }
872
      if (taosHashPut(newUser.writeDbs, alterReq.objname, len, alterReq.objname, TSDB_DB_FNAME_LEN) != 0) {
S
Shengliang Guan 已提交
873 874 875 876 877 878 879 880 881 882 883 884
        mndReleaseDb(pMnode, pDb);
        goto _OVER;
      }
    } else {
      while (1) {
        SDbObj *pDb = NULL;
        pIter = sdbFetch(pSdb, SDB_DB, pIter, (void **)&pDb);
        if (pIter == NULL) break;
        int32_t len = strlen(pDb->name) + 1;
        taosHashPut(newUser.writeDbs, pDb->name, len, pDb->name, TSDB_DB_FNAME_LEN);
        sdbRelease(pSdb, pDb);
      }
S
Shengliang Guan 已提交
885 886 887
    }
  }

S
Shengliang Guan 已提交
888
  if (alterReq.alterType == TSDB_ALTER_USER_REMOVE_READ_DB || alterReq.alterType == TSDB_ALTER_USER_REMOVE_ALL_DB) {
889 890 891
    if (strcmp(alterReq.objname, "1.*") != 0) {
      int32_t len = strlen(alterReq.objname) + 1;
      SDbObj *pDb = mndAcquireDb(pMnode, alterReq.objname);
S
Shengliang Guan 已提交
892 893 894 895
      if (pDb == NULL) {
        mndReleaseDb(pMnode, pDb);
        goto _OVER;
      }
896
      taosHashRemove(newUser.readDbs, alterReq.objname, len);
S
Shengliang Guan 已提交
897 898 899 900
    } else {
      taosHashClear(newUser.readDbs);
    }
  }
S
Shengliang Guan 已提交
901

S
Shengliang Guan 已提交
902
  if (alterReq.alterType == TSDB_ALTER_USER_REMOVE_WRITE_DB || alterReq.alterType == TSDB_ALTER_USER_REMOVE_ALL_DB) {
903 904 905
    if (strcmp(alterReq.objname, "1.*") != 0) {
      int32_t len = strlen(alterReq.objname) + 1;
      SDbObj *pDb = mndAcquireDb(pMnode, alterReq.objname);
S
Shengliang Guan 已提交
906 907 908 909
      if (pDb == NULL) {
        mndReleaseDb(pMnode, pDb);
        goto _OVER;
      }
910
      taosHashRemove(newUser.writeDbs, alterReq.objname, len);
S
Shengliang Guan 已提交
911 912 913
    } else {
      taosHashClear(newUser.writeDbs);
    }
S
Shengliang Guan 已提交
914 915
  }

C
cademfly 已提交
916
  if (alterReq.alterType == TSDB_ALTER_USER_ADD_READ_TABLE) {
X
Xiaoyu Wang 已提交
917
    if (mndTablePriviledge(pMnode, newUser.readTbs, newUser.useDbs, &alterReq, pSdb) != 0) goto _OVER;
C
table  
cademfly 已提交
918 919
  }

C
cademfly 已提交
920
  if (alterReq.alterType == TSDB_ALTER_USER_ADD_WRITE_TABLE) {
X
Xiaoyu Wang 已提交
921
    if (mndTablePriviledge(pMnode, newUser.writeTbs, newUser.useDbs, &alterReq, pSdb) != 0) goto _OVER;
C
table  
cademfly 已提交
922 923
  }

X
Xiaoyu Wang 已提交
924 925
  if (alterReq.alterType == TSDB_ALTER_USER_REMOVE_READ_TABLE) {
    if (mndRemoveTablePriviledge(pMnode, newUser.readTbs, newUser.useDbs, &alterReq, pSdb) != 0) goto _OVER;
C
cademfly 已提交
926 927
  }

X
Xiaoyu Wang 已提交
928 929
  if (alterReq.alterType == TSDB_ALTER_USER_REMOVE_WRITE_TABLE) {
    if (mndRemoveTablePriviledge(pMnode, newUser.writeTbs, newUser.useDbs, &alterReq, pSdb) != 0) goto _OVER;
C
cademfly 已提交
930 931
  }

932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951
  if (alterReq.alterType == TSDB_ALTER_USER_ADD_SUBSCRIBE_TOPIC) {
    int32_t      len = strlen(alterReq.objname) + 1;
    SMqTopicObj *pTopic = mndAcquireTopic(pMnode, alterReq.objname);
    if (pTopic == NULL) {
      mndReleaseTopic(pMnode, pTopic);
      goto _OVER;
    }
    taosHashPut(newUser.topics, pTopic->name, len, pTopic->name, TSDB_TOPIC_FNAME_LEN);
  }

  if (alterReq.alterType == TSDB_ALTER_USER_REMOVE_SUBSCRIBE_TOPIC) {
    int32_t      len = strlen(alterReq.objname) + 1;
    SMqTopicObj *pTopic = mndAcquireTopic(pMnode, alterReq.objname);
    if (pTopic == NULL) {
      mndReleaseTopic(pMnode, pTopic);
      goto _OVER;
    }
    taosHashRemove(newUser.topics, alterReq.objname, len);
  }

S
Shengliang Guan 已提交
952
  code = mndAlterUser(pMnode, pUser, &newUser, pReq);
S
Shengliang Guan 已提交
953
  if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
S
Shengliang Guan 已提交
954

955
_OVER:
S
Shengliang Guan 已提交
956
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
S
Shengliang Guan 已提交
957
    mError("user:%s, failed to alter since %s", alterReq.user, terrstr());
S
Shengliang Guan 已提交
958 959
  }

X
Xiaoyu Wang 已提交
960
  tFreeSAlterUserReq(&alterReq);
S
Shengliang Guan 已提交
961 962
  mndReleaseUser(pMnode, pOperUser);
  mndReleaseUser(pMnode, pUser);
963
  mndUserFreeObj(&newUser);
S
Shengliang Guan 已提交
964 965

  return code;
S
Shengliang Guan 已提交
966 967
}

S
Shengliang Guan 已提交
968
static int32_t mndDropUser(SMnode *pMnode, SRpcMsg *pReq, SUserObj *pUser) {
969
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_ROLLBACK, TRN_CONFLICT_NOTHING, pReq, "drop-user");
S
Shengliang Guan 已提交
970 971 972 973
  if (pTrans == NULL) {
    mError("user:%s, failed to drop since %s", pUser->user, terrstr());
    return -1;
  }
974
  mInfo("trans:%d, used to drop user:%s", pTrans->id, pUser->user);
S
Shengliang Guan 已提交
975

976 977 978
  SSdbRaw *pCommitRaw = mndUserActionEncode(pUser);
  if (pCommitRaw == NULL || mndTransAppendCommitlog(pTrans, pCommitRaw) != 0) {
    mError("trans:%d, failed to append commit log since %s", pTrans->id, terrstr());
S
Shengliang Guan 已提交
979 980 981
    mndTransDrop(pTrans);
    return -1;
  }
S
Shengliang Guan 已提交
982
  (void)sdbSetRawStatus(pCommitRaw, SDB_STATUS_DROPPED);
S
Shengliang Guan 已提交
983 984 985 986 987 988 989 990 991 992 993

  if (mndTransPrepare(pMnode, pTrans) != 0) {
    mError("trans:%d, failed to prepare since %s", pTrans->id, terrstr());
    mndTransDrop(pTrans);
    return -1;
  }

  mndTransDrop(pTrans);
  return 0;
}

S
Shengliang Guan 已提交
994 995
static int32_t mndProcessDropUserReq(SRpcMsg *pReq) {
  SMnode      *pMnode = pReq->info.node;
S
Shengliang Guan 已提交
996 997 998 999
  int32_t      code = -1;
  SUserObj    *pUser = NULL;
  SDropUserReq dropReq = {0};

S
Shengliang Guan 已提交
1000
  if (tDeserializeSDropUserReq(pReq->pCont, pReq->contLen, &dropReq) != 0) {
S
Shengliang Guan 已提交
1001
    terrno = TSDB_CODE_INVALID_MSG;
1002
    goto _OVER;
S
Shengliang Guan 已提交
1003
  }
S
Shengliang Guan 已提交
1004

1005
  mInfo("user:%s, start to drop", dropReq.user);
1006 1007 1008
  if (mndCheckOperPrivilege(pMnode, pReq->info.conn.user, MND_OPER_DROP_USER) != 0) {
    goto _OVER;
  }
S
Shengliang Guan 已提交
1009

S
Shengliang Guan 已提交
1010
  if (dropReq.user[0] == 0) {
S
Shengliang Guan 已提交
1011
    terrno = TSDB_CODE_MND_INVALID_USER_FORMAT;
1012
    goto _OVER;
S
Shengliang Guan 已提交
1013 1014
  }

S
Shengliang Guan 已提交
1015
  pUser = mndAcquireUser(pMnode, dropReq.user);
S
Shengliang Guan 已提交
1016 1017
  if (pUser == NULL) {
    terrno = TSDB_CODE_MND_USER_NOT_EXIST;
1018
    goto _OVER;
S
Shengliang Guan 已提交
1019 1020
  }

S
Shengliang Guan 已提交
1021
  code = mndDropUser(pMnode, pReq, pUser);
S
Shengliang Guan 已提交
1022
  if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
S
Shengliang Guan 已提交
1023

1024
_OVER:
S
Shengliang Guan 已提交
1025
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
S
Shengliang Guan 已提交
1026
    mError("user:%s, failed to drop since %s", dropReq.user, terrstr());
S
Shengliang Guan 已提交
1027 1028
  }

S
Shengliang Guan 已提交
1029 1030
  mndReleaseUser(pMnode, pUser);
  return code;
S
Shengliang Guan 已提交
1031 1032
}

S
Shengliang Guan 已提交
1033 1034
static int32_t mndProcessGetUserAuthReq(SRpcMsg *pReq) {
  SMnode         *pMnode = pReq->info.node;
S
Shengliang Guan 已提交
1035 1036 1037 1038 1039
  int32_t         code = -1;
  SUserObj       *pUser = NULL;
  SGetUserAuthReq authReq = {0};
  SGetUserAuthRsp authRsp = {0};

S
Shengliang Guan 已提交
1040
  if (tDeserializeSGetUserAuthReq(pReq->pCont, pReq->contLen, &authReq) != 0) {
S
Shengliang Guan 已提交
1041
    terrno = TSDB_CODE_INVALID_MSG;
1042
    goto _OVER;
S
Shengliang Guan 已提交
1043
  }
S
Shengliang Guan 已提交
1044 1045 1046 1047 1048 1049

  mTrace("user:%s, start to get auth", authReq.user);

  pUser = mndAcquireUser(pMnode, authReq.user);
  if (pUser == NULL) {
    terrno = TSDB_CODE_MND_USER_NOT_EXIST;
1050
    goto _OVER;
S
Shengliang Guan 已提交
1051 1052
  }

D
dapan 已提交
1053 1054 1055
  code = mndSetUserAuthRsp(pMnode, pUser, &authRsp);
  if (code) {
    goto _OVER;
S
Shengliang Guan 已提交
1056 1057
  }

S
Shengliang Guan 已提交
1058
  int32_t contLen = tSerializeSGetUserAuthRsp(NULL, 0, &authRsp);
S
Shengliang Guan 已提交
1059 1060 1061
  void   *pRsp = rpcMallocCont(contLen);
  if (pRsp == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
1062
    goto _OVER;
S
Shengliang Guan 已提交
1063 1064
  }

S
Shengliang Guan 已提交
1065
  tSerializeSGetUserAuthRsp(pRsp, contLen, &authRsp);
S
Shengliang Guan 已提交
1066

S
Shengliang Guan 已提交
1067 1068
  pReq->info.rsp = pRsp;
  pReq->info.rspLen = contLen;
S
Shengliang Guan 已提交
1069 1070
  code = 0;

1071
_OVER:
1072

S
Shengliang Guan 已提交
1073
  mndReleaseUser(pMnode, pUser);
S
Shengliang Guan 已提交
1074
  tFreeSGetUserAuthRsp(&authRsp);
S
Shengliang Guan 已提交
1075 1076 1077 1078

  return code;
}

S
Shengliang Guan 已提交
1079 1080
static int32_t mndRetrieveUsers(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rows) {
  SMnode   *pMnode = pReq->info.node;
S
Shengliang Guan 已提交
1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091
  SSdb     *pSdb = pMnode->pSdb;
  int32_t   numOfRows = 0;
  SUserObj *pUser = NULL;
  int32_t   cols = 0;
  char     *pWrite;

  while (numOfRows < rows) {
    pShow->pIter = sdbFetch(pSdb, SDB_USER, pShow->pIter, (void **)&pUser);
    if (pShow->pIter == NULL) break;

    cols = 0;
1092
    SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols);
1093
    char             name[TSDB_USER_LEN + VARSTR_HEADER_SIZE] = {0};
1094
    STR_WITH_MAXSIZE_TO_VARSTR(name, pUser->user, pShow->pMeta->pSchemas[cols].bytes);
1095
    colDataSetVal(pColInfo, numOfRows, (const char *)name, false);
1096

wafwerar's avatar
wafwerar 已提交
1097 1098
    cols++;
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols);
1099
    colDataSetVal(pColInfo, numOfRows, (const char *)&pUser->superUser, false);
1100

1101 1102
    cols++;
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols);
1103
    colDataSetVal(pColInfo, numOfRows, (const char *)&pUser->enable, false);
1104 1105 1106

    cols++;
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols);
1107
    colDataSetVal(pColInfo, numOfRows, (const char *)&pUser->sysInfo, false);
1108

wafwerar's avatar
wafwerar 已提交
1109 1110
    cols++;
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols);
1111
    colDataSetVal(pColInfo, numOfRows, (const char *)&pUser->createdTime, false);
S
Shengliang Guan 已提交
1112 1113 1114 1115 1116

    numOfRows++;
    sdbRelease(pSdb, pUser);
  }

1117
  pShow->numOfRows += numOfRows;
S
Shengliang Guan 已提交
1118 1119 1120 1121 1122 1123
  return numOfRows;
}

static void mndCancelGetNextUser(SMnode *pMnode, void *pIter) {
  SSdb *pSdb = pMnode->pSdb;
  sdbCancelFetch(pSdb, pIter);
S
Shengliang Guan 已提交
1124
}
D
dapan 已提交
1125

X
Xiaoyu Wang 已提交
1126 1127 1128
static void mndLoopHash(SHashObj *hash, char *priType, SSDataBlock *pBlock, int32_t *numOfRows, char *user,
                        SShowObj *pShow) {
  char   *value = taosHashIterate(hash, NULL);
C
cademfly 已提交
1129
  int32_t cols = 0;
X
Xiaoyu Wang 已提交
1130

C
cademfly 已提交
1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141
  while (value != NULL) {
    cols = 0;
    char userName[TSDB_USER_LEN + VARSTR_HEADER_SIZE] = {0};
    STR_WITH_MAXSIZE_TO_VARSTR(userName, user, pShow->pMeta->pSchemas[cols].bytes);
    SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
    colDataSetVal(pColInfo, *numOfRows, (const char *)userName, false);

    char privilege[20] = {0};
    STR_WITH_MAXSIZE_TO_VARSTR(privilege, priType, pShow->pMeta->pSchemas[cols].bytes);
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
    colDataSetVal(pColInfo, *numOfRows, (const char *)privilege, false);
X
Xiaoyu Wang 已提交
1142

C
cademfly 已提交
1143 1144
    size_t keyLen = 0;
    void  *key = taosHashGetKey(value, &keyLen);
C
cademfly 已提交
1145

X
Xiaoyu Wang 已提交
1146
    char dbName[TSDB_DB_NAME_LEN] = {0};
C
cademfly 已提交
1147
    mndExtractShortDbNameFromStbFullName(key, dbName);
X
Xiaoyu Wang 已提交
1148
    char dbNameContent[TSDB_DB_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
C
cademfly 已提交
1149 1150 1151 1152
    STR_WITH_MAXSIZE_TO_VARSTR(dbNameContent, dbName, pShow->pMeta->pSchemas[cols].bytes);
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
    colDataSetVal(pColInfo, *numOfRows, (const char *)dbNameContent, false);

X
Xiaoyu Wang 已提交
1153
    char tableName[TSDB_TABLE_NAME_LEN] = {0};
C
cademfly 已提交
1154
    mndExtractTbNameFromStbFullName(key, tableName, TSDB_TABLE_NAME_LEN);
X
Xiaoyu Wang 已提交
1155
    char tableNameContent[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
C
cademfly 已提交
1156 1157 1158
    STR_WITH_MAXSIZE_TO_VARSTR(tableNameContent, tableName, pShow->pMeta->pSchemas[cols].bytes);
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
    colDataSetVal(pColInfo, *numOfRows, (const char *)tableNameContent, false);
C
cademfly 已提交
1159

X
Xiaoyu Wang 已提交
1160
    if (strcmp("t", value) != 0) {
C
cademfly 已提交
1161 1162
      SNode  *pAst = NULL;
      int32_t sqlLen = 0;
X
Xiaoyu Wang 已提交
1163
      char    sql[TSDB_EXPLAIN_RESULT_ROW_SIZE] = {0};
C
cademfly 已提交
1164

X
Xiaoyu Wang 已提交
1165
      if (nodesStringToNode(value, &pAst) == 0) {
C
cademfly 已提交
1166
        nodesNodeToSQL(pAst, sql, TSDB_EXPLAIN_RESULT_ROW_SIZE, &sqlLen);
C
cademfly 已提交
1167
        nodesDestroyNode(pAst);
X
Xiaoyu Wang 已提交
1168
      } else {
C
cademfly 已提交
1169 1170 1171 1172
        sqlLen = 5;
        sprintf(sql, "error");
      }

X
Xiaoyu Wang 已提交
1173
      char obj[TSDB_PRIVILEDGE_CONDITION_LEN + VARSTR_HEADER_SIZE] = {0};
C
cademfly 已提交
1174
      STR_WITH_MAXSIZE_TO_VARSTR(obj, sql, pShow->pMeta->pSchemas[cols].bytes);
C
cademfly 已提交
1175 1176 1177

      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
      colDataSetVal(pColInfo, *numOfRows, (const char *)obj, false);
X
Xiaoyu Wang 已提交
1178
    } else {
1179
      char condition[TSDB_PRIVILEDGE_CONDITION_LEN + VARSTR_HEADER_SIZE] = {0};
C
cademfly 已提交
1180 1181 1182 1183
      STR_WITH_MAXSIZE_TO_VARSTR(condition, "", pShow->pMeta->pSchemas[cols].bytes);
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
      colDataSetVal(pColInfo, *numOfRows, (const char *)condition, false);
    }
X
Xiaoyu Wang 已提交
1184

C
cademfly 已提交
1185 1186 1187 1188 1189
    (*numOfRows)++;
    value = taosHashIterate(hash, value);
  }
}

S
Shengliang Guan 已提交
1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204
static int32_t mndRetrievePrivileges(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rows) {
  SMnode   *pMnode = pReq->info.node;
  SSdb     *pSdb = pMnode->pSdb;
  int32_t   numOfRows = 0;
  SUserObj *pUser = NULL;
  int32_t   cols = 0;
  char     *pWrite;

  while (numOfRows < rows) {
    pShow->pIter = sdbFetch(pSdb, SDB_USER, pShow->pIter, (void **)&pUser);
    if (pShow->pIter == NULL) break;

    int32_t numOfReadDbs = taosHashGetSize(pUser->readDbs);
    int32_t numOfWriteDbs = taosHashGetSize(pUser->writeDbs);
    int32_t numOfTopics = taosHashGetSize(pUser->topics);
C
cademfly 已提交
1205 1206 1207
    int32_t numOfReadTbs = taosHashGetSize(pUser->readTbs);
    int32_t numOfWriteTbs = taosHashGetSize(pUser->writeTbs);
    if (numOfRows + numOfReadDbs + numOfWriteDbs + numOfTopics + numOfReadTbs + numOfWriteTbs >= rows) break;
S
Shengliang Guan 已提交
1208

1209 1210
    if (pUser->superUser) {
      cols = 0;
X
Xiaoyu Wang 已提交
1211
      char userName[TSDB_USER_LEN + VARSTR_HEADER_SIZE] = {0};
1212
      STR_WITH_MAXSIZE_TO_VARSTR(userName, pUser->user, pShow->pMeta->pSchemas[cols].bytes);
X
Xiaoyu Wang 已提交
1213
      SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1214
      colDataSetVal(pColInfo, numOfRows, (const char *)userName, false);
1215 1216 1217 1218

      char privilege[20] = {0};
      STR_WITH_MAXSIZE_TO_VARSTR(privilege, "all", pShow->pMeta->pSchemas[cols].bytes);
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1219
      colDataSetVal(pColInfo, numOfRows, (const char *)privilege, false);
1220 1221 1222 1223

      char objName[20] = {0};
      STR_WITH_MAXSIZE_TO_VARSTR(objName, "all", pShow->pMeta->pSchemas[cols].bytes);
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1224
      colDataSetVal(pColInfo, numOfRows, (const char *)objName, false);
1225

X
Xiaoyu Wang 已提交
1226
      char tableName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
C
cademfly 已提交
1227 1228 1229 1230
      STR_WITH_MAXSIZE_TO_VARSTR(tableName, "", pShow->pMeta->pSchemas[cols].bytes);
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
      colDataSetVal(pColInfo, numOfRows, (const char *)tableName, false);

X
Xiaoyu Wang 已提交
1231
      char condition[TSDB_PRIVILEDGE_CONDITION_LEN + VARSTR_HEADER_SIZE] = {0};
C
cademfly 已提交
1232 1233 1234 1235
      STR_WITH_MAXSIZE_TO_VARSTR(condition, "", pShow->pMeta->pSchemas[cols].bytes);
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
      colDataSetVal(pColInfo, numOfRows, (const char *)condition, false);

1236 1237 1238
      numOfRows++;
    }

S
Shengliang Guan 已提交
1239 1240 1241
    char *db = taosHashIterate(pUser->readDbs, NULL);
    while (db != NULL) {
      cols = 0;
X
Xiaoyu Wang 已提交
1242
      char userName[TSDB_USER_LEN + VARSTR_HEADER_SIZE] = {0};
S
Shengliang Guan 已提交
1243
      STR_WITH_MAXSIZE_TO_VARSTR(userName, pUser->user, pShow->pMeta->pSchemas[cols].bytes);
X
Xiaoyu Wang 已提交
1244
      SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1245
      colDataSetVal(pColInfo, numOfRows, (const char *)userName, false);
S
Shengliang Guan 已提交
1246 1247 1248 1249

      char privilege[20] = {0};
      STR_WITH_MAXSIZE_TO_VARSTR(privilege, "read", pShow->pMeta->pSchemas[cols].bytes);
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1250
      colDataSetVal(pColInfo, numOfRows, (const char *)privilege, false);
S
Shengliang Guan 已提交
1251 1252 1253 1254 1255 1256 1257

      SName name = {0};
      char  objName[TSDB_DB_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
      tNameFromString(&name, db, T_NAME_ACCT | T_NAME_DB);
      tNameGetDbName(&name, varDataVal(objName));
      varDataSetLen(objName, strlen(varDataVal(objName)));
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1258
      colDataSetVal(pColInfo, numOfRows, (const char *)objName, false);
S
Shengliang Guan 已提交
1259

1260
      char tableName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
C
cademfly 已提交
1261 1262 1263 1264
      STR_WITH_MAXSIZE_TO_VARSTR(tableName, "", pShow->pMeta->pSchemas[cols].bytes);
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
      colDataSetVal(pColInfo, numOfRows, (const char *)tableName, false);

1265
      char condition[TSDB_PRIVILEDGE_CONDITION_LEN + VARSTR_HEADER_SIZE] = {0};
C
cademfly 已提交
1266 1267 1268 1269
      STR_WITH_MAXSIZE_TO_VARSTR(condition, "", pShow->pMeta->pSchemas[cols].bytes);
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
      colDataSetVal(pColInfo, numOfRows, (const char *)condition, false);

S
Shengliang Guan 已提交
1270 1271 1272 1273 1274 1275 1276
      numOfRows++;
      db = taosHashIterate(pUser->readDbs, db);
    }

    db = taosHashIterate(pUser->writeDbs, NULL);
    while (db != NULL) {
      cols = 0;
1277
      char userName[TSDB_USER_LEN + VARSTR_HEADER_SIZE] = {0};
S
Shengliang Guan 已提交
1278
      STR_WITH_MAXSIZE_TO_VARSTR(userName, pUser->user, pShow->pMeta->pSchemas[cols].bytes);
1279
      SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1280
      colDataSetVal(pColInfo, numOfRows, (const char *)userName, false);
S
Shengliang Guan 已提交
1281 1282 1283 1284

      char privilege[20] = {0};
      STR_WITH_MAXSIZE_TO_VARSTR(privilege, "write", pShow->pMeta->pSchemas[cols].bytes);
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1285
      colDataSetVal(pColInfo, numOfRows, (const char *)privilege, false);
S
Shengliang Guan 已提交
1286 1287 1288 1289 1290 1291 1292

      SName name = {0};
      char  objName[TSDB_DB_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
      tNameFromString(&name, db, T_NAME_ACCT | T_NAME_DB);
      tNameGetDbName(&name, varDataVal(objName));
      varDataSetLen(objName, strlen(varDataVal(objName)));
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1293
      colDataSetVal(pColInfo, numOfRows, (const char *)objName, false);
S
Shengliang Guan 已提交
1294

1295
      char tableName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
C
cademfly 已提交
1296 1297 1298 1299
      STR_WITH_MAXSIZE_TO_VARSTR(tableName, "", pShow->pMeta->pSchemas[cols].bytes);
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
      colDataSetVal(pColInfo, numOfRows, (const char *)tableName, false);

1300
      char condition[TSDB_PRIVILEDGE_CONDITION_LEN + VARSTR_HEADER_SIZE] = {0};
C
cademfly 已提交
1301 1302 1303 1304
      STR_WITH_MAXSIZE_TO_VARSTR(condition, "", pShow->pMeta->pSchemas[cols].bytes);
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
      colDataSetVal(pColInfo, numOfRows, (const char *)condition, false);

S
Shengliang Guan 已提交
1305 1306 1307 1308
      numOfRows++;
      db = taosHashIterate(pUser->writeDbs, db);
    }

C
cademfly 已提交
1309
    mndLoopHash(pUser->readTbs, "read", pBlock, &numOfRows, pUser->user, pShow);
C
table  
cademfly 已提交
1310

C
cademfly 已提交
1311
    mndLoopHash(pUser->writeTbs, "write", pBlock, &numOfRows, pUser->user, pShow);
C
table  
cademfly 已提交
1312

S
Shengliang Guan 已提交
1313 1314 1315
    char *topic = taosHashIterate(pUser->topics, NULL);
    while (topic != NULL) {
      cols = 0;
1316
      char userName[TSDB_USER_LEN + VARSTR_HEADER_SIZE] = {0};
S
Shengliang Guan 已提交
1317
      STR_WITH_MAXSIZE_TO_VARSTR(userName, pUser->user, pShow->pMeta->pSchemas[cols].bytes);
1318
      SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1319
      colDataSetVal(pColInfo, numOfRows, (const char *)userName, false);
S
Shengliang Guan 已提交
1320 1321 1322 1323

      char privilege[20] = {0};
      STR_WITH_MAXSIZE_TO_VARSTR(privilege, "subscribe", pShow->pMeta->pSchemas[cols].bytes);
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1324
      colDataSetVal(pColInfo, numOfRows, (const char *)privilege, false);
S
Shengliang Guan 已提交
1325 1326 1327 1328 1329

      char topicName[TSDB_TOPIC_NAME_LEN + VARSTR_HEADER_SIZE + 5] = {0};
      tstrncpy(varDataVal(topicName), mndGetDbStr(topic), TSDB_TOPIC_NAME_LEN - 2);
      varDataSetLen(topicName, strlen(varDataVal(topicName)));
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1330
      colDataSetVal(pColInfo, numOfRows, (const char *)topicName, false);
S
Shengliang Guan 已提交
1331

1332
      char tableName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
C
cademfly 已提交
1333 1334 1335 1336
      STR_WITH_MAXSIZE_TO_VARSTR(tableName, "", pShow->pMeta->pSchemas[cols].bytes);
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
      colDataSetVal(pColInfo, numOfRows, (const char *)tableName, false);

1337
      char condition[TSDB_PRIVILEDGE_CONDITION_LEN + VARSTR_HEADER_SIZE] = {0};
C
cademfly 已提交
1338 1339 1340 1341
      STR_WITH_MAXSIZE_TO_VARSTR(condition, "", pShow->pMeta->pSchemas[cols].bytes);
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
      colDataSetVal(pColInfo, numOfRows, (const char *)condition, false);

S
Shengliang Guan 已提交
1342
      numOfRows++;
1343
      topic = taosHashIterate(pUser->topics, topic);
S
Shengliang Guan 已提交
1344
    }
1345

S
Shengliang Guan 已提交
1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356
    sdbRelease(pSdb, pUser);
  }

  pShow->numOfRows += numOfRows;
  return numOfRows;
}

static void mndCancelGetNextPrivileges(SMnode *pMnode, void *pIter) {
  SSdb *pSdb = pMnode->pSdb;
  sdbCancelFetch(pSdb, pIter);
}
1357

1358 1359
int32_t mndValidateUserAuthInfo(SMnode *pMnode, SUserAuthVersion *pUsers, int32_t numOfUses, void **ppRsp,
                                int32_t *pRspLen) {
D
dapan 已提交
1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374
  SUserAuthBatchRsp batchRsp = {0};
  batchRsp.pArray = taosArrayInit(numOfUses, sizeof(SGetUserAuthRsp));
  if (batchRsp.pArray == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return -1;
  }

  int32_t code = 0;
  for (int32_t i = 0; i < numOfUses; ++i) {
    SUserObj *pUser = mndAcquireUser(pMnode, pUsers[i].user);
    if (pUser == NULL) {
      mError("user:%s, failed to auth user since %s", pUsers[i].user, terrstr());
      continue;
    }

D
dapan1121 已提交
1375
    pUsers[i].version = ntohl(pUsers[i].version);
D
dapan 已提交
1376 1377 1378 1379
    if (pUser->authVersion <= pUsers[i].version) {
      mndReleaseUser(pMnode, pUser);
      continue;
    }
1380

D
dapan 已提交
1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394 1395
    SGetUserAuthRsp rsp = {0};
    code = mndSetUserAuthRsp(pMnode, pUser, &rsp);
    if (code) {
      mndReleaseUser(pMnode, pUser);
      tFreeSGetUserAuthRsp(&rsp);
      goto _OVER;
    }

    taosArrayPush(batchRsp.pArray, &rsp);
    mndReleaseUser(pMnode, pUser);
  }

  if (taosArrayGetSize(batchRsp.pArray) <= 0) {
    *ppRsp = NULL;
    *pRspLen = 0;
1396

D
dapan 已提交
1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419
    tFreeSUserAuthBatchRsp(&batchRsp);
    return 0;
  }

  int32_t rspLen = tSerializeSUserAuthBatchRsp(NULL, 0, &batchRsp);
  void   *pRsp = taosMemoryMalloc(rspLen);
  if (pRsp == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    tFreeSUserAuthBatchRsp(&batchRsp);
    return -1;
  }
  tSerializeSUserAuthBatchRsp(pRsp, rspLen, &batchRsp);

  *ppRsp = pRsp;
  *pRspLen = rspLen;

  tFreeSUserAuthBatchRsp(&batchRsp);
  return 0;

_OVER:

  *ppRsp = NULL;
  *pRspLen = 0;
1420

D
dapan 已提交
1421 1422 1423
  tFreeSUserAuthBatchRsp(&batchRsp);
  return code;
}
1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471

int32_t mndUserRemoveDb(SMnode *pMnode, STrans *pTrans, char *db) {
  int32_t   code = 0;
  SSdb     *pSdb = pMnode->pSdb;
  int32_t   len = strlen(db) + 1;
  void     *pIter = NULL;
  SUserObj *pUser = NULL;
  SUserObj  newUser = {0};

  while (1) {
    pIter = sdbFetch(pSdb, SDB_USER, pIter, (void **)&pUser);
    if (pIter == NULL) break;

    code = -1;
    if (mndUserDupObj(pUser, &newUser) != 0) break;

    bool inRead = (taosHashGet(newUser.readDbs, db, len) != NULL);
    bool inWrite = (taosHashGet(newUser.writeDbs, db, len) != NULL);
    if (inRead || inWrite) {
      (void)taosHashRemove(newUser.readDbs, db, len);
      (void)taosHashRemove(newUser.writeDbs, db, len);

      SSdbRaw *pCommitRaw = mndUserActionEncode(&newUser);
      if (pCommitRaw == NULL || mndTransAppendCommitlog(pTrans, pCommitRaw) != 0) break;
      (void)sdbSetRawStatus(pCommitRaw, SDB_STATUS_READY);
    }

    mndUserFreeObj(&newUser);
    sdbRelease(pSdb, pUser);
    code = 0;
  }

  if (pUser != NULL) sdbRelease(pSdb, pUser);
  if (pIter != NULL) sdbCancelFetch(pSdb, pIter);
  mndUserFreeObj(&newUser);
  return code;
}

int32_t mndUserRemoveTopic(SMnode *pMnode, STrans *pTrans, char *topic) {
  int32_t   code = 0;
  SSdb     *pSdb = pMnode->pSdb;
  int32_t   len = strlen(topic) + 1;
  void     *pIter = NULL;
  SUserObj *pUser = NULL;
  SUserObj  newUser = {0};

  while (1) {
    pIter = sdbFetch(pSdb, SDB_USER, pIter, (void **)&pUser);
1472 1473 1474
    if (pIter == NULL) {
      break;
    }
1475 1476

    code = -1;
1477 1478 1479
    if (mndUserDupObj(pUser, &newUser) != 0) {
      break;
    }
1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498

    bool inTopic = (taosHashGet(newUser.topics, topic, len) != NULL);
    if (inTopic) {
      (void)taosHashRemove(newUser.topics, topic, len);
      SSdbRaw *pCommitRaw = mndUserActionEncode(&newUser);
      if (pCommitRaw == NULL || mndTransAppendCommitlog(pTrans, pCommitRaw) != 0) break;
      (void)sdbSetRawStatus(pCommitRaw, SDB_STATUS_READY);
    }

    mndUserFreeObj(&newUser);
    sdbRelease(pSdb, pUser);
    code = 0;
  }

  if (pUser != NULL) sdbRelease(pSdb, pUser);
  if (pIter != NULL) sdbCancelFetch(pSdb, pIter);
  mndUserFreeObj(&newUser);
  return code;
}