You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

ptsocket.cpp 8.1KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328
  1. #include "ptsocket.h"
  2. PTSocket::PTSocket(QObject* p) : QTcpSocket(p)
  3. {
  4. #ifdef PT_DEBUG
  5. m_willPing = true;
  6. dbgSetWillHandshake(true);
  7. #endif
  8. setSocketOption(QAbstractSocket::LowDelayOption, 1);
  9. m_packetSize = 0;
  10. m_timeoutTimer.setSingleShot(true);
  11. setTimeout(5000);
  12. m_pingTimer.setSingleShot(true);
  13. setPingInterval(1000 * 7);
  14. m_state = Disconnected;
  15. m_isServerSide = false;
  16. connect(&m_timeoutTimer, SIGNAL(timeout()), this, SLOT(m_timedout()));
  17. connect(&m_pingTimer, SIGNAL(timeout()), this, SLOT(m_ping()));
  18. connect(this, SIGNAL(stateChanged(QAbstractSocket::SocketState)), this, SLOT(m_stateChanged(QAbstractSocket::SocketState)));
  19. connect(this, SIGNAL(error(QAbstractSocket::SocketError)), this, SLOT(m_socketError(QAbstractSocket::SocketError)));
  20. connect(this, SIGNAL(readyRead()), this, SLOT(m_readyRead()));
  21. }
  22. PTSocket::State PTSocket::getState() const
  23. {
  24. return m_state;
  25. }
  26. int PTSocket::getTimeout() const
  27. {
  28. return m_timeout;
  29. }
  30. int PTSocket::getPingInterval() const
  31. {
  32. return m_pingInterval;
  33. }
  34. bool PTSocket::isServerSide() const
  35. {
  36. return m_isServerSide;
  37. }
  38. QString PTSocket::getError() const
  39. {
  40. return (m_error.isEmpty()? errorString() : m_error);
  41. }
  42. QByteArray PTSocket::handshakeData()
  43. {
  44. return "LibPTSocketv1";
  45. }
  46. void PTSocket::setTimeout(int t)
  47. {
  48. m_timeout = t;
  49. if(m_timeoutTimer.isActive())
  50. m_timeoutTimer.start(m_timeout);
  51. else
  52. m_timeoutTimer.setInterval(m_timeout);
  53. }
  54. void PTSocket::setPingInterval(int p)
  55. {
  56. m_pingInterval = p;
  57. if(m_pingTimer.isActive())
  58. m_pingTimer.start(m_pingInterval);
  59. else
  60. m_pingTimer.setInterval(m_pingInterval);
  61. }
  62. void PTSocket::setIsServerSide(bool s)
  63. {
  64. m_isServerSide = s;
  65. }
  66. void PTSocket::send(int pktType, QByteArray data)
  67. {
  68. send(false, pktType, data);
  69. }
  70. void PTSocket::disconnectFromServer()
  71. {
  72. disconnectFromHost();
  73. }
  74. void PTSocket::dbgSetWillPing(bool w)
  75. {
  76. #ifdef PT_DEBUG
  77. m_willPing = w;
  78. #endif
  79. }
  80. void PTSocket::dbgSetWillHandshake(bool w)
  81. {
  82. #ifdef PT_DEBUG
  83. m_willHandshake = w;
  84. m_willHandshakeFake1 = false;
  85. m_willHandshakeFake2 = false;
  86. #endif
  87. }
  88. void PTSocket::dbgSetWillHandshakeFake1(bool w)
  89. {
  90. #ifdef PT_DEBUG
  91. m_willHandshakeFake1 = w;
  92. if(!w)
  93. m_willHandshake = true;
  94. else
  95. m_willHandshake = false;
  96. m_willHandshakeFake2 = false;
  97. #endif
  98. }
  99. void PTSocket::dbgSetWillHandshakeFake2(bool w)
  100. {
  101. #ifdef PT_DEBUG
  102. m_willHandshakeFake2 = w;
  103. if(!w)
  104. m_willHandshake = true;
  105. else
  106. m_willHandshake = false;
  107. m_willHandshakeFake1 = false;
  108. #endif
  109. }
  110. void PTSocket::send(bool sys, int pktType, QByteArray data)
  111. {
  112. if(data.isEmpty())
  113. data = "\0";
  114. data.prepend(intToByteArray(pktType));
  115. data.prepend((char)sys);
  116. data.prepend(intToByteArray(data.size()));
  117. write(data);
  118. }
  119. void PTSocket::m_ping()
  120. {
  121. m_pingTimer.stop();
  122. if(m_timeoutTimer.isActive())
  123. return;
  124. #ifdef PT_DEBUG
  125. if(m_willPing)
  126. send(true, PingRequest);
  127. #else
  128. send(true, PingRequest);
  129. #endif
  130. m_timeoutTimer.start();
  131. }
  132. void PTSocket::m_readyRead()
  133. {
  134. m_timeoutTimer.stop();
  135. m_pingTimer.start();
  136. if(m_state == Handshaked)
  137. {
  138. if(m_packetSize == 0)
  139. {
  140. if((unsigned)bytesAvailable() >= sizeof(int))
  141. m_packetSize = byteArrayToInt(read(sizeof(int)));
  142. else if(bytesAvailable() != 0)
  143. m_timeoutTimer.start();
  144. }
  145. if(m_packetSize != 0)
  146. {
  147. if((unsigned)bytesAvailable() >= m_packetSize)
  148. {
  149. bool sys = (bool)read(1).at(0);
  150. QByteArray p = read(sizeof(int));
  151. int pktType = byteArrayToInt(p);
  152. QByteArray data = read(m_packetSize - 1 - p.size());
  153. if(sys)
  154. {
  155. if(pktType == PingRequest)
  156. {
  157. #ifdef PT_DEBUG
  158. if(m_willPing)
  159. send(true, PingAnswer);
  160. #else
  161. send(true, PingAnswer);
  162. #endif
  163. }
  164. //else if(pktType == PingAnswer)
  165. }
  166. else
  167. emit packetReceived(pktType, data);
  168. m_packetSize = 0;
  169. m_readyRead();
  170. }
  171. else
  172. m_timeoutTimer.start();
  173. }
  174. }
  175. else
  176. {
  177. QByteArray handshake = handshakeData();
  178. if(bytesAvailable() >= handshake.size())
  179. {
  180. QByteArray data = read(handshake.size());
  181. if(data == handshake)
  182. {
  183. setState(Handshaked);
  184. m_readyRead();
  185. }
  186. else
  187. m_handshakeError();
  188. }
  189. else
  190. m_timeoutTimer.start();
  191. }
  192. }
  193. void PTSocket::m_timedout()
  194. {
  195. if(m_state == Disconnected)
  196. return;
  197. if(m_state == HostLookUp)
  198. setError("Unable to find the host " + peerName() + " in the time (" + QString::number(m_timeout) + " ms)");
  199. else if(m_state == Connecting)
  200. setError("The " + QString(isServerSide()? "client " + peerAddress().toString() : "server " + peerName()) + " didn't answer to the connection request in the time (" + QString::number(m_timeout) + " ms)");
  201. else if(m_state == Connected)
  202. setError("The " + QString(isServerSide()? "client " + peerAddress().toString() : "server " + peerName()) + " didn't answer to the handshake in the time (" + QString::number(m_timeout) + " ms)");
  203. else if(m_state == Handshaked)
  204. setError("The " + QString(isServerSide()? "client " + peerAddress().toString() : "server " + peerName()) + " timed out (" + QString::number(m_timeout) + " ms)");
  205. else if(m_state == Disconnecting)
  206. abort();
  207. }
  208. void PTSocket::m_handshakeError()
  209. {
  210. setError("The " + QString(isServerSide()? "client " + peerAddress().toString() : "server " + peerName()) + " didn't answer to the handshake correclty");
  211. }
  212. void PTSocket::m_stateChanged(QAbstractSocket::SocketState s)
  213. {
  214. if(s == QAbstractSocket::HostLookupState)
  215. setState(HostLookUp);
  216. else if(s == QAbstractSocket::ConnectingState)
  217. setState(Connecting);
  218. else if(s == QAbstractSocket::ConnectedState)
  219. {
  220. m_packetSize = 0;
  221. setState(Connected);
  222. #ifdef PT_DEBUG
  223. if(m_willHandshakeFake1)
  224. write("LibPTSocketv");
  225. else if(m_willHandshakeFake2)
  226. write("42LibPTSocketv4242");
  227. else if(m_willHandshake)
  228. write(handshakeData());
  229. #else
  230. write(handshakeData());
  231. #endif
  232. }
  233. else if(s == QAbstractSocket::ClosingState)
  234. setState(Disconnecting);
  235. m_timeoutTimer.start();
  236. if(s == QAbstractSocket::UnconnectedState)
  237. {
  238. if(error() == QAbstractSocket::UnknownSocketError || error() == QAbstractSocket::RemoteHostClosedError)
  239. {
  240. m_timeoutTimer.stop();
  241. setState(Disconnected);
  242. }
  243. }
  244. }
  245. void PTSocket::m_socketError(QAbstractSocket::SocketError s)
  246. {
  247. if(s != QAbstractSocket::RemoteHostClosedError)
  248. setState(Error);
  249. }
  250. void PTSocket::setState(PTSocket::State s)
  251. {
  252. if(s == m_state)
  253. return;
  254. if(s == Error && (m_state == Disconnected || m_state == Disconnecting))
  255. return;
  256. if(m_state == Error && (s == Disconnected || s == Disconnecting))
  257. return;
  258. m_state = s;
  259. if(s == Error)
  260. abort();
  261. else
  262. m_error.clear();
  263. if(s == Handshaked)
  264. m_pingTimer.start();
  265. else
  266. m_pingTimer.stop();
  267. emit stateChanged(m_state);
  268. }
  269. void PTSocket::setError(QString e)
  270. {
  271. if(m_state == Error || m_state == Disconnected || m_state == Disconnecting)
  272. return;
  273. m_error = e;
  274. setState(Error);
  275. }
  276. QByteArray PTSocket::intToByteArray(int a)
  277. {
  278. QByteArray data;
  279. for(unsigned char i = 0; i < sizeof(a); ++i)
  280. data.append((char)(a >> 8 * i));
  281. return data;
  282. }
  283. int PTSocket::byteArrayToInt(QByteArray d)
  284. {
  285. int a = 0;
  286. for(unsigned char i = 0; i < d.size(); ++i)
  287. a |= (unsigned int)((unsigned char)d.at(i) << i * 8);
  288. return a;
  289. }
  290. QDebug operator <<(QDebug dbg, const PTSocket::State s)
  291. {
  292. const QMetaObject & mo = PTSocket::staticMetaObject;
  293. QMetaEnum me = mo.enumerator(mo.indexOfEnumerator("State"));
  294. return dbg<<me.valueToKey(s);
  295. }