mirror of
https://github.com/ukui/apt.git
synced 2026-03-09 09:35:45 -07:00
Every method opts in to getting the encoded URI passed along while keeping compat in case we are operated by an older acquire system. Effectively this is just a change for the http-based methods as the others just decode the URI as they work with files directly.
95 lines
2.3 KiB
C++
95 lines
2.3 KiB
C++
// -*- mode: cpp; mode: fold -*-
|
|
// Description /*{{{*/
|
|
/* ######################################################################
|
|
|
|
Copy URI - This method takes a uri like a file: uri and copies it
|
|
to the destination file.
|
|
|
|
##################################################################### */
|
|
/*}}}*/
|
|
// Include Files /*{{{*/
|
|
#include <config.h>
|
|
|
|
#include "aptmethod.h"
|
|
#include <apt-pkg/configuration.h>
|
|
#include <apt-pkg/error.h>
|
|
#include <apt-pkg/fileutl.h>
|
|
#include <apt-pkg/hashes.h>
|
|
#include <apt-pkg/strutl.h>
|
|
|
|
#include <string>
|
|
#include <sys/stat.h>
|
|
#include <sys/time.h>
|
|
|
|
#include <apti18n.h>
|
|
/*}}}*/
|
|
|
|
class CopyMethod : public aptMethod
|
|
{
|
|
virtual bool Fetch(FetchItem *Itm) APT_OVERRIDE;
|
|
|
|
public:
|
|
CopyMethod() : aptMethod("copy", "1.0", SingleInstance | SendConfig | SendURIEncoded)
|
|
{
|
|
SeccompFlags = aptMethod::BASE;
|
|
}
|
|
};
|
|
|
|
// CopyMethod::Fetch - Fetch a file /*{{{*/
|
|
// ---------------------------------------------------------------------
|
|
/* */
|
|
bool CopyMethod::Fetch(FetchItem *Itm)
|
|
{
|
|
// this ensures that relative paths work in copy
|
|
std::string const File = DecodeSendURI(Itm->Uri.substr(Itm->Uri.find(':')+1));
|
|
|
|
// Stat the file and send a start message
|
|
struct stat Buf;
|
|
if (stat(File.c_str(),&Buf) != 0)
|
|
return _error->Errno("stat",_("Failed to stat"));
|
|
|
|
// Forumulate a result and send a start message
|
|
FetchResult Res;
|
|
Res.Size = Buf.st_size;
|
|
Res.Filename = Itm->DestFile;
|
|
Res.LastModified = Buf.st_mtime;
|
|
Res.IMSHit = false;
|
|
URIStart(Res);
|
|
|
|
// just calc the hashes if the source and destination are identical
|
|
if (File == Itm->DestFile || Itm->DestFile == "/dev/null")
|
|
{
|
|
CalculateHashes(Itm, Res);
|
|
URIDone(Res);
|
|
return true;
|
|
}
|
|
|
|
// See if the file exists
|
|
FileFd From(File,FileFd::ReadOnly);
|
|
FileFd To(Itm->DestFile,FileFd::WriteAtomic);
|
|
To.EraseOnFailure();
|
|
|
|
// Copy the file
|
|
if (CopyFile(From,To) == false)
|
|
{
|
|
To.OpFail();
|
|
return false;
|
|
}
|
|
|
|
From.Close();
|
|
To.Close();
|
|
|
|
if (TransferModificationTimes(File.c_str(), Res.Filename.c_str(), Res.LastModified) == false)
|
|
return false;
|
|
|
|
CalculateHashes(Itm, Res);
|
|
URIDone(Res);
|
|
return true;
|
|
}
|
|
/*}}}*/
|
|
|
|
int main()
|
|
{
|
|
return CopyMethod().Run();
|
|
}
|