From f77fae1335c562b720c4a6d4aa710f8627eaaff6 Mon Sep 17 00:00:00 2001 From: Curtis McEnroe Date: Wed, 6 Sep 2017 14:01:48 -0400 Subject: Move C code to bin, Makefile, AGPL --- bin/hnel.c | 127 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 127 insertions(+) create mode 100644 bin/hnel.c (limited to 'bin/hnel.c') diff --git a/bin/hnel.c b/bin/hnel.c new file mode 100644 index 00000000..1f079e8a --- /dev/null +++ b/bin/hnel.c @@ -0,0 +1,127 @@ +/* PTY wrapper for preserving HJKL in Tarmak layouts. + * + * Copyright (c) 2017, Curtis McEnroe + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU Affero General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU Affero General Public License for more details. + * + * You should have received a copy of the GNU Affero General Public License + * along with this program. If not, see . + */ + +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#if defined __FreeBSD__ +#include +#elif defined __linux__ +#include +#else +#include +#endif + +static char table[256] = { + ['n'] = 'j', ['N'] = 'J', [CTRL('N')] = CTRL('J'), + ['e'] = 'k', ['E'] = 'K', [CTRL('E')] = CTRL('K'), + ['j'] = 'e', ['J'] = 'E', [CTRL('J')] = CTRL('E'), + ['k'] = 'n', ['K'] = 'N', [CTRL('K')] = CTRL('N'), +}; + +static ssize_t writeAll(int fd, const char *buf, size_t len) { + ssize_t writeLen; + while (0 < (writeLen = write(fd, buf, len))) { + buf += writeLen; + len -= writeLen; + } + return writeLen; +} + +static struct termios saveTerm; + +static void restoreTerm(void) { + tcsetattr(STDERR_FILENO, TCSADRAIN, &saveTerm); +} + +int main(int argc, char *argv[]) { + int error; + + if (argc < 2) return EX_USAGE; + + error = tcgetattr(STDERR_FILENO, &saveTerm); + if (error) err(EX_IOERR, "tcgetattr"); + atexit(restoreTerm); + + struct termios raw; + cfmakeraw(&raw); + error = tcsetattr(STDERR_FILENO, TCSADRAIN, &raw); + if (error) err(EX_IOERR, "tcsetattr"); + + struct winsize window; + error = ioctl(STDERR_FILENO, TIOCGWINSZ, &window); + if (error) err(EX_IOERR, "ioctl(%d, TIOCGWINSZ)", STDERR_FILENO); + + int master; + pid_t pid = forkpty(&master, NULL, NULL, &window); + if (pid < 0) err(EX_OSERR, "forkpty"); + + if (!pid) { + execvp(argv[1], argv + 1); + err(EX_OSERR, "%s", argv[1]); + } + + bool enable = true; + + struct pollfd fds[2] = { + { .fd = STDIN_FILENO, .events = POLLIN }, + { .fd = master, .events = POLLIN }, + }; + while (0 < poll(fds, 2, -1)) { + char buf[4096]; + ssize_t len; + + if (fds[0].revents) { + len = read(STDIN_FILENO, buf, sizeof(buf)); + if (len < 0) err(EX_IOERR, "read(%d)", STDIN_FILENO); + + if (len == 1) { + if (buf[0] == CTRL('S')) { + enable = !enable; + continue; + } + + unsigned char c = buf[0]; + if (enable && table[c]) buf[0] = table[c]; + } + + len = writeAll(master, buf, len); + if (len < 0) err(EX_IOERR, "write(%d)", master); + } + + if (fds[1].revents) { + len = read(master, buf, sizeof(buf)); + if (len < 0) err(EX_IOERR, "read(%d)", master); + len = writeAll(STDOUT_FILENO, buf, len); + if (len < 0) err(EX_IOERR, "write(%d)", STDOUT_FILENO); + } + + int status; + pid_t dead = waitpid(pid, &status, WNOHANG); + if (dead < 0) err(EX_OSERR, "waitpid(%d)", pid); + if (dead) return WIFEXITED(status) ? WEXITSTATUS(status) : EX_SOFTWARE; + } + err(EX_IOERR, "poll"); +} -- cgit 1.4.1 >June McEnroe 2019-07-11Don't do carriage return on line feedJune McEnroe 2019-07-11Interpret 256color-style SGRsJune McEnroe 2019-07-11Use inline style rather than <b>, <i>, <u>June McEnroe 2019-07-11Factor out clearJune McEnroe 2019-07-11Add bright option to shottyJune McEnroe 2019-07-11Output <b>, <i>, <u> in shottyJune McEnroe 2019-07-10Ignore SM and RMJune McEnroe 2019-07-09Add shotty man page and build itJune McEnroe 2019-07-09Add up -cJune McEnroe 2019-07-09Add options for default colors to shottyJune McEnroe 2019-07-08Use char literals consistentlyJune McEnroe