15 #include "data.h"
30 void perf_data__close_dir(struct perf_data *data)  in perf_data__close_dir()  argument
32 	close_dir(data->dir.files, data->dir.nr);  in perf_data__close_dir()
35 int perf_data__create_dir(struct perf_data *data, int nr)  in perf_data__create_dir()  argument
40 	if (WARN_ON(!data->is_dir))  in perf_data__create_dir()
47 	data->dir.version = PERF_DIR_VERSION;  in perf_data__create_dir()
48 	data->dir.files   = files;  in perf_data__create_dir()
49 	data->dir.nr      = nr;  in perf_data__create_dir()
54 		ret = asprintf(&file->path, "%s/data.%d", data->path, i);  in perf_data__create_dir()
72 int perf_data__open_dir(struct perf_data *data)  in perf_data__open_dir()  argument
81 	 * Directory containing a single regular perf data file which is already  in perf_data__open_dir()
84 	if (perf_data__is_single_file(data))  in perf_data__open_dir()
87 	if (WARN_ON(!data->is_dir))  in perf_data__open_dir()
91 	if (WARN_ON(data->dir.version != PERF_DIR_VERSION))  in perf_data__open_dir()
94 	dir = opendir(data->path);  in perf_data__open_dir()
103 		snprintf(path, sizeof(path), "%s/%s", data->path, dent->d_name);  in perf_data__open_dir()
107 		if (!S_ISREG(st.st_mode) || strncmp(dent->d_name, "data.", 5))  in perf_data__open_dir()
134 	data->dir.files = files;  in perf_data__open_dir()
135 	data->dir.nr    = nr;  in perf_data__open_dir()
143 int perf_data__update_dir(struct perf_data *data)  in perf_data__update_dir()  argument
147 	if (WARN_ON(!data->is_dir))  in perf_data__update_dir()
150 	for (i = 0; i < data->dir.nr; i++) {  in perf_data__update_dir()
151 		struct perf_data_file *file = &data->dir.files[i];  in perf_data__update_dir()
163 static bool check_pipe(struct perf_data *data)  in check_pipe()  argument
167 	int fd = perf_data__is_read(data) ?  in check_pipe()
170 	if (!data->path) {  in check_pipe()
174 		if (!strcmp(data->path, "-"))  in check_pipe()
179 		if (data->use_stdio) {  in check_pipe()
182 			mode = perf_data__is_read(data) ? "r" : "w";  in check_pipe()
183 			data->file.fptr = fdopen(fd, mode);  in check_pipe()
185 			if (data->file.fptr == NULL) {  in check_pipe()
186 				data->file.fd = fd;  in check_pipe()
187 				data->use_stdio = false;  in check_pipe()
190 			data->file.fd = fd;  in check_pipe()
194 	return data->is_pipe = is_pipe;  in check_pipe()
197 static int check_backup(struct perf_data *data)  in check_backup()  argument
201 	if (perf_data__is_read(data))  in check_backup()
204 	if (!stat(data->path, &st) && st.st_size) {  in check_backup()
209 			 data->path);  in check_backup()
213 			pr_err("Can't remove old data: %s (%s)\n",  in check_backup()
220 		if (rename(data->path, oldname)) {  in check_backup()
221 			pr_err("Can't move data: %s (%s to %s)\n",  in check_backup()
223 			       data->path, oldname);  in check_backup()
231 static bool is_dir(struct perf_data *data)  in is_dir()  argument
235 	if (stat(data->path, &st))  in is_dir()
241 static int open_file_read(struct perf_data *data)  in open_file_read()  argument
243 	int flags = data->in_place_update ? O_RDWR : O_RDONLY;  in open_file_read()
248 	fd = open(data->file.path, flags);  in open_file_read()
252 		pr_err("failed to open %s: %s", data->file.path,  in open_file_read()
254 		if (err == ENOENT && !strcmp(data->file.path, "perf.data"))  in open_file_read()
263 	if (!data->force && st.st_uid && (st.st_uid != geteuid())) {  in open_file_read()
265 		       data->file.path);  in open_file_read()
270 		pr_info("zero-sized data (%s), nothing to do!\n",  in open_file_read()
271 			data->file.path);  in open_file_read()
275 	data->file.size = st.st_size;  in open_file_read()
283 static int open_file_write(struct perf_data *data)  in open_file_write()  argument
288 	fd = open(data->file.path, O_CREAT|O_RDWR|O_TRUNC|O_CLOEXEC,  in open_file_write()
292 		pr_err("failed to open %s : %s\n", data->file.path,  in open_file_write()
298 static int open_file(struct perf_data *data)  in open_file()  argument
302 	fd = perf_data__is_read(data) ?  in open_file()
303 	     open_file_read(data) : open_file_write(data);  in open_file()
306 		zfree(&data->file.path);  in open_file()
310 	data->file.fd = fd;  in open_file()
314 static int open_file_dup(struct perf_data *data)  in open_file_dup()  argument
316 	data->file.path = strdup(data->path);  in open_file_dup()
317 	if (!data->file.path)  in open_file_dup()
320 	return open_file(data);  in open_file_dup()
323 static int open_dir(struct perf_data *data)  in open_dir()  argument
328 	 * So far we open only the header, so we can read the data version and  in open_dir()
331 	if (asprintf(&data->file.path, "%s/data", data->path) < 0)  in open_dir()
334 	if (perf_data__is_write(data) &&  in open_dir()
335 	    mkdir(data->path, S_IRWXU) < 0)  in open_dir()
338 	ret = open_file(data);  in open_dir()
341 	if (ret && perf_data__is_write(data))  in open_dir()
342 		rm_rf_perf_data(data->path);  in open_dir()
347 int perf_data__open(struct perf_data *data)  in perf_data__open()  argument
349 	if (check_pipe(data))  in perf_data__open()
353 	data->use_stdio = false;  in perf_data__open()
355 	if (!data->path)  in perf_data__open()
356 		data->path = "perf.data";  in perf_data__open()
358 	if (check_backup(data))  in perf_data__open()
361 	if (perf_data__is_read(data))  in perf_data__open()
362 		data->is_dir = is_dir(data);  in perf_data__open()
364 	return perf_data__is_dir(data) ?  in perf_data__open()
365 	       open_dir(data) : open_file_dup(data);  in perf_data__open()
368 void perf_data__close(struct perf_data *data)  in perf_data__close()  argument
370 	if (perf_data__is_dir(data))  in perf_data__close()
371 		perf_data__close_dir(data);  in perf_data__close()
373 	zfree(&data->file.path);  in perf_data__close()
375 	if (data->use_stdio)  in perf_data__close()
376 		fclose(data->file.fptr);  in perf_data__close()
378 		close(data->file.fd);  in perf_data__close()
381 ssize_t perf_data__read(struct perf_data *data, void *buf, size_t size)  in perf_data__read()  argument
383 	if (data->use_stdio) {  in perf_data__read()
384 		if (fread(buf, size, 1, data->file.fptr) == 1)  in perf_data__read()
386 		return feof(data->file.fptr) ? 0 : -1;  in perf_data__read()
388 	return readn(data->file.fd, buf, size);  in perf_data__read()
397 ssize_t perf_data__write(struct perf_data *data,  in perf_data__write()  argument
400 	if (data->use_stdio) {  in perf_data__write()
401 		if (fwrite(buf, size, 1, data->file.fptr) == 1)  in perf_data__write()
405 	return perf_data_file__write(&data->file, buf, size);  in perf_data__write()
408 int perf_data__switch(struct perf_data *data,  in perf_data__switch()  argument
415 	if (check_pipe(data))  in perf_data__switch()
417 	if (perf_data__is_read(data))  in perf_data__switch()
420 	if (asprintf(new_filepath, "%s.%s", data->path, postfix) < 0)  in perf_data__switch()
427 	if (rename(data->path, *new_filepath))  in perf_data__switch()
428 		pr_warning("Failed to rename %s to %s\n", data->path, *new_filepath);  in perf_data__switch()
431 		close(data->file.fd);  in perf_data__switch()
432 		ret = perf_data__open(data);  in perf_data__switch()
436 		if (lseek(data->file.fd, pos, SEEK_SET) == (off_t)-1) {  in perf_data__switch()
443 	ret = data->file.fd;  in perf_data__switch()
448 unsigned long perf_data__size(struct perf_data *data)  in perf_data__size()  argument
450 	u64 size = data->file.size;  in perf_data__size()
453 	if (perf_data__is_single_file(data))  in perf_data__size()
456 	for (i = 0; i < data->dir.nr; i++) {  in perf_data__size()
457 		struct perf_data_file *file = &data->dir.files[i];  in perf_data__size()
465 int perf_data__make_kcore_dir(struct perf_data *data, char *buf, size_t buf_sz)  in perf_data__make_kcore_dir()  argument
469 	if (!data->is_dir)  in perf_data__make_kcore_dir()
472 	ret = snprintf(buf, buf_sz, "%s/kcore_dir", data->path);  in perf_data__make_kcore_dir()
479 char *perf_data__kallsyms_name(struct perf_data *data)  in perf_data__kallsyms_name()  argument
484 	if (!data->is_dir)  in perf_data__kallsyms_name()
487 	if (asprintf(&kallsyms_name, "%s/kcore_dir/kallsyms", data->path) < 0)  in perf_data__kallsyms_name()