taskwarrior/src/TransportSSH.cpp
Johannes Schlatow 042d7b40de Feature #462: url support
- added uri class for proper uri and path handling
2010-10-06 18:52:12 +02:00

122 lines
3.1 KiB
C++

////////////////////////////////////////////////////////////////////////////////
// taskwarrior - a command line task list manager.
//
// Copyright 2010, Johannes Schlatow.
// All rights reserved.
//
// This program is free software; you can redistribute it and/or modify it under
// the terms of the GNU General Public License as published by the Free Software
// Foundation; either version 2 of the License, or (at your option) any later
// version.
//
// This program is distributed in the hope that it will be useful, but WITHOUT
// ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
// FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
// details.
//
// You should have received a copy of the GNU General Public License along with
// this program; if not, write to the
//
// Free Software Foundation, Inc.,
// 51 Franklin Street, Fifth Floor,
// Boston, MA
// 02110-1301
// USA
//
////////////////////////////////////////////////////////////////////////////////
#include "TransportSSH.h"
////////////////////////////////////////////////////////////////////////////////
TransportSSH::TransportSSH(const Uri& uri) : Transport(uri)
{
executable = "scp";
}
////////////////////////////////////////////////////////////////////////////////
void TransportSSH::send(const std::string& source)
{
if (uri.host == "") {
throw std::string ("Hostname is empty");
}
// Is there more than one file to transfer?
// Then path has to end with a '/'
if ( (source.find ("*") != std::string::npos)
|| (source.find ("?") != std::string::npos)
|| (source.find (" ") != std::string::npos) )
{
std::string::size_type pos;
pos = uri.path.find_last_of ("/");
if (pos != uri.path.length()-1)
{
uri.path = uri.path.substr (0, pos+1);
}
}
// cmd line is: scp [-p port] [user@]host:path
if (uri.port != "")
{
arguments.push_back ("-P");
arguments.push_back (uri.port);
}
arguments.push_back (source);
if (uri.user != "")
{
arguments.push_back (uri.user + "@" + uri.host + ":" + uri.path);
}
else
{
arguments.push_back (uri.host + ":" + uri.path);
}
if (execute())
throw std::string ("Failed to run scp!");
}
////////////////////////////////////////////////////////////////////////////////
void TransportSSH::recv(std::string target)
{
if (uri.host == "") {
throw std::string ("Hostname is empty");
}
// Is there more than one file to transfer?
// Then target has to end with a '/'
if ( (uri.path.find ("*") != std::string::npos)
|| (uri.path.find ("?") != std::string::npos) )
{
std::string::size_type pos;
pos = target.find_last_of ("/");
if (pos != target.length()-1)
{
target = target.substr( 0, pos+1);
}
}
// cmd line is: scp [-p port] [user@]host:path
if (uri.port != "")
{
arguments.push_back ("-P");
arguments.push_back (uri.port);
}
if (uri.user != "")
{
arguments.push_back (uri.user + "@" + uri.host + ":" + uri.path);
}
else
{
arguments.push_back (uri.host + ":" + uri.path);
}
arguments.push_back (target);
if (execute())
throw std::string ("Failed to run scp!");
}
////////////////////////////////////////////////////////////////////////////////