| /* |
| * Copyright (C) 2017 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #ifndef CHRE_HOST_SOCKET_SERVER_H_ |
| #define CHRE_HOST_SOCKET_SERVER_H_ |
| |
| #include <poll.h> |
| |
| #include <atomic> |
| #include <functional> |
| #include <map> |
| #include <mutex> |
| #include <vector> |
| |
| #include <android-base/macros.h> |
| #include <cutils/sockets.h> |
| |
| namespace android::chre { |
| |
| class SocketServer { |
| public: |
| SocketServer(); |
| |
| /** |
| * Defines the function signature of the callback given to run() which |
| * receives message data sent in by a client. |
| * |
| * @param clientId A unique identifier for the client that sent this request |
| * (assigned locally) |
| * @param data Pointer to buffer containing the raw message data |
| * @param len Number of bytes of data received |
| */ |
| typedef std::function<void(uint16_t clientId, void *data, size_t len)> |
| ClientMessageCallback; |
| |
| /** |
| * Opens the socket, and runs the receive loop until an error is encountered, |
| * or SIGINT/SIGTERM is received. Masks off all other signals. |
| * |
| * @param socketName Android socket name to use when listening |
| * @param allowSocketCreation If true, allow creation of the socket rather |
| * than strictly inheriting it from init (used primarily for |
| * development purposes) |
| * @param clientMessageCallback Callback to be invoked when a message is |
| * received from a client |
| */ |
| void run(const char *socketName, bool allowSocketCreation, |
| ClientMessageCallback clientMessageCallback); |
| |
| /** |
| * Delivers data to all connected clients. This method is thread-safe. |
| * |
| * @param data Pointer to buffer containing message data |
| * @param length Number of bytes of data to send |
| */ |
| void sendToAllClients(const void *data, size_t length); |
| |
| /** |
| * Sends a message to one client, specified via its unique client ID. This |
| * method is thread-safe. |
| * |
| * @param data |
| * @param length |
| * @param clientId |
| * |
| * @return true if the message was successfully sent to the specified client |
| */ |
| bool sendToClientById(const void *data, size_t length, uint16_t clientId); |
| |
| static void shutdownServer() { |
| sSignalReceived = true; |
| } |
| |
| private: |
| DISALLOW_COPY_AND_ASSIGN(SocketServer); |
| |
| static constexpr size_t kMaxActiveClients = 8; |
| static constexpr int kMaxPendingConnectionRequests = |
| static_cast<int>(kMaxActiveClients); |
| static constexpr size_t kMaxPacketSize = 1024 * 1024; |
| |
| // This is the same value as defined in |
| // host/hal_generic/common/hal_client_id.h. It is redefined here to avoid |
| // adding dependency path at multiple places for such a temporary change, |
| // which will be removed after migrating generic HAL to multiclient HAL. |
| static constexpr uint16_t kMaxHalClientId = 0x1ff; |
| |
| int mSockFd = INVALID_SOCKET; |
| // Socket client id and Hal client id are using the same field in the fbs |
| // message. To keep their id range disjoint enables message routing for both |
| // at the same time. There are 0xffff - 0x01ff = 0xfe00 (65024) socket |
| // client ids to use, which should be more than enough. |
| uint16_t mNextClientId = kMaxHalClientId + 1; |
| // TODO: std::vector-ify this |
| struct pollfd mPollFds[1 + kMaxActiveClients] = {}; |
| |
| struct ClientData { |
| uint16_t clientId; |
| }; |
| |
| // Maps from socket FD to ClientData |
| std::map<int, ClientData> mClients; |
| |
| // A buffer to read packets into. Allocated here to prevent a large object on |
| // the stack. |
| std::vector<uint8_t> mRecvBuffer = std::vector<uint8_t>(kMaxPacketSize); |
| |
| // Ensures that mClients can be safely iterated over from other threads |
| // without worrying about potential modification from the RX thread |
| std::mutex mClientsMutex; |
| |
| ClientMessageCallback mClientMessageCallback; |
| |
| void acceptClientConnection(); |
| |
| void disconnectClient(int clientSocket); |
| |
| void handleClientData(int clientSocket); |
| |
| bool sendToClientSocket(const void *data, size_t length, int clientSocket, |
| uint16_t clientId); |
| |
| void serviceSocket(); |
| |
| static std::atomic<bool> sSignalReceived; |
| }; |
| |
| } // namespace android::chre |
| |
| #endif // CHRE_HOST_SOCKET_SERVER_H_ |