connection.c 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782
  1. /* Copyright 2001,2002 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. char *conn_type_to_string[] = {
  8. "", /* 0 */
  9. "OP listener", /* 1 */
  10. "OP", /* 2 */
  11. "OR listener", /* 3 */
  12. "OR", /* 4 */
  13. "Exit", /* 5 */
  14. "App listener",/* 6 */
  15. "App", /* 7 */
  16. "Dir listener",/* 8 */
  17. "Dir", /* 9 */
  18. "DNS master", /* 10 */
  19. };
  20. char *conn_state_to_string[][15] = {
  21. { }, /* no type associated with 0 */
  22. { "ready" }, /* op listener, 0 */
  23. { "awaiting keys", /* op, 0 */
  24. "open", /* 1 */
  25. "close", /* 2 */
  26. "close_wait" }, /* 3 */
  27. { "ready" }, /* or listener, 0 */
  28. { "connecting (as OP)", /* or, 0 */
  29. "sending keys (as OP)", /* 1 */
  30. "connecting (as client)", /* 2 */
  31. "sending auth (as client)", /* 3 */
  32. "waiting for auth (as client)", /* 4 */
  33. "sending nonce (as client)", /* 5 */
  34. "waiting for auth (as server)", /* 6 */
  35. "sending auth (as server)", /* 7 */
  36. "waiting for nonce (as server)",/* 8 */
  37. "open" }, /* 9 */
  38. { "waiting for dest info", /* exit, 0 */
  39. "connecting", /* 1 */
  40. "open" }, /* 2 */
  41. { "ready" }, /* app listener, 0 */
  42. { "awaiting dest info", /* app, 0 */
  43. "waiting for OR connection", /* 1 */
  44. "open" }, /* 2 */
  45. { "ready" }, /* dir listener, 0 */
  46. { "connecting", /* 0 */
  47. "sending command", /* 1 */
  48. "reading", /* 2 */
  49. "awaiting command", /* 3 */
  50. "writing" }, /* 4 */
  51. { "open" }, /* dns master, 0 */
  52. };
  53. /********* END VARIABLES ************/
  54. /**************************************************************/
  55. int tv_cmp(struct timeval *a, struct timeval *b) {
  56. if (a->tv_sec > b->tv_sec)
  57. return 1;
  58. if (a->tv_sec < b->tv_sec)
  59. return -1;
  60. if (a->tv_usec > b->tv_usec)
  61. return 1;
  62. if (a->tv_usec < b->tv_usec)
  63. return -1;
  64. return 0;
  65. }
  66. void tv_add(struct timeval *a, struct timeval *b) {
  67. a->tv_usec += b->tv_usec;
  68. a->tv_sec += b->tv_sec + (a->tv_usec / 1000000);
  69. a->tv_usec %= 1000000;
  70. }
  71. void tv_addms(struct timeval *a, long ms) {
  72. a->tv_usec += (ms * 1000) % 1000000;
  73. a->tv_sec += ((ms * 1000) / 1000000) + (a->tv_usec / 1000000);
  74. a->tv_usec %= 1000000;
  75. }
  76. /**************************************************************/
  77. connection_t *connection_new(int type) {
  78. connection_t *conn;
  79. struct timeval now;
  80. if(gettimeofday(&now,NULL) < 0)
  81. return NULL;
  82. conn = (connection_t *)malloc(sizeof(connection_t));
  83. if(!conn)
  84. return NULL;
  85. memset(conn,0,sizeof(connection_t)); /* zero it out to start */
  86. conn->type = type;
  87. if(buf_new(&conn->inbuf, &conn->inbuflen, &conn->inbuf_datalen) < 0 ||
  88. buf_new(&conn->outbuf, &conn->outbuflen, &conn->outbuf_datalen) < 0)
  89. return NULL;
  90. conn->receiver_bucket = 10240; /* should be enough to do the handshake */
  91. conn->bandwidth = conn->receiver_bucket / 10; /* give it a default */
  92. conn->timestamp_created = now.tv_sec;
  93. conn->timestamp_lastread = now.tv_sec;
  94. conn->timestamp_lastwritten = now.tv_sec;
  95. if (connection_speaks_cells(conn)) {
  96. conn->f_crypto = crypto_new_cipher_env(CRYPTO_CIPHER_DES);
  97. if (!conn->f_crypto) {
  98. free((void *)conn);
  99. return NULL;
  100. }
  101. conn->b_crypto = crypto_new_cipher_env(CRYPTO_CIPHER_DES);
  102. if (!conn->b_crypto) {
  103. crypto_free_cipher_env(conn->f_crypto);
  104. free((void *)conn);
  105. return NULL;
  106. }
  107. }
  108. return conn;
  109. }
  110. void connection_free(connection_t *conn) {
  111. assert(conn);
  112. buf_free(conn->inbuf);
  113. buf_free(conn->outbuf);
  114. if(conn->address)
  115. free(conn->address);
  116. if(conn->dest_addr)
  117. free(conn->dest_addr);
  118. if(connection_speaks_cells(conn)) {
  119. if (conn->f_crypto)
  120. crypto_free_cipher_env(conn->f_crypto);
  121. if (conn->b_crypto)
  122. crypto_free_cipher_env(conn->b_crypto);
  123. }
  124. if (conn->pkey)
  125. crypto_free_pk_env(conn->pkey);
  126. if(conn->s > 0) {
  127. log(LOG_INFO,"connection_free(): closing fd %d.",conn->s);
  128. close(conn->s);
  129. }
  130. free(conn);
  131. }
  132. int connection_create_listener(struct sockaddr_in *bindaddr, int type) {
  133. connection_t *conn;
  134. int s;
  135. int one=1;
  136. s = socket(PF_INET,SOCK_STREAM,IPPROTO_TCP);
  137. if (s < 0)
  138. {
  139. log(LOG_ERR,"connection_create_listener(): Socket creation failed.");
  140. return -1;
  141. }
  142. setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &one, sizeof(one));
  143. if(bind(s,(struct sockaddr *)bindaddr,sizeof(*bindaddr)) < 0) {
  144. perror("bind ");
  145. log(LOG_ERR,"Could not bind to port %u.",ntohs(bindaddr->sin_port));
  146. return -1;
  147. }
  148. if(listen(s,SOMAXCONN) < 0) {
  149. log(LOG_ERR,"Could not listen on port %u.",ntohs(bindaddr->sin_port));
  150. return -1;
  151. }
  152. fcntl(s, F_SETFL, O_NONBLOCK); /* set s to non-blocking */
  153. conn = connection_new(type);
  154. if(!conn) {
  155. log(LOG_DEBUG,"connection_create_listener(): connection_new failed. Giving up.");
  156. return -1;
  157. }
  158. conn->s = s;
  159. if(connection_add(conn) < 0) { /* no space, forget it */
  160. log(LOG_DEBUG,"connection_create_listener(): connection_add failed. Giving up.");
  161. connection_free(conn);
  162. return -1;
  163. }
  164. log(LOG_DEBUG,"connection_create_listener(): Listening on port %u.",ntohs(bindaddr->sin_port));
  165. conn->state = LISTENER_STATE_READY;
  166. connection_start_reading(conn);
  167. return 0;
  168. }
  169. int connection_handle_listener_read(connection_t *conn, int new_type, int new_state) {
  170. int news; /* the new socket */
  171. connection_t *newconn;
  172. struct sockaddr_in remote; /* information about the remote peer when connecting to other routers */
  173. int remotelen = sizeof(struct sockaddr_in); /* length of the remote address */
  174. news = accept(conn->s,(struct sockaddr *)&remote,&remotelen);
  175. if (news == -1) { /* accept() error */
  176. if(errno==EAGAIN)
  177. return 0; /* he hung up before we could accept(). that's fine. */
  178. /* else there was a real error. */
  179. log(LOG_ERR,"connection_handle_listener_read(): accept() failed. Closing.");
  180. return -1;
  181. }
  182. log(LOG_INFO,"Connection accepted on socket %d (child of fd %d).",news, conn->s);
  183. fcntl(news, F_SETFL, O_NONBLOCK); /* set news to non-blocking */
  184. newconn = connection_new(new_type);
  185. newconn->s = news;
  186. if(!connection_speaks_cells(newconn)) {
  187. newconn->receiver_bucket = -1;
  188. newconn->bandwidth = -1;
  189. }
  190. newconn->address = strdup(inet_ntoa(remote.sin_addr)); /* remember the remote address */
  191. newconn->addr = ntohl(remote.sin_addr.s_addr);
  192. newconn->port = ntohs(remote.sin_port);
  193. if(connection_add(newconn) < 0) { /* no space, forget it */
  194. connection_free(newconn);
  195. return 0; /* no need to tear down the parent */
  196. }
  197. log(LOG_DEBUG,"connection_handle_listener_read(): socket %d entered state %d.",newconn->s, new_state);
  198. newconn->state = new_state;
  199. connection_start_reading(newconn);
  200. return 0;
  201. }
  202. int retry_all_connections(int role, uint16_t or_listenport,
  203. uint16_t op_listenport, uint16_t ap_listenport, uint16_t dir_listenport) {
  204. /* start all connections that should be up but aren't */
  205. struct sockaddr_in bindaddr; /* where to bind */
  206. if(role & ROLE_OR_CONNECT_ALL) {
  207. router_retry_connections();
  208. }
  209. memset(&bindaddr,0,sizeof(struct sockaddr_in));
  210. bindaddr.sin_family = AF_INET;
  211. bindaddr.sin_addr.s_addr = htonl(INADDR_ANY); /* anyone can connect */
  212. if(role & ROLE_OR_LISTEN) {
  213. bindaddr.sin_port = htons(or_listenport);
  214. if(!connection_get_by_type(CONN_TYPE_OR_LISTENER)) {
  215. connection_or_create_listener(&bindaddr);
  216. }
  217. }
  218. if(role & ROLE_OP_LISTEN) {
  219. bindaddr.sin_port = htons(op_listenport);
  220. if(!connection_get_by_type(CONN_TYPE_OP_LISTENER)) {
  221. connection_op_create_listener(&bindaddr);
  222. }
  223. }
  224. if(role & ROLE_DIR_LISTEN) {
  225. bindaddr.sin_port = htons(dir_listenport);
  226. if(!connection_get_by_type(CONN_TYPE_DIR_LISTENER)) {
  227. connection_dir_create_listener(&bindaddr);
  228. }
  229. }
  230. if(role & ROLE_AP_LISTEN) {
  231. bindaddr.sin_port = htons(ap_listenport);
  232. inet_aton("127.0.0.1", &(bindaddr.sin_addr)); /* the AP listens only on localhost! */
  233. if(!connection_get_by_type(CONN_TYPE_AP_LISTENER)) {
  234. connection_ap_create_listener(&bindaddr);
  235. }
  236. }
  237. return 0;
  238. }
  239. int connection_read_to_buf(connection_t *conn) {
  240. int read_result;
  241. struct timeval now;
  242. if(connection_speaks_cells(conn)) {
  243. assert(conn->receiver_bucket >= 0);
  244. }
  245. if(!connection_speaks_cells(conn)) {
  246. assert(conn->receiver_bucket < 0);
  247. }
  248. if(gettimeofday(&now,NULL) < 0)
  249. return -1;
  250. conn->timestamp_lastread = now.tv_sec;
  251. read_result = read_to_buf(conn->s, conn->receiver_bucket, &conn->inbuf, &conn->inbuflen,
  252. &conn->inbuf_datalen, &conn->inbuf_reached_eof);
  253. // log(LOG_DEBUG,"connection_read_to_buf(): read_to_buf returned %d.",read_result);
  254. if(read_result >= 0 && connection_speaks_cells(conn)) {
  255. conn->receiver_bucket -= read_result;
  256. if(conn->receiver_bucket <= 0) {
  257. // log(LOG_DEBUG,"connection_read_to_buf() stopping reading, receiver bucket full.");
  258. connection_stop_reading(conn);
  259. /* If we're not in 'open' state here, then we're never going to finish the
  260. * handshake, because we'll never increment the receiver_bucket. But we
  261. * can't check for that here, because the buf we just read might have enough
  262. * on it to finish the handshake. So we check for that in check_conn_read().
  263. */
  264. }
  265. }
  266. return read_result;
  267. }
  268. int connection_fetch_from_buf(char *string, int len, connection_t *conn) {
  269. return fetch_from_buf(string, len, &conn->inbuf, &conn->inbuflen, &conn->inbuf_datalen);
  270. }
  271. int connection_find_on_inbuf(char *string, int len, connection_t *conn) {
  272. return find_on_inbuf(string, len, conn->inbuf, conn->inbuf_datalen);
  273. }
  274. int connection_wants_to_flush(connection_t *conn) {
  275. return conn->outbuf_flushlen;
  276. }
  277. int connection_outbuf_too_full(connection_t *conn) {
  278. return (conn->outbuf_flushlen > 10*CELL_PAYLOAD_SIZE);
  279. }
  280. int connection_flush_buf(connection_t *conn) {
  281. return flush_buf(conn->s, &conn->outbuf, &conn->outbuflen, &conn->outbuf_flushlen, &conn->outbuf_datalen);
  282. }
  283. int connection_write_to_buf(char *string, int len, connection_t *conn) {
  284. struct timeval now;
  285. if(gettimeofday(&now,NULL) < 0)
  286. return -1;
  287. if(!len)
  288. return 0;
  289. if(conn->marked_for_close)
  290. return 0;
  291. conn->timestamp_lastwritten = now.tv_sec;
  292. if( (!connection_speaks_cells(conn)) ||
  293. (!connection_state_is_open(conn)) ||
  294. (options.LinkPadding == 0) ) {
  295. /* connection types other than or and op, or or/op not in 'open' state, should flush immediately */
  296. /* also flush immediately if we're not doing LinkPadding, since otherwise it will never flush */
  297. connection_start_writing(conn);
  298. conn->outbuf_flushlen += len;
  299. }
  300. return write_to_buf(string, len, &conn->outbuf, &conn->outbuflen, &conn->outbuf_datalen);
  301. }
  302. int connection_receiver_bucket_should_increase(connection_t *conn) {
  303. assert(conn);
  304. if(!connection_speaks_cells(conn))
  305. return 0; /* edge connections don't use receiver_buckets */
  306. if(conn->receiver_bucket > 10*conn->bandwidth)
  307. return 0;
  308. return 1;
  309. }
  310. void connection_increment_receiver_bucket (connection_t *conn) {
  311. assert(conn);
  312. if(connection_receiver_bucket_should_increase(conn)) {
  313. /* yes, the receiver_bucket can become overfull here. But not by much. */
  314. conn->receiver_bucket += conn->bandwidth*1.1;
  315. if(connection_state_is_open(conn)) {
  316. /* if we're in state 'open', then start reading again */
  317. connection_start_reading(conn);
  318. }
  319. }
  320. }
  321. int connection_speaks_cells(connection_t *conn) {
  322. assert(conn);
  323. if(conn->type == CONN_TYPE_OR || conn->type == CONN_TYPE_OP)
  324. return 1;
  325. return 0;
  326. }
  327. int connection_is_listener(connection_t *conn) {
  328. if(conn->type == CONN_TYPE_OP_LISTENER ||
  329. conn->type == CONN_TYPE_OR_LISTENER ||
  330. conn->type == CONN_TYPE_AP_LISTENER ||
  331. conn->type == CONN_TYPE_DIR_LISTENER)
  332. return 1;
  333. return 0;
  334. }
  335. int connection_state_is_open(connection_t *conn) {
  336. assert(conn);
  337. if((conn->type == CONN_TYPE_OR && conn->state == OR_CONN_STATE_OPEN) ||
  338. (conn->type == CONN_TYPE_OP && conn->state == OP_CONN_STATE_OPEN) ||
  339. (conn->type == CONN_TYPE_AP && conn->state == AP_CONN_STATE_OPEN) ||
  340. (conn->type == CONN_TYPE_EXIT && conn->state == EXIT_CONN_STATE_OPEN))
  341. return 1;
  342. return 0;
  343. }
  344. void connection_send_cell(connection_t *conn) {
  345. cell_t cell;
  346. int bytes_in_full_flushlen;
  347. /* this function only gets called if options.LinkPadding is 1 */
  348. assert(options.LinkPadding == 1);
  349. assert(conn);
  350. if(!connection_speaks_cells(conn)) {
  351. /* this conn doesn't speak cells. do nothing. */
  352. return;
  353. }
  354. if(!connection_state_is_open(conn)) {
  355. /* it's not in 'open' state, all data should already be waiting to be flushed */
  356. assert(conn->outbuf_datalen == conn->outbuf_flushlen);
  357. return;
  358. }
  359. #if 0 /* use to send evenly spaced cells, but not padding */
  360. if(conn->outbuf_datalen - conn->outbuf_flushlen >= sizeof(cell_t)) {
  361. conn->outbuf_flushlen += sizeof(cell_t); /* instruct it to send a cell */
  362. connection_start_writing(conn);
  363. }
  364. #endif
  365. connection_increment_send_timeval(conn); /* update when we'll send the next cell */
  366. bytes_in_full_flushlen = conn->bandwidth / 100; /* 10ms worth */
  367. if(bytes_in_full_flushlen < 10*sizeof(cell_t))
  368. bytes_in_full_flushlen = 10*sizeof(cell_t); /* but at least 10 cells worth */
  369. if(conn->outbuf_flushlen > bytes_in_full_flushlen - sizeof(cell_t)) {
  370. /* if we would exceed bytes_in_full_flushlen by adding a new cell */
  371. return;
  372. }
  373. if(conn->outbuf_datalen - conn->outbuf_flushlen < sizeof(cell_t)) {
  374. /* we need to queue a padding cell first */
  375. memset(&cell,0,sizeof(cell_t));
  376. cell.command = CELL_PADDING;
  377. connection_write_cell_to_buf(&cell, conn);
  378. }
  379. conn->outbuf_flushlen += sizeof(cell_t); /* instruct it to send a cell */
  380. connection_start_writing(conn);
  381. }
  382. void connection_increment_send_timeval(connection_t *conn) {
  383. /* add "1000000 * sizeof(cell_t) / conn->bandwidth" microseconds to conn->send_timeval */
  384. /* FIXME should perhaps use ceil() of this. For now I simply add 1. */
  385. tv_addms(&conn->send_timeval, 1+1000 * sizeof(cell_t) / conn->bandwidth);
  386. }
  387. void connection_init_timeval(connection_t *conn) {
  388. assert(conn);
  389. if(gettimeofday(&conn->send_timeval,NULL) < 0)
  390. return;
  391. connection_increment_send_timeval(conn);
  392. }
  393. int connection_send_destroy(aci_t aci, connection_t *conn) {
  394. cell_t cell;
  395. assert(conn);
  396. if(!connection_speaks_cells(conn)) {
  397. log(LOG_INFO,"connection_send_destroy(): Aci %d: At an edge. Marking connection for close.", aci);
  398. conn->marked_for_close = 1;
  399. return 0;
  400. }
  401. cell.aci = aci;
  402. cell.command = CELL_DESTROY;
  403. log(LOG_INFO,"connection_send_destroy(): Sending destroy (aci %d).",aci);
  404. return connection_write_cell_to_buf(&cell, conn);
  405. }
  406. int connection_write_cell_to_buf(const cell_t *cellp, connection_t *conn) {
  407. char networkcell[CELL_NETWORK_SIZE];
  408. char *n = networkcell;
  409. memset(n,0,CELL_NETWORK_SIZE); /* zero it out to start */
  410. *(aci_t *)n = htons(cellp->aci);
  411. *(n+2) = cellp->command;
  412. *(n+3) = cellp->length;
  413. /* seq is reserved, leave zero */
  414. memcpy(n+8,cellp->payload,CELL_PAYLOAD_SIZE);
  415. if(connection_encrypt_cell(n,conn)<0) {
  416. return -1;
  417. }
  418. return connection_write_to_buf(n, CELL_NETWORK_SIZE, conn);
  419. }
  420. int connection_encrypt_cell(char *cellp, connection_t *conn) {
  421. char cryptcell[CELL_NETWORK_SIZE];
  422. #if 0
  423. int x;
  424. char *px;
  425. printf("Sending: Cell header plaintext: ");
  426. px = (char *)cellp;
  427. for(x=0;x<8;x++) {
  428. printf("%u ",px[x]);
  429. }
  430. printf("\n");
  431. #endif
  432. if(crypto_cipher_encrypt(conn->f_crypto, cellp, CELL_NETWORK_SIZE, cryptcell)) {
  433. log(LOG_ERR,"Could not encrypt cell for connection %s:%u.",conn->address,conn->port);
  434. return -1;
  435. }
  436. #if 0
  437. printf("Sending: Cell header crypttext: ");
  438. px = (char *)&newcell;
  439. for(x=0;x<8;x++) {
  440. printf("%u ",px[x]);
  441. }
  442. printf("\n");
  443. #endif
  444. memcpy(cellp,cryptcell,CELL_NETWORK_SIZE);
  445. return 0;
  446. }
  447. int connection_process_inbuf(connection_t *conn) {
  448. assert(conn);
  449. switch(conn->type) {
  450. case CONN_TYPE_OP:
  451. return connection_op_process_inbuf(conn);
  452. case CONN_TYPE_OR:
  453. return connection_or_process_inbuf(conn);
  454. case CONN_TYPE_EXIT:
  455. return connection_exit_process_inbuf(conn);
  456. case CONN_TYPE_AP:
  457. return connection_ap_process_inbuf(conn);
  458. case CONN_TYPE_DIR:
  459. return connection_dir_process_inbuf(conn);
  460. case CONN_TYPE_DNSMASTER:
  461. return connection_dns_process_inbuf(conn);
  462. default:
  463. log(LOG_DEBUG,"connection_process_inbuf() got unexpected conn->type.");
  464. return -1;
  465. }
  466. }
  467. int connection_package_raw_inbuf(connection_t *conn) {
  468. int amount_to_process;
  469. cell_t cell;
  470. circuit_t *circ;
  471. assert(conn);
  472. assert(!connection_speaks_cells(conn));
  473. /* this function should never get called if the receive_topicwindow is 0 */
  474. repeat_connection_package_raw_inbuf:
  475. amount_to_process = conn->inbuf_datalen;
  476. if(!amount_to_process)
  477. return 0;
  478. if(amount_to_process > CELL_PAYLOAD_SIZE - TOPIC_HEADER_SIZE) {
  479. cell.length = CELL_PAYLOAD_SIZE - TOPIC_HEADER_SIZE;
  480. } else {
  481. cell.length = amount_to_process;
  482. }
  483. if(connection_fetch_from_buf(cell.payload+TOPIC_HEADER_SIZE, cell.length, conn) < 0)
  484. return -1;
  485. circ = circuit_get_by_conn(conn);
  486. if(!circ) {
  487. log(LOG_DEBUG,"connection_package_raw_inbuf(): conn has no circuits!");
  488. return -1;
  489. }
  490. log(LOG_DEBUG,"connection_package_raw_inbuf(): (%d) Packaging %d bytes (%d waiting).",conn->s,cell.length, amount_to_process);
  491. *(uint16_t *)(cell.payload+2) = htons(conn->topic_id);
  492. *cell.payload = TOPIC_COMMAND_DATA;
  493. cell.length += TOPIC_HEADER_SIZE;
  494. cell.command = CELL_DATA;
  495. if(conn->type == CONN_TYPE_EXIT) {
  496. cell.aci = circ->p_aci;
  497. if(circuit_deliver_data_cell_from_edge(&cell, circ, EDGE_EXIT) < 0) {
  498. log(LOG_DEBUG,"connection_package_raw_inbuf(): circuit_deliver_data_cell_from_edge (backward) failed. Closing.");
  499. circuit_close(circ);
  500. return 0;
  501. }
  502. assert(conn->n_receive_topicwindow > 0);
  503. if(--conn->n_receive_topicwindow <= 0) { /* is it 0 after decrement? */
  504. connection_stop_reading(conn);
  505. log(LOG_DEBUG,"connection_package_raw_inbuf(): receive_topicwindow at exit reached 0.");
  506. return 0; /* don't process the inbuf any more */
  507. }
  508. log(LOG_DEBUG,"connection_package_raw_inbuf(): receive_topicwindow at exit is %d",conn->n_receive_topicwindow);
  509. } else { /* send it forward. we're an AP */
  510. assert(conn->type == CONN_TYPE_AP);
  511. cell.aci = circ->n_aci;
  512. if(circuit_deliver_data_cell_from_edge(&cell, circ, EDGE_AP) < 0) {
  513. log(LOG_DEBUG,"connection_package_raw_inbuf(): circuit_deliver_data_cell_from_edge (forward) failed. Closing.");
  514. circuit_close(circ);
  515. return 0;
  516. }
  517. assert(conn->p_receive_topicwindow > 0);
  518. if(--conn->p_receive_topicwindow <= 0) { /* is it 0 after decrement? */
  519. connection_stop_reading(conn);
  520. log(LOG_DEBUG,"connection_package_raw_inbuf(): receive_topicwindow at AP reached 0.");
  521. return 0; /* don't process the inbuf any more */
  522. }
  523. log(LOG_DEBUG,"connection_package_raw_inbuf(): receive_topicwindow at AP is %d",conn->p_receive_topicwindow);
  524. }
  525. if(amount_to_process > CELL_PAYLOAD_SIZE - TOPIC_HEADER_SIZE) {
  526. log(LOG_DEBUG,"connection_package_raw_inbuf(): recursing.");
  527. goto repeat_connection_package_raw_inbuf;
  528. }
  529. return 0;
  530. }
  531. int connection_consider_sending_sendme(connection_t *conn, int edge_type) {
  532. circuit_t *circ;
  533. cell_t cell;
  534. if(connection_outbuf_too_full(conn))
  535. return 0;
  536. circ = circuit_get_by_conn(conn);
  537. if(!circ) {
  538. /* this can legitimately happen if the destroy has already arrived and torn down the circuit */
  539. log(LOG_DEBUG,"connection_consider_sending_sendme(): No circuit associated with conn. Skipping.");
  540. return 0;
  541. }
  542. *(uint16_t *)(cell.payload+2) = htons(conn->topic_id);
  543. *cell.payload = TOPIC_COMMAND_SENDME;
  544. cell.length += TOPIC_HEADER_SIZE;
  545. cell.command = CELL_DATA;
  546. if(edge_type == EDGE_EXIT) { /* we're at an exit */
  547. if(conn->p_receive_topicwindow < TOPICWINDOW_START - TOPICWINDOW_INCREMENT) {
  548. log(LOG_DEBUG,"connection_consider_sending_sendme(): Outbuf %d, Queueing topic sendme back.", conn->outbuf_flushlen);
  549. conn->p_receive_topicwindow += TOPICWINDOW_INCREMENT;
  550. cell.aci = circ->p_aci;
  551. if(circuit_deliver_data_cell_from_edge(&cell, circ, edge_type) < 0) {
  552. log(LOG_DEBUG,"connection_consider_sending_sendme(): circuit_deliver_data_cell_from_edge (backward) failed. Closing.");
  553. circuit_close(circ);
  554. return 0;
  555. }
  556. }
  557. } else { /* we're at an AP */
  558. assert(edge_type == EDGE_AP);
  559. if(conn->n_receive_topicwindow < TOPICWINDOW_START-TOPICWINDOW_INCREMENT) {
  560. log(LOG_DEBUG,"connection_consider_sending_sendme(): Outbuf %d, Queueing topic sendme forward.", conn->outbuf_flushlen);
  561. conn->n_receive_topicwindow += TOPICWINDOW_INCREMENT;
  562. cell.aci = circ->n_aci;
  563. if(circuit_deliver_data_cell_from_edge(&cell, circ, edge_type) < 0) {
  564. log(LOG_DEBUG,"connection_consider_sending_sendme(): circuit_deliver_data_cell_from_edge (forward) failed. Closing.");
  565. circuit_close(circ);
  566. return 0;
  567. }
  568. }
  569. }
  570. return 0;
  571. }
  572. int connection_finished_flushing(connection_t *conn) {
  573. assert(conn);
  574. // log(LOG_DEBUG,"connection_finished_flushing() entered. Socket %u.", conn->s);
  575. switch(conn->type) {
  576. case CONN_TYPE_AP:
  577. return connection_ap_finished_flushing(conn);
  578. case CONN_TYPE_OP:
  579. return connection_op_finished_flushing(conn);
  580. case CONN_TYPE_OR:
  581. return connection_or_finished_flushing(conn);
  582. case CONN_TYPE_EXIT:
  583. return connection_exit_finished_flushing(conn);
  584. case CONN_TYPE_DIR:
  585. return connection_dir_finished_flushing(conn);
  586. case CONN_TYPE_DNSMASTER:
  587. return connection_dns_finished_flushing(conn);
  588. default:
  589. log(LOG_DEBUG,"connection_finished_flushing() got unexpected conn->type.");
  590. return -1;
  591. }
  592. }
  593. int connection_process_cell_from_inbuf(connection_t *conn) {
  594. /* check if there's a whole cell there.
  595. * if yes, pull it off, decrypt it, and process it.
  596. */
  597. char crypted[CELL_NETWORK_SIZE];
  598. char outbuf[1024];
  599. // int x;
  600. cell_t cell;
  601. if(conn->inbuf_datalen < CELL_NETWORK_SIZE) /* entire response available? */
  602. return 0; /* not yet */
  603. if(connection_fetch_from_buf(crypted,CELL_NETWORK_SIZE,conn) < 0) {
  604. return -1;
  605. }
  606. #if 0
  607. printf("Cell header crypttext: ");
  608. for(x=0;x<8;x++) {
  609. printf("%u ",crypted[x]);
  610. }
  611. printf("\n");
  612. #endif
  613. /* decrypt */
  614. if(crypto_cipher_decrypt(conn->b_crypto,crypted,CELL_NETWORK_SIZE,outbuf)) {
  615. log(LOG_ERR,"connection_process_cell_from_inbuf(): Decryption failed, dropping.");
  616. return connection_process_inbuf(conn); /* process the remainder of the buffer */
  617. }
  618. // log(LOG_DEBUG,"connection_process_cell_from_inbuf(): Cell decrypted (%d bytes).",outlen);
  619. #if 0
  620. printf("Cell header plaintext: ");
  621. for(x=0;x<8;x++) {
  622. printf("%u ",outbuf[x]);
  623. }
  624. printf("\n");
  625. #endif
  626. /* retrieve cell info from outbuf (create the host-order struct from the network-order string) */
  627. memset(&cell,0,sizeof(cell_t)); /* zero it out to start */
  628. cell.aci = ntohs(*(aci_t *)outbuf);
  629. cell.command = *(outbuf+2);
  630. cell.length = *(outbuf+3);
  631. memcpy(cell.payload, outbuf+8, CELL_PAYLOAD_SIZE);
  632. // log(LOG_DEBUG,"connection_process_cell_from_inbuf(): Decrypted cell is of type %u (ACI %u).",cellp->command,cellp->aci);
  633. command_process_cell(&cell, conn);
  634. return connection_process_inbuf(conn); /* process the remainder of the buffer */
  635. }