StreamPeerIO.c 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718
  1. /**
  2. * @file StreamPeerIO.c
  3. * @author Ambroz Bizjak <ambrop7@gmail.com>
  4. *
  5. * @section LICENSE
  6. *
  7. * This file is part of BadVPN.
  8. *
  9. * BadVPN is free software: you can redistribute it and/or modify
  10. * it under the terms of the GNU General Public License version 2
  11. * as published by the Free Software Foundation.
  12. *
  13. * BadVPN is distributed in the hope that it will be useful,
  14. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  16. * GNU General Public License for more details.
  17. *
  18. * You should have received a copy of the GNU General Public License along
  19. * with this program; if not, write to the Free Software Foundation, Inc.,
  20. * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  21. */
  22. #include <stdlib.h>
  23. #include <openssl/rand.h>
  24. #include <prio.h>
  25. #include <ssl.h>
  26. #include <sslerr.h>
  27. #include <misc/offset.h>
  28. #include <misc/debug.h>
  29. #include <misc/jenkins_hash.h>
  30. #include <misc/byteorder.h>
  31. #include <system/BLog.h>
  32. #include <client/StreamPeerIO.h>
  33. #include <generated/blog_channel_StreamPeerIO.h>
  34. #define STREAMPEERIO_COMPONENT_SEND_SINK 0
  35. #define STREAMPEERIO_COMPONENT_RECEIVE_SOURCE 1
  36. #define STREAMPEERIO_COMPONENT_RECEIVE_DECODER 2
  37. #define MODE_NONE 0
  38. #define MODE_CONNECT 1
  39. #define MODE_LISTEN 2
  40. #define CONNECT_STATE_CONNECTING 0
  41. #define CONNECT_STATE_HANDSHAKE 1
  42. #define CONNECT_STATE_SENDING 2
  43. #define CONNECT_STATE_SENT 3
  44. #define CONNECT_STATE_FINISHED 4
  45. #define LISTEN_STATE_LISTENER 0
  46. #define LISTEN_STATE_GOTCLIENT 1
  47. #define LISTEN_STATE_FINISHED 2
  48. static int init_persistent_io (StreamPeerIO *pio, PacketPassInterface *user_recv_if);
  49. static void free_persistent_io (StreamPeerIO *pio);
  50. static void connecting_connect_handler (StreamPeerIO *pio, int event);
  51. static SECStatus client_auth_certificate_callback (StreamPeerIO *pio, PRFileDesc *fd, PRBool checkSig, PRBool isServer);
  52. static SECStatus client_client_auth_data_callback (StreamPeerIO *pio, PRFileDesc *fd, CERTDistNames *caNames, CERTCertificate **pRetCert, SECKEYPrivateKey **pRetKey);
  53. static void connecting_try_handshake (StreamPeerIO *pio);
  54. static void connecting_handshake_read_handler (StreamPeerIO *pio, PRInt16 event);
  55. static void connecting_pwsender_handler (StreamPeerIO *pio, int is_error);
  56. static void error_handler (StreamPeerIO *pio, int component, const void *data);
  57. static void listener_handler_client (StreamPeerIO *pio, sslsocket *sock);
  58. static int init_io (StreamPeerIO *pio, sslsocket *sock);
  59. static void free_io (StreamPeerIO *pio);
  60. static int compare_certificate (StreamPeerIO *pio, CERTCertificate *cert);
  61. static void reset_state (StreamPeerIO *pio);
  62. static void cleanup_socket (sslsocket *sock, int ssl);
  63. #define COMPONENT_SOURCE 1
  64. #define COMPONENT_SINK 2
  65. #define COMPONENT_DECODER 3
  66. void connecting_connect_handler (StreamPeerIO *pio, int event)
  67. {
  68. ASSERT(event == BSOCKET_CONNECT)
  69. ASSERT(pio->mode == MODE_CONNECT)
  70. ASSERT(pio->connect.state == CONNECT_STATE_CONNECTING)
  71. DebugObject_Access(&pio->d_obj);
  72. // remove connect event handler
  73. BSocket_RemoveEventHandler(&pio->connect.sock.sock, BSOCKET_CONNECT);
  74. // check connection result
  75. int res = BSocket_GetConnectResult(&pio->connect.sock.sock);
  76. if (res != 0) {
  77. BLog(BLOG_NOTICE, "Connection failed (%d)", res);
  78. goto fail0;
  79. }
  80. if (pio->ssl) {
  81. // create BSocket NSPR file descriptor
  82. BSocketPRFileDesc_Create(&pio->connect.sock.bottom_prfd, &pio->connect.sock.sock);
  83. // create SSL file descriptor from the socket's BSocketPRFileDesc
  84. if (!(pio->connect.sock.ssl_prfd = SSL_ImportFD(NULL, &pio->connect.sock.bottom_prfd))) {
  85. ASSERT_FORCE(PR_Close(&pio->connect.sock.bottom_prfd) == PR_SUCCESS)
  86. goto fail0;
  87. }
  88. // set client mode
  89. if (SSL_ResetHandshake(pio->connect.sock.ssl_prfd, PR_FALSE) != SECSuccess) {
  90. BLog(BLOG_ERROR, "SSL_ResetHandshake failed");
  91. goto fail_ssl1;
  92. }
  93. // set verify peer certificate hook
  94. if (SSL_AuthCertificateHook(pio->connect.sock.ssl_prfd, (SSLAuthCertificate)client_auth_certificate_callback, pio) != SECSuccess) {
  95. BLog(BLOG_ERROR, "SSL_AuthCertificateHook failed");
  96. goto fail_ssl1;
  97. }
  98. // set client certificate callback
  99. if (SSL_GetClientAuthDataHook(pio->connect.sock.ssl_prfd, (SSLGetClientAuthData)client_client_auth_data_callback, pio) != SECSuccess) {
  100. BLog(BLOG_ERROR, "SSL_GetClientAuthDataHook failed");
  101. goto fail_ssl1;
  102. }
  103. // initialize BPRFileDesc on SSL file descriptor
  104. BPRFileDesc_Init(&pio->connect.sock.ssl_bprfd, pio->connect.sock.ssl_prfd);
  105. // add event handler for driving handshake
  106. BPRFileDesc_AddEventHandler(&pio->connect.sock.ssl_bprfd, PR_POLL_READ, (BPRFileDesc_handler)connecting_handshake_read_handler, pio);
  107. // change state
  108. pio->connect.state = CONNECT_STATE_HANDSHAKE;
  109. // start handshake
  110. connecting_try_handshake(pio);
  111. return;
  112. } else {
  113. // init password sender
  114. PasswordSender_Init(&pio->connect.pwsender, pio->connect.password, 0, &pio->connect.sock.sock, NULL, (PasswordSender_handler)connecting_pwsender_handler, pio, pio->reactor);
  115. // change state
  116. pio->connect.state = CONNECT_STATE_SENDING;
  117. return;
  118. }
  119. // cleanup
  120. fail_ssl1:
  121. ASSERT_FORCE(PR_Close(pio->connect.sock.ssl_prfd) == PR_SUCCESS)
  122. fail0:
  123. reset_state(pio);
  124. // report error
  125. pio->handler_error(pio->user);
  126. return;
  127. }
  128. SECStatus client_auth_certificate_callback (StreamPeerIO *pio, PRFileDesc *fd, PRBool checkSig, PRBool isServer)
  129. {
  130. ASSERT(pio->ssl)
  131. ASSERT(pio->mode == MODE_CONNECT)
  132. ASSERT(pio->connect.state == CONNECT_STATE_HANDSHAKE)
  133. DebugObject_Access(&pio->d_obj);
  134. // This callback is used to bypass checking the server's domain name, as peers
  135. // don't have domain names. We byte-compare the certificate to the one reported
  136. // by the server anyway.
  137. CERTCertificate *server_cert = SSL_PeerCertificate(pio->connect.sock.ssl_prfd);
  138. if (!server_cert) {
  139. BLog(BLOG_ERROR, "SSL_PeerCertificate failed");
  140. PORT_SetError(SSL_ERROR_BAD_CERTIFICATE);
  141. return SECFailure;
  142. }
  143. SECStatus verify_result = CERT_VerifyCertNow(CERT_GetDefaultCertDB(), server_cert, PR_TRUE, certUsageSSLServer, SSL_RevealPinArg(pio->connect.sock.ssl_prfd));
  144. if (verify_result == SECFailure) {
  145. goto out;
  146. }
  147. // compare to certificate provided by the server
  148. if (!compare_certificate(pio, server_cert)) {
  149. verify_result = SECFailure;
  150. PORT_SetError(SSL_ERROR_BAD_CERTIFICATE);
  151. goto out;
  152. }
  153. out:
  154. CERT_DestroyCertificate(server_cert);
  155. return verify_result;
  156. }
  157. SECStatus client_client_auth_data_callback (StreamPeerIO *pio, PRFileDesc *fd, CERTDistNames *caNames, CERTCertificate **pRetCert, SECKEYPrivateKey **pRetKey)
  158. {
  159. ASSERT(pio->ssl)
  160. ASSERT(pio->mode == MODE_CONNECT)
  161. ASSERT(pio->connect.state == CONNECT_STATE_HANDSHAKE)
  162. DebugObject_Access(&pio->d_obj);
  163. if (!(*pRetCert = CERT_DupCertificate(pio->connect.ssl_cert))) {
  164. return SECFailure;
  165. }
  166. if (!(*pRetKey = SECKEY_CopyPrivateKey(pio->connect.ssl_key))) {
  167. CERT_DestroyCertificate(*pRetCert);
  168. return SECFailure;
  169. }
  170. return SECSuccess;
  171. }
  172. void connecting_try_handshake (StreamPeerIO *pio)
  173. {
  174. ASSERT(pio->ssl)
  175. ASSERT(pio->mode == MODE_CONNECT)
  176. ASSERT(pio->connect.state == CONNECT_STATE_HANDSHAKE)
  177. if (SSL_ForceHandshake(pio->connect.sock.ssl_prfd) != SECSuccess) {
  178. PRErrorCode error = PR_GetError();
  179. if (error == PR_WOULD_BLOCK_ERROR) {
  180. BPRFileDesc_EnableEvent(&pio->connect.sock.ssl_bprfd, PR_POLL_READ);
  181. return;
  182. }
  183. BLog(BLOG_NOTICE, "SSL_ForceHandshake failed (%d)", (int)error);
  184. goto fail0;
  185. }
  186. // remove client certificate callback
  187. if (SSL_GetClientAuthDataHook(pio->connect.sock.ssl_prfd, NULL, NULL) != SECSuccess) {
  188. BLog(BLOG_ERROR, "SSL_GetClientAuthDataHook failed");
  189. goto fail0;
  190. }
  191. // remove verify peer certificate callback
  192. if (SSL_AuthCertificateHook(pio->connect.sock.ssl_prfd, NULL, NULL) != SECSuccess) {
  193. BLog(BLOG_ERROR, "SSL_AuthCertificateHook failed");
  194. goto fail0;
  195. }
  196. // remove read handler
  197. BPRFileDesc_RemoveEventHandler(&pio->connect.sock.ssl_bprfd, PR_POLL_READ);
  198. // init password sender
  199. PasswordSender_Init(&pio->connect.pwsender, pio->connect.password, 1, NULL, &pio->connect.sock.ssl_bprfd, (PasswordSender_handler)connecting_pwsender_handler, pio, pio->reactor);
  200. // change state
  201. pio->connect.state = CONNECT_STATE_SENDING;
  202. return;
  203. // cleanup
  204. fail0:
  205. reset_state(pio);
  206. // report error
  207. pio->handler_error(pio->user);
  208. return;
  209. }
  210. void connecting_handshake_read_handler (StreamPeerIO *pio, PRInt16 event)
  211. {
  212. DebugObject_Access(&pio->d_obj);
  213. connecting_try_handshake(pio);
  214. return;
  215. }
  216. static void connecting_pwsender_handler (StreamPeerIO *pio, int is_error)
  217. {
  218. ASSERT(pio->mode == MODE_CONNECT)
  219. ASSERT(pio->connect.state == CONNECT_STATE_SENDING)
  220. DebugObject_Access(&pio->d_obj);
  221. if (is_error) {
  222. BLog(BLOG_NOTICE, "error sending password");
  223. BLog(BLOG_NOTICE,"BSocket error %d", BSocket_GetError(&pio->connect.sock.sock));
  224. if (pio->ssl) {
  225. BLog(BLOG_NOTICE, "NSPR error %d", (int)PR_GetError());
  226. }
  227. goto fail0;
  228. }
  229. // free password sender
  230. PasswordSender_Free(&pio->connect.pwsender);
  231. // change state
  232. pio->connect.state = CONNECT_STATE_SENT;
  233. // setup i/o
  234. if (!init_io(pio, &pio->connect.sock)) {
  235. goto fail0;
  236. }
  237. // change state
  238. pio->connect.state = CONNECT_STATE_FINISHED;
  239. return;
  240. fail0:
  241. reset_state(pio);
  242. // report error
  243. pio->handler_error(pio->user);
  244. return;
  245. }
  246. void error_handler (StreamPeerIO *pio, int component, const void *data)
  247. {
  248. ASSERT(pio->sock)
  249. DebugObject_Access(&pio->d_obj);
  250. switch (component) {
  251. case COMPONENT_SOURCE:
  252. case COMPONENT_SINK:
  253. BLog(BLOG_NOTICE,"BSocket error %d", BSocket_GetError(&pio->sock->sock));
  254. if (pio->ssl) {
  255. BLog(BLOG_NOTICE, "NSPR error %d", (int)PR_GetError());
  256. }
  257. break;
  258. case COMPONENT_DECODER:
  259. BLog(BLOG_NOTICE, "decoder error %d", *((int *)data));
  260. break;
  261. default:
  262. ASSERT(0);
  263. }
  264. // cleanup
  265. reset_state(pio);
  266. // report error
  267. pio->handler_error(pio->user);
  268. return;
  269. }
  270. void listener_handler_client (StreamPeerIO *pio, sslsocket *sock)
  271. {
  272. ASSERT(pio->mode == MODE_LISTEN)
  273. ASSERT(pio->listen.state == LISTEN_STATE_LISTENER)
  274. DebugObject_Access(&pio->d_obj);
  275. // remember socket
  276. pio->listen.sock = sock;
  277. // change state
  278. pio->listen.state = LISTEN_STATE_GOTCLIENT;
  279. // check ceritficate
  280. if (pio->ssl) {
  281. CERTCertificate *peer_cert = SSL_PeerCertificate(pio->listen.sock->ssl_prfd);
  282. if (!peer_cert) {
  283. BLog(BLOG_ERROR, "SSL_PeerCertificate failed");
  284. goto fail0;
  285. }
  286. // compare certificate to the one provided by the server
  287. if (!compare_certificate(pio, peer_cert)) {
  288. CERT_DestroyCertificate(peer_cert);
  289. goto fail0;
  290. }
  291. CERT_DestroyCertificate(peer_cert);
  292. }
  293. // setup i/o
  294. if (!init_io(pio, pio->listen.sock)) {
  295. goto fail0;
  296. }
  297. // change state
  298. pio->listen.state = LISTEN_STATE_FINISHED;
  299. return;
  300. // cleanup
  301. fail0:
  302. reset_state(pio);
  303. // report error
  304. pio->handler_error(pio->user);
  305. return;
  306. }
  307. int init_persistent_io (StreamPeerIO *pio, PacketPassInterface *user_recv_if)
  308. {
  309. // init error domain
  310. FlowErrorDomain_Init(&pio->ioerrdomain, (FlowErrorDomain_handler)error_handler, pio);
  311. // init sending objects
  312. PacketCopier_Init(&pio->output_user_copier, pio->payload_mtu, BReactor_PendingGroup(pio->reactor));
  313. PacketProtoEncoder_Init(&pio->output_user_ppe, PacketCopier_GetOutput(&pio->output_user_copier), BReactor_PendingGroup(pio->reactor));
  314. PacketPassConnector_Init(&pio->output_connector, PACKETPROTO_ENCLEN(pio->payload_mtu), BReactor_PendingGroup(pio->reactor));
  315. if (!SinglePacketBuffer_Init(&pio->output_user_spb, PacketProtoEncoder_GetOutput(&pio->output_user_ppe), PacketPassConnector_GetInput(&pio->output_connector), BReactor_PendingGroup(pio->reactor))) {
  316. goto fail1;
  317. }
  318. // init receiveing objects
  319. StreamRecvConnector_Init(&pio->input_connector, BReactor_PendingGroup(pio->reactor));
  320. if (!PacketProtoDecoder_Init(
  321. &pio->input_decoder, FlowErrorReporter_Create(&pio->ioerrdomain, COMPONENT_DECODER),
  322. StreamRecvConnector_GetOutput(&pio->input_connector), user_recv_if, BReactor_PendingGroup(pio->reactor)
  323. )) {
  324. goto fail2;
  325. }
  326. return 1;
  327. // free receiveing objects
  328. fail2:
  329. StreamRecvConnector_Free(&pio->input_connector);
  330. // free sending objects
  331. SinglePacketBuffer_Free(&pio->output_user_spb);
  332. fail1:
  333. PacketPassConnector_Free(&pio->output_connector);
  334. PacketProtoEncoder_Free(&pio->output_user_ppe);
  335. PacketCopier_Free(&pio->output_user_copier);
  336. return 0;
  337. }
  338. void free_persistent_io (StreamPeerIO *pio)
  339. {
  340. // free receiveing objects
  341. PacketProtoDecoder_Free(&pio->input_decoder);
  342. StreamRecvConnector_Free(&pio->input_connector);
  343. // free sending objects
  344. SinglePacketBuffer_Free(&pio->output_user_spb);
  345. PacketPassConnector_Free(&pio->output_connector);
  346. PacketProtoEncoder_Free(&pio->output_user_ppe);
  347. PacketCopier_Free(&pio->output_user_copier);
  348. }
  349. int init_io (StreamPeerIO *pio, sslsocket *sock)
  350. {
  351. ASSERT(!pio->sock)
  352. // init sending
  353. StreamPassInterface *sink_interface;
  354. if (pio->ssl) {
  355. PRStreamSink_Init(
  356. &pio->output_sink.ssl, FlowErrorReporter_Create(&pio->ioerrdomain, COMPONENT_SINK),
  357. &sock->ssl_bprfd, BReactor_PendingGroup(pio->reactor)
  358. );
  359. sink_interface = PRStreamSink_GetInput(&pio->output_sink.ssl);
  360. } else {
  361. StreamSocketSink_Init(
  362. &pio->output_sink.plain, FlowErrorReporter_Create(&pio->ioerrdomain, COMPONENT_SINK),
  363. &sock->sock, BReactor_PendingGroup(pio->reactor)
  364. );
  365. sink_interface = StreamSocketSink_GetInput(&pio->output_sink.plain);
  366. }
  367. PacketStreamSender_Init(&pio->output_pss, sink_interface, PACKETPROTO_ENCLEN(pio->payload_mtu), BReactor_PendingGroup(pio->reactor));
  368. PacketPassConnector_ConnectOutput(&pio->output_connector, PacketStreamSender_GetInput(&pio->output_pss));
  369. // init receiving
  370. StreamRecvInterface *source_interface;
  371. if (pio->ssl) {
  372. PRStreamSource_Init(
  373. &pio->input_source.ssl, FlowErrorReporter_Create(&pio->ioerrdomain, COMPONENT_SOURCE),
  374. &sock->ssl_bprfd, BReactor_PendingGroup(pio->reactor)
  375. );
  376. source_interface = PRStreamSource_GetOutput(&pio->input_source.ssl);
  377. } else {
  378. StreamSocketSource_Init(
  379. &pio->input_source.plain, FlowErrorReporter_Create(&pio->ioerrdomain, COMPONENT_SOURCE),
  380. &sock->sock, BReactor_PendingGroup(pio->reactor)
  381. );
  382. source_interface = StreamSocketSource_GetOutput(&pio->input_source.plain);
  383. }
  384. StreamRecvConnector_ConnectInput(&pio->input_connector, source_interface);
  385. pio->sock = sock;
  386. return 1;
  387. }
  388. void free_io (StreamPeerIO *pio)
  389. {
  390. ASSERT(pio->sock)
  391. // reset decoder
  392. PacketProtoDecoder_Reset(&pio->input_decoder);
  393. // free receiving
  394. StreamRecvConnector_DisconnectInput(&pio->input_connector);
  395. if (pio->ssl) {
  396. PRStreamSource_Free(&pio->input_source.ssl);
  397. } else {
  398. StreamSocketSource_Free(&pio->input_source.plain);
  399. }
  400. // free sending
  401. PacketPassConnector_DisconnectOutput(&pio->output_connector);
  402. PacketStreamSender_Free(&pio->output_pss);
  403. if (pio->ssl) {
  404. PRStreamSink_Free(&pio->output_sink.ssl);
  405. } else {
  406. StreamSocketSink_Free(&pio->output_sink.plain);
  407. }
  408. pio->sock = NULL;
  409. }
  410. int compare_certificate (StreamPeerIO *pio, CERTCertificate *cert)
  411. {
  412. ASSERT(pio->ssl)
  413. PRArenaPool *arena = PORT_NewArena(DER_DEFAULT_CHUNKSIZE);
  414. if (!arena) {
  415. BLog(BLOG_ERROR, "WARNING: PORT_NewArena failed");
  416. return 0;
  417. }
  418. // encode server certificate
  419. SECItem der;
  420. der.len = 0;
  421. der.data = NULL;
  422. if (!SEC_ASN1EncodeItem(arena, &der, cert, SEC_ASN1_GET(CERT_CertificateTemplate))) {
  423. BLog(BLOG_ERROR, "SEC_ASN1EncodeItem failed");
  424. PORT_FreeArena(arena, PR_FALSE);
  425. return 0;
  426. }
  427. // byte compare
  428. if (der.len != pio->ssl_peer_cert_len || memcmp(der.data, pio->ssl_peer_cert, der.len)) {
  429. BLog(BLOG_NOTICE, "Client certificate doesn't match");
  430. PORT_FreeArena(arena, PR_FALSE);
  431. return 0;
  432. }
  433. PORT_FreeArena(arena, PR_FALSE);
  434. return 1;
  435. }
  436. void reset_state (StreamPeerIO *pio)
  437. {
  438. // free resources
  439. switch (pio->mode) {
  440. case MODE_NONE:
  441. break;
  442. case MODE_LISTEN:
  443. switch (pio->listen.state) {
  444. case LISTEN_STATE_FINISHED:
  445. free_io(pio);
  446. case LISTEN_STATE_GOTCLIENT:
  447. cleanup_socket(pio->listen.sock, pio->ssl);
  448. free(pio->listen.sock);
  449. break;
  450. case LISTEN_STATE_LISTENER:
  451. PasswordListener_RemoveEntry(pio->listen.listener, &pio->listen.pwentry);
  452. break;
  453. default:
  454. ASSERT(0);
  455. }
  456. pio->mode = MODE_NONE;
  457. break;
  458. case MODE_CONNECT:
  459. switch (pio->connect.state) {
  460. case CONNECT_STATE_FINISHED:
  461. free_io(pio);
  462. case CONNECT_STATE_SENT:
  463. case CONNECT_STATE_SENDING:
  464. if (pio->connect.state == CONNECT_STATE_SENDING) {
  465. PasswordSender_Free(&pio->connect.pwsender);
  466. }
  467. case CONNECT_STATE_HANDSHAKE:
  468. if (pio->ssl) {
  469. BPRFileDesc_Free(&pio->connect.sock.ssl_bprfd);
  470. ASSERT_FORCE(PR_Close(pio->connect.sock.ssl_prfd) == PR_SUCCESS)
  471. }
  472. case CONNECT_STATE_CONNECTING:
  473. BSocket_Free(&pio->connect.sock.sock);
  474. break;
  475. default:
  476. ASSERT(0);
  477. }
  478. pio->mode = MODE_NONE;
  479. break;
  480. default:
  481. ASSERT(0);
  482. }
  483. ASSERT(!pio->sock)
  484. }
  485. void cleanup_socket (sslsocket *sock, int ssl)
  486. {
  487. if (ssl) {
  488. // free BPRFileDesc
  489. BPRFileDesc_Free(&sock->ssl_bprfd);
  490. // free SSL NSPR file descriptor
  491. ASSERT_FORCE(PR_Close(sock->ssl_prfd) == PR_SUCCESS)
  492. }
  493. // free socket
  494. BSocket_Free(&sock->sock);
  495. }
  496. int StreamPeerIO_Init (
  497. StreamPeerIO *pio,
  498. BReactor *reactor,
  499. int ssl,
  500. uint8_t *ssl_peer_cert,
  501. int ssl_peer_cert_len,
  502. int payload_mtu,
  503. PacketPassInterface *user_recv_if,
  504. StreamPeerIO_handler_error handler_error,
  505. void *user
  506. )
  507. {
  508. ASSERT(ssl == 0 || ssl == 1)
  509. ASSERT(payload_mtu >= 0)
  510. ASSERT(PacketPassInterface_GetMTU(user_recv_if) >= payload_mtu)
  511. // init arguments
  512. pio->reactor = reactor;
  513. pio->ssl = ssl;
  514. if (pio->ssl) {
  515. pio->ssl_peer_cert = ssl_peer_cert;
  516. pio->ssl_peer_cert_len = ssl_peer_cert_len;
  517. }
  518. pio->payload_mtu = payload_mtu;
  519. pio->handler_error = handler_error;
  520. pio->user = user;
  521. // init persistent I/O modules
  522. if (!init_persistent_io(pio, user_recv_if)) {
  523. return 0;
  524. }
  525. // set mode none
  526. pio->mode = MODE_NONE;
  527. // set no socket
  528. pio->sock = NULL;
  529. DebugObject_Init(&pio->d_obj);
  530. return 1;
  531. }
  532. void StreamPeerIO_Free (StreamPeerIO *pio)
  533. {
  534. DebugObject_Free(&pio->d_obj);
  535. // reset state
  536. reset_state(pio);
  537. // free persistent I/O modules
  538. free_persistent_io(pio);
  539. }
  540. PacketPassInterface * StreamPeerIO_GetSendInput (StreamPeerIO *pio)
  541. {
  542. DebugObject_Access(&pio->d_obj);
  543. return PacketCopier_GetInput(&pio->output_user_copier);
  544. }
  545. int StreamPeerIO_Connect (StreamPeerIO *pio, BAddr addr, uint64_t password, CERTCertificate *ssl_cert, SECKEYPrivateKey *ssl_key)
  546. {
  547. ASSERT(BAddr_IsRecognized(&addr) && !BAddr_IsInvalid(&addr))
  548. DebugObject_Access(&pio->d_obj);
  549. // reset state
  550. reset_state(pio);
  551. // create socket
  552. if (BSocket_Init(&pio->connect.sock.sock, pio->reactor, addr.type, BSOCKET_TYPE_STREAM) < 0) {
  553. BLog(BLOG_ERROR, "BSocket_Init failed");
  554. goto fail0;
  555. }
  556. // attempt connection
  557. if (BSocket_Connect(&pio->connect.sock.sock, &addr) >= 0 || BSocket_GetError(&pio->connect.sock.sock) != BSOCKET_ERROR_IN_PROGRESS) {
  558. BLog(BLOG_NOTICE, "BSocket_Connect failed");
  559. goto fail1;
  560. }
  561. // waiting for connection result
  562. BSocket_AddEventHandler(&pio->connect.sock.sock, BSOCKET_CONNECT, (BSocket_handler)connecting_connect_handler, pio);
  563. BSocket_EnableEvent(&pio->connect.sock.sock, BSOCKET_CONNECT);
  564. // remember data
  565. if (pio->ssl) {
  566. pio->connect.ssl_cert = ssl_cert;
  567. pio->connect.ssl_key = ssl_key;
  568. }
  569. pio->connect.password = htol64(password);
  570. // set state
  571. pio->mode = MODE_CONNECT;
  572. pio->connect.state = CONNECT_STATE_CONNECTING;
  573. return 1;
  574. fail1:
  575. BSocket_Free(&pio->connect.sock.sock);
  576. fail0:
  577. return 0;
  578. }
  579. void StreamPeerIO_Listen (StreamPeerIO *pio, PasswordListener *listener, uint64_t *password)
  580. {
  581. ASSERT(listener->ssl == pio->ssl)
  582. DebugObject_Access(&pio->d_obj);
  583. // reset state
  584. reset_state(pio);
  585. // add PasswordListener entry
  586. uint64_t newpass = PasswordListener_AddEntry(listener, &pio->listen.pwentry, (PasswordListener_handler_client)listener_handler_client, pio);
  587. // remember data
  588. pio->listen.listener = listener;
  589. // set state
  590. pio->mode = MODE_LISTEN;
  591. pio->listen.state = LISTEN_STATE_LISTENER;
  592. *password = newpass;
  593. }