[proxy,modules] check return of pf_modules_run_hook

This commit is contained in:
Armin Novak
2023-06-28 11:54:46 +02:00
committed by akallabeth
parent de7fea7347
commit 80058caa4d
2 changed files with 6 additions and 3 deletions

View File

@@ -765,7 +765,8 @@ static BOOL pf_client_connect(freerdp* instance)
if (!freerdp_connect(instance))
{
pf_modules_run_hook(pc->pdata->module, HOOK_TYPE_CLIENT_LOGIN_FAILURE, pc->pdata, pc);
if (!pf_modules_run_hook(pc->pdata->module, HOOK_TYPE_CLIENT_LOGIN_FAILURE, pc->pdata, pc))
goto out;
if (!retry)
goto out;

View File

@@ -578,7 +578,8 @@ static DWORD WINAPI pf_server_handle_peer(LPVOID arg)
pdata = ps->pdata;
WINPR_ASSERT(pdata);
pf_modules_run_hook(pdata->module, HOOK_TYPE_SERVER_SESSION_INITIALIZE, pdata, client);
if (!pf_modules_run_hook(pdata->module, HOOK_TYPE_SERVER_SESSION_INITIALIZE, pdata, client))
goto out_free_peer;
WINPR_ASSERT(client->Initialize);
client->Initialize(client);
@@ -586,7 +587,8 @@ static DWORD WINAPI pf_server_handle_peer(LPVOID arg)
PROXY_LOG_INFO(TAG, ps, "new connection: proxy address: %s, client address: %s",
pdata->config->Host, client->hostname);
pf_modules_run_hook(pdata->module, HOOK_TYPE_SERVER_SESSION_STARTED, pdata, client);
if (!pf_modules_run_hook(pdata->module, HOOK_TYPE_SERVER_SESSION_STARTED, pdata, client))
goto out_free_peer;
while (1)
{