GuillaumeHemmen/AutoTrackR2
1
0
Fork 0
mirror of https://github.com/BubbaGumpShrump/AutoTrackR2.git synced 2025-05-25 09:35:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added checks for streamlink response for web handler

Browse source 
I've also put in a check to make sure if we are going to accept a response from the api that it it's valid.
This commit is contained in:
Heavy Bob 2025-04-10 02:46:35 +10:00
parent d7057efe15
commit 5994743c73
1 changed files with 39 additions and 15 deletions

54
AutoTrackR2/WebHandler.cs
View file

@ -175,23 +175,47 @@ public static class WebHandler
var responseContent = await response.Content.ReadAsStringAsync();
Console.WriteLine($"Response: {responseContent}");
// Check if the response contains a streamer field
try
// Only process streamer data if streamlink is enabled
if (ConfigManager.StreamlinkEnabled == 1)
{
var responseData = JsonSerializer.Deserialize<Dictionary<string, JsonElement>>(responseContent);
if (responseData != null && responseData.TryGetValue("streamer", out JsonElement streamerElement))
{
string streamerHandle = streamerElement.GetString() ?? string.Empty;
if (!string.IsNullOrEmpty(streamerHandle))
{
TrackREventDispatcher.OnStreamlinkRecordEvent(streamerHandle);
}
}
}
catch (Exception ex)
{
Console.WriteLine($"Error parsing streamer from response: {ex.Message}");
ProcessStreamerResponse(responseContent);
}
}
}
private static void ProcessStreamerResponse(string responseContent)
{
try
{
var responseData = JsonSerializer.Deserialize<Dictionary<string, JsonElement>>(responseContent);
if (responseData != null && responseData.TryGetValue("streamer", out JsonElement streamerElement))
{
string streamerHandle = streamerElement.GetString() ?? string.Empty;
if (!string.IsNullOrEmpty(streamerHandle))
{
// Sanitize the streamer handle before using it, this is to prevent any malicious instructions.
string sanitizedHandle = SanitizeStreamerHandle(streamerHandle);
if (!string.IsNullOrEmpty(sanitizedHandle))
{
TrackREventDispatcher.OnStreamlinkRecordEvent(sanitizedHandle);
}
}
}
}
catch (Exception ex)
{
Console.WriteLine($"Error parsing streamer from response: {ex.Message}");
}
}
private static string SanitizeStreamerHandle(string handle)
{
// Twitch usernames 4-25 characters, letters, numbers and underscores.
if (Regex.IsMatch(handle, @"^[a-zA-Z0-9_]{4,25}$"))
{
return handle.ToLower(); // Api won't return anything other than lowercase but just in case.
}
return string.Empty; // Reject invalid handles
}
}