directory.c 7.7 KB
Newer Older
1
/* Copyright 2001,2002 Roger Dingledine, Matej Pfajfar. */
2
3
4
5
6
7
8
/* See LICENSE for licensing information */
/* $Id$ */

#include "or.h"

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

Roger Dingledine's avatar
Roger Dingledine committed
9
10
11
12
13
static int directory_send_command(connection_t *conn);
static void directory_rebuild(void);
static int directory_handle_command(connection_t *conn);
static int directory_handle_reading(connection_t *conn);

14
15
16
17
18
19
/********* 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;
20
static int reading_headers=0;
21
static int directory_dirty=1;
22

23
static char getstring[] = "GET / HTTP/1.0\r\n\r\n";
Roger Dingledine's avatar
Roger Dingledine committed
24
static char poststring[] = "POST / HTTP/1.0\r\n\r\n";
25
static char answerstring[] = "HTTP/1.0 200 OK\r\n\r\n";
26
27
28
29
30
31
32
33
34

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

void directory_initiate_fetch(routerinfo_t *router) {
  connection_t *conn;

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

35
36
  if(connection_get_by_type(CONN_TYPE_DIR)) { /* there's already a fetch running */
    log_fn(LOG_DEBUG,"Canceling fetch, dir conn already active.");
37
    return;
38
  }
39

40
  log_fn(LOG_DEBUG,"initiating directory fetch");
41

42
43
44
45
46
  conn = connection_new(CONN_TYPE_DIR);
  if(!conn)
    return;

  /* set up conn so it's got all the data we need to remember */
Roger Dingledine's avatar
Roger Dingledine committed
47
48
  conn->addr = router->addr;
  conn->port = router->dir_port;
49
50
51
  conn->address = strdup(router->address);
  conn->receiver_bucket = -1; /* edge connections don't do receiver buckets */
  conn->bandwidth = -1;
Nick Mathewson's avatar
Nick Mathewson committed
52
53
54
  if (router->signing_pkey)
    conn->pkey = crypto_pk_dup_key(router->signing_pkey);
  else {
55
    log_fn(LOG_ERR, "No signing key known for directory %s; signature won't be checked", conn->address);
Nick Mathewson's avatar
Nick Mathewson committed
56
57
    conn->pkey = NULL;
  }
58

59
  if(connection_add(conn) < 0) { /* no space, forget it */
60
61
62
63
    connection_free(conn);
    return;
  }

64
65
  switch(connection_connect(conn, router->address, router->addr, router->dir_port)) {
    case -1:
66
      router_forget_router(conn->addr, conn->port); /* don't try him again */
67
68
      connection_free(conn);
      return;
69
70
    case 0:
      connection_set_poll_socket(conn);
71
72
73
      connection_watch_events(conn, POLLIN | POLLOUT | POLLERR);
      /* writable indicates finish, readable indicates broken link,
         error indicates broken link in windowsland. */
74
75
      conn->state = DIR_CONN_STATE_CONNECTING;
      return;
76
    /* case 1: fall through */
77
78
  }

79
  connection_set_poll_socket(conn);
80
81
82
83
84
85
  if(directory_send_command(conn) < 0) {
    connection_remove(conn);
    connection_free(conn);
  }
}

Roger Dingledine's avatar
Roger Dingledine committed
86
static int directory_send_command(connection_t *conn) {
87
88
89
90

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

  if(connection_write_to_buf(getstring, strlen(getstring), conn) < 0) {
91
    log_fn(LOG_DEBUG,"Couldn't write command to buffer.");
92
93
94
95
96
97
98
    return -1;
  }

  conn->state = DIR_CONN_STATE_SENDING_COMMAND;
  return 0;
}

99
100
101
void directory_set_dirty(void) {
  directory_dirty = 1;
}
102

Roger Dingledine's avatar
Roger Dingledine committed
103
static void directory_rebuild(void) {
104
  if(directory_dirty) {
105
106
107
108
109
    if (dump_signed_directory_to_string(the_directory, MAX_DIR_SIZE,
                                        get_signing_privatekey())) {
      log(LOG_ERR, "Error writing directory");
      return;
    }
110
111
112
113
114
115
    log(LOG_INFO,"New directory:\n%s",the_directory);
    directorylen = strlen(the_directory);
    directory_dirty = 0;
  } else {
    log(LOG_INFO,"Directory still clean, reusing.");
  }
116
117
118
119
120
121
122
123
}

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) {
124
      log_fn(LOG_DEBUG,"conn reached eof, not reading. Closing.");
125
126
127
      return -1;
    }
    /* eof reached, kill it, but first process the_directory and learn about new routers. */
128
    log_fn(LOG_DEBUG,"Received directory (size %d)\n%s", directorylen, the_directory);
