1 /*
2 * Copyright (C) 2017 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #include <ios>
18 #include <linux/netlink.h>
19
20 #include "netdutils/Math.h"
21 #include "netdutils/Netlink.h"
22
23 namespace android {
24 namespace netdutils {
25
forEachNetlinkMessage(const Slice buf,const std::function<void (const nlmsghdr &,const Slice)> & onMsg)26 void forEachNetlinkMessage(const Slice buf,
27 const std::function<void(const nlmsghdr&, const Slice)>& onMsg) {
28 Slice tail = buf;
29 while (tail.size() >= sizeof(nlmsghdr)) {
30 nlmsghdr hdr = {};
31 extract(tail, hdr);
32 const auto len = std::max<size_t>(hdr.nlmsg_len, sizeof(hdr));
33 onMsg(hdr, drop(take(tail, len), sizeof(hdr)));
34 tail = drop(tail, align(len, 2));
35 }
36 }
37
forEachNetlinkAttribute(const Slice buf,const std::function<void (const nlattr &,const Slice)> & onAttr)38 void forEachNetlinkAttribute(const Slice buf,
39 const std::function<void(const nlattr&, const Slice)>& onAttr) {
40 Slice tail = buf;
41 while (tail.size() >= sizeof(nlattr)) {
42 nlattr hdr = {};
43 extract(tail, hdr);
44 const auto len = std::max<size_t>(hdr.nla_len, sizeof(hdr));
45 onAttr(hdr, drop(take(tail, len), sizeof(hdr)));
46 tail = drop(tail, align(len, 2));
47 }
48 }
49
50 } // namespace netdutils
51 } // namespace android
52
operator ==(const sockaddr_nl & lhs,const sockaddr_nl & rhs)53 bool operator==(const sockaddr_nl& lhs, const sockaddr_nl& rhs) {
54 return (lhs.nl_family == rhs.nl_family) && (lhs.nl_pid == rhs.nl_pid) &&
55 (lhs.nl_groups == rhs.nl_groups);
56 }
57
operator !=(const sockaddr_nl & lhs,const sockaddr_nl & rhs)58 bool operator!=(const sockaddr_nl& lhs, const sockaddr_nl& rhs) {
59 return !(lhs == rhs);
60 }
61
operator <<(std::ostream & os,const nlmsghdr & hdr)62 std::ostream& operator<<(std::ostream& os, const nlmsghdr& hdr) {
63 return os << std::hex << "nlmsghdr["
64 << "len: 0x" << hdr.nlmsg_len << ", type: 0x" << hdr.nlmsg_type << ", flags: 0x"
65 << hdr.nlmsg_flags << ", seq: 0x" << hdr.nlmsg_seq << ", pid: 0x" << hdr.nlmsg_pid
66 << "]" << std::dec;
67 }
68
operator <<(std::ostream & os,const nlattr & attr)69 std::ostream& operator<<(std::ostream& os, const nlattr& attr) {
70 return os << std::hex << "nlattr["
71 << "len: 0x" << attr.nla_len << ", type: 0x" << attr.nla_type << "]" << std::dec;
72 }
73
operator <<(std::ostream & os,const sockaddr_nl & addr)74 std::ostream& operator<<(std::ostream& os, const sockaddr_nl& addr) {
75 return os << std::hex << "sockaddr_nl["
76 << "family: " << addr.nl_family << ", pid: " << addr.nl_pid
77 << ", groups: " << addr.nl_groups << "]" << std::dec;
78 }
79