aboutsummaryrefslogtreecommitdiff
path: root/src/io.h
blob: c36a5b023c5194a84f87a8fb3746994df3e0bfa2 (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
#ifndef CHANGE_IO_H_
#define CHANGE_IO_H_

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

#include <cstring>

namespace io {

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

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

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

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

	private:
		int fd;

};

std::string get_file_name(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);
}

}

#endif  // CHANGE_IO_H_