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 relevent
	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 <apt-pkg/acquire-method.h>
	17	#include <apt-pkg/error.h>
	18	#include <apt-pkg/hashes.h>
	19	#include <apt-pkg/fileutl.h>
	20
	21	#include <sys/stat.h>
	22	#include <unistd.h>
	23	#include <apti18n.h>
	24	/}}}/
	25
	26	class FileMethod : public pkgAcqMethod
	27	{
	28	virtual bool Fetch(FetchItem *Itm);
	29
	30	public:
	31
	32	FileMethod() : pkgAcqMethod("1.0",SingleInstance \| LocalOnly) {};
	33	};
	34
	35	// FileMethod::Fetch - Fetch a file /{{{/
	36	// ---------------------------------------------------------------------
	37	/* */
	38	bool FileMethod::Fetch(FetchItem *Itm)
	39	{
	40	URI Get = Itm->Uri;
	41	string File = Get.Path;
	42	FetchResult Res;
	43	if (Get.Host.empty() == false)
	44	return _error->Error(_("Invalid URI, local URIS must not start with //"));
	45
	46	// See if the file exists
	47	struct stat Buf;
	48	if (stat(File.c_str(),&Buf) == 0)
	49	{
	50	Res.Size = Buf.st_size;
	51	Res.Filename = File;
	52	Res.LastModified = Buf.st_mtime;
	53	Res.IMSHit = false;
	54	if (Itm->LastModified == Buf.st_mtime && Itm->LastModified != 0)
	55	Res.IMSHit = true;
	56	}
	57
	58	// See if we can compute a file without a .gz exentsion
	59	string::size_type Pos = File.rfind(".gz");
	60	if (Pos + 3 == File.length())
	61	{
	62	File = string(File,0,Pos);
	63	if (stat(File.c_str(),&Buf) == 0)
	64	{
	65	FetchResult AltRes;
	66	AltRes.Size = Buf.st_size;
	67	AltRes.Filename = File;
	68	AltRes.LastModified = Buf.st_mtime;
	69	AltRes.IMSHit = false;
	70	if (Itm->LastModified == Buf.st_mtime && Itm->LastModified != 0)
	71	AltRes.IMSHit = true;
	72
	73	URIDone(Res,&AltRes);
	74	return true;
	75	}
	76	}
	77
	78	if (Res.Filename.empty() == true)
	79	return _error->Error(_("File not found"));
	80
	81	Hashes Hash;
	82	FileFd Fd(Res.Filename, FileFd::ReadOnly);
	83	Hash.AddFD(Fd.Fd(), Fd.Size());
	84	Res.TakeHashes(Hash);
	85	URIDone(Res);
	86	return true;
	87	}
	88	/}}}/
	89
	90	int main()
	91	{
	92	setlocale(LC_ALL, "");
	93
	94	FileMethod Mth;
	95	return Mth.Run();
	96	}