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 int server_mark_socket_exec(bool exec
, bool usr
) {
23 if (stat(sockaddr
.sun_path
, &sb
) == -1)
25 mode_t mode
= sb
.st_mode
;
26 mode_t flag
= usr
? S_IXUSR
: S_IXGRP
;
31 return chmod(sockaddr
.sun_path
, mode
);
34 static int server_create_socket(const char *name
) {
35 int socket
= create_socket(name
);
38 socklen_t socklen
= offsetof(struct sockaddr_un
, sun_path
) + strlen(sockaddr
.sun_path
) + 1;
39 mode_t mode
= S_IRUSR
|S_IWUSR
;
41 if (bind(socket
, (struct sockaddr
*)&sockaddr
, socklen
) == -1)
43 if (fchmod(socket
, mode
) == -1 && chmod(sockaddr
.sun_path
, mode
) == -1)
45 if (listen(socket
, 5) == -1)
49 unlink(sockaddr
.sun_path
);
53 static int server_set_socket_non_blocking(int sock
) {
55 if ((flags
= fcntl(sock
, F_GETFL
, 0)) == -1)
57 return fcntl(sock
, F_SETFL
, flags
| O_NONBLOCK
);
60 static Client
*server_accept_client() {
61 int newfd
= accept(server
.socket
, NULL
, NULL
);
64 Client
*c
= client_malloc(newfd
);
68 server_mark_socket_exec(true, true);
69 server_set_socket_non_blocking(newfd
);
71 c
->state
= STATE_CONNECTED
;
72 c
->next
= server
.clients
;
74 server
.read_pty
= true;
78 static bool server_read_pty(Packet
*pkt
) {
79 pkt
->type
= MSG_CONTENT
;
80 ssize_t len
= read(server
.pty
, pkt
->u
.msg
, sizeof(pkt
->u
.msg
));
83 else if (errno
!= EAGAIN
&& errno
!= EINTR
)
84 server
.running
= false;
85 print_packet("server-read-pty:", pkt
);
89 static bool server_write_pty(Packet
*pkt
) {
90 print_packet("server-write-pty:", pkt
);
91 size_t size
= pkt
->len
;
92 if (write_all(server
.pty
, pkt
->u
.msg
, size
) == size
)
95 server
.running
= false;
99 static bool server_recv_packet(Client
*c
, Packet
*pkt
) {
100 if (recv_packet(c
->socket
, pkt
)) {
101 print_packet("server-recv:", pkt
);
104 debug("server-recv: FAILED\n");
105 c
->state
= STATE_DISCONNECTED
;
109 static bool server_send_packet(Client
*c
, Packet
*pkt
) {
110 print_packet("server-send:", pkt
);
111 if (send_packet(c
->socket
, pkt
))
114 c
->state
= STATE_DISCONNECTED
;
118 static void server_pty_died_handler(int sig
) {
122 while ((pid
= waitpid(-1, &server
.exit_status
, WNOHANG
)) != 0) {
125 server
.exit_status
= WEXITSTATUS(server
.exit_status
);
126 server_mark_socket_exec(true, false);
129 debug("server pty died: %d\n", server
.exit_status
);
133 static void server_sigterm_handler(int sig
) {
134 exit(EXIT_FAILURE
); /* invoke atexit handler */
137 static void server_sigusr1_handler(int sig
) {
138 int socket
= server_create_socket(server
.session_name
);
141 close(server
.socket
);
142 server
.socket
= socket
;
146 static void server_atexit_handler() {
147 unlink(sockaddr
.sun_path
);
150 static void server_mainloop() {
151 atexit(server_atexit_handler
);
152 fd_set new_readfds
, new_writefds
;
153 FD_ZERO(&new_readfds
);
154 FD_ZERO(&new_writefds
);
155 FD_SET(server
.socket
, &new_readfds
);
156 int new_fdmax
= server
.socket
;
157 bool exit_packet_delivered
= false;
160 FD_SET_MAX(server
.pty
, &new_readfds
, new_fdmax
);
162 while (server
.clients
|| !exit_packet_delivered
) {
163 int fdmax
= new_fdmax
;
164 fd_set readfds
= new_readfds
;
165 fd_set writefds
= new_writefds
;
166 FD_SET_MAX(server
.socket
, &readfds
, fdmax
);
168 if (select(fdmax
+1, &readfds
, &writefds
, NULL
, NULL
) == -1) {
171 die("server-mainloop");
174 FD_ZERO(&new_readfds
);
175 FD_ZERO(&new_writefds
);
176 new_fdmax
= server
.socket
;
178 bool pty_data
= false;
180 Packet server_packet
, client_packet
;
182 if (FD_ISSET(server
.socket
, &readfds
))
183 server_accept_client();
185 if (FD_ISSET(server
.pty
, &readfds
))
186 pty_data
= server_read_pty(&server_packet
);
188 for (Client
**prev_next
= &server
.clients
, *c
= server
.clients
; c
;) {
189 if (FD_ISSET(c
->socket
, &readfds
) && server_recv_packet(c
, &client_packet
)) {
190 switch (client_packet
.type
) {
192 server_write_pty(&client_packet
);
195 c
->readonly
= client_packet
.u
.b
;
198 c
->state
= STATE_ATTACHED
;
200 if (!c
->readonly
&& (client_packet
.type
== MSG_REDRAW
|| c
== server
.clients
)) {
201 debug("server-ioct: TIOCSWINSZ\n");
202 ioctl(server
.pty
, TIOCSWINSZ
, &client_packet
.u
.ws
);
204 kill(-server
.pid
, SIGWINCH
);
208 c
->state
= STATE_DISCONNECTED
;
210 default: /* ignore package */
215 if (c
->state
== STATE_DISCONNECTED
) {
216 bool first
= (c
== server
.clients
);
220 if (first
&& server
.clients
) {
225 server_send_packet(server
.clients
, &pkt
);
226 } else if (!server
.clients
) {
227 server_mark_socket_exec(false, true);
232 FD_SET_MAX(c
->socket
, &new_readfds
, new_fdmax
);
235 server_send_packet(c
, &server_packet
);
236 if (!server
.running
) {
237 if (server
.exit_status
!= -1) {
240 .u
.i
= server
.exit_status
,
241 .len
= sizeof(pkt
.u
.i
),
243 if (server_send_packet(c
, &pkt
))
244 exit_packet_delivered
= true;
246 FD_SET_MAX(c
->socket
, &new_writefds
, new_fdmax
);
248 FD_SET_MAX(c
->socket
, &new_writefds
, new_fdmax
);
251 prev_next
= &c
->next
;
255 if (server
.running
&& server
.read_pty
)
256 FD_SET_MAX(server
.pty
, &new_readfds
, new_fdmax
);