From 097890c3b2709eee1c826639ae3018ed376da099 Mon Sep 17 00:00:00 2001
From: rodri <rgl@antares-labs.eu>
Date: Sun, 1 Sep 2024 15:55:20 +0000
Subject: debug/plmon: consume as many files as arguments.

---
 debug/plmon.c | 56 ++++++++++++++++++++++++++++++++++++--------------------
 1 file changed, 36 insertions(+), 20 deletions(-)

(limited to 'debug/plmon.c')

diff --git a/debug/plmon.c b/debug/plmon.c
index 354ddb6..5c60b25 100644
--- a/debug/plmon.c
+++ b/debug/plmon.c
@@ -90,6 +90,27 @@ addt(char *n, Slot s)
 	t->times[t->ntime-1] = s;
 }
 
+static void
+consume(Biobuf *bin)
+{
+	Slot s;
+	char *line, *f[3];
+	int nf;
+
+	while((line = Brdline(bin, '\n')) != nil){
+		line[Blinelen(bin)-1] = 0;
+		nf = tokenize(line, f, 3);
+		if(nf != 3)
+			continue;
+		s.t0 = strtoull(f[1], nil, 10);
+		s.t1 = strtoull(f[2], nil, 10);
+		if(s.t0 >= s.t1)
+			continue;
+		addt(f[0], s);
+	}
+	Bterm(bin);
+}
+
 static void
 printsched(void)
 {
@@ -316,7 +337,7 @@ redraw(void)
 void
 usage(void)
 {
-	fprint(2, "usage: %s [file]\n", argv0);
+	fprint(2, "usage: %s [file ...]\n", argv0);
 	exits("usage");
 }
 
@@ -326,33 +347,28 @@ threadmain(int argc, char *argv[])
 	Mousectl *mc;
 	Keyboardctl *kc;
 	Biobuf *bin;
-	Slot s;
 	Rune r;
-	char *line, *f[3];
-	ulong nf;
 	int i;
 
 	ARGBEGIN{
 	default: usage();
 	}ARGEND
-	if(argc > 1)
-		usage();
 
-	bin = argc? Bopen(argv[0], OREAD): Bfdopen(0, OREAD);
-	if(bin == nil)
-		sysfatal("Bfdopen: %r");
-	while((line = Brdline(bin, '\n')) != nil){
-		line[Blinelen(bin)-1] = 0;
-		nf = tokenize(line, f, 3);
-		if(nf != 3)
-			continue;
-		s.t0 = strtoull(f[1], nil, 10);
-		s.t1 = strtoull(f[2], nil, 10);
-		if(s.t0 >= s.t1)
-			continue;
-		addt(f[0], s);
+	if(argc < 1){
+		bin = Bfdopen(0, OREAD);
+		if(bin == nil)
+			sysfatal("Bfdopen: %r");
+		consume(bin);
 	}
-	Bterm(bin);
+	for(i = 0; i < argc; i++){
+		bin = Bopen(argv[i], OREAD);
+		if(bin == nil)
+			sysfatal("Bopen: %r");
+		consume(bin);
+	}
+
+	if(sched.ntask < 1)
+		sysfatal("no tasks found");
 
 	print("loaded %lud tasks:\n", sched.ntask);
 	for(i = 0; i < sched.ntask; i++)
-- 
cgit v1.2.3