17c804472Sopenharmony_ci/* 27c804472Sopenharmony_ci * Copyright (c) 2023 Huawei Device Co., Ltd. 37c804472Sopenharmony_ci * Licensed under the Apache License, Version 2.0 (the "License"); 47c804472Sopenharmony_ci * you may not use this file except in compliance with the License. 57c804472Sopenharmony_ci * You may obtain a copy of the License at 67c804472Sopenharmony_ci * 77c804472Sopenharmony_ci * http://www.apache.org/licenses/LICENSE-2.0 87c804472Sopenharmony_ci * 97c804472Sopenharmony_ci * Unless required by applicable law or agreed to in writing, software 107c804472Sopenharmony_ci * distributed under the License is distributed on an "AS IS" BASIS, 117c804472Sopenharmony_ci * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 127c804472Sopenharmony_ci * See the License for the specific language governing permissions and 137c804472Sopenharmony_ci * limitations under the License. 147c804472Sopenharmony_ci */ 157c804472Sopenharmony_ci 167c804472Sopenharmony_ci#ifndef WEBSOCKETSERVER_H 177c804472Sopenharmony_ci#define WEBSOCKETSERVER_H 187c804472Sopenharmony_ci 197c804472Sopenharmony_ci#include <thread> 207c804472Sopenharmony_ci#include <csignal> 217c804472Sopenharmony_ci#include <mutex> 227c804472Sopenharmony_ci#include "libwebsockets.h" 237c804472Sopenharmony_ci 247c804472Sopenharmony_ciclass WebSocketServer { 257c804472Sopenharmony_cipublic: 267c804472Sopenharmony_ci WebSocketServer& operator=(const WebSocketServer&) = delete; 277c804472Sopenharmony_ci WebSocketServer(const WebSocketServer&) = delete; 287c804472Sopenharmony_ci static WebSocketServer& GetInstance(); 297c804472Sopenharmony_ci void SetServerPort(int port); 307c804472Sopenharmony_ci static int ProtocolCallback(struct lws* wsi, enum lws_callback_reasons reason, void* user, void* in, size_t len); 317c804472Sopenharmony_ci void StartWebsocketListening(); 327c804472Sopenharmony_ci void Run(); 337c804472Sopenharmony_ci size_t WriteData(unsigned char* data, size_t length); 347c804472Sopenharmony_ci enum class WebSocketState { INIT = -1, UNWRITEABLE = 0, WRITEABLE = 1 }; 357c804472Sopenharmony_ci static WebSocketState webSocketWritable; 367c804472Sopenharmony_ci static uint8_t* firstImageBuffer; 377c804472Sopenharmony_ci static uint64_t firstImagebufferSize; 387c804472Sopenharmony_ci std::mutex mutex; 397c804472Sopenharmony_ci 407c804472Sopenharmony_ciprivate: 417c804472Sopenharmony_ci WebSocketServer(); 427c804472Sopenharmony_ci virtual ~WebSocketServer(); 437c804472Sopenharmony_ci static void SignalHandler(int sig); 447c804472Sopenharmony_ci std::unique_ptr<std::thread> serverThread; 457c804472Sopenharmony_ci int serverPort; 467c804472Sopenharmony_ci const char* serverHostname = "127.0.0.1"; 477c804472Sopenharmony_ci int websocketMaxConn = 1024; 487c804472Sopenharmony_ci static lws* webSocket; 497c804472Sopenharmony_ci static std::atomic<bool> interrupted; 507c804472Sopenharmony_ci static const int MAX_PAYLOAD_SIZE = 6400000; 517c804472Sopenharmony_ci static const int WEBSOCKET_SERVER_TIMEOUT = 1000; 527c804472Sopenharmony_ci struct lws_protocols protocols[2]; 537c804472Sopenharmony_ci}; 547c804472Sopenharmony_ci 557c804472Sopenharmony_ci#endif // WEBSOCKETSERVER_H 56