From ea82284a78165b6bf0d4e2df1927f0def20dc86f Mon Sep 17 00:00:00 2001
From: "kenneth.skovhede@gmail.com"
 <kenneth.skovhede@gmail.com@59da171f-624f-0410-aa54-27559c288bec>
Date: Thu, 15 Jan 2009 14:47:05 +0000
Subject: [PATCH] A not-so-robust-but-working version of the SSH backend is
 ready.

git-svn-id: https://duplicati.googlecode.com/svn/trunk@65 59da171f-624f-0410-aa54-27559c288bec
---
 Duplicati/CommandLine/UnitTest.cs             |   4 +
 Duplicati/Library/Backend/FTP/FTPBackend.cs   |   4 +-
 .../SSH/Duplicati.Library.Backend.SSH.csproj  |   4 +
 Duplicati/Library/Backend/SSH/SSHBackend.cs   | 199 +++++++++++++++++-
 4 files changed, 203 insertions(+), 8 deletions(-)

diff --git a/Duplicati/CommandLine/UnitTest.cs b/Duplicati/CommandLine/UnitTest.cs
index 7e7482900..b3444ea0a 100644
--- a/Duplicati/CommandLine/UnitTest.cs
+++ b/Duplicati/CommandLine/UnitTest.cs
@@ -93,7 +93,11 @@ namespace Duplicati.CommandLine
 
                 Console.WriteLine("Backing up the full copy: " + folders[0]);
                 using (new Timer("Full backup of " + folders[0]))
