X-Git-Url: https://git.tokkee.org/?p=sysdb.git;a=blobdiff_plain;f=src%2Futils%2Funixsock.c;h=1a4373abb4ee92bc38b0cb23982f36d93b61f490;hp=564aba70bab4c4a592e86da6bf4be93187f50513;hb=39a45905e0b237e458b1826ff9b4fad1c4a59550;hpb=bef7167f0dc1fd405e35d5cbffb3c0820945a9ea

diff --git a/src/utils/unixsock.c b/src/utils/unixsock.c
index 564aba7..1a4373a 100644
--- a/src/utils/unixsock.c
+++ b/src/utils/unixsock.c
@@ -25,8 +25,12 @@
  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
+#if HAVE_CONFIG_H
+#	include "config.h"
+#endif /* HAVE_CONFIG_H */
+
+#include "utils/error.h"
 #include "utils/unixsock.h"
-#include "utils/string.h"
 
 #include <assert.h>
 #include <errno.h>
@@ -113,11 +117,11 @@ sdb_unixsock_parse_cell(char *string, int type, sdb_data_t *data)
 		case SDB_TYPE_BINARY:
 			/* we don't support any binary information containing 0-bytes */
 			data->data.binary.length = strlen(string);
-			data->data.binary.datum = (const unsigned char *)string;
+			data->data.binary.datum = (unsigned char *)string;
 			break;
 		default:
-			fprintf(stderr, "unixsock: Unexpected type %i while "
-					"parsing query result.\n", type);
+			sdb_log(SDB_LOG_ERR, "unixsock: Unexpected type %i while "
+					"parsing query result.", type);
 			return -1;
 	}
 
