test/libapt/fileutl_test.cc

   1 #include <config.h>
   2
   3 #include <apt-pkg/error.h>
   4 #include <apt-pkg/fileutl.h>
   5
   6 #include <string>
   7 #include <vector>
   8 #include <stdlib.h>
   9 #include <sys/stat.h>
  10
  11 #include "assert.h"
  12
  13 // regression test for permission bug LP: #1304657
  14 static bool
  15 TestFileFdOpenPermissions(mode_t a_umask, mode_t ExpectedFilePermission)
  16 {
  17    FileFd f;
  18    struct stat buf;
  19    static const char* fname = "test.txt";
  20
  21    umask(a_umask);
  22    f.Open(fname, FileFd::ReadWrite|FileFd::Atomic);
  23    f.Close();
  24    if (stat(fname, &buf) < 0)
  25    {
  26       _error->Errno("stat", "failed to stat");
  27       _error->DumpErrors();
  28       return false;
  29    }
  30    unlink(fname);
  31    equals(buf.st_mode & 0777, ExpectedFilePermission);
  32    return true;
  33 }
  34
  35 int main()
  36 {
  37    std::vector<std::string> files;
  38
  39    if (TestFileFdOpenPermissions(0002, 0664) == false ||
  40        TestFileFdOpenPermissions(0022, 0644) == false ||
  41        TestFileFdOpenPermissions(0077, 0600) == false ||
  42        TestFileFdOpenPermissions(0026, 0640) == false)
  43    {
  44       return 1;
  45    }
  46
  47    // normal match
  48    files = Glob("*.lst");
  49    if (files.size() != 1)
  50    {
  51       _error->DumpErrors();
  52       return 1;
  53    }
  54
  55    // not there
  56    files = Glob("xxxyyyzzz");
  57    if (files.size() != 0 || _error->PendingError())
  58    {
  59       _error->DumpErrors();
  60       return 1;
  61    }
  62
  63    // many matches (number is a bit random)
  64    files = Glob("*.cc");
  65    if (files.size() < 10)
  66    {
  67       _error->DumpErrors();
  68       return 1;
  69    }
  70
  71    // GetTempDir()
  72    unsetenv("TMPDIR");
  73    equals(GetTempDir(), "/tmp");
  74
  75    setenv("TMPDIR", "", 1);
  76    equals(GetTempDir(), "/tmp");
  77
  78    setenv("TMPDIR", "/not-there-no-really-not", 1);
  79    equals(GetTempDir(), "/tmp");
  80
  81    setenv("TMPDIR", "/usr", 1);
  82    equals(GetTempDir(), "/usr");
  83
  84    return 0;
  85 }