1 #pragma once 2 3 /* 4 * Copyright (C) 2017 The Android Open Source Project 5 * 6 * Licensed under the Apache License, Version 2.0 (the "License"); 7 * you may not use this file except in compliance with the License. 8 * You may obtain a copy of the License at 9 * 10 * http://www.apache.org/licenses/LICENSE-2.0 11 * 12 * Unless required by applicable law or agreed to in writing, software 13 * distributed under the License is distributed on an "AS IS" BASIS, 14 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 15 * See the License for the specific language governing permissions and 16 * limitations under the License. 17 */ 18 19 20 #include <condition_variable> 21 #include <memory> 22 #include <mutex> 23 #include <unordered_map> 24 25 #include "common/libs/threads/thread_annotations.h" 26 #include "host/frontend/vnc_server/vnc_utils.h" 27 28 namespace cvd { 29 namespace vnc { 30 31 class VncClientConnection; 32 class FrameBufferWatcher; 33 using StripePtrVec = std::vector<std::shared_ptr<const Stripe>>; 34 using SeqNumberVec = std::vector<StripeSeqNumber>; 35 36 SeqNumberVec MakeSeqNumberVec(); 37 38 class BlackBoard { 39 private: 40 struct ClientFBUState { 41 bool ready_to_receive{}; 42 ScreenOrientation orientation{}; 43 std::condition_variable new_frame_cv; 44 SeqNumberVec stripe_seq_nums = MakeSeqNumberVec(); 45 bool closed{}; 46 }; 47 48 public: 49 class Registerer { 50 public: Registerer(BlackBoard * bb,const VncClientConnection * conn)51 Registerer(BlackBoard* bb, const VncClientConnection* conn) 52 : bb_{bb}, conn_{conn} { 53 bb->Register(conn); 54 } ~Registerer()55 ~Registerer() { bb_->Unregister(conn_); } 56 Registerer(const Registerer&) = delete; 57 Registerer& operator=(const Registerer&) = delete; 58 59 private: 60 BlackBoard* bb_{}; 61 const VncClientConnection* conn_{}; 62 }; 63 64 BlackBoard() = default; 65 BlackBoard(const BlackBoard&) = delete; 66 BlackBoard& operator=(const BlackBoard&) = delete; 67 68 bool NoNewStripesFor(const SeqNumberVec& seq_nums) const REQUIRES(m_); 69 void NewStripeReady(int index, StripeSeqNumber seq_num); 70 void Register(const VncClientConnection* conn); 71 void Unregister(const VncClientConnection* conn); 72 73 StripePtrVec WaitForSenderWork(const VncClientConnection* conn); 74 75 void WaitForAtLeastOneClientConnection(); 76 77 void FrameBufferUpdateRequestReceived(const VncClientConnection* conn); 78 // Setting orientation implies needing the entire screen 79 void SetOrientation(const VncClientConnection* conn, 80 ScreenOrientation orientation); 81 void SignalClientNeedsEntireScreen(const VncClientConnection* conn); 82 83 void StopWaiting(const VncClientConnection* conn); 84 85 void set_frame_buffer_watcher(FrameBufferWatcher* frame_buffer_watcher); 86 87 // quality_level must be the value received from the client, in the range 88 // [kJpegMinQualityEncoding, kJpegMaxQualityEncoding], else it is ignored. 89 void set_jpeg_quality_level(int quality_level); 90 jpeg_quality_level()91 int jpeg_quality_level() const { 92 std::lock_guard<std::mutex> guard(m_); 93 return jpeg_quality_level_; 94 } 95 96 private: 97 ClientFBUState& GetStateForClient(const VncClientConnection* conn) 98 REQUIRES(m_); 99 static void ResetToZero(SeqNumberVec* seq_nums); 100 101 mutable std::mutex m_; 102 SeqNumberVec most_recent_stripe_seq_nums_ GUARDED_BY(m_) = MakeSeqNumberVec(); 103 std::unordered_map<const VncClientConnection*, ClientFBUState> clients_ 104 GUARDED_BY(m_); 105 int jpeg_quality_level_ GUARDED_BY(m_) = 100; 106 std::condition_variable new_client_cv_; 107 // NOTE the FrameBufferWatcher pointer itself should be 108 // guarded, but not the pointee. GUARDED_BY(m_)109 FrameBufferWatcher* frame_buffer_watcher_ GUARDED_BY(m_){}; 110 }; 111 112 } // namespace vnc 113 } // namespace cvd 114