private void Begin()
{
_webClient.DownloadFileCompleted += DownloadCompleted;
if (!ValidateUri(DownloadUri))
{
throw new DownloadException("The specified url cannot be found");
}
if (!String.IsNullOrEmpty(SaveFileAs))
{
ValidateSaveAs(SaveFileAs);
}
if (!Headless)
{
Show();
TopMost = true;
}
Text = String.Format("0% of {0} Completed", Path.GetFileName(DownloadUri.ToString()));
lblProgress.Text = "";
lblFileName.Text = String.Format("{0} from {1}", Path.GetFileName(DownloadUri.ToString()), PathName);
_webClient.DownloadProgressChanged += DownloadProgressChanged;
_webClient.DownloadFileCompleted -= DownloadComplete;
_webClient.DownloadFileCompleted += DownloadComplete;
progressBar1.Value = 0;
Log.Info("Download started");
_timer.Interval = 1000;
_timer.Tick += TimerTick;
_webClient.Proxy = null;
_webClient.DownloadFileAsync(DownloadUri, FileToSave, SaveFileAs);
_timer.Start();
}
public override int GetHashCode()
{
unchecked
{
return(((DownloadUri != null ? DownloadUri.GetHashCode() : 0) * 397) ^ (DownloadPath != null ? DownloadPath.GetHashCode() : 0));
}
}
private static Dictionary <string, List <DownloadUri> > GetAllUpdates(IUpdate iUpdate, bool wsusEnabled = false)
{
var bundleDict = new Dictionary <string, List <DownloadUri> >();
foreach (IUpdate bundle in iUpdate.BundledUpdates)
{
foreach (IUpdateDownloadContent udc in bundle.DownloadContents)
{
var downloadContents = new List <DownloadUri>();
if (String.IsNullOrEmpty(udc.DownloadUrl))
{
continue;
}
var downloadUri = new DownloadUri();
string[] sha1Hash;
if (wsusEnabled)
{
//This uses WSUS Server
var sha1Tmp = udc.DownloadUrl.Split('/');
sha1Hash = sha1Tmp[sha1Tmp.Length - 1].Split('.');
}
else
{
//Not a WSUS Server
var sha1Tmp = udc.DownloadUrl.Split('_');
sha1Hash = sha1Tmp[sha1Tmp.Length - 1].Split('.');
}
downloadUri.Hash = (sha1Hash[0] ?? (sha1Hash[0] = ""));
downloadUri.Uri = udc.DownloadUrl;
var tempHold = downloadUri.Uri.Split(new[] { '/' });
downloadUri.FileName = tempHold[tempHold.Length - 1];
downloadUri.FileSize = GetUriFileSize(downloadUri.Uri);
downloadContents.Add(downloadUri);
if (!bundleDict.ContainsKey(bundle.Title))
{
bundleDict.Add(bundle.Title, downloadContents);
}
}
if (bundle.BundledUpdates.Count > 0)
{
var valuesReturned = GetAllUpdates(bundle);
foreach (var data in valuesReturned)
{
if (!bundleDict.ContainsKey(data.Key))
{
bundleDict.Add(data.Key, data.Value);
}
}
}
}
return(bundleDict);
}
public DownloadMessage(Uri uri, string targetPath, E_FileDisposition enumDisposition = E_FileDisposition.OverwriteAlways, int htmlDepth = 0)
{
ID = unchecked (Interlocked.Increment(ref RequestID));
DownloadUri = uri ?? throw new NullReferenceException("DownloadMessage requires non-null Uri"); // assume caller has lowercased strings
Url = DownloadUri.ToString(); // set alternate format as 1-off
var segs = uri.Segments; // split the url (except querystring)
TargetPath = (string.IsNullOrWhiteSpace(Path.GetFileNameWithoutExtension(targetPath)))
? Path.Combine(targetPath, segs[segs.Length - 1] + ".html") // default filename to final segment of Url
: targetPath;
EnumDisposition = enumDisposition;
HtmlDepth = htmlDepth;
}
private void DownloadProgressChanged(object sender, DownloadProgressChangedEventArgs e)
{
DownloadStatus = Status.Downloading;
DownloadSize = e.TotalBytesToReceive;
DataDownloaded = e.BytesReceived;
Text = String.Format("{0}% of {1} Completed", e.ProgressPercentage, Path.GetFileName(DownloadUri.ToString()));
progressBar1.Value = e.ProgressPercentage;
DownloadProgress = String.Format("{0} of {1}", Utils.BitsToString(Convert.ToInt32(e.BytesReceived)), Utils.BitsToString(Convert.ToInt32(e.TotalBytesToReceive)));
lblProgress.Text = String.Format("{0} at {1}", DownloadProgress, DownloadSpeedAsString);
lblFileName.Text = String.Format("{0} from {1}", Path.GetFileName(DownloadUri.ToString()), PathName);
}
public static List <DownloadUri> GrabLocalUpdateBundle(string updateIdentity, string updateName)
{
if (Directory.Exists(Settings.SavedUpdatesDirectory))
{
var directories = Directory.GetDirectories(Settings.SavedUpdatesDirectory);
var downloaduri = new DownloadUri();
var downloaduriList = new List <DownloadUri>();
foreach (var dir in directories)
{
var stripped = dir.Split(new[] { '\\' });
if (stripped[stripped.Length - 1] != updateIdentity)
{
continue;
}
var dirContent = Directory.GetFiles(dir);
foreach (var item in dirContent)
{
var strip = item.Split(new[] { '\\' });
var filename = strip[strip.Length - 1];
if (updateName.Trim().ToLower() == filename.Trim().ToLower())
{
var serialized = JsonConvert.DeserializeObject <Operations.LocalBundleContent>(File.ReadAllText(item));
foreach (var data in serialized.Value)
{
downloaduri.FileName = data.FileName;
downloaduri.FileSize = Convert.ToInt32(data.FileSize);
downloaduri.Hash = data.Hash;
downloaduri.Uri = data.Uri;
downloaduriList.Add(downloaduri);
}
return(downloaduriList);
}
}
}
}
return(null);
}
public static Operations.SavedOpData DownloadFile(Operations.SavedOpData update, UpdateDirectories updateTypePath)
{
var uris = new List <DownloadUri>();
string updateDirectory;
switch (updateTypePath)
{
case UpdateDirectories.SupportedAppDir:
updateDirectory = Settings.SupportedAppDirectory;
break;
case UpdateDirectories.CustomAppDir:
updateDirectory = Settings.CustomAppDirectory;
break;
case UpdateDirectories.OSUpdateDir:
updateDirectory = Settings.UpdateDirectory;
break;
default:
updateDirectory = Settings.UpdateDirectory;
break;
}
var savedir = Path.Combine(updateDirectory, update.filedata_app_id);
foreach (var uriData in update.filedata_app_uris)
{
var tempDownloadUri = new DownloadUri();
tempDownloadUri.FileName = uriData.file_name;
tempDownloadUri.FileSize = uriData.file_size;
tempDownloadUri.Hash = uriData.file_hash;
//add uri link from WSUS if its enabled
if (WSUS.IsWSUSEnabled())
{
var tempAvilApp = Agent.RV.WindowsApps.WindowsUpdates.GetAvailableUpdates();
foreach (var application in tempAvilApp)
{
if (application.Name == update.filedata_app_name)
{
foreach (var availUri in application.FileData)
{
tempDownloadUri.Uris.Add(availUri.Uri);
}
}
}
}
foreach (var uri in uriData.file_uris)
{
tempDownloadUri.Uris.Add(uri);
}
uris.Add(tempDownloadUri);
}
if (Directory.Exists(savedir))
{
Directory.Delete(savedir, true);
}
Directory.CreateDirectory(savedir);
foreach (var file in uris)
{
// Just in case the web server is using a self-signed cert.
// Webclient won't validate the SSL/TLS cerficate if it's not trusted.
var tempCallback = ServicePointManager.ServerCertificateValidationCallback;
ServicePointManager.ServerCertificateValidationCallback = delegate { return(true); };
var filepath = Path.Combine(savedir, file.FileName);
try
{
using (var client = new WebClient())
{
if (Settings.Proxy != null)
{
client.Proxy = Settings.Proxy;
}
var downloaded = false;
foreach (var uriSingle in file.Uris)
{
try
{
var splitted = uriSingle.Split(new[] { "//" }, StringSplitOptions.RemoveEmptyEntries);
var splitted2 = splitted[1].Split(new[] { '/' });
var relayserver = splitted2[0];
if (downloaded)
{
break;
}
Logger.Log("Attempting to download {1} from {0} with file size of {2}.", LogLevel.Info, relayserver, file.FileName, file.FileSize);
client.DownloadFile(uriSingle, filepath);
if (File.Exists(filepath))
{
var localFileHashMd5 = RvUtils.Md5HashFile(filepath).ToLower();
var localFileHashSha1 = RvUtils.Sha1HashFile(filepath).ToLower();
var localFileHashSha256 = RvUtils.Sha256HashFile(filepath).ToLower();
Logger.Log("Download Complete, {0}", LogLevel.Info, file.FileName);
Logger.Log("Checking Hashes...");
Logger.Log("Incoming Hash: {0}", LogLevel.Info, file.Hash);
Logger.Log("Local MD5 Hash: {0}", LogLevel.Info, localFileHashMd5);
Logger.Log("Local SHA1 Hash: {0}", LogLevel.Info, localFileHashSha1);
Logger.Log("Local SHA256 Hash: {0}", LogLevel.Info, localFileHashSha256);
downloaded = true;
if (localFileHashMd5 != file.Hash.ToLower() && localFileHashSha1 != file.Hash.ToLower() && localFileHashSha256 != file.Hash.ToLower())
{
Logger.Log("Local file {0} Hash did not match remote's. Retrying with a different server.", LogLevel.Info, file.FileName);
update.error = "Local file Hash did not match remote. Bad file integrity. ";
update.success = false.ToString().ToLower();
downloaded = false;
}
}
else
{
Logger.Log("File {0} did not download. Retrying with a different server.", LogLevel.Info, file.FileName);
update.error = "File did not download successfully, it was not found on disk. Please check download server.";
update.success = false.ToString().ToLower();
downloaded = false;
}
}
catch (Exception e)
{
Logger.Log("File {0} failed to download correctly... Possible connection issue, Retrying with a different server.", LogLevel.Info, file.FileName);
update.error = "File did not download correctly, Exception message: " + e.Message + ". Please check download server connectivity.";
update.success = false.ToString().ToLower();
downloaded = false;
}
}
//Check if the file was successfully downloaded and return.
if (downloaded)
{
update.error = String.Empty;
update.success = true.ToString().ToLower();
}
}
}
catch (Exception e)
{
//Critical exception occurred.
update.error = "Application did not download, Exception occured, refer to log for details.";
update.success = false.ToString().ToLower();
Logger.Log("One or more Application Files were not downloaded successfully; {0}.",
LogLevel.Error, file.FileName);
Logger.LogException(e);
return(update);
}
ServicePointManager.ServerCertificateValidationCallback = tempCallback;
}
return(update);
}
public static Operations.SavedOpData DownloadUpdate(Operations.SavedOpData update)
{
var uris = new List <DownloadUri>();
foreach (var uriData in update.filedata_app_uris)
{
var tempDownloadUri = new DownloadUri();
tempDownloadUri.FileName = uriData.file_name;
tempDownloadUri.FileSize = uriData.file_size;
tempDownloadUri.Hash = uriData.file_hash;
foreach (var uri in uriData.file_uris)
{
tempDownloadUri.Uris.Add(uri);
}
uris.Add(tempDownloadUri);
}
try
{
if (Directory.Exists(AgentUpdateDirectory))
{
Directory.Delete(AgentUpdateDirectory, true);
}
Directory.CreateDirectory(AgentUpdateDirectory);
}catch {}
foreach (var file in uris)
{
// Just in case the web server is using a self-signed cert.
// Webclient won't validate the SSL/TLS cerficate if it's not trusted.
var tempCallback = ServicePointManager.ServerCertificateValidationCallback;
ServicePointManager.ServerCertificateValidationCallback = delegate { return(true); };
var filepath = Path.Combine(AgentUpdateDirectory, file.FileName);
try
{
using (var client = new WebClient())
{
if (Settings.Proxy != null)
{
client.Proxy = Settings.Proxy;
}
var downloaded = false;
foreach (var uriSingle in file.Uris)
{
try
{
var splitted = uriSingle.Split(new[] { "//" }, StringSplitOptions.RemoveEmptyEntries);
var splitted2 = splitted[1].Split(new[] { '/' });
var relayserver = splitted2[0];
if (downloaded)
{
break;
}
Logger.Log("Attempting to download {1} from {0} with file size of {2}.", LogLevel.Info, relayserver, file.FileName, file.FileSize);
byte NetThrottle = byte.Parse(update.net_throttle);
if (NetThrottle > 0 || NetThrottle == null)
{
Agent.RV.Utils.NetworkThrottle.DownloadThrottle(filepath, uriSingle, file.FileName, NetThrottle);
}
else
{
client.DownloadFile(uriSingle, filepath);
}
if (File.Exists(filepath))
{
var localFileHash = RvUtils.Md5HashFile(filepath).ToLower();
Logger.Log("Download Complete, {0}", LogLevel.Info, file.FileName);
Logger.Log("Checking MD5 Hash...");
Logger.Log("Incoming Hash: {0}", LogLevel.Info, file.Hash);
Logger.Log("Local Hash: {0}", LogLevel.Info, localFileHash);
downloaded = true;
if (localFileHash != file.Hash.ToLower())
{
Logger.Log("Local file {0} Hash did not match remote's. Retrying with a different server.", LogLevel.Info, file.FileName);
update.error = "Local file Hash did not match remote. Bad file integrity. ";
update.success = false.ToString().ToLower();
downloaded = false;
}
}
else
{
Logger.Log("File {0} did not download. Retrying with a different server.", LogLevel.Info, file.FileName);
update.error = "File did not download successfully, it was not found on disk. Please check download server.";
update.success = false.ToString().ToLower();
downloaded = false;
}
}
catch (Exception e)
{
Logger.Log("File {0} failed to download correctly... Possible connection issue, Retrying with a different server.", LogLevel.Info, file.FileName);
update.error = "File did not download correctly, Exception message: " + e.Message + ". Please check download server connectivity.";
update.success = false.ToString().ToLower();
downloaded = false;
}
}
//Check if the file was successfully downloaded and return.
if (downloaded)
{
update.error = String.Empty;
update.success = true.ToString().ToLower();
}
}
}
catch (Exception e)
{
//Critical exception occurred.
update.error = "Agent update did not download successfully, Exception occured, refer to log for details.";
update.success = false.ToString().ToLower();
Logger.Log("One or more Agent update Files were not downloaded successfully; {0}.",
LogLevel.Error, file.FileName);
Logger.LogException(e);
return(update);
}
ServicePointManager.ServerCertificateValidationCallback = tempCallback;
}
return(update);
}
