16 #include "data.h"
31 void perf_data__close_dir(struct perf_data *data)  in perf_data__close_dir()  argument
33 	close_dir(data->dir.files, data->dir.nr);  in perf_data__close_dir()
36 int perf_data__create_dir(struct perf_data *data, int nr)  in perf_data__create_dir()  argument
41 	if (WARN_ON(!data->is_dir))  in perf_data__create_dir()
51 		ret = asprintf(&file->path, "%s/data.%d", data->path, i);  in perf_data__create_dir()
66 	data->dir.version = PERF_DIR_VERSION;  in perf_data__create_dir()
67 	data->dir.files   = files;  in perf_data__create_dir()
68 	data->dir.nr      = nr;  in perf_data__create_dir()
76 int perf_data__open_dir(struct perf_data *data)  in perf_data__open_dir()  argument
85 	 * Directory containing a single regular perf data file which is already  in perf_data__open_dir()
88 	if (perf_data__is_single_file(data))  in perf_data__open_dir()
91 	if (WARN_ON(!data->is_dir))  in perf_data__open_dir()
95 	if (WARN_ON(data->dir.version != PERF_DIR_VERSION))  in perf_data__open_dir()
98 	dir = opendir(data->path);  in perf_data__open_dir()
107 		snprintf(path, sizeof(path), "%s/%s", data->path, dent->d_name);  in perf_data__open_dir()
111 		if (!S_ISREG(st.st_mode) || strncmp(dent->d_name, "data.", 5))  in perf_data__open_dir()
138 	data->dir.files = files;  in perf_data__open_dir()
139 	data->dir.nr    = nr;  in perf_data__open_dir()
147 int perf_data__update_dir(struct perf_data *data)  in perf_data__update_dir()  argument
151 	if (WARN_ON(!data->is_dir))  in perf_data__update_dir()
154 	for (i = 0; i < data->dir.nr; i++) {  in perf_data__update_dir()
155 		struct perf_data_file *file = &data->dir.files[i];  in perf_data__update_dir()
167 static bool check_pipe(struct perf_data *data)  in check_pipe()  argument
171 	int fd = perf_data__is_read(data) ?  in check_pipe()
174 	if (!data->path) {  in check_pipe()
178 		if (!strcmp(data->path, "-"))  in check_pipe()
183 		if (data->use_stdio) {  in check_pipe()
186 			mode = perf_data__is_read(data) ? "r" : "w";  in check_pipe()
187 			data->file.fptr = fdopen(fd, mode);  in check_pipe()
189 			if (data->file.fptr == NULL) {  in check_pipe()
190 				data->file.fd = fd;  in check_pipe()
191 				data->use_stdio = false;  in check_pipe()
194 			data->file.fd = fd;  in check_pipe()
198 	return data->is_pipe = is_pipe;  in check_pipe()
201 static int check_backup(struct perf_data *data)  in check_backup()  argument
205 	if (perf_data__is_read(data))  in check_backup()
208 	if (!stat(data->path, &st) && st.st_size) {  in check_backup()
213 			 data->path);  in check_backup()
217 			pr_err("Can't remove old data: %s (%s)\n",  in check_backup()
224 		if (rename(data->path, oldname)) {  in check_backup()
225 			pr_err("Can't move data: %s (%s to %s)\n",  in check_backup()
227 			       data->path, oldname);  in check_backup()
235 static bool is_dir(struct perf_data *data)  in is_dir()  argument
239 	if (stat(data->path, &st))  in is_dir()
245 static int open_file_read(struct perf_data *data)  in open_file_read()  argument
247 	int flags = data->in_place_update ? O_RDWR : O_RDONLY;  in open_file_read()
252 	fd = open(data->file.path, flags);  in open_file_read()
256 		pr_err("failed to open %s: %s", data->file.path,  in open_file_read()
258 		if (err == ENOENT && !strcmp(data->file.path, "perf.data"))  in open_file_read()
267 	if (!data->force && st.st_uid && (st.st_uid != geteuid())) {  in open_file_read()
269 		       data->file.path);  in open_file_read()
274 		pr_info("zero-sized data (%s), nothing to do!\n",  in open_file_read()
275 			data->file.path);  in open_file_read()
279 	data->file.size = st.st_size;  in open_file_read()
287 static int open_file_write(struct perf_data *data)  in open_file_write()  argument
292 	fd = open(data->file.path, O_CREAT|O_RDWR|O_TRUNC|O_CLOEXEC,  in open_file_write()
296 		pr_err("failed to open %s : %s\n", data->file.path,  in open_file_write()
302 static int open_file(struct perf_data *data)  in open_file()  argument
306 	fd = perf_data__is_read(data) ?  in open_file()
307 	     open_file_read(data) : open_file_write(data);  in open_file()
310 		zfree(&data->file.path);  in open_file()
314 	data->file.fd = fd;  in open_file()
318 static int open_file_dup(struct perf_data *data)  in open_file_dup()  argument
320 	data->file.path = strdup(data->path);  in open_file_dup()
321 	if (!data->file.path)  in open_file_dup()
324 	return open_file(data);  in open_file_dup()
327 static int open_dir(struct perf_data *data)  in open_dir()  argument
332 	 * So far we open only the header, so we can read the data version and  in open_dir()
335 	if (asprintf(&data->file.path, "%s/data", data->path) < 0)  in open_dir()
338 	if (perf_data__is_write(data) &&  in open_dir()
339 	    mkdir(data->path, S_IRWXU) < 0)  in open_dir()
342 	ret = open_file(data);  in open_dir()
345 	if (ret && perf_data__is_write(data))  in open_dir()
346 		rm_rf_perf_data(data->path);  in open_dir()
351 int perf_data__open(struct perf_data *data)  in perf_data__open()  argument
353 	if (check_pipe(data))  in perf_data__open()
357 	data->use_stdio = false;  in perf_data__open()
359 	if (!data->path)  in perf_data__open()
360 		data->path = "perf.data";  in perf_data__open()
362 	if (check_backup(data))  in perf_data__open()
365 	if (perf_data__is_read(data))  in perf_data__open()
366 		data->is_dir = is_dir(data);  in perf_data__open()
368 	return perf_data__is_dir(data) ?  in perf_data__open()
369 	       open_dir(data) : open_file_dup(data);  in perf_data__open()
372 void perf_data__close(struct perf_data *data)  in perf_data__close()  argument
374 	if (perf_data__is_dir(data))  in perf_data__close()
375 		perf_data__close_dir(data);  in perf_data__close()
377 	zfree(&data->file.path);  in perf_data__close()
379 	if (data->use_stdio)  in perf_data__close()
380 		fclose(data->file.fptr);  in perf_data__close()
382 		close(data->file.fd);  in perf_data__close()
385 ssize_t perf_data__read(struct perf_data *data, void *buf, size_t size)  in perf_data__read()  argument
387 	if (data->use_stdio) {  in perf_data__read()
388 		if (fread(buf, size, 1, data->file.fptr) == 1)  in perf_data__read()
390 		return feof(data->file.fptr) ? 0 : -1;  in perf_data__read()
392 	return readn(data->file.fd, buf, size);  in perf_data__read()
401 ssize_t perf_data__write(struct perf_data *data,  in perf_data__write()  argument
404 	if (data->use_stdio) {  in perf_data__write()
405 		if (fwrite(buf, size, 1, data->file.fptr) == 1)  in perf_data__write()
409 	return perf_data_file__write(&data->file, buf, size);  in perf_data__write()
412 int perf_data__switch(struct perf_data *data,  in perf_data__switch()  argument
419 	if (check_pipe(data))  in perf_data__switch()
421 	if (perf_data__is_read(data))  in perf_data__switch()
424 	if (asprintf(new_filepath, "%s.%s", data->path, postfix) < 0)  in perf_data__switch()
431 	if (rename(data->path, *new_filepath))  in perf_data__switch()
432 		pr_warning("Failed to rename %s to %s\n", data->path, *new_filepath);  in perf_data__switch()
435 		close(data->file.fd);  in perf_data__switch()
436 		ret = perf_data__open(data);  in perf_data__switch()
440 		if (lseek(data->file.fd, pos, SEEK_SET) == (off_t)-1) {  in perf_data__switch()
447 	ret = data->file.fd;  in perf_data__switch()
452 unsigned long perf_data__size(struct perf_data *data)  in perf_data__size()  argument
454 	u64 size = data->file.size;  in perf_data__size()
457 	if (perf_data__is_single_file(data))  in perf_data__size()
460 	for (i = 0; i < data->dir.nr; i++) {  in perf_data__size()
461 		struct perf_data_file *file = &data->dir.files[i];  in perf_data__size()
469 int perf_data__make_kcore_dir(struct perf_data *data, char *buf, size_t buf_sz)  in perf_data__make_kcore_dir()  argument
473 	if (!data->is_dir)  in perf_data__make_kcore_dir()
476 	ret = snprintf(buf, buf_sz, "%s/kcore_dir", data->path);  in perf_data__make_kcore_dir()
502 char *perf_data__kallsyms_name(struct perf_data *data)  in perf_data__kallsyms_name()  argument
507 	if (!data->is_dir)  in perf_data__kallsyms_name()
510 	if (asprintf(&kallsyms_name, "%s/kcore_dir/kallsyms", data->path) < 0)  in perf_data__kallsyms_name()
521 char *perf_data__guest_kallsyms_name(struct perf_data *data, pid_t machine_pid)  in perf_data__guest_kallsyms_name()  argument
526 	if (!data->is_dir)  in perf_data__guest_kallsyms_name()
529 	if (asprintf(&kallsyms_name, "%s/kcore_dir__%d/kallsyms", data->path, machine_pid) < 0)  in perf_data__guest_kallsyms_name()