@@ -125,14 +129,14 @@ sdb_unixsock_parse_cell(char *string, int type, sdb_data_t *data)
 			|| (type == SDB_TYPE_DATETIME)) {
 		if (errno || (string == endptr)) {
 			char errbuf[1024];
-			fprintf(stderr, "unixsock: Failed to parse string '%s' "
-					"as numeric value (type %i): %s\n", string, type,
+			sdb_log(SDB_LOG_ERR, "unixsock: Failed to parse string "
+					"'%s' as numeric value (type %i): %s", string, type,
 					sdb_strerror(errno, errbuf, sizeof(errbuf)));
 			return -1;
 		}
 		else if (endptr && (*endptr != '\0'))
-			fprintf(stderr, "unixsock: Ignoring garbage after number "
-					"while parsing numeric value (type %i): %s.\n",
+			sdb_log(SDB_LOG_WARNING, "unixsock: Ignoring garbage after "
+					"number while parsing numeric value (type %i): %s.",
 					type, endptr);
 	}
 
@@ -157,9 +161,10 @@ sdb_unixsock_client_process_one_line(sdb_unixsock_client_t *client,
 		char *next;
 
 		if (! line) { /* this must no happen */
-			fprintf(stderr, "unixsock: Unexpected EOL while parsing line "
-					"(expected %i columns delimited by '%s'; got %i): %s\n",
-					column_count, delim, /* last line number */ i, orig_line);
+			sdb_log(SDB_LOG_ERR, "unixsock: Unexpected EOL while "
+					"parsing line (expected %i columns delimited by '%s'; "
+					"got %i): %s", column_count, delim,
+					/* last line number */ i, orig_line);
 			return -1;
 		}
 
@@ -230,7 +235,7 @@ sdb_unixsock_client_connect(sdb_unixsock_client_t *client)
 	fd = socket(AF_UNIX, SOCK_STREAM, /* protocol = */ 0);
 	if (fd < 0) {
 		char errbuf[1024];
-		fprintf(stderr, "unixsock: Failed to open socket: %s\n",
+		sdb_log(SDB_LOG_ERR, "unixsock: Failed to open socket: %s",
 				sdb_strerror(errno, errbuf, sizeof(errbuf)));
 		return -1;
 	}
@@ -241,7 +246,7 @@ sdb_unixsock_client_connect(sdb_unixsock_client_t *client)
 
 	if (connect(fd, (struct sockaddr *)&sa, sizeof(sa))) {
 		char errbuf[1024];
-		fprintf(stderr, "unixsock: Failed to connect to %s: %s\n",
+		sdb_log(SDB_LOG_ERR, "unixsock: Failed to connect to %s: %s",
 				sa.sun_path, sdb_strerror(errno, errbuf, sizeof(errbuf)));
 		close(fd);
 		return -1;
@@ -250,12 +255,16 @@ sdb_unixsock_client_connect(sdb_unixsock_client_t *client)
 	client->fh = fdopen(fd, "r+");
 	if (! client->fh) {
 		char errbuf[1024];
-		fprintf(stderr, "unixsock: Failed to open I/O stream for %s: %s\n",
-				sa.sun_path, sdb_strerror(errno, errbuf, sizeof(errbuf)));
+		sdb_log(SDB_LOG_ERR, "unixsock: Failed to open I/O "
+				"stream for %s: %s", sa.sun_path,
+				sdb_strerror(errno, errbuf, sizeof(errbuf)));
 		close(fd);
 		return -1;
 	}
 
+	/* enable line-buffering */
+	setvbuf(client->fh, NULL, _IOLBF, 0);
+
 	client->shutdown = 0;
 	return 0;
 } /* sdb_unixsock_client_connect */
@@ -275,8 +284,9 @@ sdb_unixsock_client_send(sdb_unixsock_client_t *client,
 	status = fprintf(client->fh, "%s\r\n", msg);
 	if (status < 0) {
 		char errbuf[1024];
-		fprintf(stderr, "unixsock: Failed to write to socket (%s): %s\n",
-				client->path, sdb_strerror(errno, errbuf, sizeof(errbuf)));
+		sdb_log(SDB_LOG_ERR, "unixsock: Failed to write to "
+				"socket (%s): %s", client->path,
+				sdb_strerror(errno, errbuf, sizeof(errbuf)));
 		return status;
 	}
 	return status;
@@ -286,25 +296,35 @@ char *
 sdb_unixsock_client_recv(sdb_unixsock_client_t *client,
 		char *buffer, size_t buflen)
 {
+	char *tmp;
+
 	if ((! client) || (! client->fh) || (! buffer))
 		return NULL;
 
 	if (client->shutdown & SDB_SHUT_RD) /* reconnect */
 		sdb_unixsock_client_connect(client);
 
-	buffer = fgets(buffer, (int)buflen - 1, client->fh);
-	if (! buffer) {
-		if (! feof(client->fh)) {
-			char errbuf[1024];
-			fprintf(stderr, "unixsock: Failed to read from socket (%s): %s\n",
-					client->path, sdb_strerror(errno, errbuf, sizeof(errbuf)));
+	tmp = NULL;
+	while (tmp == NULL) {
+		errno = 0;
+		tmp = fgets(buffer, (int)buflen - 1, client->fh);
+		if (! tmp) {
+			if ((errno == EAGAIN) || (errno == EINTR))
+				continue;
+
+			if (! feof(client->fh)) {
+				char errbuf[1024];
+				sdb_log(SDB_LOG_ERR, "unixsock: Failed to read "
+						"from socket (%s): %s", client->path,
+						sdb_strerror(errno, errbuf, sizeof(errbuf)));
+			}
+			return NULL;
 		}
-		return buffer;
 	}
 	buffer[buflen - 1] = '\0';
 
 	buflen = strlen(buffer);
-	while ((buffer[buflen - 1] == '\n') || (buffer[buflen - 1] == '\r')) {
+	while (buflen && ((buffer[buflen - 1] == '\n') || (buffer[buflen - 1] == '\r'))) {
 		buffer[buflen - 1] = '\0';
 		--buflen;
 	}
@@ -336,9 +356,9 @@ sdb_unixsock_client_process_lines(sdb_unixsock_client_t *client,
 			types[i] = va_arg(ap, int);
 
 			if ((types[i] < 1) || (types[i] > SDB_TYPE_BINARY)) {
-				fprintf(stderr, "unixsock: Unknown column type %i while "
-						"processing response from the UNIX socket @ %s.\n",
-						types[i], client->path);
+				sdb_log(SDB_LOG_ERR, "unixsock: Unknown column "
+						"type %i while processing response from the "
+						"UNIX socket @ %s.", types[i], client->path);
 				va_end(ap);
 				free(types);
 				return -1;
@@ -369,9 +389,9 @@ sdb_unixsock_client_process_lines(sdb_unixsock_client_t *client,
 		column_count = sdb_unixsock_get_column_count(line, delim);
 
 		if ((n_cols >= 0) && (n_cols != column_count)) {
-			fprintf(stderr, "unixsock: number of columns (%i) does not "
-					"match the number of requested columns (%i) while "
-					"processing response from the UNIX socket @ %s: %s\n",
+			sdb_log(SDB_LOG_ERR, "unixsock: number of columns (%i) "
+					"does not match the number of requested columns (%i) "
+					"while processing response from the UNIX socket @ %s: %s",
 					column_count, n_cols, client->path, line);
 			continue;
 		}
@@ -390,8 +410,8 @@ sdb_unixsock_client_process_lines(sdb_unixsock_client_t *client,
 			|| ((max_lines < 0) && (! sdb_unixsock_client_eof(client)))
 			|| sdb_unixsock_client_error(client)) {
 		char errbuf[1024];
-		fprintf(stderr, "unixsock: Unexpected end of data while reading "
-				"from socket (%s): %s\n", client->path,
+		sdb_log(SDB_LOG_ERR, "unixsock: Unexpected end of data while "
+				"reading from socket (%s): %s", client->path,
 				sdb_strerror(errno, errbuf, sizeof(errbuf)));
 		return -1;
 	}