public AutoUpdateResult CheckForUpdates(int lobbyId)
 {
     try
     {
         //Check current version, send updated files
         return(AutoUpdateResult.RetrieveFileList(lobbyId));
     }
     catch (Exception error)
     {
         Error.Write(error);
         throw;
     }
 }
        public void GetPendingUpdateQueues()
        {
            LobbyResult lobbyResult = new LobbyResult()
            {
                Host             = "www.test.com",
                LobbyId          = 1,
                LobbyIdSpecified = true,
                Name             = "TestUpdate"
            };



            string testPath = Path.Combine(Allegiance.CommunitySecuritySystem.Client.Integration.AllegianceRegistry.LobbyPath, lobbyResult.Name);

            if (Directory.Exists(testPath) == false)
            {
                Directory.CreateDirectory(testPath);
            }

            File.WriteAllText(Path.Combine(testPath, "test1.txt"), "this is a test file. It is for testing.");

            var encryption = new Encryption <SHA1>();
            var checksum   = encryption.Calculate(Path.Combine(testPath, "test1.txt"));

            AutoUpdateResult autoUpdateResult = new AutoUpdateResult()
            {
                AutoUpdateBaseAddress = "http://www.pork.com",
                Files = new FindAutoUpdateFilesResult[]
                {
                    new FindAutoUpdateFilesResult()
                    {
                        AutoUpdateFileId          = 1,
                        AutoUpdateFileIdSpecified = true,
                        CurrentVersion            = "1.0.0.0",
                        DateCreated           = DateTime.Parse("4/8/2014"),
                        DateCreatedSpecified  = true,
                        DateModified          = DateTime.Parse("4/8/2014"),
                        DateModifiedSpecified = true,
                        Filename             = "test1-notfound.txt",
                        IsProtected          = false,
                        IsProtectedSpecified = true,
                        LobbyId          = 1,
                        LobbyIdSpecified = true,
                        ValidChecksum    = checksum
                    }
                }
            };

            if (File.Exists("autoupdate.ds") == true)
            {
                File.Delete("autoupdate.ds");
            }

            List <FindAutoUpdateFilesResult> result = Client.Service.AutoUpdate.ProcessPendingUpdates(lobbyResult, autoUpdateResult);

            Assert.AreEqual(1, result.Count, "File was not found, one update should have applied.");

            autoUpdateResult.Files[0].Filename = "test1.txt";

            result = Client.Service.AutoUpdate.ProcessPendingUpdates(lobbyResult, autoUpdateResult);

            Assert.AreEqual(0, result.Count, "No updates should have been done, checksums match.");

            autoUpdateResult.Files[0].ValidChecksum = "INVALID_CHECKSUM";

            result = Client.Service.AutoUpdate.ProcessPendingUpdates(lobbyResult, autoUpdateResult);

            Assert.AreEqual(1, result.Count, "One update should have been done, checksum miss-match with no prior update record.");

            var autoUpdate = DataStore.Open("autoupdate.ds", "Ga46^#a042");

            var lastResults = new Dictionary <string, FindAutoUpdateFilesResult>();
            //foreach (var res in result)
            //	lastResults.Add(res.Filename, res);

            string dataKey = "Files_" + lobbyResult.LobbyId;

            autoUpdate[dataKey] = lastResults;

            autoUpdate.Save();

            result = Client.Service.AutoUpdate.ProcessPendingUpdates(lobbyResult, autoUpdateResult);

            Assert.AreEqual(1, result.Count, "One update should have been done, checksum miss-match prior update dictionary exists, but no record for file.");

            autoUpdateResult.Files[0].ValidChecksum = checksum;

            foreach (var res in result)
            {
                lastResults.Add(res.Filename, res);
            }

            autoUpdate[dataKey] = lastResults;

            autoUpdate.Save();

            result = Client.Service.AutoUpdate.ProcessPendingUpdates(lobbyResult, autoUpdateResult);

            Assert.AreEqual(0, result.Count, "No updates should have been done, versions match.");

            autoUpdateResult.Files[0].CurrentVersion = "1.0.0.1";

            result = Client.Service.AutoUpdate.ProcessPendingUpdates(lobbyResult, autoUpdateResult);

            Assert.AreEqual(1, result.Count, "File version miss-match with previous version of file, update applied.");

            autoUpdateResult.Files[0].CurrentVersion = "1.0.0.0";
            autoUpdateResult.Files[0].IsProtected    = false;
            autoUpdateResult.Files[0].ValidChecksum  = "INVALID_CHECKSUM";

            result = Client.Service.AutoUpdate.ProcessPendingUpdates(lobbyResult, autoUpdateResult);

            Assert.AreEqual(1, result.Count, "updates should have been done, versions are same, checksum is different, file is not protected and the user has not modified the file.");

            autoUpdateResult.Files[0].DateModified = DateTime.Parse("1/1/2015");

            result = Client.Service.AutoUpdate.ProcessPendingUpdates(lobbyResult, autoUpdateResult);

            Assert.AreEqual(0, result.Count, "updates should not have been done, versions are same, checksum is different, file is not protected and the user has modified the file since the last update.");

            //autoUpdateResult.Files[0].DateModified = DateTime.Parse("4/8/2014");

            autoUpdateResult.Files[0].CurrentVersion = "1.0.0.0";
            autoUpdateResult.Files[0].IsProtected    = true;
            autoUpdateResult.Files[0].ValidChecksum  = "INVALID_CHECKSUM";

            result = Client.Service.AutoUpdate.ProcessPendingUpdates(lobbyResult, autoUpdateResult);

            Assert.AreEqual(1, result.Count, "updates should have been done, versions are same, checksum is different, file is protected and the user has modified the file.");
        }
        public void GetPendingUpdateQueues()
        {
            LobbyResult lobbyResult = new LobbyResult()
            {
                Host = "www.test.com",
                LobbyId = 1,
                LobbyIdSpecified = true,
                Name = "TestUpdate"
            };

            string testPath = Path.Combine(Allegiance.CommunitySecuritySystem.Client.Integration.AllegianceRegistry.LobbyPath, lobbyResult.Name);

            if(Directory.Exists(testPath) == false)
                Directory.CreateDirectory(testPath);

            File.WriteAllText(Path.Combine(testPath, "test1.txt"), "this is a test file. It is for testing.");

            var encryption = new Encryption<SHA1>();
            var checksum = encryption.Calculate(Path.Combine(testPath, "test1.txt"));

            AutoUpdateResult autoUpdateResult = new AutoUpdateResult()
            {
                AutoUpdateBaseAddress = "http://www.pork.com",
                Files = new FindAutoUpdateFilesResult[]
                {
                    new FindAutoUpdateFilesResult()
                    {
                        AutoUpdateFileId = 1,
                        AutoUpdateFileIdSpecified = true,
                         CurrentVersion = "1.0.0.0",
                         DateCreated = DateTime.Parse("4/8/2014"),
                         DateCreatedSpecified = true,
                         DateModified = DateTime.Parse("4/8/2014"),
                         DateModifiedSpecified = true,
                         Filename = "test1-notfound.txt",
                         IsProtected = false,
                         IsProtectedSpecified = true,
                         LobbyId = 1,
                         LobbyIdSpecified = true,
                         ValidChecksum = checksum
                    }
                }
            };

            if (File.Exists("autoupdate.ds") == true)
                File.Delete("autoupdate.ds");

            List<FindAutoUpdateFilesResult> result = Client.Service.AutoUpdate.ProcessPendingUpdates(lobbyResult, autoUpdateResult);

            Assert.AreEqual(1, result.Count, "File was not found, one update should have applied.");

            autoUpdateResult.Files[0].Filename = "test1.txt";

            result = Client.Service.AutoUpdate.ProcessPendingUpdates(lobbyResult, autoUpdateResult);

            Assert.AreEqual(0, result.Count, "No updates should have been done, checksums match.");

            autoUpdateResult.Files[0].ValidChecksum = "INVALID_CHECKSUM";

            result = Client.Service.AutoUpdate.ProcessPendingUpdates(lobbyResult, autoUpdateResult);

            Assert.AreEqual(1, result.Count, "One update should have been done, checksum miss-match with no prior update record.");

            var autoUpdate = DataStore.Open("autoupdate.ds", "Ga46^#a042");

            var lastResults = new Dictionary<string, FindAutoUpdateFilesResult>();
            //foreach (var res in result)
            //	lastResults.Add(res.Filename, res);

            string dataKey = "Files_" + lobbyResult.LobbyId;
            autoUpdate[dataKey] = lastResults;

            autoUpdate.Save();

            result = Client.Service.AutoUpdate.ProcessPendingUpdates(lobbyResult, autoUpdateResult);

            Assert.AreEqual(1, result.Count, "One update should have been done, checksum miss-match prior update dictionary exists, but no record for file.");

            autoUpdateResult.Files[0].ValidChecksum = checksum;

            foreach (var res in result)
                lastResults.Add(res.Filename, res);

            autoUpdate[dataKey] = lastResults;

            autoUpdate.Save();

            result = Client.Service.AutoUpdate.ProcessPendingUpdates(lobbyResult, autoUpdateResult);

            Assert.AreEqual(0, result.Count, "No updates should have been done, versions match.");

            autoUpdateResult.Files[0].CurrentVersion = "1.0.0.1";

            result = Client.Service.AutoUpdate.ProcessPendingUpdates(lobbyResult, autoUpdateResult);

            Assert.AreEqual(1, result.Count, "File version miss-match with previous version of file, update applied.");

            autoUpdateResult.Files[0].CurrentVersion = "1.0.0.0";
            autoUpdateResult.Files[0].IsProtected = false;
            autoUpdateResult.Files[0].ValidChecksum = "INVALID_CHECKSUM";

            result = Client.Service.AutoUpdate.ProcessPendingUpdates(lobbyResult, autoUpdateResult);

            Assert.AreEqual(1, result.Count, "updates should have been done, versions are same, checksum is different, file is not protected and the user has not modified the file.");

            autoUpdateResult.Files[0].DateModified = DateTime.Parse("1/1/2015");

            result = Client.Service.AutoUpdate.ProcessPendingUpdates(lobbyResult, autoUpdateResult);

            Assert.AreEqual(0, result.Count, "updates should not have been done, versions are same, checksum is different, file is not protected and the user has modified the file since the last update.");

            //autoUpdateResult.Files[0].DateModified = DateTime.Parse("4/8/2014");

            autoUpdateResult.Files[0].CurrentVersion = "1.0.0.0";
            autoUpdateResult.Files[0].IsProtected = true;
            autoUpdateResult.Files[0].ValidChecksum = "INVALID_CHECKSUM";

            result = Client.Service.AutoUpdate.ProcessPendingUpdates(lobbyResult, autoUpdateResult);

            Assert.AreEqual(1, result.Count, "updates should have been done, versions are same, checksum is different, file is protected and the user has modified the file.");
        }
