1 #define FD_SET_MAX(fd, set, maxfd) do { \
7 static Client
*client_malloc(int socket
) {
8 Client
*c
= calloc(1, sizeof(Client
));
15 static void client_free(Client
*c
) {
16 if (c
&& c
->socket
> 0)
21 static void server_sink_client() {
22 if (!server
.clients
|| !server
.clients
->next
)
24 Client
*target
= server
.clients
;
25 server
.clients
= target
->next
;
26 Client
*dst
= server
.clients
;
33 static void server_mark_socket_exec(bool exec
, bool usr
) {
35 if (stat(sockaddr
.sun_path
, &sb
) == -1)
37 mode_t mode
= sb
.st_mode
;
38 mode_t flag
= usr
? S_IXUSR
: S_IXGRP
;
43 chmod(sockaddr
.sun_path
, mode
);
46 static int server_create_socket(const char *name
) {
47 if (!set_socket_name(&sockaddr
, name
))
49 int fd
= socket(AF_UNIX
, SOCK_STREAM
, 0);
52 socklen_t socklen
= offsetof(struct sockaddr_un
, sun_path
) + strlen(sockaddr
.sun_path
) + 1;
53 mode_t mask
= umask(S_IXUSR
|S_IRWXG
|S_IRWXO
);
54 int r
= bind(fd
, (struct sockaddr
*)&sockaddr
, socklen
);
57 if (r
== -1 || chmod(sockaddr
.sun_path
, S_ISVTX
|S_IRUSR
|S_IWUSR
) == -1) {
62 if (listen(fd
, 5) == -1) {
63 unlink(sockaddr
.sun_path
);
71 static int server_set_socket_non_blocking(int sock
) {
73 if ((flags
= fcntl(sock
, F_GETFL
, 0)) == -1)
75 return fcntl(sock
, F_SETFL
, flags
| O_NONBLOCK
);
78 static Client
*server_accept_client(void) {
79 int newfd
= accept(server
.socket
, NULL
, NULL
);
80 if (newfd
== -1 || server_set_socket_non_blocking(newfd
) == -1)
82 Client
*c
= client_malloc(newfd
);
86 server_mark_socket_exec(true, true);
88 c
->state
= STATE_CONNECTED
;
89 c
->next
= server
.clients
;
91 server
.read_pty
= true;
99 static bool server_read_pty(Packet
*pkt
) {
100 pkt
->type
= MSG_CONTENT
;
101 ssize_t len
= read(server
.pty
, pkt
->u
.msg
, sizeof(pkt
->u
.msg
));
105 server
.running
= false;
106 else if (len
== -1 && errno
!= EAGAIN
&& errno
!= EINTR
&& errno
!= EWOULDBLOCK
)
107 server
.running
= false;
108 print_packet("server-read-pty:", pkt
);
112 static bool server_write_pty(Packet
*pkt
) {
113 print_packet("server-write-pty:", pkt
);
114 size_t size
= pkt
->len
;
115 if (write_all(server
.pty
, pkt
->u
.msg
, size
) == size
)
118 server
.running
= false;
122 static bool server_recv_packet(Client
*c
, Packet
*pkt
) {
123 if (recv_packet(c
->socket
, pkt
)) {
124 print_packet("server-recv:", pkt
);
127 debug("server-recv: FAILED\n");
128 c
->state
= STATE_DISCONNECTED
;
132 static bool server_send_packet(Client
*c
, Packet
*pkt
) {
133 print_packet("server-send:", pkt
);
134 if (send_packet(c
->socket
, pkt
))
137 c
->state
= STATE_DISCONNECTED
;
141 static void server_pty_died_handler(int sig
) {
145 while ((pid
= waitpid(-1, &server
.exit_status
, WNOHANG
)) != 0) {
148 server
.exit_status
= WEXITSTATUS(server
.exit_status
);
149 server_mark_socket_exec(true, false);
152 debug("server pty died: %d\n", server
.exit_status
);
156 static void server_sigterm_handler(int sig
) {
157 exit(EXIT_FAILURE
); /* invoke atexit handler */
160 static void server_sigusr1_handler(int sig
) {
161 int socket
= server_create_socket(server
.session_name
);
164 close(server
.socket
);
165 server
.socket
= socket
;
169 static void server_atexit_handler(void) {
170 unlink(sockaddr
.sun_path
);
173 static void server_mainloop(void) {
174 atexit(server_atexit_handler
);
175 fd_set new_readfds
, new_writefds
;
176 FD_ZERO(&new_readfds
);
177 FD_ZERO(&new_writefds
);
178 FD_SET(server
.socket
, &new_readfds
);
179 int new_fdmax
= server
.socket
;
180 bool exit_packet_delivered
= false;
183 FD_SET_MAX(server
.pty
, &new_readfds
, new_fdmax
);
185 while (server
.clients
|| !exit_packet_delivered
) {
186 int fdmax
= new_fdmax
;
187 fd_set readfds
= new_readfds
;
188 fd_set writefds
= new_writefds
;
189 FD_SET_MAX(server
.socket
, &readfds
, fdmax
);
191 if (select(fdmax
+1, &readfds
, &writefds
, NULL
, NULL
) == -1) {
194 die("server-mainloop");
197 FD_ZERO(&new_readfds
);
198 FD_ZERO(&new_writefds
);
199 new_fdmax
= server
.socket
;
201 bool pty_data
= false;
203 Packet server_packet
, client_packet
;
205 if (FD_ISSET(server
.socket
, &readfds
))
206 server_accept_client();
208 if (FD_ISSET(server
.pty
, &readfds
))
209 pty_data
= server_read_pty(&server_packet
);
211 for (Client
**prev_next
= &server
.clients
, *c
= server
.clients
; c
;) {
212 if (FD_ISSET(c
->socket
, &readfds
) && server_recv_packet(c
, &client_packet
)) {
213 switch (client_packet
.type
) {
215 server_write_pty(&client_packet
);
218 c
->flags
= client_packet
.u
.i
;
219 if (c
->flags
& CLIENT_LOWPRIORITY
)
220 server_sink_client();
223 c
->state
= STATE_ATTACHED
;
225 if (!(c
->flags
& CLIENT_READONLY
) && (client_packet
.type
== MSG_REDRAW
|| c
== server
.clients
)) {
226 debug("server-ioct: TIOCSWINSZ\n");
227 ioctl(server
.pty
, TIOCSWINSZ
, &client_packet
.u
.ws
);
229 kill(-server
.pid
, SIGWINCH
);
232 exit_packet_delivered
= true;
235 c
->state
= STATE_DISCONNECTED
;
237 default: /* ignore package */
242 if (c
->state
== STATE_DISCONNECTED
) {
243 bool first
= (c
== server
.clients
);
247 if (first
&& server
.clients
) {
252 server_send_packet(server
.clients
, &pkt
);
253 } else if (!server
.clients
) {
254 server_mark_socket_exec(false, true);
259 FD_SET_MAX(c
->socket
, &new_readfds
, new_fdmax
);
262 server_send_packet(c
, &server_packet
);
263 if (!server
.running
) {
264 if (server
.exit_status
!= -1) {
267 .u
.i
= server
.exit_status
,
268 .len
= sizeof(pkt
.u
.i
),
270 if (!server_send_packet(c
, &pkt
))
271 FD_SET_MAX(c
->socket
, &new_writefds
, new_fdmax
);
273 FD_SET_MAX(c
->socket
, &new_writefds
, new_fdmax
);
276 prev_next
= &c
->next
;
280 if (server
.running
&& server
.read_pty
)
281 FD_SET_MAX(server
.pty
, &new_readfds
, new_fdmax
);