terminal: prevent returning invalid pointer

parent 26ed61e0
...@@ -115,8 +115,8 @@ int lxc_terminal_signalfd_cb(int fd, uint32_t events, void *cbdata, ...@@ -115,8 +115,8 @@ int lxc_terminal_signalfd_cb(int fd, uint32_t events, void *cbdata,
struct lxc_terminal_state *lxc_terminal_signal_init(int srcfd, int dstfd) struct lxc_terminal_state *lxc_terminal_signal_init(int srcfd, int dstfd)
{ {
__do_close_prot_errno int signal_fd = -EBADF;
__do_free struct lxc_terminal_state *ts = NULL; __do_free struct lxc_terminal_state *ts = NULL;
bool istty = false;
int ret; int ret;
sigset_t mask; sigset_t mask;
...@@ -132,49 +132,40 @@ struct lxc_terminal_state *lxc_terminal_signal_init(int srcfd, int dstfd) ...@@ -132,49 +132,40 @@ struct lxc_terminal_state *lxc_terminal_signal_init(int srcfd, int dstfd)
ret = sigemptyset(&mask); ret = sigemptyset(&mask);
if (ret < 0) { if (ret < 0) {
SYSERROR("Failed to initialize an empty signal set"); SYSERROR("Failed to initialize an empty signal set");
goto on_error; return NULL;
} }
istty = (isatty(srcfd) == 1); if (isatty(srcfd)) {
if (!istty) {
INFO("fd %d does not refer to a tty device", srcfd);
} else {
ret = sigaddset(&mask, SIGWINCH); ret = sigaddset(&mask, SIGWINCH);
if (ret < 0) if (ret < 0)
SYSNOTICE("Failed to add SIGWINCH to signal set"); SYSNOTICE("Failed to add SIGWINCH to signal set");
} else {
INFO("fd %d does not refer to a tty device", srcfd);
} }
/* Exit the mainloop cleanly on SIGTERM. */ /* Exit the mainloop cleanly on SIGTERM. */
ret = sigaddset(&mask, SIGTERM); ret = sigaddset(&mask, SIGTERM);
if (ret < 0) { if (ret < 0) {
SYSERROR("Failed to add SIGWINCH to signal set"); SYSERROR("Failed to add SIGWINCH to signal set");
goto on_error; return NULL;
} }
ret = pthread_sigmask(SIG_BLOCK, &mask, &ts->oldmask); ret = pthread_sigmask(SIG_BLOCK, &mask, &ts->oldmask);
if (ret < 0) { if (ret < 0) {
WARN("Failed to block signals"); WARN("Failed to block signals");
goto on_error; return NULL;
} }
ts->sigfd = signalfd(-1, &mask, SFD_CLOEXEC); signal_fd = signalfd(-1, &mask, SFD_CLOEXEC);
if (ts->sigfd < 0) { if (signal_fd < 0) {
WARN("Failed to create signal fd"); WARN("Failed to create signal fd");
(void)pthread_sigmask(SIG_SETMASK, &ts->oldmask, NULL); (void)pthread_sigmask(SIG_SETMASK, &ts->oldmask, NULL);
goto on_error; return NULL;
}
DEBUG("Created signal fd %d", ts->sigfd);
return ts;
on_error:
ERROR("Failed to create signal fd");
if (ts->sigfd >= 0) {
close(ts->sigfd);
ts->sigfd = -1;
} }
ts->sigfd = move_fd(signal_fd);
TRACE("Created signal fd %d", ts->sigfd);
return NULL; return move_ptr(ts);
} }
/** /**
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment