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, int numBytes) const;
53	void						onInfoLogData				(const deUint8* bytes, int 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, int dataSize);
108
109	de::Socket&					m_socket;
110	TcpIpLinkState&				m_state;
111
112	std::vector<deUint8>		m_curMsgBuf;
113	int							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