gfwleak/pxz-hos-client-cpp-module
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🐞fix(hos_set_thread_sum): 修复fd_info引起的内存错误

Browse Source
This commit is contained in:
彭宣正
2021-04-16 17:07:15 +08:00
parent 82dc6b00e7
commit 20e6f942cd
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/hos_client.cpp
View File

@@ -204,10 +204,6 @@ void hos_set_thread_sum(hos_client_handle client, size_t thread_sum)
return; return;
} }
for (size_t i = 0; i < thread_sum; i++)
{
fd_info[i][0] = 65533;
}
if (hos_cache) if (hos_cache)
{ {
hos_cache = (size_t *)realloc(hos_cache, thread_sum * sizeof(size_t)); hos_cache = (size_t *)realloc(hos_cache, thread_sum * sizeof(size_t));
@@ -221,6 +217,7 @@ void hos_set_thread_sum(hos_client_handle client, size_t thread_sum)
{ {
hash_hos_info = (hos_info_t **)calloc(thread_sum, sizeof(hos_info_t*)); hash_hos_info = (hos_info_t **)calloc(thread_sum, sizeof(hos_info_t*));
} }
if (fd_info) if (fd_info)
{ {
fd_info = (size_t (*) [MAX_HOS_CLIENT_FD_NUM + 1])realloc(fd_info, thread_sum * sizeof(size_t [MAX_HOS_CLIENT_FD_NUM + 1])); fd_info = (size_t (*) [MAX_HOS_CLIENT_FD_NUM + 1])realloc(fd_info, thread_sum * sizeof(size_t [MAX_HOS_CLIENT_FD_NUM + 1]));
@@ -228,6 +225,12 @@ void hos_set_thread_sum(hos_client_handle client, size_t thread_sum)
{ {
fd_info = (size_t (*) [MAX_HOS_CLIENT_FD_NUM + 1])calloc(thread_sum, sizeof(size_t [MAX_HOS_CLIENT_FD_NUM + 1])); fd_info = (size_t (*) [MAX_HOS_CLIENT_FD_NUM + 1])calloc(thread_sum, sizeof(size_t [MAX_HOS_CLIENT_FD_NUM + 1]));
} }
for (size_t i = 0; i < thread_sum; i++)
{
fd_info[i][0] = 65533;
}
client->thread_sum = thread_sum; client->thread_sum = thread_sum;
return ; return ;
} }