Grab abnd log agent messages to stderr

knizhnik · gsmolk · commit 6a721aa00425 · 2019-08-09T01:45:23.000+03:00
diff --git a/src/utils/file.c b/src/utils/file.c
@@ -20,6 +20,7 @@ static __thread unsigned long fio_fdset = 0;
 static __thread void* fio_stdin_buffer;
 static __thread int fio_stdout = 0;
 static __thread int fio_stdin = 0;
+static __thread int fio_stderr = 0;
 
 fio_location MyLocation;
 
@@ -38,10 +39,28 @@ typedef struct
 #define fio_fileno(f) (((size_t)f - 1) | FIO_PIPE_MARKER)
 
 /* Use specified file descriptors as stdin/stdout for FIO functions */
-void fio_redirect(int in, int out)
+void fio_redirect(int in, int out, int err)
 {
 	fio_stdin = in;
 	fio_stdout = out;
+	fio_stderr = err;
+}
+
+void fio_error(int rc, int size, char const* file, int line)
+{
+	if (remote_agent)
+	{
+		fprintf(stderr, "%s:%d: proceeds %d bytes instead of %d: %s\n", file, line, rc, size, rc >= 0 ? "end of data" :  strerror(errno));
+		exit(EXIT_FAILURE);
+	}
+	else
+	{
+		char buf[PRINTF_BUF_SIZE];
+		int err_size = read(fio_stderr, buf, sizeof(buf));
+		if (err_size > 0)
+			elog(LOG, "Agent error: %s", buf);
+		elog(ERROR, "Communication error: %s", rc >= 0 ? "end of data" :  strerror(errno));
+	}
 }
 
 /* Check if file descriptor is local or remote (created by FIO) */
diff --git a/src/utils/file.h b/src/utils/file.h
@@ -49,7 +49,7 @@ typedef enum
 #define PAGE_CHECKSUM_MISMATCH (-256)
 
 #define SYS_CHECK(cmd) do if ((cmd) < 0) { fprintf(stderr, "%s:%d: (%s) %s\n", __FILE__, __LINE__, #cmd, strerror(errno)); exit(EXIT_FAILURE); } while (0)
-#define IO_CHECK(cmd, size) do { int _rc = (cmd); if (_rc != (size)) { if (remote_agent) { fprintf(stderr, "%s:%d: proceeds %d bytes instead of %d: %s\n", __FILE__, __LINE__, _rc, (int)(size), _rc >= 0 ? "end of data" :  strerror(errno)); exit(EXIT_FAILURE); } else elog(ERROR, "Communication error: %s", _rc >= 0 ? "end of data" :  strerror(errno)); } } while (0)
+#define IO_CHECK(cmd, size) do { int _rc = (cmd); if (_rc != (size)) fio_error(_rc, size, __FILE__, __LINE__); } while (0)
 
 typedef struct
 {
@@ -64,7 +64,7 @@ extern fio_location MyLocation;
 /* Check if FILE handle is local or remote (created by FIO) */
 #define fio_is_remote_file(file) ((size_t)(file) <= FIO_FDMAX)
 
-extern void    fio_redirect(int in, int out);
+extern void    fio_redirect(int in, int out, int err);
 extern void    fio_communicate(int in, int out);
 
 extern FILE*   fio_fopen(char const* name, char const* mode, fio_location location);
@@ -77,6 +77,7 @@ extern int     fio_fseek(FILE* f, off_t offs);
 extern int     fio_ftruncate(FILE* f, off_t size);
 extern int     fio_fclose(FILE* f);
 extern int     fio_ffstat(FILE* f, struct stat* st);
+extern void    fio_error(int rc, int size, char const* file, int line);
 
 struct pgFile;
 extern  int    fio_send_pages(FILE* in, FILE* out, struct pgFile *file, XLogRecPtr horizonLsn, 
diff --git a/src/utils/remote.c b/src/utils/remote.c
@@ -110,6 +110,7 @@ bool launch_agent(void)
 	int ssh_argc;
 	int outfd[2];
 	int infd[2];
+	int errfd[2];
 
 	ssh_argc = 0;
 #ifdef WIN32
@@ -195,20 +196,25 @@ bool launch_agent(void)
 #else
 	SYS_CHECK(pipe(infd));
 	SYS_CHECK(pipe(outfd));
+	SYS_CHECK(pipe(errfd));
 
 	SYS_CHECK(child_pid = fork());
 
 	if (child_pid == 0) { /* child */
 		SYS_CHECK(close(STDIN_FILENO));
 		SYS_CHECK(close(STDOUT_FILENO));
+		SYS_CHECK(close(STDERR_FILENO));
 
 		SYS_CHECK(dup2(outfd[0], STDIN_FILENO));
 		SYS_CHECK(dup2(infd[1],  STDOUT_FILENO));
+		SYS_CHECK(dup2(errfd[1], STDERR_FILENO));
 
 		SYS_CHECK(close(infd[0]));
 		SYS_CHECK(close(infd[1]));
 		SYS_CHECK(close(outfd[0]));
 		SYS_CHECK(close(outfd[1]));
+		SYS_CHECK(close(errfd[0]));
+		SYS_CHECK(close(errfd[1]));
 
 		if (execvp(ssh_argv[0], ssh_argv) < 0)
 			return false;
@@ -217,9 +223,10 @@ bool launch_agent(void)
 		elog(LOG, "Spawn agent %d version %s", child_pid, PROGRAM_VERSION);
 		SYS_CHECK(close(infd[1]));  /* These are being used by the child */
 		SYS_CHECK(close(outfd[0]));
+		SYS_CHECK(close(errfd[1]));
 		/*atexit(kill_child);*/
 
-		fio_redirect(infd[0], outfd[1]); /* write to stdout */
+		fio_redirect(infd[0], outfd[1], errfd[0]); /* write to stdout */
 	}
 	return true;
 }
