src/or/command.c

   1 /* Copyright 2001,2002,2003 Roger Dingledine, Matej Pfajfar. */
   2 /* See LICENSE for licensing information */
   3 /* $Id$ */
   4
   5 #include "or.h"
   6
   7 extern or_options_t options; /* command-line and config-file options */
   8
   9 unsigned long stats_n_padding_cells_processed = 0;
  10 unsigned long stats_n_create_cells_processed = 0;
  11 unsigned long stats_n_created_cells_processed = 0;
  12 unsigned long stats_n_relay_cells_processed = 0;
  13 unsigned long stats_n_destroy_cells_processed = 0;
  14
  15 static void command_process_create_cell(cell_t *cell, connection_t *conn);
  16 static void command_process_created_cell(cell_t *cell, connection_t *conn);
  17 static void command_process_relay_cell(cell_t *cell, connection_t *conn);
  18 static void command_process_destroy_cell(cell_t *cell, connection_t *conn);
  19
  20 static void command_time_process_cell(cell_t *cell, connection_t *conn,
  21                                int *num, int *time,
  22                                void (*func)(cell_t *, connection_t *)) {
  23   struct timeval start, end;
  24   long time_passed;
  25
  26   *num += 1;
  27
  28   tor_gettimeofday(&start);
  29
  30   (*func)(cell, conn);
  31
  32   tor_gettimeofday(&end);
  33   time_passed = tv_udiff(&start, &end) ;
  34
  35   if (time_passed > 10000) { /* more than 10ms */
  36     log_fn(LOG_INFO,"That call just took %ld ms.",time_passed/1000);
  37   }
  38   *time += time_passed;
  39 }
  40
  41 void command_process_cell(cell_t *cell, connection_t *conn) {
  42   static int num_create=0, num_created=0, num_relay=0, num_destroy=0;
  43   static int create_time=0, created_time=0, relay_time=0, destroy_time=0;
  44   static time_t current_second = 0; /* from previous calls to time */
  45   time_t now = time(NULL);
  46
  47   if(now > current_second) { /* the second has rolled over */
  48     /* print stats */
  49     log(LOG_INFO,"At end of second: %d creates (%d ms), %d createds (%d ms), %d relays (%d ms), %d destroys (%d ms)",
  50       num_create, create_time/1000,
  51       num_created, created_time/1000,
  52       num_relay, relay_time/1000,
  53       num_destroy, destroy_time/1000);
  54
  55     /* zero out stats */
  56     num_create = num_created = num_relay = num_destroy = 0;
  57     create_time = created_time = relay_time = destroy_time = 0;
  58
  59     /* remember which second it is, for next time */
  60     current_second = now;
  61   }
  62
  63   switch(cell->command) {
  64     case CELL_PADDING:
  65       ++stats_n_padding_cells_processed;
  66       /* do nothing */
  67       break;
  68     case CELL_CREATE:
  69       ++stats_n_create_cells_processed;
  70       command_time_process_cell(cell, conn, &num_create, &create_time,
  71                                 command_process_create_cell);
  72       break;
  73     case CELL_CREATED:
  74       ++stats_n_created_cells_processed;
  75       command_time_process_cell(cell, conn, &num_created, &created_time,
  76                                 command_process_created_cell);
  77       break;
  78     case CELL_RELAY:
  79       ++stats_n_relay_cells_processed;
  80       command_time_process_cell(cell, conn, &num_relay, &relay_time,
  81                                 command_process_relay_cell);
  82       break;
  83     case CELL_DESTROY:
  84       ++stats_n_destroy_cells_processed;
  85       command_time_process_cell(cell, conn, &num_destroy, &destroy_time,
  86                                 command_process_destroy_cell);
  87       break;
  88     default:
  89       log_fn(LOG_WARN,"Cell of unknown type (%d) received. Dropping.", cell->command);
  90       break;
  91   }
  92 }
  93
  94 static void command_process_create_cell(cell_t *cell, connection_t *conn) {
  95   circuit_t *circ;
  96
  97   circ = circuit_get_by_circ_id_conn(cell->circ_id, conn);
  98
  99   if(circ) {
 100     log_fn(LOG_WARN,"received CREATE cell (circID %d) for known circ. Dropping.", cell->circ_id);
 101     return;
 102   }
 103
 104   circ = circuit_new(cell->circ_id, conn);
 105   circ->state = CIRCUIT_STATE_ONIONSKIN_PENDING;
 106   circ->purpose = CIRCUIT_PURPOSE_OR;
 107
 108   memcpy(circ->onionskin, cell->payload, ONIONSKIN_CHALLENGE_LEN);
 109
 110   /* hand it off to the cpuworkers, and then return */
 111   if(assign_to_cpuworker(NULL, CPUWORKER_TASK_ONION, circ) < 0) {
 112     log_fn(LOG_WARN,"Failed to hand off onionskin. Closing.");
 113     circuit_mark_for_close(circ);
 114     return;
 115   }
 116   log_fn(LOG_DEBUG,"success: handed off onionskin.");
 117 }
 118
 119 static void command_process_created_cell(cell_t *cell, connection_t *conn) {
 120   circuit_t *circ;
 121
 122   circ = circuit_get_by_circ_id_conn(cell->circ_id, conn);
 123
 124   if(!circ) {
 125     log_fn(LOG_INFO,"(circID %d) unknown circ (probably got a destroy earlier). Dropping.", cell->circ_id);
 126     return;
 127   }
 128
 129   if(circ->n_circ_id != cell->circ_id) {
 130     log_fn(LOG_WARN,"got created cell from OPward? Closing.");
 131     circuit_mark_for_close(circ);
 132     return;
 133   }
 134
 135   if(CIRCUIT_IS_ORIGIN(circ)) { /* we're the OP. Handshake this. */
 136     log_fn(LOG_DEBUG,"at OP. Finishing handshake.");
 137     if(circuit_finish_handshake(circ, cell->payload) < 0) {
 138       log_fn(LOG_WARN,"circuit_finish_handshake failed.");
 139       circuit_mark_for_close(circ);
 140       return;
 141     }
 142     log_fn(LOG_DEBUG,"Moving to next skin.");
 143     if(circuit_send_next_onion_skin(circ) < 0) {
 144       log_fn(LOG_INFO,"circuit_send_next_onion_skin failed.");
 145       circuit_mark_for_close(circ); /* XXX push this circuit_close lower */
 146       return;
 147     }
 148   } else { /* pack it into an extended relay cell, and send it. */
 149     log_fn(LOG_DEBUG,"Converting created cell to extended relay cell, sending.");
 150     connection_edge_send_command(NULL, circ, RELAY_COMMAND_EXTENDED,
 151                                  cell->payload, ONIONSKIN_REPLY_LEN, NULL);
 152   }
 153 }
 154
 155 static void command_process_relay_cell(cell_t *cell, connection_t *conn) {
 156   circuit_t *circ;
 157
 158   circ = circuit_get_by_circ_id_conn(cell->circ_id, conn);
 159
 160   if(!circ) {
 161     log_fn(LOG_INFO,"unknown circuit %d on connection to %s:%d. Dropping.",
 162            cell->circ_id, conn->address, conn->port);
 163     return;
 164   }
 165
 166   if(circ->state == CIRCUIT_STATE_ONIONSKIN_PENDING) {
 167     log_fn(LOG_WARN,"circuit in create_wait. Closing.");
 168     circuit_mark_for_close(circ);
 169     return;
 170   }
 171
 172   if(cell->circ_id == circ->p_circ_id) { /* it's an outgoing cell */
 173     if(circuit_receive_relay_cell(cell, circ, CELL_DIRECTION_OUT) < 0) {
 174       log_fn(LOG_WARN,"circuit_receive_relay_cell (forward) failed. Closing.");
 175       circuit_mark_for_close(circ);
 176       return;
 177     }
 178   } else { /* it's an ingoing cell */
 179     if(circuit_receive_relay_cell(cell, circ, CELL_DIRECTION_IN) < 0) {
 180       log_fn(LOG_WARN,"circuit_receive_relay_cell (backward) failed. Closing.");
 181       circuit_mark_for_close(circ);
 182       return;
 183     }
 184   }
 185 }
 186
 187 static void command_process_destroy_cell(cell_t *cell, connection_t *conn) {
 188   circuit_t *circ;
 189
 190   circ = circuit_get_by_circ_id_conn(cell->circ_id, conn);
 191
 192   if(!circ) {
 193     log_fn(LOG_INFO,"unknown circuit %d on connection to %s:%d. Dropping.",
 194            cell->circ_id, conn->address, conn->port);
 195     return;
 196   }
 197
 198   log_fn(LOG_INFO,"Received for circID %d.",cell->circ_id);
 199   if(circ->state == CIRCUIT_STATE_ONIONSKIN_PENDING) {
 200     onion_pending_remove(circ);
 201   }
 202
 203   if(cell->circ_id == circ->p_circ_id || CIRCUIT_IS_ORIGIN(circ)) {
 204     /* either the destroy came from behind, or we're the AP */
 205     circ->p_conn = NULL;
 206     circuit_mark_for_close(circ);
 207   } else { /* the destroy came from ahead */
 208     circ->n_conn = NULL;
 209     log_fn(LOG_DEBUG, "Delivering 'truncated' back.");
 210     connection_edge_send_command(NULL, circ, RELAY_COMMAND_TRUNCATED,
 211                                  NULL, 0, NULL);
 212   }
 213 }
 214
 215 /*
 216   Local Variables:
 217   mode:c
 218   indent-tabs-mode:nil
 219   c-basic-offset:2
 220   End:
 221 */