connection.c 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810
  1. /* Copyright 2001,2002,2003 Roger Dingledine, Matej Pfajfar. */
  2. /* See LICENSE for licensing information */
  3. /* $Id$ */
  4. #include "or.h"
  5. /********* START VARIABLES **********/
  6. extern or_options_t options; /* command-line and config-file options */
  7. extern int global_read_bucket;
  8. char *conn_type_to_string[] = {
  9. "", /* 0 */
  10. "OP listener", /* 1 */
  11. "OP", /* 2 */
  12. "OR listener", /* 3 */
  13. "OR", /* 4 */
  14. "Exit", /* 5 */
  15. "App listener",/* 6 */
  16. "App", /* 7 */
  17. "Dir listener",/* 8 */
  18. "Dir", /* 9 */
  19. "DNS worker", /* 10 */
  20. "CPU worker", /* 11 */
  21. };
  22. char *conn_state_to_string[][_CONN_TYPE_MAX+1] = {
  23. { NULL }, /* no type associated with 0 */
  24. { NULL }, /* op listener, obsolete */
  25. { NULL }, /* op, obsolete */
  26. { "ready" }, /* or listener, 0 */
  27. { "", /* OR, 0 */
  28. "connect()ing", /* 1 */
  29. "handshaking", /* 2 */
  30. "open" }, /* 3 */
  31. { "", /* exit, 0 */
  32. "waiting for dest info", /* 1 */
  33. "connecting", /* 2 */
  34. "open" }, /* 3 */
  35. { "ready" }, /* app listener, 0 */
  36. { "", /* 0 */
  37. "", /* 1 */
  38. "", /* 2 */
  39. "", /* 3 */
  40. "awaiting dest info", /* app, 4 */
  41. "waiting for OR connection", /* 5 */
  42. "open" }, /* 6 */
  43. { "ready" }, /* dir listener, 0 */
  44. { "", /* dir, 0 */
  45. "connecting (fetch)", /* 1 */
  46. "connecting (upload)", /* 2 */
  47. "client sending fetch", /* 3 */
  48. "client sending upload", /* 4 */
  49. "client reading fetch", /* 5 */
  50. "client reading upload", /* 6 */
  51. "awaiting command", /* 7 */
  52. "writing" }, /* 8 */
  53. { "", /* dns worker, 0 */
  54. "idle", /* 1 */
  55. "busy" }, /* 2 */
  56. { "", /* cpu worker, 0 */
  57. "idle", /* 1 */
  58. "busy with onion", /* 2 */
  59. "busy with handshake" }, /* 3 */
  60. };
  61. /********* END VARIABLES ************/
  62. static int connection_init_accepted_conn(connection_t *conn);
  63. static int connection_handle_listener_read(connection_t *conn, int new_type);
  64. /**************************************************************/
  65. connection_t *connection_new(int type) {
  66. connection_t *conn;
  67. time_t now = time(NULL);
  68. conn = (connection_t *)tor_malloc(sizeof(connection_t));
  69. memset(conn,0,sizeof(connection_t)); /* zero it out to start */
  70. conn->s = -1; /* give it a default of 'not used' */
  71. conn->type = type;
  72. if(!connection_is_listener(conn)) { /* listeners never use their buf */
  73. conn->inbuf = buf_new();
  74. conn->outbuf = buf_new();
  75. }
  76. conn->timestamp_created = now;
  77. conn->timestamp_lastread = now;
  78. conn->timestamp_lastwritten = now;
  79. return conn;
  80. }
  81. void connection_free(connection_t *conn) {
  82. assert(conn);
  83. if(!connection_is_listener(conn)) {
  84. buf_free(conn->inbuf);
  85. buf_free(conn->outbuf);
  86. }
  87. tor_free(conn->address);
  88. if(connection_speaks_cells(conn)) {
  89. directory_set_dirty();
  90. if (conn->tls)
  91. tor_tls_free(conn->tls);
  92. }
  93. if (conn->onion_pkey)
  94. crypto_free_pk_env(conn->onion_pkey);
  95. if (conn->link_pkey)
  96. crypto_free_pk_env(conn->link_pkey);
  97. if (conn->identity_pkey)
  98. crypto_free_pk_env(conn->identity_pkey);
  99. tor_free(conn->nickname);
  100. if(conn->s >= 0) {
  101. log_fn(LOG_INFO,"closing fd %d.",conn->s);
  102. close(conn->s);
  103. }
  104. free(conn);
  105. }
  106. int connection_create_listener(char *bindaddress, uint16_t bindport, int type) {
  107. struct sockaddr_in bindaddr; /* where to bind */
  108. struct hostent *rent;
  109. connection_t *conn;
  110. int s; /* the socket we're going to make */
  111. int one=1;
  112. memset(&bindaddr,0,sizeof(struct sockaddr_in));
  113. bindaddr.sin_family = AF_INET;
  114. bindaddr.sin_port = htons(bindport);
  115. rent = gethostbyname(bindaddress);
  116. if (!rent) {
  117. log_fn(LOG_WARN,"Can't resolve BindAddress %s",bindaddress);
  118. return -1;
  119. }
  120. if(rent->h_length != 4)
  121. return -1; /* XXX complain */
  122. memcpy(&(bindaddr.sin_addr.s_addr),rent->h_addr,rent->h_length);
  123. s = socket(PF_INET,SOCK_STREAM,IPPROTO_TCP);
  124. if (s < 0) {
  125. log_fn(LOG_WARN,"Socket creation failed.");
  126. return -1;
  127. }
  128. setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &one, sizeof(one));
  129. if(bind(s,(struct sockaddr *)&bindaddr,sizeof(bindaddr)) < 0) {
  130. log_fn(LOG_WARN,"Could not bind to port %u: %s",bindport,strerror(errno));
  131. return -1;
  132. }
  133. if(listen(s,SOMAXCONN) < 0) {
  134. log_fn(LOG_WARN,"Could not listen on port %u: %s",bindport,strerror(errno));
  135. return -1;
  136. }
  137. set_socket_nonblocking(s);
  138. conn = connection_new(type);
  139. conn->s = s;
  140. if(connection_add(conn) < 0) { /* no space, forget it */
  141. log_fn(LOG_WARN,"connection_add failed. Giving up.");
  142. connection_free(conn);
  143. return -1;
  144. }
  145. log_fn(LOG_DEBUG,"%s listening on port %u.",conn_type_to_string[type], bindport);
  146. conn->state = LISTENER_STATE_READY;
  147. connection_start_reading(conn);
  148. return 0;
  149. }
  150. static int connection_handle_listener_read(connection_t *conn, int new_type) {
  151. int news; /* the new socket */
  152. connection_t *newconn;
  153. struct sockaddr_in remote; /* information about the remote peer when connecting to other routers */
  154. int remotelen = sizeof(struct sockaddr_in); /* length of the remote address */
  155. #ifdef MS_WINDOWS
  156. int e;
  157. #endif
  158. news = accept(conn->s,(struct sockaddr *)&remote,&remotelen);
  159. if (news == -1) { /* accept() error */
  160. if(ERRNO_EAGAIN(errno)) {
  161. #ifdef MS_WINDOWS
  162. e = correct_socket_errno(conn->s);
  163. if (ERRNO_EAGAIN(e))
  164. return 0;
  165. #else
  166. return 0; /* he hung up before we could accept(). that's fine. */
  167. #endif
  168. }
  169. /* else there was a real error. */
  170. log_fn(LOG_WARN,"accept() failed. Closing listener.");
  171. return -1;
  172. }
  173. log(LOG_INFO,"Connection accepted on socket %d (child of fd %d).",news, conn->s);
  174. set_socket_nonblocking(news);
  175. newconn = connection_new(new_type);
  176. newconn->s = news;
  177. newconn->address = tor_strdup(inet_ntoa(remote.sin_addr)); /* remember the remote address */
  178. newconn->addr = ntohl(remote.sin_addr.s_addr);
  179. newconn->port = ntohs(remote.sin_port);
  180. if(connection_add(newconn) < 0) { /* no space, forget it */
  181. connection_free(newconn);
  182. return 0; /* no need to tear down the parent */
  183. }
  184. if(connection_init_accepted_conn(newconn) < 0) {
  185. newconn->marked_for_close = 1;
  186. return 0;
  187. }
  188. return 0;
  189. }
  190. static int connection_init_accepted_conn(connection_t *conn) {
  191. connection_start_reading(conn);
  192. switch(conn->type) {
  193. case CONN_TYPE_OR:
  194. return connection_tls_start_handshake(conn, 1);
  195. case CONN_TYPE_AP:
  196. conn->state = AP_CONN_STATE_SOCKS_WAIT;
  197. break;
  198. case CONN_TYPE_DIR:
  199. conn->state = DIR_CONN_STATE_SERVER_COMMAND_WAIT;
  200. break;
  201. }
  202. return 0;
  203. }
  204. /* take conn, make a nonblocking socket; try to connect to
  205. * addr:port (they arrive in *host order*). If fail, return -1. Else
  206. * assign s to conn->s: if connected return 1, if eagain return 0.
  207. * address is used to make the logs useful.
  208. */
  209. int connection_connect(connection_t *conn, char *address, uint32_t addr, uint16_t port) {
  210. int s;
  211. struct sockaddr_in dest_addr;
  212. s=socket(PF_INET,SOCK_STREAM,IPPROTO_TCP);
  213. if (s < 0) {
  214. log_fn(LOG_WARN,"Error creating network socket.");
  215. return -1;
  216. }
  217. set_socket_nonblocking(s);
  218. memset((void *)&dest_addr,0,sizeof(dest_addr));
  219. dest_addr.sin_family = AF_INET;
  220. dest_addr.sin_port = htons(port);
  221. dest_addr.sin_addr.s_addr = htonl(addr);
  222. log_fn(LOG_DEBUG,"Connecting to %s:%u.",address,port);
  223. if(connect(s,(struct sockaddr *)&dest_addr,sizeof(dest_addr)) < 0) {
  224. if(!ERRNO_CONN_EINPROGRESS(errno)) {
  225. /* yuck. kill it. */
  226. perror("connect");
  227. log_fn(LOG_INFO,"Connect() to %s:%u failed.",address,port);
  228. return -1;
  229. } else {
  230. /* it's in progress. set state appropriately and return. */
  231. conn->s = s;
  232. log_fn(LOG_DEBUG,"connect in progress, socket %d.",s);
  233. return 0;
  234. }
  235. }
  236. /* it succeeded. we're connected. */
  237. log_fn(LOG_INFO,"Connection to %s:%u established.",address,port);
  238. conn->s = s;
  239. return 1;
  240. }
  241. /* start all connections that should be up but aren't */
  242. int retry_all_connections(void) {
  243. if(options.ORPort) {
  244. router_retry_connections();
  245. }
  246. if(options.ORPort && !connection_get_by_type(CONN_TYPE_OR_LISTENER)) {
  247. if(connection_create_listener(options.ORBindAddress, options.ORPort,
  248. CONN_TYPE_OR_LISTENER) < 0)
  249. return -1;
  250. }
  251. if(options.DirPort && !connection_get_by_type(CONN_TYPE_DIR_LISTENER)) {
  252. if(connection_create_listener(options.DirBindAddress, options.DirPort,
  253. CONN_TYPE_DIR_LISTENER) < 0)
  254. return -1;
  255. }
  256. if(options.SocksPort && !connection_get_by_type(CONN_TYPE_AP_LISTENER)) {
  257. if(connection_create_listener(options.SocksBindAddress, options.SocksPort,
  258. CONN_TYPE_AP_LISTENER) < 0)
  259. return -1;
  260. }
  261. return 0;
  262. }
  263. int connection_handle_read(connection_t *conn) {
  264. conn->timestamp_lastread = time(NULL);
  265. switch(conn->type) {
  266. case CONN_TYPE_OR_LISTENER:
  267. return connection_handle_listener_read(conn, CONN_TYPE_OR);
  268. case CONN_TYPE_AP_LISTENER:
  269. return connection_handle_listener_read(conn, CONN_TYPE_AP);
  270. case CONN_TYPE_DIR_LISTENER:
  271. return connection_handle_listener_read(conn, CONN_TYPE_DIR);
  272. }
  273. if(connection_read_to_buf(conn) < 0) {
  274. if(conn->type == CONN_TYPE_DIR &&
  275. (conn->state == DIR_CONN_STATE_CONNECTING_FETCH ||
  276. conn->state == DIR_CONN_STATE_CONNECTING_UPLOAD)) {
  277. /* it's a directory server and connecting failed: forget about this router */
  278. /* XXX I suspect pollerr may make Windows not get to this point. :( */
  279. router_mark_as_down(conn->nickname);
  280. }
  281. return -1;
  282. }
  283. if(connection_process_inbuf(conn) < 0) {
  284. // log_fn(LOG_DEBUG,"connection_process_inbuf returned -1.");
  285. return -1;
  286. }
  287. return 0;
  288. }
  289. /* return -1 if we want to break conn, else return 0 */
  290. int connection_read_to_buf(connection_t *conn) {
  291. int result;
  292. int at_most;
  293. if(options.LinkPadding) {
  294. at_most = global_read_bucket;
  295. } else {
  296. /* do a rudimentary round-robin so one connection can't hog a thickpipe */
  297. if(connection_speaks_cells(conn)) {
  298. at_most = 30*(CELL_NETWORK_SIZE);
  299. } else {
  300. at_most = 30*(CELL_PAYLOAD_SIZE - RELAY_HEADER_SIZE);
  301. }
  302. if(at_most > global_read_bucket)
  303. at_most = global_read_bucket;
  304. }
  305. if(connection_speaks_cells(conn) && conn->state != OR_CONN_STATE_CONNECTING) {
  306. if(conn->state == OR_CONN_STATE_HANDSHAKING)
  307. return connection_tls_continue_handshake(conn);
  308. /* else open, or closing */
  309. if(at_most > conn->receiver_bucket)
  310. at_most = conn->receiver_bucket;
  311. result = read_to_buf_tls(conn->tls, at_most, conn->inbuf);
  312. switch(result) {
  313. case TOR_TLS_ERROR:
  314. case TOR_TLS_CLOSE:
  315. log_fn(LOG_INFO,"tls error. breaking.");
  316. return -1; /* XXX deal with close better */
  317. case TOR_TLS_WANTWRITE:
  318. connection_start_writing(conn);
  319. return 0;
  320. case TOR_TLS_WANTREAD: /* we're already reading */
  321. case TOR_TLS_DONE: /* no data read, so nothing to process */
  322. return 0;
  323. }
  324. } else {
  325. result = read_to_buf(conn->s, at_most, conn->inbuf,
  326. &conn->inbuf_reached_eof);
  327. // log(LOG_DEBUG,"connection_read_to_buf(): read_to_buf returned %d.",read_result);
  328. if(result < 0)
  329. return -1;
  330. }
  331. global_read_bucket -= result; assert(global_read_bucket >= 0);
  332. if(global_read_bucket == 0) {
  333. log_fn(LOG_DEBUG,"global bucket exhausted. Pausing.");
  334. conn->wants_to_read = 1;
  335. connection_stop_reading(conn);
  336. return 0;
  337. }
  338. if(connection_speaks_cells(conn) && conn->state == OR_CONN_STATE_OPEN) {
  339. conn->receiver_bucket -= result; assert(conn->receiver_bucket >= 0);
  340. if(conn->receiver_bucket == 0) {
  341. log_fn(LOG_DEBUG,"receiver bucket exhausted. Pausing.");
  342. conn->wants_to_read = 1;
  343. connection_stop_reading(conn);
  344. return 0;
  345. }
  346. }
  347. return 0;
  348. }
  349. int connection_fetch_from_buf(char *string, int len, connection_t *conn) {
  350. return fetch_from_buf(string, len, conn->inbuf);
  351. }
  352. int connection_find_on_inbuf(char *string, int len, connection_t *conn) {
  353. return find_on_inbuf(string, len, conn->inbuf);
  354. }
  355. int connection_wants_to_flush(connection_t *conn) {
  356. return conn->outbuf_flushlen;
  357. }
  358. int connection_outbuf_too_full(connection_t *conn) {
  359. return (conn->outbuf_flushlen > 10*CELL_PAYLOAD_SIZE);
  360. }
  361. int connection_flush_buf(connection_t *conn) {
  362. return flush_buf(conn->s, conn->outbuf, &conn->outbuf_flushlen);
  363. }
  364. /* return -1 if you want to break the conn, else return 0 */
  365. int connection_handle_write(connection_t *conn) {
  366. if(connection_is_listener(conn)) {
  367. log_fn(LOG_WARN,"Got a listener socket. Can't happen!");
  368. return -1;
  369. }
  370. conn->timestamp_lastwritten = time(NULL);
  371. if(connection_speaks_cells(conn) && conn->state != OR_CONN_STATE_CONNECTING) {
  372. if(conn->state == OR_CONN_STATE_HANDSHAKING) {
  373. connection_stop_writing(conn);
  374. return connection_tls_continue_handshake(conn);
  375. }
  376. /* else open, or closing */
  377. switch(flush_buf_tls(conn->tls, conn->outbuf, &conn->outbuf_flushlen)) {
  378. case TOR_TLS_ERROR:
  379. case TOR_TLS_CLOSE:
  380. log_fn(LOG_INFO,"tls error. breaking.");
  381. return -1; /* XXX deal with close better */
  382. case TOR_TLS_WANTWRITE:
  383. log_fn(LOG_DEBUG,"wanted write.");
  384. /* we're already writing */
  385. return 0;
  386. case TOR_TLS_WANTREAD:
  387. /* Make sure to avoid a loop if the receive buckets are empty. */
  388. log_fn(LOG_DEBUG,"wanted read.");
  389. if(!connection_is_reading(conn)) {
  390. connection_stop_writing(conn);
  391. conn->wants_to_write = 1;
  392. /* we'll start reading again when the next second arrives,
  393. * and then also start writing again.
  394. */
  395. }
  396. /* else no problem, we're already reading */
  397. return 0;
  398. /* case TOR_TLS_DONE:
  399. * for TOR_TLS_DONE, fall through to check if the flushlen
  400. * is empty, so we can stop writing.
  401. */
  402. }
  403. } else {
  404. if(flush_buf(conn->s, conn->outbuf, &conn->outbuf_flushlen) < 0)
  405. return -1;
  406. /* conns in CONNECTING state will fall through... */
  407. }
  408. if(!connection_wants_to_flush(conn)) /* it's done flushing */
  409. if(connection_finished_flushing(conn) < 0) /* ...and get handled here. */
  410. return -1;
  411. return 0;
  412. }
  413. void connection_write_to_buf(const char *string, int len, connection_t *conn) {
  414. if(!len || conn->marked_for_close)
  415. return;
  416. if( (!connection_speaks_cells(conn)) ||
  417. (!connection_state_is_open(conn)) ||
  418. (options.LinkPadding == 0) ) {
  419. /* connection types other than or, or or not in 'open' state, should flush immediately */
  420. /* also flush immediately if we're not doing LinkPadding, since otherwise it will never flush */
  421. connection_start_writing(conn);
  422. conn->outbuf_flushlen += len;
  423. }
  424. if(write_to_buf(string, len, conn->outbuf) < 0) {
  425. log_fn(LOG_WARN,"write_to_buf failed. Closing connection (fd %d).", conn->s);
  426. conn->marked_for_close = 1;
  427. }
  428. }
  429. connection_t *connection_exact_get_by_addr_port(uint32_t addr, uint16_t port) {
  430. int i, n;
  431. connection_t *conn;
  432. connection_t **carray;
  433. get_connection_array(&carray,&n);
  434. for(i=0;i<n;i++) {
  435. conn = carray[i];
  436. if(conn->addr == addr && conn->port == port && !conn->marked_for_close)
  437. return conn;
  438. }
  439. return NULL;
  440. }
  441. connection_t *connection_twin_get_by_addr_port(uint32_t addr, uint16_t port) {
  442. /* Find a connection to the router described by addr and port,
  443. * or alternately any router which knows its key.
  444. * This connection *must* be in 'open' state.
  445. * If not, return NULL.
  446. */
  447. int i, n;
  448. connection_t *conn;
  449. routerinfo_t *router;
  450. connection_t **carray;
  451. /* first check if it's there exactly */
  452. conn = connection_exact_get_by_addr_port(addr,port);
  453. if(conn && connection_state_is_open(conn)) {
  454. log(LOG_INFO,"connection_twin_get_by_addr_port(): Found exact match.");
  455. return conn;
  456. }
  457. /* now check if any of the other open connections are a twin for this one */
  458. router = router_get_by_addr_port(addr,port);
  459. if(!router)
  460. return NULL;
  461. get_connection_array(&carray,&n);
  462. for(i=0;i<n;i++) {
  463. conn = carray[i];
  464. assert(conn);
  465. if(connection_state_is_open(conn) &&
  466. !conn->marked_for_close &&
  467. !crypto_pk_cmp_keys(conn->onion_pkey, router->onion_pkey)) {
  468. log(LOG_INFO,"connection_twin_get_by_addr_port(): Found twin (%s).",conn->address);
  469. return conn;
  470. }
  471. }
  472. return NULL;
  473. }
  474. connection_t *connection_get_by_type(int type) {
  475. int i, n;
  476. connection_t *conn;
  477. connection_t **carray;
  478. get_connection_array(&carray,&n);
  479. for(i=0;i<n;i++) {
  480. conn = carray[i];
  481. if(conn->type == type && !conn->marked_for_close)
  482. return conn;
  483. }
  484. return NULL;
  485. }
  486. connection_t *connection_get_by_type_state(int type, int state) {
  487. int i, n;
  488. connection_t *conn;
  489. connection_t **carray;
  490. get_connection_array(&carray,&n);
  491. for(i=0;i<n;i++) {
  492. conn = carray[i];
  493. if(conn->type == type && conn->state == state && !conn->marked_for_close)
  494. return conn;
  495. }
  496. return NULL;
  497. }
  498. connection_t *connection_get_by_type_state_lastwritten(int type, int state) {
  499. int i, n;
  500. connection_t *conn, *best=NULL;
  501. connection_t **carray;
  502. get_connection_array(&carray,&n);
  503. for(i=0;i<n;i++) {
  504. conn = carray[i];
  505. if(conn->type == type && conn->state == state && !conn->marked_for_close)
  506. if(!best || conn->timestamp_lastwritten < best->timestamp_lastwritten)
  507. best = conn;
  508. }
  509. return best;
  510. }
  511. int connection_receiver_bucket_should_increase(connection_t *conn) {
  512. assert(conn);
  513. if(!connection_speaks_cells(conn))
  514. return 0; /* edge connections don't use receiver_buckets */
  515. if(conn->state != OR_CONN_STATE_OPEN)
  516. return 0; /* only open connections play the rate limiting game */
  517. assert(conn->bandwidth > 0);
  518. if(conn->receiver_bucket > 9*conn->bandwidth)
  519. return 0;
  520. return 1;
  521. }
  522. int connection_is_listener(connection_t *conn) {
  523. if(conn->type == CONN_TYPE_OR_LISTENER ||
  524. conn->type == CONN_TYPE_AP_LISTENER ||
  525. conn->type == CONN_TYPE_DIR_LISTENER)
  526. return 1;
  527. return 0;
  528. }
  529. int connection_state_is_open(connection_t *conn) {
  530. assert(conn);
  531. if((conn->type == CONN_TYPE_OR && conn->state == OR_CONN_STATE_OPEN) ||
  532. (conn->type == CONN_TYPE_AP && conn->state == AP_CONN_STATE_OPEN) ||
  533. (conn->type == CONN_TYPE_EXIT && conn->state == EXIT_CONN_STATE_OPEN))
  534. return 1;
  535. return 0;
  536. }
  537. int connection_send_destroy(aci_t aci, connection_t *conn) {
  538. cell_t cell;
  539. assert(conn);
  540. if(!connection_speaks_cells(conn)) {
  541. log_fn(LOG_INFO,"Aci %d: At an edge. Marking connection for close.", aci);
  542. connection_edge_end(conn, END_STREAM_REASON_DESTROY, conn->cpath_layer);
  543. /* if they already sent a destroy, they know. XXX can just close? */
  544. return 0;
  545. }
  546. memset(&cell, 0, sizeof(cell_t));
  547. cell.aci = aci;
  548. cell.command = CELL_DESTROY;
  549. log_fn(LOG_INFO,"Sending destroy (aci %d).",aci);
  550. connection_or_write_cell_to_buf(&cell, conn);
  551. return 0;
  552. }
  553. int connection_process_inbuf(connection_t *conn) {
  554. assert(conn);
  555. switch(conn->type) {
  556. case CONN_TYPE_OR:
  557. return connection_or_process_inbuf(conn);
  558. case CONN_TYPE_EXIT:
  559. case CONN_TYPE_AP:
  560. return connection_edge_process_inbuf(conn);
  561. case CONN_TYPE_DIR:
  562. return connection_dir_process_inbuf(conn);
  563. case CONN_TYPE_DNSWORKER:
  564. return connection_dns_process_inbuf(conn);
  565. case CONN_TYPE_CPUWORKER:
  566. return connection_cpu_process_inbuf(conn);
  567. default:
  568. log_fn(LOG_WARN,"got unexpected conn->type %d.", conn->type);
  569. return -1;
  570. }
  571. }
  572. int connection_finished_flushing(connection_t *conn) {
  573. assert(conn);
  574. // log_fn(LOG_DEBUG,"entered. Socket %u.", conn->s);
  575. switch(conn->type) {
  576. case CONN_TYPE_OR:
  577. return connection_or_finished_flushing(conn);
  578. case CONN_TYPE_AP:
  579. case CONN_TYPE_EXIT:
  580. return connection_edge_finished_flushing(conn);
  581. case CONN_TYPE_DIR:
  582. return connection_dir_finished_flushing(conn);
  583. case CONN_TYPE_DNSWORKER:
  584. return connection_dns_finished_flushing(conn);
  585. case CONN_TYPE_CPUWORKER:
  586. return connection_cpu_finished_flushing(conn);
  587. default:
  588. log_fn(LOG_WARN,"got unexpected conn->type %d.", conn->type);
  589. return -1;
  590. }
  591. }
  592. void assert_connection_ok(connection_t *conn, time_t now)
  593. {
  594. return;
  595. assert(conn);
  596. assert(conn->type >= _CONN_TYPE_MIN);
  597. assert(conn->type <= _CONN_TYPE_MAX);
  598. /* XXX check: wants_to_read, wants_to_write, s, poll_index,
  599. * marked_for_close. */
  600. /* buffers */
  601. assert(conn->inbuf);
  602. assert(conn->outbuf);
  603. assert(!now || conn->timestamp_lastread <= now);
  604. assert(!now || conn->timestamp_lastwritten <= now);
  605. assert(conn->timestamp_created <= conn->timestamp_lastread);
  606. assert(conn->timestamp_created <= conn->timestamp_lastwritten);
  607. /* XXX Fix this; no longer so.*/
  608. #if 0
  609. if(conn->type != CONN_TYPE_OR && conn->type != CONN_TYPE_DIR)
  610. assert(!conn->pkey);
  611. /* pkey is set if we're a dir client, or if we're an OR in state OPEN
  612. * connected to another OR.
  613. */
  614. #endif
  615. if (conn->type != CONN_TYPE_OR) {
  616. assert(!conn->tls);
  617. } else {
  618. if(conn->state == OR_CONN_STATE_OPEN) {
  619. assert(conn->bandwidth > 0);
  620. assert(conn->receiver_bucket >= 0);
  621. assert(conn->receiver_bucket <= 10*conn->bandwidth);
  622. }
  623. assert(conn->addr && conn->port);
  624. assert(conn->address);
  625. if (conn->state != OR_CONN_STATE_CONNECTING)
  626. assert(conn->tls);
  627. }
  628. if (conn->type != CONN_TYPE_EXIT && conn->type != CONN_TYPE_AP) {
  629. assert(!conn->stream_id[0]);
  630. assert(!conn->next_stream);
  631. assert(!conn->cpath_layer);
  632. assert(!conn->package_window);
  633. assert(!conn->deliver_window);
  634. assert(!conn->done_sending);
  635. assert(!conn->done_receiving);
  636. } else {
  637. assert(!conn->next_stream ||
  638. conn->next_stream->type == CONN_TYPE_EXIT ||
  639. conn->next_stream->type == CONN_TYPE_AP);
  640. if(conn->type == CONN_TYPE_AP && conn->state == AP_CONN_STATE_OPEN)
  641. assert(conn->cpath_layer);
  642. if(conn->cpath_layer)
  643. assert_cpath_layer_ok(conn->cpath_layer);
  644. /* XXX unchecked, package window, deliver window. */
  645. }
  646. switch(conn->type)
  647. {
  648. case CONN_TYPE_OR_LISTENER:
  649. case CONN_TYPE_AP_LISTENER:
  650. case CONN_TYPE_DIR_LISTENER:
  651. assert(conn->state == LISTENER_STATE_READY);
  652. break;
  653. case CONN_TYPE_OR:
  654. assert(conn->state >= _OR_CONN_STATE_MIN &&
  655. conn->state <= _OR_CONN_STATE_MAX);
  656. break;
  657. case CONN_TYPE_EXIT:
  658. assert(conn->state >= _EXIT_CONN_STATE_MIN &&
  659. conn->state <= _EXIT_CONN_STATE_MAX);
  660. break;
  661. case CONN_TYPE_AP:
  662. assert(conn->state >= _AP_CONN_STATE_MIN &&
  663. conn->state <= _AP_CONN_STATE_MAX);
  664. break;
  665. case CONN_TYPE_DIR:
  666. assert(conn->state >= _DIR_CONN_STATE_MIN &&
  667. conn->state <= _DIR_CONN_STATE_MAX);
  668. break;
  669. case CONN_TYPE_DNSWORKER:
  670. assert(conn->state == DNSWORKER_STATE_IDLE ||
  671. conn->state == DNSWORKER_STATE_BUSY);
  672. case CONN_TYPE_CPUWORKER:
  673. assert(conn->state >= _CPUWORKER_STATE_MIN &&
  674. conn->state <= _CPUWORKER_STATE_MAX);
  675. break;
  676. default:
  677. assert(0);
  678. }
  679. }
  680. /*
  681. Local Variables:
  682. mode:c
  683. indent-tabs-mode:nil
  684. c-basic-offset:2
  685. End:
  686. */