duplicati/Duplicati/Library/Main/ForestHash/Operation/RestoreControlFilesHandler.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;

namespace Duplicati.Library.Main.ForestHash.Operation
{
    internal class RestoreControlFilesHandler : IDisposable
    {
        private FhOptions m_options;
        private RestoreStatistics m_stat;
        private string m_backendurl;
        private string m_target;

        public RestoreControlFilesHandler(string backendurl, FhOptions options, RestoreStatistics stat, string target)
        {
            m_options = options;
            m_stat = stat;
            m_target = target;
            m_backendurl = backendurl;
        }

        public void Run()
        {
            using (var tmpdb = new Utility.TempFile())
            using (var db = new Database.LocalDatabase(System.IO.File.Exists(m_options.Fhdbpath) ? m_options.Fhdbpath : (string)tmpdb, "RestoreControlFiles"))
            using (var backend = new FhBackend(m_backendurl, m_options, m_stat, db))
            {
            	try
            	{
	                var files = from file in backend.List()
	                            let p = Volumes.VolumeBase.ParseFilename(file)
	                            where p != null && p.FileType == RemoteVolumeType.Files
	                            orderby p.Time
	                            select file;
	
	                Exception lastEx = new Exception("No suitable files found on remote target");
	
	                foreach(var file in files)
	                    try
	                    {
	                        long size;
	                        string hash;
	                        RemoteVolumeType type;
	                        RemoteVolumeState state;
	                        if (!db.GetRemoteVolume(file.Name, out hash, out size, out type, out state))
	                            size = file.Size;
	
							using (var tmpfile = backend.Get(file.Name, size, hash))
	                        using (var tmp = new Volumes.FilesetVolumeReader(RestoreHandler.GetCompressionModule(file.Name), tmpfile, m_options))
	                            foreach (var cf in tmp.ControlFiles)
	                                using (var ts = System.IO.File.Create(System.IO.Path.Combine(m_target, cf.Key)))
	                                    Utility.Utility.CopyStream(cf.Value, ts);
	                        
	                        lastEx = null;
	                        break;
	                    }
	                    catch(Exception ex)
	                    {
	                        lastEx = ex;
	                    }
	
	                if (lastEx != null)
	                    throw lastEx;
	        	}
	        	finally
	        	{
	        		backend.WaitForComplete(db, null);
	        	}
            }
        }

        public void Dispose()
        {
        }
    }
}