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

#include "or.h"

7
static int directory_send_command(connection_t *conn, int command);
Roger Dingledine's avatar
Roger Dingledine committed
8
9
static int directory_handle_command(connection_t *conn);

10
11
12
13
/********* START VARIABLES **********/

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

14
static char fetchstring[] = "GET / HTTP/1.0\r\n\r\n";
15
static char answerstring[] = "HTTP/1.0 200 OK\r\n\r\n";
16
17
static char the_directory[MAX_DIR_SIZE+1];
static int directorylen=0;
18
19
20

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

21
void directory_initiate_command(routerinfo_t *router, int command) {
22
23
  connection_t *conn;

24
25
26
  if(!router) { /* i guess they didn't have one in mind for me to use */
    log_fn(LOG_WARNING,"No running dirservers known. This is really bad.");
    /* XXX never again will a directory fetch work. Should we exit here, or what? */
27
    return;
28
  }
29

30
31
#if 0 /* there's no problem with parallel get/posts now. whichever 'get' ends
         last is the directory. */
32
33
  if(connection_get_by_type(CONN_TYPE_DIR)) { /* there's already a dir conn running */
    log_fn(LOG_DEBUG,"Canceling connect, dir conn already active.");
34
    return;
35
  }
36
#endif
37

38
39
  if(command == DIR_CONN_STATE_CONNECTING_FETCH)
    log_fn(LOG_DEBUG,"initiating directory fetch");
40
  else
41
    log_fn(LOG_DEBUG,"initiating directory upload");
42

43
44
45
  conn = connection_new(CONN_TYPE_DIR);

  /* set up conn so it's got all the data we need to remember */
Roger Dingledine's avatar
Roger Dingledine committed
46
47
  conn->addr = router->addr;
  conn->port = router->dir_port;
48
  conn->address = strdup(router->address);
49
  conn->nickname = strdup(router->nickname);
50
51
  if (router->identity_pkey)
    conn->identity_pkey = crypto_pk_dup_key(router->identity_pkey);
Nick Mathewson's avatar
Nick Mathewson committed
52
  else {
53
    log_fn(LOG_WARNING, "No signing key known for dirserver %s; signature won't be checked", conn->address);
54
    conn->identity_pkey = NULL;
55
    /* XXX is there really any situation where router doesn't have an identity_pkey? */
Nick Mathewson's avatar
Nick Mathewson committed
56
  }
57

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

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

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

85
static int directory_send_command(connection_t *conn, int command) {
86
87
  const char *s;
  char tmp[8192];
88
89
90

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

91
  switch(command) {
92
93
    case DIR_CONN_STATE_CONNECTING_FETCH:
      if(connection_write_to_buf(fetchstring, strlen(fetchstring), conn) < 0) {
94
        log_fn(LOG_WARNING,"Couldn't write fetch to buffer.");
95
96
        return -1;
      }
97
      conn->state = DIR_CONN_STATE_CLIENT_SENDING_FETCH;
98
      break;
99
    case DIR_CONN_STATE_CONNECTING_UPLOAD:
100
101
      s = router_get_my_descriptor();
      if(!s) {
102
        log_fn(LOG_WARNING,"Failed to get my descriptor.");
103
104
        return -1;
      }
105
106
107
      snprintf(tmp, sizeof(tmp), "POST / HTTP/1.0\r\nContent-Length: %d\r\n\r\n%s",
               strlen(s), s);
      if(connection_write_to_buf(tmp, strlen(tmp), conn) < 0) {
108
        log_fn(LOG_WARNING,"Couldn't write post/descriptor to buffer.");
109
110
        return -1;
      }
111
      conn->state = DIR_CONN_STATE_CLIENT_SENDING_UPLOAD;
112
      break;
113
114
115
116
117
118
119
120
121
  }
  return 0;
}

int connection_dir_process_inbuf(connection_t *conn) {

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

  if(conn->inbuf_reached_eof) {
122
    switch(conn->state) {
123
      case DIR_CONN_STATE_CLIENT_READING_FETCH:
124
        /* kill it, but first process the_directory and learn about new routers. */
125
        switch(fetch_from_buf_http(conn->inbuf,
126
                                   NULL, 0, the_directory, MAX_DIR_SIZE)) {
127
          case -1: /* overflow */
128
            log_fn(LOG_WARNING,"'fetch' response too large. Failing.");
129
130
            return -1;
          case 0:
131
            log_fn(LOG_INFO,"'fetch' response not all here, but we're at eof. Closing.");
132
133
134
            return -1;
          /* case 1, fall through */
        }
135
        /* XXX check headers, at least make sure returned 2xx */
136
        directorylen = strlen(the_directory);
137
        log_fn(LOG_INFO,"Received directory (size %d):\n%s", directorylen, the_directory);
138
        if(directorylen == 0) {
139
          log_fn(LOG_INFO,"Empty directory. Ignoring.");
140
141
          return -1;
        }
142
        if(router_get_dir_from_string(the_directory, conn->identity_pkey) < 0){
143
          log_fn(LOG_INFO,"...but parsing failed. Ignoring.");
144
        } else {
145
          log_fn(LOG_INFO,"and got an %s directory; updated routers.", 
146
              conn->identity_pkey ? "authenticated" : "unauthenticated");
147
148
149
150
151
        }
        if(options.OnionRouter) { /* connect to them all */
          router_retry_connections();
        }
        return -1;
152
      case DIR_CONN_STATE_CLIENT_READING_UPLOAD:
153
        /* XXX make sure there's a 200 OK on the buffer */
154
        log_fn(LOG_INFO,"eof while reading upload response. Finished.");
155
156
        return -1;
      default:
157
        log_fn(LOG_INFO,"conn reached eof, not reading. Closing.");
158
        return -1;
Roger Dingledine's avatar
cleanup    
Roger Dingledine committed
159
    }
160
161
  }

162
163
164
165
  if(conn->state == DIR_CONN_STATE_SERVER_COMMAND_WAIT)
    return directory_handle_command(conn);

  /* XXX for READ states, might want to make sure inbuf isn't too big */
166

167
  log_fn(LOG_DEBUG,"Got data, not eof. Leaving on inbuf.");
168
169
170
  return 0;
}

Roger Dingledine's avatar
Roger Dingledine committed
171
static int directory_handle_command(connection_t *conn) {
172
  char headers[1024];
173
  char body[50000]; /* XXX */
Roger Dingledine's avatar
Roger Dingledine committed
174
  size_t dlen;
175
  const char *cp;
176
177
178

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

179
  switch(fetch_from_buf_http(conn->inbuf,
180
                             headers, sizeof(headers), body, sizeof(body))) {
181
    case -1: /* overflow */
182
      log_fn(LOG_WARNING,"input too large. Failing.");
183
184
185
186
187
      return -1;
    case 0:
      log_fn(LOG_DEBUG,"command not all here yet.");
      return 0;
    /* case 1, fall through */
188
189
  }

190
191
  log_fn(LOG_DEBUG,"headers '%s', body '%s'.",headers,body);
  if(!strncasecmp(headers,"GET",3)) {
192
    /* XXX should check url and http version */
193
    log_fn(LOG_DEBUG,"Received GET command.");
194

Roger Dingledine's avatar
Roger Dingledine committed
195
    dlen = dirserv_get_directory(&cp);
196

Roger Dingledine's avatar
Roger Dingledine committed
197
    if(dlen == 0) {
198
      log_fn(LOG_WARNING,"My directory is empty. Closing.");
199
      return -1; /* XXX send some helpful http error code */
200
    }
201

202
203
    log_fn(LOG_DEBUG,"Dumping directory to client."); 
    if((connection_write_to_buf(answerstring, strlen(answerstring), conn) < 0) ||
Roger Dingledine's avatar
Roger Dingledine committed
204
       (connection_write_to_buf(cp, dlen, conn) < 0)) {
205
      log_fn(LOG_WARNING,"Failed to write answerstring+directory to outbuf.");
206
207
208
209
      return -1;
    }
    conn->state = DIR_CONN_STATE_SERVER_WRITING;
    return 0;
210
211
  }

212
  if(!strncasecmp(headers,"POST",4)) {
213
    /* XXX should check url and http version */
214
    log_fn(LOG_DEBUG,"Received POST command.");
215
216
217
218
219
    cp = body;
    if(dirserv_add_descriptor(&cp) < 0) {
      log_fn(LOG_WARNING,"dirserv_add_descriptor() failed. Dropping.");
      return -1; /* XXX should write an http failed code */
    }
220
    dirserv_get_directory(&cp); /* rebuild and write to disk */
221
    if(connection_write_to_buf(answerstring, strlen(answerstring), conn) < 0) {
222
      log_fn(LOG_WARNING,"Failed to write answerstring to outbuf.");
223
224
225
226
      return -1;
    }
    conn->state = DIR_CONN_STATE_SERVER_WRITING;
    return 0;
227
228
  }

229
  log_fn(LOG_WARNING,"Got headers with unknown command. Closing.");
230
  return -1;
231
232
233
234
235
236
237
238
}

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

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

  switch(conn->state) {
239
240
    case DIR_CONN_STATE_CONNECTING_FETCH:
    case DIR_CONN_STATE_CONNECTING_UPLOAD:
241
      if (getsockopt(conn->s, SOL_SOCKET, SO_ERROR, (void*)&e, &len) < 0)  { /* not yet */
242
        if(!ERRNO_CONN_EINPROGRESS(errno)) {
243
          log_fn(LOG_DEBUG,"in-progress connect failed. Removing.");
244
          router_mark_as_down(conn->nickname); /* don't try him again */
245
246
247
248
249
250
251
          return -1;
        } else {
          return 0; /* no change, see if next time is better */
        }
      }
      /* the connect has finished. */

252
      log_fn(LOG_INFO,"Dir connection to router %s:%u established.",
253
254
          conn->address,conn->port);

255
      return directory_send_command(conn, conn->state);
256
257
258
    case DIR_CONN_STATE_CLIENT_SENDING_FETCH:
      log_fn(LOG_DEBUG,"client finished sending fetch command.");
      conn->state = DIR_CONN_STATE_CLIENT_READING_FETCH;
259
260
      connection_watch_events(conn, POLLIN);
      return 0;
261
262
263
    case DIR_CONN_STATE_CLIENT_SENDING_UPLOAD:
      log_fn(LOG_DEBUG,"client finished sending upload command.");
      conn->state = DIR_CONN_STATE_CLIENT_READING_UPLOAD;
264
265
      connection_watch_events(conn, POLLIN);
      return 0;
266
    case DIR_CONN_STATE_SERVER_WRITING:
267
      log_fn(LOG_INFO,"Finished writing server response. Closing.");
268
269
      return -1; /* kill it */
    default:
270
271
      log_fn(LOG_WARNING,"BUG: called in unexpected state.");
      return -1;
272
273
274
275
  }
  return 0;
}

276
277
278
279
280
281
282
/*
  Local Variables:
  mode:c
  indent-tabs-mode:nil
  c-basic-offset:2
  End:
*/