129
    if(directorylen == 0) {
130
      log_fn(LOG_DEBUG,"Empty directory. Ignoring.");
131
132
      return -1;
    }
133
    if(router_get_dir_from_string(the_directory, conn->pkey) < 0) {
134
      log_fn(LOG_DEBUG,"...but parsing failed. Ignoring.");
Nick Mathewson's avatar
Nick Mathewson committed
135
    } else {
Roger Dingledine's avatar
Roger Dingledine committed
136
      log_fn(LOG_DEBUG,"and got an %s directory; updated routers.", 
Nick Mathewson's avatar
Nick Mathewson committed
137
          conn->pkey ? "authenticated" : "unauthenticated");
138
    }
139

140
    if(options.OnionRouter) { /* connect to them all */
141
      router_retry_connections();
Roger Dingledine's avatar
cleanup    
Roger Dingledine committed
142
    }
143
144
145
146
147
148
149
150
151
    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:
152
      log_fn(LOG_DEBUG,"Got data while writing; Ignoring.");
153
154
155
156
157
158
      break;
  }

  return 0;
}

Roger Dingledine's avatar
Roger Dingledine committed
159
static int directory_handle_command(connection_t *conn) {
160
  char buf[15];
161
162
163

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

164
  if(conn->inbuf_datalen < (int)strlen(getstring)) { /* entire response available? */
165
    log_fn(LOG_DEBUG,"Entire command not here yet. Waiting.");
166
167
168
    return 0; /* not yet */
  }

169
  connection_fetch_from_buf(buf,strlen(getstring),conn);
170
171

  if(strncasecmp(buf,getstring,strlen("GET / HTTP/"))) {
172
    log_fn(LOG_DEBUG,"Command doesn't seem to be a get. Closing,");
173
174
    return -1;
  }
175

176
177
  directory_rebuild(); /* rebuild it now, iff it's dirty */

178
  if(directorylen == 0) {
179
    log_fn(LOG_DEBUG,"My directory is empty. Closing.");
180
181
182
    return -1;
  }

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

  conn->state = DIR_CONN_STATE_WRITING;
  return 0;
}

Roger Dingledine's avatar
Roger Dingledine committed
194
static int directory_handle_reading(connection_t *conn) {
195
  int amt;
196
  char *headers;
197
198
199

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

200
201
202
203
  if(reading_headers) {
    amt = connection_find_on_inbuf("\r\n\r\n", 4, conn);
    if(amt < 0) /* not there yet */
      return 0;
204
    headers = tor_malloc(amt+1);
205
    connection_fetch_from_buf(headers,amt,conn);
206
207
208
209
210
    headers[amt] = 0; /* null terminate it, */
    free(headers); /* and then throw it away */
    reading_headers = 0;
  }

211
212
213
  amt = conn->inbuf_datalen;

  if(amt + directorylen >= MAX_DIR_SIZE) {
214
    log_fn(LOG_DEBUG,"Directory too large. Failing messily.");
215
216
217
    return -1;
  }

218
  log_fn(LOG_DEBUG,"Pulling %d bytes in at offset %d.",
219
220
    amt, directorylen);

221
  connection_fetch_from_buf(the_directory+directorylen,amt,conn);
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236

  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:
237
      if (getsockopt(conn->s, SOL_SOCKET, SO_ERROR, (void*)&e, &len) < 0)  { /* not yet */
238
        if(!ERRNO_CONN_EINPROGRESS(errno)) {
239
          log_fn(LOG_DEBUG,"in-progress connect failed. Removing.");
240
          router_forget_router(conn->addr, conn->port); /* don't try him again */
241
242
243
244
245
246
247
          return -1;
        } else {
          return 0; /* no change, see if next time is better */
        }
      }
      /* the connect has finished. */

248
      log_fn(LOG_DEBUG,"Dir connection to router %s:%u established.",
249
250
251
252
          conn->address,conn->port);

      return directory_send_command(conn);
    case DIR_CONN_STATE_SENDING_COMMAND:
253
      log_fn(LOG_DEBUG,"client finished sending command.");
254
      directorylen = 0;
255
      reading_headers = 1;
256
257
258
259
      conn->state = DIR_CONN_STATE_READING;
      connection_watch_events(conn, POLLIN);
      return 0;
    case DIR_CONN_STATE_WRITING:
260
      log_fn(LOG_DEBUG,"Finished writing directory. Closing.");
261
262
      return -1; /* kill it */
    default:
263
      log_fn(LOG_DEBUG,"BUG: called in unexpected state.");
264
265
266
267
268
269
      return 0;
  }

  return 0;
}

270
271
272
273
274
275
276
/*
  Local Variables:
  mode:c
  indent-tabs-mode:nil
  c-basic-offset:2
  End:
*/