From 7392e731558500d8532d4b45e1fb9b095208090c Mon Sep 17 00:00:00 2001
From: Shengliang Guan <slguan@taosdata.com>
Date: Mon, 18 Oct 2021 14:00:35 +0800
Subject: [PATCH] default user

---
 source/server/mnode/inc/mnodeDef.h  |   2 +
 source/server/mnode/src/mnodeSdb.c  |   1 +
 source/server/mnode/src/mnodeUser.c | 115 +++++++++++++++++++++++++++-
 3 files changed, 115 insertions(+), 3 deletions(-)

diff --git a/source/server/mnode/inc/mnodeDef.h b/source/server/mnode/inc/mnodeDef.h
index 3c1f4b2a16..a380be33f1 100644
--- a/source/server/mnode/inc/mnodeDef.h
+++ b/source/server/mnode/inc/mnodeDef.h
@@ -20,6 +20,7 @@
 #include "tlog.h"
 #include "trpc.h"
 #include "ttimer.h"
+#include "thash.h"
 #include "cJSON.h"
 #include "mnode.h"
 
@@ -177,6 +178,7 @@ typedef struct SUserObj {
   int64_t   createdTime;
   int64_t   updateTime;
   int8_t    rootAuth;
+  SHashObj *prohibitDbHash;
   SAcctObj *pAcct;
 } SUserObj;
 
diff --git a/source/server/mnode/src/mnodeSdb.c b/source/server/mnode/src/mnodeSdb.c
index 870ab63d4a..f676729469 100644
--- a/source/server/mnode/src/mnodeSdb.c
+++ b/source/server/mnode/src/mnodeSdb.c
@@ -148,6 +148,7 @@ static int32_t sdbReadDataFile() {
     pHead->status = MN_SDB_STAT_AVAIL;
 
     sdbInsertRow(pHead->type, pHead);
+    free(pHead);
     cJSON_Delete(root);
     root = NULL;
   }
diff --git a/source/server/mnode/src/mnodeUser.c b/source/server/mnode/src/mnodeUser.c
index 58110e5ef2..381a16ae26 100644
--- a/source/server/mnode/src/mnodeUser.c
+++ b/source/server/mnode/src/mnodeUser.c
@@ -15,7 +15,116 @@
 
 #define _DEFAULT_SOURCE
 #include "os.h"
-#include "mnodeInt.h"
+#include "tkey.h"
+#include "tglobal.h"
+#include "mnodeSdb.h"
 
