]> git.saurik.com Git - apt.git/blame - test/libapt/fileutl_test.cc
Use _apt as our unprivileged user name
[apt.git] / test / libapt / fileutl_test.cc
CommitLineData
453b82a3
DK
1#include <config.h>
2
488011fa
MV
3#include <apt-pkg/error.h>
4#include <apt-pkg/fileutl.h>
f00832cc 5#include <apt-pkg/strutl.h>
f7feb041 6#include <apt-pkg/aptconfiguration.h>
488011fa 7
488011fa
MV
8#include <string>
9#include <vector>
488011fa 10#include <stdlib.h>
f7feb041 11#include <string.h>
488011fa 12
f00832cc 13#include <gtest/gtest.h>
488011fa 14
f00832cc 15#include "file-helpers.h"
f7feb041 16
f00832cc
DK
17static void TestFileFd(mode_t const a_umask, mode_t const ExpectedFilePermission,
18 unsigned int const filemode, APT::Configuration::Compressor const &compressor)
f22b65b4 19{
f00832cc
DK
20 std::string trace;
21 strprintf(trace, "TestFileFd: Compressor: %s umask: %#o permission: %#o mode: %d", compressor.Name.c_str(), a_umask, ExpectedFilePermission, filemode);
22 SCOPED_TRACE(trace);
23
bb93178b
DK
24 static const char* fname = "apt-filefd-test.txt";
25 if (FileExists(fname) == true)
f00832cc 26 EXPECT_EQ(0, unlink(fname));
f22b65b4 27
f00832cc 28 FileFd f;
f22b65b4 29 umask(a_umask);
f00832cc
DK
30 EXPECT_TRUE(f.Open(fname, filemode, compressor));
31 EXPECT_TRUE(f.IsOpen());
32 EXPECT_FALSE(f.Failed());
33 EXPECT_EQ(umask(a_umask), a_umask);
f7feb041 34
bb93178b 35 std::string test = "This is a test!\n";
f00832cc
DK
36 EXPECT_TRUE(f.Write(test.c_str(), test.size()));
37 EXPECT_TRUE(f.IsOpen());
38 EXPECT_FALSE(f.Failed());
f7feb041 39
f22b65b4 40 f.Close();
f00832cc
DK
41 EXPECT_FALSE(f.IsOpen());
42 EXPECT_FALSE(f.Failed());
43
44 EXPECT_TRUE(f.Open(fname, FileFd::ReadOnly, compressor));
45 EXPECT_TRUE(f.IsOpen());
46 EXPECT_FALSE(f.Failed());
47 EXPECT_FALSE(f.Eof());
48 EXPECT_NE(0, f.FileSize());
49 EXPECT_FALSE(f.Failed());
50 EXPECT_NE(0, f.ModificationTime());
51 EXPECT_FALSE(f.Failed());
f7feb041 52
bb93178b 53 // ensure the memory is as predictably messed up
f00832cc 54#define APT_INIT_READBACK \
bb93178b
DK
55 char readback[20]; \
56 memset(readback, 'D', sizeof(readback)/sizeof(readback[0])); \
57 readback[19] = '\0';
f00832cc
DK
58#define EXPECT_N_STR(expect, actual) \
59 EXPECT_EQ(0, strncmp(expect, actual, strlen(expect)));
60
f7feb041 61 {
bb93178b 62 APT_INIT_READBACK
f7feb041 63 char const * const expect = "This";
f00832cc
DK
64 EXPECT_TRUE(f.Read(readback, strlen(expect)));
65 EXPECT_FALSE(f.Failed());
66 EXPECT_FALSE(f.Eof());
67 EXPECT_N_STR(expect, readback);
68 EXPECT_EQ(strlen(expect), f.Tell());
f7feb041
DK
69 }
70 {
bb93178b
DK
71 APT_INIT_READBACK
72 char const * const expect = "test!\n";
f00832cc
DK
73 EXPECT_TRUE(f.Skip((test.size() - f.Tell()) - strlen(expect)));
74 EXPECT_TRUE(f.Read(readback, strlen(expect)));
75 EXPECT_FALSE(f.Failed());
76 EXPECT_FALSE(f.Eof());
77 EXPECT_N_STR(expect, readback);
78 EXPECT_EQ(test.size(), f.Tell());
f7feb041 79 }
bb93178b
DK
80 {
81 APT_INIT_READBACK
f00832cc
DK
82 EXPECT_TRUE(f.Seek(0));
83 EXPECT_FALSE(f.Eof());
84 EXPECT_TRUE(f.Read(readback, 20, true));
85 EXPECT_FALSE(f.Failed());
86 EXPECT_TRUE(f.Eof());
87 EXPECT_N_STR(test.c_str(), readback);
88 EXPECT_EQ(f.Size(), f.Tell());
bb93178b
DK
89 }
90 {
91 APT_INIT_READBACK
f00832cc
DK
92 EXPECT_TRUE(f.Seek(0));
93 EXPECT_FALSE(f.Eof());
94 EXPECT_TRUE(f.Read(readback, test.size(), true));
95 EXPECT_FALSE(f.Failed());
96 EXPECT_FALSE(f.Eof());
97 EXPECT_N_STR(test.c_str(), readback);
98 EXPECT_EQ(f.Size(), f.Tell());
bb93178b
DK
99 }
100 {
101 APT_INIT_READBACK
f00832cc
DK
102 EXPECT_TRUE(f.Seek(0));
103 EXPECT_FALSE(f.Eof());
bb93178b 104 unsigned long long actual;
f00832cc
DK
105 EXPECT_TRUE(f.Read(readback, 20, &actual));
106 EXPECT_FALSE(f.Failed());
107 EXPECT_TRUE(f.Eof());
108 EXPECT_EQ(test.size(), actual);
109 EXPECT_N_STR(test.c_str(), readback);
110 EXPECT_EQ(f.Size(), f.Tell());
bb93178b
DK
111 }
112 {
113 APT_INIT_READBACK
f00832cc
DK
114 EXPECT_TRUE(f.Seek(0));
115 EXPECT_FALSE(f.Eof());
bb93178b 116 f.ReadLine(readback, 20);
f00832cc
DK
117 EXPECT_FALSE(f.Failed());
118 EXPECT_FALSE(f.Eof());
119 EXPECT_EQ(test, readback);
120 EXPECT_EQ(f.Size(), f.Tell());
bb93178b
DK
121 }
122 {
123 APT_INIT_READBACK
f00832cc
DK
124 EXPECT_TRUE(f.Seek(0));
125 EXPECT_FALSE(f.Eof());
bb93178b
DK
126 char const * const expect = "This";
127 f.ReadLine(readback, strlen(expect) + 1);
f00832cc
DK
128 EXPECT_FALSE(f.Failed());
129 EXPECT_FALSE(f.Eof());
130 EXPECT_N_STR(expect, readback);
131 EXPECT_EQ(strlen(expect), f.Tell());
bb93178b
DK
132 }
133#undef APT_INIT_READBACK
f7feb041
DK
134
135 f.Close();
f00832cc
DK
136 EXPECT_FALSE(f.IsOpen());
137 EXPECT_FALSE(f.Failed());
f7feb041
DK
138
139 // regression test for permission bug LP: #1304657
f00832cc
DK
140 struct stat buf;
141 EXPECT_EQ(0, stat(fname, &buf));
142 EXPECT_EQ(0, unlink(fname));
143 EXPECT_EQ(ExpectedFilePermission, buf.st_mode & 0777);
f22b65b4
MV
144}
145
f00832cc 146static void TestFileFd(unsigned int const filemode)
488011fa 147{
bb93178b
DK
148 std::vector<APT::Configuration::Compressor> compressors = APT::Configuration::getCompressors();
149
150 // testing the (un)compress via pipe, as the 'real' compressors are usually built in via libraries
151 compressors.push_back(APT::Configuration::Compressor("rev", ".reversed", "rev", NULL, NULL, 42));
152 //compressors.push_back(APT::Configuration::Compressor("cat", ".ident", "cat", NULL, NULL, 42));
153
f7feb041
DK
154 for (std::vector<APT::Configuration::Compressor>::const_iterator c = compressors.begin(); c != compressors.end(); ++c)
155 {
156 if ((filemode & FileFd::ReadWrite) == FileFd::ReadWrite &&
157 (c->Name.empty() != true && c->Binary.empty() != true))
158 continue;
f00832cc
DK
159 TestFileFd(0002, 0664, filemode, *c);
160 TestFileFd(0022, 0644, filemode, *c);
161 TestFileFd(0077, 0600, filemode, *c);
162 TestFileFd(0026, 0640, filemode, *c);
f7feb041 163 }
f7feb041 164}
488011fa 165
f00832cc
DK
166TEST(FileUtlTest, FileFD)
167{
168 std::string const startdir = SafeGetCWD();
169 EXPECT_FALSE(startdir.empty());
170 std::string tempdir;
171 createTemporaryDirectory("filefd", tempdir);
172 EXPECT_EQ(0, chdir(tempdir.c_str()));
173
174 TestFileFd(FileFd::WriteOnly | FileFd::Create);
175 TestFileFd(FileFd::WriteOnly | FileFd::Create | FileFd::Empty);
176 TestFileFd(FileFd::WriteOnly | FileFd::Create | FileFd::Exclusive);
177 TestFileFd(FileFd::WriteOnly | FileFd::Atomic);
178 TestFileFd(FileFd::WriteOnly | FileFd::Create | FileFd::Atomic);
179 // short-hands for ReadWrite with these modes
180 TestFileFd(FileFd::WriteEmpty);
181 TestFileFd(FileFd::WriteAny);
182 TestFileFd(FileFd::WriteTemp);
183 TestFileFd(FileFd::WriteAtomic);
184
185 EXPECT_EQ(0, chdir(startdir.c_str()));
186 removeDirectory(tempdir);
187}
188TEST(FileUtlTest, Glob)
f7feb041 189{
f7feb041 190 std::vector<std::string> files;
488011fa 191 // normal match
f00832cc
DK
192 files = Glob("*akefile");
193 EXPECT_EQ(1, files.size());
488011fa
MV
194
195 // not there
196 files = Glob("xxxyyyzzz");
f00832cc
DK
197 EXPECT_TRUE(files.empty());
198 EXPECT_FALSE(_error->PendingError());
488011fa
MV
199
200 // many matches (number is a bit random)
201 files = Glob("*.cc");
f00832cc
DK
202 EXPECT_LT(10, files.size());
203}
204TEST(FileUtlTest, GetTempDir)
205{
206 char const * const envtmp = getenv("TMPDIR");
207 std::string old_tmpdir;
208 if (envtmp != NULL)
209 old_tmpdir = envtmp;
488011fa 210
a077861a 211 unsetenv("TMPDIR");
f00832cc 212 EXPECT_EQ("/tmp", GetTempDir());
a077861a
MV
213
214 setenv("TMPDIR", "", 1);
f00832cc 215 EXPECT_EQ("/tmp", GetTempDir());
a077861a
MV
216
217 setenv("TMPDIR", "/not-there-no-really-not", 1);
f00832cc 218 EXPECT_EQ("/tmp", GetTempDir());
a077861a
MV
219
220 setenv("TMPDIR", "/usr", 1);
f00832cc 221 EXPECT_EQ("/usr", GetTempDir());
a077861a 222
f00832cc
DK
223 unsetenv("TMPDIR");
224 if (old_tmpdir.empty() == false)
225 setenv("TMPDIR", old_tmpdir.c_str(), 1);
488011fa 226}
7ad2a347
MV
227TEST(FileUtlTest, Popen)
228{
229 FileFd Fd;
230 pid_t Child;
231 char buf[1024];
232 std::string s;
233 unsigned long long n = 0;
234 std::vector<std::string> OpenFds;
235
236 // count Fds to ensure we don't have a resource leak
237 if(FileExists("/proc/self/fd"))
238 OpenFds = Glob("/proc/self/fd/*");
239
240 // output something
241 const char* Args[10] = {"/bin/echo", "meepmeep", NULL};
242 bool res = Popen(Args, Fd, Child, FileFd::ReadOnly);
243 Fd.Read(buf, sizeof(buf)-1, &n);
244 buf[n] = 0;
245 EXPECT_NE(n, 0);
246 EXPECT_EQ(res, true);
247 EXPECT_STREQ(buf, "meepmeep\n");
248
249 // wait for the child to exit and cleanup
250 ExecWait(Child, "PopenRead");
251 Fd.Close();
252
253 // ensure that after a close all is good again
254 if(FileExists("/proc/self/fd"))
255 EXPECT_EQ(Glob("/proc/self/fd/*").size(), OpenFds.size());
256
257
258 // ReadWrite is not supported
259 res = Popen(Args, Fd, Child, FileFd::ReadWrite);
260 EXPECT_EQ(res, false);
261 _error->Discard();
262
263 // write something
264 Args[0] = "/bin/bash";
265 Args[1] = "-c";
266 Args[2] = "read";
267 Args[3] = NULL;
268 res = Popen(Args, Fd, Child, FileFd::WriteOnly);
269 s = "\n";
270 Fd.Write(s.c_str(), s.size());
271 Fd.Close();
272 ExecWait(Child, "PopenWrite");
273}
53ac87ac
MV
274TEST(FileUtlTest, flAbsPath)
275{
a111a024 276 std::string cwd = SafeGetCWD();
53ac87ac
MV
277 int res = chdir("/bin/");
278 EXPECT_EQ(res, 0);
279 std::string p = flAbsPath("ls");
280 EXPECT_EQ(p, "/bin/ls");
a111a024
MV
281
282 res = chdir(cwd.c_str());
283 EXPECT_EQ(res, 0);
53ac87ac 284}