git.saurik.com Git - apt.git/blame_incremental

... / ...

Commit	Line	Data
	1	// -- mode: cpp; mode: fold --
	2	// Description /{{{/
	3	// $Id: file.cc,v 1.9.2.1 2004/01/16 18:58:50 mdz Exp $
	4	/* ######################################################################
	5
	6	File URI method for APT
	7
	8	This simply checks that the file specified exists, if so the relevant
	9	information is returned. If a .gz filename is specified then the file
	10	name with .gz removed will also be checked and information about it
	11	will be returned in Alt-*
	12
	13	##################################################################### */
	14	/}}}/
	15	// Include Files /{{{/
	16	#include <config.h>
	17
	18	#include <apt-pkg/acquire-method.h>
	19	#include <apt-pkg/aptconfiguration.h>
	20	#include <apt-pkg/error.h>
	21	#include <apt-pkg/hashes.h>
	22	#include <apt-pkg/fileutl.h>
	23	#include <apt-pkg/strutl.h>
	24
	25	#include <string>
	26	#include <sys/stat.h>
	27
	28	#include <apti18n.h>
	29	/}}}/
	30
	31	class FileMethod : public pkgAcqMethod
	32	{
	33	virtual bool Fetch(FetchItem *Itm) APT_OVERRIDE;
	34	virtual bool Configuration(std::string Message) APT_OVERRIDE;
	35
	36	public:
	37
	38	FileMethod() : pkgAcqMethod("1.0",SingleInstance \| SendConfig \| LocalOnly) {};
	39	};
	40	bool FileMethod::Configuration(std::string Message)
	41	{
	42	if (pkgAcqMethod::Configuration(Message) == false)
	43	return false;
	44
	45	DropPrivsOrDie();
	46
	47	return true;
	48	}
	49
	50	// FileMethod::Fetch - Fetch a file /{{{/
	51	// ---------------------------------------------------------------------
	52	/* */
	53	bool FileMethod::Fetch(FetchItem *Itm)
	54	{
	55	URI Get = Itm->Uri;
	56	std::string File = Get.Path;
	57	FetchResult Res;
	58	if (Get.Host.empty() == false)
	59	return _error->Error(_("Invalid URI, local URIS must not start with //"));
	60
	61	struct stat Buf;
	62	// deal with destination files which might linger around
	63	if (lstat(Itm->DestFile.c_str(), &Buf) == 0)
	64	{
	65	if ((Buf.st_mode & S_IFREG) != 0)
	66	{
	67	if (Itm->LastModified == Buf.st_mtime && Itm->LastModified != 0)
	68	{
	69	HashStringList const hsl = Itm->ExpectedHashes;
	70	if (Itm->ExpectedHashes.VerifyFile(File))
	71	{
	72	Res.Filename = Itm->DestFile;
	73	Res.IMSHit = true;
	74	}
	75	}
	76	}
	77	}
	78	if (Res.IMSHit != true)
	79	unlink(Itm->DestFile.c_str());
	80
	81	// See if the file exists
	82	if (stat(File.c_str(),&Buf) == 0)
	83	{
	84	Res.Size = Buf.st_size;
	85	Res.Filename = File;
	86	Res.LastModified = Buf.st_mtime;
	87	Res.IMSHit = false;
	88	if (Itm->LastModified == Buf.st_mtime && Itm->LastModified != 0)
	89	{
	90	unsigned long long const filesize = Itm->ExpectedHashes.FileSize();
	91	if (filesize != 0 && filesize == Res.Size)
	92	Res.IMSHit = true;
	93	}
	94
	95	Hashes Hash(Itm->ExpectedHashes);
	96	FileFd Fd(File, FileFd::ReadOnly);
	97	Hash.AddFD(Fd);
	98	Res.TakeHashes(Hash);
	99	}
	100	if (Res.IMSHit == false)
	101	URIStart(Res);
	102
	103	// See if the uncompressed file exists and reuse it
	104	FetchResult AltRes;
	105	AltRes.Filename.clear();
	106	std::vector<std::string> extensions = APT::Configuration::getCompressorExtensions();
	107	for (std::vector<std::string>::const_iterator ext = extensions.begin(); ext != extensions.end(); ++ext)
	108	{
	109	if (APT::String::Endswith(File, *ext) == true)
	110	{
	111	std::string const unfile = File.substr(0, File.length() - ext->length() - 1);
	112	if (stat(unfile.c_str(),&Buf) == 0)
	113	{
	114	AltRes.Size = Buf.st_size;
	115	AltRes.Filename = unfile;
	116	AltRes.LastModified = Buf.st_mtime;
	117	AltRes.IMSHit = false;
	118	if (Itm->LastModified == Buf.st_mtime && Itm->LastModified != 0)
	119	AltRes.IMSHit = true;
	120	break;
	121	}
	122	// no break here as we could have situations similar to '.gz' vs '.tar.gz' here
	123	}
	124	}
	125
	126	if (AltRes.Filename.empty() == false)
	127	URIDone(Res,&AltRes);
	128	else if (Res.Filename.empty() == false)
	129	URIDone(Res);
	130	else
	131	return _error->Error(_("File not found"));
	132
	133	return true;
	134	}
	135	/}}}/
	136
	137	int main()
	138	{
	139	setlocale(LC_ALL, "");
	140
	141	FileMethod Mth;
	142	return Mth.Run();
	143	}