Merge pull request #15086 from taosdata/fix/TS-1686

fix: resolve a segmentation fault issue of taos on alter table... set tag with a number as key

src/client/src/tscSQLParser.c

@@ -3972,10 +3972,6 @@ int32_t doGetColumnIndexByName(SStrToken* pToken, SQueryInfo* pQueryInfo, SColum
   const char* msg0 = "ambiguous column name";
   const char* msg1 = "invalid column name";
 
-  if (pToken->n == 0) {
-    return TSDB_CODE_TSC_INVALID_OPERATION;
-  }
-
   int16_t tsWinColumnIndex;
   if (isTablenameToken(pToken)) {
     pIndex->columnIndex = TSDB_TBNAME_COLUMN_INDEX;
@@ -4061,6 +4057,12 @@ int32_t getTableIndexByName(SStrToken* pToken, SQueryInfo* pQueryInfo, SColumnIn
 }
 
 int32_t getColumnIndexByName(const SStrToken* pToken, SQueryInfo* pQueryInfo, SColumnIndex* pIndex, char* msg) {
+  const char* msg0 = "invalid column name";
+
+  if (pToken->n == 0) {
+    return invalidOperationMsg(msg, msg0);
+  }
+
   if (pQueryInfo->pTableMetaInfo == NULL || pQueryInfo->numOfTables == 0) {
     return TSDB_CODE_TSC_INVALID_OPERATION;
   }
@@ -7399,6 +7401,9 @@ int32_t setAlterTableInfo(SSqlObj* pSql, struct SSqlInfo* pInfo) {
     }
 
     SColumnIndex idx = COLUMN_INDEX_INITIALIZER;
+    if (pItem->pVar.nType != TSDB_DATA_TYPE_BINARY) {
+        return invalidOperationMsg(pMsg, msg17);
+    }
     SStrToken    name = {.z = pItem->pVar.pz, .n = pItem->pVar.nLen};
 
     if (getColumnIndexByName(&name, pQueryInfo, &idx, tscGetErrorMsgPayload(pCmd)) != TSDB_CODE_SUCCESS) {
@@ -7472,6 +7477,9 @@ int32_t setAlterTableInfo(SSqlObj* pSql, struct SSqlInfo* pInfo) {
     int16_t       numOfTags = tscGetNumOfTags(pTableMeta);
 
     SColumnIndex columnIndex = COLUMN_INDEX_INITIALIZER;
+    if (item->pVar.nType != TSDB_DATA_TYPE_BINARY) {
+        return invalidOperationMsg(pMsg, msg17);
+    }
     SStrToken    name = {.z = item->pVar.pz, .n = item->pVar.nLen};
     if (getColumnIndexByName(&name, pQueryInfo, &columnIndex, tscGetErrorMsgPayload(pCmd)) != TSDB_CODE_SUCCESS) {
       return TSDB_CODE_TSC_INVALID_OPERATION;
@@ -7617,6 +7625,9 @@ int32_t setAlterTableInfo(SSqlObj* pSql, struct SSqlInfo* pInfo) {
     tVariantListItem* pItem = taosArrayGet(pAlterSQL->varList, 0);
 
     SColumnIndex columnIndex = COLUMN_INDEX_INITIALIZER;
+    if (pItem->pVar.nType != TSDB_DATA_TYPE_BINARY) {
+        return invalidOperationMsg(pMsg, msg17);
+    }
     SStrToken    name = {.z = pItem->pVar.pz, .n = pItem->pVar.nLen};
 
     if (getColumnIndexByName(&name, pQueryInfo, &columnIndex, tscGetErrorMsgPayload(pCmd)) != TSDB_CODE_SUCCESS) {
@@ -9665,7 +9676,9 @@ int32_t tscGetExprFilters(SSqlCmd* pCmd, SQueryInfo* pQueryInfo, SArray* pSelect
     SStrToken* pToken = &pParam->pNode->columnName;
 
     SColumnIndex idx = COLUMN_INDEX_INITIALIZER;
-    getColumnIndexByName(pToken, pQueryInfo, &idx, tscGetErrorMsgPayload(pCmd));
+    if (getColumnIndexByName(pToken, pQueryInfo, &idx, tscGetErrorMsgPayload(pCmd)) != TSDB_CODE_SUCCESS) {
+        return TSDB_CODE_TSC_INVALID_OPERATION;
+    }
     STableMetaInfo* pTableMetaInfo = tscGetMetaInfo(pQueryInfo, idx.tableIndex);
     schema = *tscGetTableColumnSchema(pTableMetaInfo->pTableMeta, idx.columnIndex);
   } else {

tests/pytest/table/tagNameCaseSensitive.py

@@ -65,7 +65,7 @@ class TDTestCase:
         tdSql.query("select * from `STB6`")
         tdSql.checkRows(6)
 
-        tdSql.execute("delete from `STB6` where ` ` = 1 and ts = '2022-06-24 11:17:31.000'")
+        tdSql.execute("delete from `STB6` where ` ` = 1 and ts = 1656040651000")
         tdSql.checkAffectedRows(1)
         tdSql.query("select * from `STB6`")
         tdSql.checkRows(5)
@@ -74,6 +74,10 @@ class TDTestCase:
         tdSql.query("select * from `STB6`")
         tdSql.checkRows(2)
 
+        tdSql.execute("alter table `STB6` add tag `1` int")
+        tdSql.execute("create table t1 using `STB6`(`1`) tags(1)")
+        tdSql.error("alter table t1 set tag 1=2222")
+
         tdSql.error("alter table `STB6` add tag `` nchar(20)")
 
     def stop(self):
@@ -82,4 +86,4 @@ class TDTestCase:
 
 
 tdCases.addWindows(__file__, TDTestCase())
-tdCases.addLinux(__file__, TDTestCase())
\ No newline at end of file
+tdCases.addLinux(__file__, TDTestCase())