1 /* 2 * Copyright (c) 2023 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 #ifndef ARKCOMPILER_TOOLCHAIN_WEBSOCKET_CLIENT_WEBSOCKET_CLIENT_H 17 #define ARKCOMPILER_TOOLCHAIN_WEBSOCKET_CLIENT_WEBSOCKET_CLIENT_H 18 19 #include "http.h" 20 #include "websocket_base.h" 21 22 #include <array> 23 #include <atomic> 24 #include <iostream> 25 #include <map> 26 #include <memory> 27 28 namespace OHOS::ArkCompiler::Toolchain { 29 class WebSocketClient final : public WebSocketBase { 30 public: 31 ~WebSocketClient() noexcept override = default; 32 33 void Close(); 34 35 bool InitToolchainWebSocketForPort(int port, uint32_t timeoutLimit = 5); 36 bool InitToolchainWebSocketForSockName(const std::string &sockName, uint32_t timeoutLimit = 5); 37 bool ClientSendWSUpgradeReq(); 38 bool ClientRecvWSUpgradeRsp(); 39 40 std::string GetSocketStateString(); 41 42 private: 43 bool DecodeMessage(WebSocketFrame& wsFrame) const override; 44 45 void CloseOnInitFailure(); 46 bool ValidateIncomingFrame(const WebSocketFrame& wsFrame) const override; 47 std::string CreateFrame(bool isLast, FrameType frameType) const override; 48 std::string CreateFrame(bool isLast, FrameType frameType, const std::string& payload) const override; 49 std::string CreateFrame(bool isLast, FrameType frameType, std::string&& payload) const override; 50 51 private: 52 static constexpr std::array<std::string_view, 4> SOCKET_STATE_NAMES = { 53 "connecting", 54 "open", 55 "closing", 56 "closed" 57 }; 58 59 // May replace default websocket-key with randomly generated, as required by spec. 60 static constexpr char CLIENT_WEBSOCKET_UPGRADE_REQ[] = "GET / HTTP/1.1\r\n" 61 "Connection: Upgrade\r\n" 62 "Pragma: no-cache\r\n" 63 "Cache-Control: no-cache\r\n" 64 "Upgrade: websocket\r\n" 65 "Sec-WebSocket-Version: 13\r\n" 66 "Accept-Encoding: gzip, deflate, br\r\n" 67 "Sec-WebSocket-Key: 64b4B+s5JDlgkdg7NekJ+g==\r\n" 68 "Sec-WebSocket-Extensions: permessage-deflate\r\n" 69 "\r\n"; 70 71 static constexpr int NET_SUCCESS = 1; 72 static constexpr uint8_t MASK_KEY[] = {0xa, 0xb, 0xc, 0xd}; 73 }; 74 } // namespace OHOS::ArkCompiler::Toolchain 75 76 #endif // ARKCOMPILER_TOOLCHAIN_WEBSOCKET_CLIENT_WEBSOCKET_CLIENT_H 77