directory.c 8.75 KB
Newer Older
1
/* Copyright 2001,2002 Roger Dingledine, Matej Pfajfar. */
2
3
4
5
6
7
8
9
10
11
12
13
14
/* See LICENSE for licensing information */
/* $Id$ */

#include "or.h"

#define MAX_DIR_SIZE 50000 /* XXX, big enough? */

/********* START VARIABLES **********/

extern or_options_t options; /* command-line and config-file options */

static char the_directory[MAX_DIR_SIZE+1];
static int directorylen=0;
15
static int reading_headers=0;
16

17
18
static char getstring[] = "GET / HTTP/1.0\r\n\r\n";
static char answerstring[] = "HTTP/1.0 200 OK\r\n\r\n";
19
20
21
22
23
24
25
26
27
28
29

/********* END VARIABLES ************/

void directory_initiate_fetch(routerinfo_t *router) {
  connection_t *conn;
  struct sockaddr_in router_addr;
  int s;

  if(!router) /* i guess they didn't have one in mind for me to use */
    return;

30
31
32
33
34
  if(connection_get_by_type(CONN_TYPE_DIR)) /* there's already a fetch running */
    return;

  log(LOG_DEBUG,"directory_initiate_fetch(): initiating directory fetch");

35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
  conn = connection_new(CONN_TYPE_DIR);
  if(!conn)
    return;

  /* set up conn so it's got all the data we need to remember */
  conn->addr = router->addr, conn->port = router->or_port; /* NOTE we store or_port here always */
  conn->address = strdup(router->address);
  conn->receiver_bucket = -1; /* edge connections don't do receiver buckets */
  conn->bandwidth = -1;

  s=socket(PF_INET,SOCK_STREAM,IPPROTO_TCP);
  if (s < 0)
  { 
    log(LOG_ERR,"directory_initiate_fetch(): Error creating network socket.");
    connection_free(conn);
    return;
  }
  fcntl(s, F_SETFL, O_NONBLOCK); /* set s to non-blocking */

  memset((void *)&router_addr,0,sizeof(router_addr));
  router_addr.sin_family = AF_INET;
  router_addr.sin_port = htons(router->dir_port);
57
  router_addr.sin_addr.s_addr = htonl(router->addr);
58

59
  log(LOG_DEBUG,"directory_initiate_fetch(): Trying to connect to %s:%u.",router->address,router->dir_port);
60
61
62
63

  if(connect(s,(struct sockaddr *)&router_addr,sizeof(router_addr)) < 0){
    if(errno != EINPROGRESS){
      /* yuck. kill it. */
64
      router_forget_router(conn->addr, conn->port); /* don't try him again */
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
      connection_free(conn);
      return;
    } else {
      /* it's in progress. set state appropriately and return. */
      conn->s = s;

      if(connection_add(conn) < 0) { /* no space, forget it */
        connection_free(conn);
        return;
      }

      log(LOG_DEBUG,"directory_initiate_fetch(): connect in progress.");
      connection_watch_events(conn, POLLIN | POLLOUT); /* writable indicates finish, readable indicates broken link */
      conn->state = DIR_CONN_STATE_CONNECTING;
      return;
    }
  }

  /* it succeeded. we're connected. */
  conn->s = s;

  if(connection_add(conn) < 0) { /* no space, forget it */
    connection_free(conn);
    return;
  }

  log(LOG_DEBUG,"directory_initiate_fetch(): Connection to router %s:%u established.",router->address,router->dir_port);

  if(directory_send_command(conn) < 0) {
    connection_remove(conn);
    connection_free(conn);
  }
}

int directory_send_command(connection_t *conn) {

  assert(conn && conn->type == CONN_TYPE_DIR);

  if(connection_write_to_buf(getstring, strlen(getstring), conn) < 0) {
    log(LOG_DEBUG,"directory_send_command(): Couldn't write command to buffer.");
    return -1;
  }

  conn->state = DIR_CONN_STATE_SENDING_COMMAND;
  return 0;
}

void directory_rebuild(void) {

  dump_directory_to_string(the_directory, MAX_DIR_SIZE);
115
  log(LOG_DEBUG,"New directory:\n%s",the_directory);  
116
117
118
119
120
121
122
123
124
125
126
127
128
129
  directorylen = strlen(the_directory);

}

int connection_dir_process_inbuf(connection_t *conn) {

  assert(conn && conn->type == CONN_TYPE_DIR);

  if(conn->inbuf_reached_eof) {
    if(conn->state != DIR_CONN_STATE_READING) {
      log(LOG_DEBUG,"connection_dir_process_inbuf(): conn reached eof, not reading. Closing.");
      return -1;
    }
    /* eof reached, kill it, but first process the_directory and learn about new routers. */
Roger Dingledine's avatar
cleanup    
Roger Dingledine committed
130
131
//    log(LOG_DEBUG,"connection_dir_process_inbuf(): conn reached eof. Processing directory.");
    log(LOG_DEBUG,"connection_dir_process_inbuf(): Received directory (size %d)\n%s", directorylen, the_directory);
132
133
134
135
    if(directorylen == 0) {
      log(LOG_DEBUG,"connection_dir_process_inbuf(): Empty directory. Ignoring.");
      return -1;
    }
136
    if(router_get_list_from_string(the_directory) < 0) {
137
138
      log(LOG_DEBUG,"connection_dir_process_inbuf(): ...but parsing failed. Ignoring.");
    }
Roger Dingledine's avatar
cleanup    
Roger Dingledine committed
139
    if(options.Role & ROLE_OR_CONNECT_ALL) { /* connect to them all */
140
      router_retry_connections();
Roger Dingledine's avatar
cleanup    
Roger Dingledine committed
141
    }
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
    return -1;
  }

  switch(conn->state) {
    case DIR_CONN_STATE_COMMAND_WAIT:
      return directory_handle_command(conn);
    case DIR_CONN_STATE_READING:
      return directory_handle_reading(conn);
    default:
      log(LOG_DEBUG,"connection_dir_process_inbuf(): Got data while writing; Ignoring.");
      break;
  }

  return 0;
}

