mndUser.c 48.7 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 159
    size_t valueLen = 0;
    valueLen = strlen(stb);
    size += sizeof(int32_t);
    size += keyLen;
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 239 240 241
  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)
    useDb = taosHashIterate(pUser->writeTbs, useDb);
  }

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 372
      char *value = taosMemoryCalloc(valuelen, sizeof(char));
      memset(value, 0, keyLen);
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 393

    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));
    }
C
table  
cademfly 已提交
394 395
  }

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

  terrno = 0;

401
_OVER:
402
  if (terrno != 0) {
403 404 405 406 407
    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 已提交
408 409
      taosHashCleanup(pUser->readTbs);
      taosHashCleanup(pUser->writeTbs);
X
Xiaoyu Wang 已提交
410
      taosHashCleanup(pUser->useDbs);
411
    }
wafwerar's avatar
wafwerar 已提交
412
    taosMemoryFreeClear(pRow);
413 414
    return NULL;
  }
S
Shengliang Guan 已提交
415

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

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

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

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

C
cademfly 已提交
435 436 437 438 439 440 441 442 443 444 445
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 已提交
446
    char  *key = taosHashGetKey(tb, &keyLen);
C
cademfly 已提交
447 448 449 450 451 452 453 454 455 456 457 458 459 460

    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;
}

461 462 463 464 465 466 467 468
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 已提交
469 470
  pNew->readTbs = mndDupTableHash(pUser->readTbs);
  pNew->writeTbs = mndDupTableHash(pUser->writeTbs);
471
  pNew->topics = mndDupTopicHash(pUser->topics);
X
Xiaoyu Wang 已提交
472
  pNew->useDbs = mndDupDbHash(pUser->useDbs);
473 474 475 476 477 478 479 480 481
  taosRUnLockLatch(&pUser->lock);

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

static void mndUserFreeObj(SUserObj *pUser) {
482 483
  taosHashCleanup(pUser->readDbs);
  taosHashCleanup(pUser->writeDbs);
484
  taosHashCleanup(pUser->topics);
C
cademfly 已提交
485 486
  taosHashCleanup(pUser->readTbs);
  taosHashCleanup(pUser->writeTbs);
X
Xiaoyu Wang 已提交
487
  taosHashCleanup(pUser->useDbs);
488 489
  pUser->readDbs = NULL;
  pUser->writeDbs = NULL;
490
  pUser->topics = NULL;
C
cademfly 已提交
491 492
  pUser->readTbs = NULL;
  pUser->writeTbs = NULL;
X
Xiaoyu Wang 已提交
493
  pUser->useDbs = NULL;
494 495 496 497 498
}

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

S
Shengliang Guan 已提交
502
static int32_t mndUserActionUpdate(SSdb *pSdb, SUserObj *pOld, SUserObj *pNew) {
S
Shengliang Guan 已提交
503
  mTrace("user:%s, perform update action, old row:%p new row:%p", pOld->user, pOld, pNew);
S
Shengliang Guan 已提交
504
  taosWLockLatch(&pOld->lock);
S
Shengliang Guan 已提交
505
  pOld->updateTime = pNew->updateTime;
D
dapan1121 已提交
506
  pOld->authVersion = pNew->authVersion;
507 508
  pOld->sysInfo = pNew->sysInfo;
  pOld->enable = pNew->enable;
S
Shengliang Guan 已提交
509
  memcpy(pOld->pass, pNew->pass, TSDB_PASSWORD_LEN);
wafwerar's avatar
wafwerar 已提交
510 511
  TSWAP(pOld->readDbs, pNew->readDbs);
  TSWAP(pOld->writeDbs, pNew->writeDbs);
512
  TSWAP(pOld->topics, pNew->topics);
C
cademfly 已提交
513 514
  TSWAP(pOld->readTbs, pNew->readTbs);
  TSWAP(pOld->writeTbs, pNew->writeTbs);
X
Xiaoyu Wang 已提交
515
  TSWAP(pOld->useDbs, pNew->useDbs);
S
Shengliang Guan 已提交
516
  taosWUnLockLatch(&pOld->lock);
517

S
Shengliang Guan 已提交
518 519 520
  return 0;
}

521
SUserObj *mndAcquireUser(SMnode *pMnode, const char *userName) {
S
Shengliang Guan 已提交
522 523 524
  SSdb     *pSdb = pMnode->pSdb;
  SUserObj *pUser = sdbAcquire(pSdb, SDB_USER, userName);
  if (pUser == NULL) {
dengyihao's avatar
dengyihao 已提交
525 526 527 528 529
    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 已提交
530 531
  }
  return pUser;
S
Shengliang Guan 已提交
532
}
S
Shengliang Guan 已提交
533

S
Shengliang Guan 已提交
534 535 536
void mndReleaseUser(SMnode *pMnode, SUserObj *pUser) {
  SSdb *pSdb = pMnode->pSdb;
  sdbRelease(pSdb, pUser);
S
Shengliang Guan 已提交
537 538
}

