[PATCH trace-cmd V6 1/7] trace-cmd: Support -N option for trace-cmd extract

[Masami Hiramatsu <masami.hiramatsu.pt@hitachi.com>]

From: Yoshihiro YUNOMAE <yoshihiro.yunomae.ez@hitachi.com>

There is no reason to prohibit supporting -N command in extract
mode, since both record and extract read trace logs from ftrace
and save it.

Signed-off-by: Yoshihiro YUNOMAE <yoshihiro.yunomae.ez@hitachi.com>
Signed-off-by: Masami Hiramatsu <masami.hiramatsu.pt@hitachi.com>
---
 trace-record.c |    8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/trace-record.c b/trace-record.c
index 9874a12..c387aff 100644
--- a/trace-record.c
+++ b/trace-record.c
@@ -520,6 +520,7 @@ static void stop_threads(enum trace_type type)
 
 static int create_recorder(struct buffer_instance *instance, int cpu,
 			   enum trace_type type, int *brass);
+static void setup_network(void);
 
 static void flush_threads(void)
 {
@@ -529,6 +530,9 @@ static void flush_threads(void)
 	if (!cpu_count)
 		return;
 
+	if (host)
+		setup_network();
+
 	for (i = 0; i < cpu_count; i++) {
 		/* Extract doesn't support sub buffers yet */
 		ret = create_recorder(&top_instance, i, TRACE_TYPE_EXTRACT, NULL);
@@ -4047,8 +4051,8 @@ void trace_record (int argc, char **argv)
 			rt_prio = atoi(optarg);
 			break;
 		case 'N':
-			if (!record)
-				die("-N only available with record");
+			if (!record && !extract)
+				die("-N only available with record or extract");
 			if (output)
 				die("-N incompatible with -o");
 			host = optarg;