mirrors
/
FreeRDP
mirror of https://github.com/FreeRDP/FreeRDP
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

PoC: waiting until an AVD instance is up

This commit is contained in:
Eduard Mielieshkin 2023-09-12 19:21:46 +03:00 committed by akallabeth
parent 1afaaa82d3
commit 4dcf35c0cb
1 changed files with 68 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

68
libfreerdp/core/gateway/arm.c
View File

@ -56,6 +56,9 @@
#include <cjson/cJSON.h>
#endif
#include <time.h>
#include <string.h>
struct rdp_arm
{
rdpContext* context;
@ -65,6 +68,8 @@ struct rdp_arm
typedef struct rdp_arm rdpArm;
short int gateway_retry = 0;
#define TAG FREERDP_TAG("core.gateway.arm")
#ifdef WITH_AAD
@ -855,6 +860,69 @@ BOOL arm_resolve_endpoint(rdpContext* context, DWORD timeout)
if (!res)
goto arm_error;
}
else if (StatusCode == HTTP_STATUS_BAD_REQUEST)
{
const size_t response_length = http_response_get_body_length(response);
char* msg = calloc(response_length + 1, sizeof(char));
if (!msg)
goto arm_error;
memcpy(msg, http_response_get_body(response), response_length);
WLog_DBG(TAG, "Got HTTP Response data: %s", msg);
cJSON *json = cJSON_ParseWithLength((const char*)msg, response_length);
if (json == NULL)
{
const char *error_ptr = cJSON_GetErrorPtr();
if (error_ptr != NULL)
{
WLog_ERR(TAG, "NullPoException: %s", error_ptr);
free(msg);
goto arm_error;
}
}
const cJSON *gateway_code_str = NULL;
gateway_code_str = cJSON_GetObjectItemCaseSensitive(json, "Code");
if (!cJSON_IsString(gateway_code_str) || (gateway_code_str->valuestring == NULL))
{
WLog_ERR(TAG, "Response has no \"Code\" property");
http_response_log_error_status(WLog_Get(TAG), WLOG_ERROR, response);
if (json) {
cJSON_Delete(json);
}
free(msg);
goto arm_error;
}
if (strcmp(gateway_code_str->valuestring, "E_PROXY_ORCHESTRATION_LB_SESSIONHOST_DEALLOCATED") == 0) {
int max_retries = 10;
int code_execution_interval = 60;
if (gateway_retry >= max_retries) {
WLog_ERR(TAG, "Timeout reached. Exiting loop.");
if (json) {
cJSON_Delete(json);
}
free(msg);
goto arm_error;
}
WLog_INFO(TAG, "Please wait... VM is starting running.");
if (json) {
cJSON_Delete(json);
}
free(msg);
sleep(code_execution_interval);
arm_resolve_endpoint(context, timeout);
}
else
{
cJSON_Delete(json);
free(msg);
http_response_log_error_status(WLog_Get(TAG), WLOG_ERROR, response);
goto arm_error;
}
}
else
{
http_response_log_error_status(WLog_Get(TAG), WLOG_ERROR, response);