S
Shengliang Guan 已提交
539
static int32_t mndCreateUser(SMnode *pMnode, char *acct, SCreateUserReq *pCreate, SRpcMsg *pReq) {
S
Shengliang Guan 已提交
540
  SUserObj userObj = {0};
S
Shengliang Guan 已提交
541 542
  taosEncryptPass_c((uint8_t *)pCreate->pass, strlen(pCreate->pass), userObj.pass);
  tstrncpy(userObj.user, pCreate->user, TSDB_USER_LEN);
S
Shengliang Guan 已提交
543 544 545
  tstrncpy(userObj.acct, acct, TSDB_USER_LEN);
  userObj.createdTime = taosGetTimestampMs();
  userObj.updateTime = userObj.createdTime;
546
  userObj.superUser = 0;  // pCreate->superUser;
547 548
  userObj.sysInfo = pCreate->sysInfo;
  userObj.enable = pCreate->enable;
S
Shengliang Guan 已提交
549

550
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_ROLLBACK, TRN_CONFLICT_NOTHING, pReq, "create-user");
S
Shengliang Guan 已提交
551
  if (pTrans == NULL) {
S
Shengliang Guan 已提交
552
    mError("user:%s, failed to create since %s", pCreate->user, terrstr());
S
Shengliang Guan 已提交
553 554
    return -1;
  }
555
  mInfo("trans:%d, used to create user:%s", pTrans->id, pCreate->user);
S
Shengliang Guan 已提交
556

557 558 559
  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 已提交
560
    mndTransDrop(pTrans);
S
Shengliang Guan 已提交
561
    return -1;
S
Shengliang Guan 已提交
562
  }
S
Shengliang Guan 已提交
563
  (void)sdbSetRawStatus(pCommitRaw, SDB_STATUS_READY);
S
Shengliang Guan 已提交
564

S
Shengliang Guan 已提交
565
  if (mndTransPrepare(pMnode, pTrans) != 0) {
S
Shengliang Guan 已提交
566
    mError("trans:%d, failed to prepare since %s", pTrans->id, terrstr());
S
Shengliang Guan 已提交
567
    mndTransDrop(pTrans);
S
Shengliang Guan 已提交
568
    return -1;
S
Shengliang Guan 已提交
569 570
  }

S
Shengliang Guan 已提交
571
  mndTransDrop(pTrans);
S
Shengliang Guan 已提交
572
  return 0;
S
Shengliang Guan 已提交
573 574
}

S
Shengliang Guan 已提交
575 576
static int32_t mndProcessCreateUserReq(SRpcMsg *pReq) {
  SMnode        *pMnode = pReq->info.node;
S
Shengliang Guan 已提交
577 578 579 580 581
  int32_t        code = -1;
  SUserObj      *pUser = NULL;
  SUserObj      *pOperUser = NULL;
  SCreateUserReq createReq = {0};

S
Shengliang Guan 已提交
582
  if (tDeserializeSCreateUserReq(pReq->pCont, pReq->contLen, &createReq) != 0) {
S
Shengliang Guan 已提交
583
    terrno = TSDB_CODE_INVALID_MSG;
584
    goto _OVER;
S
Shengliang Guan 已提交
585
  }
S
Shengliang Guan 已提交
586

587
  mInfo("user:%s, start to create", createReq.user);
588 589 590
  if (mndCheckOperPrivilege(pMnode, pReq->info.conn.user, MND_OPER_CREATE_USER) != 0) {
    goto _OVER;
  }
S
Shengliang Guan 已提交
591

S
Shengliang Guan 已提交
592
  if (createReq.user[0] == 0) {
S
Shengliang Guan 已提交
593
    terrno = TSDB_CODE_MND_INVALID_USER_FORMAT;
594
    goto _OVER;
S
Shengliang Guan 已提交
595 596
  }

S
Shengliang Guan 已提交
597
  if (createReq.pass[0] == 0) {
S
Shengliang Guan 已提交
598
    terrno = TSDB_CODE_MND_INVALID_PASS_FORMAT;
599
    goto _OVER;
S
Shengliang Guan 已提交
600 601
  }

S
Shengliang Guan 已提交
602
  pUser = mndAcquireUser(pMnode, createReq.user);
S
Shengliang Guan 已提交
603
  if (pUser != NULL) {
S
Shengliang Guan 已提交
604
    terrno = TSDB_CODE_MND_USER_ALREADY_EXIST;
605
    goto _OVER;
S
Shengliang Guan 已提交
606 607
  }

608
  pOperUser = mndAcquireUser(pMnode, pReq->info.conn.user);
S
Shengliang Guan 已提交
609
  if (pOperUser == NULL) {
S
Shengliang Guan 已提交
610
    terrno = TSDB_CODE_MND_NO_USER_FROM_CONN;
611
    goto _OVER;
S
Shengliang Guan 已提交
612 613
  }

C
Cary Xu 已提交
614 615 616 617
  if ((terrno = grantCheck(TSDB_GRANT_USER)) != 0) {
    code = terrno;
    goto _OVER;
  }
618

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

622
_OVER:
S
Shengliang Guan 已提交
623
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
S
Shengliang Guan 已提交
624
    mError("user:%s, failed to create since %s", createReq.user, terrstr());
S
Shengliang Guan 已提交
625 626
  }

