mirror of
https://github.com/tmate-io/tmate.git
synced 2025-02-23 13:51:40 +01:00
Switch jobs over to use a bufferevent.
This commit is contained in:
parent
ea8c8c5f33
commit
4d6091379b
2
Makefile
2
Makefile
@ -35,7 +35,7 @@ SRCS= attributes.c buffer-poll.c buffer.c cfg.c \
|
||||
layout-set.c layout.c log.c job.c \
|
||||
mode-key.c names.c options-cmd.c options.c paste.c procname.c \
|
||||
resize.c screen-redraw.c screen-write.c screen.c session.c status.c \
|
||||
server-fn.c server.c server-client.c server-window.c server-job.c \
|
||||
server-fn.c server.c server-client.c server-window.c \
|
||||
tmux.c tty-keys.c tty-term.c tty.c utf8.c \
|
||||
window-choose.c window-clock.c window-copy.c window-more.c window.c \
|
||||
xterm-keys.c xmalloc.c
|
||||
|
@ -77,31 +77,33 @@ cmd_run_shell_callback(struct job *job)
|
||||
{
|
||||
struct cmd_run_shell_data *cdata = job->data;
|
||||
struct cmd_ctx *ctx = &cdata->ctx;
|
||||
char *cmd, *msg, *line, *buf;
|
||||
size_t off, len, llen;
|
||||
char *cmd, *msg, *line;
|
||||
size_t size;
|
||||
int retcode;
|
||||
u_int lines;
|
||||
|
||||
buf = BUFFER_OUT(job->out);
|
||||
len = BUFFER_USED(job->out);
|
||||
lines = 0;
|
||||
do {
|
||||
if ((line = evbuffer_readline(job->event->input)) != NULL) {
|
||||
ctx->print(ctx, "%s", line);
|
||||
lines++;
|
||||
}
|
||||
} while (line != NULL);
|
||||
|
||||
size = EVBUFFER_LENGTH(job->event->input);
|
||||
if (size != 0) {
|
||||
line = xmalloc(size + 1);
|
||||
memcpy(line, EVBUFFER_DATA(job->event->input), size);
|
||||
line[size] = '\0';
|
||||
|
||||
ctx->print(ctx, "%s", line);
|
||||
lines++;
|
||||
|
||||
xfree(line);
|
||||
}
|
||||
|
||||
cmd = cdata->cmd;
|
||||
|
||||
if (len != 0) {
|
||||
line = buf;
|
||||
for (off = 0; off < len; off++) {
|
||||
if (buf[off] == '\n') {
|
||||
llen = buf + off - line;
|
||||
if (llen > INT_MAX)
|
||||
break;
|
||||
ctx->print(ctx, "%.*s", (int) llen, line);
|
||||
line = buf + off + 1;
|
||||
}
|
||||
}
|
||||
llen = buf + len - line;
|
||||
if (llen > 0 && llen < INT_MAX)
|
||||
ctx->print(ctx, "%.*s", (int) llen, line);
|
||||
}
|
||||
|
||||
msg = NULL;
|
||||
if (WIFEXITED(job->status)) {
|
||||
if ((retcode = WEXITSTATUS(job->status)) != 0)
|
||||
@ -111,7 +113,7 @@ cmd_run_shell_callback(struct job *job)
|
||||
xasprintf(&msg, "'%s' terminated by signal %d", cmd, retcode);
|
||||
}
|
||||
if (msg != NULL) {
|
||||
if (len != 0)
|
||||
if (lines != 0)
|
||||
ctx->print(ctx, "%s", msg);
|
||||
else
|
||||
ctx->info(ctx, "%s", msg);
|
||||
|
51
job.c
51
job.c
@ -17,6 +17,7 @@
|
||||
*/
|
||||
|
||||
#include <sys/types.h>
|
||||
#include <sys/socket.h>
|
||||
|
||||
#include <fcntl.h>
|
||||
#include <paths.h>
|
||||
@ -35,6 +36,8 @@ struct joblist all_jobs = SLIST_HEAD_INITIALIZER(&all_jobs);
|
||||
|
||||
RB_GENERATE(jobs, job, entry, job_cmp);
|
||||
|
||||
void job_callback(struct bufferevent *, short, void *);
|
||||
|
||||
int
|
||||
job_cmp(struct job *job1, struct job *job2)
|
||||
{
|
||||
@ -86,14 +89,13 @@ job_add(struct jobs *jobs, int flags, struct client *c, const char *cmd,
|
||||
job->client = c;
|
||||
|
||||
job->fd = -1;
|
||||
job->out = buffer_create(BUFSIZ);
|
||||
memset(&job->event, 0, sizeof job->event);
|
||||
job->event = NULL;
|
||||
|
||||
job->callbackfn = callbackfn;
|
||||
job->freefn = freefn;
|
||||
job->data = data;
|
||||
|
||||
job->flags = flags|JOB_DONE;
|
||||
job->flags = flags;
|
||||
|
||||
if (jobs != NULL)
|
||||
RB_INSERT(jobs, jobs, job);
|
||||
@ -125,9 +127,9 @@ job_free(struct job *job)
|
||||
|
||||
if (job->fd != -1)
|
||||
close(job->fd);
|
||||
if (job->out != NULL)
|
||||
buffer_destroy(job->out);
|
||||
event_del(&job->event);
|
||||
|
||||
if (job->event != NULL)
|
||||
bufferevent_free(job->event);
|
||||
|
||||
xfree(job);
|
||||
}
|
||||
@ -138,11 +140,10 @@ job_run(struct job *job)
|
||||
{
|
||||
int nullfd, out[2], mode;
|
||||
|
||||
if (!(job->flags & JOB_DONE))
|
||||
if (job->fd != -1 || job->pid != -1)
|
||||
return (0);
|
||||
job->flags &= ~JOB_DONE;
|
||||
|
||||
if (pipe(out) != 0)
|
||||
if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, out) != 0)
|
||||
return (-1);
|
||||
|
||||
switch (job->pid = fork()) {
|
||||
@ -181,13 +182,41 @@ job_run(struct job *job)
|
||||
if (fcntl(job->fd, F_SETFD, FD_CLOEXEC) == -1)
|
||||
fatal("fcntl failed");
|
||||
|
||||
if (BUFFER_USED(job->out) != 0)
|
||||
buffer_remove(job->out, BUFFER_USED(job->out));
|
||||
if (job->event != NULL)
|
||||
bufferevent_free(job->event);
|
||||
job->event =
|
||||
bufferevent_new(job->fd, NULL, NULL, job_callback, job);
|
||||
bufferevent_enable(job->event, EV_READ);
|
||||
|
||||
return (0);
|
||||
}
|
||||
}
|
||||
|
||||
/* Job buffer error callback. */
|
||||
void
|
||||
job_callback(unused struct bufferevent *bufev, unused short events, void *data)
|
||||
{
|
||||
struct job *job = data;
|
||||
|
||||
bufferevent_disable(job->event, EV_READ);
|
||||
close(job->fd);
|
||||
job->fd = -1;
|
||||
|
||||
if (job->pid == -1 && job->callbackfn != NULL)
|
||||
job->callbackfn(job);
|
||||
}
|
||||
|
||||
/* Job died (waitpid() returned its pid). */
|
||||
void
|
||||
job_died(struct job *job, int status)
|
||||
{
|
||||
job->status = status;
|
||||
job->pid = -1;
|
||||
|
||||
if (job->fd == -1 && job->callbackfn != NULL)
|
||||
job->callbackfn(job);
|
||||
}
|
||||
|
||||
/* Kill a job. */
|
||||
void
|
||||
job_kill(struct job *job)
|
||||
|
77
server-job.c
77
server-job.c
@ -1,77 +0,0 @@
|
||||
/* $OpenBSD$ */
|
||||
|
||||
/*
|
||||
* Copyright (c) 2009 Nicholas Marriott <nicm@users.sourceforge.net>
|
||||
*
|
||||
* Permission to use, copy, modify, and distribute this software for any
|
||||
* purpose with or without fee is hereby granted, provided that the above
|
||||
* copyright notice and this permission notice appear in all copies.
|
||||
*
|
||||
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
||||
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
||||
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
||||
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
||||
* WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER
|
||||
* IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
|
||||
* OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
||||
*/
|
||||
|
||||
#include <sys/types.h>
|
||||
|
||||
#include <event.h>
|
||||
#include <unistd.h>
|
||||
|
||||
#include "tmux.h"
|
||||
|
||||
/* Register jobs for poll. */
|
||||
void
|
||||
server_job_prepare(void)
|
||||
{
|
||||
struct job *job;
|
||||
|
||||
SLIST_FOREACH(job, &all_jobs, lentry) {
|
||||
if (job->fd == -1)
|
||||
continue;
|
||||
event_del(&job->event);
|
||||
event_set(
|
||||
&job->event, job->fd, EV_READ, server_job_callback, job);
|
||||
event_add(&job->event, NULL);
|
||||
}
|
||||
}
|
||||
|
||||
/* Process a single job event. */
|
||||
void
|
||||
server_job_callback(int fd, short events, void *data)
|
||||
{
|
||||
struct job *job = data;
|
||||
|
||||
if (job->fd == -1)
|
||||
return;
|
||||
|
||||
if (buffer_poll(fd, events, job->out, NULL) != 0) {
|
||||
close(job->fd);
|
||||
job->fd = -1;
|
||||
}
|
||||
}
|
||||
|
||||
/* Job functions that happen once a loop. */
|
||||
void
|
||||
server_job_loop(void)
|
||||
{
|
||||
struct job *job;
|
||||
|
||||
restart:
|
||||
SLIST_FOREACH(job, &all_jobs, lentry) {
|
||||
if (job->flags & JOB_DONE || job->fd != -1 || job->pid != -1)
|
||||
continue;
|
||||
job->flags |= JOB_DONE;
|
||||
|
||||
if (job->callbackfn != NULL) {
|
||||
job->callbackfn(job);
|
||||
if ((!job->flags & JOB_PERSIST)) {
|
||||
job_free(job);
|
||||
goto restart;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
6
server.c
6
server.c
@ -214,14 +214,12 @@ server_loop(void)
|
||||
while (!server_should_shutdown()) {
|
||||
server_update_socket();
|
||||
|
||||
server_job_prepare();
|
||||
server_window_prepare();
|
||||
server_client_prepare();
|
||||
|
||||
event_loopexit(&tv);
|
||||
event_loop(EVLOOP_ONCE);
|
||||
|
||||
server_job_loop();
|
||||
server_window_loop();
|
||||
server_client_loop();
|
||||
|
||||
@ -470,8 +468,8 @@ server_child_exited(pid_t pid, int status)
|
||||
|
||||
SLIST_FOREACH(job, &all_jobs, lentry) {
|
||||
if (pid == job->pid) {
|
||||
job->pid = -1;
|
||||
job->status = status;
|
||||
job_died(job, status); /* might free job */
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
13
status.c
13
status.c
@ -486,22 +486,25 @@ status_job(struct client *c, char **iptr)
|
||||
void
|
||||
status_job_callback(struct job *job)
|
||||
{
|
||||
char *buf;
|
||||
char *line, *buf;
|
||||
size_t len;
|
||||
|
||||
len = BUFFER_USED(job->out);
|
||||
buf = NULL;
|
||||
if ((line = evbuffer_readline(job->event->input)) == NULL) {
|
||||
len = EVBUFFER_LENGTH(job->event->input);
|
||||
buf = xmalloc(len + 1);
|
||||
if (len != 0)
|
||||
buffer_read(job->out, buf, len);
|
||||
memcpy(buf, EVBUFFER_DATA(job->event->input), len);
|
||||
buf[len] = '\0';
|
||||
buf[strcspn(buf, "\n")] = '\0';
|
||||
}
|
||||
|
||||
if (job->data != NULL)
|
||||
xfree(job->data);
|
||||
else
|
||||
server_redraw_client(job->client);
|
||||
job->data = xstrdup(buf);
|
||||
job->data = xstrdup(line);
|
||||
|
||||
if (buf != NULL)
|
||||
xfree(buf);
|
||||
}
|
||||
|
||||
|
12
tmux.h
12
tmux.h
@ -661,16 +661,14 @@ struct job {
|
||||
struct client *client;
|
||||
|
||||
int fd;
|
||||
struct event event;
|
||||
struct buffer *out;
|
||||
struct bufferevent *event;
|
||||
|
||||
void (*callbackfn)(struct job *);
|
||||
void (*freefn)(void *);
|
||||
void *data;
|
||||
|
||||
int flags;
|
||||
#define JOB_DONE 0x1
|
||||
#define JOB_PERSIST 0x2 /* don't free after callback */
|
||||
#define JOB_PERSIST 0x1 /* don't free after callback */
|
||||
|
||||
RB_ENTRY(job) entry;
|
||||
SLIST_ENTRY(job) lentry;
|
||||
@ -1306,6 +1304,7 @@ struct job *job_add(struct jobs *, int, struct client *,
|
||||
void job_remove(struct jobs *, struct job *);
|
||||
void job_free(struct job *);
|
||||
int job_run(struct job *);
|
||||
void job_died(struct job *, int);
|
||||
void job_kill(struct job *);
|
||||
|
||||
/* environ.c */
|
||||
@ -1589,11 +1588,6 @@ void server_client_prepare(void);
|
||||
void server_client_callback(int, short, void *);
|
||||
void server_client_loop(void);
|
||||
|
||||
/* server-job.c */
|
||||
void server_job_prepare(void);
|
||||
void server_job_callback(int, short, void *);
|
||||
void server_job_loop(void);
|
||||
|
||||
/* server-window.c */
|
||||
void server_window_prepare(void);
|
||||
void server_window_callback(int, short, void *);
|
||||
|
Loading…
Reference in New Issue
Block a user