public MainWindow()
{
try
{
//' Initial logging statement
WriteLogFile("ConfigMgr OSD Monitor started");
//' Read command line arguments and determine application run mode
string[] cmdLineArgs = Environment.GetCommandLineArgs();
string convertedParameter = string.Empty;
switch (cmdLineArgs[1])
{
case "OSCOM":
convertedParameter = "END";
break;
case "OSFULL":
convertedParameter = "FULLOS";
break;
case "OSPE":
convertedParameter = "WINPE";
break;
}
WriteLogFile(String.Format("Passed command line argument: {0}", convertedParameter));
if (cmdLineArgs.Length == 1)
{
WriteLogFile("No command line argument was given, please use either WINPE, FULLOS or END");
Application.Current.Shutdown();
}
else if (cmdLineArgs.Length == 2)
{
List <string> approvedArguments = new List <string> {
"OSFULL", "OSPE", "OSCOM"
};
string cmdLine = cmdLineArgs[1];
if (approvedArguments.Contains(cmdLine))
{
//' Ensure that task sequence environment can be loaded
TSEnvironment.TestTSEnvironment();
WriteLogFile("Successfully loaded the Microsoft.SMS.TSEnvironment object");
//' Get computer macaddress information
macAddress = Computer.GetMacAddress();
if (String.IsNullOrEmpty(macAddress))
{
macAddress = null;
}
//' Get computer smbios guid information
uuid = Computer.GetSMBIOSGUID();
if (String.IsNullOrEmpty(uuid))
{
uuid = null;
}
try
{
//' Determine if a task sequence for operating system deployment is running
sequenceType = TSEnvironment.GetTSVariable("_SMSTSType");
if (sequenceType == "2")
{
WriteLogFile("Detected that a task sequence is currently running at application startup");
}
else
{
WriteLogFile("Detected that no task sequence is currently running at application startup");
}
WriteLogFile("Successfully determined task sequence operational state");
}
catch (System.Exception ex)
{
WriteLogFile(String.Format("An error occurred while reading TSEnvironment variable _SMSTSType at application startup. Error: {0}", ex.Message));
Application.Current.Shutdown();
}
//' Initialize application and show window
InitializeComponent();
if (convertedParameter == "WINPE" || convertedParameter == "FULLOS")
{
if (convertedParameter == "WINPE")
{
try
{
//' Gather dart data
if (Properties.Settings.Default.GatherDaRTData == true)
{
bool isWinpe;
Boolean.TryParse(TSEnvironment.GetTSVariable("_SMSTSInWinPE"), out isWinpe);
if (isWinpe == true)
{
string systemDrive = String.Format("{0}\\", Environment.GetEnvironmentVariable("SystemDrive"));
WriteLogFile("Gather DaRT data application setting was enabled, attempting to locate and read DaRT information");
LoadDartData(systemDrive);
}
}
}
catch (System.Exception ex)
{
WriteLogFile(String.Format("An error occurred while reading DaRT information from inv32.xml. Error: {0}", ex.Message));
}
}
//' Handle session ending event from operating system
SystemEvents.SessionEnding += new SessionEndingEventHandler(SystemEvents_SessionEnding);
WriteLogFile("Successfully configured event handler for restarts of computer");
//' Construct new timer to fire accordingly to application settings
WriteLogFile(String.Format("Creating a new timer to send monitoring data to web service every '{0}' ms", Properties.Settings.Default.MonitorIntervalMilliseconds));
timer = new System.Windows.Forms.Timer
{
Interval = Properties.Settings.Default.MonitorIntervalMilliseconds
};
timer.Tick += new EventHandler(Timer_Tick);
timer.Start();
}
else
{
AddMonitoringDataEnd();
WriteLogFile("Successfully posted deployment completed monitoring data to web service, application will now shutdown");
Application.Current.Shutdown();
}
}
else
{
WriteLogFile("Unsupported command line argument detected, please see documentation for supported command line arguments");
Application.Current.Shutdown();
}
}
else
{
WriteLogFile("Unhandled amount of command line arguments passed to application, please see documentation for supported command line arguments");
Application.Current.Shutdown();
}
}
catch (System.Exception ex)
{
MessageBox.Show("Unable to load Task Sequence environment", "Load failure", MessageBoxButton.OK, MessageBoxImage.Error);
WriteLogFile(String.Format("Unable to load Microsoft.SMS.TSEnvironment, this application should be executed from within a task sequence environment", ex.Message));
Application.Current.Shutdown();
}
}
async public static void AddMonitoringData(MonitoringSessionState monitoringState, string modifiedTime, string computerName, bool monitorEnd)
{
WriteLogFile("Monitoring timer was triggered, reading task sequence environment data");
try
{
//' Determine unique id for deployment
string uniqueId = TSEnvironment.GetTSVariable(Properties.Settings.Default.UniqueMonitoringTSVariableName);
if (String.IsNullOrEmpty(uniqueId))
{
uniqueId = Guid.NewGuid().ToString();
TSEnvironment.SetTSVariable(Properties.Settings.Default.UniqueMonitoringTSVariableName, uniqueId);
}
try
{
//' Read additional task sequence information if the environment is reachable
bool tsEnvironment = TSEnvironment.TestTSEnvironment();
if (tsEnvironment == true)
{
if (!String.IsNullOrEmpty(sequenceType))
{
if (sequenceType == "2")
{
//' Read task sequence variable for deployment ID value
deploymentId = TSEnvironment.GetTSVariable("_SMSTSAdvertID");
if (String.IsNullOrEmpty(deploymentId))
{
deploymentId = null;
}
//' Read task sequence variable for current step name value
stepName = TSEnvironment.GetTSVariable("_SMSTSCurrentActionName");
if (String.IsNullOrEmpty(stepName))
{
stepName = null;
}
//' Read task sequence variable for current step value
currentStep = TSEnvironment.GetTSVariable("_SMSTSNextInstructionPointer");
if (String.IsNullOrEmpty(currentStep))
{
currentStep = null;
}
//' Read task sequence variable for total steps value
totalSteps = TSEnvironment.GetTSVariable("_SMSTSInstructionTableSize");
if (String.IsNullOrEmpty(totalSteps))
{
totalSteps = null;
}
}
}
}
}
catch (System.Exception ex)
{
WriteLogFile(String.Format("An warning occurred while reading additional task sequence data, this exception happened within the add monitoring method and was unexpected. Error: {0}", ex.Message));
}
//' Determine details message parameter
switch (monitoringState)
{
case MonitoringSessionState.Running:
details = "Task sequence is running";
severity = 1;
break;
case MonitoringSessionState.Reboot:
details = "Computer is being restarted";
severity = 1;
break;
case MonitoringSessionState.NotStarted:
details = "Task sequence has not started";
severity = 1;
break;
case MonitoringSessionState.Failed:
details = "Deployment failed";
severity = 2;
break;
case MonitoringSessionState.Completed:
details = "Deployment completed";
severity = 1;
break;
default:
severity = 1;
break;
}
//' Determine if end time should be set
if (monitorEnd == true)
{
endTime = GetTimeNow();
}
try
{
//' Add monitoring data through web service call
AddCMOSDMonitorDataResponse monitorResponse = await webService.AddCMOSDMonitorDataAsync(Properties.Settings.Default.WebServiceSecretKey, uniqueId, computerName, uuid, macAddress, severity, modifiedTime, deploymentId, stepName, currentStep, totalSteps, startTime, endTime, details, dartIp, dartPort, dartTicket);
WriteLogFile("Successfully posted monitoring data to web service");
}
catch (System.Exception ex)
{
WriteLogFile(String.Format("An error occurred while attempting to post monitoring data to web service. Error: {0}", ex.Message));
}
}
catch (System.Exception ex)
{
WriteLogFile(String.Format("An error occurred while collecting data for web service post. Will be retried in the next cycle. Error: {0}", ex.Message));
}
}