1 /*
2  * Copyright 2019, 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 #pragma once
18 
19 #include "result.h"
20 
21 #include <netlink/netlink.h>
22 
23 #include <type_traits>
24 
25 class NetlinkMessage {
26 public:
27     NetlinkMessage();
28     NetlinkMessage(NetlinkMessage&& other);
29     ~NetlinkMessage();
30 
31     NetlinkMessage& operator=(NetlinkMessage&& other);
32 
33     bool initGeneric(int family, uint8_t command, int version);
34 
35     template<typename T,
36              typename = typename std::enable_if<!std::is_pointer<T>::value>::type>
addAttribute(int attribute,const T & value)37     bool addAttribute(int attribute, const T& value) {
38        return nla_put(mMessage, attribute, sizeof(T), &value) == 0;
39     }
40     template<typename T,
41              typename = typename std::enable_if<std::is_pointer<T>::value>::type>
addAttribute(int attribute,T data,size_t numElements)42     bool addAttribute(int attribute, T data, size_t numElements) {
43         return nla_put(mMessage,
44                        attribute,
45                        sizeof(*data) * numElements,
46                        data) == 0;
47     }
48 
49     uint32_t getSeqNum() const;
50 
get()51     struct nl_msg* get() { return mMessage; }
52 
53 private:
54     NetlinkMessage(const NetlinkMessage&) = delete;
55     NetlinkMessage& operator=(const NetlinkMessage&) = delete;
56 
57     struct nl_msg* mMessage = nullptr;
58 };
59 
60