X-Git-Url: https://git.cworth.org/git?p=ttt;a=blobdiff_plain;f=src%2Fx.c;h=c35f4c88955f3147271d86c1c3113140aeafb496;hp=1695374c8c72a3c6423a69b69d15728c575055ee;hb=95898262b4ce4a2a3d36f70a4e6cc8188decc142;hpb=2cd70db8433cc9d02a4ca784190260889c835198 diff --git a/src/x.c b/src/x.c index 1695374..c35f4c8 100644 --- a/src/x.c +++ b/src/x.c @@ -19,9 +19,7 @@ * Author: Carl Worth */ -#include "ttt.h" - -#include +#include "x.h" void xasprintf (char **strp, const char *fmt, ...) @@ -119,6 +117,22 @@ xrealloc (void *ptr, size_t size) return ret; } +FILE * +xfopen (const char *path, const char *mode) +{ + FILE *ret; + + ret = fopen (path, mode); + + if (ret == NULL) { + fprintf (stderr, "Error: fopen of %s failed: %s. Aborting.\n", + path, strerror (errno)); + exit (1); + } + + return ret; +} + FILE * xfdopen (int filedes, const char *mode) { @@ -127,18 +141,35 @@ xfdopen (int filedes, const char *mode) ret = fdopen (filedes, mode); if (ret == NULL) { - fprintf (stderr, "Error: fdopen failed: %s. Aborting.\n", strerror (errno)); + fprintf (stderr, "Error: fdopen failed: %s. Aborting.\n", + strerror (errno)); exit (1); } return ret; } +void +xdup2 (int oldfd, int newfd) +{ + int ret; + + ret = dup2 (oldfd, newfd); + if (ret == -1) { + printf ("Error: dup2 failed: %s. Aborting.\n", + strerror (errno)); + exit (1); + } +} + char * xstrdup (const char *s) { char *ret; + if (s == NULL) + return NULL; + ret = strdup (s); if (ret == NULL) { @@ -190,6 +221,29 @@ xbind (int sockfd, const struct sockaddr *my_addr, socklen_t addrlen) } } +ttt_status_t +xconnect (int sockfd, const struct sockaddr *serv_addr, socklen_t addrlen) +{ + int ret; + + ret = connect (sockfd, serv_addr, addrlen); + if (ret == -1) { + switch (errno) { + case ECONNREFUSED: + return TTT_STATUS_CONNECTION_REFUSED; + case EHOSTUNREACH: + case ENETUNREACH: + return TTT_STATUS_NETWORK_UNREACHABLE; + default: + fprintf (stderr, "Error: connect failed (errno = %d): %s. Aborting.\n", + errno, strerror (errno)); + exit (1); + } + } + + return TTT_STATUS_SUCCESS; +} + void xlisten (int s, int backlog) { @@ -236,3 +290,33 @@ xselect (int n, return ret; } + +ssize_t +xread (int fd, void *buf, size_t count) +{ + int ret; + + ret = read (fd, buf, count); + if (ret == -1) { + fprintf (stderr, "Error: read failed: %s. Aborting.\n", + strerror (errno)); + exit (1); + } + + return ret; +} + +ssize_t +xwrite (int fd, const void *buf, size_t count) +{ + int ret; + + ret = write (fd, buf, count); + if (ret == -1) { + fprintf (stderr, "Error: write failed: %s. Aborting.\n", + strerror (errno)); + exit (1); + } + + return ret; +}