S
Shengliang Guan 已提交
627 628 629 630
  mndReleaseUser(pMnode, pUser);
  mndReleaseUser(pMnode, pOperUser);

  return code;
S
Shengliang Guan 已提交
631 632
}

S
Shengliang Guan 已提交
633
static int32_t mndAlterUser(SMnode *pMnode, SUserObj *pOld, SUserObj *pNew, SRpcMsg *pReq) {
634
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_ROLLBACK, TRN_CONFLICT_NOTHING, pReq, "alter-user");
S
Shengliang Guan 已提交
635
  if (pTrans == NULL) {
S
Shengliang Guan 已提交
636
    mError("user:%s, failed to alter since %s", pOld->user, terrstr());
S
Shengliang Guan 已提交
637 638
    return -1;
  }
639
  mInfo("trans:%d, used to alter user:%s", pTrans->id, pOld->user);
S
Shengliang Guan 已提交
640

641 642 643
  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 已提交
644 645 646
    mndTransDrop(pTrans);
    return -1;
  }
S
Shengliang Guan 已提交
647
  (void)sdbSetRawStatus(pCommitRaw, SDB_STATUS_READY);
S
Shengliang Guan 已提交
648 649 650 651 652 653 654 655 656 657 658

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

  mndTransDrop(pTrans);
  return 0;
}

659
SHashObj *mndDupObjHash(SHashObj *pOld, int32_t dataLen) {
S
Shengliang Guan 已提交
660 661
  SHashObj *pNew =
      taosHashInit(taosHashGetSize(pOld), taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_ENTRY_LOCK);
S
Shengliang Guan 已提交
662 663 664 665 666 667 668 669
  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;
670
    if (taosHashPut(pNew, db, len, db, dataLen) != 0) {
S
Shengliang Guan 已提交
671 672
      taosHashCancelIterate(pOld, db);
      taosHashCleanup(pNew);
S
Shengliang Guan 已提交
673
      terrno = TSDB_CODE_OUT_OF_MEMORY;
S
Shengliang Guan 已提交
674 675 676 677 678 679 680 681
      return NULL;
    }
    db = taosHashIterate(pOld, db);
  }

  return pNew;
}

682 683 684 685
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 已提交
686 687
static int32_t mndTablePriviledge(SMnode *pMnode, SHashObj *hash, SHashObj *useDbHash, SAlterUserReq *alterReq,
                                  SSdb *pSdb) {
X
Xiaoyu Wang 已提交
688 689
  void *pIter = NULL;
  char  tbFName[TSDB_TABLE_FNAME_LEN] = {0};
C
cademfly 已提交
690

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

X
Xiaoyu Wang 已提交
694
  if (alterReq->tagCond != NULL && alterReq->tagCondLen != 0) {
695
    char *value = taosHashGet(hash, tbFName, len);
X
Xiaoyu Wang 已提交
696
    if (value != NULL) {
697 698 699 700
      terrno = TSDB_CODE_MND_PRIVILEDGE_EXIST;
      return -1;
    }

X
Xiaoyu Wang 已提交
701
    int32_t condLen = alterReq->tagCondLen;
702 703 704
    if (taosHashPut(hash, tbFName, len, alterReq->tagCond, condLen) != 0) {
      return -1;
    }
X
Xiaoyu Wang 已提交
705
  } else {
706 707 708
    if (taosHashPut(hash, tbFName, len, "t", 2) != 0) {
      return -1;
    }
C
cademfly 已提交
709
  }
C
cademfly 已提交
710

X
Xiaoyu Wang 已提交
711 712 713 714 715 716 717 718 719 720
  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 已提交
721 722 723
  return 0;
}

X
Xiaoyu Wang 已提交
724 725
static int32_t mndRemoveTablePriviledge(SMnode *pMnode, SHashObj *hash, SHashObj *useDbHash, SAlterUserReq *alterReq,
                                        SSdb *pSdb) {
X
Xiaoyu Wang 已提交
726 727
  void *pIter = NULL;
  char  tbFName[TSDB_TABLE_FNAME_LEN] = {0};
C
cademfly 已提交
728 729
  snprintf(tbFName, sizeof(tbFName), "%s.%s", alterReq->objname, alterReq->tabName);
  int32_t len = strlen(tbFName) + 1;
C
cademfly 已提交
730

C
cademfly 已提交
731 732 733
  if (taosHashRemove(hash, tbFName, len) != 0) {
    return -1;
  }
C
cademfly 已提交
734

X
Xiaoyu Wang 已提交
735 736 737 738 739 740 741 742 743 744 745 746 747
  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 已提交
748 749 750
  return 0;
}