예제 #4
0
        public static List <FindAutoUpdateFilesResult> ProcessPendingUpdates(LobbyResult lobby, AutoUpdateResult results)
        {
            //Initialize Checksum class to use SHA1
            var encryption = new Encryption <SHA1>();

            //var root        = AllegianceRegistry.EXEPath; //Allegiance root directory
            var    autoUpdate  = DataStore.Open("autoupdate.ds", "Ga46^#a042");
            string dataKey     = "Files_" + lobby.LobbyId;
            var    lastResults = autoUpdate[dataKey] as Dictionary <string, FindAutoUpdateFilesResult>;
            List <FindAutoUpdateFilesResult> updateQueue = new List <FindAutoUpdateFilesResult>();

            //Check files which need update
            foreach (var file in results.Files)
            {
                DebugDetector.AssertCheckRunning();

                // You can put this in if you are testing the launcher with the production CSS server.
                // Turn off file protection on the launcher on the CSS server's Auto Update version of the launcher,
                // then drop the debug version of the launcher into your local game directory. Then you can
                // launch the launcher from the debugger and work with it.
//#if DEBUG
//                if (file.Filename.EndsWith("Launcher.exe", StringComparison.InvariantCultureIgnoreCase) == true
//                    || file.Filename.EndsWith("Launcher.pdb", StringComparison.InvariantCultureIgnoreCase) == true)
//                    continue;
//#endif

                string path;

                if (string.Equals(file.Filename, GlobalSettings.ClientExecutableName) ||
                    string.Equals(file.Filename, GlobalSettings.ClientExecutablePDB))
                {
                    path = Path.Combine(AllegianceRegistry.LobbyPath, file.Filename);
                }
                else
                {
                    path = GetLobbyPath(lobby, file.Filename);
                }

                //Check that all files exist
                if (!File.Exists(path))
                {
                    Log.Write("File did not exist: " + path + ", will update.");
                    updateQueue.Add(file);
                    continue;
                }

                //Check that all file versions match
                if (lastResults != null && lastResults.ContainsKey(file.Filename) &&
                    (file.CurrentVersion != lastResults[file.Filename].CurrentVersion))
                {
                    Log.Write("File version mismatch, will update: " + file.Filename + ", server version: " + file.CurrentVersion + ", local version: " + lastResults[file.Filename].CurrentVersion);

                    updateQueue.Add(file);
                    continue;
                }

                // Test for checksum match, and if they don't replace the file if:
                // * This is the first time AutoUpdate has run for this installation.
                // * This is the first time this file has been seen by AutoUpdate
                // * The file has the protected flag turned on.
                var checksum = encryption.Calculate(path);
                if (!string.Equals(file.ValidChecksum, checksum))
                {
                    // If there haven't been any updates at all, and the checksums don't match, then overwrite the unknown file. (old installation / left over files from a previous uninstall)
                    if (lastResults == null)
                    {
                        Log.Write("No prior autoupdate records, will updated: " + path);
                        updateQueue.Add(file);
                        continue;
                    }

                    // If there wasn't a prior update applied for the file, and the checksums don't match, then overwrite the unknown file. (An older file is on disk, and was re-added to the auto update system)
                    if (!lastResults.ContainsKey(file.Filename))
                    {
                        Log.Write("No record of file in autoupdate history, will update: " + path);
                        updateQueue.Add(file);
                        continue;
                    }

                    // If the file was changed on the server, but not modified by the user, then update.
                    if (lastResults[file.Filename].DateModified == file.DateModified)
                    {
                        Log.Write("file was not user modified, and a change was detected from AU, will update: " + path);
                        updateQueue.Add(file);
                        continue;
                    }

                    // If the file is protected and the hashes don't match, then update.
                    if (file.IsProtected)
                    {
                        Log.Write("File checksum mismatch, will update: " + path + ", server checksum: " + file.ValidChecksum + ", local checksum: " + checksum);
                        updateQueue.Add(file);
                        continue;
                    }
                }
            }

            return(updateQueue);
        }
		public static List<FindAutoUpdateFilesResult> ProcessPendingUpdates(LobbyResult lobby, AutoUpdateResult results)
		{
			//Initialize Checksum class to use SHA1
			var encryption = new Encryption<SHA1>();

			//var root        = AllegianceRegistry.EXEPath; //Allegiance root directory
			var autoUpdate = DataStore.Open("autoupdate.ds", "Ga46^#a042");
			string dataKey = "Files_" + lobby.LobbyId;
			var lastResults = autoUpdate[dataKey] as Dictionary<string, FindAutoUpdateFilesResult>;
			List<FindAutoUpdateFilesResult> updateQueue = new List<FindAutoUpdateFilesResult>();

			//Check files which need update
			foreach (var file in results.Files)
			{
				DebugDetector.AssertCheckRunning();

				// You can put this in if you are testing the launcher with the production CSS server. 
				// Turn off file protection on the launcher on the CSS server's Auto Update version of the launcher, 
				// then drop the debug version of the launcher into your local game directory. Then you can 
				// launch the launcher from the debugger and work with it.
//#if DEBUG
//                if (file.Filename.EndsWith("Launcher.exe", StringComparison.InvariantCultureIgnoreCase) == true
//                    || file.Filename.EndsWith("Launcher.pdb", StringComparison.InvariantCultureIgnoreCase) == true)
//                    continue;
//#endif 

				string path;

				if (string.Equals(file.Filename, GlobalSettings.ClientExecutableName)
					|| string.Equals(file.Filename, GlobalSettings.ClientExecutablePDB))
					path = Path.Combine(AllegianceRegistry.LobbyPath, file.Filename);
				else
					path = GetLobbyPath(lobby, file.Filename);

				//Check that all files exist
				if (!File.Exists(path))
				{
					Log.Write("File did not exist: " + path + ", will update.");
					updateQueue.Add(file);
					continue;
				}

				//Check that all file versions match
				if (lastResults != null && lastResults.ContainsKey(file.Filename)
					&& (file.CurrentVersion != lastResults[file.Filename].CurrentVersion))
				{
					Log.Write("File version mismatch, will update: " + file.Filename + ", server version: " + file.CurrentVersion + ", local version: " + lastResults[file.Filename].CurrentVersion);

					updateQueue.Add(file);
					continue;
				}

				// Test for checksum match, and if they don't replace the file if:
				// * This is the first time AutoUpdate has run for this installation.
				// * This is the first time this file has been seen by AutoUpdate
				// * The file has the protected flag turned on.
				var checksum = encryption.Calculate(path);
				if (!string.Equals(file.ValidChecksum, checksum))
				{
					// If there haven't been any updates at all, and the checksums don't match, then overwrite the unknown file. (old installation / left over files from a previous uninstall)
					if (lastResults == null)
					{
						Log.Write("No prior autoupdate records, will updated: " + path);
						updateQueue.Add(file);
						continue;
					}

					// If there wasn't a prior update applied for the file, and the checksums don't match, then overwrite the unknown file. (An older file is on disk, and was re-added to the auto update system)
					if (!lastResults.ContainsKey(file.Filename))
					{
						Log.Write("No record of file in autoupdate history, will update: " + path);
						updateQueue.Add(file);
						continue;
					}

					// If the file was changed on the server, but not modified by the user, then update.
					if (lastResults[file.Filename].DateModified == file.DateModified)
					{
						Log.Write("file was not user modified, and a change was detected from AU, will update: " + path);
						updateQueue.Add(file);
						continue;
					}

					// If the file is protected and the hashes don't match, then update.
					if (file.IsProtected)
					{
						Log.Write("File checksum mismatch, will update: " + path + ", server checksum: " + file.ValidChecksum + ", local checksum: " + checksum);
						updateQueue.Add(file);
						continue;
					}
				}
			}

			return updateQueue;
		}