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
) {
23 if (stat(sockaddr
.sun_path
, &sb
) == -1)
25 mode_t mode
= sb
.st_mode
;
30 return chmod(sockaddr
.sun_path
, mode
);
33 static int server_create_socket(const char *name
) {
34 int socket
= create_socket(name
);
37 socklen_t socklen
= offsetof(struct sockaddr_un
, sun_path
) + strlen(sockaddr
.sun_path
) + 1;
38 mode_t mode
= S_IRUSR
|S_IWUSR
;
40 if (bind(socket
, (struct sockaddr
*)&sockaddr
, socklen
) == -1)
42 if (fchmod(socket
, mode
) == -1 && chmod(sockaddr
.sun_path
, mode
) == -1)
44 if (listen(socket
, 5) == -1)
48 unlink(sockaddr
.sun_path
);
52 static int server_set_socket_non_blocking(int sock
) {
54 if ((flags
= fcntl(sock
, F_GETFL
, 0)) == -1)
56 return fcntl(sock
, F_SETFL
, flags
| O_NONBLOCK
);
59 static Client
*server_accept_client() {
60 int newfd
= accept(server
.socket
, NULL
, NULL
);
63 Client
*c
= client_malloc(newfd
);
67 server_mark_socket_exec(true);
68 server_set_socket_non_blocking(newfd
);
70 c
->state
= STATE_CONNECTED
;
71 c
->next
= server
.clients
;
76 static bool server_read_pty(Packet
*pkt
) {
77 pkt
->type
= MSG_CONTENT
;
78 ssize_t len
= read(server
.pty
, pkt
->u
.msg
, sizeof(pkt
->u
.msg
));
81 else if (errno
!= EAGAIN
&& errno
!= EINTR
)
82 server
.running
= false;
83 print_packet("server-read-pty:", pkt
);
87 static bool server_write_pty(Packet
*pkt
) {
88 print_packet("server-write-pty:", pkt
);
89 size_t size
= pkt
->len
;
90 if (write_all(server
.pty
, pkt
->u
.msg
, size
) == size
)
93 server
.running
= false;
97 static bool server_recv_packet(Client
*c
, Packet
*pkt
) {
98 if (recv_packet(c
->socket
, pkt
)) {
99 print_packet("server-recv:", pkt
);
102 debug("server-recv: FAILED\n");
103 c
->state
= STATE_DISCONNECTED
;
107 static bool server_send_packet(Client
*c
, Packet
*pkt
) {
108 print_packet("server-send:", pkt
);
109 if (send_packet(c
->socket
, pkt
))
112 c
->state
= STATE_DISCONNECTED
;
116 static void server_pty_died_handler(int sig
) {
120 while ((pid
= waitpid(-1, &server
.exit_status
, WNOHANG
)) != 0) {
123 server
.exit_status
= WEXITSTATUS(server
.exit_status
);
126 debug("server pty died: %d\n", server
.exit_status
);
130 static void server_sigterm_handler(int sig
) {
131 exit(EXIT_FAILURE
); /* invoke atexit handler */
134 static void server_sigusr1_handler(int sig
) {
135 int socket
= server_create_socket(server
.session_name
);
138 close(server
.socket
);
139 server
.socket
= socket
;
143 static void server_atexit_handler() {
144 unlink(sockaddr
.sun_path
);
147 static void server_mainloop() {
148 atexit(server_atexit_handler
);
149 fd_set new_readfds
, new_writefds
;
150 FD_ZERO(&new_readfds
);
151 FD_ZERO(&new_writefds
);
152 FD_SET(server
.socket
, &new_readfds
);
153 int new_fdmax
= server
.socket
;
154 Packet
*exit_pkt
= NULL
;
157 int fdmax
= new_fdmax
;
158 fd_set readfds
= new_readfds
;
159 fd_set writefds
= new_writefds
;
160 FD_SET_MAX(server
.socket
, &readfds
, fdmax
);
162 if (select(fdmax
+1, &readfds
, &writefds
, NULL
, NULL
) == -1) {
165 die("server-mainloop");
168 FD_ZERO(&new_readfds
);
169 FD_ZERO(&new_writefds
);
170 new_fdmax
= server
.socket
;
172 bool pty_data
= false;
174 Packet server_packet
, client_packet
;
176 if (FD_ISSET(server
.socket
, &readfds
))
177 server_accept_client();
179 if (FD_ISSET(server
.pty
, &readfds
))
180 pty_data
= server_read_pty(&server_packet
);
182 if (!server
.running
&& server
.exit_status
!= -1 && server
.clients
) { /* application terminated */
185 .u
.i
= server
.exit_status
,
186 .len
= sizeof(pkt
.u
.i
),
189 server
.exit_status
= -1;
192 for (Client
**prev_next
= &server
.clients
, *c
= server
.clients
; c
;) {
193 if (c
->state
== STATE_DISCONNECTED
) {
194 bool first
= (c
== server
.clients
);
198 if (first
&& server
.clients
) {
203 server_send_packet(server
.clients
, &pkt
);
204 } else if (!server
.clients
) {
205 server_mark_socket_exec(false);
210 if (FD_ISSET(c
->socket
, &readfds
) && server_recv_packet(c
, &client_packet
)) {
211 switch (client_packet
.type
) {
213 server_write_pty(&client_packet
);
216 c
->readonly
= client_packet
.u
.b
;
219 c
->state
= STATE_ATTACHED
;
221 if (!c
->readonly
&& (client_packet
.type
== MSG_REDRAW
|| c
== server
.clients
)) {
222 debug("server-ioct: TIOCSWINSZ\n");
223 ioctl(server
.pty
, TIOCSWINSZ
, &client_packet
.u
.ws
);
225 kill(-server
.pid
, SIGWINCH
);
228 c
->state
= STATE_DETACHED
;
230 default: /* ignore package */
235 FD_SET_MAX(c
->socket
, &new_readfds
, new_fdmax
);
238 server_send_packet(c
, &server_packet
);
240 server_send_packet(c
, exit_pkt
);
242 prev_next
= &c
->next
;
247 FD_SET_MAX(server
.pty
, &new_readfds
, new_fdmax
);