public DtoApiIntResponse StartUdpSender(DtoMulticastArgs mArgs)
{
return(new DtoApiIntResponse()
{
Value = new MulticastArguments().GenerateProcessArguments(mArgs)
});
}
private int GenerateProcessArguments()
{
var multicastArgs = new DtoMulticastArgs();
multicastArgs.schema = new ServiceClientPartition(_imageProfile).GetImageSchema();
multicastArgs.Environment = _imageProfile.Image.Environment;
multicastArgs.ImageName = _imageProfile.Image.Name;
multicastArgs.Port = _multicastSession.Port.ToString();
var comServer = new ServiceClientComServer().GetServer(_multicastSession.ComServerId);
if (_isOnDemand)
{
multicastArgs.ExtraArgs = string.IsNullOrEmpty(_imageProfile.SenderArguments)
? comServer.MulticastSenderArguments
: _imageProfile.SenderArguments;
if (!string.IsNullOrEmpty(_clientCount))
{
multicastArgs.clientCount = _clientCount;
}
}
else
{
multicastArgs.ExtraArgs = string.IsNullOrEmpty(_imageProfile.SenderArguments)
? comServer.MulticastSenderArguments
: _imageProfile.SenderArguments;
multicastArgs.clientCount = _computers.Count.ToString();
}
if (_multicastServerId == null)
{
return(0);
}
var pid = new MulticastArguments().RunOnComServer(multicastArgs, comServer);
if (pid == 0)
{
return(pid);
}
var activeMulticastSessionServices = new ServiceActiveMulticastSession();
if (_isOnDemand)
{
_multicastSession.Pid = pid;
_multicastSession.Name = _group.Name;
activeMulticastSessionServices.AddActiveMulticastSession(_multicastSession);
}
else
{
_multicastSession.Pid = pid;
activeMulticastSessionServices.UpdateActiveMulticastSession(_multicastSession);
}
return(pid);
}
public int RunOnComServer(DtoMulticastArgs mArgs, EntityClientComServer comServer)
{
var intercomKey = ServiceSetting.GetSettingValue(SettingStrings.IntercomKeyEncrypted);
var decryptedKey = new EncryptionServices().DecryptText(intercomKey);
var pid = new APICall().ClientComServerApi.StartUdpSender(comServer.Url, "", decryptedKey, mArgs);
return(pid);
}
public int StartUdpSender(string url, string serverName, string interComKey, DtoMulticastArgs mArgs)
{
Request.Method = Method.POST;
Request.Resource = "Imaging/StartUdpSender";
Request.AddJsonBody(mArgs);
var responseData = new ApiRequest(new Uri(url)).ExecuteHMACInterCom <DtoApiIntResponse>(Request, serverName, interComKey);
if (responseData != null)
{
return(responseData.Value);
}
else
{
return(0);
}
}
public int GenerateProcessArguments(DtoMulticastArgs mArgs)
{
var guid = ConfigurationManager.AppSettings["ComServerUniqueId"];
_thisComServer = new ServiceClientComServer().GetServerByGuid(guid);
if (_thisComServer == null)
{
log.Error($"Com Server With Guid {guid} Not Found");
return(0);
}
var schemaCounter = -1;
var multicastHdCounter = 0;
string processArguments = null;
foreach (var hd in mArgs.schema.HardDrives)
{
schemaCounter++;
if (!hd.Active)
{
continue;
}
multicastHdCounter++;
var x = 0;
foreach (var part in mArgs.schema.HardDrives[schemaCounter].Partitions)
{
if (!part.Active)
{
continue;
}
string imageFile = null;
foreach (var ext in new[] { "ntfs", "fat", "extfs", "hfsp", "imager", "winpe", "xfs" })
{
imageFile = new FilesystemServices().GetMulticastFileNameWithFullPath(mArgs.ImageName,
schemaCounter.ToString(), part.Number, ext, _thisComServer.LocalStoragePath);
if (!string.IsNullOrEmpty(imageFile))
{
break;
}
//Look for lvm
if (part.VolumeGroup == null)
{
continue;
}
if (part.VolumeGroup.LogicalVolumes == null)
{
continue;
}
foreach (var lv in part.VolumeGroup.LogicalVolumes.Where(lv => lv.Active))
{
imageFile = new FilesystemServices().GetMulticastLVMFileNameWithFullPath(mArgs.ImageName,
schemaCounter.ToString(), lv.VolumeGroup, lv.Name, ext, _thisComServer.LocalStoragePath);
}
}
if (string.IsNullOrEmpty(imageFile))
{
continue;
}
if (mArgs.Environment == "winpe" &&
mArgs.schema.HardDrives[schemaCounter].Table.ToLower() == "gpt")
{
if (part.Type.ToLower() == "system" || part.Type.ToLower() == "recovery" ||
part.Type.ToLower() == "reserved")
{
continue;
}
}
if (mArgs.Environment == "winpe" &&
mArgs.schema.HardDrives[schemaCounter].Table.ToLower() == "mbr")
{
if (part.Number == mArgs.schema.HardDrives[schemaCounter].Boot &&
mArgs.schema.HardDrives[schemaCounter].Partitions.Length > 1)
{
continue;
}
}
x++;
var minReceivers = "";
if (!string.IsNullOrEmpty(mArgs.clientCount))
{
minReceivers = " --min-receivers " + mArgs.clientCount;
}
var isUnix = Environment.OSVersion.ToString().Contains("Unix");
string compAlg;
var stdout = "";
switch (Path.GetExtension(imageFile))
{
case ".lz4":
compAlg = isUnix ? "lz4 -d " : "lz4.exe\" -d ";
stdout = " - ";
break;
case ".gz":
if (isUnix)
{
compAlg = "gzip -c -d ";
stdout = "";
}
else
{
compAlg = "7za.exe\" x ";
stdout = " -so ";
}
break;
case ".uncp":
compAlg = "none";
break;
case ".wim":
compAlg = "none";
break;
default:
return(0);
}
if (isUnix)
{
string prefix = null;
if (multicastHdCounter == 1)
{
prefix = x == 1 ? " -c \"" : " ; ";
}
else
{
prefix = " ; ";
}
if (compAlg == "none" ||
_thisComServer.DecompressImageOn == "client")
{
processArguments += prefix + "cat " + "\"" + imageFile + "\"" + " | udp-sender" +
" --portbase " + mArgs.Port + minReceivers + " " +
" --ttl 32 --interface " + _thisComServer.MulticastInterfaceIp + " --mcast-rdv-address " + _thisComServer.MulticastInterfaceIp +
mArgs.ExtraArgs;
}
else
{
processArguments += prefix + compAlg + "\"" + imageFile + "\"" + stdout + " | udp-sender" +
" --portbase " + mArgs.Port + minReceivers + " " +
" --ttl 32 --interface " + _thisComServer.MulticastInterfaceIp + " --mcast-rdv-address " + _thisComServer.MulticastInterfaceIp +
mArgs.ExtraArgs;
}
}
else
{
var appPath = HttpContext.Current.Server.MapPath("~") + Path.DirectorySeparatorChar + "private" +
Path.DirectorySeparatorChar + "apps" + Path.DirectorySeparatorChar;
string prefix = null;
if (multicastHdCounter == 1)
{
prefix = x == 1 ? " /c \"" : " & ";
}
else
{
prefix = " & ";
}
if (compAlg == "none" ||
_thisComServer.DecompressImageOn == "client")
{
processArguments += prefix + "\"" + appPath +
"udp-sender.exe" + "\"" + " --file " + "\"" + imageFile + "\"" +
" --portbase " + mArgs.Port + minReceivers + " " +
" --ttl 32 --interface " + _thisComServer.MulticastInterfaceIp + " --mcast-rdv-address " + _thisComServer.MulticastInterfaceIp +
mArgs.ExtraArgs;
}
else
{
processArguments += prefix + "\"" + appPath + compAlg + "\"" + imageFile + "\"" + stdout +
" | " + "\"" + appPath +
"udp-sender.exe" + "\"" +
" --portbase " + mArgs.Port + minReceivers + " " +
" --ttl 32 --interface " + _thisComServer.MulticastInterfaceIp + " --mcast-rdv-address " + _thisComServer.MulticastInterfaceIp +
mArgs.ExtraArgs;
}
}
}
}
processArguments += "\"";
return(StartMulticastSender(processArguments, mArgs.groupName));
}
