titor/mimiclaw
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(search): prevent web_search output overflow and expose tavily secret template

Browse Source 
Signed-off-by: Asklv <boironic@gmail.com>
This commit is contained in:
Asklv
2026-03-03 01:25:30 +08:00
parent 443cb97f59
commit 7edcd31f8a
2 changed files with 18 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
main/mimi_secrets.h.example
View File

@@ -33,3 +33,5 @@
/* Brave Search API */ /* Brave Search API */
#define MIMI_SECRET_SEARCH_KEY "" #define MIMI_SECRET_SEARCH_KEY ""
/* Tavily Search API */
#define MIMI_SECRET_TAVILY_KEY ""

20
main/tools/tool_web_search.c
View File

@@ -139,19 +139,25 @@ static void format_results(cJSON *root, char *output, size_t output_size)
cJSON *item; cJSON *item;
cJSON_ArrayForEach(item, results) { cJSON_ArrayForEach(item, results) {
if (idx >= SEARCH_RESULT_COUNT) break; if (idx >= SEARCH_RESULT_COUNT) break;
if (off >= output_size - 1) break;
cJSON *title = cJSON_GetObjectItem(item, "title"); cJSON *title = cJSON_GetObjectItem(item, "title");
cJSON *url = cJSON_GetObjectItem(item, "url"); cJSON *url = cJSON_GetObjectItem(item, "url");
cJSON *desc = cJSON_GetObjectItem(item, "description"); cJSON *desc = cJSON_GetObjectItem(item, "description");
off += snprintf(output + off, output_size - off, int written = snprintf(output + off, output_size - off,
"%d. %s\n %s\n %s\n\n", "%d. %s\n %s\n %s\n\n",
idx + 1, idx + 1,
(title && cJSON_IsString(title)) ? title->valuestring : "(no title)", (title && cJSON_IsString(title)) ? title->valuestring : "(no title)",
(url && cJSON_IsString(url)) ? url->valuestring : "", (url && cJSON_IsString(url)) ? url->valuestring : "",
(desc && cJSON_IsString(desc)) ? desc->valuestring : ""); (desc && cJSON_IsString(desc)) ? desc->valuestring : "");
if (off >= output_size - 1) break; if (written < 0) break;
if ((size_t)written >= output_size - off) {
off = output_size - 1;
break;
}
off += (size_t)written;
idx++; idx++;
} }
} }
@@ -169,19 +175,25 @@ static void format_tavily_results(cJSON *root, char *output, size_t output_size)
cJSON *item; cJSON *item;
cJSON_ArrayForEach(item, results) { cJSON_ArrayForEach(item, results) {
if (idx >= SEARCH_RESULT_COUNT) break; if (idx >= SEARCH_RESULT_COUNT) break;
if (off >= output_size - 1) break;
cJSON *title = cJSON_GetObjectItem(item, "title"); cJSON *title = cJSON_GetObjectItem(item, "title");
cJSON *url = cJSON_GetObjectItem(item, "url"); cJSON *url = cJSON_GetObjectItem(item, "url");
cJSON *content = cJSON_GetObjectItem(item, "content"); cJSON *content = cJSON_GetObjectItem(item, "content");
off += snprintf(output + off, output_size - off, int written = snprintf(output + off, output_size - off,
"%d. %s\n %s\n %s\n\n", "%d. %s\n %s\n %s\n\n",
idx + 1, idx + 1,
(title && cJSON_IsString(title)) ? title->valuestring : "(no title)", (title && cJSON_IsString(title)) ? title->valuestring : "(no title)",
(url && cJSON_IsString(url)) ? url->valuestring : "", (url && cJSON_IsString(url)) ? url->valuestring : "",
(content && cJSON_IsString(content)) ? content->valuestring : ""); (content && cJSON_IsString(content)) ? content->valuestring : "");
if (off >= output_size - 1) break; if (written < 0) break;
if ((size_t)written >= output_size - off) {
off = output_size - 1;
break;
}
off += (size_t)written;
idx++; idx++;
} }
} }