S
Shengliang Guan 已提交
751 752
static int32_t mndProcessAlterUserReq(SRpcMsg *pReq) {
  SMnode       *pMnode = pReq->info.node;
S
Shengliang Guan 已提交
753 754
  SSdb         *pSdb = pMnode->pSdb;
  void         *pIter = NULL;
S
Shengliang Guan 已提交
755 756 757
  int32_t       code = -1;
  SUserObj     *pUser = NULL;
  SUserObj     *pOperUser = NULL;
S
Shengliang Guan 已提交
758
  SUserObj      newUser = {0};
S
Shengliang Guan 已提交
759 760
  SAlterUserReq alterReq = {0};

S
Shengliang Guan 已提交
761
  if (tDeserializeSAlterUserReq(pReq->pCont, pReq->contLen, &alterReq) != 0) {
S
Shengliang Guan 已提交
762
    terrno = TSDB_CODE_INVALID_MSG;
763
    goto _OVER;
S
Shengliang Guan 已提交
764
  }
S
Shengliang Guan 已提交
765

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

S
Shengliang Guan 已提交
768
  if (alterReq.user[0] == 0) {
S
Shengliang Guan 已提交
769
    terrno = TSDB_CODE_MND_INVALID_USER_FORMAT;
770 771 772 773 774
    goto _OVER;
  }

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

S
Shengliang Guan 已提交
778
  pUser = mndAcquireUser(pMnode, alterReq.user);
S
Shengliang Guan 已提交
779 780
  if (pUser == NULL) {
    terrno = TSDB_CODE_MND_USER_NOT_EXIST;
781
    goto _OVER;
S
Shengliang Guan 已提交
782 783
  }

784
  pOperUser = mndAcquireUser(pMnode, pReq->info.conn.user);
S
Shengliang Guan 已提交
785 786
  if (pOperUser == NULL) {
    terrno = TSDB_CODE_MND_NO_USER_FROM_CONN;
787
    goto _OVER;
S
Shengliang Guan 已提交
788 789
  }

790
  if (mndCheckAlterUserPrivilege(pOperUser, pUser, &alterReq) != 0) {
S
Shengliang Guan 已提交
791 792 793
    goto _OVER;
  }

794
  if (mndUserDupObj(pUser, &newUser) != 0) goto _OVER;
S
Shengliang Guan 已提交
795 796 797 798

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

  if (alterReq.alterType == TSDB_ALTER_USER_SUPERUSER) {
S
Shengliang Guan 已提交
803
    newUser.superUser = alterReq.superUser;
S
Shengliang Guan 已提交
804 805
  }

806 807 808 809 810 811 812 813
  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 已提交
814
  if (alterReq.alterType == TSDB_ALTER_USER_ADD_READ_DB || alterReq.alterType == TSDB_ALTER_USER_ADD_ALL_DB) {
815 816 817
    if (strcmp(alterReq.objname, "1.*") != 0) {
      int32_t len = strlen(alterReq.objname) + 1;
      SDbObj *pDb = mndAcquireDb(pMnode, alterReq.objname);
S
Shengliang Guan 已提交
818 819 820 821
      if (pDb == NULL) {
        mndReleaseDb(pMnode, pDb);
        goto _OVER;
      }
822
      if (taosHashPut(newUser.readDbs, alterReq.objname, len, alterReq.objname, TSDB_DB_FNAME_LEN) != 0) {
S
Shengliang Guan 已提交
823 824 825 826 827 828 829 830 831 832 833 834
        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 已提交
835
    }
S
Shengliang Guan 已提交
836 837 838
  }

  if (alterReq.alterType == TSDB_ALTER_USER_ADD_WRITE_DB || alterReq.alterType == TSDB_ALTER_USER_ADD_ALL_DB) {
839 840 841
    if (strcmp(alterReq.objname, "1.*") != 0) {
      int32_t len = strlen(alterReq.objname) + 1;
      SDbObj *pDb = mndAcquireDb(pMnode, alterReq.objname);
S
Shengliang Guan 已提交
842 843 844 845
      if (pDb == NULL) {
        mndReleaseDb(pMnode, pDb);
        goto _OVER;
      }
846
      if (taosHashPut(newUser.writeDbs, alterReq.objname, len, alterReq.objname, TSDB_DB_FNAME_LEN) != 0) {
S
Shengliang Guan 已提交
847 848 849 850 851 852 853 854 855 856 857 858
        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 已提交
859 860 861
    }
  }

S
Shengliang Guan 已提交
862
  if (alterReq.alterType == TSDB_ALTER_USER_REMOVE_READ_DB || alterReq.alterType == TSDB_ALTER_USER_REMOVE_ALL_DB) {
863 864 865
    if (strcmp(alterReq.objname, "1.*") != 0) {
      int32_t len = strlen(alterReq.objname) + 1;
      SDbObj *pDb = mndAcquireDb(pMnode, alterReq.objname);
S
Shengliang Guan 已提交
866 867 868 869
      if (pDb == NULL) {
        mndReleaseDb(pMnode, pDb);
        goto _OVER;
      }
870
      taosHashRemove(newUser.readDbs, alterReq.objname, len);
S
Shengliang Guan 已提交
871 872 873 874
    } else {
      taosHashClear(newUser.readDbs);
    }
  }
S
Shengliang Guan 已提交
875

S
Shengliang Guan 已提交
876
  if (alterReq.alterType == TSDB_ALTER_USER_REMOVE_WRITE_DB || alterReq.alterType == TSDB_ALTER_USER_REMOVE_ALL_DB) {
877 878 879
    if (strcmp(alterReq.objname, "1.*") != 0) {
      int32_t len = strlen(alterReq.objname) + 1;
      SDbObj *pDb = mndAcquireDb(pMnode, alterReq.objname);
S
Shengliang Guan 已提交
880 881 882 883
      if (pDb == NULL) {
        mndReleaseDb(pMnode, pDb);
        goto _OVER;
      }
884
      taosHashRemove(newUser.writeDbs, alterReq.objname, len);
S
Shengliang Guan 已提交
885 886 887
    } else {
      taosHashClear(newUser.writeDbs);
    }
S
Shengliang Guan 已提交
888 889
  }

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

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

X
Xiaoyu Wang 已提交
898 899
  if (alterReq.alterType == TSDB_ALTER_USER_REMOVE_READ_TABLE) {
    if (mndRemoveTablePriviledge(pMnode, newUser.readTbs, newUser.useDbs, &alterReq, pSdb) != 0) goto _OVER;
C
cademfly 已提交
900 901
  }

X
Xiaoyu Wang 已提交
902 903
  if (alterReq.alterType == TSDB_ALTER_USER_REMOVE_WRITE_TABLE) {
    if (mndRemoveTablePriviledge(pMnode, newUser.writeTbs, newUser.useDbs, &alterReq, pSdb) != 0) goto _OVER;
C
cademfly 已提交
904 905
  }

906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925
  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 已提交
926
  code = mndAlterUser(pMnode, pUser, &newUser, pReq);
S
Shengliang Guan 已提交
927
  if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
S
Shengliang Guan 已提交
928

929
_OVER:
S
Shengliang Guan 已提交
930
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
S
Shengliang Guan 已提交
931
    mError("user:%s, failed to alter since %s", alterReq.user, terrstr());
S
Shengliang Guan 已提交
932 933
  }

S
Shengliang Guan 已提交
934 935
  mndReleaseUser(pMnode, pOperUser);
  mndReleaseUser(pMnode, pUser);
936
  mndUserFreeObj(&newUser);
S
Shengliang Guan 已提交
937 938

  return code;
S
Shengliang Guan 已提交
939 940
}

S
Shengliang Guan 已提交
941
static int32_t mndDropUser(SMnode *pMnode, SRpcMsg *pReq, SUserObj *pUser) {
942
  STrans *pTrans = mndTransCreate(pMnode, TRN_POLICY_ROLLBACK, TRN_CONFLICT_NOTHING, pReq, "drop-user");
S
Shengliang Guan 已提交
943 944 945 946
  if (pTrans == NULL) {
    mError("user:%s, failed to drop since %s", pUser->user, terrstr());
    return -1;
  }
947
  mInfo("trans:%d, used to drop user:%s", pTrans->id, pUser->user);
S
Shengliang Guan 已提交
948

949 950 951
  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 已提交
952 953 954
    mndTransDrop(pTrans);
    return -1;
  }
S
Shengliang Guan 已提交
955
  (void)sdbSetRawStatus(pCommitRaw, SDB_STATUS_DROPPED);
S
Shengliang Guan 已提交
956 957 958 959 960 961 962 963 964 965 966

  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 已提交
967 968
static int32_t mndProcessDropUserReq(SRpcMsg *pReq) {
  SMnode      *pMnode = pReq->info.node;
S
Shengliang Guan 已提交
969 970 971 972
  int32_t      code = -1;
  SUserObj    *pUser = NULL;
  SDropUserReq dropReq = {0};

S
Shengliang Guan 已提交
973
  if (tDeserializeSDropUserReq(pReq->pCont, pReq->contLen, &dropReq) != 0) {
S
Shengliang Guan 已提交
974
    terrno = TSDB_CODE_INVALID_MSG;
975
    goto _OVER;
S
Shengliang Guan 已提交
976
  }
S
Shengliang Guan 已提交
977

978
  mInfo("user:%s, start to drop", dropReq.user);
979 980 981
  if (mndCheckOperPrivilege(pMnode, pReq->info.conn.user, MND_OPER_DROP_USER) != 0) {
    goto _OVER;
  }
S
Shengliang Guan 已提交
982

S
Shengliang Guan 已提交
983
  if (dropReq.user[0] == 0) {
S
Shengliang Guan 已提交
984
    terrno = TSDB_CODE_MND_INVALID_USER_FORMAT;
985
    goto _OVER;
S
Shengliang Guan 已提交
986 987
  }

S
Shengliang Guan 已提交
988
  pUser = mndAcquireUser(pMnode, dropReq.user);
S
Shengliang Guan 已提交
989 990
  if (pUser == NULL) {
    terrno = TSDB_CODE_MND_USER_NOT_EXIST;
991
    goto _OVER;
S
Shengliang Guan 已提交
992 993
  }

S
Shengliang Guan 已提交
994
  code = mndDropUser(pMnode, pReq, pUser);
S
Shengliang Guan 已提交
995
  if (code == 0) code = TSDB_CODE_ACTION_IN_PROGRESS;
S
Shengliang Guan 已提交
996

997
_OVER:
S
Shengliang Guan 已提交
998
  if (code != 0 && code != TSDB_CODE_ACTION_IN_PROGRESS) {
S
Shengliang Guan 已提交
999
    mError("user:%s, failed to drop since %s", dropReq.user, terrstr());
S
Shengliang Guan 已提交
1000 1001
  }

S
Shengliang Guan 已提交
1002 1003
  mndReleaseUser(pMnode, pUser);
  return code;
S
Shengliang Guan 已提交
1004 1005
}

S
Shengliang Guan 已提交
1006 1007
static int32_t mndProcessGetUserAuthReq(SRpcMsg *pReq) {
  SMnode         *pMnode = pReq->info.node;
S
Shengliang Guan 已提交
1008 1009 1010 1011 1012
  int32_t         code = -1;
  SUserObj       *pUser = NULL;
  SGetUserAuthReq authReq = {0};
  SGetUserAuthRsp authRsp = {0};

S
Shengliang Guan 已提交
1013
  if (tDeserializeSGetUserAuthReq(pReq->pCont, pReq->contLen, &authReq) != 0) {
S
Shengliang Guan 已提交
1014
    terrno = TSDB_CODE_INVALID_MSG;
1015
    goto _OVER;
S
Shengliang Guan 已提交
1016
  }
S
Shengliang Guan 已提交
1017 1018 1019 1020 1021 1022

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

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

D
dapan 已提交
1026 1027 1028
  code = mndSetUserAuthRsp(pMnode, pUser, &authRsp);
  if (code) {
    goto _OVER;
S
Shengliang Guan 已提交
1029 1030
  }

S
Shengliang Guan 已提交
1031
  int32_t contLen = tSerializeSGetUserAuthRsp(NULL, 0, &authRsp);
S
Shengliang Guan 已提交
1032 1033 1034
  void   *pRsp = rpcMallocCont(contLen);
  if (pRsp == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
1035
    goto _OVER;
S
Shengliang Guan 已提交
1036 1037
  }

S
Shengliang Guan 已提交
1038
  tSerializeSGetUserAuthRsp(pRsp, contLen, &authRsp);
S
Shengliang Guan 已提交
1039

S
Shengliang Guan 已提交
1040 1041
  pReq->info.rsp = pRsp;
  pReq->info.rspLen = contLen;
S
Shengliang Guan 已提交
1042 1043
  code = 0;

1044
_OVER:
1045

S
Shengliang Guan 已提交
1046
  mndReleaseUser(pMnode, pUser);
S
Shengliang Guan 已提交
1047
  tFreeSGetUserAuthRsp(&authRsp);
S
Shengliang Guan 已提交
1048 1049 1050 1051

  return code;
}

S
Shengliang Guan 已提交
1052 1053
static int32_t mndRetrieveUsers(SRpcMsg *pReq, SShowObj *pShow, SSDataBlock *pBlock, int32_t rows) {
  SMnode   *pMnode = pReq->info.node;
S
Shengliang Guan 已提交
1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064
  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;
1065
    SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols);
1066
    char             name[TSDB_USER_LEN + VARSTR_HEADER_SIZE] = {0};
1067
    STR_WITH_MAXSIZE_TO_VARSTR(name, pUser->user, pShow->pMeta->pSchemas[cols].bytes);
1068
    colDataSetVal(pColInfo, numOfRows, (const char *)name, false);
1069

wafwerar's avatar
wafwerar 已提交
1070 1071
    cols++;
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols);
1072
    colDataSetVal(pColInfo, numOfRows, (const char *)&pUser->superUser, false);
