summaryrefslogtreecommitdiff
path: root/ACE/ace/Monitor_Control/BSD_Network_Interface_Monitor.cpp
blob: ffb2f3f9cd70536c6fc1874a7de3813e41072673 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
// $Id$

#include "ace/Monitor_Control/BSD_Network_Interface_Monitor.h"

#if defined (__FreeBSD__) || defined (__NetBSD__) || defined (__OpenBSD__)

#include "ace/Log_Msg.h"
#include "ace/OS_NS_stdio.h"
#include <sys/types.h>
#include <sys/socket.h>
#include <ifaddrs.h>

ACE_BEGIN_VERSIONED_NAMESPACE_DECL

namespace ACE
{
  namespace Monitor_Control
  {
    BSD_Network_Interface_Monitor::BSD_Network_Interface_Monitor (
      const ACE_TCHAR *lookup_str)
      : value_ (0UL),
        lookup_str_ (lookup_str)
    {
    }

    void
    BSD_Network_Interface_Monitor::update_i (void)
    {
      ACE_UINT64 count = 0;

      int fd = socket (AF_INET, SOCK_DGRAM, 0);
      if (fd == -1) 
        {
          ACE_ERROR ((LM_ERROR, ACE_TEXT ("socket failed\n")));
          return;
        }

      struct ifaddrs *ifa, *ifap;
      if (getifaddrs (&ifap) < 0) 
        {
          ACE_ERROR ((LM_ERROR, ACE_TEXT ("getifaddrs failed\n")));
          close (fd);
          return;     
        }

      char *p = 0;
      for (ifa = ifap; ifa != 0; ifa = ifa->ifa_next) 
        {
          if (p && strcmp (p, ifa->ifa_name) == 0)
            continue;
          p = ifa->ifa_name; 

    struct ifdatareq ifdr;
    memset (&ifdr, 0, sizeof (ifdr));
    strncpy (ifdr.ifdr_name, ifa->ifa_name, sizeof (ifdr));

    if (ioctl (fd, SIOCGIFDATA, &ifdr) == -1) 
      {
        ACE_ERROR ((LM_ERROR, ACE_TEXT ("SIOCGIFDATA failed\n")));
      }

    struct if_data * const ifi = &ifdr.ifdr_data;

    if (this->lookup_str_ == "ibytes") {
        count += ifi->ifi_ibytes;
    } else if (this->lookup_str_ == "ipackets") {
        count += ifi->ifi_ipackets;
    } else if (this->lookup_str_ == "obytes") {
        count += ifi->ifi_obytes;
    } else if (this->lookup_str_ == "opackets") {
        count += ifi->ifi_opackets;
    }
        }

      freeifaddrs (ifap);

      close (fd);

      value_ = count;
    }
  }
}

ACE_END_VERSIONED_NAMESPACE_DECL

#endif /* defined (__FreeBSD__) || defined (__NetBSD__) || defined (__OpenBSD__) */