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
16 #include "hilog_input_socket_server.h"
17
18 #include <__threading_support>
19 #include <atomic>
20 #include <functional>
21 #include <sys/prctl.h>
22 #include <sys/socket.h>
23 #include <thread>
24 #include <vector>
25
26 namespace OHOS {
27 namespace HiviewDFX {
~HilogInputSocketServer()28 HilogInputSocketServer::~HilogInputSocketServer()
29 {
30 StopServingThread();
31 }
32
RunServingThread()33 HilogInputSocketServer::ServerThreadState HilogInputSocketServer::RunServingThread()
34 {
35 if (m_serverThread.get_id() != std::thread().get_id()) {
36 return ServerThreadState::ALREADY_STARTED;
37 }
38 m_stopServer.store(false);
39 m_serverThread = std::thread([this]() {
40 ServingThread();
41 });
42 if (m_serverThread.get_id() != std::thread().get_id()) {
43 return ServerThreadState::JUST_STARTED;
44 }
45 return ServerThreadState::CAN_NOT_START;
46 }
47
StopServingThread()48 void HilogInputSocketServer::StopServingThread()
49 {
50 if (m_serverThread.get_id() == std::thread().get_id()) {
51 return;
52 }
53 std::thread tmp;
54 std::swap(m_serverThread, tmp);
55 if (tmp.joinable()) {
56 m_stopServer.store(true);
57 tmp.join();
58 }
59 }
60
ServingThread()61 void HilogInputSocketServer::ServingThread()
62 {
63 prctl(PR_SET_NAME, "hilogd.server");
64 int ret;
65 std::vector<char> data(maxPacketLength);
66 #ifndef __RECV_MSG_WITH_UCRED_
67 while ((ret = RecvPacket(data)) >= 0) {
68 if (ret > 0) {
69 m_packetHandler(data, ret);
70 }
71 if (m_stopServer.load()) {
72 break;
73 }
74 }
75 #else
76 ucred cred;
77 while ((ret = RecvPacket(data, &cred)) >= 0) {
78 if (ret > 0) {
79 m_packetHandler(cred, data, ret);
80 }
81 if (m_stopServer.load()) {
82 break;
83 }
84 }
85 #endif
86 }
87 } // namespace HiviewDFX
88 } // namespace OHOS
89