add check for tag range

apache · Apr 19, 2024 · 4903094 · 4903094
1 parent 7debc56
commit 4903094
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/bthread/bthread.cpp b/src/bthread/bthread.cpp
@@ -389,6 +389,9 @@ int bthread_setconcurrency_by_tag(int num, bthread_tag_t tag) {
         bthread::never_set_bthread_concurrency_by_tag = false;
         return 0;
     }
+    if (tag < BTHREAD_TAG_DEFAULT || tag >= FLAGS_task_group_ntags) {
+        return EPERM;
+    }
     BAIDU_SCOPED_LOCK(bthread::g_task_control_mutex);
     auto c = bthread::get_task_control();
     if (c == NULL) {