-int32_t mnodeInitUser() { return 0; }
-void    mnodeCleanupUser() {}
\ No newline at end of file
+static int32_t mnodeCreateDefaultUser(char *acct, char *user, char *pass) {
+  int32_t code = TSDB_CODE_SUCCESS;
+
+  SUserObj userObj = {0};
+  tstrncpy(userObj.user, user, TSDB_USER_LEN);
+  tstrncpy(userObj.acct, acct, TSDB_USER_LEN);
+  taosEncryptPass((uint8_t *)pass, strlen(pass), userObj.pass);
+  userObj.createdTime = taosGetTimestampMs();
+  userObj.updateTime = taosGetTimestampMs();
+
+  if (strcmp(user, TSDB_DEFAULT_USER) == 0) {
+    userObj.rootAuth = 1;
+  }
+
+  sdbInsertRow(MN_SDB_USER, &userObj);
+}
+
+static void mnodeCreateDefaultUsers() {
+  mnodeCreateDefaultUser(TSDB_DEFAULT_USER, TSDB_DEFAULT_USER, TSDB_DEFAULT_PASS);
+  mnodeCreateDefaultUser(TSDB_DEFAULT_USER, "monitor", tsInternalPass);
+  mnodeCreateDefaultUser(TSDB_DEFAULT_USER, "_" TSDB_DEFAULT_USER, tsInternalPass);
+}
+
+int32_t mnodeEncodeUser(SUserObj *pUser, char *buf, int32_t maxLen) {
+  int32_t len = 0;
+  char   *base64 = base64_encode((const unsigned char *)pUser->pass, TSDB_KEY_LEN);
+
+  len += snprintf(buf + len, maxLen - len, "{\"type\":%d, ", MN_SDB_USER);
+  len += snprintf(buf + len, maxLen - len, "\"user\":\"%s\", ", pUser->user);
+  len += snprintf(buf + len, maxLen - len, "\"auth\":\"%24s\", ", base64);
+  len += snprintf(buf + len, maxLen - len, "\"acct\":\"%s\", ", pUser->acct);
+  len += snprintf(buf + len, maxLen - len, "\"createdTime\":\"%" PRIu64 "\", ", pUser->createdTime);
+  len += snprintf(buf + len, maxLen - len, "\"updateTime\":\"%" PRIu64 "\"}\n", pUser->updateTime);
+
+  free(base64);
+  return len;
+}
+
+SUserObj *mnodeDecodeUser(cJSON *root) {
+  int32_t   code = -1;
+  SUserObj *pUser = calloc(1, sizeof(SUserObj));
+
+  cJSON *user = cJSON_GetObjectItem(root, "user");
+  if (!user || user->type != cJSON_String) {
+    mError("failed to parse user since user not found");
+    goto DECODE_USER_OVER;
+  }
+  tstrncpy(pUser->user, user->valuestring, TSDB_USER_LEN);
+
+  if (strcmp(pUser->user, TSDB_DEFAULT_USER) == 0) {
+    pUser->rootAuth = 1;
+  }
+
+  cJSON *pass = cJSON_GetObjectItem(root, "auth");
+  if (!pass || pass->type != cJSON_String) {
+    mError("user:%s, failed to parse since auth not found", pUser->user);
+    goto DECODE_USER_OVER;
+  }
+
+  int32_t outlen = 0;
+  char   *base64 = (char *)base64_decode(pass->valuestring, strlen(pass->valuestring), &outlen);
+  if (outlen != TSDB_KEY_LEN) {
+    mError("user:%s, failed to parse since invalid auth format", pUser->user);
+    free(base64);
+    goto DECODE_USER_OVER;
+  } else {
+    memcpy(pUser->pass, base64, outlen);
+    free(base64);
+  }
+
+  cJSON *acct = cJSON_GetObjectItem(root, "acct");
+  if (!acct || acct->type != cJSON_String) {
+    mError("user:%s, failed to parse since acct not found", pUser->user);
+    goto DECODE_USER_OVER;
+  }
+  tstrncpy(pUser->acct, acct->valuestring, TSDB_USER_LEN);
+
+  cJSON *createdTime = cJSON_GetObjectItem(root, "createdTime");
+  if (!createdTime || createdTime->type != cJSON_String) {
+    mError("user:%s, failed to parse since createdTime not found", pUser->user);
+    goto DECODE_USER_OVER;
+  }
+  pUser->createdTime = atol(createdTime->valuestring);
+
+  cJSON *updateTime = cJSON_GetObjectItem(root, "updateTime");
+  if (!updateTime || updateTime->type != cJSON_String) {
+    mError("user:%s, failed to parse since updateTime not found", pUser->user);
+    goto DECODE_USER_OVER;
+  }
+  pUser->updateTime = atol(updateTime->valuestring);
+
+  code = 0;
+  mTrace("user:%s, parse success", pUser->user);
+
+DECODE_USER_OVER:
+  if (code != 0) {
+    free(pUser);
+    pUser = NULL;
+  }
+  return pUser;
+}
+
+int32_t mnodeInitUser() {
+  sdbSetFp(MN_SDB_USER, MN_KEY_BINARY, mnodeCreateDefaultUsers, (SdbEncodeFp)mnodeEncodeUser,
+           (SdbDecodeFp)(mnodeDecodeUser), sizeof(SUserObj));
+  return 0;
+}
+
+void mnodeCleanupUser() {}
\ No newline at end of file