blob: cb550c17c0e6b7e7f118f605ec6b21011fc12c9c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
#include "TestUtil.h"
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <cerrno>
#include <cstring>
#include <sstream>
#include <fstream>
#include "a2io.h"
#include "File.h"
#include "StringFormat.h"
#include "FatalException.h"
namespace aria2 {
void createFile(const std::string& path, size_t length)
{
File(File(path).getDirname()).mkdirs();
int fd = creat(path.c_str(), OPEN_MODE);
if(fd == -1) {
throw FATAL_EXCEPTION(StringFormat("Could not create file=%s. cause:%s",
path.c_str(), strerror(errno)).str());
}
if(-1 == ftruncate(fd, length)) {
throw FATAL_EXCEPTION(StringFormat("ftruncate failed. cause:%s",
strerror(errno)).str());
}
close(fd);
}
std::string readFile(const std::string& path)
{
std::stringstream ss;
std::ifstream in(path.c_str(), std::ios::binary);
char buf[4096];
while(1) {
in.read(buf, sizeof(buf));
ss.write(buf, in.gcount());
if(in.gcount() != sizeof(buf)) {
break;
}
}
return ss.str();
}
};
|