chiark / gitweb /
site: Make transport_record_peers cope with multiple addresses
[secnet.git] / site.c
1 /* site.c - manage communication with a remote network site */
2
3 /* The 'site' code doesn't know anything about the structure of the
4    packets it's transmitting.  In fact, under the new netlink
5    configuration scheme it doesn't need to know anything at all about
6    IP addresses, except how to contact its peer.  This means it could
7    potentially be used to tunnel other protocols too (IPv6, IPX, plain
8    old Ethernet frames) if appropriate netlink code can be written
9    (and that ought not to be too hard, eg. using the TUN/TAP device to
10    pretend to be an Ethernet interface).  */
11
12 /* At some point in the future the netlink code will be asked for
13    configuration information to go in the PING/PONG packets at the end
14    of the key exchange. */
15
16 #include "secnet.h"
17 #include <stdio.h>
18 #include <string.h>
19 #include <limits.h>
20 #include <assert.h>
21 #include <sys/socket.h>
22
23 #include <sys/mman.h>
24 #include "util.h"
25 #include "unaligned.h"
26 #include "magic.h"
27
28 #define SETUP_BUFFER_LEN 2048
29
30 #define DEFAULT_KEY_LIFETIME                  (3600*1000) /* [ms] */
31 #define DEFAULT_KEY_RENEGOTIATE_GAP           (5*60*1000) /* [ms] */
32 #define DEFAULT_SETUP_RETRIES 5
33 #define DEFAULT_SETUP_RETRY_INTERVAL             (2*1000) /* [ms] */
34 #define DEFAULT_WAIT_TIME                       (20*1000) /* [ms] */
35
36 #define DEFAULT_MOBILE_KEY_LIFETIME      (2*24*3600*1000) /* [ms] */
37 #define DEFAULT_MOBILE_KEY_RENEGOTIATE_GAP (12*3600*1000) /* [ms] */
38 #define DEFAULT_MOBILE_SETUP_RETRIES 30
39 #define DEFAULT_MOBILE_SETUP_RETRY_INTERVAL      (1*1000) /* [ms] */
40 #define DEFAULT_MOBILE_WAIT_TIME                (10*1000) /* [ms] */
41
42 #define DEFAULT_MOBILE_PEER_EXPIRY            (2*60)      /* [s] */
43 #define DEFAULT_MOBILE_PEERS_MAX 3 /* send at most this many copies (default) */
44
45 /* Each site can be in one of several possible states. */
46
47 /* States:
48    SITE_STOP         - nothing is allowed to happen; tunnel is down;
49                        all session keys have been erased
50      -> SITE_RUN upon external instruction
51    SITE_RUN          - site up, maybe with valid key
52      -> SITE_RESOLVE upon outgoing packet and no valid key
53          we start name resolution for the other end of the tunnel
54      -> SITE_SENTMSG2 upon valid incoming message 1 and suitable time
55          we send an appropriate message 2
56    SITE_RESOLVE      - waiting for name resolution
57      -> SITE_SENTMSG1 upon successful resolution
58          we send an appropriate message 1
59      -> SITE_SENTMSG2 upon valid incoming message 1 (then abort resolution)
60          we abort resolution and 
61      -> SITE_WAIT on timeout or resolution failure
62    SITE_SENTMSG1
63      -> SITE_SENTMSG2 upon valid incoming message 1 from higher priority end
64      -> SITE_SENTMSG3 upon valid incoming message 2
65      -> SITE_WAIT on timeout
66    SITE_SENTMSG2
67      -> SITE_SENTMSG4 upon valid incoming message 3
68      -> SITE_WAIT on timeout
69    SITE_SENTMSG3
70      -> SITE_SENTMSG5 upon valid incoming message 4
71      -> SITE_WAIT on timeout
72    SITE_SENTMSG4
73      -> SITE_RUN upon valid incoming message 5
74      -> SITE_WAIT on timeout
75    SITE_SENTMSG5
76      -> SITE_RUN upon valid incoming message 6
77      -> SITE_WAIT on timeout
78    SITE_WAIT         - failed to establish key; do nothing for a while
79      -> SITE_RUN on timeout
80    */
81
82 #define SITE_STOP     0
83 #define SITE_RUN      1
84 #define SITE_RESOLVE  2
85 #define SITE_SENTMSG1 3
86 #define SITE_SENTMSG2 4
87 #define SITE_SENTMSG3 5
88 #define SITE_SENTMSG4 6
89 #define SITE_SENTMSG5 7
90 #define SITE_WAIT     8
91
92 int32_t site_max_start_pad = 4*4;
93
94 static cstring_t state_name(uint32_t state)
95 {
96     switch (state) {
97     case 0: return "STOP";
98     case 1: return "RUN";
99     case 2: return "RESOLVE";
100     case 3: return "SENTMSG1";
101     case 4: return "SENTMSG2";
102     case 5: return "SENTMSG3";
103     case 6: return "SENTMSG4";
104     case 7: return "SENTMSG5";
105     case 8: return "WAIT";
106     default: return "*bad state*";
107     }
108 }
109
110 #define NONCELEN 8
111
112 #define LOG_UNEXPECTED    0x00000001
113 #define LOG_SETUP_INIT    0x00000002
114 #define LOG_SETUP_TIMEOUT 0x00000004
115 #define LOG_ACTIVATE_KEY  0x00000008
116 #define LOG_TIMEOUT_KEY   0x00000010
117 #define LOG_SEC           0x00000020
118 #define LOG_STATE         0x00000040
119 #define LOG_DROP          0x00000080
120 #define LOG_DUMP          0x00000100
121 #define LOG_ERROR         0x00000400
122 #define LOG_PEER_ADDRS    0x00000800
123
124 static struct flagstr log_event_table[]={
125     { "unexpected", LOG_UNEXPECTED },
126     { "setup-init", LOG_SETUP_INIT },
127     { "setup-timeout", LOG_SETUP_TIMEOUT },
128     { "activate-key", LOG_ACTIVATE_KEY },
129     { "timeout-key", LOG_TIMEOUT_KEY },
130     { "security", LOG_SEC },
131     { "state-change", LOG_STATE },
132     { "packet-drop", LOG_DROP },
133     { "dump-packets", LOG_DUMP },
134     { "errors", LOG_ERROR },
135     { "peer-addrs", LOG_PEER_ADDRS },
136     { "default", LOG_SETUP_INIT|LOG_SETUP_TIMEOUT|
137       LOG_ACTIVATE_KEY|LOG_TIMEOUT_KEY|LOG_SEC|LOG_ERROR },
138     { "all", 0xffffffff },
139     { NULL, 0 }
140 };
141
142
143 /***** TRANSPORT PEERS declarations *****/
144
145 /* Details of "mobile peer" semantics:
146
147  | Note: this comment is wishful thinking right now.  It will be
148  | implemented in subsequent commits.
149
150    - We use the same data structure for the different configurations,
151      but manage it with different algorithms.
152    
153    - We record up to mobile_peers_max peer address/port numbers
154      ("peers") for key setup, and separately up to mobile_peers_max
155      for data transfer.
156
157    - In general, we make a new set of addrs (see below) when we start
158      a new key exchange; the key setup addrs become the data transport
159      addrs when key setup complets.
160
161    If our peer is mobile:
162
163    - We send to all recent addresses of incoming packets, plus
164      initially all configured addresses (which we also expire).
165
166    - So, we record addrs of good incoming packets, as follows:
167       1. expire any peers last seen >120s ("mobile-peer-expiry") ago
168       2. add the peer of the just received packet to the applicable list
169          (possibly evicting the oldest entries to make room)
170      NB that we do not expire peers until an incoming packet arrives.
171
172    - If the peer has a configured address or name, we record them the
173      same way, but only as a result of our own initiation of key
174      setup.  (We might evict some incoming packet addrs to make room.)
175
176    - The default number of addrs to keep is 3, or 4 if we have a
177      configured name or address.  That's space for two configured
178      addresses (one IPv6 and one IPv4), plus two received addresses.
179
180    - Outgoing packets are sent to every recorded address in the
181      applicable list.  Any unsupported[1] addresses are deleted from
182      the list right away.  (This should only happen to configured
183      addresses, of course, but there is no need to check that.)
184
185    - When we successfully complete a key setup, we merge the key setup
186      peers into the data transfer peers.
187
188    [1] An unsupported address is one for whose AF we don't have a
189      socket (perhaps because we got EAFNOSUPPORT or some such) or for
190      which sendto gives ENETUNREACH.
191
192    If neither end is mobile:
193
194    - When peer initiated the key exchange, we use the incoming packet
195      address.
196
197    - When we initiate the key exchange, we try configured addresses
198      until we get one which isn't unsupported then fixate on that.
199
200    - When we complete a key setup, we replace the data transport peers
201      with those from the key setup.
202
203    If we are mobile:
204
205    - We can't tell when local network setup changes so we can't cache
206      the unsupported addrs and completely remove the spurious calls to
207      sendto, but we can optimise things a bit by deprioritising addrs
208      which seem to be unsupported.
209
210    - Use only configured addresses.  (Except, that if our peer
211      initiated a key exchange we use the incoming packet address until
212      our name resolution completes.)
213
214    - When we send a packet, try each address in turn; if addr
215      supported, put that address to the end of the list for future
216      packets, and go onto the next address.
217
218    - When we complete a key setup, we replace the data transport peers
219      with those from the key setup.
220
221    */
222
223 #define MAX_MOBILE_PEERS_MAX 5 /* send at most this many copies, compiled max */
224
225 typedef struct {
226     struct timeval last;
227     struct comm_addr addr;
228 } transport_peer;
229
230 typedef struct {
231 /* configuration information */
232 /* runtime information */
233     int npeers;
234     transport_peer peers[MAX_MOBILE_PEERS_MAX];
235 } transport_peers;
236
237 /* Basic operations on transport peer address sets */
238 static void transport_peers_clear(struct site *st, transport_peers *peers);
239 static int transport_peers_valid(transport_peers *peers);
240 static void transport_peers_copy(struct site *st, transport_peers *dst,
241                                  const transport_peers *src);
242
243 /* Record address of incoming setup packet; resp. data packet. */
244 static void transport_setup_msgok(struct site *st, const struct comm_addr *a);
245 static void transport_data_msgok(struct site *st, const struct comm_addr *a);
246
247 /* Initialise the setup addresses.  Called before we send the first
248  * packet in a key exchange.  If we are the initiator, as a result of
249  * resolve completing (or being determined not to be relevant) or an
250  * incoming PROD; if we are the responder, as a result of the MSG1. */
251 static bool_t transport_compute_setupinit_peers(struct site *st,
252         const struct comm_addr *configured_addr /* 0 if none or not found */,
253         const struct comm_addr *incoming_packet_addr /* 0 if none */);
254
255 /* Called if we are the responder in a key setup, when the resolve
256  * completes.  transport_compute_setupinit_peers will hvae been called
257  * earlier.  If _complete is called, we are still doing the key setup
258  * (and we should use the new values for both the rest of the key
259  * setup and the ongoing data exchange); if _tardy is called, the key
260  * setup is done (either completed or not) and only the data peers are
261  * relevant */
262 static void transport_resolve_complete(struct site *st,
263                                       const struct comm_addr *a);
264 static void transport_resolve_complete_tardy(struct site *st,
265                                              const struct comm_addr *ca_use);
266
267 static void transport_xmit(struct site *st, transport_peers *peers,
268                            struct buffer_if *buf, bool_t candebug);
269
270  /***** END of transport peers declarations *****/
271
272
273 struct data_key {
274     struct transform_inst_if *transform;
275     uint64_t key_timeout; /* End of life of current key */
276     uint32_t remote_session_id;
277 };
278
279 struct site {
280     closure_t cl;
281     struct site_if ops;
282 /* configuration information */
283     string_t localname;
284     string_t remotename;
285     bool_t local_mobile, peer_mobile; /* Mobile client support */
286     int32_t transport_peers_max;
287     string_t tunname; /* localname<->remotename by default, used in logs */
288     string_t address; /* DNS name for bootstrapping, optional */
289     int remoteport; /* Port for bootstrapping, optional */
290     uint32_t mtu_target;
291     struct netlink_if *netlink;
292     struct comm_if **comms;
293     int ncomms;
294     struct resolver_if *resolver;
295     struct log_if *log;
296     struct random_if *random;
297     struct rsaprivkey_if *privkey;
298     struct rsapubkey_if *pubkey;
299     struct transform_if **transforms;
300     int ntransforms;
301     struct dh_if *dh;
302     struct hash_if *hash;
303
304     uint32_t index; /* Index of this site */
305     uint32_t local_capabilities;
306     int32_t setup_retries; /* How many times to send setup packets */
307     int32_t setup_retry_interval; /* Initial timeout for setup packets */
308     int32_t wait_timeout; /* How long to wait if setup unsuccessful */
309     int32_t mobile_peer_expiry; /* How long to remember 2ary addresses */
310     int32_t key_lifetime; /* How long a key lasts once set up */
311     int32_t key_renegotiate_time; /* If we see traffic (or a keepalive)
312                                       after this time, initiate a new
313                                       key exchange */
314
315     bool_t setup_priority; /* Do we have precedence if both sites emit
316                               message 1 simultaneously? */
317     uint32_t log_events;
318
319 /* runtime information */
320     uint32_t state;
321     uint64_t now; /* Most recently seen time */
322     bool_t allow_send_prod;
323     bool_t resolving;
324
325     /* The currently established session */
326     struct data_key current;
327     struct data_key auxiliary_key;
328     bool_t auxiliary_is_new;
329     uint64_t renegotiate_key_time; /* When we can negotiate a new key */
330     uint64_t auxiliary_renegotiate_key_time;
331     transport_peers peers; /* Current address(es) of peer for data traffic */
332
333     /* The current key setup protocol exchange.  We can only be
334        involved in one of these at a time.  There's a potential for
335        denial of service here (the attacker keeps sending a setup
336        packet; we keep trying to continue the exchange, and have to
337        timeout before we can listen for another setup packet); perhaps
338        we should keep a list of 'bad' sources for setup packets. */
339     uint32_t remote_capabilities;
340     uint16_t remote_adv_mtu;
341     struct transform_if *chosen_transform;
342     uint32_t setup_session_id;
343     transport_peers setup_peers;
344     uint8_t localN[NONCELEN]; /* Nonces for key exchange */
345     uint8_t remoteN[NONCELEN];
346     struct buffer_if buffer; /* Current outgoing key exchange packet */
347     struct buffer_if scratch;
348     int32_t retries; /* Number of retries remaining */
349     uint64_t timeout; /* Timeout for current state */
350     uint8_t *dhsecret;
351     uint8_t *sharedsecret;
352     uint32_t sharedsecretlen, sharedsecretallocd;
353     struct transform_inst_if *new_transform; /* For key setup/verify */
354 };
355
356 static uint32_t event_log_priority(struct site *st, uint32_t event)
357 {
358     if (!(event&st->log_events))
359         return 0;
360     switch(event) {
361     case LOG_UNEXPECTED:    return M_INFO;
362     case LOG_SETUP_INIT:    return M_INFO;
363     case LOG_SETUP_TIMEOUT: return M_NOTICE;
364     case LOG_ACTIVATE_KEY:  return M_INFO;
365     case LOG_TIMEOUT_KEY:   return M_INFO;
366     case LOG_SEC:           return M_SECURITY;
367     case LOG_STATE:         return M_DEBUG;
368     case LOG_DROP:          return M_DEBUG;
369     case LOG_DUMP:          return M_DEBUG;
370     case LOG_ERROR:         return M_ERR;
371     case LOG_PEER_ADDRS:    return M_DEBUG;
372     default:                return M_ERR;
373     }
374 }
375
376 static void vslog(struct site *st, uint32_t event, cstring_t msg, va_list ap)
377 FORMAT(printf,3,0);
378 static void vslog(struct site *st, uint32_t event, cstring_t msg, va_list ap)
379 {
380     uint32_t class;
381
382     class=event_log_priority(st, event);
383     if (class) {
384         slilog_part(st->log,class,"%s: ",st->tunname);
385         vslilog_part(st->log,class,msg,ap);
386         slilog_part(st->log,class,"\n");
387     }
388 }
389
390 static void slog(struct site *st, uint32_t event, cstring_t msg, ...)
391 FORMAT(printf,3,4);
392 static void slog(struct site *st, uint32_t event, cstring_t msg, ...)
393 {
394     va_list ap;
395     va_start(ap,msg);
396     vslog(st,event,msg,ap);
397     va_end(ap);
398 }
399
400 static void logtimeout(struct site *st, const char *fmt, ...)
401 FORMAT(printf,2,3);
402 static void logtimeout(struct site *st, const char *fmt, ...)
403 {
404     uint32_t class=event_log_priority(st,LOG_SETUP_TIMEOUT);
405     if (!class)
406         return;
407
408     va_list ap;
409     va_start(ap,fmt);
410
411     slilog_part(st->log,class,"%s: ",st->tunname);
412     vslilog_part(st->log,class,fmt,ap);
413
414     const char *delim;
415     int i;
416     for (i=0, delim=" (tried ";
417          i<st->setup_peers.npeers;
418          i++, delim=", ") {
419         transport_peer *peer=&st->setup_peers.peers[i];
420         const char *s=comm_addr_to_string(&peer->addr);
421         slilog_part(st->log,class,"%s%s",delim,s);
422     }
423
424     slilog_part(st->log,class,")\n");
425     va_end(ap);
426 }
427
428 static void set_link_quality(struct site *st);
429 static void delete_keys(struct site *st, cstring_t reason, uint32_t loglevel);
430 static void delete_one_key(struct site *st, struct data_key *key,
431                            const char *reason /* may be 0 meaning don't log*/,
432                            const char *which /* ignored if !reasonn */,
433                            uint32_t loglevel /* ignored if !reasonn */);
434 static bool_t initiate_key_setup(struct site *st, cstring_t reason,
435                                  const struct comm_addr *prod_hint);
436 static void enter_state_run(struct site *st);
437 static bool_t enter_state_resolve(struct site *st);
438 static bool_t enter_new_state(struct site *st,uint32_t next);
439 static void enter_state_wait(struct site *st);
440 static void activate_new_key(struct site *st);
441
442 static bool_t is_transform_valid(struct transform_inst_if *transform)
443 {
444     return transform && transform->valid(transform->st);
445 }
446
447 static bool_t current_valid(struct site *st)
448 {
449     return is_transform_valid(st->current.transform);
450 }
451
452 #define DEFINE_CALL_TRANSFORM(fwdrev)                                   \
453 static int call_transform_##fwdrev(struct site *st,                     \
454                                    struct transform_inst_if *transform, \
455                                    struct buffer_if *buf,               \
456                                    const char **errmsg)                 \
457 {                                                                       \
458     if (!is_transform_valid(transform)) {                               \
459         *errmsg="transform not set up";                                 \
460         return 1;                                                       \
461     }                                                                   \
462     return transform->fwdrev(transform->st,buf,errmsg);                 \
463 }
464
465 DEFINE_CALL_TRANSFORM(forwards)
466 DEFINE_CALL_TRANSFORM(reverse)
467
468 static void dispose_transform(struct transform_inst_if **transform_var)
469 {
470     struct transform_inst_if *transform=*transform_var;
471     if (transform) {
472         transform->delkey(transform->st);
473         transform->destroy(transform->st);
474     }
475     *transform_var = 0;
476 }    
477
478 #define CHECK_AVAIL(b,l) do { if ((b)->size<(l)) return False; } while(0)
479 #define CHECK_EMPTY(b) do { if ((b)->size!=0) return False; } while(0)
480 #define CHECK_TYPE(b,t) do { uint32_t type; \
481     CHECK_AVAIL((b),4); \
482     type=buf_unprepend_uint32((b)); \
483     if (type!=(t)) return False; } while(0)
484
485 static _Bool type_is_msg34(uint32_t type)
486 {
487     return
488         type == LABEL_MSG3 ||
489         type == LABEL_MSG3BIS ||
490         type == LABEL_MSG4;
491 }
492
493 struct parsedname {
494     int32_t len;
495     uint8_t *name;
496     struct buffer_if extrainfo;
497 };
498
499 struct msg {
500     uint8_t *hashstart;
501     uint32_t dest;
502     uint32_t source;
503     struct parsedname remote;
504     struct parsedname local;
505     uint32_t remote_capabilities;
506     uint16_t remote_mtu;
507     int capab_transformnum;
508     uint8_t *nR;
509     uint8_t *nL;
510     int32_t pklen;
511     char *pk;
512     int32_t hashlen;
513     int32_t siglen;
514     char *sig;
515 };
516
517 static void set_new_transform(struct site *st, char *pk)
518 {
519     /* Make room for the shared key */
520     st->sharedsecretlen=st->chosen_transform->keylen?:st->dh->ceil_len;
521     assert(st->sharedsecretlen);
522     if (st->sharedsecretlen > st->sharedsecretallocd) {
523         st->sharedsecretallocd=st->sharedsecretlen;
524         st->sharedsecret=realloc(st->sharedsecret,st->sharedsecretallocd);
525     }
526     if (!st->sharedsecret) fatal_perror("site:sharedsecret");
527
528     /* Generate the shared key */
529     st->dh->makeshared(st->dh->st,st->dhsecret,st->dh->len,pk,
530                        st->sharedsecret,st->sharedsecretlen);
531
532     /* Set up the transform */
533     struct transform_if *generator=st->chosen_transform;
534     struct transform_inst_if *generated=generator->create(generator->st);
535     generated->setkey(generated->st,st->sharedsecret,
536                       st->sharedsecretlen,st->setup_priority);
537     dispose_transform(&st->new_transform);
538     st->new_transform=generated;
539
540     slog(st,LOG_SETUP_INIT,"key exchange negotiated transform"
541          " %d (capabilities ours=%#"PRIx32" theirs=%#"PRIx32")",
542          st->chosen_transform->capab_transformnum,
543          st->local_capabilities, st->remote_capabilities);
544 }
545
546 struct xinfoadd {
547     int32_t lenpos, afternul;
548 };
549 static void append_string_xinfo_start(struct buffer_if *buf,
550                                       struct xinfoadd *xia,
551                                       const char *str)
552     /* Helps construct one of the names with additional info as found
553      * in MSG1..4.  Call this function first, then append all the
554      * desired extra info (not including the nul byte) to the buffer,
555      * then call append_string_xinfo_done. */
556 {
557     xia->lenpos = buf->size;
558     buf_append_string(buf,str);
559     buf_append_uint8(buf,0);
560     xia->afternul = buf->size;
561 }
562 static void append_string_xinfo_done(struct buffer_if *buf,
563                                      struct xinfoadd *xia)
564 {
565     /* we just need to adjust the string length */
566     if (buf->size == xia->afternul) {
567         /* no extra info, strip the nul too */
568         buf_unappend_uint8(buf);
569     } else {
570         put_uint16(buf->start+xia->lenpos, buf->size-(xia->lenpos+2));
571     }
572 }
573
574 /* Build any of msg1 to msg4. msg5 and msg6 are built from the inside
575    out using a transform of config data supplied by netlink */
576 static bool_t generate_msg(struct site *st, uint32_t type, cstring_t what)
577 {
578     void *hst;
579     uint8_t *hash;
580     string_t dhpub, sig;
581
582     st->retries=st->setup_retries;
583     BUF_ALLOC(&st->buffer,what);
584     buffer_init(&st->buffer,0);
585     buf_append_uint32(&st->buffer,
586         (type==LABEL_MSG1?0:st->setup_session_id));
587     buf_append_uint32(&st->buffer,st->index);
588     buf_append_uint32(&st->buffer,type);
589
590     struct xinfoadd xia;
591     append_string_xinfo_start(&st->buffer,&xia,st->localname);
592     if ((st->local_capabilities & CAPAB_EARLY) || (type != LABEL_MSG1)) {
593         buf_append_uint32(&st->buffer,st->local_capabilities);
594     }
595     if (type_is_msg34(type)) {
596         buf_append_uint16(&st->buffer,st->mtu_target);
597     }
598     append_string_xinfo_done(&st->buffer,&xia);
599
600     buf_append_string(&st->buffer,st->remotename);
601     memcpy(buf_append(&st->buffer,NONCELEN),st->localN,NONCELEN);
602     if (type==LABEL_MSG1) return True;
603     memcpy(buf_append(&st->buffer,NONCELEN),st->remoteN,NONCELEN);
604     if (type==LABEL_MSG2) return True;
605
606     if (hacky_par_mid_failnow()) return False;
607
608     if (type==LABEL_MSG3BIS)
609         buf_append_uint8(&st->buffer,st->chosen_transform->capab_transformnum);
610
611     dhpub=st->dh->makepublic(st->dh->st,st->dhsecret,st->dh->len);
612     buf_append_string(&st->buffer,dhpub);
613     free(dhpub);
614     hash=safe_malloc(st->hash->len, "generate_msg");
615     hst=st->hash->init();
616     st->hash->update(hst,st->buffer.start,st->buffer.size);
617     st->hash->final(hst,hash);
618     sig=st->privkey->sign(st->privkey->st,hash,st->hash->len);
619     buf_append_string(&st->buffer,sig);
620     free(sig);
621     free(hash);
622     return True;
623 }
624
625 static bool_t unpick_name(struct buffer_if *msg, struct parsedname *nm)
626 {
627     CHECK_AVAIL(msg,2);
628     nm->len=buf_unprepend_uint16(msg);
629     CHECK_AVAIL(msg,nm->len);
630     nm->name=buf_unprepend(msg,nm->len);
631     uint8_t *nul=memchr(nm->name,0,nm->len);
632     if (!nul) {
633         buffer_readonly_view(&nm->extrainfo,0,0);
634     } else {
635         buffer_readonly_view(&nm->extrainfo, nul+1, msg->start-(nul+1));
636         nm->len=nul-nm->name;
637     }
638     return True;
639 }
640
641 static bool_t unpick_msg(struct site *st, uint32_t type,
642                          struct buffer_if *msg, struct msg *m)
643 {
644     m->capab_transformnum=-1;
645     m->hashstart=msg->start;
646     CHECK_AVAIL(msg,4);
647     m->dest=buf_unprepend_uint32(msg);
648     CHECK_AVAIL(msg,4);
649     m->source=buf_unprepend_uint32(msg);
650     CHECK_TYPE(msg,type);
651     if (!unpick_name(msg,&m->remote)) return False;
652     m->remote_capabilities=0;
653     m->remote_mtu=0;
654     if (m->remote.extrainfo.size) {
655         CHECK_AVAIL(&m->remote.extrainfo,4);
656         m->remote_capabilities=buf_unprepend_uint32(&m->remote.extrainfo);
657     }
658     if (type_is_msg34(type) && m->remote.extrainfo.size) {
659         CHECK_AVAIL(&m->remote.extrainfo,2);
660         m->remote_mtu=buf_unprepend_uint16(&m->remote.extrainfo);
661     }
662     if (!unpick_name(msg,&m->local)) return False;
663     if (type==LABEL_PROD) {
664         CHECK_EMPTY(msg);
665         return True;
666     }
667     CHECK_AVAIL(msg,NONCELEN);
668     m->nR=buf_unprepend(msg,NONCELEN);
669     if (type==LABEL_MSG1) {
670         CHECK_EMPTY(msg);
671         return True;
672     }
673     CHECK_AVAIL(msg,NONCELEN);
674     m->nL=buf_unprepend(msg,NONCELEN);
675     if (type==LABEL_MSG2) {
676         CHECK_EMPTY(msg);
677         return True;
678     }
679     if (type==LABEL_MSG3BIS) {
680         CHECK_AVAIL(msg,1);
681         m->capab_transformnum = buf_unprepend_uint8(msg);
682     } else {
683         m->capab_transformnum = CAPAB_TRANSFORMNUM_ANCIENT;
684     }
685     CHECK_AVAIL(msg,2);
686     m->pklen=buf_unprepend_uint16(msg);
687     CHECK_AVAIL(msg,m->pklen);
688     m->pk=buf_unprepend(msg,m->pklen);
689     m->hashlen=msg->start-m->hashstart;
690     CHECK_AVAIL(msg,2);
691     m->siglen=buf_unprepend_uint16(msg);
692     CHECK_AVAIL(msg,m->siglen);
693     m->sig=buf_unprepend(msg,m->siglen);
694     CHECK_EMPTY(msg);
695     return True;
696 }
697
698 static bool_t name_matches(const struct parsedname *nm, const char *expected)
699 {
700     int expected_len=strlen(expected);
701     return
702         nm->len == expected_len &&
703         !memcmp(nm->name, expected, expected_len);
704 }    
705
706 static bool_t check_msg(struct site *st, uint32_t type, struct msg *m,
707                         cstring_t *error)
708 {
709     if (type==LABEL_MSG1) return True;
710
711     /* Check that the site names and our nonce have been sent
712        back correctly, and then store our peer's nonce. */ 
713     if (!name_matches(&m->remote,st->remotename)) {
714         *error="wrong remote site name";
715         return False;
716     }
717     if (!name_matches(&m->local,st->localname)) {
718         *error="wrong local site name";
719         return False;
720     }
721     if (memcmp(m->nL,st->localN,NONCELEN)!=0) {
722         *error="wrong locally-generated nonce";
723         return False;
724     }
725     if (type==LABEL_MSG2) return True;
726     if (!consttime_memeq(m->nR,st->remoteN,NONCELEN)!=0) {
727         *error="wrong remotely-generated nonce";
728         return False;
729     }
730     /* MSG3 has complicated rules about capabilities, which are
731      * handled in process_msg3. */
732     if (type==LABEL_MSG3 || type==LABEL_MSG3BIS) return True;
733     if (m->remote_capabilities!=st->remote_capabilities) {
734         *error="remote capabilities changed";
735         return False;
736     }
737     if (type==LABEL_MSG4) return True;
738     *error="unknown message type";
739     return False;
740 }
741
742 static bool_t generate_msg1(struct site *st)
743 {
744     st->random->generate(st->random->st,NONCELEN,st->localN);
745     return generate_msg(st,LABEL_MSG1,"site:MSG1");
746 }
747
748 static bool_t process_msg1(struct site *st, struct buffer_if *msg1,
749                            const struct comm_addr *src, struct msg *m)
750 {
751     /* We've already determined we're in an appropriate state to
752        process an incoming MSG1, and that the MSG1 has correct values
753        of A and B. */
754
755     st->setup_session_id=m->source;
756     st->remote_capabilities=m->remote_capabilities;
757     memcpy(st->remoteN,m->nR,NONCELEN);
758     return True;
759 }
760
761 static bool_t generate_msg2(struct site *st)
762 {
763     st->random->generate(st->random->st,NONCELEN,st->localN);
764     return generate_msg(st,LABEL_MSG2,"site:MSG2");
765 }
766
767 static bool_t process_msg2(struct site *st, struct buffer_if *msg2,
768                            const struct comm_addr *src)
769 {
770     struct msg m;
771     cstring_t err;
772
773     if (!unpick_msg(st,LABEL_MSG2,msg2,&m)) return False;
774     if (!check_msg(st,LABEL_MSG2,&m,&err)) {
775         slog(st,LOG_SEC,"msg2: %s",err);
776         return False;
777     }
778     st->setup_session_id=m.source;
779     st->remote_capabilities=m.remote_capabilities;
780
781     /* Select the transform to use */
782
783     uint32_t remote_transforms = st->remote_capabilities & CAPAB_TRANSFORM_MASK;
784     if (!remote_transforms)
785         /* old secnets only had this one transform */
786         remote_transforms = 1UL << CAPAB_TRANSFORMNUM_ANCIENT;
787
788     struct transform_if *ti;
789     int i;
790     for (i=0; i<st->ntransforms; i++) {
791         ti=st->transforms[i];
792         if ((1UL << ti->capab_transformnum) & remote_transforms)
793             goto transform_found;
794     }
795     slog(st,LOG_ERROR,"no transforms in common"
796          " (us %#"PRIx32"; them: %#"PRIx32")",
797          st->local_capabilities & CAPAB_TRANSFORM_MASK,
798          remote_transforms);
799     return False;
800  transform_found:
801     st->chosen_transform=ti;
802
803     memcpy(st->remoteN,m.nR,NONCELEN);
804     return True;
805 }
806
807 static bool_t generate_msg3(struct site *st)
808 {
809     /* Now we have our nonce and their nonce. Think of a secret key,
810        and create message number 3. */
811     st->random->generate(st->random->st,st->dh->len,st->dhsecret);
812     return generate_msg(st,
813                         (st->remote_capabilities & CAPAB_TRANSFORM_MASK
814                          ? LABEL_MSG3BIS : LABEL_MSG3),
815                         "site:MSG3");
816 }
817
818 static bool_t process_msg3_msg4(struct site *st, struct msg *m)
819 {
820     uint8_t *hash;
821     void *hst;
822
823     /* Check signature and store g^x mod m */
824     hash=safe_malloc(st->hash->len, "process_msg3_msg4");
825     hst=st->hash->init();
826     st->hash->update(hst,m->hashstart,m->hashlen);
827     st->hash->final(hst,hash);
828     /* Terminate signature with a '0' - cheating, but should be ok */
829     m->sig[m->siglen]=0;
830     if (!st->pubkey->check(st->pubkey->st,hash,st->hash->len,m->sig)) {
831         slog(st,LOG_SEC,"msg3/msg4 signature failed check!");
832         free(hash);
833         return False;
834     }
835     free(hash);
836
837     st->remote_adv_mtu=m->remote_mtu;
838
839     return True;
840 }
841
842 static bool_t process_msg3(struct site *st, struct buffer_if *msg3,
843                            const struct comm_addr *src, uint32_t msgtype)
844 {
845     struct msg m;
846     cstring_t err;
847
848     assert(msgtype==LABEL_MSG3 || msgtype==LABEL_MSG3BIS);
849
850     if (!unpick_msg(st,msgtype,msg3,&m)) return False;
851     if (!check_msg(st,msgtype,&m,&err)) {
852         slog(st,LOG_SEC,"msg3: %s",err);
853         return False;
854     }
855     uint32_t capab_adv_late = m.remote_capabilities
856         & ~st->remote_capabilities & CAPAB_EARLY;
857     if (capab_adv_late) {
858         slog(st,LOG_SEC,"msg3 impermissibly adds early capability flag(s)"
859              " %#"PRIx32" (was %#"PRIx32", now %#"PRIx32")",
860              capab_adv_late, st->remote_capabilities, m.remote_capabilities);
861         return False;
862     }
863     st->remote_capabilities|=m.remote_capabilities;
864
865     struct transform_if *ti;
866     int i;
867     for (i=0; i<st->ntransforms; i++) {
868         ti=st->transforms[i];
869         if (ti->capab_transformnum == m.capab_transformnum)
870             goto transform_found;
871     }
872     slog(st,LOG_SEC,"peer chose unknown-to-us transform %d!",
873          m.capab_transformnum);
874     return False;
875  transform_found:
876     st->chosen_transform=ti;
877
878     if (!process_msg3_msg4(st,&m))
879         return False;
880
881     /* Terminate their DH public key with a '0' */
882     m.pk[m.pklen]=0;
883     /* Invent our DH secret key */
884     st->random->generate(st->random->st,st->dh->len,st->dhsecret);
885
886     /* Generate the shared key and set up the transform */
887     set_new_transform(st,m.pk);
888
889     return True;
890 }
891
892 static bool_t generate_msg4(struct site *st)
893 {
894     /* We have both nonces, their public key and our private key. Generate
895        our public key, sign it and send it to them. */
896     return generate_msg(st,LABEL_MSG4,"site:MSG4");
897 }
898
899 static bool_t process_msg4(struct site *st, struct buffer_if *msg4,
900                            const struct comm_addr *src)
901 {
902     struct msg m;
903     cstring_t err;
904
905     if (!unpick_msg(st,LABEL_MSG4,msg4,&m)) return False;
906     if (!check_msg(st,LABEL_MSG4,&m,&err)) {
907         slog(st,LOG_SEC,"msg4: %s",err);
908         return False;
909     }
910     
911     if (!process_msg3_msg4(st,&m))
912         return False;
913
914     /* Terminate their DH public key with a '0' */
915     m.pk[m.pklen]=0;
916
917     /* Generate the shared key and set up the transform */
918     set_new_transform(st,m.pk);
919
920     return True;
921 }
922
923 struct msg0 {
924     uint32_t dest;
925     uint32_t source;
926     uint32_t type;
927 };
928
929 static bool_t unpick_msg0(struct site *st, struct buffer_if *msg0,
930                           struct msg0 *m)
931 {
932     CHECK_AVAIL(msg0,4);
933     m->dest=buf_unprepend_uint32(msg0);
934     CHECK_AVAIL(msg0,4);
935     m->source=buf_unprepend_uint32(msg0);
936     CHECK_AVAIL(msg0,4);
937     m->type=buf_unprepend_uint32(msg0);
938     return True;
939     /* Leaves transformed part of buffer untouched */
940 }
941
942 static bool_t generate_msg5(struct site *st)
943 {
944     cstring_t transform_err;
945
946     BUF_ALLOC(&st->buffer,"site:MSG5");
947     /* We are going to add four words to the message */
948     buffer_init(&st->buffer,calculate_max_start_pad());
949     /* Give the netlink code an opportunity to put its own stuff in the
950        message (configuration information, etc.) */
951     buf_prepend_uint32(&st->buffer,LABEL_MSG5);
952     if (call_transform_forwards(st,st->new_transform,
953                                 &st->buffer,&transform_err))
954         return False;
955     buf_prepend_uint32(&st->buffer,LABEL_MSG5);
956     buf_prepend_uint32(&st->buffer,st->index);
957     buf_prepend_uint32(&st->buffer,st->setup_session_id);
958
959     st->retries=st->setup_retries;
960     return True;
961 }
962
963 static bool_t process_msg5(struct site *st, struct buffer_if *msg5,
964                            const struct comm_addr *src,
965                            struct transform_inst_if *transform)
966 {
967     struct msg0 m;
968     cstring_t transform_err;
969
970     if (!unpick_msg0(st,msg5,&m)) return False;
971
972     if (call_transform_reverse(st,transform,msg5,&transform_err)) {
973         /* There's a problem */
974         slog(st,LOG_SEC,"process_msg5: transform: %s",transform_err);
975         return False;
976     }
977     /* Buffer should now contain untransformed PING packet data */
978     CHECK_AVAIL(msg5,4);
979     if (buf_unprepend_uint32(msg5)!=LABEL_MSG5) {
980         slog(st,LOG_SEC,"MSG5/PING packet contained wrong label");
981         return False;
982     }
983     /* Older versions of secnet used to write some config data here
984      * which we ignore.  So we don't CHECK_EMPTY */
985     return True;
986 }
987
988 static void create_msg6(struct site *st, struct transform_inst_if *transform,
989                         uint32_t session_id)
990 {
991     cstring_t transform_err;
992
993     BUF_ALLOC(&st->buffer,"site:MSG6");
994     /* We are going to add four words to the message */
995     buffer_init(&st->buffer,calculate_max_start_pad());
996     /* Give the netlink code an opportunity to put its own stuff in the
997        message (configuration information, etc.) */
998     buf_prepend_uint32(&st->buffer,LABEL_MSG6);
999     int problem = call_transform_forwards(st,transform,
1000                                           &st->buffer,&transform_err);
1001     assert(!problem);
1002     buf_prepend_uint32(&st->buffer,LABEL_MSG6);
1003     buf_prepend_uint32(&st->buffer,st->index);
1004     buf_prepend_uint32(&st->buffer,session_id);
1005 }
1006
1007 static bool_t generate_msg6(struct site *st)
1008 {
1009     if (!is_transform_valid(st->new_transform))
1010         return False;
1011     create_msg6(st,st->new_transform,st->setup_session_id);
1012     st->retries=1; /* Peer will retransmit MSG5 if this packet gets lost */
1013     return True;
1014 }
1015
1016 static bool_t process_msg6(struct site *st, struct buffer_if *msg6,
1017                            const struct comm_addr *src)
1018 {
1019     struct msg0 m;
1020     cstring_t transform_err;
1021
1022     if (!unpick_msg0(st,msg6,&m)) return False;
1023
1024     if (call_transform_reverse(st,st->new_transform,msg6,&transform_err)) {
1025         /* There's a problem */
1026         slog(st,LOG_SEC,"process_msg6: transform: %s",transform_err);
1027         return False;
1028     }
1029     /* Buffer should now contain untransformed PING packet data */
1030     CHECK_AVAIL(msg6,4);
1031     if (buf_unprepend_uint32(msg6)!=LABEL_MSG6) {
1032         slog(st,LOG_SEC,"MSG6/PONG packet contained invalid data");
1033         return False;
1034     }
1035     /* Older versions of secnet used to write some config data here
1036      * which we ignore.  So we don't CHECK_EMPTY */
1037     return True;
1038 }
1039
1040 static bool_t decrypt_msg0(struct site *st, struct buffer_if *msg0,
1041                            const struct comm_addr *src)
1042 {
1043     cstring_t transform_err, auxkey_err, newkey_err="n/a";
1044     struct msg0 m;
1045     uint32_t problem;
1046
1047     if (!unpick_msg0(st,msg0,&m)) return False;
1048
1049     /* Keep a copy so we can try decrypting it with multiple keys */
1050     buffer_copy(&st->scratch, msg0);
1051
1052     problem = call_transform_reverse(st,st->current.transform,
1053                                      msg0,&transform_err);
1054     if (!problem) {
1055         if (!st->auxiliary_is_new)
1056             delete_one_key(st,&st->auxiliary_key,
1057                            "peer has used new key","auxiliary key",LOG_SEC);
1058         return True;
1059     }
1060     if (problem==2)
1061         goto skew;
1062
1063     buffer_copy(msg0, &st->scratch);
1064     problem = call_transform_reverse(st,st->auxiliary_key.transform,
1065                                      msg0,&auxkey_err);
1066     if (problem==0) {
1067         slog(st,LOG_DROP,"processing packet which uses auxiliary key");
1068         if (st->auxiliary_is_new) {
1069             /* We previously timed out in state SENTMSG5 but it turns
1070              * out that our peer did in fact get our MSG5 and is
1071              * using the new key.  So we should switch to it too. */
1072             /* This is a bit like activate_new_key. */
1073             struct data_key t;
1074             t=st->current;
1075             st->current=st->auxiliary_key;
1076             st->auxiliary_key=t;
1077
1078             delete_one_key(st,&st->auxiliary_key,"peer has used new key",
1079                            "previous key",LOG_SEC);
1080             st->auxiliary_is_new=0;
1081             st->renegotiate_key_time=st->auxiliary_renegotiate_key_time;
1082         }
1083         return True;
1084     }
1085     if (problem==2)
1086         goto skew;
1087
1088     if (st->state==SITE_SENTMSG5) {
1089         buffer_copy(msg0, &st->scratch);
1090         problem = call_transform_reverse(st,st->new_transform,
1091                                          msg0,&newkey_err);
1092         if (!problem) {
1093             /* It looks like we didn't get the peer's MSG6 */
1094             /* This is like a cut-down enter_new_state(SITE_RUN) */
1095             slog(st,LOG_STATE,"will enter state RUN (MSG0 with new key)");
1096             BUF_FREE(&st->buffer);
1097             st->timeout=0;
1098             activate_new_key(st);
1099             return True; /* do process the data in this packet */
1100         }
1101         if (problem==2)
1102             goto skew;
1103     }
1104
1105     slog(st,LOG_SEC,"transform: %s (aux: %s, new: %s)",
1106          transform_err,auxkey_err,newkey_err);
1107     initiate_key_setup(st,"incoming message would not decrypt",0);
1108     send_nak(src,m.dest,m.source,m.type,msg0,"message would not decrypt");
1109     return False;
1110
1111  skew:
1112     slog(st,LOG_DROP,"transform: %s (merely skew)",transform_err);
1113     return False;
1114 }
1115
1116 static bool_t process_msg0(struct site *st, struct buffer_if *msg0,
1117                            const struct comm_addr *src)
1118 {
1119     uint32_t type;
1120
1121     if (!decrypt_msg0(st,msg0,src))
1122         return False;
1123
1124     CHECK_AVAIL(msg0,4);
1125     type=buf_unprepend_uint32(msg0);
1126     switch(type) {
1127     case LABEL_MSG7:
1128         /* We must forget about the current session. */
1129         delete_keys(st,"request from peer",LOG_SEC);
1130         return True;
1131     case LABEL_MSG9:
1132         /* Deliver to netlink layer */
1133         st->netlink->deliver(st->netlink->st,msg0);
1134         transport_data_msgok(st,src);
1135         /* See whether we should start negotiating a new key */
1136         if (st->now > st->renegotiate_key_time)
1137             initiate_key_setup(st,"incoming packet in renegotiation window",0);
1138         return True;
1139     default:
1140         slog(st,LOG_SEC,"incoming encrypted message of type %08x "
1141              "(unknown)",type);
1142         break;
1143     }
1144     return False;
1145 }
1146
1147 static void dump_packet(struct site *st, struct buffer_if *buf,
1148                         const struct comm_addr *addr, bool_t incoming)
1149 {
1150     uint32_t dest=get_uint32(buf->start);
1151     uint32_t source=get_uint32(buf->start+4);
1152     uint32_t msgtype=get_uint32(buf->start+8);
1153
1154     if (st->log_events & LOG_DUMP)
1155         slilog(st->log,M_DEBUG,"%s: %s: %08x<-%08x: %08x:",
1156                st->tunname,incoming?"incoming":"outgoing",
1157                dest,source,msgtype);
1158 }
1159
1160 static uint32_t site_status(void *st)
1161 {
1162     return 0;
1163 }
1164
1165 static bool_t send_msg(struct site *st)
1166 {
1167     if (st->retries>0) {
1168         transport_xmit(st, &st->setup_peers, &st->buffer, True);
1169         st->timeout=st->now+st->setup_retry_interval;
1170         st->retries--;
1171         return True;
1172     } else if (st->state==SITE_SENTMSG5) {
1173         logtimeout(st,"timed out sending MSG5, stashing new key");
1174         /* We stash the key we have produced, in case it turns out that
1175          * our peer did see our MSG5 after all and starts using it. */
1176         /* This is a bit like some of activate_new_key */
1177         struct transform_inst_if *t;
1178         t=st->auxiliary_key.transform;
1179         st->auxiliary_key.transform=st->new_transform;
1180         st->new_transform=t;
1181         dispose_transform(&st->new_transform);
1182
1183         st->auxiliary_is_new=1;
1184         st->auxiliary_key.key_timeout=st->now+st->key_lifetime;
1185         st->auxiliary_renegotiate_key_time=st->now+st->key_renegotiate_time;
1186         st->auxiliary_key.remote_session_id=st->setup_session_id;
1187
1188         enter_state_wait(st);
1189         return False;
1190     } else {
1191         logtimeout(st,"timed out sending key setup packet "
1192             "(in state %s)",state_name(st->state));
1193         enter_state_wait(st);
1194         return False;
1195     }
1196 }
1197
1198 static void site_resolve_callback(void *sst, struct in_addr *address)
1199 {
1200     struct site *st=sst;
1201     struct comm_addr ca_buf, *ca_use;
1202
1203     st->resolving=False;
1204
1205     if (address) {
1206         FILLZERO(ca_buf);
1207         ca_buf.comm=st->comms[0];
1208         ca_buf.sin.sin_family=AF_INET;
1209         ca_buf.sin.sin_port=htons(st->remoteport);
1210         ca_buf.sin.sin_addr=*address;
1211         ca_use=&ca_buf;
1212         slog(st,LOG_STATE,"resolution of %s completed: %s",
1213              st->address, comm_addr_to_string(ca_use));;
1214     } else {
1215         slog(st,LOG_ERROR,"resolution of %s failed",st->address);
1216         ca_use=0;
1217     }
1218
1219     switch (st->state) {
1220     case SITE_RESOLVE:
1221         if (transport_compute_setupinit_peers(st,ca_use,0)) {
1222             enter_new_state(st,SITE_SENTMSG1);
1223         } else {
1224             /* Can't figure out who to try to to talk to */
1225             slog(st,LOG_SETUP_INIT,
1226                  "key exchange failed: cannot find peer address");
1227             enter_state_run(st);
1228         }
1229         break;
1230     case SITE_SENTMSG1: case SITE_SENTMSG2:
1231     case SITE_SENTMSG3: case SITE_SENTMSG4:
1232     case SITE_SENTMSG5:
1233         if (ca_use) {
1234             /* We start using the address immediately for data too.
1235              * It's best to store it in st->peers now because we might
1236              * go via SENTMSG5, WAIT, and a MSG0, straight into using
1237              * the new key (without updating the data peer addrs). */
1238             transport_resolve_complete(st,ca_use);
1239         } else if (st->local_mobile) {
1240             /* We can't let this rest because we may have a peer
1241              * address which will break in the future. */
1242             slog(st,LOG_SETUP_INIT,"resolution of %s failed: "
1243                  "abandoning key exchange",st->address);
1244             enter_state_wait(st);
1245         } else {
1246             slog(st,LOG_SETUP_INIT,"resolution of %s failed: "
1247                  " continuing to use source address of peer's packets"
1248                  " for key exchange and ultimately data",
1249                  st->address);
1250         }
1251         break;
1252     case SITE_RUN:
1253         if (ca_use) {
1254             slog(st,LOG_SETUP_INIT,"resolution of %s completed tardily,"
1255                  " updating peer address(es)",st->address);
1256             transport_resolve_complete_tardy(st,ca_use);
1257         } else if (st->local_mobile) {
1258             /* Not very good.  We should queue (another) renegotiation
1259              * so that we can update the peer address. */
1260             st->key_renegotiate_time=st->now+st->wait_timeout;
1261         } else {
1262             slog(st,LOG_SETUP_INIT,"resolution of %s failed: "
1263                  " continuing to use source address of peer's packets",
1264                  st->address);
1265         }
1266         break;
1267     case SITE_WAIT:
1268     case SITE_STOP:
1269         /* oh well */
1270         break;
1271     }
1272 }
1273
1274 static bool_t initiate_key_setup(struct site *st, cstring_t reason,
1275                                  const struct comm_addr *prod_hint)
1276 {
1277     /* Reentrancy hazard: can call enter_new_state/enter_state_* */
1278     if (st->state!=SITE_RUN) return False;
1279     slog(st,LOG_SETUP_INIT,"initiating key exchange (%s)",reason);
1280     if (st->address) {
1281         slog(st,LOG_SETUP_INIT,"resolving peer address");
1282         return enter_state_resolve(st);
1283     } else if (transport_compute_setupinit_peers(st,0,prod_hint)) {
1284         return enter_new_state(st,SITE_SENTMSG1);
1285     }
1286     slog(st,LOG_SETUP_INIT,"key exchange failed: no address for peer");
1287     return False;
1288 }
1289
1290 static void activate_new_key(struct site *st)
1291 {
1292     struct transform_inst_if *t;
1293
1294     /* We have three transform instances, which we swap between old,
1295        active and setup */
1296     t=st->auxiliary_key.transform;
1297     st->auxiliary_key.transform=st->current.transform;
1298     st->current.transform=st->new_transform;
1299     st->new_transform=t;
1300     dispose_transform(&st->new_transform);
1301
1302     st->timeout=0;
1303     st->auxiliary_is_new=0;
1304     st->auxiliary_key.key_timeout=st->current.key_timeout;
1305     st->current.key_timeout=st->now+st->key_lifetime;
1306     st->renegotiate_key_time=st->now+st->key_renegotiate_time;
1307     transport_peers_copy(st,&st->peers,&st->setup_peers);
1308     st->current.remote_session_id=st->setup_session_id;
1309
1310     /* Compute the inter-site MTU.  This is min( our_mtu, their_mtu ).
1311      * But their mtu be unspecified, in which case we just use ours. */
1312     uint32_t intersite_mtu=
1313         MIN(st->mtu_target, st->remote_adv_mtu ?: ~(uint32_t)0);
1314     st->netlink->set_mtu(st->netlink->st,intersite_mtu);
1315
1316     slog(st,LOG_ACTIVATE_KEY,"new key activated"
1317          " (mtu ours=%"PRId32" theirs=%"PRId32" intersite=%"PRId32")",
1318          st->mtu_target, st->remote_adv_mtu, intersite_mtu);
1319     enter_state_run(st);
1320 }
1321
1322 static void delete_one_key(struct site *st, struct data_key *key,
1323                            cstring_t reason, cstring_t which, uint32_t loglevel)
1324 {
1325     if (!is_transform_valid(key->transform)) return;
1326     if (reason) slog(st,loglevel,"%s deleted (%s)",which,reason);
1327     dispose_transform(&key->transform);
1328     key->key_timeout=0;
1329 }
1330
1331 static void delete_keys(struct site *st, cstring_t reason, uint32_t loglevel)
1332 {
1333     if (current_valid(st)) {
1334         slog(st,loglevel,"session closed (%s)",reason);
1335
1336         delete_one_key(st,&st->current,0,0,0);
1337         set_link_quality(st);
1338     }
1339     delete_one_key(st,&st->auxiliary_key,0,0,0);
1340 }
1341
1342 static void state_assert(struct site *st, bool_t ok)
1343 {
1344     if (!ok) fatal("site:state_assert");
1345 }
1346
1347 static void enter_state_stop(struct site *st)
1348 {
1349     st->state=SITE_STOP;
1350     st->timeout=0;
1351     delete_keys(st,"entering state STOP",LOG_TIMEOUT_KEY);
1352     dispose_transform(&st->new_transform);
1353 }
1354
1355 static void set_link_quality(struct site *st)
1356 {
1357     uint32_t quality;
1358     if (current_valid(st))
1359         quality=LINK_QUALITY_UP;
1360     else if (st->state==SITE_WAIT || st->state==SITE_STOP)
1361         quality=LINK_QUALITY_DOWN;
1362     else if (st->address)
1363         quality=LINK_QUALITY_DOWN_CURRENT_ADDRESS;
1364     else if (transport_peers_valid(&st->peers))
1365         quality=LINK_QUALITY_DOWN_STALE_ADDRESS;
1366     else
1367         quality=LINK_QUALITY_DOWN;
1368
1369     st->netlink->set_quality(st->netlink->st,quality);
1370 }
1371
1372 static void enter_state_run(struct site *st)
1373 {
1374     slog(st,LOG_STATE,"entering state RUN");
1375     st->state=SITE_RUN;
1376     st->timeout=0;
1377
1378     st->setup_session_id=0;
1379     transport_peers_clear(st,&st->setup_peers);
1380     memset(st->localN,0,NONCELEN);
1381     memset(st->remoteN,0,NONCELEN);
1382     dispose_transform(&st->new_transform);
1383     memset(st->dhsecret,0,st->dh->len);
1384     memset(st->sharedsecret,0,st->sharedsecretlen);
1385     set_link_quality(st);
1386 }
1387
1388 static bool_t ensure_resolving(struct site *st)
1389 {
1390     /* Reentrancy hazard: may call site_resolve_callback and hence
1391      * enter_new_state, enter_state_* and generate_msg*. */
1392     if (st->resolving)
1393         return True;
1394
1395     assert(st->address);
1396
1397     /* resolver->request might reentrantly call site_resolve_callback
1398      * which will clear st->resolving, so we need to set it beforehand
1399      * rather than afterwards; also, it might return False, in which
1400      * case we have to clear ->resolving again. */
1401     st->resolving=True;
1402     bool_t ok = st->resolver->request(st->resolver->st,st->address,
1403                                       site_resolve_callback,st);
1404     if (!ok)
1405         st->resolving=False;
1406
1407     return ok;
1408 }
1409
1410 static bool_t enter_state_resolve(struct site *st)
1411 {
1412     /* Reentrancy hazard!  See ensure_resolving. */
1413     state_assert(st,st->state==SITE_RUN);
1414     slog(st,LOG_STATE,"entering state RESOLVE");
1415     st->state=SITE_RESOLVE;
1416     return ensure_resolving(st);
1417 }
1418
1419 static bool_t enter_new_state(struct site *st, uint32_t next)
1420 {
1421     bool_t (*gen)(struct site *st);
1422     int r;
1423
1424     slog(st,LOG_STATE,"entering state %s",state_name(next));
1425     switch(next) {
1426     case SITE_SENTMSG1:
1427         state_assert(st,st->state==SITE_RUN || st->state==SITE_RESOLVE);
1428         gen=generate_msg1;
1429         break;
1430     case SITE_SENTMSG2:
1431         state_assert(st,st->state==SITE_RUN || st->state==SITE_RESOLVE ||
1432                      st->state==SITE_SENTMSG1 || st->state==SITE_WAIT);
1433         gen=generate_msg2;
1434         break;
1435     case SITE_SENTMSG3:
1436         state_assert(st,st->state==SITE_SENTMSG1);
1437         BUF_FREE(&st->buffer);
1438         gen=generate_msg3;
1439         break;
1440     case SITE_SENTMSG4:
1441         state_assert(st,st->state==SITE_SENTMSG2);
1442         BUF_FREE(&st->buffer);
1443         gen=generate_msg4;
1444         break;
1445     case SITE_SENTMSG5:
1446         state_assert(st,st->state==SITE_SENTMSG3);
1447         BUF_FREE(&st->buffer);
1448         gen=generate_msg5;
1449         break;
1450     case SITE_RUN:
1451         state_assert(st,st->state==SITE_SENTMSG4);
1452         BUF_FREE(&st->buffer);
1453         gen=generate_msg6;
1454         break;
1455     default:
1456         gen=NULL;
1457         fatal("enter_new_state(%s): invalid new state",state_name(next));
1458         break;
1459     }
1460
1461     if (hacky_par_start_failnow()) return False;
1462
1463     r= gen(st) && send_msg(st);
1464
1465     hacky_par_end(&r,
1466                   st->setup_retries, st->setup_retry_interval,
1467                   send_msg, st);
1468     
1469     if (r) {
1470         st->state=next;
1471         if (next==SITE_RUN) {
1472             BUF_FREE(&st->buffer); /* Never reused */
1473             st->timeout=0; /* Never retransmit */
1474             activate_new_key(st);
1475         }
1476         return True;
1477     }
1478     slog(st,LOG_ERROR,"error entering state %s",state_name(next));
1479     st->buffer.free=False; /* Unconditionally use the buffer; it may be
1480                               in either state, and enter_state_wait() will
1481                               do a BUF_FREE() */
1482     enter_state_wait(st);
1483     return False;
1484 }
1485
1486 /* msg7 tells our peer that we're about to forget our key */
1487 static bool_t send_msg7(struct site *st, cstring_t reason)
1488 {
1489     cstring_t transform_err;
1490
1491     if (current_valid(st) && st->buffer.free
1492         && transport_peers_valid(&st->peers)) {
1493         BUF_ALLOC(&st->buffer,"site:MSG7");
1494         buffer_init(&st->buffer,calculate_max_start_pad());
1495         buf_append_uint32(&st->buffer,LABEL_MSG7);
1496         buf_append_string(&st->buffer,reason);
1497         if (call_transform_forwards(st, st->current.transform,
1498                                     &st->buffer, &transform_err))
1499             goto free_out;
1500         buf_prepend_uint32(&st->buffer,LABEL_MSG0);
1501         buf_prepend_uint32(&st->buffer,st->index);
1502         buf_prepend_uint32(&st->buffer,st->current.remote_session_id);
1503         transport_xmit(st,&st->peers,&st->buffer,True);
1504         BUF_FREE(&st->buffer);
1505     free_out:
1506         return True;
1507     }
1508     return False;
1509 }
1510
1511 /* We go into this state if our peer becomes uncommunicative. Similar to
1512    the "stop" state, we forget all session keys for a while, before
1513    re-entering the "run" state. */
1514 static void enter_state_wait(struct site *st)
1515 {
1516     slog(st,LOG_STATE,"entering state WAIT");
1517     st->timeout=st->now+st->wait_timeout;
1518     st->state=SITE_WAIT;
1519     set_link_quality(st);
1520     BUF_FREE(&st->buffer); /* will have had an outgoing packet in it */
1521     /* XXX Erase keys etc. */
1522 }
1523
1524 static void generate_prod(struct site *st, struct buffer_if *buf)
1525 {
1526     buffer_init(buf,0);
1527     buf_append_uint32(buf,0);
1528     buf_append_uint32(buf,0);
1529     buf_append_uint32(buf,LABEL_PROD);
1530     buf_append_string(buf,st->localname);
1531     buf_append_string(buf,st->remotename);
1532 }
1533
1534 static void generate_send_prod(struct site *st,
1535                                const struct comm_addr *source)
1536 {
1537     if (!st->allow_send_prod) return; /* too soon */
1538     if (!(st->state==SITE_RUN || st->state==SITE_RESOLVE ||
1539           st->state==SITE_WAIT)) return; /* we'd ignore peer's MSG1 */
1540
1541     slog(st,LOG_SETUP_INIT,"prodding peer for key exchange");
1542     st->allow_send_prod=0;
1543     generate_prod(st,&st->scratch);
1544     dump_packet(st,&st->scratch,source,False);
1545     source->comm->sendmsg(source->comm->st, &st->scratch, source);
1546 }
1547
1548 static inline void site_settimeout(uint64_t timeout, int *timeout_io)
1549 {
1550     if (timeout) {
1551         int64_t offset=timeout-*now;
1552         if (offset<0) offset=0;
1553         if (offset>INT_MAX) offset=INT_MAX;
1554         if (*timeout_io<0 || offset<*timeout_io)
1555             *timeout_io=offset;
1556     }
1557 }
1558
1559 static int site_beforepoll(void *sst, struct pollfd *fds, int *nfds_io,
1560                            int *timeout_io)
1561 {
1562     struct site *st=sst;
1563
1564     *nfds_io=0; /* We don't use any file descriptors */
1565     st->now=*now;
1566
1567     /* Work out when our next timeout is. The earlier of 'timeout' or
1568        'current.key_timeout'. A stored value of '0' indicates no timeout
1569        active. */
1570     site_settimeout(st->timeout, timeout_io);
1571     site_settimeout(st->current.key_timeout, timeout_io);
1572     site_settimeout(st->auxiliary_key.key_timeout, timeout_io);
1573
1574     return 0; /* success */
1575 }
1576
1577 static void check_expiry(struct site *st, struct data_key *key,
1578                          const char *which)
1579 {
1580     if (key->key_timeout && *now>key->key_timeout) {
1581         delete_one_key(st,key,"maximum life exceeded",which,LOG_TIMEOUT_KEY);
1582     }
1583 }
1584
1585 /* NB site_afterpoll will be called before site_beforepoll is ever called */
1586 static void site_afterpoll(void *sst, struct pollfd *fds, int nfds)
1587 {
1588     struct site *st=sst;
1589
1590     st->now=*now;
1591     if (st->timeout && *now>st->timeout) {
1592         st->timeout=0;
1593         if (st->state>=SITE_SENTMSG1 && st->state<=SITE_SENTMSG5) {
1594             if (!hacky_par_start_failnow())
1595                 send_msg(st);
1596         } else if (st->state==SITE_WAIT) {
1597             enter_state_run(st);
1598         } else {
1599             slog(st,LOG_ERROR,"site_afterpoll: unexpected timeout, state=%d",
1600                  st->state);
1601         }
1602     }
1603     check_expiry(st,&st->current,"current key");
1604     check_expiry(st,&st->auxiliary_key,"auxiliary key");
1605 }
1606
1607 /* This function is called by the netlink device to deliver packets
1608    intended for the remote network. The packet is in "raw" wire
1609    format, but is guaranteed to be word-aligned. */
1610 static void site_outgoing(void *sst, struct buffer_if *buf)
1611 {
1612     struct site *st=sst;
1613     cstring_t transform_err;
1614     
1615     if (st->state==SITE_STOP) {
1616         BUF_FREE(buf);
1617         return;
1618     }
1619
1620     st->allow_send_prod=1;
1621
1622     /* In all other states we consider delivering the packet if we have
1623        a valid key and a valid address to send it to. */
1624     if (current_valid(st) && transport_peers_valid(&st->peers)) {
1625         /* Transform it and send it */
1626         if (buf->size>0) {
1627             buf_prepend_uint32(buf,LABEL_MSG9);
1628             if (call_transform_forwards(st, st->current.transform,
1629                                         buf, &transform_err))
1630                 goto free_out;
1631             buf_prepend_uint32(buf,LABEL_MSG0);
1632             buf_prepend_uint32(buf,st->index);
1633             buf_prepend_uint32(buf,st->current.remote_session_id);
1634             transport_xmit(st,&st->peers,buf,False);
1635         }
1636     free_out:
1637         BUF_FREE(buf);
1638         return;
1639     }
1640
1641     slog(st,LOG_DROP,"discarding outgoing packet of size %d",buf->size);
1642     BUF_FREE(buf);
1643     initiate_key_setup(st,"outgoing packet",0);
1644 }
1645
1646 static bool_t named_for_us(struct site *st, const struct buffer_if *buf_in,
1647                            uint32_t type, struct msg *m)
1648     /* For packets which are identified by the local and remote names.
1649      * If it has our name and our peer's name in it it's for us. */
1650 {
1651     struct buffer_if buf[1];
1652     buffer_readonly_clone(buf,buf_in);
1653     return unpick_msg(st,type,buf,m)
1654         && name_matches(&m->remote,st->remotename)
1655         && name_matches(&m->local,st->localname);
1656 }
1657
1658 /* This function is called by the communication device to deliver
1659    packets from our peers.
1660    It should return True if the packet is recognised as being for
1661    this current site instance (and should therefore not be processed
1662    by other sites), even if the packet was otherwise ignored. */
1663 static bool_t site_incoming(void *sst, struct buffer_if *buf,
1664                             const struct comm_addr *source)
1665 {
1666     struct site *st=sst;
1667
1668     if (buf->size < 12) return False;
1669
1670     uint32_t dest=get_uint32(buf->start);
1671     uint32_t msgtype=get_uint32(buf->start+8);
1672     struct msg named_msg;
1673
1674     if (msgtype==LABEL_MSG1) {
1675         if (!named_for_us(st,buf,msgtype,&named_msg))
1676             return False;
1677         /* It's a MSG1 addressed to us. Decide what to do about it. */
1678         dump_packet(st,buf,source,True);
1679         if (st->state==SITE_RUN || st->state==SITE_RESOLVE ||
1680             st->state==SITE_WAIT) {
1681             /* We should definitely process it */
1682             transport_compute_setupinit_peers(st,0,source);
1683             if (process_msg1(st,buf,source,&named_msg)) {
1684                 slog(st,LOG_SETUP_INIT,"key setup initiated by peer");
1685                 bool_t entered=enter_new_state(st,SITE_SENTMSG2);
1686                 if (entered && st->address && st->local_mobile)
1687                     /* We must do this as the very last thing, because
1688                        the resolver callback might reenter us. */
1689                     ensure_resolving(st);
1690             } else {
1691                 slog(st,LOG_ERROR,"failed to process incoming msg1");
1692             }
1693             BUF_FREE(buf);
1694             return True;
1695         } else if (st->state==SITE_SENTMSG1) {
1696             /* We've just sent a message 1! They may have crossed on
1697                the wire. If we have priority then we ignore the
1698                incoming one, otherwise we process it as usual. */
1699             if (st->setup_priority) {
1700                 BUF_FREE(buf);
1701                 slog(st,LOG_DUMP,"crossed msg1s; we are higher "
1702                      "priority => ignore incoming msg1");
1703                 return True;
1704             } else {
1705                 slog(st,LOG_DUMP,"crossed msg1s; we are lower "
1706                      "priority => use incoming msg1");
1707                 if (process_msg1(st,buf,source,&named_msg)) {
1708                     BUF_FREE(&st->buffer); /* Free our old message 1 */
1709                     transport_setup_msgok(st,source);
1710                     enter_new_state(st,SITE_SENTMSG2);
1711                 } else {
1712                     slog(st,LOG_ERROR,"failed to process an incoming "
1713                          "crossed msg1 (we have low priority)");
1714                 }
1715                 BUF_FREE(buf);
1716                 return True;
1717             }
1718         }
1719         /* The message 1 was received at an unexpected stage of the
1720            key setup. XXX POLICY - what do we do? */
1721         slog(st,LOG_UNEXPECTED,"unexpected incoming message 1");
1722         BUF_FREE(buf);
1723         return True;
1724     }
1725     if (msgtype==LABEL_PROD) {
1726         if (!named_for_us(st,buf,msgtype,&named_msg))
1727             return False;
1728         dump_packet(st,buf,source,True);
1729         if (st->state!=SITE_RUN) {
1730             slog(st,LOG_DROP,"ignoring PROD when not in state RUN");
1731         } else if (current_valid(st)) {
1732             slog(st,LOG_DROP,"ignoring PROD when we think we have a key");
1733         } else {
1734             initiate_key_setup(st,"peer sent PROD packet",source);
1735         }
1736         BUF_FREE(buf);
1737         return True;
1738     }
1739     if (dest==st->index) {
1740         /* Explicitly addressed to us */
1741         if (msgtype!=LABEL_MSG0) dump_packet(st,buf,source,True);
1742         switch (msgtype) {
1743         case LABEL_NAK:
1744             /* If the source is our current peer then initiate a key setup,
1745                because our peer's forgotten the key */
1746             if (get_uint32(buf->start+4)==st->current.remote_session_id) {
1747                 bool_t initiated;
1748                 initiated = initiate_key_setup(st,"received a NAK",source);
1749                 if (!initiated) generate_send_prod(st,source);
1750             } else {
1751                 slog(st,LOG_SEC,"bad incoming NAK");
1752             }
1753             break;
1754         case LABEL_MSG0:
1755             process_msg0(st,buf,source);
1756             break;
1757         case LABEL_MSG1:
1758             /* Setup packet: should not have been explicitly addressed
1759                to us */
1760             slog(st,LOG_SEC,"incoming explicitly addressed msg1");
1761             break;
1762         case LABEL_MSG2:
1763             /* Setup packet: expected only in state SENTMSG1 */
1764             if (st->state!=SITE_SENTMSG1) {
1765                 slog(st,LOG_UNEXPECTED,"unexpected MSG2");
1766             } else if (process_msg2(st,buf,source)) {
1767                 transport_setup_msgok(st,source);
1768                 enter_new_state(st,SITE_SENTMSG3);
1769             } else {
1770                 slog(st,LOG_SEC,"invalid MSG2");
1771             }
1772             break;
1773         case LABEL_MSG3:
1774         case LABEL_MSG3BIS:
1775             /* Setup packet: expected only in state SENTMSG2 */
1776             if (st->state!=SITE_SENTMSG2) {
1777                 slog(st,LOG_UNEXPECTED,"unexpected MSG3");
1778             } else if (process_msg3(st,buf,source,msgtype)) {
1779                 transport_setup_msgok(st,source);
1780                 enter_new_state(st,SITE_SENTMSG4);
1781             } else {
1782                 slog(st,LOG_SEC,"invalid MSG3");
1783             }
1784             break;
1785         case LABEL_MSG4:
1786             /* Setup packet: expected only in state SENTMSG3 */
1787             if (st->state!=SITE_SENTMSG3) {
1788                 slog(st,LOG_UNEXPECTED,"unexpected MSG4");
1789             } else if (process_msg4(st,buf,source)) {
1790                 transport_setup_msgok(st,source);
1791                 enter_new_state(st,SITE_SENTMSG5);
1792             } else {
1793                 slog(st,LOG_SEC,"invalid MSG4");
1794             }
1795             break;
1796         case LABEL_MSG5:
1797             /* Setup packet: expected only in state SENTMSG4 */
1798             /* (may turn up in state RUN if our return MSG6 was lost
1799                and the new key has already been activated. In that
1800                case we discard it. The peer will realise that we
1801                are using the new key when they see our data packets.
1802                Until then the peer's data packets to us get discarded. */
1803             if (st->state==SITE_SENTMSG4) {
1804                 if (process_msg5(st,buf,source,st->new_transform)) {
1805                     transport_setup_msgok(st,source);
1806                     enter_new_state(st,SITE_RUN);
1807                 } else {
1808                     slog(st,LOG_SEC,"invalid MSG5");
1809                 }
1810             } else if (st->state==SITE_RUN) {
1811                 if (process_msg5(st,buf,source,st->current.transform)) {
1812                     slog(st,LOG_DROP,"got MSG5, retransmitting MSG6");
1813                     transport_setup_msgok(st,source);
1814                     create_msg6(st,st->current.transform,
1815                                 st->current.remote_session_id);
1816                     transport_xmit(st,&st->peers,&st->buffer,True);
1817                     BUF_FREE(&st->buffer);
1818                 } else {
1819                     slog(st,LOG_SEC,"invalid MSG5 (in state RUN)");
1820                 }
1821             } else {
1822                 slog(st,LOG_UNEXPECTED,"unexpected MSG5");
1823             }
1824             break;
1825         case LABEL_MSG6:
1826             /* Setup packet: expected only in state SENTMSG5 */
1827             if (st->state!=SITE_SENTMSG5) {
1828                 slog(st,LOG_UNEXPECTED,"unexpected MSG6");
1829             } else if (process_msg6(st,buf,source)) {
1830                 BUF_FREE(&st->buffer); /* Free message 5 */
1831                 transport_setup_msgok(st,source);
1832                 activate_new_key(st);
1833             } else {
1834                 slog(st,LOG_SEC,"invalid MSG6");
1835             }
1836             break;
1837         default:
1838             slog(st,LOG_SEC,"received message of unknown type 0x%08x",
1839                  msgtype);
1840             break;
1841         }
1842         BUF_FREE(buf);
1843         return True;
1844     }
1845
1846     return False;
1847 }
1848
1849 static void site_control(void *vst, bool_t run)
1850 {
1851     struct site *st=vst;
1852     if (run) enter_state_run(st);
1853     else enter_state_stop(st);
1854 }
1855
1856 static void site_phase_hook(void *sst, uint32_t newphase)
1857 {
1858     struct site *st=sst;
1859
1860     /* The program is shutting down; tell our peer */
1861     send_msg7(st,"shutting down");
1862 }
1863
1864 static list_t *site_apply(closure_t *self, struct cloc loc, dict_t *context,
1865                           list_t *args)
1866 {
1867     static uint32_t index_sequence;
1868     struct site *st;
1869     item_t *item;
1870     dict_t *dict;
1871     int i;
1872
1873     st=safe_malloc(sizeof(*st),"site_apply");
1874
1875     st->cl.description="site";
1876     st->cl.type=CL_SITE;
1877     st->cl.apply=NULL;
1878     st->cl.interface=&st->ops;
1879     st->ops.st=st;
1880     st->ops.control=site_control;
1881     st->ops.status=site_status;
1882
1883     /* First parameter must be a dict */
1884     item=list_elem(args,0);
1885     if (!item || item->type!=t_dict)
1886         cfgfatal(loc,"site","parameter must be a dictionary\n");
1887     
1888     dict=item->data.dict;
1889     st->localname=dict_read_string(dict, "local-name", True, "site", loc);
1890     st->remotename=dict_read_string(dict, "name", True, "site", loc);
1891
1892     st->peer_mobile=dict_read_bool(dict,"mobile",False,"site",loc,False);
1893     st->local_mobile=
1894         dict_read_bool(dict,"local-mobile",False,"site",loc,False);
1895
1896     /* Sanity check (which also allows the 'sites' file to include
1897        site() closures for all sites including our own): refuse to
1898        talk to ourselves */
1899     if (strcmp(st->localname,st->remotename)==0) {
1900         Message(M_DEBUG,"site %s: local-name==name -> ignoring this site\n",
1901                 st->localname);
1902         if (st->peer_mobile != st->local_mobile)
1903             cfgfatal(loc,"site","site %s's peer-mobile=%d"
1904                     " but our local-mobile=%d\n",
1905                     st->localname, st->peer_mobile, st->local_mobile);
1906         free(st);
1907         return NULL;
1908     }
1909     if (st->peer_mobile && st->local_mobile) {
1910         Message(M_WARNING,"site %s: site is mobile but so are we"
1911                 " -> ignoring this site\n", st->remotename);
1912         free(st);
1913         return NULL;
1914     }
1915
1916     assert(index_sequence < 0xffffffffUL);
1917     st->index = ++index_sequence;
1918     st->local_capabilities = 0;
1919     st->netlink=find_cl_if(dict,"link",CL_NETLINK,True,"site",loc);
1920
1921 #define GET_CLOSURE_LIST(dictkey,things,nthings,CL_TYPE) do{            \
1922     list_t *things##_cfg=dict_lookup(dict,dictkey);                     \
1923     if (!things##_cfg)                                                  \
1924         cfgfatal(loc,"site","closure list \"%s\" not found\n",dictkey); \
1925     st->nthings=list_length(things##_cfg);                              \
1926     st->things=safe_malloc_ary(sizeof(*st->things),st->nthings,dictkey "s"); \
1927     assert(st->nthings);                                                \
1928     for (i=0; i<st->nthings; i++) {                                     \
1929         item_t *item=list_elem(things##_cfg,i);                         \
1930         if (item->type!=t_closure)                                      \
1931             cfgfatal(loc,"site","%s is not a closure\n",dictkey);       \
1932         closure_t *cl=item->data.closure;                               \
1933         if (cl->type!=CL_TYPE)                                          \
1934             cfgfatal(loc,"site","%s closure wrong type\n",dictkey);     \
1935         st->things[i]=cl->interface;                                    \
1936     }                                                                   \
1937 }while(0)
1938
1939     GET_CLOSURE_LIST("comm",comms,ncomms,CL_COMM);
1940
1941     st->resolver=find_cl_if(dict,"resolver",CL_RESOLVER,True,"site",loc);
1942     st->log=find_cl_if(dict,"log",CL_LOG,True,"site",loc);
1943     st->random=find_cl_if(dict,"random",CL_RANDOMSRC,True,"site",loc);
1944
1945     st->privkey=find_cl_if(dict,"local-key",CL_RSAPRIVKEY,True,"site",loc);
1946     st->address=dict_read_string(dict, "address", False, "site", loc);
1947     if (st->address)
1948         st->remoteport=dict_read_number(dict,"port",True,"site",loc,0);
1949     else st->remoteport=0;
1950     st->pubkey=find_cl_if(dict,"key",CL_RSAPUBKEY,True,"site",loc);
1951
1952     GET_CLOSURE_LIST("transform",transforms,ntransforms,CL_TRANSFORM);
1953
1954     st->dh=find_cl_if(dict,"dh",CL_DH,True,"site",loc);
1955     st->hash=find_cl_if(dict,"hash",CL_HASH,True,"site",loc);
1956
1957 #define DEFAULT(D) (st->peer_mobile || st->local_mobile \
1958                     ? DEFAULT_MOBILE_##D : DEFAULT_##D)
1959 #define CFG_NUMBER(k,D) dict_read_number(dict,(k),False,"site",loc,DEFAULT(D));
1960
1961     st->key_lifetime=         CFG_NUMBER("key-lifetime",  KEY_LIFETIME);
1962     st->setup_retries=        CFG_NUMBER("setup-retries", SETUP_RETRIES);
1963     st->setup_retry_interval= CFG_NUMBER("setup-timeout", SETUP_RETRY_INTERVAL);
1964     st->wait_timeout=         CFG_NUMBER("wait-time",     WAIT_TIME);
1965     st->mtu_target= dict_read_number(dict,"mtu-target",False,"site",loc,0);
1966
1967     st->mobile_peer_expiry= dict_read_number(
1968        dict,"mobile-peer-expiry",False,"site",loc,DEFAULT_MOBILE_PEER_EXPIRY);
1969
1970     st->transport_peers_max= !st->peer_mobile ? 1 : dict_read_number(
1971         dict,"mobile-peers-max",False,"site",loc,DEFAULT_MOBILE_PEERS_MAX);
1972     if (st->transport_peers_max<1 ||
1973         st->transport_peers_max>=MAX_MOBILE_PEERS_MAX) {
1974         cfgfatal(loc,"site","mobile-peers-max must be in range 1.."
1975                  STRING(MAX_MOBILE_PEERS_MAX) "\n");
1976     }
1977
1978     if (st->key_lifetime < DEFAULT(KEY_RENEGOTIATE_GAP)*2)
1979         st->key_renegotiate_time=st->key_lifetime/2;
1980     else
1981         st->key_renegotiate_time=st->key_lifetime-DEFAULT(KEY_RENEGOTIATE_GAP);
1982     st->key_renegotiate_time=dict_read_number(
1983         dict,"renegotiate-time",False,"site",loc,st->key_renegotiate_time);
1984     if (st->key_renegotiate_time > st->key_lifetime) {
1985         cfgfatal(loc,"site",
1986                  "renegotiate-time must be less than key-lifetime\n");
1987     }
1988
1989     st->log_events=string_list_to_word(dict_lookup(dict,"log-events"),
1990                                        log_event_table,"site");
1991
1992     st->resolving=False;
1993     st->allow_send_prod=0;
1994
1995     st->tunname=safe_malloc(strlen(st->localname)+strlen(st->remotename)+5,
1996                             "site_apply");
1997     sprintf(st->tunname,"%s<->%s",st->localname,st->remotename);
1998
1999     /* The information we expect to see in incoming messages of type 1 */
2000     /* fixme: lots of unchecked overflows here, but the results are only
2001        corrupted packets rather than undefined behaviour */
2002     st->setup_priority=(strcmp(st->localname,st->remotename)>0);
2003
2004     buffer_new(&st->buffer,SETUP_BUFFER_LEN);
2005
2006     buffer_new(&st->scratch,SETUP_BUFFER_LEN);
2007     BUF_ALLOC(&st->scratch,"site:scratch");
2008
2009     /* We are interested in poll(), but only for timeouts. We don't have
2010        any fds of our own. */
2011     register_for_poll(st, site_beforepoll, site_afterpoll, 0, "site");
2012     st->timeout=0;
2013
2014     st->remote_capabilities=0;
2015     st->chosen_transform=0;
2016     st->current.key_timeout=0;
2017     st->auxiliary_key.key_timeout=0;
2018     transport_peers_clear(st,&st->peers);
2019     transport_peers_clear(st,&st->setup_peers);
2020     /* XXX mlock these */
2021     st->dhsecret=safe_malloc(st->dh->len,"site:dhsecret");
2022     st->sharedsecretlen=st->sharedsecretallocd=0;
2023     st->sharedsecret=0;
2024
2025     for (i=0; i<st->ntransforms; i++) {
2026         struct transform_if *ti=st->transforms[i];
2027         uint32_t capbit = 1UL << ti->capab_transformnum;
2028         if (st->local_capabilities & capbit)
2029             slog(st,LOG_ERROR,"transformnum capability bit"
2030                  " %d (%#"PRIx32") reused", ti->capab_transformnum, capbit);
2031         st->local_capabilities |= capbit;
2032     }
2033
2034     /* We need to register the remote networks with the netlink device */
2035     uint32_t netlink_mtu; /* local virtual interface mtu */
2036     st->netlink->reg(st->netlink->st, site_outgoing, st, &netlink_mtu);
2037     if (!st->mtu_target)
2038         st->mtu_target=netlink_mtu;
2039     
2040     for (i=0; i<st->ncomms; i++)
2041         st->comms[i]->request_notify(st->comms[i]->st, st, site_incoming);
2042
2043     st->current.transform=0;
2044     st->auxiliary_key.transform=0;
2045     st->new_transform=0;
2046     st->auxiliary_is_new=0;
2047
2048     enter_state_stop(st);
2049
2050     add_hook(PHASE_SHUTDOWN,site_phase_hook,st);
2051
2052     return new_closure(&st->cl);
2053 }
2054
2055 void site_module(dict_t *dict)
2056 {
2057     add_closure(dict,"site",site_apply);
2058 }
2059
2060
2061 /***** TRANSPORT PEERS definitions *****/
2062
2063 static void transport_peers_debug(struct site *st, transport_peers *dst,
2064                                   const char *didwhat,
2065                                   int nargs, const struct comm_addr *args,
2066                                   size_t stride) {
2067     int i;
2068     char *argp;
2069
2070     if (!(st->log_events & LOG_PEER_ADDRS))
2071         return; /* an optimisation */
2072
2073     slog(st, LOG_PEER_ADDRS, "peers (%s) %s nargs=%d => npeers=%d",
2074          (dst==&st->peers ? "data" :
2075           dst==&st->setup_peers ? "setup" : "UNKNOWN"),
2076          didwhat, nargs, dst->npeers);
2077
2078     for (i=0, argp=(void*)args;
2079          i<nargs;
2080          i++, (argp+=stride?stride:sizeof(*args))) {
2081         const struct comm_addr *ca=(void*)argp;
2082         slog(st, LOG_PEER_ADDRS, " args: addrs[%d]=%s",
2083              i, comm_addr_to_string(ca));
2084     }
2085     for (i=0; i<dst->npeers; i++) {
2086         struct timeval diff;
2087         timersub(tv_now,&dst->peers[i].last,&diff);
2088         const struct comm_addr *ca=&dst->peers[i].addr;
2089         slog(st, LOG_PEER_ADDRS, " peers: addrs[%d]=%s T-%ld.%06ld",
2090              i, comm_addr_to_string(ca),
2091              (unsigned long)diff.tv_sec, (unsigned long)diff.tv_usec);
2092     }
2093 }
2094
2095 static bool_t transport_addrs_equal(const struct comm_addr *a,
2096                                     const struct comm_addr *b) {
2097     return !memcmp(a,b,sizeof(*a));
2098 }
2099
2100 static void transport_peers_expire(struct site *st, transport_peers *peers) {
2101     /* peers must be sorted first */
2102     int previous_peers=peers->npeers;
2103     struct timeval oldest;
2104     oldest.tv_sec  = tv_now->tv_sec - st->mobile_peer_expiry;
2105     oldest.tv_usec = tv_now->tv_usec;
2106     while (peers->npeers>1 &&
2107            timercmp(&peers->peers[peers->npeers-1].last, &oldest, <))
2108         peers->npeers--;
2109     if (peers->npeers != previous_peers)
2110         transport_peers_debug(st,peers,"expire", 0,0,0);
2111 }
2112
2113 static bool_t transport_peer_record_one(struct site *st, transport_peers *peers,
2114                                         const struct comm_addr *ca,
2115                                         const struct timeval *tv) {
2116     /* returns false if output is full */
2117     int search;
2118
2119     if (peers->npeers > st->transport_peers_max)
2120         return 0;
2121
2122     for (search=0; search<peers->npeers; search++)
2123         if (transport_addrs_equal(&peers->peers[search].addr, ca))
2124             return 1;
2125
2126     peers->peers[peers->npeers].addr = *ca;
2127     peers->peers[peers->npeers].last = *tv;
2128     peers->npeers++;
2129     return 1;
2130 }
2131
2132 static void transport_record_peers(struct site *st, transport_peers *peers,
2133                                    const struct comm_addr *addrs, int naddrs,
2134                                    const char *m) {
2135     /* We add addrs into peers.  The new entries end up at the front
2136      * and displace entries towards the end (perhaps even off the end).
2137      * Any existing matching entries are moved up to the front.
2138      * Caller must first call transport_peers_expire. */
2139
2140     if (naddrs==1 && peers->npeers>=1 &&
2141         transport_addrs_equal(&addrs[0], &peers->peers[0].addr)) {
2142         /* optimisation, also avoids debug for trivial updates */
2143         peers->peers[0].last = *tv_now;
2144         return;
2145     }
2146
2147     int old_npeers=peers->npeers;
2148     transport_peers old_peers[old_npeers];
2149     memcpy(old_peers,peers->peers,sizeof(old_npeers));
2150
2151     peers->npeers=0;
2152     int i;
2153     for (i=0; i<naddrs; i++) {
2154         if (!transport_peer_record_one(st,peers, &addrs[i], tv_now))
2155             break;
2156     }
2157     for (i=0; i<old_npeers; i++) {
2158         const transport_peer *old=&old_peers->peers[i];
2159         if (!transport_peer_record_one(st,peers, &old->addr, &old->last))
2160             break;
2161     }
2162
2163     transport_peers_debug(st,peers,m, naddrs,addrs,0);
2164 }
2165
2166 static bool_t transport_compute_setupinit_peers(struct site *st,
2167         const struct comm_addr *configured_addr /* 0 if none or not found */,
2168         const struct comm_addr *incoming_packet_addr /* 0 if none */) {
2169
2170     if (!configured_addr && !incoming_packet_addr &&
2171         !transport_peers_valid(&st->peers))
2172         return False;
2173
2174     slog(st,LOG_SETUP_INIT,
2175          "using:%s%s %d old peer address(es)",
2176          configured_addr ? " configured address;" : "",
2177          incoming_packet_addr ? " incoming packet address;" : "",
2178          st->peers.npeers);
2179
2180     /* Non-mobile peers have st->peers.npeers==0 or ==1, since they
2181      * have transport_peers_max==1.  The effect is that this code
2182      * always uses the configured address if supplied, or otherwise
2183      * the address of the incoming PROD, or the existing data peer if
2184      * one exists; this is as desired. */
2185
2186     transport_peers_copy(st,&st->setup_peers,&st->peers);
2187     transport_peers_expire(st,&st->setup_peers);
2188
2189     if (incoming_packet_addr)
2190         transport_record_peers(st,&st->setup_peers,
2191                                incoming_packet_addr,1, "incoming");
2192
2193     if (configured_addr)
2194         transport_record_peers(st,&st->setup_peers,
2195                               configured_addr,1, "setupinit");
2196
2197     assert(transport_peers_valid(&st->setup_peers));
2198     return True;
2199 }
2200
2201 static void transport_setup_msgok(struct site *st, const struct comm_addr *a) {
2202     if (st->peer_mobile) {
2203         transport_peers_expire(st,&st->setup_peers);
2204         transport_record_peers(st,&st->setup_peers,a,1,"setupmsg");
2205     }
2206 }
2207 static void transport_data_msgok(struct site *st, const struct comm_addr *a) {
2208     if (st->peer_mobile) {
2209         transport_peers_expire(st,&st->setup_peers);
2210         transport_record_peers(st,&st->peers,a,1,"datamsg");
2211     }
2212 }
2213
2214 static int transport_peers_valid(transport_peers *peers) {
2215     return peers->npeers;
2216 }
2217 static void transport_peers_clear(struct site *st, transport_peers *peers) {
2218     peers->npeers= 0;
2219     transport_peers_debug(st,peers,"clear",0,0,0);
2220 }
2221 static void transport_peers_copy(struct site *st, transport_peers *dst,
2222                                  const transport_peers *src) {
2223     dst->npeers=src->npeers;
2224     memcpy(dst->peers, src->peers, sizeof(*dst->peers) * dst->npeers);
2225     transport_peers_debug(st,dst,"copy",
2226                           src->npeers, &src->peers->addr, sizeof(*src->peers));
2227 }
2228
2229 static void transport_resolve_complete(struct site *st,
2230                                        const struct comm_addr *ca_use) {
2231     transport_peers_expire(st,&st->peers);
2232     transport_record_peers(st,&st->peers,ca_use,1,"resolved data");
2233     transport_peers_expire(st,&st->setup_peers);
2234     transport_record_peers(st,&st->setup_peers,ca_use,1,"resolved setup");
2235 }
2236
2237 static void transport_resolve_complete_tardy(struct site *st,
2238                                              const struct comm_addr *ca_use) {
2239     transport_peers_expire(st,&st->peers);
2240     transport_record_peers(st,&st->peers,ca_use,1,"resolved tardily");
2241 }
2242
2243 static void transport_peers__copy_by_mask(transport_peer *out, int *nout_io,
2244                                           unsigned mask,
2245                                           const transport_peers *inp) {
2246     /* out and in->peers may be the same region, or nonoverlapping */
2247     const transport_peer *in=inp->peers;
2248     int slot;
2249     for (slot=0; slot<inp->npeers; slot++) {
2250         if (!(mask & (1U << slot)))
2251             continue;
2252         if (!(out==in && slot==*nout_io))
2253             memcpy(&out[*nout_io], &in[slot], sizeof(out[0]));
2254         (*nout_io)++;
2255     }
2256 }
2257
2258 void transport_xmit(struct site *st, transport_peers *peers,
2259                     struct buffer_if *buf, bool_t candebug) {
2260     int slot;
2261     transport_peers_expire(st, peers);
2262     unsigned failed=0; /* bitmask */
2263     assert(MAX_MOBILE_PEERS_MAX < sizeof(unsigned)*CHAR_BIT);
2264
2265     int nfailed=0;
2266     for (slot=0; slot<peers->npeers; slot++) {
2267         transport_peer *peer=&peers->peers[slot];
2268         if (candebug)
2269             dump_packet(st, buf, &peer->addr, False);
2270         bool_t ok =
2271             peer->addr.comm->sendmsg(peer->addr.comm->st, buf, &peer->addr);
2272         if (!ok) {
2273             failed |= 1U << slot;
2274             nfailed++;
2275         }
2276         if (ok && !st->peer_mobile)
2277             break;
2278     }
2279     /* Now we need to demote/delete failing addrs: if we are mobile we
2280      * merely demote them; otherwise we delete them. */
2281     if (st->local_mobile) {
2282         unsigned expected = ((1U << nfailed)-1) << (peers->npeers-nfailed);
2283         /* `expected' has all the failures at the end already */
2284         if (failed != expected) {
2285             int fslot=0;
2286             transport_peer failedpeers[nfailed];
2287             transport_peers__copy_by_mask(failedpeers, &fslot, failed,peers);
2288             assert(fslot == nfailed);
2289             int wslot=0;
2290             transport_peers__copy_by_mask(peers->peers,&wslot,~failed,peers);
2291             assert(wslot+nfailed == peers->npeers);
2292             memcpy(peers->peers+wslot,failedpeers,sizeof(failedpeers));
2293         }
2294     } else {
2295         if (failed && peers->npeers > 1) {
2296             int wslot=0;
2297             transport_peers__copy_by_mask(peers->peers,&wslot,~failed,peers);
2298             peers->npeers=wslot;
2299         }
2300     }
2301 }
2302
2303 /***** END of transport peers declarations *****/