int directory_handle_command(connection_t *conn) {
159
  char buf[15];
160
161
162
163
164
165
166
167

  assert(conn && conn->type == CONN_TYPE_DIR);

  if(conn->inbuf_datalen < strlen(getstring)) { /* entire response available? */
    log(LOG_DEBUG,"directory_handle_command(): Entire command not here yet. Waiting.");
    return 0; /* not yet */
  }

168
169
170
171
172
173
174
175
  if(connection_fetch_from_buf(buf,strlen(getstring),conn) < 0) {
    return -1;    
  }

  if(strncasecmp(buf,getstring,strlen("GET / HTTP/"))) {
    log(LOG_DEBUG,"directory_handle_command(): Command doesn't seem to be a get. Closing,");
    return -1;
  }
176
177
178
179
180
181
182

  if(directorylen == 0) {
    log(LOG_DEBUG,"directory_handle_command(): My directory is empty. Closing.");
    return -1;
  }

  log(LOG_DEBUG,"directory_handle_command(): Dumping directory to client."); 
183
184
  if((connection_write_to_buf(answerstring, strlen(answerstring), conn) < 0) ||
     (connection_write_to_buf(the_directory, directorylen, conn) < 0)) {
185
186
187
188
189
190
191
192
193
194
    log(LOG_DEBUG,"directory_handle_command(): my outbuf is full. Oops.");
    return -1;
  }

  conn->state = DIR_CONN_STATE_WRITING;
  return 0;
}

int directory_handle_reading(connection_t *conn) {
  int amt;
195
  char *headers;
196
197
198

  assert(conn && conn->type == CONN_TYPE_DIR);

199
200
201
202
203
204
205
206
207
208
209
210
211
212
  if(reading_headers) {
    amt = connection_find_on_inbuf("\r\n\r\n", 4, conn);
    if(amt < 0) /* not there yet */
      return 0;
    headers = malloc(amt+1);
    if(connection_fetch_from_buf(headers,amt,conn) < 0) {
      log(LOG_DEBUG,"directory_handle_reading(): fetch_from_buf failed (reading headers).");
      return -1;
    }
    headers[amt] = 0; /* null terminate it, */
    free(headers); /* and then throw it away */
    reading_headers = 0;
  }

213
214
215
216
217
218
219
220
221
222
223
  amt = conn->inbuf_datalen;

  if(amt + directorylen >= MAX_DIR_SIZE) {
    log(LOG_DEBUG,"directory_handle_reading(): Directory too large. Failing messily.");
    return -1;
  }

  log(LOG_DEBUG,"directory_handle_reading(): Pulling %d bytes in at offset %d.",
    amt, directorylen);

  if(connection_fetch_from_buf(the_directory+directorylen,amt,conn) < 0) {
224
    log(LOG_DEBUG,"directory_handle_reading(): fetch_from_buf failed (reading dir).");
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
    return -1;    
  }

  directorylen += amt;

  the_directory[directorylen] = 0;

  return 0;
}

int connection_dir_finished_flushing(connection_t *conn) {
  int e, len=sizeof(e);

  assert(conn && conn->type == CONN_TYPE_DIR);

  switch(conn->state) {
    case DIR_CONN_STATE_CONNECTING:
      if (getsockopt(conn->s, SOL_SOCKET, SO_ERROR, &e, &len) < 0)  { /* not yet */
        if(errno != EINPROGRESS){
          /* yuck. kill it. */
          log(LOG_DEBUG,"connection_dir_finished_flushing(): in-progress connect failed. Removing.");
246
          router_forget_router(conn->addr, conn->port); /* don't try him again */
247
248
249
250
251
252
253
          return -1;
        } else {
          return 0; /* no change, see if next time is better */
        }
      }
      /* the connect has finished. */

254
      log(LOG_DEBUG,"connection_dir_finished_flushing(): Dir connection to router %s:%u established.",
255
256
257
258
259
260
          conn->address,conn->port);

      return directory_send_command(conn);
    case DIR_CONN_STATE_SENDING_COMMAND:
      log(LOG_DEBUG,"connection_dir_finished_flushing(): client finished sending command.");
      directorylen = 0;
261
      reading_headers = 1;
262
263
264
265
266
267
268
269
270
271
272
273
274
275
      conn->state = DIR_CONN_STATE_READING;
      connection_watch_events(conn, POLLIN);
      return 0;
    case DIR_CONN_STATE_WRITING:
      log(LOG_DEBUG,"connection_dir_finished_flushing(): Finished writing directory. Closing.");
      return -1; /* kill it */
    default:
      log(LOG_DEBUG,"Bug: connection_dir_finished_flushing() called in unexpected state.");
      return 0;
  }

  return 0;
}

276
int connection_dir_create_listener(struct sockaddr_in *bindaddr) {
277
  log(LOG_DEBUG,"connection_create_dir_listener starting");
278
  return connection_create_listener(bindaddr, CONN_TYPE_DIR_LISTENER);
279
280
281
282
283
284
285
}

int connection_dir_handle_listener_read(connection_t *conn) {
  log(LOG_NOTICE,"Dir: Received a connection request. Waiting for command.");
  return connection_handle_listener_read(conn, CONN_TYPE_DIR, DIR_CONN_STATE_COMMAND_WAIT);
}