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 <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                                                                         /*}}}*/