aboutsummaryrefslogtreecommitdiff
path: root/src/utility/io.h
blob: 4c09b30a2c7b096f225e9711979e141eaf154e3d (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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
#ifndef CHANGE_SRC_UTILITY_IO_H_
#define CHANGE_SRC_UTILITY_IO_H_

#include <unistd.h>
#include <dlfcn.h>
#include <fcntl.h>
#include <sys/stat.h>

#include <cstring>

namespace utility {

class FileDescriptorGuard {
	public:
		FileDescriptorGuard(const std::string& path) {
			this->fd_ = open(path.c_str(), O_CREAT | O_WRONLY, S_IRUSR | S_IWUSR);

			if ( !this->fd_ ) {
				this->fd_ = STDERR_FILENO;
			}
		}

		~FileDescriptorGuard() {
			close(this->fd_);
		}

		operator int() {
			return this->fd_;
		}

	private:
		int fd_;

};

std::string get_file_path(int fd) {
	char proc_link[20];
	char file_name[256];

	snprintf(proc_link, sizeof(proc_link), "/proc/self/fd/%d", fd);
	const ssize_t name_size = readlink(proc_link, file_name, sizeof(file_name));

	if ( name_size > 0 ) {
		file_name[name_size] = '\0';

		return std::string(file_name);
	} else {
		return std::string();
	}
}

bool is_regular_file(int fd) {
	struct stat fd_stat;
	fstat(fd, &fd_stat);

	return S_ISREG(fd_stat.st_mode);
}

bool is_regular_file(const char* path) {
	struct stat fd_stat;
	lstat(path, &fd_stat);

	return S_ISREG(fd_stat.st_mode);
}

}

#endif  // CHANGE_SRC_UTILITY_IO_H_