From 0e03940802cebefdf6b0597a154bd9395e1af4d2 Mon Sep 17 00:00:00 2001
From: Olivier Gayot <duskcoder@gmail.com>
Date: Wed, 18 Jun 2014 15:45:45 +0200
Subject: Add the vanilla version of the project

This version can still be found here:

http://www.roland-riegel.de/nload/index.html
---
 src/devreader-bsd.cpp | 175 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 175 insertions(+)
 create mode 100644 src/devreader-bsd.cpp

(limited to 'src/devreader-bsd.cpp')

diff --git a/src/devreader-bsd.cpp b/src/devreader-bsd.cpp
new file mode 100644
index 0000000..b542704
--- /dev/null
+++ b/src/devreader-bsd.cpp
@@ -0,0 +1,175 @@
+/***************************************************************************
+ *                                                                         *
+ *   This program is free software; you can redistribute it and/or modify  *
+ *   it under the terms of the GNU General Public License as published by  *
+ *   the Free Software Foundation; either version 2 of the License, or     *
+ *   (at your option) any later version.                                   * 
+ *                                                                         *
+ ***************************************************************************/
+
+/***************************************************************************
+ * Free/Net/OpenBSD specific network code taken out of gkrellm:            *
+ * Copyright by Bill Wilson <bill@gkrellm.net>                             *
+ * FreeBSD code by Hajimu Umemoto <ume@mahoroba.org>                       *
+ * NetBSD code by Anthony Mallet <anthony.mallet@useless-ficus.net>        *
+ * Hajimu Umemoto merged Free/Net/OpenBSD code                             *
+ * adapted for nload by Roland Riegel <feedback@roland-riegel.de>          *
+ ***************************************************************************/
+
+#include "devreader-bsd.h"
+
+#include <stdlib.h>
+#include <stdio.h>
+#include <string.h>
+#include <sys/types.h>
+#include <sys/ioctl.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+#include <net/if.h>
+#include <unistd.h>
+#include <sys/param.h>
+#include <sys/sysctl.h>
+#include <net/if_dl.h>
+#include <net/route.h>
+
+#include <string>
+#include <list>
+
+using namespace std;
+
+DevReaderBsd::DevReaderBsd(const string& deviceName)
+    : DevReader(deviceName)
+{
+}
+
+DevReaderBsd::~DevReaderBsd()
+{
+}
+
+list<string> DevReaderBsd::findAllDevices()
+{
+    list<string> interfaceNames;
+
+    struct if_msghdr *ifm, *nextifm;
+    struct sockaddr_dl *sdl;
+    char *lim, *next;
+    size_t needed;
+    int mib[] = {CTL_NET, PF_ROUTE, 0, 0, NET_RT_IFLIST, 0};
+    char *buf = 0;
+    
+    if(sysctl(mib, 6, NULL, &needed, NULL, 0) < 0)
+        return interfaceNames;
+            
+    buf = (char *) malloc(needed);
+    if(buf == NULL)
+        return interfaceNames;
+            
+    if(sysctl(mib, 6, buf, &needed, NULL, 0) < 0)
+        return interfaceNames;
+            
+    lim = buf + needed;
+            
+    next = buf;
+    while(next < lim)
+    {
+        ifm = (struct if_msghdr *) next;
+        if(ifm->ifm_type != RTM_IFINFO)
+            break;
+                    
+        next += ifm->ifm_msglen;
+                    
+        while(next < lim)
+        {
+            nextifm = (struct if_msghdr *) next;
+            if(nextifm->ifm_type != RTM_NEWADDR)
+                break;
+            next += nextifm->ifm_msglen;
+        }
+                    
+        sdl = (struct sockaddr_dl *) (ifm + 1);
+        if(sdl->sdl_family != AF_LINK)
+            continue;
+        
+        interfaceNames.push_back(string(sdl->sdl_data));
+    }
+
+    free(buf);
+
+    return interfaceNames;
+}
+
+void DevReaderBsd::readFromDevice(DataFrame& dataFrame)
+{
+    struct if_msghdr *ifm, *nextifm;
+    struct sockaddr_dl *sdl;
+    char *lim, *next;
+    size_t needed;
+    int mib[] = {CTL_NET, PF_ROUTE, 0, 0, NET_RT_IFLIST, 0};
+    char *buf = 0;
+    
+    if(m_deviceName.empty())
+        return;
+	
+    do
+    {
+        if(sysctl(mib, 6, NULL, &needed, NULL, 0) < 0)
+            break;
+		
+        buf = (char *) malloc(needed);
+        if(buf == NULL)
+            break;
+		
+        if(sysctl(mib, 6, buf, &needed, NULL, 0) < 0)
+            break;
+		
+        lim = buf + needed;
+		
+        next = buf;
+        while(next < lim)
+        {
+            ifm = (struct if_msghdr *) next;
+            if(ifm->ifm_type != RTM_IFINFO)
+                break;
+			
+            next += ifm->ifm_msglen;
+			
+            while(next < lim)
+            {
+                nextifm = (struct if_msghdr *) next;
+                if(nextifm->ifm_type != RTM_NEWADDR)
+                    break;
+                next += nextifm->ifm_msglen;
+            }
+			
+            if(ifm->ifm_flags & IFF_UP)
+            {
+                sdl = (struct sockaddr_dl *) (ifm + 1);
+                if(sdl->sdl_family != AF_LINK)
+                    continue;
+                    
+                if(strncmp(m_deviceName.c_str(), sdl->sdl_data, sdl->sdl_nlen) == 0)
+                {
+                    dataFrame.setTotalDataIn(ifm->ifm_data.ifi_ibytes);
+                    dataFrame.setTotalDataOut(ifm->ifm_data.ifi_obytes);
+
+                    dataFrame.setTotalPacketsIn(ifm->ifm_data.ifi_ipackets);
+                    dataFrame.setTotalPacketsOut(ifm->ifm_data.ifi_opackets);
+
+                    dataFrame.setTotalErrorsIn(ifm->ifm_data.ifi_ierrors);
+                    dataFrame.setTotalErrorsOut(ifm->ifm_data.ifi_oerrors);
+
+                    dataFrame.setTotalDropsIn(0); // ifi_iqdrops ?
+                    dataFrame.setTotalDropsOut(0);
+
+                    dataFrame.setValid(true);
+
+                    break;
+                }
+            }
+        }
+    } while(0);
+    
+    free(buf);
+}
+
-- 
cgit v1.2.3