|
Packit |
8f70b4 |
/*
|
|
Packit |
8f70b4 |
* lftp - file transfer program
|
|
Packit |
8f70b4 |
*
|
|
Packit |
8f70b4 |
* Copyright (c) 1996-2015 by Alexander V. Lukyanov (lav@yars.free.net)
|
|
Packit |
8f70b4 |
*
|
|
Packit |
8f70b4 |
* This program is free software; you can redistribute it and/or modify
|
|
Packit |
8f70b4 |
* it under the terms of the GNU General Public License as published by
|
|
Packit |
8f70b4 |
* the Free Software Foundation; either version 3 of the License, or
|
|
Packit |
8f70b4 |
* (at your option) any later version.
|
|
Packit |
8f70b4 |
*
|
|
Packit |
8f70b4 |
* This program is distributed in the hope that it will be useful,
|
|
Packit |
8f70b4 |
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
Packit |
8f70b4 |
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
Packit |
8f70b4 |
* GNU General Public License for more details.
|
|
Packit |
8f70b4 |
*
|
|
Packit |
8f70b4 |
* You should have received a copy of the GNU General Public License
|
|
Packit |
8f70b4 |
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
Packit |
8f70b4 |
*/
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
#include <config.h>
|
|
Packit |
8f70b4 |
#include <sys/types.h>
|
|
Packit |
8f70b4 |
#include <sys/stat.h>
|
|
Packit |
8f70b4 |
#include <fcntl.h>
|
|
Packit |
8f70b4 |
#include <stddef.h>
|
|
Packit |
8f70b4 |
#ifdef HAVE_UNISTD_H
|
|
Packit |
8f70b4 |
#include <unistd.h>
|
|
Packit |
8f70b4 |
#endif
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
#include "GetJob.h"
|
|
Packit |
8f70b4 |
#include "misc.h"
|
|
Packit |
8f70b4 |
#include "ArgV.h"
|
|
Packit |
8f70b4 |
#include "url.h"
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
#define super CopyJobEnv
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
FileCopyPeer *GetJob::SrcLocal(const char *src)
|
|
Packit |
8f70b4 |
{
|
|
Packit |
8f70b4 |
const char *f=(cwd && src[0]!='/') ? dir_file(cwd,src) : src;
|
|
Packit |
8f70b4 |
return FileCopyPeerFDStream::NewGet(f);
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
FileCopyPeer *GetJob::DstLocal(const char *dst)
|
|
Packit |
8f70b4 |
{
|
|
Packit |
8f70b4 |
bool clobber=(cont || QueryBool("xfer:clobber",0));
|
|
Packit |
8f70b4 |
int flags=O_WRONLY|O_CREAT|(truncate_target_first?O_TRUNC:0)|(clobber?0:O_EXCL);
|
|
Packit |
8f70b4 |
dst=expand_home_relative(dst);
|
|
Packit |
8f70b4 |
const char *f=(cwd && dst[0]!='/') ? dir_file(cwd,dst) : dst;
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
FileStream *local=new FileStream(f,flags);
|
|
Packit |
8f70b4 |
FileCopyPeerFDStream *p=new FileCopyPeerFDStream(local,FileCopyPeer::PUT);
|
|
Packit |
8f70b4 |
p->CloseWhenDone();
|
|
Packit |
8f70b4 |
return p;
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
bool GetJob::IsRemoteNonURL(const ParsedURL &url,FA::open_mode mode)
|
|
Packit |
8f70b4 |
{
|
|
Packit |
8f70b4 |
// store & put || !store & get
|
|
Packit |
8f70b4 |
return (!url.proto && ((mode==FA::STORE)^!reverse));
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
bool GetJob::IsLocalNonURL(const ParsedURL &url,FA::open_mode mode)
|
|
Packit |
8f70b4 |
{
|
|
Packit |
8f70b4 |
// store & get || !store & put
|
|
Packit |
8f70b4 |
return (!url.proto && ((mode==FA::STORE)^reverse));
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
bool GetJob::IsLocal(const ParsedURL &url)
|
|
Packit |
8f70b4 |
{
|
|
Packit |
8f70b4 |
return !url.proto || !strcasecmp(url.proto,"file");
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
// create copy peer from a cloned session
|
|
Packit |
8f70b4 |
FileCopyPeer *GetJob::CreateCopyPeer(FileAccess *session,const char *path,FA::open_mode mode)
|
|
Packit |
8f70b4 |
{
|
|
Packit |
8f70b4 |
ParsedURL url(path,true);
|
|
Packit |
8f70b4 |
if(IsRemoteNonURL(url,mode))
|
|
Packit |
8f70b4 |
return new FileCopyPeerFA(session,path,mode);
|
|
Packit |
8f70b4 |
Delete(session); // delete cloned session.
|
|
Packit |
8f70b4 |
return CreateCopyPeer(url,path,mode);
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
// create copy peer using a session reference
|
|
Packit |
8f70b4 |
FileCopyPeer *GetJob::CreateCopyPeer(const FileAccessRef& session,const char *path,FA::open_mode mode)
|
|
Packit |
8f70b4 |
{
|
|
Packit |
8f70b4 |
ParsedURL url(path,true);
|
|
Packit |
8f70b4 |
if(IsRemoteNonURL(url,mode)) {
|
|
Packit |
8f70b4 |
if(parallel>1) // need to clone the session for parallel transfers
|
|
Packit |
8f70b4 |
return new FileCopyPeerFA(session->Clone(),path,mode);
|
|
Packit |
8f70b4 |
return new FileCopyPeerFA(session,path,mode);
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
return CreateCopyPeer(url,path,mode);
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
FileCopyPeer *GetJob::CreateCopyPeer(const ParsedURL &url,const char *path,FA::open_mode mode)
|
|
Packit |
8f70b4 |
{
|
|
Packit |
8f70b4 |
if(IsLocalNonURL(url,mode))
|
|
Packit |
8f70b4 |
return CreateCopyPeer(path,mode);
|
|
Packit |
8f70b4 |
if(IsLocal(url))
|
|
Packit |
8f70b4 |
return CreateCopyPeer(url.path,mode);
|
|
Packit |
8f70b4 |
return new FileCopyPeerFA(&url,mode);
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
FileCopyPeer *GetJob::CreateCopyPeer(const char *path,FA::open_mode mode)
|
|
Packit |
8f70b4 |
{
|
|
Packit |
8f70b4 |
return mode==FA::STORE ? DstLocal(path) : SrcLocal(path);
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
void GetJob::NextFile()
|
|
Packit |
8f70b4 |
{
|
|
Packit |
8f70b4 |
try_next:
|
|
Packit |
8f70b4 |
if(!args)
|
|
Packit |
8f70b4 |
return;
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
const char *src=args->getnext();
|
|
Packit |
8f70b4 |
const char *dst=args->getnext();
|
|
Packit |
8f70b4 |
if(!src || !dst)
|
|
Packit |
8f70b4 |
return;
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
FileCopyPeer *dst_peer=CreateCopyPeer(session,dst,FA::STORE);
|
|
Packit |
8f70b4 |
if(!dst_peer)
|
|
Packit |
8f70b4 |
goto try_next;
|
|
Packit |
8f70b4 |
if(make_dirs)
|
|
Packit |
8f70b4 |
dst_peer->MakeTargetDir();
|
|
Packit |
8f70b4 |
FileCopyPeer *src_peer=CreateCopyPeer(session,src,FA::RETRIEVE);
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
FileCopy *c=FileCopy::New(src_peer,dst_peer,cont);
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
if(delete_files)
|
|
Packit |
8f70b4 |
c->RemoveSourceLater();
|
|
Packit |
8f70b4 |
if(remove_target_first)
|
|
Packit |
8f70b4 |
c->RemoveTargetFirst();
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
AddCopier(c,src);
|
|
Packit |
8f70b4 |
}
|
|
Packit |
8f70b4 |
|
|
Packit |
8f70b4 |
GetJob::GetJob(FileAccess *s,ArgV *a,bool c)
|
|
Packit |
8f70b4 |
: CopyJobEnv(s,a,c)
|
|
Packit |
8f70b4 |
{
|
|
Packit |
8f70b4 |
make_dirs=false;
|
|
Packit |
8f70b4 |
delete_files=false;
|
|
Packit |
8f70b4 |
remove_target_first=false;
|
|
Packit |
8f70b4 |
truncate_target_first=!cont;
|
|
Packit |
8f70b4 |
reverse=false;
|
|
Packit |
8f70b4 |
}
|