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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
/*
* Copyright (c) 2009, Jay Loden, Giampaolo Rodola'. All rights reserved.
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
// Networks related functions. Original code was refactored and moved
// from psutil/_psutil_osx.c in 2023. This is the GIT blame before the move:
// https://github.com/giampaolo/psutil/blame/efd7ed3/psutil/_psutil_osx.c
#include <Python.h>
#include <sys/socket.h>
#include <sys/sysctl.h>
#include <net/if.h>
#include <net/if_dl.h>
#include <net/route.h>
#include "../../arch/all/init.h"
PyObject *
psutil_net_io_counters(PyObject *self, PyObject *args) {
char *buf = NULL, *lim, *next;
struct if_msghdr *ifm;
int mib[6];
size_t len = 0;
PyObject *py_ifc_info = NULL;
PyObject *py_retdict = PyDict_New();
if (py_retdict == NULL)
return NULL;
mib[0] = CTL_NET; // networking subsystem
mib[1] = PF_ROUTE; // type of information
mib[2] = 0; // protocol (IPPROTO_xxx)
mib[3] = 0; // address family
mib[4] = NET_RT_IFLIST2; // operation
mib[5] = 0;
if (psutil_sysctl_malloc(mib, 6, &buf, &len) != 0)
goto error;
lim = buf + len;
for (next = buf; next < lim; ) {
// Check we have enough space for if_msghdr.
if ((size_t)(lim - next) < sizeof(struct if_msghdr)) {
psutil_debug("struct xfile size mismatch");
}
ifm = (struct if_msghdr *)next;
if (ifm->ifm_msglen == 0 || next + ifm->ifm_msglen > lim) {
psutil_debug("ifm_msglen size mismatch");
}
next += ifm->ifm_msglen;
if (ifm->ifm_type == RTM_IFINFO2) {
py_ifc_info = NULL;
struct if_msghdr2 *if2m = (struct if_msghdr2 *)ifm;
if ((char *)if2m + sizeof(struct if_msghdr2) > lim) {
psutil_debug("if_msghdr2 + sockaddr_dl mismatch");
}
struct sockaddr_dl *sdl = (struct sockaddr_dl *)(if2m + 1);
if ((char *)sdl + sizeof(struct sockaddr_dl) > lim) {
psutil_debug("not enough buffer for sockaddr_dl");
}
char ifc_name[32];
size_t namelen = sdl->sdl_nlen;
if (namelen >= sizeof(ifc_name))
namelen = sizeof(ifc_name) - 1;
strncpy(ifc_name, sdl->sdl_data, namelen);
ifc_name[namelen] = '\0';
py_ifc_info = Py_BuildValue(
"(KKKKKKKi)",
(unsigned long long)if2m->ifm_data.ifi_obytes,
(unsigned long long)if2m->ifm_data.ifi_ibytes,
(unsigned long long)if2m->ifm_data.ifi_opackets,
(unsigned long long)if2m->ifm_data.ifi_ipackets,
(unsigned long long)if2m->ifm_data.ifi_ierrors,
(unsigned long long)if2m->ifm_data.ifi_oerrors,
(unsigned long long)if2m->ifm_data.ifi_iqdrops,
0); // dropout not supported
if (!py_ifc_info)
goto error;
if (PyDict_SetItemString(py_retdict, ifc_name, py_ifc_info)) {
Py_CLEAR(py_ifc_info);
goto error;
}
Py_CLEAR(py_ifc_info);
}
}
free(buf);
return py_retdict;
error:
Py_XDECREF(py_ifc_info);
Py_DECREF(py_retdict);
if (buf != NULL)
free(buf);
return NULL;
}
|
