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

... / ...

Commit	Line	Data
	1	// -- mode: cpp; mode: fold --
	2	// Description /{{{/
	3	// $Id: dirstream.cc,v 1.3.2.1 2004/01/16 18:58:50 mdz Exp $
	4	/* ######################################################################
	5
	6	Directory Stream
	7
	8	This class provides a simple basic extractor that can be used for
	9	a number of purposes.
	10
	11	##################################################################### */
	12	/}}}/
	13	// Include Files /{{{/
	14	#include <apt-pkg/dirstream.h>
	15	#include <apt-pkg/error.h>
	16
	17	#include <fcntl.h>
	18	#include <sys/stat.h>
	19	#include <sys/types.h>
	20	#include <errno.h>
	21	#include <utime.h>
	22	#include <unistd.h>
	23	#include <apti18n.h>
	24	/}}}/
	25
	26	// DirStream::DoItem - Process an item /{{{/
	27	// ---------------------------------------------------------------------
	28	/* This is a very simple extractor, it does not deal with things like
	29	overwriting directories with files and so on. */
	30	bool pkgDirStream::DoItem(Item &Itm,int &Fd)
	31	{
	32	switch (Itm.Type)
	33	{
	34	case Item::File:
	35	{
	36	/* Open the output file, NDELAY is used to prevent this from
	37	blowing up on device special files.. */
	38	int iFd = open(Itm.Name,O_NDELAY\|O_WRONLY\|O_CREAT\|O_TRUNC\|O_APPEND,
	39	Itm.Mode);
	40	if (iFd < 0)
	41	return _error->Errno("open",_("Failed to write file %s"),
	42	Itm.Name);
	43
	44	// fchmod deals with umask and fchown sets the ownership
	45	if (fchmod(iFd,Itm.Mode) != 0)
	46	return _error->Errno("fchmod",_("Failed to write file %s"),
	47	Itm.Name);
	48	if (fchown(iFd,Itm.UID,Itm.GID) != 0 && errno != EPERM)
	49	return _error->Errno("fchown",_("Failed to write file %s"),
	50	Itm.Name);
	51	Fd = iFd;
	52	return true;
	53	}
	54
	55	case Item::HardLink:
	56	case Item::SymbolicLink:
	57	case Item::CharDevice:
	58	case Item::BlockDevice:
	59	case Item::Directory:
	60	{
	61	struct stat Buf;
	62	// check if the dir is already there, if so return true
	63	if (stat(Itm.Name,&Buf) == 0)
	64	{
	65	if(S_ISDIR(Buf.st_mode))
	66	return true;
	67	// something else is there already, return false
	68	return false;
	69	}
	70	// nothing here, create the dir
	71	if(mkdir(Itm.Name,Itm.Mode) < 0)
	72	return false;
	73	return true;
	74	break;
	75	}
	76	case Item::FIFO:
	77	break;
	78	}
	79
	80	return true;
	81	}
	82	/}}}/
	83	// DirStream::FinishedFile - Finished processing a file /{{{/
	84	// ---------------------------------------------------------------------
	85	/* */
	86	bool pkgDirStream::FinishedFile(Item &Itm,int Fd)
	87	{
	88	if (Fd < 0)
	89	return true;
	90
	91	if (close(Fd) != 0)
	92	return _error->Errno("close",_("Failed to close file %s"),Itm.Name);
	93
	94	/* Set the modification times. The only way it can fail is if someone
	95	has futzed with our file, which is intolerable :> */
	96	struct utimbuf Time;
	97	Time.actime = Itm.MTime;
	98	Time.modtime = Itm.MTime;
	99	if (utime(Itm.Name,&Time) != 0)
	100	_error->Errno("utime",_("Failed to close file %s"),Itm.Name);
	101
	102	return true;
	103	}
	104	/}}}/
	105	// DirStream::Fail - Failed processing a file /{{{/
	106	// ---------------------------------------------------------------------
	107	/* */
	108	bool pkgDirStream::Fail(Item &Itm,int Fd)
	109	{
	110	if (Fd < 0)
	111	return true;
	112
	113	close(Fd);
	114	return false;
	115	}
	116	/}}}/