apt-inst/dirstream.cc

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