From ee442870809ebfd9eefd8868314cafa5c001d925 Mon Sep 17 00:00:00 2001 From: Curtis McEnroe Date: Fri, 6 Jan 2017 11:49:29 -0500 Subject: Implement pbcopy and pbpaste in C Ted Unangst broke my netcat implementation of pbpaste with this commit: , which, when /dev/null is attached to stdin, causes nc to exit and never read from the socket. Turns out the core functionality of netcat can be implemented in about 50 lines of C. --- .bin/pbcopy.c | 52 ++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 52 insertions(+) create mode 100755 .bin/pbcopy.c (limited to '.bin/pbcopy.c') diff --git a/.bin/pbcopy.c b/.bin/pbcopy.c new file mode 100755 index 00000000..37c334ca --- /dev/null +++ b/.bin/pbcopy.c @@ -0,0 +1,52 @@ +#if 0 +cc -Wall -Wextra -pedantic $@ -o $(dirname $0)/pbcopy $0 && \ +exec cc -Wall -Wextra -pedantic -DPBPASTE $@ -o $(dirname $0)/pbpaste $0 +#endif + +#include +#include +#include +#include +#include +#include + +int main() { + int error; + + int client = socket(PF_INET, SOCK_STREAM, 0); + if (client < 0) err(EX_OSERR, "socket"); + + struct sockaddr_in addr = { + .sin_family = AF_INET, + .sin_port = htons(7062), + .sin_addr = { .s_addr = htonl(0x7f000001) }, + }; + + error = connect(client, (struct sockaddr *)&addr, sizeof(addr)); + if (error) err(EX_OSERR, "connect"); + +#ifdef PBPASTE + int fdIn = client; + int fdOut = STDOUT_FILENO; + error = shutdown(client, SHUT_WR); + if (error) err(EX_OSERR, "shutdown"); +#else + int fdIn = STDIN_FILENO; + int fdOut = client; +#endif + + char readBuf[4096]; + ssize_t readLen; + while (0 < (readLen = read(fdIn, readBuf, sizeof(readBuf)))) { + char *writeBuf = readBuf; + ssize_t writeLen; + while (0 < (writeLen = write(fdOut, writeBuf, readLen))) { + writeBuf += writeLen; + readLen -= writeLen; + } + if (writeLen < 0) err(EX_IOERR, "write"); + } + if (readLen < 0) err(EX_IOERR, "read"); + + return 0; +} -- cgit 1.4.1