Altered version of the sys-mon-set widgets with more padding

- Link to original version: http://tracesof.net/uebersicht-widgets/#sys-mon-set
2019-07-27 21:46:03 +03:00
commit 918aa40339
10 changed files with 491 additions and 0 deletions
@@ -0,0 +1,166 @@
+/*
+ * Print interface statistics in bytes per second
+ * 
+ * Parameters: 
+ * -t #		Time interval to compute, default 2 seconds
+ * -i <ifname>	Interface name, defaults to 'en0' 
+ * -s		Print raw stats
+ * 
+ * Output: Two numbers separated by a space. First value is input bytes per
+ * second. Second value is output bytes per second
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/sysctl.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <net/if.h>
+#include <net/if_var.h>
+#include <net/if_dl.h>
+#include <net/if_types.h>
+#include <net/if_mib.h>
+#include <net/route.h>
+
+#include <netinet/in.h>
+#include <netinet/in_var.h>
+#include <unistd.h>
+
+static u_int64_t opackets = 0;
+static u_int64_t ipackets = 0;
+static u_int64_t obytes = 0;
+static u_int64_t ibytes = 0;
+static int	raw_stats = 0;
+
+int
+get_stats(char *interface)
+{
+	int		ret = 0;
+	char		name      [32];
+	int		mib        [6];
+	char           *buf = NULL, *lim, *next;
+	size_t		len;
+	struct if_msghdr *ifm;
+	unsigned int	ifindex = 0;
+
+	if (interface != 0)
+		ifindex = if_nametoindex(interface);
+
+	mib[0] = CTL_NET;
+	mib[1] = PF_ROUTE;
+	mib[2] = 0;
+	mib[3] = 0;
+	mib[4] = NET_RT_IFLIST2;
+	mib[5] = 0;
+
+	if (sysctl(mib, 6, NULL, &len, NULL, 0) < 0)
+		return ret;
+	if ((buf = malloc(len)) == NULL) {
+		printf("malloc failed\n");
+		exit(1);
+	}
+	if (sysctl(mib, 6, buf, &len, NULL, 0) < 0) {
+		if (buf)
+			free(buf);
+		return ret;
+	}
+	lim = buf + len;
+	for (next = buf; next < lim;) {
+		ifm = (struct if_msghdr *)next;
+		next += ifm->ifm_msglen;
+
+		if (ifm->ifm_type == RTM_IFINFO2) {
+			struct if_msghdr2 *if2m = (struct if_msghdr2 *)ifm;
+			struct sockaddr_dl *sdl = (struct sockaddr_dl *)(if2m + 1);
+
+			strncpy(name, sdl->sdl_data, sdl->sdl_nlen);
+			name[sdl->sdl_nlen] = 0;
+			if (interface != 0 && if2m->ifm_index != ifindex)
+				continue;
+
+			/*
+			 * Get the interface stats.  These may get overriden
+			 * below on a per-interface basis.
+			 */
+			opackets = if2m->ifm_data.ifi_opackets;
+			ipackets = if2m->ifm_data.ifi_ipackets;
+			obytes = if2m->ifm_data.ifi_obytes;
+			ibytes = if2m->ifm_data.ifi_ibytes;
+			if (ret == 0) {
+				ret = 1;
+				if (raw_stats) {
+					printf("%5s %10s %14s ",
+					       "Name", "Ipkts", "IBytes");
+					printf("%10s %14s\n", "Opkts", "Obytes");
+				}
+			}
+			if (raw_stats) {
+				printf("%-5s %10llu ", name, ipackets);
+				printf("%14llu ", ibytes);
+				printf("%10llu ", opackets);
+				printf("%14llu\n", obytes);
+			}
+		}
+	}
+
+	free(buf);
+
+	return ret;
+}
+
+void
+usage(void)
+{
+	printf("arguments:\n\t-t #\t\tTime interval to compute, default 2 seconds\n");
+	printf("\t-i <ifname>\tInterface name, defaults to 'en0'\n");
+	printf("\t-s\t\tPrint raw stats\n");
+	exit(1);
+}
+
+int
+main(int argc, char *argv[])
+{
+	char           *ifname = "en0";
+	int		r;
+	int		sleeptime = 2;
+
+	int		bflag     , ch, fd;
+
+	bflag = 0;
+	while ((ch = getopt(argc, argv, "st:i:")) != -1) {
+		switch (ch) {
+		case 't':
+			sleeptime = atoi(optarg);
+			if (sleeptime < 1) {
+				sleeptime = 1;
+			}
+			break;
+		case 'i':
+			ifname = optarg;
+			break;
+		case 's':
+			raw_stats = 1;
+			break;
+		default:
+			usage();
+		}
+	}
+	argc -= optind;
+	argv += optind;
+
+	r = get_stats(ifname);
+	if (r) {
+		u_int64_t	ib = ibytes;
+		u_int64_t	ob = obytes;
+		u_int64_t	diffo, diffi;
+
+		sleep(sleeptime);
+		get_stats(ifname);
+		diffo = (obytes - ob) / (u_int64_t) sleeptime;
+		diffi = (ibytes - ib) / (u_int64_t) sleeptime;
+		printf("%llu %llu\n", diffi, diffo);
+	} else {
+		printf("No interface %s\n", ifname);
+	}
+}