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
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
|
/*
* Copyright (C) 2011, 2012 Red Hat Inc.
*
* David Malcolm <dmalcolm@redhat.com>
*
* This application 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; version 2.
*
* This application is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* General Public License for more details.
*/
/* Python object corresponding to a (struct rtnl_addr) */
#include <Python.h>
#include "structmember.h"
#include <netlink/route/rtnl.h>
#include "etherinfo_struct.h"
#include "etherinfo.h"
/* IPv4 Addresses: */
static PyObject *
PyNetlinkIPv4Address_from_rtnl_addr(struct nl_object *nl_obj, struct rtnl_addr *addr)
{
PyNetlinkIPv4Address *py_obj;
char buf[INET_ADDRSTRLEN+1];
struct nl_addr *brdcst;
py_obj = PyObject_New(PyNetlinkIPv4Address,
ðtool_netlink_ipv4_address_Type);
if (!py_obj) {
return NULL;
}
/* Set ipv4_address: */
memset(&buf, 0, sizeof(buf));
if (!inet_ntop(AF_INET, nl_addr_get_binary_addr((struct nl_addr *)addr),
buf, sizeof(buf))) {
PyErr_SetFromErrno(PyExc_RuntimeError);
goto error;
}
py_obj->ipv4_address = PyString_FromString(buf);
if (!py_obj->ipv4_address) {
goto error;
}
/* Set ipv4_netmask: */
py_obj->ipv4_netmask = rtnl_addr_get_prefixlen((struct rtnl_addr*)nl_obj);
/* Set ipv4_broadcast: */
py_obj->ipv4_broadcast = NULL;
brdcst = rtnl_addr_get_broadcast((struct rtnl_addr*)nl_obj);
if( brdcst ) {
memset(&buf, 0, sizeof(buf));
if (!inet_ntop(AF_INET, nl_addr_get_binary_addr(brdcst),
buf, sizeof(buf))) {
PyErr_SetFromErrno(PyExc_RuntimeError);
goto error;
}
py_obj->ipv4_broadcast = PyString_FromString(buf);
if (!py_obj->ipv4_broadcast) {
goto error;
}
}
return (PyObject*)py_obj;
error:
Py_DECREF(py_obj);
return NULL;
}
static void
netlink_ipv4_address_dealloc(PyNetlinkIPv4Address *obj)
{
Py_DECREF(obj->ipv4_address);
Py_XDECREF(obj->ipv4_broadcast);
/* We can call PyObject_Del directly rather than calling through
tp_free since the type is not subtypable (Py_TPFLAGS_BASETYPE is
not set): */
PyObject_Del(obj);
}
static PyObject*
netlink_ipv4_address_repr(PyNetlinkIPv4Address *obj)
{
PyObject *result = PyString_FromString("ethtool.NetlinkIPv4Address(address='");
PyString_Concat(&result, obj->ipv4_address);
PyString_ConcatAndDel(&result,
PyString_FromFormat("', netmask=%d",
obj->ipv4_netmask));
if (obj->ipv4_broadcast) {
PyString_ConcatAndDel(&result, PyString_FromString(", broadcast='"));
PyString_Concat(&result, obj->ipv4_broadcast);
PyString_ConcatAndDel(&result, PyString_FromString("'"));
}
PyString_ConcatAndDel(&result, PyString_FromString(")"));
return result;
}
static PyMemberDef _ethtool_netlink_ipv4_address_members[] = {
{"address",
T_OBJECT_EX,
offsetof(PyNetlinkIPv4Address, ipv4_address),
0,
NULL},
{"netmask",
T_INT,
offsetof(PyNetlinkIPv4Address, ipv4_netmask),
0,
NULL},
{"broadcast",
T_OBJECT, /* can be NULL */
offsetof(PyNetlinkIPv4Address, ipv4_broadcast),
0,
NULL},
{NULL} /* End of member list */
};
PyTypeObject ethtool_netlink_ipv4_address_Type = {
PyVarObject_HEAD_INIT(0, 0)
.tp_name = "ethtool.NetlinkIPv4Address",
.tp_basicsize = sizeof(PyNetlinkIPv4Address),
.tp_dealloc = (destructor)netlink_ipv4_address_dealloc,
.tp_repr = (reprfunc)netlink_ipv4_address_repr,
.tp_members = _ethtool_netlink_ipv4_address_members,
};
/* Factory function, in case we want to generalize this to add IPv6 support */
PyObject *
make_python_address_from_rtnl_addr(struct nl_object *obj,
struct rtnl_addr *addr)
{
int family;
assert(addr);
family = nl_addr_get_family((struct nl_addr *)addr);
switch( family ) {
case AF_INET:
return PyNetlinkIPv4Address_from_rtnl_addr(obj, addr);
/*
For now, we just support IPv4 addresses.
*/
default:
return PyErr_SetFromErrno(PyExc_RuntimeError);
}
}
/*
Local variables:
c-basic-offset: 8
indent-tabs-mode: y
End:
*/
|
