github
/
abduco
mirror of https://github.com/martanne/abduco
1
0
Fork 1
Code Issues Releases Wiki Activity

Make error handling more robust

This commit is contained in:
Marc André Tanner 2015-02-16 23:28:46 +01:00
parent d1a44a9c1a
commit e4a6b01fdc
2 changed files with 13 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
abduco.c
View File

@ -316,9 +316,9 @@ static bool create_session(const char *name, char * const argv[]) {
case 0: /* child = user application process */
close(server.socket);
close(server_pipe[0]);
fcntl(client_pipe[1], F_SETFD, FD_CLOEXEC);
fcntl(server_pipe[1], F_SETFD, FD_CLOEXEC);
execvp(argv[0], argv);
if (fcntl(client_pipe[1], F_SETFD, FD_CLOEXEC) == 0 &&
fcntl(server_pipe[1], F_SETFD, FD_CLOEXEC) == 0)
execvp(argv[0], argv);
snprintf(errormsg, sizeof(errormsg), "server-execvp: %s: %s\n",
argv[0], strerror(errno));
write_all(client_pipe[1], errormsg, strlen(errormsg));

17
server.c
View File

@ -18,17 +18,17 @@ static void client_free(Client *c) {
free(c);
}
static int server_mark_socket_exec(bool exec, bool usr) {
static void server_mark_socket_exec(bool exec, bool usr) {
struct stat sb;
if (stat(sockaddr.sun_path, &sb) == -1)
return -1;
return;
mode_t mode = sb.st_mode;
mode_t flag = usr ? S_IXUSR : S_IXGRP;
if (exec)
mode |= flag;
else
mode &= ~flag;
return chmod(sockaddr.sun_path, mode);
chmod(sockaddr.sun_path, mode);
}
static int server_create_socket(const char *name) {
@ -63,20 +63,23 @@ static int server_set_socket_non_blocking(int sock) {
static Client *server_accept_client(void) {
int newfd = accept(server.socket, NULL, NULL);
if (newfd == -1)
return NULL;
if (newfd == -1 || server_set_socket_non_blocking(newfd) == -1)
goto error;
Client *c = client_malloc(newfd);
if (!c)
return NULL;
goto error;
if (!server.clients)
server_mark_socket_exec(true, true);
server_set_socket_non_blocking(newfd);
c->socket = newfd;
c->state = STATE_CONNECTED;
c->next = server.clients;
server.clients = c;
server.read_pty = true;
return c;
error:
if (newfd != -1)
close(newfd);
return NULL;
}
static bool server_read_pty(Packet *pkt) {