1 /*
2  * Copyright (c) 2021 Huawei Device Co., Ltd.
3  * Licensed under the Apache License, Version 2.0 (the "License");
4  * you may not use this file except in compliance with the License.
5  * You may obtain a copy of the License at
6  *
7  *     http://www.apache.org/licenses/LICENSE-2.0
8  *
9  * Unless required by applicable law or agreed to in writing, software
10  * distributed under the License is distributed on an "AS IS" BASIS,
11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  * See the License for the specific language governing permissions and
13  * limitations under the License.
14  */
15 #ifndef OHOS_STORAGE_DAEMON_NETLINK_LISTENER_H
16 #define OHOS_STORAGE_DAEMON_NETLINK_LISTENER_H
17 
18 #include <memory>
19 #include <thread>
20 
21 #include <poll.h>
22 
23 #include "ipc/storage_daemon.h"
24 
25 namespace OHOS {
26 namespace StorageDaemon {
27 class NetlinkListener {
28 public:
29     NetlinkListener(int32_t socket);
30     int32_t StartListener();
31     int32_t StopListener();
32 
33 protected:
34     virtual void OnEvent(char *msg) = 0;
35 
36 private:
37     int32_t socketFd_ { -1 };
38     int32_t socketPipe_[2] { -1, -1 };
39     std::unique_ptr<std::thread> socketThread_;
40     void RecvUeventMsg();
41     int32_t ReadMsg(int32_t fd_count, struct pollfd ufds[2]);
42     void RunListener();
43     static void EventProcess(void*);
44 };
45 } // STORAGE_DAEMON
46 } // OHOS
47 
48 #endif // OHOS_STORAGE_DAEMON_NETLINK_LISTENER_H
49