1#ifndef _XETCPIPLINK_HPP 2#define _XETCPIPLINK_HPP 3/*------------------------------------------------------------------------- 4 * drawElements Quality Program Test Executor 5 * ------------------------------------------ 6 * 7 * Copyright 2014 The Android Open Source Project 8 * 9 * Licensed under the Apache License, Version 2.0 (the "License"); 10 * you may not use this file except in compliance with the License. 11 * You may obtain a copy of the License at 12 * 13 * http://www.apache.org/licenses/LICENSE-2.0 14 * 15 * Unless required by applicable law or agreed to in writing, software 16 * distributed under the License is distributed on an "AS IS" BASIS, 17 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 18 * See the License for the specific language governing permissions and 19 * limitations under the License. 20 * 21 *//*! 22 * \file 23 * \brief Tcp/Ip communication link. 24 *//*--------------------------------------------------------------------*/ 25 26#include "xeDefs.hpp" 27#include "xeCommLink.hpp" 28#include "deSocket.hpp" 29#include "deRingBuffer.hpp" 30#include "deBlockBuffer.hpp" 31#include "xsProtocol.hpp" 32#include "deThread.hpp" 33#include "deTimer.h" 34 35#include <vector> 36 37namespace xe 38{ 39 40class TcpIpLinkState 41{ 42public: 43 TcpIpLinkState (CommLinkState initialState, const char* initialErr); 44 ~TcpIpLinkState (void); 45 46 CommLinkState getState (void) const; 47 CommLinkState getState (std::string& error) const; 48 49 void setCallbacks (CommLink::StateChangedFunc stateChangedCallback, CommLink::LogDataFunc testLogDataCallback, CommLink::LogDataFunc infoLogDataCallback, void* userPtr); 50 51 void setState (CommLinkState state, const char* error = ""); 52 void onTestLogData (const deUint8* bytes, size_t numBytes) const; 53 void onInfoLogData (const deUint8* bytes, size_t numBytes) const; 54 55 void onKeepaliveReceived (void); 56 deUint64 getLastKeepaliveRecevied (void) const; 57 58private: 59 mutable de::Mutex m_lock; 60 volatile CommLinkState m_state; 61 std::string m_error; 62 63 volatile deUint64 m_lastKeepaliveReceived; 64 65 volatile CommLink::StateChangedFunc m_stateChangedCallback; 66 volatile CommLink::LogDataFunc m_testLogDataCallback; 67 volatile CommLink::LogDataFunc m_infoLogDataCallback; 68 void* volatile m_userPtr; 69}; 70 71class TcpIpSendThread : public de::Thread 72{ 73public: 74 TcpIpSendThread (de::Socket& socket, TcpIpLinkState& state); 75 ~TcpIpSendThread (void); 76 77 void start (void); 78 void run (void); 79 void stop (void); 80 81 bool isRunning (void) const { return m_isRunning; } 82 83 de::BlockBuffer<deUint8>& getBuffer (void) { return m_buffer; } 84 85private: 86 de::Socket& m_socket; 87 TcpIpLinkState& m_state; 88 89 de::BlockBuffer<deUint8> m_buffer; 90 91 bool m_isRunning; 92}; 93 94class TcpIpRecvThread : public de::Thread 95{ 96public: 97 TcpIpRecvThread (de::Socket& socket, TcpIpLinkState& state); 98 ~TcpIpRecvThread (void); 99 100 void start (void); 101 void run (void); 102 void stop (void); 103 104 bool isRunning (void) const { return m_isRunning; } 105 106private: 107 void handleMessage (xs::MessageType messageType, const deUint8* data, size_t dataSize); 108 109 de::Socket& m_socket; 110 TcpIpLinkState& m_state; 111 112 std::vector<deUint8> m_curMsgBuf; 113 size_t m_curMsgPos; 114 115 bool m_isRunning; 116}; 117 118class TcpIpLink : public CommLink 119{ 120public: 121 TcpIpLink (void); 122 ~TcpIpLink (void); 123 124 // TcpIpLink -specific API 125 void connect (const de::SocketAddress& address); 126 void disconnect (void); 127 128 // CommLink API 129 void reset (void); 130 131 CommLinkState getState (void) const; 132 CommLinkState getState (std::string& error) const; 133 134 void setCallbacks (StateChangedFunc stateChangedCallback, LogDataFunc testLogDataCallback, LogDataFunc infoLogDataCallback, void* userPtr); 135 136 void startTestProcess (const char* name, const char* params, const char* workingDir, const char* caseList); 137 void stopTestProcess (void); 138 139private: 140 void closeConnection (void); 141 142 static void keepaliveTimerCallback (void* ptr); 143 144 de::Socket m_socket; 145 TcpIpLinkState m_state; 146 147 TcpIpSendThread m_sendThread; 148 TcpIpRecvThread m_recvThread; 149 150 deTimer* m_keepaliveTimer; 151}; 152 153} // xe 154 155#endif // _XETCPIPLINK_HPP 156