1 files changed, 34 insertions, 37 deletions

diff --git a/src/handlers.c b/src/handlers.c
index 10599a3..fdd16f8 100644
--- a/src/handlers.c
+++ b/src/handlers.c
@@ -6,6 +6,7 @@
 #include <stdlib.h>
 #include <stdio.h>
 #include <sys/socket.h>
+#include <arpa/inet.h>
 #include <string.h>
 #include <syslog.h>
 #include <limits.h>
@@ -19,21 +20,13 @@
 #include "packet.h"
 #include "netascii.h"
 
-typedef enum opmode {
-	NOT_FOUND,
-	NETASCII,
-	OCTAL,
-	MAIL
-// any additional modes
-} nbd_opmode;
-
 nbd_opmode get_mode(char *mode) {
 	nbd_opmode opmode;
-	if (strncmp((const char *)mode, (const char *)&"netascii", NBD_NBTPD_ARGS_MODE_MAX) == 0) {
+	if (strncmp(mode, (char *)&"netascii", NBD_NBTPD_ARGS_MODE_MAX) == 0) {
 		opmode = NETASCII;
-	} else if (strncmp((const char *)mode, (const char *)&"octal", NBD_NBTPD_ARGS_MODE_MAX) == 0) {
+	} else if (strncmp(mode, (char *)&"octal", NBD_NBTPD_ARGS_MODE_MAX) == 0) {
 		opmode = OCTAL;
-	} else if (strncmp((const char *)mode, (const char *)&"mail", NBD_NBTPD_ARGS_MODE_MAX) == 0) {
+	} else if (strncmp(mode, (char *)&"mail", NBD_NBTPD_ARGS_MODE_MAX) == 0) {
 		opmode = MAIL;
 	} else {
 		opmode = NOT_FOUND;
@@ -47,57 +40,51 @@ void *read_req_resp(void *args) {
 	nbd_nbtpd_args *argptr = (nbd_nbtpd_args *)args;
 	if (!is_netascii_str((char *)&argptr->path)) {
 		argptr->err = 1;
-		goto nbd_nbtpd_thread_read_cleanup_ps;
+		goto pre_socket;
 	}
 	fname = realpath((char *)&argptr->path, NULL);
 	if (fname == NULL) {
 		syslog(LOG_ERR, "unable to get real path: %s", strerror(errno));
 		argptr->err = 1;
-		goto nbd_nbtpd_thread_read_cleanup_ps;
+		goto pre_socket;
 	}
-	/*
-	 * POSIX does not define what happens when getcwd() is given NULL.
-	 * This means that illumos libc and GNU libc both return a newly
-	 * malloc()'d string, while FreeBSD libc does not.  Also, FreeBSD uses
-	 * constant MAXPATHLEN defined in <sys/param.h> rather than PATH_MAX.
-	 */
+/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
+ * POSIX does not define what happens when getcwd() is given NULL.     *
+ * This means that illumos libc and GNU libc both return a newly       *
+ * malloc()'d string, while FreeBSD libc does not.  Also, FreeBSD uses *
+ * constant MAXPATHLEN defined in <sys/param.h> rather than PATH_MAX.  *
+ * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
 #ifdef __FreeBSD__
 	wd = malloc(MAXPATHLEN);
 	if (wd == NULL) {
 		syslog(LOG_ERR, "unable to allocate PATH_MAX memory: %s", strerror(errno));
 		argptr->err = 0;
-		goto nbd_nbtpd_thread_read_cleanup_ps;
+		goto pre_socket;
 	}
 	if (getcwd(wd, MAXPATHLEN) == NULL) {
 		argptr->err = 0;
-		goto nbd_nbtpd_thread_read_cleanup_ps;
+		goto pre_socket;
 	}
 #else
 	if ((wd = getcwd(NULL, PATH_MAX)) == NULL) {
 		argptr->err = 0;
-		goto nbd_nbtpd_thread_read_cleanup_ps;
+		goto pre_socket;
 	}
 #endif
 	if (!strncmp(wd, fname, sizeof(wd) - 1)) {
 		argptr->err = 2;
-		goto nbd_nbtpd_thread_read_cleanup_ps;
+		goto pre_socket;
 	}
 	if (!is_netascii_str((char *)&argptr->mode)) {
 		argptr->err = 4;
-		goto nbd_nbtpd_thread_read_cleanup_ps;
+		goto pre_socket;
 	}
 	nbd_opmode opmode = get_mode((char *)&argptr->mode);
-	switch (opmode) {
-		case NETASCII:
-			break;
-		case OCTAL:
-			break;
-		default:
-			argptr->err = 4;
-			goto nbd_nbtpd_thread_read_cleanup_ps;
+	if ((opmode != NETASCII) && (opmode != OCTAL)) {
+		argptr->err = 4;
+		goto pre_socket;
 	}
 	fp = fopen(fname, "r");
-
 	buf = malloc(512);
 	if (buf == NULL) {
 		syslog(
@@ -106,24 +93,34 @@ void *read_req_resp(void *args) {
 			strerror(errno)
 		);
 		argptr->err = 0;
-		goto nbd_nbtpd_thread_read_cleanup_ps;
+		goto pre_socket;
 	}
 	memset(buf, '\0', 512);
 	int s = socket(AF_INET, SOCK_DGRAM, 0);
 	if (s <= 0) {
 		syslog(LOG_ERR, "unable to define socket: %s", strerror(errno));
-		goto nbd_nbtpd_thread_read_cleanup;
+		goto cleanup;
+	}
+	if (connect(s, (struct sockaddr *)&argptr->client, sizeof(struct sockaddr)) < 0) {
+		syslog(
+			LOG_ERR,
+			"unable to connect to client %s: %s",
+			inet_ntoa(argptr->client.sin_addr),
+			strerror(errno)
+		);
+		goto socket_clean;
 	}
 
+socket_clean:
 	close(s);
-nbd_nbtpd_thread_read_cleanup:
+cleanup:
 	fclose(fp);
 	free(buf);
 	free(fname);
 	free(wd);
 	free(args);
 	return (void *)NULL;
-nbd_nbtpd_thread_read_cleanup_ps:
+pre_socket:
 	if (fp != NULL) {
 		fclose(fp);
 	}