forked from team-unstablers/ulalaca-xrdp
-
Notifications
You must be signed in to change notification settings - Fork 1
/
UnixSocket.hpp
115 lines (90 loc) · 2.54 KB
/
UnixSocket.hpp
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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
//
// Created by cheesekun on 2/28/22.
//
#ifndef ULALACA_UNIXSOCKET_HPP
#define ULALACA_UNIXSOCKET_HPP
#include <string>
#include <sys/socket.h>
#include <sys/un.h>
#include "SystemCallException.hpp"
using FD = int;
class UnixSocketBase {
public:
/**
* 파일 디스크립터를 반환합니다.
*
* @note 임의로 닫지 마십시오.
*/
virtual FD descriptor() = 0;
/**
* 소켓으로부터 데이터를 읽습니다.
*
* @param buffer 입력 버퍼
* @param size 읽고자 하는 데이터의 크기
*
* @return 버퍼에 실제로 읽힌 데이터의 크기를 반환합니다.
* 플랫폼에 따라 오류가 발생하면 음수 값이 반환될 수 있습니다.
*/
ssize_t read(void *buffer, size_t size);
/**
* 소켓에 데이터를 씁니다.
*
* @param buffer 출력 버퍼
* @param size 쓰고자 하는 데이터의 크기
*
* @return 소켓에 실제로 쓰여진 데이터의 크기를 반환합니다.
* 플랫폼에 따라 오류가 발생하면 음수 값이 반환될 수 있습니다.
*/
ssize_t write(const void *buffer, size_t size);
/**
* 소켓을 닫습니다.
*/
void close();
};
class UnixSocketConnection: public UnixSocketBase {
public:
explicit UnixSocketConnection(FD descriptor, sockaddr_un clientAddress);
FD descriptor() override;
private:
FD _descriptor;
sockaddr_un _clientAddress;
};
class UnixSocket: public UnixSocketBase {
public:
explicit UnixSocket(const std::string path);
/**
* bind(2)를 호출합니다.
*
* @see man 2 bind
* @throws SystemCallException
* bind()의 리턴 값이 -1일 시 errno를 담은 예외를 던집니다.
*/
void bind();
/**
* listen(2)를 호출합니다.
*
* @see man 2 listen
* @throws SystemCallException
* listen()의 리턴 값이 -1일 시 errno를 담은 예외를 던집니다.
*/
void listen();
/**
* accept(2)를 호출합니다.
*
* @see man 2 accept
* @return (File Descriptor, sockaddr_un)의 pair를 반환합니다.
* @throws SystemCallException
* accept()의 리턴 값이 -1일 시 errno를 담은 예외를 던집니다.
*/
UnixSocketConnection accept();
/**
* connect(2)를 호출합니다.
*/
void connect();
FD descriptor() override;
static FD createSocket();
private:
std::string _path;
FD _descriptor;
};
#endif //ULALACA_UNIXSOCKET_HPP