+                {
+                    options["full"] = "";
                     Duplicati.Library.Main.Interface.Backup(folders[0], target, options);
+                    options.Remove("full");
+                }
 
                 for (int i = 1; i < folders.Length; i++)
                 {
diff --git a/Duplicati/Library/Backend/FTP/FTPBackend.cs b/Duplicati/Library/Backend/FTP/FTPBackend.cs
index 3be88bfe1..5b1b801e8 100644
--- a/Duplicati/Library/Backend/FTP/FTPBackend.cs
+++ b/Duplicati/Library/Backend/FTP/FTPBackend.cs
@@ -75,7 +75,7 @@ namespace Duplicati.Library.Backend
         #endregion
 
 
-        private Match MatchLine(string line)
+        private static Match MatchLine(string line)
         {
             Match m = null;
             foreach (Regex s in PARSEFORMATS)
@@ -85,7 +85,7 @@ namespace Duplicati.Library.Backend
             return null;
         }
 
-        private FileEntry ParseLine(string line)
+        public static FileEntry ParseLine(string line)
         {
             Match m = MatchLine(line);
             if (m == null)
diff --git a/Duplicati/Library/Backend/SSH/Duplicati.Library.Backend.SSH.csproj b/Duplicati/Library/Backend/SSH/Duplicati.Library.Backend.SSH.csproj
index 10e18cf2a..4b2e45c7f 100644
--- a/Duplicati/Library/Backend/SSH/Duplicati.Library.Backend.SSH.csproj
+++ b/Duplicati/Library/Backend/SSH/Duplicati.Library.Backend.SSH.csproj
@@ -41,6 +41,10 @@
       <Project>{994F3D0A-E91E-4EEE-80EE-064485E09909}</Project>
       <Name>Duplicati.Library.Backend.BackendInterface</Name>
     </ProjectReference>
+    <ProjectReference Include="..\FTP\Duplicati.Library.Backend.FTP.csproj">
+      <Project>{F61679A9-E5DE-468A-B5A4-05F92D0143D2}</Project>
+      <Name>Duplicati.Library.Backend.FTP</Name>
+    </ProjectReference>
   </ItemGroup>
   <Import Project="$(MSBuildBinPath)\Microsoft.CSharp.targets" />
   <!-- To modify your build process, add your task inside one of the targets below and uncomment it. 
diff --git a/Duplicati/Library/Backend/SSH/SSHBackend.cs b/Duplicati/Library/Backend/SSH/SSHBackend.cs
index 4e6d3d334..95885630b 100644
--- a/Duplicati/Library/Backend/SSH/SSHBackend.cs
+++ b/Duplicati/Library/Backend/SSH/SSHBackend.cs
@@ -20,15 +20,22 @@
 using System;
 using System.Collections.Generic;
 using System.Text;
+using System.Text.RegularExpressions;
 
 namespace Duplicati.Library.Backend
 {
     public class SSH : IBackendInterface
     {
-
+        private string m_server;
         private string m_path;
+        private string m_username;
+        private string m_password;
         Dictionary<string, string> m_options;
 
+        private string m_sftp;
+        private string m_scp;
+        private string m_ssh_options;
+
         public SSH()
         {
         }
@@ -36,7 +43,55 @@ namespace Duplicati.Library.Backend
         public SSH(string url, Dictionary<string, string> options)
         {
             m_options = options;
-            //m_path = ExtractUrlParts(url, m_options);
+            Uri u = new Uri(url);
+            if (!string.IsNullOrEmpty(u.UserInfo))
+            {
+                if (u.UserInfo.IndexOf(":") >= 0)
+                {
+                    m_username = u.UserInfo.Substring(0, u.UserInfo.IndexOf(":"));
+                    m_password = u.UserInfo.Substring(u.UserInfo.IndexOf(":") + 1);
+                }
+                else
+                {
+                    m_username = u.UserInfo;
+                    if (options.ContainsKey("ftp_password"))
+                        m_password = options["ftp_password"];
+                }
+            }
+
+            m_path = u.AbsolutePath.Substring(1);
+            if (!m_path.EndsWith("/"))
+                m_path += "/";
+
+            m_server = u.Host;
+
+            if (options.ContainsKey("sftp-command"))
+                m_sftp = options["sftp-command"];
+            else
+            {
+                if (System.Environment.OSVersion.Platform == PlatformID.Unix || System.Environment.OSVersion.Platform == PlatformID.MacOSX)
+                    m_sftp = "sftp";
+                else
+                    m_sftp = "psftp.exe";
+            }
+
+            if (options.ContainsKey("scp-command"))
+                m_scp = options["scp-command"];
+            else
+            {
+                if (System.Environment.OSVersion.Platform == PlatformID.Unix || System.Environment.OSVersion.Platform == PlatformID.MacOSX)
+                    m_scp = "scp";
+                else
+                    m_scp = "pscp.exe";
+            }
+
+            if (options.ContainsKey("ssh-options"))
+                m_ssh_options = options["ssh-options"];
+            else
+                m_ssh_options = "-C";
+
+            if (!u.IsDefaultPort)
+                m_ssh_options += " -P " + u.Port;
         }
 
         #region IBackendInterface Members
@@ -53,24 +108,156 @@ namespace Duplicati.Library.Backend
 
         public List<FileEntry> List()
         {
-            throw new Exception("The method or operation is not implemented.");
+            List<FileEntry> files = new List<FileEntry>();
+
+            using (System.Diagnostics.Process p = GetConnection(true, null))
+            {
+                p.StandardInput.WriteLine("ls");
+                p.StandardInput.WriteLine("exit");
+
+                string s;
+                while ((s = p.StandardOutput.ReadLine()) != null)
+                {
+                    FileEntry fe = FTP.ParseLine(s);
+                    if (fe != null && fe.Name != "." && fe.Name != "..")
+                        files.Add(fe);
+                }
+
+                if (!p.WaitForExit(5000))
+                {
+                    p.Close();
+                    throw new Exception("Timeout while closing session");
+                }
+                
+                return files;
+            }
+
         }
 
+
         public void Put(string remotename, string filename)
         {
-            throw new Exception("The method or operation is not implemented.");
+            using (System.Diagnostics.Process p = GetConnection(false, "\"" + filename + "\" \"" + BuildEscapedPath(remotename) + "\""))
+            {
+                if (!p.WaitForExit(5 * 60 * 1000))
+                {
+                    p.Close();
+                    throw new Exception("Timeout while uploading file");
+                }
+
+                if (p.StandardError.Peek() != -1)
+                    throw new Exception(p.StandardError.ReadToEnd());
+            }
         }
 
         public void Get(string remotename, string filename)
         {
-            throw new Exception("The method or operation is not implemented.");
+            using (System.Diagnostics.Process p = GetConnection(false, "\"" + BuildEscapedPath(remotename) + "\" \"" + filename + "\""))
+            {
+                if (!p.WaitForExit(5 * 60 * 1000))
+                {
+                    p.Close();
+                    throw new Exception("Timeout while uploading file");
+                }
+
+                if (p.StandardError.Peek() != -1)
+                    throw new Exception(p.StandardError.ReadToEnd());
+            }
         }
 
         public void Delete(string remotename)
         {
-            throw new Exception("The method or operation is not implemented.");
+            using (System.Diagnostics.Process p = GetConnection(true, null))
+            {
+                p.StandardInput.WriteLine("rm \"" + remotename + "\"");
+                p.StandardInput.WriteLine("exit");
+                if (!p.WaitForExit(5000))
+                {
+                    p.Close();
+                    throw new Exception("Timeout while closing session");
+                }
+            }
         }
 
         #endregion
+
+        private string BuildEscapedPath(string remotename)
+        {
+            string path = m_path + remotename;
+
+            path = m_server + ":" + path;
+            if (!string.IsNullOrEmpty(m_username))
+                path = m_username + "@" + path;
+
+            return path;
+        }
+
+        private System.Diagnostics.Process GetConnection(bool sftp, string args)
+        {
+            System.Diagnostics.Process p = new System.Diagnostics.Process();
+            p.StartInfo.CreateNoWindow = true;
+            p.StartInfo.RedirectStandardError = true;
+            p.StartInfo.RedirectStandardInput = true;
+            p.StartInfo.RedirectStandardOutput = true;
+            p.StartInfo.UseShellExecute = false;
+            p.StartInfo.WindowStyle = System.Diagnostics.ProcessWindowStyle.Hidden;
+            p.StartInfo.FileName = sftp ? m_sftp : m_scp;
+            p.StartInfo.Arguments = m_ssh_options;
+
+            if (sftp)
+                p.StartInfo.Arguments += " " + m_server;
+
+            if (args != null)
+                p.StartInfo.Arguments += " " + args;
+
+            p.Start();
+
+            //TODO: This is not the most robust way of dealing with
+            // the psftp and pscp commands. It is likely not very portable either
+
+            System.Text.StringBuilder prompts = new StringBuilder();
+
+            char[] tmp = new char[1000];
+            int t = p.StandardOutput.Read(tmp, 0, tmp.Length);
+            string greeting = new string(tmp, 0, t);
+            prompts.Append(greeting);
+
+            if (!string.IsNullOrEmpty(m_username) && greeting.Trim() == "login as:")
+            {
+                p.StandardInput.WriteLine(m_username);
+                t = p.StandardOutput.Read(tmp, 0, tmp.Length);
+                greeting = new string(tmp, 0, t);
+                prompts.Append(greeting);
+            }
+
+            if (!string.IsNullOrEmpty(m_password) && greeting.Trim().ToLower().EndsWith(" password:"))
+            {
+                p.StandardInput.WriteLine(m_password);
+
+                t = p.StandardOutput.Read(tmp, 0, tmp.Length);
+                greeting = new string(tmp, 0, t);
+                prompts.Append(greeting);
+            }
+
+            if (sftp && !greeting.Trim().ToLower().StartsWith("remote working directory"))
+            {
+                //Sometime the message comes a little delayed later
+                do
+                {
+                    t = p.StandardOutput.Read(tmp, 0, tmp.Length);
+                    greeting = new string(tmp, 0, t);
+                    prompts.Append(greeting);
+                } while (greeting.Length > 0 && greeting.Length < 4);
+
+                if (!greeting.Trim().ToLower().StartsWith("remote working directory"))
+                    throw new Exception("Failed to login, prompts were: " + prompts.ToString());
+            }
+
+            if (sftp)
+                if (!string.IsNullOrEmpty(m_path))
+                    p.StandardInput.WriteLine("cd \"" + m_path + "\"");
+
+            return p;
+        }
     }
 }