1073

1074 1075
    cols++;
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols);
1076
    colDataSetVal(pColInfo, numOfRows, (const char *)&pUser->enable, false);
1077 1078 1079

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

wafwerar's avatar
wafwerar 已提交
1082 1083
    cols++;
    pColInfo = taosArrayGet(pBlock->pDataBlock, cols);
1084
    colDataSetVal(pColInfo, numOfRows, (const char *)&pUser->createdTime, false);
S
Shengliang Guan 已提交
1085 1086 1087 1088 1089

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

1090
  pShow->numOfRows += numOfRows;
S
Shengliang Guan 已提交
1091 1092 1093 1094 1095 1096
  return numOfRows;
}

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

X
Xiaoyu Wang 已提交
1099 1100 1101
static void mndLoopHash(SHashObj *hash, char *priType, SSDataBlock *pBlock, int32_t *numOfRows, char *user,
                        SShowObj *pShow) {
  char   *value = taosHashIterate(hash, NULL);
C
cademfly 已提交
1102
  int32_t cols = 0;
X
Xiaoyu Wang 已提交
1103

C
cademfly 已提交
1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114
  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 已提交
1115

C
cademfly 已提交
1116 1117
    size_t keyLen = 0;
    void  *key = taosHashGetKey(value, &keyLen);
C
cademfly 已提交
1118

X
Xiaoyu Wang 已提交
1119
    char dbName[TSDB_DB_NAME_LEN] = {0};
C
cademfly 已提交
1120
    mndExtractShortDbNameFromStbFullName(key, dbName);
X
Xiaoyu Wang 已提交
1121
    char dbNameContent[TSDB_DB_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
C
cademfly 已提交
1122 1123 1124 1125
    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 已提交
1126
    char tableName[TSDB_TABLE_NAME_LEN] = {0};
C
cademfly 已提交
1127
    mndExtractTbNameFromStbFullName(key, tableName, TSDB_TABLE_NAME_LEN);
X
Xiaoyu Wang 已提交
1128
    char tableNameContent[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
C
cademfly 已提交
1129 1130 1131
    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 已提交
1132

X
Xiaoyu Wang 已提交
1133
    if (strcmp("t", value) != 0) {
C
cademfly 已提交
1134 1135
      SNode  *pAst = NULL;
      int32_t sqlLen = 0;
X
Xiaoyu Wang 已提交
1136
      char    sql[TSDB_EXPLAIN_RESULT_ROW_SIZE] = {0};
C
cademfly 已提交
1137

X
Xiaoyu Wang 已提交
1138
      if (nodesStringToNode(value, &pAst) == 0) {
C
cademfly 已提交
1139
        nodesNodeToSQL(pAst, sql, TSDB_EXPLAIN_RESULT_ROW_SIZE, &sqlLen);
C
cademfly 已提交
1140
        nodesDestroyNode(pAst);
X
Xiaoyu Wang 已提交
1141
      } else {
C
cademfly 已提交
1142 1143 1144 1145
        sqlLen = 5;
        sprintf(sql, "error");
      }

X
Xiaoyu Wang 已提交
1146 1147
      // char *obj = taosMemoryMalloc(sqlLen + VARSTR_HEADER_SIZE + 1);
      char obj[TSDB_PRIVILEDGE_CONDITION_LEN + VARSTR_HEADER_SIZE] = {0};
C
cademfly 已提交
1148
      STR_WITH_MAXSIZE_TO_VARSTR(obj, sql, pShow->pMeta->pSchemas[cols].bytes);
C
cademfly 已提交
1149 1150 1151

      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
      colDataSetVal(pColInfo, *numOfRows, (const char *)obj, false);
X
Xiaoyu Wang 已提交
1152 1153
      // taosMemoryFree(obj);
    } else {
C
cademfly 已提交
1154 1155 1156 1157 1158
      char condition[20] = {0};
      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 已提交
1159

C
cademfly 已提交
1160 1161 1162 1163 1164
    (*numOfRows)++;
    value = taosHashIterate(hash, value);
  }
}

S
Shengliang Guan 已提交
1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179
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 已提交
1180 1181 1182
    int32_t numOfReadTbs = taosHashGetSize(pUser->readTbs);
    int32_t numOfWriteTbs = taosHashGetSize(pUser->writeTbs);
    if (numOfRows + numOfReadDbs + numOfWriteDbs + numOfTopics + numOfReadTbs + numOfWriteTbs >= rows) break;
S
Shengliang Guan 已提交
1183

1184 1185
    if (pUser->superUser) {
      cols = 0;
X
Xiaoyu Wang 已提交
1186
      char userName[TSDB_USER_LEN + VARSTR_HEADER_SIZE] = {0};
1187
      STR_WITH_MAXSIZE_TO_VARSTR(userName, pUser->user, pShow->pMeta->pSchemas[cols].bytes);
X
Xiaoyu Wang 已提交
1188
      SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1189
      colDataSetVal(pColInfo, numOfRows, (const char *)userName, false);
1190 1191 1192 1193

      char privilege[20] = {0};
      STR_WITH_MAXSIZE_TO_VARSTR(privilege, "all", pShow->pMeta->pSchemas[cols].bytes);
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1194
      colDataSetVal(pColInfo, numOfRows, (const char *)privilege, false);
1195 1196 1197 1198

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

X
Xiaoyu Wang 已提交
1201
      char tableName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
C
cademfly 已提交
1202 1203 1204 1205
      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 已提交
1206
      char condition[TSDB_PRIVILEDGE_CONDITION_LEN + VARSTR_HEADER_SIZE] = {0};
C
cademfly 已提交
1207 1208 1209 1210
      STR_WITH_MAXSIZE_TO_VARSTR(condition, "", pShow->pMeta->pSchemas[cols].bytes);
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
      colDataSetVal(pColInfo, numOfRows, (const char *)condition, false);

1211 1212 1213
      numOfRows++;
    }

S
Shengliang Guan 已提交
1214 1215 1216
    char *db = taosHashIterate(pUser->readDbs, NULL);
    while (db != NULL) {
      cols = 0;
X
Xiaoyu Wang 已提交
1217
      char userName[TSDB_USER_LEN + VARSTR_HEADER_SIZE] = {0};
S
Shengliang Guan 已提交
1218
      STR_WITH_MAXSIZE_TO_VARSTR(userName, pUser->user, pShow->pMeta->pSchemas[cols].bytes);
X
Xiaoyu Wang 已提交
1219
      SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1220
      colDataSetVal(pColInfo, numOfRows, (const char *)userName, false);
S
Shengliang Guan 已提交
1221 1222 1223 1224

      char privilege[20] = {0};
      STR_WITH_MAXSIZE_TO_VARSTR(privilege, "read", pShow->pMeta->pSchemas[cols].bytes);
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1225
      colDataSetVal(pColInfo, numOfRows, (const char *)privilege, false);
S
Shengliang Guan 已提交
1226 1227 1228 1229 1230 1231 1232

      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++);
1233
      colDataSetVal(pColInfo, numOfRows, (const char *)objName, false);
S
Shengliang Guan 已提交
1234

C
cademfly 已提交
1235 1236 1237 1238 1239 1240 1241 1242 1243 1244
      char tableName[20] = {0};
      STR_WITH_MAXSIZE_TO_VARSTR(tableName, "", pShow->pMeta->pSchemas[cols].bytes);
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
      colDataSetVal(pColInfo, numOfRows, (const char *)tableName, false);

      char condition[20] = {0};
      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 已提交
1245 1246 1247 1248 1249 1250 1251
      numOfRows++;
      db = taosHashIterate(pUser->readDbs, db);
    }

    db = taosHashIterate(pUser->writeDbs, NULL);
    while (db != NULL) {
      cols = 0;
1252
      char userName[TSDB_USER_LEN + VARSTR_HEADER_SIZE] = {0};
S
Shengliang Guan 已提交
1253
      STR_WITH_MAXSIZE_TO_VARSTR(userName, pUser->user, pShow->pMeta->pSchemas[cols].bytes);
1254
      SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1255
      colDataSetVal(pColInfo, numOfRows, (const char *)userName, false);
S
Shengliang Guan 已提交
1256 1257 1258 1259

      char privilege[20] = {0};
      STR_WITH_MAXSIZE_TO_VARSTR(privilege, "write", pShow->pMeta->pSchemas[cols].bytes);
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1260
      colDataSetVal(pColInfo, numOfRows, (const char *)privilege, false);
S
Shengliang Guan 已提交
1261 1262 1263 1264 1265 1266 1267

      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++);
1268
      colDataSetVal(pColInfo, numOfRows, (const char *)objName, false);
S
Shengliang Guan 已提交
1269

C
cademfly 已提交
1270 1271 1272 1273 1274 1275 1276 1277 1278 1279
      char tableName[20] = {0};
      STR_WITH_MAXSIZE_TO_VARSTR(tableName, "", pShow->pMeta->pSchemas[cols].bytes);
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
      colDataSetVal(pColInfo, numOfRows, (const char *)tableName, false);

      char condition[20] = {0};
      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 已提交
1280 1281 1282 1283
      numOfRows++;
      db = taosHashIterate(pUser->writeDbs, db);
    }

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

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

