memory/shm_file.cc - GCC Code Coverage Report

memory/shm_file.cc

Line	Branch	Exec	Source
1			#include "shm_file.h"
2
3			#include <algorithm>
4			#include <iostream>
5			#ifdef __linux__
6			# include <sys/stat.h>
7			#endif
8
9			namespace base {
10			namespace {
11
12			constexpr const char* kFsDaxType = "FS_DAX";
13			constexpr const char* kAnonyDramType = "ANONY_DRAM";
14			constexpr const char* kDevDaxType = "DEV_DAX";
15
16		1242	json BuildShmFileConfig(
17			const std::string& type, const std::string& filename, int64 size) {
18	15/30 ✓ Branch 3 taken 1242 times. ✗ Branch 4 not taken. ✓ Branch 6 taken 1242 times. ✗ Branch 7 not taken. ✓ Branch 9 taken 1242 times. ✗ Branch 10 not taken. ✓ Branch 12 taken 1242 times. ✗ Branch 13 not taken. ✓ Branch 15 taken 1242 times. ✗ Branch 16 not taken. ✓ Branch 19 taken 1242 times. ✗ Branch 20 not taken. ✓ Branch 22 taken 1242 times. ✗ Branch 23 not taken. ✓ Branch 24 taken 3726 times. ✓ Branch 25 taken 1242 times. ✓ Branch 27 taken 2484 times. ✓ Branch 28 taken 1242 times. ✓ Branch 30 taken 2484 times. ✓ Branch 31 taken 1242 times. ✓ Branch 33 taken 2484 times. ✓ Branch 34 taken 1242 times. ✗ Branch 36 not taken. ✗ Branch 37 not taken. ✗ Branch 39 not taken. ✗ Branch 40 not taken. ✗ Branch 42 not taken. ✗ Branch 43 not taken. ✗ Branch 45 not taken. ✗ Branch 46 not taken.	12420	return json{{"type", type}, {"filename", filename}, {"size", size}};
19			}
20
21			} // namespace
22
23		1248	std::unique_ptr<ShmFile> ShmFile::New(const json& config) {
24	3/16 ✓ Branch 1 taken 1248 times. ✗ Branch 2 not taken. ✗ Branch 3 not taken. ✓ Branch 4 taken 1248 times. ✗ Branch 6 not taken. ✗ Branch 7 not taken. ✗ Branch 9 not taken. ✗ Branch 10 not taken. ✗ Branch 12 not taken. ✗ Branch 13 not taken. ✗ Branch 15 not taken. ✗ Branch 16 not taken. ✗ Branch 19 not taken. ✓ Branch 20 taken 1248 times. ✗ Branch 22 not taken. ✗ Branch 23 not taken.	1248	CHECK(config.contains("type")) << "ShmFile config requires type";
25	2/4 ✓ Branch 1 taken 1248 times. ✗ Branch 2 not taken. ✓ Branch 4 taken 1248 times. ✗ Branch 5 not taken.	1248	const std::string type = config.at("type").get<std::string>();
26			using SF = base::Factory<ShmFile, const json&>;
27	1/2 ✓ Branch 1 taken 1248 times. ✗ Branch 2 not taken.	1248	std::unique_ptr<ShmFile> shm_file(SF::NewInstance(type, config));
28	3/4 ✓ Branch 2 taken 1248 times. ✗ Branch 3 not taken. ✓ Branch 4 taken 2 times. ✓ Branch 5 taken 1246 times.	1248	if (!shm_file->Initialize(config)) {
29		2	return nullptr;
30			}
31		1246	return shm_file;
32		1248	}
33
34		1242	json ShmFile::ConfigForMedium(
35			const std::string& medium, const std::string& filename, int64 size) {
36	1/2 ✓ Branch 1 taken 1242 times. ✗ Branch 2 not taken.	1242	if (medium == "DRAM")
37	2/4 ✓ Branch 2 taken 1242 times. ✗ Branch 3 not taken. ✓ Branch 5 taken 1242 times. ✗ Branch 6 not taken.	1242	return BuildShmFileConfig(kAnonyDramType, filename, size);
38			// return BuildShmFileConfig(kDevDaxType, filename, size);
39	0/2 ✗ Branch 1 not taken. ✗ Branch 2 not taken.	✗	if (medium == "SSD")
40	0/4 ✗ Branch 2 not taken. ✗ Branch 3 not taken. ✗ Branch 5 not taken. ✗ Branch 6 not taken.	✗	return BuildShmFileConfig(kFsDaxType, filename, size);
41	0/6 ✗ Branch 2 not taken. ✗ Branch 3 not taken. ✗ Branch 5 not taken. ✗ Branch 6 not taken. ✗ Branch 8 not taken. ✗ Branch 9 not taken.	✗	LOG(FATAL) << "Unsupported ShmFile medium: " << medium;
42			return json();
43			}
44
45		1248	std::string ShmFile::ConfigFilename(const json& config) {
46	2/14 ✗ Branch 1 not taken. ✓ Branch 2 taken 1248 times. ✗ Branch 4 not taken. ✗ Branch 5 not taken. ✗ Branch 7 not taken. ✗ Branch 8 not taken. ✗ Branch 10 not taken. ✗ Branch 11 not taken. ✗ Branch 13 not taken. ✗ Branch 14 not taken. ✗ Branch 17 not taken. ✓ Branch 18 taken 1248 times. ✗ Branch 20 not taken. ✗ Branch 21 not taken.	1248	CHECK(config.contains("filename")) << "ShmFile config requires filename";
47		1248	return config.at("filename").get<std::string>();
48			}
49
50		1248	int64 ShmFile::ConfigSize(const json& config) {
51	2/14 ✗ Branch 1 not taken. ✓ Branch 2 taken 1248 times. ✗ Branch 4 not taken. ✗ Branch 5 not taken. ✗ Branch 7 not taken. ✗ Branch 8 not taken. ✗ Branch 10 not taken. ✗ Branch 11 not taken. ✗ Branch 13 not taken. ✗ Branch 14 not taken. ✗ Branch 17 not taken. ✓ Branch 18 taken 1248 times. ✗ Branch 20 not taken. ✗ Branch 21 not taken.	1248	CHECK(config.contains("size")) << "ShmFile config requires size";
52		1248	return config.at("size").get<int64>();
53			}
54
55		6	bool FsDaxShmFile::Initialize(const json& config) {
56	1/2 ✓ Branch 1 taken 6 times. ✗ Branch 2 not taken.	6	Clear();
57
58	1/2 ✓ Branch 1 taken 6 times. ✗ Branch 2 not taken.	6	const std::string filename = ConfigFilename(config);
59	1/2 ✓ Branch 1 taken 6 times. ✗ Branch 2 not taken.	6	const int64 size = ConfigSize(config);
60
61		6	std::error_code ec;
62	1/2 ✓ Branch 1 taken 6 times. ✗ Branch 2 not taken.	6	bool file_exists = fs::exists(filename, ec);
63	1/2 ✗ Branch 1 not taken. ✓ Branch 2 taken 6 times.	6	if (ec) {
64	0/14 ✗ Branch 1 not taken. ✗ Branch 2 not taken. ✗ Branch 4 not taken. ✗ Branch 5 not taken. ✗ Branch 7 not taken. ✗ Branch 8 not taken. ✗ Branch 10 not taken. ✗ Branch 11 not taken. ✗ Branch 13 not taken. ✗ Branch 14 not taken. ✗ Branch 16 not taken. ✗ Branch 17 not taken. ✗ Branch 19 not taken. ✗ Branch 20 not taken.	✗	LOG(ERROR) << "fs::exists failed for " << filename << ": " << ec.message();
65		✗	return false;
66			}
67
68	2/2 ✓ Branch 0 taken 4 times. ✓ Branch 1 taken 2 times.	6	if (!file_exists) {
69	3/6 ✓ Branch 1 taken 4 times. ✗ Branch 2 not taken. ✓ Branch 4 taken 4 times. ✗ Branch 5 not taken. ✓ Branch 7 taken 4 times. ✗ Branch 8 not taken.	4	fs::create_directories(fs::path(filename).parent_path(), ec);
70	1/2 ✗ Branch 1 not taken. ✓ Branch 2 taken 4 times.	4	if (ec) {
71	0/10 ✗ Branch 1 not taken. ✗ Branch 2 not taken. ✗ Branch 4 not taken. ✗ Branch 5 not taken. ✗ Branch 7 not taken. ✗ Branch 8 not taken. ✗ Branch 10 not taken. ✗ Branch 11 not taken. ✗ Branch 13 not taken. ✗ Branch 14 not taken.	✗	LOG(ERROR) << "fs::create_directories failed: " << ec.message();
72		✗	return false;
73			}
74	6/12 ✓ Branch 1 taken 4 times. ✗ Branch 2 not taken. ✓ Branch 4 taken 4 times. ✗ Branch 5 not taken. ✓ Branch 7 taken 4 times. ✗ Branch 8 not taken. ✓ Branch 10 taken 4 times. ✗ Branch 11 not taken. ✓ Branch 13 taken 4 times. ✗ Branch 14 not taken. ✓ Branch 16 taken 4 times. ✗ Branch 17 not taken.	4	LOG(INFO) << "Create ShmFile: " << filename << ", size: " << size;
75			}
76
77	1/2 ✓ Branch 2 taken 6 times. ✗ Branch 3 not taken.	6	fd_ = open(filename.c_str(), O_RDWR \| O_CREAT, 0666);
78	1/2 ✗ Branch 0 not taken. ✓ Branch 1 taken 6 times.	6	if (fd_ < 0) {
79	0/12 ✗ Branch 1 not taken. ✗ Branch 2 not taken. ✗ Branch 4 not taken. ✗ Branch 5 not taken. ✗ Branch 7 not taken. ✗ Branch 8 not taken. ✗ Branch 10 not taken. ✗ Branch 11 not taken. ✗ Branch 13 not taken. ✗ Branch 14 not taken. ✗ Branch 17 not taken. ✗ Branch 18 not taken.	✗	LOG(ERROR) << "Failed to open file " << filename << ": " << strerror(errno);
80		✗	return false;
81			}
82
83	2/2 ✓ Branch 0 taken 4 times. ✓ Branch 1 taken 2 times.	6	if (!file_exists) {
84	1/2 ✓ Branch 1 taken 4 times. ✗ Branch 2 not taken.	4	int ret = posix_fallocate(fd_, 0, size);
85	1/2 ✗ Branch 0 not taken. ✓ Branch 1 taken 4 times.	4	if (ret != 0) {
86	0/8 ✗ Branch 1 not taken. ✗ Branch 2 not taken. ✗ Branch 4 not taken. ✗ Branch 5 not taken. ✗ Branch 7 not taken. ✗ Branch 8 not taken. ✗ Branch 11 not taken. ✗ Branch 12 not taken.	✗	LOG(ERROR) << "posix_fallocate failed: " << strerror(ret);
87	0/2 ✗ Branch 1 not taken. ✗ Branch 2 not taken.	✗	close(fd_);
88		✗	fd_ = -1;
89		✗	return false;
90			}
91			}
92
93			struct stat st;
94	1/2 ✗ Branch 1 not taken. ✓ Branch 2 taken 6 times.	6	if (fstat(fd_, &st) != 0) {
95	0/8 ✗ Branch 1 not taken. ✗ Branch 2 not taken. ✗ Branch 4 not taken. ✗ Branch 5 not taken. ✗ Branch 7 not taken. ✗ Branch 8 not taken. ✗ Branch 11 not taken. ✗ Branch 12 not taken.	✗	LOG(ERROR) << "fstat failed: " << strerror(errno);
96	0/2 ✗ Branch 1 not taken. ✗ Branch 2 not taken.	✗	close(fd_);
97		✗	fd_ = -1;
98		✗	return false;
99			}
100		6	size_ = st.st_size;
101
102	2/2 ✓ Branch 0 taken 2 times. ✓ Branch 1 taken 4 times.	6	if (size_ != size) {
103	6/12 ✓ Branch 1 taken 2 times. ✗ Branch 2 not taken. ✓ Branch 4 taken 2 times. ✗ Branch 5 not taken. ✓ Branch 7 taken 2 times. ✗ Branch 8 not taken. ✓ Branch 10 taken 2 times. ✗ Branch 11 not taken. ✓ Branch 13 taken 2 times. ✗ Branch 14 not taken. ✓ Branch 16 taken 2 times. ✗ Branch 17 not taken.	2	LOG(ERROR) << "Size Error: " << size_ << " vs " << size;
104	1/2 ✓ Branch 1 taken 2 times. ✗ Branch 2 not taken.	2	close(fd_);
105		2	fd_ = -1;
106		2	size_ = 0;
107		2	return false;
108			}
109
110		4	data_ = reinterpret_cast<char*>(
111		4	mmap(nullptr, size_, PROT_READ \| PROT_WRITE, MAP_SHARED, fd_, 0));
112	2/10 ✓ Branch 3 taken 4 times. ✗ Branch 4 not taken. ✗ Branch 5 not taken. ✓ Branch 6 taken 4 times. ✗ Branch 9 not taken. ✗ Branch 10 not taken. ✗ Branch 12 not taken. ✗ Branch 13 not taken. ✗ Branch 15 not taken. ✗ Branch 16 not taken.	4	CHECK_NE(data_, MAP_FAILED) << "map failed";
113
114	1/2 ✓ Branch 1 taken 4 times. ✗ Branch 2 not taken.	4	filename_ = filename;
115	6/12 ✓ Branch 1 taken 4 times. ✗ Branch 2 not taken. ✓ Branch 4 taken 4 times. ✗ Branch 5 not taken. ✓ Branch 7 taken 4 times. ✗ Branch 8 not taken. ✓ Branch 10 taken 4 times. ✗ Branch 11 not taken. ✓ Branch 13 taken 4 times. ✗ Branch 14 not taken. ✓ Branch 16 taken 4 times. ✗ Branch 17 not taken.	8	LOG(INFO) << "mmap shm file: " << filename << ", size: " << size_
116	2/4 ✓ Branch 1 taken 4 times. ✗ Branch 2 not taken. ✓ Branch 4 taken 4 times. ✗ Branch 5 not taken.	4	<< ", fd: " << fd_;
117		4	return true;
118		6	}
119
120		12	void FsDaxShmFile::Clear() {
121	2/2 ✓ Branch 0 taken 4 times. ✓ Branch 1 taken 8 times.	12	if (fd_ >= 0) {
122	5/10 ✓ Branch 2 taken 4 times. ✗ Branch 3 not taken. ✓ Branch 5 taken 4 times. ✗ Branch 6 not taken. ✓ Branch 8 taken 4 times. ✗ Branch 9 not taken. ✓ Branch 11 taken 4 times. ✗ Branch 12 not taken. ✓ Branch 14 taken 4 times. ✗ Branch 15 not taken.	8	LOG(INFO) << "ummap shm file: " << filename_ << ", size: " << size_
123	2/4 ✓ Branch 1 taken 4 times. ✗ Branch 2 not taken. ✓ Branch 4 taken 4 times. ✗ Branch 5 not taken.	4	<< ", fd: " << fd_;
124		4	munmap(data_, size_);
125		4	close(fd_);
126			}
127		12	filename_.clear();
128		12	data_ = NULL;
129		12	size_ = 0;
130		12	fd_ = -1;
131		12	}
132
133		1242	bool AnonyDramShmFile::Initialize(const json& config) {
134	1/2 ✓ Branch 1 taken 1242 times. ✗ Branch 2 not taken.	1242	Clear();
135
136	1/2 ✓ Branch 1 taken 1242 times. ✗ Branch 2 not taken.	1242	const std::string filename = ConfigFilename(config);
137	1/2 ✓ Branch 1 taken 1242 times. ✗ Branch 2 not taken.	1242	const int64 size = ConfigSize(config);
138
139		1242	data_ = reinterpret_cast<char*>(mmap(
140			nullptr,
141			size,
142			PROT_READ \| PROT_WRITE,
143			MAP_PRIVATE \| MAP_ANONYMOUS,
144			-1,
145			0));
146	2/10 ✓ Branch 3 taken 1242 times. ✗ Branch 4 not taken. ✗ Branch 5 not taken. ✓ Branch 6 taken 1242 times. ✗ Branch 9 not taken. ✗ Branch 10 not taken. ✗ Branch 12 not taken. ✗ Branch 13 not taken. ✗ Branch 15 not taken. ✗ Branch 16 not taken.	1242	CHECK_NE(data_, MAP_FAILED) << "anonymous DRAM mmap failed";
147	1/2 ✓ Branch 1 taken 1242 times. ✗ Branch 2 not taken.	1242	filename_ = filename;
148		1242	size_ = size;
149	2/4 ✓ Branch 1 taken 1242 times. ✗ Branch 2 not taken. ✓ Branch 4 taken 1242 times. ✗ Branch 5 not taken.	2484	LOG(INFO) << "ShmFile, anonymous DRAM mmap, filename:" << filename
150	4/8 ✓ Branch 1 taken 1242 times. ✗ Branch 2 not taken. ✓ Branch 4 taken 1242 times. ✗ Branch 5 not taken. ✓ Branch 7 taken 1242 times. ✗ Branch 8 not taken. ✓ Branch 10 taken 1242 times. ✗ Branch 11 not taken.	1242	<< ", size:" << size;
151		1242	return true;
152		1242	}
153
154		2484	void AnonyDramShmFile::Clear() {
155	2/2 ✓ Branch 0 taken 1242 times. ✓ Branch 1 taken 1242 times.	2484	if (data_ != NULL) {
156	5/10 ✓ Branch 2 taken 1242 times. ✗ Branch 3 not taken. ✓ Branch 5 taken 1242 times. ✗ Branch 6 not taken. ✓ Branch 8 taken 1242 times. ✗ Branch 9 not taken. ✓ Branch 11 taken 1242 times. ✗ Branch 12 not taken. ✓ Branch 14 taken 1242 times. ✗ Branch 15 not taken.	1242	LOG(INFO) << "ummap anonymous DRAM: " << filename_ << ", size: " << size_;
157		1242	munmap(data_, size_);
158			}
159		2484	filename_.clear();
160		2484	data_ = NULL;
161		2484	size_ = 0;
162		2484	}
163
164		✗	bool DevDaxShmFile::Initialize(const json& config) {
165	0/2 ✗ Branch 1 not taken. ✗ Branch 2 not taken.	✗	Clear();
166
167	0/2 ✗ Branch 1 not taken. ✗ Branch 2 not taken.	✗	const std::string filename = ConfigFilename(config);
168	0/2 ✗ Branch 1 not taken. ✗ Branch 2 not taken.	✗	const int64 size = ConfigSize(config);
169
170			{
171			static std::mutex m;
172	0/2 ✗ Branch 1 not taken. ✗ Branch 2 not taken.	✗	std::lock_guard<std::mutex> _(m);
173		✗	data_ =
174	0/4 ✗ Branch 1 not taken. ✗ Branch 2 not taken. ✗ Branch 4 not taken. ✗ Branch 5 not taken.	✗	(char*)PMMmapRegisterCenter::GetInstance()->Register(filename, size);
175	0/2 ✗ Branch 1 not taken. ✗ Branch 2 not taken.	✗	filename_ = filename;
176		✗	}
177
178		✗	std::error_code ec;
179	0/4 ✗ Branch 1 not taken. ✗ Branch 2 not taken. ✗ Branch 5 not taken. ✗ Branch 6 not taken.	✗	if (!fs::exists(filename, ec)) {
180	0/6 ✗ Branch 1 not taken. ✗ Branch 2 not taken. ✗ Branch 4 not taken. ✗ Branch 5 not taken. ✗ Branch 7 not taken. ✗ Branch 8 not taken.	✗	fs::create_directories(fs::path(filename).parent_path(), ec);
181	0/12 ✗ Branch 1 not taken. ✗ Branch 2 not taken. ✗ Branch 4 not taken. ✗ Branch 5 not taken. ✗ Branch 7 not taken. ✗ Branch 8 not taken. ✗ Branch 10 not taken. ✗ Branch 11 not taken. ✗ Branch 13 not taken. ✗ Branch 14 not taken. ✗ Branch 16 not taken. ✗ Branch 17 not taken.	✗	LOG(INFO) << "Create ShmFile: " << filename << ", size: " << size;
182	0/2 ✗ Branch 1 not taken. ✗ Branch 2 not taken.	✗	std::ofstream output(filename);
183	0/2 ✗ Branch 1 not taken. ✗ Branch 2 not taken.	✗	output.write("a", 1);
184	0/2 ✗ Branch 1 not taken. ✗ Branch 2 not taken.	✗	output.close();
185		✗	}
186		✗	size_ = size;
187		✗	return true;
188		✗	}
189
190		✗	void DevDaxShmFile::Clear() {
191		✗	filename_.clear();
192		✗	data_ = NULL;
193		✗	size_ = 0;
194		✗	}
195
196			FACTORY_REGISTER(ShmFile, FS_DAX, FsDaxShmFile, const json&);
197			FACTORY_REGISTER(ShmFile, ANONY_DRAM, AnonyDramShmFile, const json&);
198			FACTORY_REGISTER(ShmFile, DEV_DAX, DevDaxShmFile, const json&);
199
200			} // namespace base
201