diff options
-rw-r--r-- | liberty-proto.c | 2 | ||||
-rw-r--r-- | tools/wdye/CMakeLists.txt | 4 | ||||
-rw-r--r-- | tools/wdye/test.lua | 2 | ||||
-rw-r--r-- | tools/wdye/wdye.adoc | 6 | ||||
-rw-r--r-- | tools/wdye/wdye.c | 56 |
5 files changed, 42 insertions, 28 deletions
diff --git a/liberty-proto.c b/liberty-proto.c index d4355c7..fb0fc36 100644 --- a/liberty-proto.c +++ b/liberty-proto.c @@ -1594,6 +1594,8 @@ mpd_client_parse_kv (char *line, char **value) static void mpd_client_update_poller (struct mpd_client *self) { + if (self->state != MPD_CONNECTED) + return; poller_fd_set (&self->socket_event, self->write_buffer.len ? (POLLIN | POLLOUT) : POLLIN); } diff --git a/tools/wdye/CMakeLists.txt b/tools/wdye/CMakeLists.txt index a69f7ee..7dbf09e 100644 --- a/tools/wdye/CMakeLists.txt +++ b/tools/wdye/CMakeLists.txt @@ -22,7 +22,7 @@ option (WITH_CURSES "Offer terminal sequences using Curses" "${CURSES_FOUND}") # -liconv may or may not be a part of libc find_path (iconv_INCLUDE_DIRS iconv.h) -include_directories ("${PROJECT_BINARY_DIR}" ${iconv_INCLUDE_DIRS}) +include_directories (BEFORE "${PROJECT_BINARY_DIR}" ${iconv_INCLUDE_DIRS}) file (CONFIGURE OUTPUT "${PROJECT_BINARY_DIR}/config.h" CONTENT [[ #define PROGRAM_NAME "${PROJECT_NAME}" #define PROGRAM_VERSION "${PROJECT_VERSION}" @@ -38,5 +38,5 @@ if (WITH_CURSES) target_link_libraries (wdye PUBLIC ${CURSES_LIBRARIES}) endif () -add_test (NAME simple COMMAND wdye "${PROJECT_SOURCE_DIR}/test.lua") +add_test (NAME wdye COMMAND wdye "${PROJECT_SOURCE_DIR}/test.lua") include (CTest) diff --git a/tools/wdye/test.lua b/tools/wdye/test.lua index c255c72..add8488 100644 --- a/tools/wdye/test.lua +++ b/tools/wdye/test.lua @@ -25,7 +25,9 @@ cat:send("Closing...\r"):send("\004") local v = expect(cat:eof {true}, cat:default {.5, function (p) error "expected EOF, got a timeout" end}) +assert(cat.pid > 0, "process has no ID") local s1, exit, signal = cat:wait () assert(s1 == 0 and exit == 0 and not signal, "unexpected exit status") +assert(cat.pid < 0, "process still has an ID") local s2 = cat:wait (true) assert(s1 == s2, "exit status not remembered") diff --git a/tools/wdye/wdye.adoc b/tools/wdye/wdye.adoc index 98e251e..2609849 100644 --- a/tools/wdye/wdye.adoc +++ b/tools/wdye/wdye.adoc @@ -31,7 +31,7 @@ The *env* map may be used to override environment variables, notably _TERM_. Variables evaluating to _false_ will be removed from the environment. The program's whole process group receives SIGKILL when the _process_ -is garbage-collected. +is garbage-collected, unless *wait* has collected the process group leader. wdye.expect ([pattern1, ...]) ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ @@ -57,6 +57,10 @@ PROCESS.buffer ~~~~~~~~~~~~~~ A string with the _process_' current read buffer contents. +PROCESS.pid +~~~~~~~~~~~ +An integer with the _process_' process ID, or -1 if *wait* has collected it. + PROCESS.term ~~~~~~~~~~~~ A table with the _process_' *terminfo*(5) capabilities, diff --git a/tools/wdye/wdye.c b/tools/wdye/wdye.c index b7fccd6..44dbae0 100644 --- a/tools/wdye/wdye.c +++ b/tools/wdye/wdye.c @@ -175,7 +175,7 @@ pty_fork (int *ptrfdm, char **slave_name, char *pts_name = NULL; if ((fdm = ptym_open (&pts_name)) < 0) { - error_set (e, "can’t open master pty: %s", strerror (errno)); + error_set (e, "can't open master pty: %s", strerror (errno)); return -1; } if (slave_name != NULL) @@ -194,7 +194,7 @@ pty_fork (int *ptrfdm, char **slave_name, if (setsid () < 0) exit_fatal ("setsid: %s", strerror (errno)); if ((fds = ptys_open (pts_name)) < 0) - exit_fatal ("can’t open slave pty: %s", strerror (errno)); + exit_fatal ("can't open slave pty: %s", strerror (errno)); xclose (fdm); #if defined BSD @@ -401,7 +401,7 @@ xlua_pattern_index (lua_State *L) if (!strcmp (key, "process")) lua_rawgeti (L, LUA_REGISTRYINDEX, self->ref_process); else - return luaL_argerror (L, 2, "not a readable property"); + return luaL_error (L, "not a readable property: %s", key); return 1; } @@ -411,8 +411,8 @@ xlua_pattern_index (lua_State *L) case PATTERN_REGEX: { const regmatch_t *m = self->matches + group; - if (group < 0 || group > self->regex->re_nsub - || m->rm_so < 0 || m->rm_eo < 0 || m->rm_eo > self->input.len) + if (group < 0 || (size_t) group > self->regex->re_nsub + || m->rm_so < 0 || m->rm_eo < 0 || (size_t) m->rm_eo > self->input.len) lua_pushnil (L); else lua_pushlstring (L, @@ -428,7 +428,7 @@ xlua_pattern_index (lua_State *L) lua_pushlstring (L, self->input.str, self->input.len); return 1; default: - return luaL_argerror (L, 2, "indexing unavailable for this pattern"); + return luaL_argerror (L, 1, "indexing unavailable for this pattern"); } } @@ -519,10 +519,12 @@ xlua_process_index (lua_State *L) if (!strcmp (key, "buffer")) lua_pushlstring (L, self->buffer.str, self->buffer.len); + else if (!strcmp (key, "pid")) + lua_pushinteger (L, self->pid); else if (!strcmp (key, "term")) lua_rawgeti (L, LUA_REGISTRYINDEX, self->ref_term); else - return luaL_argerror (L, 2, "not a readable property"); + return luaL_error (L, "not a readable property: %s", key); return 1; } @@ -542,7 +544,7 @@ xlua_process_send (lua_State *L) ssize_t written = write (self->terminal_fd, arg, len); if (written == -1) return luaL_error (L, "write failed: %s", strerror (errno)); - else if (written != len) + else if (written != (ssize_t) len) return luaL_error (L, "write failed: %s", "short write"); if (self->asciicast) @@ -560,7 +562,7 @@ xlua_process_send (lua_State *L) static int xlua_process_regex (lua_State *L) { - struct process *self = luaL_checkudata (L, 1, XLUA_PROCESS_METATABLE); + (void) luaL_checkudata (L, 1, XLUA_PROCESS_METATABLE); luaL_checktype (L, 2, LUA_TTABLE); if (lua_gettop (L) != 2) return luaL_error (L, "too many arguments"); @@ -593,7 +595,7 @@ xlua_process_regex (lua_State *L) static int xlua_process_exact (lua_State *L) { - struct process *self = luaL_checkudata (L, 1, XLUA_PROCESS_METATABLE); + (void) luaL_checkudata (L, 1, XLUA_PROCESS_METATABLE); luaL_checktype (L, 2, LUA_TTABLE); if (lua_gettop (L) != 2) return luaL_error (L, "too many arguments"); @@ -618,7 +620,7 @@ xlua_process_exact (lua_State *L) static int xlua_process_eof (lua_State *L) { - struct process *self = luaL_checkudata (L, 1, XLUA_PROCESS_METATABLE); + (void) luaL_checkudata (L, 1, XLUA_PROCESS_METATABLE); luaL_checktype (L, 2, LUA_TTABLE); if (lua_gettop (L) != 2) return luaL_error (L, "too many arguments"); @@ -634,7 +636,7 @@ xlua_process_eof (lua_State *L) static int xlua_process_default (lua_State *L) { - struct process *self = luaL_checkudata (L, 1, XLUA_PROCESS_METATABLE); + (void) luaL_checkudata (L, 1, XLUA_PROCESS_METATABLE); luaL_checktype (L, 2, LUA_TTABLE); if (lua_gettop (L) != 2) return luaL_error (L, "too many arguments"); @@ -1081,7 +1083,7 @@ expect_prepare_pattern (struct expect_context *ctx, struct pattern *p) { str_reset (&p->input); if (p->kind == PATTERN_REGEX) - for (int i = 0; i <= p->regex->re_nsub; i++) + for (size_t i = 0; i <= p->regex->re_nsub; i++) p->matches[i] = (regmatch_t) { .rm_so = -1, .rm_eo = -1 }; if (p->kind == PATTERN_REGEX @@ -1168,7 +1170,7 @@ pattern_match (struct pattern *self) if (regexec (self->regex, buffer->str, self->regex->re_nsub + 1, self->matches, flags)) { - for (int i = 0; i <= self->regex->re_nsub; i++) + for (size_t i = 0; i <= self->regex->re_nsub; i++) self->matches[i] = (regmatch_t) { .rm_so = -1, .rm_eo = -1 }; return false; } @@ -1374,9 +1376,9 @@ xlua_panic (lua_State *L) int main (int argc, char *argv[]) { - if (argc != 2) + if (argc < 2) { - fprintf (stderr, "Usage: %s program.lua\n", argv[0]); + fprintf (stderr, "Usage: %s program.lua [args...]\n", argv[0]); return 1; } @@ -1397,16 +1399,20 @@ main (int argc, char *argv[]) luaL_setfuncs (g.L, xlua_pattern_table, 0); lua_pop (g.L, 1); - const char *path = argv[1]; + (void) lua_checkstack (g.L, argc); + lua_pushcfunction (g.L, xlua_error_handler); - if (luaL_loadfile (g.L, path) - || lua_pcall (g.L, 0, 0, -2)) - { - print_error ("%s", lua_tostring (g.L, -1)); - lua_pop (g.L, 1); - lua_close (g.L); - return 1; - } + if (luaL_loadfile (g.L, strcmp (argv[1], "-") ? argv[1] : NULL)) + goto error; + for (int i = 2; i < argc; i++) + lua_pushstring (g.L, argv[i]); + if (lua_pcall (g.L, argc - 2, 0, 1)) + goto error; lua_close (g.L); return 0; + +error: + print_error ("%s", lua_tostring (g.L, -1)); + lua_close (g.L); + return 1; } |