blob: 32d6ecb667029dc33cfa1632fbf439deb4f3a04a (
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_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);
}
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_
|
