Blame src/GetJob.cc

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
}