From 8cdd42ddc8d5930f07a0c21e0d94e0cfd3a8cbcb Mon Sep 17 00:00:00 2001
From: Radharapu Rakesh <rakesh.radharapu@amd.com>
Date: Tue, 2 Dec 2025 12:51:38 +0530
Subject: [PATCH] tsn-utils: Add TADMA stream query support

Add -g option to display configured TADMA streams sorted by trigger time.

Signed-off-by: Radharapu Rakesh <rakesh.radharapu@amd.com>
---
 tsn-apps/tadma_prog/tadma_prog.c | 72 +++++++++++++++++++++++++++++---
 1 file changed, 66 insertions(+), 6 deletions(-)

diff --git a/tsn-apps/tadma_prog/tadma_prog.c b/tsn-apps/tadma_prog/tadma_prog.c
index d6a4c3b..d03c5f2 100644
--- a/tsn-apps/tadma_prog/tadma_prog.c
+++ b/tsn-apps/tadma_prog/tadma_prog.c
@@ -16,6 +16,8 @@
 #include <libconfig.h>
 #include <errno.h>
 
+#define MAX_STREAMS 128
+
 enum axienet_tsn_ioctl {
 	SIOCCHIOCTL = SIOCDEVPRIVATE,
 	SIOC_GET_SCHED,
@@ -30,6 +32,7 @@ enum axienet_tsn_ioctl {
 	SIOC_TADMA_STR_FLUSH,
 	SIOC_PREEMPTION_RECEIVE,
 	SIOC_TADMA_OFF,
+	SIOC_TADMA_GET_STREAMS,
 };
 
 struct tadma_stream {
@@ -40,6 +43,62 @@ struct tadma_stream {
 	unsigned char qno;
 };
 
+int sortbytrigger(const void *i1, const void *i2)  {
+	struct tadma_stream **a = (struct tadma_stream **)i1;
+	struct tadma_stream **b = (struct tadma_stream **)i2;
+	return ((*a)->trigger - (*b)->trigger);
+}
+
+int get_streams(char *ifname)
+{
+	struct tadma_stream *stream_ptrs[MAX_STREAMS];
+	struct tadma_stream streams[MAX_STREAMS];
+	struct ifreq s;
+	int ret;
+	int fd;
+	int i;
+
+	fd = socket(PF_INET, SOCK_DGRAM, IPPROTO_IP);
+	if (fd < 0) {
+		perror("socket");
+		return -1;
+	}
+
+	strcpy(s.ifr_name, ifname);
+	s.ifr_data = (void *)streams;
+	ret = ioctl(fd, SIOC_TADMA_GET_STREAMS, &s);
+	if (ret < 0) {
+		perror("TADMA get streams failed");
+		close(fd);
+		return ret;
+	}
+
+	if (ret == 0) {
+		printf("No TADMA streams configured\n");
+		close(fd);
+		return 0;
+	}
+
+	for (i = 0; i < ret; i++) {
+		stream_ptrs[i] = &streams[i];
+	}
+	qsort(stream_ptrs, ret, sizeof(struct tadma_stream*), sortbytrigger);
+
+	printf("Configured TADMA Streams:\n");
+	for (i = 0; i < ret; i++) {
+		printf("Stream %d: MAC %02x:%02x:%02x:%02x:%02x:%02x, VID %u, QNO %u, Trigger %u, Count %u\n",
+				i,
+				stream_ptrs[i]->dmac[0], stream_ptrs[i]->dmac[1], stream_ptrs[i]->dmac[2],
+				stream_ptrs[i]->dmac[3], stream_ptrs[i]->dmac[4], stream_ptrs[i]->dmac[5],
+				stream_ptrs[i]->vid,
+				stream_ptrs[i]->qno,
+				stream_ptrs[i]->trigger,
+				stream_ptrs[i]->count);
+	}
+	close(fd);
+	return 0;
+}
+
 int change_to_continuous(char *ifname)
 {
 	struct ifreq s;
@@ -129,12 +188,6 @@ int program_all_streams(char *ifname)
 	close(fd);
 }
 
-int sortbytrigger(const void *i1, const void *i2)  {
-    struct tadma_stream **a = (struct tadma_stream **)i1;
-    struct tadma_stream **b = (struct tadma_stream **)i2;
-    return ((*a)->trigger - (*b)->trigger);
-}
-
 void usage()
 {
 	printf("Usage-1 :\n tadma_prog <interface>\n");
@@ -143,6 +196,8 @@ void usage()
 	printf("	file name : user defined streams's configuration. Ex: streams_01.cfg\n");
 	printf("Usage-3 :\n tadma_prog <interface> off\n");
 	printf("	Programs TADMA back to continuous mode\n");
+	printf("Usage-4 :\n tadma_prog -g <interface>\n");
+	printf("	Display currently configured TADMA streams\n");
 	printf("-c : To use the configuration of streams in the file at mentioned path\n");
 	exit(1);
 }
@@ -173,6 +228,10 @@ int main(int argc, char **argv)
 		flush_stream(ifname);
 		change_to_continuous(ifname);
 		return 0;
+	} else if (argc == 3 && strcmp(argv[1], "-g") == 0) {
+		strncpy(ifname, argv[2], IFNAMSIZ);
+		get_streams(ifname);
+		return 0;
 	} else {
 		if(argc == 3 || argc == 4) 
 			usage();
@@ -181,6 +240,7 @@ int main(int argc, char **argv)
 		strcpy(path,"/etc/xilinx-tsn/streams.cfg");
 		strncpy(ifname, argv[1], IFNAMSIZ);
 	}
+
 	if( !config_read_file(&cfg, path) )
 	{
 		config_destroy(&cfg);