aswi2022bug-thugs-gitlab/ld_client/LDClient/detection/InfoFetcher.cs @ 6dab0250

using System.Diagnostics;
using LDClient.utils;

namespace LDClient.detection {

    public class InfoFetcher {
        
        private const string UndefinedSerialNumber = "number";

        private readonly string _f32RemExecutable;
        private readonly string _f32RemArguments;
        
        private readonly uint _maxAttempts;
        private readonly uint _waitPeriodMs;
        private readonly string _infoFilePath;

        public string HeadSerialNumber { get; private set; } = UndefinedSerialNumber;
        public string BodySerialNumber { get; private set; } = UndefinedSerialNumber;

        public InfoFetcher(uint maxAttempts, uint waitPeriodMs, string infoFilePath, string f32RemExecutable, string f32RemArguments) {
            _maxAttempts = maxAttempts;
            _waitPeriodMs = waitPeriodMs;
            _infoFilePath = infoFilePath;
            _f32RemExecutable = f32RemExecutable;
            _f32RemArguments = f32RemArguments;
        }

        public async Task<bool> FetchDataAsync() {
            Program.DefaultLogger.Info("Fetching data from the debugger.");
            var success = await SendRetrieveInfoCommandAsync(_f32RemExecutable, _f32RemArguments);
            if (!success) {
                Program.DefaultLogger.Error("Failed to fetch data from the debugger.");
                return false;
            }
            for (var i = 0; i < _maxAttempts; i++) {
                Program.DefaultLogger.Info($"{i}. attempt to parse the info file.");
                if (RetrieveDebuggerInfo(_infoFilePath)) {
                    Program.DefaultLogger.Info($"Info file has been parsed successfully.");
                    return true;
                }
                await Task.Delay((int)_waitPeriodMs);
            }
            Program.DefaultLogger.Error("Failed to parse the into file. It may have not been created.");
            return false;
        }

        private bool RetrieveDebuggerInfo(string filePath) {
            try {
                var fileContent = IoUtils.ReadFile(filePath);
                var (headSerialNumber, bodySerialNumber) = DebuggerInfoParser.Parse(fileContent);
                HeadSerialNumber = headSerialNumber;
                BodySerialNumber = bodySerialNumber;
                File.Delete(filePath);
            } catch (Exception exception) {
                Program.DefaultLogger.Error($"Failed to retrieve debugger info. File {filePath} may not exist or it does not have the right format. {exception.Message}");
                return false;
            }
            return true;
        }

        private static async Task<bool> SendRetrieveInfoCommandAsync(string executableFile, string arguments) {
            var t32RemProcess = new Process();
            t32RemProcess.StartInfo.FileName = executableFile;
            t32RemProcess.StartInfo.Arguments = arguments;
            try {
                t32RemProcess.Start();
                await t32RemProcess.WaitForExitAsync();
            } catch (Exception exception) {
                Program.DefaultLogger.Error($"Failed to run {executableFile}. {exception.Message}");
                return false;
            }
            return true;
        }
    }
}