SocketCore.cc 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755
  1. /* <!-- copyright */
  2. /*
  3. * aria2 - The high speed download utility
  4. *
  5. * Copyright (C) 2006 Tatsuhiro Tsujikawa
  6. *
  7. * This program is free software; you can redistribute it and/or modify
  8. * it under the terms of the GNU General Public License as published by
  9. * the Free Software Foundation; either version 2 of the License, or
  10. * (at your option) any later version.
  11. *
  12. * This program is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  15. * GNU General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU General Public License
  18. * along with this program; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. *
  21. * In addition, as a special exception, the copyright holders give
  22. * permission to link the code of portions of this program with the
  23. * OpenSSL library under certain conditions as described in each
  24. * individual source file, and distribute linked combinations
  25. * including the two.
  26. * You must obey the GNU General Public License in all respects
  27. * for all of the code used other than OpenSSL. If you modify
  28. * file(s) with this exception, you may extend this exception to your
  29. * version of the file(s), but you are not obligated to do so. If you
  30. * do not wish to do so, delete this exception statement from your
  31. * version. If you delete this exception statement from all source
  32. * files in the program, then also delete it here.
  33. */
  34. /* copyright --> */
  35. #include "SocketCore.h"
  36. #include "message.h"
  37. #include "a2netcompat.h"
  38. #include "DlRetryEx.h"
  39. #include "DlAbortEx.h"
  40. #include "StringFormat.h"
  41. #include "Util.h"
  42. #include <unistd.h>
  43. #include <cerrno>
  44. #include <cstring>
  45. #ifndef __MINGW32__
  46. # define SOCKET_ERRNO (errno)
  47. #else
  48. # define SOCKET_ERRNO (WSAGetLastError())
  49. #endif // __MINGW32__
  50. #ifdef __MINGW32__
  51. # define A2_EINPROGRESS WSAEWOULDBLOCK
  52. #else
  53. # define A2_EINPROGRESS EINPROGRESS
  54. #endif // __MINGW32__
  55. #ifdef __MINGW32__
  56. # define CLOSE(X) ::closesocket(sockfd)
  57. #else
  58. # define CLOSE(X) while(close(X) == -1 && errno == EINTR)
  59. #endif // __MINGW32__
  60. namespace aria2 {
  61. SocketCore::SocketCore(int sockType):_sockType(sockType), sockfd(-1) {
  62. init();
  63. }
  64. SocketCore::SocketCore(sock_t sockfd, int sockType):_sockType(sockType), sockfd(sockfd) {
  65. init();
  66. }
  67. void SocketCore::init()
  68. {
  69. #ifdef HAVE_EPOLL
  70. _epfd = -1;
  71. #endif // HAVE_EPOLL
  72. blocking = true;
  73. secure = false;
  74. #ifdef HAVE_LIBSSL
  75. // for SSL
  76. sslCtx = NULL;
  77. ssl = NULL;
  78. #endif // HAVE_LIBSSL
  79. #ifdef HAVE_LIBGNUTLS
  80. sslSession = NULL;
  81. sslXcred = NULL;
  82. peekBufMax = 4096;
  83. peekBuf = 0;
  84. peekBufLength = 0;
  85. #endif //HAVE_LIBGNUTLS
  86. }
  87. SocketCore::~SocketCore() {
  88. closeConnection();
  89. #ifdef HAVE_EPOLL
  90. if(_epfd != -1) {
  91. CLOSE(_epfd);
  92. }
  93. #endif // HAVE_EPOLL
  94. #ifdef HAVE_LIBGNUTLS
  95. delete [] peekBuf;
  96. #endif // HAVE_LIBGNUTLS
  97. }
  98. template<typename T>
  99. std::string uitos(T value)
  100. {
  101. std::string str;
  102. if(value == 0) {
  103. str = "0";
  104. return str;
  105. }
  106. while(value) {
  107. char digit = value%10+'0';
  108. str.insert(str.begin(), digit);
  109. value /= 10;
  110. }
  111. return str;
  112. }
  113. void SocketCore::bind(uint16_t port)
  114. {
  115. closeConnection();
  116. struct addrinfo hints;
  117. struct addrinfo* res;
  118. memset(&hints, 0, sizeof(hints));
  119. hints.ai_family = AF_UNSPEC;
  120. hints.ai_socktype = _sockType;
  121. hints.ai_flags = AI_PASSIVE;
  122. hints.ai_protocol = 0;
  123. int s;
  124. s = getaddrinfo(0, uitos(port).c_str(), &hints, &res);
  125. if(s) {
  126. throw DlAbortEx(StringFormat(EX_SOCKET_BIND, gai_strerror(s)).str());
  127. }
  128. struct addrinfo* rp;
  129. for(rp = res; rp; rp = rp->ai_next) {
  130. sock_t fd = socket(rp->ai_family, rp->ai_socktype, rp->ai_protocol);
  131. if(fd == -1) {
  132. continue;
  133. }
  134. SOCKOPT_T sockopt = 1;
  135. if(setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &sockopt, sizeof(socklen_t)) < 0) {
  136. CLOSE(fd);
  137. continue;
  138. }
  139. if(::bind(fd, rp->ai_addr, rp->ai_addrlen) == -1) {
  140. CLOSE(fd);
  141. continue;
  142. }
  143. sockfd = fd;
  144. break;
  145. }
  146. freeaddrinfo(res);
  147. if(sockfd == -1) {
  148. throw DlAbortEx(StringFormat(EX_SOCKET_BIND, "all addresses failed").str());
  149. }
  150. }
  151. void SocketCore::beginListen()
  152. {
  153. if(listen(sockfd, 1) == -1) {
  154. throw DlAbortEx(StringFormat(EX_SOCKET_LISTEN, errorMsg()).str());
  155. }
  156. }
  157. SocketCore* SocketCore::acceptConnection() const
  158. {
  159. struct sockaddr_storage sockaddr;
  160. socklen_t len = sizeof(sockaddr);
  161. sock_t fd;
  162. while((fd = accept(sockfd, reinterpret_cast<struct sockaddr*>(&sockaddr), &len)) == -1 && errno == EINTR);
  163. if(fd == -1) {
  164. throw DlAbortEx(StringFormat(EX_SOCKET_ACCEPT, errorMsg()).str());
  165. }
  166. return new SocketCore(fd, _sockType);
  167. }
  168. void SocketCore::getAddrInfo(std::pair<std::string, uint16_t>& addrinfo) const
  169. {
  170. struct sockaddr_storage sockaddr;
  171. socklen_t len = sizeof(sockaddr);
  172. struct sockaddr* addrp = reinterpret_cast<struct sockaddr*>(&sockaddr);
  173. if(getsockname(sockfd, addrp, &len) == -1) {
  174. throw DlAbortEx(StringFormat(EX_SOCKET_GET_NAME, errorMsg()).str());
  175. }
  176. addrinfo = Util::getNumericNameInfo(addrp, len);
  177. }
  178. void SocketCore::getPeerInfo(std::pair<std::string, uint16_t>& peerinfo) const
  179. {
  180. struct sockaddr_storage sockaddr;
  181. socklen_t len = sizeof(sockaddr);
  182. struct sockaddr* addrp = reinterpret_cast<struct sockaddr*>(&sockaddr);
  183. if(getpeername(sockfd, addrp, &len) == -1) {
  184. throw DlAbortEx(StringFormat(EX_SOCKET_GET_NAME, errorMsg()).str());
  185. }
  186. peerinfo = Util::getNumericNameInfo(addrp, len);
  187. }
  188. void SocketCore::establishConnection(const std::string& host, uint16_t port)
  189. {
  190. closeConnection();
  191. struct addrinfo hints;
  192. struct addrinfo* res;
  193. memset(&hints, 0, sizeof(hints));
  194. hints.ai_family = AF_UNSPEC;
  195. hints.ai_socktype = _sockType;
  196. hints.ai_flags = 0;
  197. hints.ai_protocol = 0;
  198. int s;
  199. s = getaddrinfo(host.c_str(), uitos(port).c_str(), &hints, &res);
  200. if(s) {
  201. throw DlAbortEx(StringFormat(EX_RESOLVE_HOSTNAME,
  202. host.c_str(), gai_strerror(s)).str());
  203. }
  204. struct addrinfo* rp;
  205. for(rp = res; rp; rp = rp->ai_next) {
  206. sock_t fd = socket(rp->ai_family, rp->ai_socktype, rp->ai_protocol);
  207. if(fd == -1) {
  208. continue;
  209. }
  210. SOCKOPT_T sockopt = 1;
  211. if(setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &sockopt, sizeof(socklen_t)) < 0) {
  212. CLOSE(fd);
  213. continue;
  214. }
  215. sockfd = fd;
  216. // make socket non-blocking mode
  217. setNonBlockingMode();
  218. if(connect(fd, rp->ai_addr, rp->ai_addrlen) == -1 &&
  219. SOCKET_ERRNO != A2_EINPROGRESS) {
  220. CLOSE(sockfd);
  221. sockfd = -1;
  222. continue;
  223. }
  224. // TODO at this point, connection may not be established and it may fail
  225. // later. In such case, next ai_addr should be tried.
  226. break;
  227. }
  228. freeaddrinfo(res);
  229. if(sockfd == -1) {
  230. throw DlAbortEx(StringFormat(EX_SOCKET_CONNECT, host.c_str(),
  231. "all addresses failed").str());
  232. }
  233. }
  234. void SocketCore::setNonBlockingMode()
  235. {
  236. #ifdef __MINGW32__
  237. static u_long flag = 1;
  238. if (::ioctlsocket(sockfd, FIONBIO, &flag) == -1) {
  239. throw DlAbortEx(StringFormat(EX_SOCKET_NONBLOCKING, errorMsg()).str());
  240. }
  241. #else
  242. int flags;
  243. while((flags = fcntl(sockfd, F_GETFL, 0)) == -1 && errno == EINTR);
  244. // TODO add error handling
  245. while(fcntl(sockfd, F_SETFL, flags|O_NONBLOCK) == -1 && errno == EINTR);
  246. #endif // __MINGW32__
  247. blocking = false;
  248. }
  249. void SocketCore::setBlockingMode()
  250. {
  251. #ifdef __MINGW32__
  252. static u_long flag = 0;
  253. if (::ioctlsocket(sockfd, FIONBIO, &flag) == -1) {
  254. throw DlAbortEx(StringFormat(EX_SOCKET_BLOCKING, errorMsg()).str());
  255. }
  256. #else
  257. int flags;
  258. while((flags = fcntl(sockfd, F_GETFL, 0)) == -1 && errno == EINTR);
  259. // TODO add error handling
  260. while(fcntl(sockfd, F_SETFL, flags&(~O_NONBLOCK)) == -1 && errno == EINTR);
  261. #endif // __MINGW32__
  262. blocking = true;
  263. }
  264. void SocketCore::closeConnection()
  265. {
  266. #ifdef HAVE_LIBSSL
  267. // for SSL
  268. if(secure) {
  269. SSL_shutdown(ssl);
  270. }
  271. #endif // HAVE_LIBSSL
  272. #ifdef HAVE_LIBGNUTLS
  273. if(secure) {
  274. gnutls_bye(sslSession, GNUTLS_SHUT_RDWR);
  275. }
  276. #endif // HAVE_LIBGNUTLS
  277. if(sockfd != -1) {
  278. CLOSE(sockfd);
  279. sockfd = -1;
  280. }
  281. #ifdef HAVE_LIBSSL
  282. // for SSL
  283. if(secure) {
  284. SSL_free(ssl);
  285. SSL_CTX_free(sslCtx);
  286. }
  287. #endif // HAVE_LIBSSL
  288. #ifdef HAVE_LIBGNUTLS
  289. if(secure) {
  290. gnutls_deinit(sslSession);
  291. gnutls_certificate_free_credentials(sslXcred);
  292. }
  293. #endif // HAVE_LIBGNUTLS
  294. }
  295. #ifdef HAVE_EPOLL
  296. void SocketCore::initEPOLL()
  297. {
  298. if((_epfd = epoll_create(1)) == -1) {
  299. throw DlRetryEx(StringFormat("epoll_create failed:%s", errorMsg()).str());
  300. }
  301. memset(&_epEvent, 0, sizeof(struct epoll_event));
  302. _epEvent.events = EPOLLIN|EPOLLOUT;
  303. _epEvent.data.fd = sockfd;
  304. if(epoll_ctl(_epfd, EPOLL_CTL_ADD, sockfd, &_epEvent) == -1) {
  305. throw DlRetryEx(StringFormat("epoll_ctl failed:%s", errorMsg()).str());
  306. }
  307. }
  308. #endif // HAVE_EPOLL
  309. bool SocketCore::isWritable(time_t timeout)
  310. {
  311. #ifdef HAVE_EPOLL
  312. if(_epfd == -1) {
  313. initEPOLL();
  314. }
  315. struct epoll_event epEvents[1];
  316. int r;
  317. while((r = epoll_wait(_epfd, epEvents, 1, 0)) == -1 && errno == EINTR);
  318. if(r > 0) {
  319. return epEvents[0].events&(EPOLLOUT|EPOLLHUP|EPOLLERR);
  320. } else if(r == 0) {
  321. return false;
  322. } else {
  323. throw DlRetryEx(StringFormat(EX_SOCKET_CHECK_WRITABLE, errorMsg()).str());
  324. }
  325. #else // !HAVE_EPOLL
  326. fd_set fds;
  327. FD_ZERO(&fds);
  328. FD_SET(sockfd, &fds);
  329. struct timeval tv;
  330. tv.tv_sec = timeout;
  331. tv.tv_usec = 0;
  332. int r = select(sockfd+1, NULL, &fds, NULL, &tv);
  333. if(r == 1) {
  334. return true;
  335. } else if(r == 0) {
  336. // time out
  337. return false;
  338. } else {
  339. if(SOCKET_ERRNO == EINPROGRESS || SOCKET_ERRNO == EINTR) {
  340. return false;
  341. } else {
  342. throw DlRetryEx(StringFormat(EX_SOCKET_CHECK_WRITABLE, errorMsg()).str());
  343. }
  344. }
  345. #endif // !HAVE_EPOLL
  346. }
  347. bool SocketCore::isReadable(time_t timeout)
  348. {
  349. #ifdef HAVE_LIBGNUTLS
  350. if(secure && peekBufLength > 0) {
  351. return true;
  352. }
  353. #endif // HAVE_LIBGNUTLS
  354. #ifdef HAVE_EPOLL
  355. if(_epfd == -1) {
  356. initEPOLL();
  357. }
  358. struct epoll_event epEvents[1];
  359. int r;
  360. while((r = epoll_wait(_epfd, epEvents, 1, 0)) == -1 && errno == EINTR);
  361. if(r > 0) {
  362. return epEvents[0].events&(EPOLLIN|EPOLLHUP|EPOLLERR);
  363. } else if(r == 0) {
  364. return false;
  365. } else {
  366. throw DlRetryEx(StringFormat(EX_SOCKET_CHECK_READABLE, errorMsg()).str());
  367. }
  368. #else // !HAVE_EPOLL
  369. fd_set fds;
  370. FD_ZERO(&fds);
  371. FD_SET(sockfd, &fds);
  372. struct timeval tv;
  373. tv.tv_sec = timeout;
  374. tv.tv_usec = 0;
  375. int r = select(sockfd+1, &fds, NULL, NULL, &tv);
  376. if(r == 1) {
  377. return true;
  378. } else if(r == 0) {
  379. // time out
  380. return false;
  381. } else {
  382. if(SOCKET_ERRNO == EINPROGRESS || SOCKET_ERRNO == EINTR) {
  383. return false;
  384. } else {
  385. throw DlRetryEx(StringFormat(EX_SOCKET_CHECK_READABLE, errorMsg()).str());
  386. }
  387. }
  388. #endif // !HAVE_EPOLL
  389. }
  390. void SocketCore::writeData(const char* data, size_t len)
  391. {
  392. ssize_t ret = 0;
  393. if(!secure) {
  394. while((ret = send(sockfd, data, len, 0)) == -1 && errno == EINTR);
  395. // TODO assuming Blocking mode.
  396. if(ret == -1 || (size_t)ret != len) {
  397. throw DlRetryEx(StringFormat(EX_SOCKET_SEND, errorMsg()).str());
  398. }
  399. } else {
  400. #ifdef HAVE_LIBSSL
  401. // for SSL
  402. // TODO handling len == 0 case required
  403. ret = SSL_write(ssl, data, len);
  404. if(ret <= 0 || (size_t)ret != len) {
  405. throw DlRetryEx(StringFormat(EX_SOCKET_SEND, ERR_error_string(ERR_get_error(), NULL)).str());
  406. }
  407. #endif // HAVE_LIBSSL
  408. #ifdef HAVE_LIBGNUTLS
  409. ret = gnutls_record_send(sslSession, data, len);
  410. if(ret < 0 || (size_t)ret != len) {
  411. throw DlRetryEx(StringFormat(EX_SOCKET_SEND, gnutls_strerror(ret)).str());
  412. }
  413. #endif // HAVE_LIBGNUTLS
  414. }
  415. }
  416. void SocketCore::readData(char* data, size_t& len)
  417. {
  418. ssize_t ret = 0;
  419. if(!secure) {
  420. while((ret = recv(sockfd, data, len, 0)) == -1 && errno == EINTR);
  421. if(ret == -1) {
  422. throw DlRetryEx(StringFormat(EX_SOCKET_RECV, errorMsg()).str());
  423. }
  424. } else {
  425. #ifdef HAVE_LIBSSL
  426. // for SSL
  427. // TODO handling len == 0 case required
  428. if ((ret = SSL_read(ssl, data, len)) <= 0) {
  429. throw DlRetryEx
  430. (StringFormat(EX_SOCKET_RECV,
  431. ERR_error_string(ERR_get_error(), 0)).str());
  432. }
  433. #endif // HAVE_LIBSSL
  434. #ifdef HAVE_LIBGNUTLS
  435. if ((ret = gnutlsRecv(data, len)) < 0) {
  436. throw DlRetryEx
  437. (StringFormat(EX_SOCKET_RECV, gnutls_strerror(ret)).str());
  438. }
  439. #endif // HAVE_LIBGNUTLS
  440. }
  441. len = ret;
  442. }
  443. void SocketCore::peekData(char* data, size_t& len)
  444. {
  445. ssize_t ret = 0;
  446. if(!secure) {
  447. while((ret = recv(sockfd, data, len, MSG_PEEK)) == -1 && errno == EINTR);
  448. if(ret == -1) {
  449. throw DlRetryEx(StringFormat(EX_SOCKET_PEEK, errorMsg()).str());
  450. }
  451. } else {
  452. #ifdef HAVE_LIBSSL
  453. // for SSL
  454. // TODO handling len == 0 case required
  455. if ((ret = SSL_peek(ssl, data, len)) < 0) {
  456. throw DlRetryEx
  457. (StringFormat(EX_SOCKET_PEEK,
  458. ERR_error_string(ERR_get_error(), 0)).str());
  459. }
  460. #endif // HAVE_LIBSSL
  461. #ifdef HAVE_LIBGNUTLS
  462. if ((ret = gnutlsPeek(data, len)) < 0) {
  463. throw DlRetryEx(StringFormat(EX_SOCKET_PEEK,
  464. gnutls_strerror(ret)).str());
  465. }
  466. #endif // HAVE_LIBGNUTLS
  467. }
  468. len = ret;
  469. }
  470. #ifdef HAVE_LIBGNUTLS
  471. size_t SocketCore::shiftPeekData(char* data, size_t len)
  472. {
  473. if(peekBufLength <= len) {
  474. memcpy(data, peekBuf, peekBufLength);
  475. size_t ret = peekBufLength;
  476. peekBufLength = 0;
  477. return ret;
  478. } else {
  479. memcpy(data, peekBuf, len);
  480. char* temp = new char[peekBufMax];
  481. memcpy(temp, peekBuf+len, peekBufLength-len);
  482. delete [] peekBuf;
  483. peekBuf = temp;
  484. peekBufLength -= len;
  485. return len;
  486. }
  487. }
  488. void SocketCore::addPeekData(char* data, size_t len)
  489. {
  490. if(peekBufLength+len > peekBufMax) {
  491. char* temp = new char[peekBufMax+len];
  492. memcpy(temp, peekBuf, peekBufLength);
  493. delete [] peekBuf;
  494. peekBuf = temp;
  495. peekBufMax = peekBufLength+len;
  496. }
  497. memcpy(peekBuf+peekBufLength, data, len);
  498. peekBufLength += len;
  499. }
  500. ssize_t SocketCore::gnutlsRecv(char* data, size_t len)
  501. {
  502. size_t plen = shiftPeekData(data, len);
  503. if(plen < len) {
  504. ssize_t ret = gnutls_record_recv(sslSession, data+plen, len-plen);
  505. if(ret < 0) {
  506. throw DlRetryEx(StringFormat(EX_SOCKET_RECV, gnutls_strerror(ret)).str());
  507. }
  508. return plen+ret;
  509. } else {
  510. return plen;
  511. }
  512. }
  513. ssize_t SocketCore::gnutlsPeek(char* data, size_t len)
  514. {
  515. if(peekBufLength >= len) {
  516. memcpy(data, peekBuf, len);
  517. return len;
  518. } else {
  519. memcpy(data, peekBuf, peekBufLength);
  520. ssize_t ret = gnutls_record_recv(sslSession, data+peekBufLength, len-peekBufLength);
  521. if(ret < 0) {
  522. throw DlRetryEx(StringFormat(EX_SOCKET_PEEK, gnutls_strerror(ret)).str());
  523. }
  524. addPeekData(data+peekBufLength, ret);
  525. return peekBufLength;
  526. }
  527. }
  528. #endif // HAVE_LIBGNUTLS
  529. void SocketCore::initiateSecureConnection()
  530. {
  531. #ifdef HAVE_LIBSSL
  532. // for SSL
  533. if(!secure) {
  534. sslCtx = SSL_CTX_new(SSLv23_client_method());
  535. if(sslCtx == NULL) {
  536. throw DlAbortEx
  537. (StringFormat(EX_SSL_INIT_FAILURE,
  538. ERR_error_string(ERR_get_error(), 0)).str());
  539. }
  540. SSL_CTX_set_mode(sslCtx, SSL_MODE_AUTO_RETRY);
  541. ssl = SSL_new(sslCtx);
  542. if(ssl == NULL) {
  543. throw DlAbortEx
  544. (StringFormat(EX_SSL_INIT_FAILURE,
  545. ERR_error_string(ERR_get_error(), 0)).str());
  546. }
  547. if(SSL_set_fd(ssl, sockfd) == 0) {
  548. throw DlAbortEx
  549. (StringFormat(EX_SSL_INIT_FAILURE,
  550. ERR_error_string(ERR_get_error(), 0)).str());
  551. }
  552. // TODO handling return value == 0 case required
  553. int e = SSL_connect(ssl);
  554. if (e <= 0) {
  555. int ssl_error = SSL_get_error(ssl, e);
  556. switch(ssl_error) {
  557. case SSL_ERROR_NONE:
  558. break;
  559. case SSL_ERROR_WANT_READ:
  560. case SSL_ERROR_WANT_WRITE:
  561. case SSL_ERROR_WANT_X509_LOOKUP:
  562. case SSL_ERROR_ZERO_RETURN:
  563. if (blocking) {
  564. throw DlAbortEx
  565. (StringFormat(EX_SSL_CONNECT_ERROR, ssl_error).str());
  566. }
  567. break;
  568. case SSL_ERROR_SYSCALL:
  569. throw DlAbortEx(EX_SSL_IO_ERROR);
  570. case SSL_ERROR_SSL:
  571. throw DlAbortEx(EX_SSL_PROTOCOL_ERROR);
  572. default:
  573. throw DlAbortEx
  574. (StringFormat(EX_SSL_UNKNOWN_ERROR, ssl_error).str());
  575. }
  576. }
  577. }
  578. #endif // HAVE_LIBSSL
  579. #ifdef HAVE_LIBGNUTLS
  580. if(!secure) {
  581. const int cert_type_priority[3] = { GNUTLS_CRT_X509,
  582. GNUTLS_CRT_OPENPGP, 0
  583. };
  584. // while we do not support X509 certificate, most web servers require
  585. // X509 stuff.
  586. gnutls_certificate_allocate_credentials (&sslXcred);
  587. gnutls_init(&sslSession, GNUTLS_CLIENT);
  588. gnutls_set_default_priority(sslSession);
  589. gnutls_kx_set_priority(sslSession, cert_type_priority);
  590. // put the x509 credentials to the current session
  591. gnutls_credentials_set(sslSession, GNUTLS_CRD_CERTIFICATE, sslXcred);
  592. gnutls_transport_set_ptr(sslSession, (gnutls_transport_ptr_t)sockfd);
  593. int ret = gnutls_handshake(sslSession);
  594. if(ret < 0) {
  595. throw DlAbortEx
  596. (StringFormat(EX_SSL_INIT_FAILURE, gnutls_strerror(ret)).str());
  597. }
  598. peekBuf = new char[peekBufMax];
  599. }
  600. #endif // HAVE_LIBGNUTLS
  601. secure = true;
  602. }
  603. /* static */ int SocketCore::error()
  604. {
  605. return SOCKET_ERRNO;
  606. }
  607. /* static */ const char *SocketCore::errorMsg()
  608. {
  609. return errorMsg(SOCKET_ERRNO);
  610. }
  611. /* static */ const char *SocketCore::errorMsg(const int err)
  612. {
  613. #ifndef __MINGW32__
  614. return strerror(err);
  615. #else
  616. static char buf[256];
  617. if (FormatMessage(
  618. FORMAT_MESSAGE_FROM_SYSTEM |
  619. FORMAT_MESSAGE_IGNORE_INSERTS,
  620. NULL,
  621. err,
  622. MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), // Default language
  623. (LPTSTR) &buf,
  624. sizeof(buf),
  625. NULL
  626. ) == 0) {
  627. snprintf(buf, sizeof(buf), EX_SOCKET_UNKNOWN_ERROR, err, err);
  628. }
  629. return buf;
  630. #endif // __MINGW32__
  631. }
  632. void SocketCore::writeData(const char* data, size_t len, const std::string& host, uint16_t port)
  633. {
  634. struct addrinfo hints;
  635. struct addrinfo* res;
  636. memset(&hints, 0, sizeof(hints));
  637. hints.ai_family = AF_UNSPEC;
  638. hints.ai_socktype = _sockType;
  639. hints.ai_flags = 0;
  640. hints.ai_protocol = 0;
  641. int s;
  642. s = getaddrinfo(host.c_str(), uitos(port).c_str(), &hints, &res);
  643. if(s) {
  644. throw DlAbortEx(StringFormat(EX_SOCKET_SEND, gai_strerror(s)).str());
  645. }
  646. struct addrinfo* rp;
  647. ssize_t r = -1;
  648. for(rp = res; rp; rp = rp->ai_next) {
  649. while((r = sendto(sockfd, data, len, 0, rp->ai_addr, rp->ai_addrlen)) == -1 && EINTR == errno);
  650. if(r == static_cast<ssize_t>(len)) {
  651. break;
  652. }
  653. }
  654. freeaddrinfo(res);
  655. if(r == -1) {
  656. throw DlAbortEx(StringFormat(EX_SOCKET_SEND, errorMsg()).str());
  657. }
  658. }
  659. ssize_t SocketCore::readDataFrom(char* data, size_t len,
  660. std::pair<std::string /* numerichost */,
  661. uint16_t /* port */>& sender)
  662. {
  663. struct sockaddr_storage sockaddr;
  664. socklen_t sockaddrlen = sizeof(struct sockaddr_storage);
  665. struct sockaddr* addrp = reinterpret_cast<struct sockaddr*>(&sockaddr);
  666. ssize_t r;
  667. while((r = recvfrom(sockfd, data, len, 0, addrp, &sockaddrlen)) == -1 &&
  668. EINTR == errno);
  669. if(r == -1) {
  670. throw DlAbortEx(StringFormat(EX_SOCKET_RECV, errorMsg()).str());
  671. }
  672. sender = Util::getNumericNameInfo(addrp, sockaddrlen);
  673. return r;
  674. }
  675. } // namespace aria2