]> CyberLeo.Net >> Repos - FreeBSD/releng/10.0.git/blob - sys/netgraph/ng_pptpgre.c
- Copy stable/10 (r259064) to releng/10.0 as part of the
[FreeBSD/releng/10.0.git] / sys / netgraph / ng_pptpgre.c
1 /*
2  * ng_pptpgre.c
3  */
4
5 /*-
6  * Copyright (c) 1996-1999 Whistle Communications, Inc.
7  * All rights reserved.
8  * 
9  * Subject to the following obligations and disclaimer of warranty, use and
10  * redistribution of this software, in source or object code forms, with or
11  * without modifications are expressly permitted by Whistle Communications;
12  * provided, however, that:
13  * 1. Any and all reproductions of the source or object code must include the
14  *    copyright notice above and the following disclaimer of warranties; and
15  * 2. No rights are granted, in any manner or form, to use Whistle
16  *    Communications, Inc. trademarks, including the mark "WHISTLE
17  *    COMMUNICATIONS" on advertising, endorsements, or otherwise except as
18  *    such appears in the above copyright notice or in the software.
19  * 
20  * THIS SOFTWARE IS BEING PROVIDED BY WHISTLE COMMUNICATIONS "AS IS", AND
21  * TO THE MAXIMUM EXTENT PERMITTED BY LAW, WHISTLE COMMUNICATIONS MAKES NO
22  * REPRESENTATIONS OR WARRANTIES, EXPRESS OR IMPLIED, REGARDING THIS SOFTWARE,
23  * INCLUDING WITHOUT LIMITATION, ANY AND ALL IMPLIED WARRANTIES OF
24  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, OR NON-INFRINGEMENT.
25  * WHISTLE COMMUNICATIONS DOES NOT WARRANT, GUARANTEE, OR MAKE ANY
26  * REPRESENTATIONS REGARDING THE USE OF, OR THE RESULTS OF THE USE OF THIS
27  * SOFTWARE IN TERMS OF ITS CORRECTNESS, ACCURACY, RELIABILITY OR OTHERWISE.
28  * IN NO EVENT SHALL WHISTLE COMMUNICATIONS BE LIABLE FOR ANY DAMAGES
29  * RESULTING FROM OR ARISING OUT OF ANY USE OF THIS SOFTWARE, INCLUDING
30  * WITHOUT LIMITATION, ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,
31  * PUNITIVE, OR CONSEQUENTIAL DAMAGES, PROCUREMENT OF SUBSTITUTE GOODS OR
32  * SERVICES, LOSS OF USE, DATA OR PROFITS, HOWEVER CAUSED AND UNDER ANY
33  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
34  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
35  * THIS SOFTWARE, EVEN IF WHISTLE COMMUNICATIONS IS ADVISED OF THE POSSIBILITY
36  * OF SUCH DAMAGE.
37  *
38  * Author: Archie Cobbs <archie@freebsd.org>
39  *
40  * $FreeBSD$
41  * $Whistle: ng_pptpgre.c,v 1.7 1999/12/08 00:10:06 archie Exp $
42  */
43
44 /*
45  * PPTP/GRE netgraph node type.
46  *
47  * This node type does the GRE encapsulation as specified for the PPTP
48  * protocol (RFC 2637, section 4).  This includes sequencing and
49  * retransmission of frames, but not the actual packet delivery nor
50  * any of the TCP control stream protocol.
51  *
52  * The "upper" hook of this node is suitable for attaching to a "ppp"
53  * node link hook.  The "lower" hook of this node is suitable for attaching
54  * to a "ksocket" node on hook "inet/raw/gre".
55  */
56
57 #include <sys/param.h>
58 #include <sys/systm.h>
59 #include <sys/kernel.h>
60 #include <sys/time.h>
61 #include <sys/lock.h>
62 #include <sys/malloc.h>
63 #include <sys/mbuf.h>
64 #include <sys/mutex.h>
65 #include <sys/endian.h>
66 #include <sys/errno.h>
67
68 #include <netinet/in.h>
69 #include <netinet/in_systm.h>
70 #include <netinet/ip.h>
71
72 #include <netgraph/ng_message.h>
73 #include <netgraph/netgraph.h>
74 #include <netgraph/ng_parse.h>
75 #include <netgraph/ng_pptpgre.h>
76
77 /* GRE packet format, as used by PPTP */
78 struct greheader {
79 #if BYTE_ORDER == LITTLE_ENDIAN
80         u_char          recursion:3;            /* recursion control */
81         u_char          ssr:1;                  /* strict source route */
82         u_char          hasSeq:1;               /* sequence number present */
83         u_char          hasKey:1;               /* key present */
84         u_char          hasRoute:1;             /* routing present */
85         u_char          hasSum:1;               /* checksum present */
86         u_char          vers:3;                 /* version */
87         u_char          flags:4;                /* flags */
88         u_char          hasAck:1;               /* acknowlege number present */
89 #elif BYTE_ORDER == BIG_ENDIAN
90         u_char          hasSum:1;               /* checksum present */
91         u_char          hasRoute:1;             /* routing present */
92         u_char          hasKey:1;               /* key present */
93         u_char          hasSeq:1;               /* sequence number present */
94         u_char          ssr:1;                  /* strict source route */
95         u_char          recursion:3;            /* recursion control */
96         u_char          hasAck:1;               /* acknowlege number present */
97         u_char          flags:4;                /* flags */
98         u_char          vers:3;                 /* version */
99 #else
100 #error BYTE_ORDER is not defined properly
101 #endif
102         u_int16_t       proto;                  /* protocol (ethertype) */
103         u_int16_t       length;                 /* payload length */
104         u_int16_t       cid;                    /* call id */
105         u_int32_t       data[0];                /* opt. seq, ack, then data */
106 };
107
108 /* The PPTP protocol ID used in the GRE 'proto' field */
109 #define PPTP_GRE_PROTO          0x880b
110
111 /* Bits that must be set a certain way in all PPTP/GRE packets */
112 #define PPTP_INIT_VALUE         ((0x2001 << 16) | PPTP_GRE_PROTO)
113 #define PPTP_INIT_MASK          0xef7fffff
114
115 /* Min and max packet length */
116 #define PPTP_MAX_PAYLOAD        (0xffff - sizeof(struct greheader) - 8)
117
118 /* All times are scaled by this (PPTP_TIME_SCALE time units = 1 sec.) */
119 #define PPTP_TIME_SCALE         1024                    /* milliseconds */
120 typedef u_int64_t               pptptime_t;
121
122 /* Acknowledgment timeout parameters and functions */
123 #define PPTP_XMIT_WIN           16                      /* max xmit window */
124 #define PPTP_MIN_TIMEOUT        (PPTP_TIME_SCALE / 83)  /* 12 milliseconds */
125 #define PPTP_MAX_TIMEOUT        (3 * PPTP_TIME_SCALE)   /* 3 seconds */
126
127 /* When we recieve a packet, we wait to see if there's an outgoing packet
128    we can piggy-back the ACK off of. These parameters determine the mimimum
129    and maxmimum length of time we're willing to wait in order to do that.
130    These have no effect unless "enableDelayedAck" is turned on. */
131 #define PPTP_MIN_ACK_DELAY      (PPTP_TIME_SCALE / 500) /* 2 milliseconds */
132 #define PPTP_MAX_ACK_DELAY      (PPTP_TIME_SCALE / 2)   /* 500 milliseconds */
133
134 /* See RFC 2637 section 4.4 */
135 #define PPTP_ACK_ALPHA(x)       (((x) + 4) >> 3)        /* alpha = 0.125 */
136 #define PPTP_ACK_BETA(x)        (((x) + 2) >> 2)        /* beta = 0.25 */
137 #define PPTP_ACK_CHI(x)         ((x) << 2)      /* chi = 4 */
138 #define PPTP_ACK_DELTA(x)       ((x) << 1)      /* delta = 2 */
139
140 #define PPTP_SEQ_DIFF(x,y)      ((int32_t)(x) - (int32_t)(y))
141
142 #define SESSHASHSIZE            0x0020
143 #define SESSHASH(x)             (((x) ^ ((x) >> 8)) & (SESSHASHSIZE - 1))
144
145 /* We keep packet retransmit and acknowlegement state in this struct */
146 struct ng_pptpgre_sess {
147         node_p                  node;           /* this node pointer */
148         hook_p                  hook;           /* hook to upper layers */
149         struct ng_pptpgre_conf  conf;           /* configuration info */
150         struct mtx              mtx;            /* session mutex */
151         u_int32_t               recvSeq;        /* last seq # we rcv'd */
152         u_int32_t               xmitSeq;        /* last seq # we sent */
153         u_int32_t               recvAck;        /* last seq # peer ack'd */
154         u_int32_t               xmitAck;        /* last seq # we ack'd */
155         int32_t                 ato;            /* adaptive time-out value */
156         int32_t                 rtt;            /* round trip time estimate */
157         int32_t                 dev;            /* deviation estimate */
158         u_int16_t               xmitWin;        /* size of xmit window */
159         struct callout          sackTimer;      /* send ack timer */
160         struct callout          rackTimer;      /* recv ack timer */
161         u_int32_t               winAck;         /* seq when xmitWin will grow */
162         pptptime_t              timeSent[PPTP_XMIT_WIN];
163         LIST_ENTRY(ng_pptpgre_sess) sessions;
164 };
165 typedef struct ng_pptpgre_sess *hpriv_p;
166
167 /* Node private data */
168 struct ng_pptpgre_private {
169         hook_p                  upper;          /* hook to upper layers */
170         hook_p                  lower;          /* hook to lower layers */
171         struct ng_pptpgre_sess  uppersess;      /* default session for compat */
172         LIST_HEAD(, ng_pptpgre_sess) sesshash[SESSHASHSIZE];
173         struct ng_pptpgre_stats stats;          /* node statistics */
174 };
175 typedef struct ng_pptpgre_private *priv_p;
176
177 /* Netgraph node methods */
178 static ng_constructor_t ng_pptpgre_constructor;
179 static ng_rcvmsg_t      ng_pptpgre_rcvmsg;
180 static ng_shutdown_t    ng_pptpgre_shutdown;
181 static ng_newhook_t     ng_pptpgre_newhook;
182 static ng_rcvdata_t     ng_pptpgre_rcvdata;
183 static ng_rcvdata_t     ng_pptpgre_rcvdata_lower;
184 static ng_disconnect_t  ng_pptpgre_disconnect;
185
186 /* Helper functions */
187 static int      ng_pptpgre_xmit(hpriv_p hpriv, item_p item);
188 static void     ng_pptpgre_start_send_ack_timer(hpriv_p hpriv);
189 static void     ng_pptpgre_start_recv_ack_timer(hpriv_p hpriv);
190 static void     ng_pptpgre_recv_ack_timeout(node_p node, hook_p hook,
191                     void *arg1, int arg2);
192 static void     ng_pptpgre_send_ack_timeout(node_p node, hook_p hook,
193                     void *arg1, int arg2);
194 static hpriv_p  ng_pptpgre_find_session(priv_p privp, u_int16_t cid);
195 static void     ng_pptpgre_reset(hpriv_p hpriv);
196 static pptptime_t ng_pptpgre_time(void);
197
198 /* Parse type for struct ng_pptpgre_conf */
199 static const struct ng_parse_struct_field ng_pptpgre_conf_type_fields[]
200         = NG_PPTPGRE_CONF_TYPE_INFO;
201 static const struct ng_parse_type ng_pptpgre_conf_type = {
202         &ng_parse_struct_type,
203         &ng_pptpgre_conf_type_fields,
204 };
205
206 /* Parse type for struct ng_pptpgre_stats */
207 static const struct ng_parse_struct_field ng_pptpgre_stats_type_fields[]
208         = NG_PPTPGRE_STATS_TYPE_INFO;
209 static const struct ng_parse_type ng_pptp_stats_type = {
210         &ng_parse_struct_type,
211         &ng_pptpgre_stats_type_fields
212 };
213
214 /* List of commands and how to convert arguments to/from ASCII */
215 static const struct ng_cmdlist ng_pptpgre_cmdlist[] = {
216         {
217           NGM_PPTPGRE_COOKIE,
218           NGM_PPTPGRE_SET_CONFIG,
219           "setconfig",
220           &ng_pptpgre_conf_type,
221           NULL
222         },
223         {
224           NGM_PPTPGRE_COOKIE,
225           NGM_PPTPGRE_GET_CONFIG,
226           "getconfig",
227           &ng_parse_hint16_type,
228           &ng_pptpgre_conf_type
229         },
230         {
231           NGM_PPTPGRE_COOKIE,
232           NGM_PPTPGRE_GET_STATS,
233           "getstats",
234           NULL,
235           &ng_pptp_stats_type
236         },
237         {
238           NGM_PPTPGRE_COOKIE,
239           NGM_PPTPGRE_CLR_STATS,
240           "clrstats",
241           NULL,
242           NULL
243         },
244         {
245           NGM_PPTPGRE_COOKIE,
246           NGM_PPTPGRE_GETCLR_STATS,
247           "getclrstats",
248           NULL,
249           &ng_pptp_stats_type
250         },
251         { 0 }
252 };
253
254 /* Node type descriptor */
255 static struct ng_type ng_pptpgre_typestruct = {
256         .version =      NG_ABI_VERSION,
257         .name =         NG_PPTPGRE_NODE_TYPE,
258         .constructor =  ng_pptpgre_constructor,
259         .rcvmsg =       ng_pptpgre_rcvmsg,
260         .shutdown =     ng_pptpgre_shutdown,
261         .newhook =      ng_pptpgre_newhook,
262         .rcvdata =      ng_pptpgre_rcvdata,
263         .disconnect =   ng_pptpgre_disconnect,
264         .cmdlist =      ng_pptpgre_cmdlist,
265 };
266 NETGRAPH_INIT(pptpgre, &ng_pptpgre_typestruct);
267
268 #define ERROUT(x)       do { error = (x); goto done; } while (0)
269
270 /************************************************************************
271                         NETGRAPH NODE STUFF
272  ************************************************************************/
273
274 /*
275  * Node type constructor
276  */
277 static int
278 ng_pptpgre_constructor(node_p node)
279 {
280         priv_p priv;
281         int i;
282
283         /* Allocate private structure */
284         priv = malloc(sizeof(*priv), M_NETGRAPH, M_WAITOK | M_ZERO);
285
286         NG_NODE_SET_PRIVATE(node, priv);
287
288         /* Initialize state */
289         mtx_init(&priv->uppersess.mtx, "ng_pptp", NULL, MTX_DEF);
290         ng_callout_init(&priv->uppersess.sackTimer);
291         ng_callout_init(&priv->uppersess.rackTimer);
292         priv->uppersess.node = node;
293
294         for (i = 0; i < SESSHASHSIZE; i++)
295             LIST_INIT(&priv->sesshash[i]);
296
297         LIST_INSERT_HEAD(&priv->sesshash[0], &priv->uppersess, sessions);
298
299         /* Done */
300         return (0);
301 }
302
303 /*
304  * Give our OK for a hook to be added.
305  */
306 static int
307 ng_pptpgre_newhook(node_p node, hook_p hook, const char *name)
308 {
309         const priv_p priv = NG_NODE_PRIVATE(node);
310
311         /* Check hook name */
312         if (strcmp(name, NG_PPTPGRE_HOOK_UPPER) == 0) {
313                 priv->upper = hook;
314                 priv->uppersess.hook = hook;
315                 NG_HOOK_SET_PRIVATE(hook, &priv->uppersess);
316         } else if (strcmp(name, NG_PPTPGRE_HOOK_LOWER) == 0) {
317                 priv->lower = hook;
318                 NG_HOOK_SET_RCVDATA(hook, ng_pptpgre_rcvdata_lower);
319         } else {
320                 static const char hexdig[16] = "0123456789abcdef";
321                 const char *hex;
322                 hpriv_p hpriv;
323                 int i, j;
324                 uint16_t cid, hash;
325
326                 /* Parse hook name to get session ID */
327                 if (strncmp(name, NG_PPTPGRE_HOOK_SESSION_P,
328                     sizeof(NG_PPTPGRE_HOOK_SESSION_P) - 1) != 0)
329                         return (EINVAL);
330                 hex = name + sizeof(NG_PPTPGRE_HOOK_SESSION_P) - 1;
331                 for (cid = i = 0; i < 4; i++) {
332                         for (j = 0; j < 16 && hex[i] != hexdig[j]; j++);
333                         if (j == 16)
334                                 return (EINVAL);
335                         cid = (cid << 4) | j;
336                 }
337                 if (hex[i] != '\0')
338                         return (EINVAL);
339
340                 hpriv = malloc(sizeof(*hpriv), M_NETGRAPH, M_NOWAIT | M_ZERO);
341                 if (hpriv == NULL)
342                         return (ENOMEM);
343         
344                 /* Initialize state */
345                 mtx_init(&hpriv->mtx, "ng_pptp", NULL, MTX_DEF);
346                 ng_callout_init(&hpriv->sackTimer);
347                 ng_callout_init(&hpriv->rackTimer);
348                 hpriv->conf.cid = cid;
349                 hpriv->node = node;
350                 hpriv->hook = hook;
351                 NG_HOOK_SET_PRIVATE(hook, hpriv);
352
353                 hash = SESSHASH(cid);
354                 LIST_INSERT_HEAD(&priv->sesshash[hash], hpriv, sessions);
355         }
356
357         return (0);
358 }
359
360 /*
361  * Receive a control message.
362  */
363 static int
364 ng_pptpgre_rcvmsg(node_p node, item_p item, hook_p lasthook)
365 {
366         const priv_p priv = NG_NODE_PRIVATE(node);
367         struct ng_mesg *resp = NULL;
368         int error = 0;
369         struct ng_mesg *msg;
370
371         NGI_GET_MSG(item, msg);
372         switch (msg->header.typecookie) {
373         case NGM_PPTPGRE_COOKIE:
374                 switch (msg->header.cmd) {
375                 case NGM_PPTPGRE_SET_CONFIG:
376                     {
377                         struct ng_pptpgre_conf *const newConf =
378                                 (struct ng_pptpgre_conf *) msg->data;
379                         hpriv_p hpriv;
380                         uint16_t hash;
381
382                         /* Check for invalid or illegal config */
383                         if (msg->header.arglen != sizeof(*newConf))
384                                 ERROUT(EINVAL);
385                         /* Try to find session by cid. */
386                         hpriv = ng_pptpgre_find_session(priv, newConf->cid);
387                         /* If not present - use upper. */
388                         if (hpriv == NULL) {
389                                 hpriv = &priv->uppersess;
390                                 LIST_REMOVE(hpriv, sessions);
391                                 hash = SESSHASH(newConf->cid);
392                                 LIST_INSERT_HEAD(&priv->sesshash[hash], hpriv,
393                                     sessions);
394                         }
395                         ng_pptpgre_reset(hpriv);        /* reset on configure */
396                         hpriv->conf = *newConf;
397                         break;
398                     }
399                 case NGM_PPTPGRE_GET_CONFIG:
400                     {
401                         hpriv_p hpriv;
402
403                         if (msg->header.arglen == 2) {
404                                 /* Try to find session by cid. */
405                                 hpriv = ng_pptpgre_find_session(priv,
406                                     *((uint16_t *)msg->data));
407                                 if (hpriv == NULL)
408                                         ERROUT(EINVAL);
409                         } else if (msg->header.arglen == 0) {
410                                 /* Use upper. */
411                                 hpriv = &priv->uppersess;
412                         } else
413                                 ERROUT(EINVAL);
414                         NG_MKRESPONSE(resp, msg, sizeof(hpriv->conf), M_NOWAIT);
415                         if (resp == NULL)
416                                 ERROUT(ENOMEM);
417                         bcopy(&hpriv->conf, resp->data, sizeof(hpriv->conf));
418                         break;
419                     }
420                 case NGM_PPTPGRE_GET_STATS:
421                 case NGM_PPTPGRE_CLR_STATS:
422                 case NGM_PPTPGRE_GETCLR_STATS:
423                     {
424                         if (msg->header.cmd != NGM_PPTPGRE_CLR_STATS) {
425                                 NG_MKRESPONSE(resp, msg,
426                                     sizeof(priv->stats), M_NOWAIT);
427                                 if (resp == NULL)
428                                         ERROUT(ENOMEM);
429                                 bcopy(&priv->stats,
430                                     resp->data, sizeof(priv->stats));
431                         }
432                         if (msg->header.cmd != NGM_PPTPGRE_GET_STATS)
433                                 bzero(&priv->stats, sizeof(priv->stats));
434                         break;
435                     }
436                 default:
437                         error = EINVAL;
438                         break;
439                 }
440                 break;
441         default:
442                 error = EINVAL;
443                 break;
444         }
445 done:
446         NG_RESPOND_MSG(error, node, item, resp);
447         NG_FREE_MSG(msg);
448         return (error);
449 }
450
451 /*
452  * Receive incoming data on a hook.
453  */
454 static int
455 ng_pptpgre_rcvdata(hook_p hook, item_p item)
456 {
457         const hpriv_p hpriv = NG_HOOK_PRIVATE(hook);
458         int rval;
459
460         /* If not configured, reject */
461         if (!hpriv->conf.enabled) {
462                 NG_FREE_ITEM(item);
463                 return (ENXIO);
464         }
465
466         mtx_lock(&hpriv->mtx);
467
468         rval = ng_pptpgre_xmit(hpriv, item);
469
470         mtx_assert(&hpriv->mtx, MA_NOTOWNED);
471
472         return (rval);
473 }
474
475 /*
476  * Hook disconnection
477  */
478 static int
479 ng_pptpgre_disconnect(hook_p hook)
480 {
481         const node_p node = NG_HOOK_NODE(hook);
482         const priv_p priv = NG_NODE_PRIVATE(node);
483         const hpriv_p hpriv = NG_HOOK_PRIVATE(hook);
484
485         /* Zero out hook pointer */
486         if (hook == priv->upper) {
487                 priv->upper = NULL;
488                 priv->uppersess.hook = NULL;
489         } else if (hook == priv->lower) {
490                 priv->lower = NULL;
491         } else {
492                 /* Reset node (stops timers) */
493                 ng_pptpgre_reset(hpriv);
494
495                 LIST_REMOVE(hpriv, sessions);
496                 mtx_destroy(&hpriv->mtx);
497                 free(hpriv, M_NETGRAPH);
498         }
499
500         /* Go away if no longer connected to anything */
501         if ((NG_NODE_NUMHOOKS(node) == 0)
502         && (NG_NODE_IS_VALID(node)))
503                 ng_rmnode_self(node);
504         return (0);
505 }
506
507 /*
508  * Destroy node
509  */
510 static int
511 ng_pptpgre_shutdown(node_p node)
512 {
513         const priv_p priv = NG_NODE_PRIVATE(node);
514
515         /* Reset node (stops timers) */
516         ng_pptpgre_reset(&priv->uppersess);
517
518         LIST_REMOVE(&priv->uppersess, sessions);
519         mtx_destroy(&priv->uppersess.mtx);
520
521         free(priv, M_NETGRAPH);
522
523         /* Decrement ref count */
524         NG_NODE_UNREF(node);
525         return (0);
526 }
527
528 /*************************************************************************
529                     TRANSMIT AND RECEIVE FUNCTIONS
530 *************************************************************************/
531
532 /*
533  * Transmit an outgoing frame, or just an ack if m is NULL.
534  */
535 static int
536 ng_pptpgre_xmit(hpriv_p hpriv, item_p item)
537 {
538         const priv_p priv = NG_NODE_PRIVATE(hpriv->node);
539         u_char buf[sizeof(struct greheader) + 2 * sizeof(u_int32_t)];
540         struct greheader *const gre = (struct greheader *)buf;
541         int grelen, error;
542         struct mbuf *m;
543
544         mtx_assert(&hpriv->mtx, MA_OWNED);
545
546         if (item) {
547                 NGI_GET_M(item, m);
548         } else {
549                 m = NULL;
550         }
551         /* Check if there's data */
552         if (m != NULL) {
553
554                 /* Check if windowing is enabled */
555                 if (hpriv->conf.enableWindowing) {
556                         /* Is our transmit window full? */
557                         if ((u_int32_t)PPTP_SEQ_DIFF(hpriv->xmitSeq,
558                             hpriv->recvAck) >= hpriv->xmitWin) {
559                                 priv->stats.xmitDrops++;
560                                 ERROUT(ENOBUFS);
561                         }
562                 }
563
564                 /* Sanity check frame length */
565                 if (m->m_pkthdr.len > PPTP_MAX_PAYLOAD) {
566                         priv->stats.xmitTooBig++;
567                         ERROUT(EMSGSIZE);
568                 }
569         } else {
570                 priv->stats.xmitLoneAcks++;
571         }
572
573         /* Build GRE header */
574         be32enc(gre, PPTP_INIT_VALUE);
575         be16enc(&gre->length, (m != NULL) ? m->m_pkthdr.len : 0);
576         be16enc(&gre->cid, hpriv->conf.peerCid);
577
578         /* Include sequence number if packet contains any data */
579         if (m != NULL) {
580                 gre->hasSeq = 1;
581                 if (hpriv->conf.enableWindowing) {
582                         hpriv->timeSent[hpriv->xmitSeq - hpriv->recvAck]
583                             = ng_pptpgre_time();
584                 }
585                 hpriv->xmitSeq++;
586                 be32enc(&gre->data[0], hpriv->xmitSeq);
587         }
588
589         /* Include acknowledgement (and stop send ack timer) if needed */
590         if (hpriv->conf.enableAlwaysAck || hpriv->xmitAck != hpriv->recvSeq) {
591                 gre->hasAck = 1;
592                 be32enc(&gre->data[gre->hasSeq], hpriv->recvSeq);
593                 hpriv->xmitAck = hpriv->recvSeq;
594                 if (hpriv->conf.enableDelayedAck)
595                         ng_uncallout(&hpriv->sackTimer, hpriv->node);
596         }
597
598         /* Prepend GRE header to outgoing frame */
599         grelen = sizeof(*gre) + sizeof(u_int32_t) * (gre->hasSeq + gre->hasAck);
600         if (m == NULL) {
601                 MGETHDR(m, M_NOWAIT, MT_DATA);
602                 if (m == NULL) {
603                         priv->stats.memoryFailures++;
604                         ERROUT(ENOBUFS);
605                 }
606                 m->m_len = m->m_pkthdr.len = grelen;
607                 m->m_pkthdr.rcvif = NULL;
608         } else {
609                 M_PREPEND(m, grelen, M_NOWAIT);
610                 if (m == NULL || (m->m_len < grelen
611                     && (m = m_pullup(m, grelen)) == NULL)) {
612                         priv->stats.memoryFailures++;
613                         ERROUT(ENOBUFS);
614                 }
615         }
616         bcopy(gre, mtod(m, u_char *), grelen);
617
618         /* Update stats */
619         priv->stats.xmitPackets++;
620         priv->stats.xmitOctets += m->m_pkthdr.len;
621
622         /*
623          * XXX: we should reset timer only after an item has been sent
624          * successfully.
625          */
626         if (hpriv->conf.enableWindowing &&
627             gre->hasSeq && hpriv->xmitSeq == hpriv->recvAck + 1)
628                 ng_pptpgre_start_recv_ack_timer(hpriv);
629
630         mtx_unlock(&hpriv->mtx);
631
632         /* Deliver packet */
633         if (item) {
634                 NG_FWD_NEW_DATA(error, item, priv->lower, m);
635         } else {
636                 NG_SEND_DATA_ONLY(error, priv->lower, m);
637         }
638
639         return (error);
640
641 done:
642         mtx_unlock(&hpriv->mtx);
643         NG_FREE_M(m);
644         if (item)
645                 NG_FREE_ITEM(item);
646         return (error);
647 }
648
649 /*
650  * Handle an incoming packet.  The packet includes the IP header.
651  */
652 static int
653 ng_pptpgre_rcvdata_lower(hook_p hook, item_p item)
654 {
655         hpriv_p hpriv;
656         node_p node = NG_HOOK_NODE(hook);
657         const priv_p priv = NG_NODE_PRIVATE(node);
658         int iphlen, grelen, extralen;
659         const struct greheader *gre;
660         const struct ip *ip;
661         int error = 0;
662         struct mbuf *m;
663
664         NGI_GET_M(item, m);
665         /* Update stats */
666         priv->stats.recvPackets++;
667         priv->stats.recvOctets += m->m_pkthdr.len;
668
669         /* Sanity check packet length */
670         if (m->m_pkthdr.len < sizeof(*ip) + sizeof(*gre)) {
671                 priv->stats.recvRunts++;
672                 ERROUT(EINVAL);
673         }
674
675         /* Safely pull up the complete IP+GRE headers */
676         if (m->m_len < sizeof(*ip) + sizeof(*gre)
677             && (m = m_pullup(m, sizeof(*ip) + sizeof(*gre))) == NULL) {
678                 priv->stats.memoryFailures++;
679                 ERROUT(ENOBUFS);
680         }
681         ip = mtod(m, const struct ip *);
682         iphlen = ip->ip_hl << 2;
683         if (m->m_len < iphlen + sizeof(*gre)) {
684                 if ((m = m_pullup(m, iphlen + sizeof(*gre))) == NULL) {
685                         priv->stats.memoryFailures++;
686                         ERROUT(ENOBUFS);
687                 }
688                 ip = mtod(m, const struct ip *);
689         }
690         gre = (const struct greheader *)((const u_char *)ip + iphlen);
691         grelen = sizeof(*gre) + sizeof(u_int32_t) * (gre->hasSeq + gre->hasAck);
692         if (m->m_pkthdr.len < iphlen + grelen) {
693                 priv->stats.recvRunts++;
694                 ERROUT(EINVAL);
695         }
696         if (m->m_len < iphlen + grelen) {
697                 if ((m = m_pullup(m, iphlen + grelen)) == NULL) {
698                         priv->stats.memoryFailures++;
699                         ERROUT(ENOBUFS);
700                 }
701                 ip = mtod(m, const struct ip *);
702                 gre = (const struct greheader *)((const u_char *)ip + iphlen);
703         }
704
705         /* Sanity check packet length and GRE header bits */
706         extralen = m->m_pkthdr.len
707             - (iphlen + grelen + gre->hasSeq * be16dec(&gre->length));
708         if (extralen < 0) {
709                 priv->stats.recvBadGRE++;
710                 ERROUT(EINVAL);
711         }
712         if ((be32dec(gre) & PPTP_INIT_MASK) != PPTP_INIT_VALUE) {
713                 priv->stats.recvBadGRE++;
714                 ERROUT(EINVAL);
715         }
716
717         hpriv = ng_pptpgre_find_session(priv, be16dec(&gre->cid));
718         if (hpriv == NULL || hpriv->hook == NULL || !hpriv->conf.enabled) {
719                 priv->stats.recvBadCID++;
720                 ERROUT(EINVAL);
721         }
722         mtx_lock(&hpriv->mtx);
723
724         /* Look for peer ack */
725         if (gre->hasAck) {
726                 const u_int32_t ack = be32dec(&gre->data[gre->hasSeq]);
727                 const int index = ack - hpriv->recvAck - 1;
728                 long sample;
729                 long diff;
730
731                 /* Sanity check ack value */
732                 if (PPTP_SEQ_DIFF(ack, hpriv->xmitSeq) > 0) {
733                         priv->stats.recvBadAcks++;
734                         goto badAck;            /* we never sent it! */
735                 }
736                 if (PPTP_SEQ_DIFF(ack, hpriv->recvAck) <= 0)
737                         goto badAck;            /* ack already timed out */
738                 hpriv->recvAck = ack;
739
740                 /* Update adaptive timeout stuff */
741                 if (hpriv->conf.enableWindowing) {
742                         sample = ng_pptpgre_time() - hpriv->timeSent[index];
743                         diff = sample - hpriv->rtt;
744                         hpriv->rtt += PPTP_ACK_ALPHA(diff);
745                         if (diff < 0)
746                                 diff = -diff;
747                         hpriv->dev += PPTP_ACK_BETA(diff - hpriv->dev);
748                             /* +2 to compensate low precision of int math */
749                         hpriv->ato = hpriv->rtt + PPTP_ACK_CHI(hpriv->dev + 2);
750                         if (hpriv->ato > PPTP_MAX_TIMEOUT)
751                                 hpriv->ato = PPTP_MAX_TIMEOUT;
752                         else if (hpriv->ato < PPTP_MIN_TIMEOUT)
753                                 hpriv->ato = PPTP_MIN_TIMEOUT;
754
755                         /* Shift packet transmit times in our transmit window */
756                         bcopy(hpriv->timeSent + index + 1, hpriv->timeSent,
757                             sizeof(*hpriv->timeSent)
758                               * (PPTP_XMIT_WIN - (index + 1)));
759
760                         /* If we sent an entire window, increase window size */
761                         if (PPTP_SEQ_DIFF(ack, hpriv->winAck) >= 0
762                             && hpriv->xmitWin < PPTP_XMIT_WIN) {
763                                 hpriv->xmitWin++;
764                                 hpriv->winAck = ack + hpriv->xmitWin;
765                         }
766
767                         /* Stop/(re)start receive ACK timer as necessary */
768                         ng_uncallout(&hpriv->rackTimer, hpriv->node);
769                         if (hpriv->recvAck != hpriv->xmitSeq)
770                                 ng_pptpgre_start_recv_ack_timer(hpriv);
771                 }
772         }
773 badAck:
774
775         /* See if frame contains any data */
776         if (gre->hasSeq) {
777                 const u_int32_t seq = be32dec(&gre->data[0]);
778
779                 /* Sanity check sequence number */
780                 if (PPTP_SEQ_DIFF(seq, hpriv->recvSeq) <= 0) {
781                         if (seq == hpriv->recvSeq)
782                                 priv->stats.recvDuplicates++;
783                         else
784                                 priv->stats.recvOutOfOrder++;
785                         mtx_unlock(&hpriv->mtx);
786                         ERROUT(EINVAL);
787                 }
788                 hpriv->recvSeq = seq;
789
790                 /* We need to acknowledge this packet; do it soon... */
791                 if (!(callout_pending(&hpriv->sackTimer))) {
792                         /* If delayed ACK is disabled, send it now */
793                         if (!hpriv->conf.enableDelayedAck) {    /* ack now */
794                                 ng_pptpgre_xmit(hpriv, NULL);
795                                 /* ng_pptpgre_xmit() drops the mutex */
796                         } else {                                /* ack later */
797                                 ng_pptpgre_start_send_ack_timer(hpriv);
798                                 mtx_unlock(&hpriv->mtx);
799                         }
800                 } else
801                         mtx_unlock(&hpriv->mtx);
802
803                 /* Trim mbuf down to internal payload */
804                 m_adj(m, iphlen + grelen);
805                 if (extralen > 0)
806                         m_adj(m, -extralen);
807
808                 mtx_assert(&hpriv->mtx, MA_NOTOWNED);
809
810                 /* Deliver frame to upper layers */
811                 NG_FWD_NEW_DATA(error, item, hpriv->hook, m);
812         } else {
813                 priv->stats.recvLoneAcks++;
814                 mtx_unlock(&hpriv->mtx);
815                 NG_FREE_ITEM(item);
816                 NG_FREE_M(m);           /* no data to deliver */
817         }
818
819         return (error);
820
821 done:
822         NG_FREE_ITEM(item);
823         NG_FREE_M(m);
824         return (error);
825 }
826
827 /*************************************************************************
828                     TIMER RELATED FUNCTIONS
829 *************************************************************************/
830
831 /*
832  * Start a timer for the peer's acknowledging our oldest unacknowledged
833  * sequence number.  If we get an ack for this sequence number before
834  * the timer goes off, we cancel the timer.  Resets currently running
835  * recv ack timer, if any.
836  */
837 static void
838 ng_pptpgre_start_recv_ack_timer(hpriv_p hpriv)
839 {
840         int remain, ticks;
841
842         /* Compute how long until oldest unack'd packet times out,
843            and reset the timer to that time. */
844         remain = (hpriv->timeSent[0] + hpriv->ato) - ng_pptpgre_time();
845         if (remain < 0)
846                 remain = 0;
847
848         /* Be conservative: timeout can happen up to 1 tick early */
849         ticks = (((remain * hz) + PPTP_TIME_SCALE - 1) / PPTP_TIME_SCALE) + 1;
850         ng_callout(&hpriv->rackTimer, hpriv->node, hpriv->hook,
851             ticks, ng_pptpgre_recv_ack_timeout, hpriv, 0);
852 }
853
854 /*
855  * The peer has failed to acknowledge the oldest unacknowledged sequence
856  * number within the time allotted.  Update our adaptive timeout parameters
857  * and reset/restart the recv ack timer.
858  */
859 static void
860 ng_pptpgre_recv_ack_timeout(node_p node, hook_p hook, void *arg1, int arg2)
861 {
862         const priv_p priv = NG_NODE_PRIVATE(node);
863         const hpriv_p hpriv = arg1;
864
865         /* Update adaptive timeout stuff */
866         priv->stats.recvAckTimeouts++;
867         hpriv->rtt = PPTP_ACK_DELTA(hpriv->rtt) + 1; /* +1 to avoid delta*0 case */
868         hpriv->ato = hpriv->rtt + PPTP_ACK_CHI(hpriv->dev);
869         if (hpriv->ato > PPTP_MAX_TIMEOUT)
870                 hpriv->ato = PPTP_MAX_TIMEOUT;
871         else if (hpriv->ato < PPTP_MIN_TIMEOUT)
872                 hpriv->ato = PPTP_MIN_TIMEOUT;
873
874         /* Reset ack and sliding window */
875         hpriv->recvAck = hpriv->xmitSeq;                /* pretend we got the ack */
876         hpriv->xmitWin = (hpriv->xmitWin + 1) / 2;      /* shrink transmit window */
877         hpriv->winAck = hpriv->recvAck + hpriv->xmitWin;        /* reset win expand time */
878 }
879
880 /*
881  * Start the send ack timer. This assumes the timer is not
882  * already running.
883  */
884 static void
885 ng_pptpgre_start_send_ack_timer(hpriv_p hpriv)
886 {
887         int ackTimeout, ticks;
888
889         /* Take 1/4 of the estimated round trip time */
890         ackTimeout = (hpriv->rtt >> 2);
891         if (ackTimeout < PPTP_MIN_ACK_DELAY)
892                 ackTimeout = PPTP_MIN_ACK_DELAY;
893         else if (ackTimeout > PPTP_MAX_ACK_DELAY)
894                 ackTimeout = PPTP_MAX_ACK_DELAY;
895
896         /* Be conservative: timeout can happen up to 1 tick early */
897         ticks = (((ackTimeout * hz) + PPTP_TIME_SCALE - 1) / PPTP_TIME_SCALE);
898         ng_callout(&hpriv->sackTimer, hpriv->node, hpriv->hook,
899             ticks, ng_pptpgre_send_ack_timeout, hpriv, 0);
900 }
901
902 /*
903  * We've waited as long as we're willing to wait before sending an
904  * acknowledgement to the peer for received frames. We had hoped to
905  * be able to piggy back our acknowledgement on an outgoing data frame,
906  * but apparently there haven't been any since. So send the ack now.
907  */
908 static void
909 ng_pptpgre_send_ack_timeout(node_p node, hook_p hook, void *arg1, int arg2)
910 {
911         const hpriv_p hpriv = arg1;
912
913         mtx_lock(&hpriv->mtx);
914         /* Send a frame with an ack but no payload */
915         ng_pptpgre_xmit(hpriv, NULL);
916         mtx_assert(&hpriv->mtx, MA_NOTOWNED);
917 }
918
919 /*************************************************************************
920                     MISC FUNCTIONS
921 *************************************************************************/
922
923 /*
924  * Find the hook with a given session ID.
925  */
926 static hpriv_p
927 ng_pptpgre_find_session(priv_p privp, u_int16_t cid)
928 {
929         uint16_t        hash = SESSHASH(cid);
930         hpriv_p hpriv = NULL;
931
932         LIST_FOREACH(hpriv, &privp->sesshash[hash], sessions) {
933                 if (hpriv->conf.cid == cid)
934                         break;
935         }
936
937         return (hpriv);
938 }
939
940 /*
941  * Reset state (must be called with lock held or from writer)
942  */
943 static void
944 ng_pptpgre_reset(hpriv_p hpriv)
945 {
946         /* Reset adaptive timeout state */
947         hpriv->ato = PPTP_MAX_TIMEOUT;
948         hpriv->rtt = PPTP_TIME_SCALE / 10;
949         if (hpriv->conf.peerPpd > 1)    /* ppd = 0 treat as = 1 */
950                 hpriv->rtt *= hpriv->conf.peerPpd;
951         hpriv->dev = 0;
952         hpriv->xmitWin = (hpriv->conf.recvWin + 1) / 2;
953         if (hpriv->xmitWin < 2)         /* often the first packet is lost */
954                 hpriv->xmitWin = 2;             /*   because the peer isn't ready */
955         else if (hpriv->xmitWin > PPTP_XMIT_WIN)
956                 hpriv->xmitWin = PPTP_XMIT_WIN;
957         hpriv->winAck = hpriv->xmitWin;
958
959         /* Reset sequence numbers */
960         hpriv->recvSeq = ~0;
961         hpriv->recvAck = ~0;
962         hpriv->xmitSeq = ~0;
963         hpriv->xmitAck = ~0;
964
965         /* Stop timers */
966         ng_uncallout(&hpriv->sackTimer, hpriv->node);
967         ng_uncallout(&hpriv->rackTimer, hpriv->node);
968 }
969
970 /*
971  * Return the current time scaled & translated to our internally used format.
972  */
973 static pptptime_t
974 ng_pptpgre_time(void)
975 {
976         struct timeval tv;
977         pptptime_t t;
978
979         microuptime(&tv);
980         t = (pptptime_t)tv.tv_sec * PPTP_TIME_SCALE;
981         t += tv.tv_usec / (1000000 / PPTP_TIME_SCALE);
982         return(t);
983 }