S
Shengliang Guan 已提交
1288 1289 1290
    char *topic = taosHashIterate(pUser->topics, NULL);
    while (topic != NULL) {
      cols = 0;
1291
      char userName[TSDB_USER_LEN + VARSTR_HEADER_SIZE] = {0};
S
Shengliang Guan 已提交
1292
      STR_WITH_MAXSIZE_TO_VARSTR(userName, pUser->user, pShow->pMeta->pSchemas[cols].bytes);
1293
      SColumnInfoData *pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1294
      colDataSetVal(pColInfo, numOfRows, (const char *)userName, false);
S
Shengliang Guan 已提交
1295 1296 1297 1298

      char privilege[20] = {0};
      STR_WITH_MAXSIZE_TO_VARSTR(privilege, "subscribe", pShow->pMeta->pSchemas[cols].bytes);
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
1299
      colDataSetVal(pColInfo, numOfRows, (const char *)privilege, false);
S
Shengliang Guan 已提交
1300 1301 1302 1303 1304

      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++);
1305
      colDataSetVal(pColInfo, numOfRows, (const char *)topicName, false);
S
Shengliang Guan 已提交
1306

C
cademfly 已提交
1307 1308 1309 1310 1311 1312 1313 1314 1315 1316
      char tableName[20] = {0};
      STR_WITH_MAXSIZE_TO_VARSTR(tableName, "", pShow->pMeta->pSchemas[cols].bytes);
      pColInfo = taosArrayGet(pBlock->pDataBlock, cols++);
      colDataSetVal(pColInfo, numOfRows, (const char *)tableName, false);

      char condition[20] = {0};
      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 已提交
1317
      numOfRows++;
1318
      topic = taosHashIterate(pUser->topics, topic);
S
Shengliang Guan 已提交
1319
    }
1320

S
Shengliang Guan 已提交
1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331
    sdbRelease(pSdb, pUser);
  }

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

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

1333 1334
int32_t mndValidateUserAuthInfo(SMnode *pMnode, SUserAuthVersion *pUsers, int32_t numOfUses, void **ppRsp,
                                int32_t *pRspLen) {
D
dapan 已提交
1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349
  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 已提交
1350
    pUsers[i].version = ntohl(pUsers[i].version);
D
dapan 已提交
1351 1352 1353 1354
    if (pUser->authVersion <= pUsers[i].version) {
      mndReleaseUser(pMnode, pUser);
      continue;
    }
1355

D
dapan 已提交
1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370
    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;
1371

D
dapan 已提交
1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394
    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;
1395

D
dapan 已提交
1396 1397 1398
  tFreeSUserAuthBatchRsp(&batchRsp);
  return code;
}
1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446

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);
1447 1448 1449
    if (pIter == NULL) {
      break;
    }
1450 1451

    code = -1;
1452 1453 1454
    if (mndUserDupObj(pUser, &newUser) != 0) {
      break;
    }
1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473

    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;
}