On 2021/11/13 18:04, Alexander Bluhm wrote:
> Hi,
> 
> To make IPsec MP safe we need refcounting for the tdb.  The diff
> below is part of something bigger we have at genua.  Although it
> does not cover timeouts and the tdb reaper yet, I want to get this
> in as a frist step.
> 
> It passes regress but there are setups that are not covered.  Bridge
> and pfsync with IPsec and TCP signature need special care.
> 
> When testing, please check for tdb leaks.  The vmstat -m tdb in use
> culumn must be 0.  It it is not, try ipsecctl -F.  If ipsecctl -sa
> shows nothing, but vmstat -m shows a used tdb, then it is a bug.
> 
> Name        Size Requests Fail    InUse Pgreq Pgrel Npage Hiwat Minpg Maxpg 
> Idle
> tdb          904       56    0        0     7     7     0     7     0     8   
>  0

Tested with a box running isakmpd at home. Config is:

isakmpd_flags=-Kv -D0=29 -D1=49 -D2=10 -D3=30 -D5=20 -D6=30 -D8=30 -D9=30 
-D10=20

ike passive esp \
        from xxx.xx.xxx.1 (xxx.xx.xxx.0/27) to 0.0.0.0/0 \
        local xxx.xx.xxx.1 \
        main auth hmac-sha2-256 enc aes group ecp256 \
        quick enc aes-128-gcm group ecp256 \
        srcid xxx

isakmpd starts at boot, 3 devices connect in and tunnels come up,
FLOWS and SAD show in ipsecctl -sa.

isakmpd crashes shortly after starting (as described in previous mails;
this is not new, it happens very often on the first start at boot),
FLOWS and SAD remain showing. At this point they still show InUse:

Name        Size Requests Fail    InUse Pgreq Pgrel Npage Hiwat Minpg Maxpg Idle
tdb         1088        9    0        6     1     0     1     1     0     8    0

After ipsecctl -F they are freed:

tdb         1088        9    0        0     1     0     1     1     0     8    1

Restart isakmpd, ipsecctl -f /etc/ipsec.conf, devices connect back in:

tdb         1088       18    0        6     2     1     1     1     0     8    0

(And at this point isakmpd is usually stable).

I will leave a shell loop running vmstat -m every few minutes to check
if it grows.

(Bad timing for testing TCPMD5 because over the last couple of days I have
just removed bgpd+ospfd from the machines that would have been good
choices to run test diffs..I might be able to get some simple setup
back up and running though)

> 
> ok?
> 
> bluhm
> 
> Index: net/if_bridge.c
> ===================================================================
> RCS file: /data/mirror/openbsd/cvs/src/sys/net/if_bridge.c,v
> retrieving revision 1.358
> diff -u -p -r1.358 if_bridge.c
> --- net/if_bridge.c   11 Nov 2021 18:08:17 -0000      1.358
> +++ net/if_bridge.c   11 Nov 2021 23:24:31 -0000
> @@ -1577,10 +1577,12 @@ bridge_ipsec(struct ifnet *ifp, struct e
>  
>                       prot = (*(tdb->tdb_xform->xf_input))(&m, tdb, hlen,
>                           off);
> +                     tdb_unref(tdb);
>                       if (prot != IPPROTO_DONE)
>                               ip_deliver(&m, &hlen, prot, af);
>                       return (1);
>               } else {
> +                     tdb_unref(tdb);
>   skiplookup:
>                       /* XXX do an input policy lookup */
>                       return (0);
> Index: net/if_pfsync.c
> ===================================================================
> RCS file: /data/mirror/openbsd/cvs/src/sys/net/if_pfsync.c,v
> retrieving revision 1.298
> diff -u -p -r1.298 if_pfsync.c
> --- net/if_pfsync.c   11 Nov 2021 12:35:01 -0000      1.298
> +++ net/if_pfsync.c   11 Nov 2021 23:24:31 -0000
> @@ -1325,11 +1325,13 @@ pfsync_update_net_tdb(struct pfsync_tdb 
>               /* Neither replay nor byte counter should ever decrease. */
>               if (pt->rpl < tdb->tdb_rpl ||
>                   pt->cur_bytes < tdb->tdb_cur_bytes) {
> +                     tdb_unref(tdb);
>                       goto bad;
>               }
>  
>               tdb->tdb_rpl = pt->rpl;
>               tdb->tdb_cur_bytes = pt->cur_bytes;
> +             tdb_unref(tdb);
>       }
>       return;
>  
> Index: net/pfkeyv2.c
> ===================================================================
> RCS file: /data/mirror/openbsd/cvs/src/sys/net/pfkeyv2.c,v
> retrieving revision 1.221
> diff -u -p -r1.221 pfkeyv2.c
> --- net/pfkeyv2.c     25 Oct 2021 18:25:01 -0000      1.221
> +++ net/pfkeyv2.c     12 Nov 2021 21:03:45 -0000
> @@ -1044,7 +1044,8 @@ pfkeyv2_sa_flush(struct tdb *tdb, void *
>       if (!(*((u_int8_t *) satype_vp)) ||
>           tdb->tdb_satype == *((u_int8_t *) satype_vp)) {
>               tdb_unlink_locked(tdb);
> -             tdb_free(tdb);
> +             tdb_unbundle(tdb);
> +             tdb_unref(tdb);
>       }
>       return (0);
>  }
> @@ -1327,7 +1328,7 @@ pfkeyv2_send(struct socket *so, void *me
>  
>                       if ((rval = pfkeyv2_get_proto_alg(newsa->tdb_satype,
>                           &newsa->tdb_sproto, &alg))) {
> -                             tdb_free(freeme);
> +                             tdb_unref(freeme);
>                               freeme = NULL;
>                               NET_UNLOCK();
>                               goto ret;
> @@ -1363,7 +1364,7 @@ pfkeyv2_send(struct socket *so, void *me
>                           headers[SADB_X_EXT_DST_MASK],
>                           headers[SADB_X_EXT_PROTOCOL],
>                           headers[SADB_X_EXT_FLOW_TYPE]))) {
> -                             tdb_free(freeme);
> +                             tdb_unref(freeme);
>                               freeme = NULL;
>                               NET_UNLOCK();
>                               goto ret;
> @@ -1386,7 +1387,7 @@ pfkeyv2_send(struct socket *so, void *me
>                       rval = tdb_init(newsa, alg, &ii);
>                       if (rval) {
>                               rval = EINVAL;
> -                             tdb_free(freeme);
> +                             tdb_unref(freeme);
>                               freeme = NULL;
>                               NET_UNLOCK();
>                               goto ret;
> @@ -1397,7 +1398,7 @@ pfkeyv2_send(struct socket *so, void *me
>                       /* Delete old version of the SA, insert new one */
>                       tdb_delete(sa2);
>                       puttdb((struct tdb *) freeme);
> -                     sa2 = freeme = NULL;
> +                     freeme = NULL;
>               } else {
>                       /*
>                        * The SA is already initialized, so we're only allowed 
> to
> @@ -1503,7 +1504,7 @@ pfkeyv2_send(struct socket *so, void *me
>                       newsa->tdb_satype = smsg->sadb_msg_satype;
>                       if ((rval = pfkeyv2_get_proto_alg(newsa->tdb_satype,
>                           &newsa->tdb_sproto, &alg))) {
> -                             tdb_free(freeme);
> +                             tdb_unref(freeme);
>                               freeme = NULL;
>                               NET_UNLOCK();
>                               goto ret;
> @@ -1541,7 +1542,7 @@ pfkeyv2_send(struct socket *so, void *me
>                           headers[SADB_X_EXT_DST_MASK],
>                           headers[SADB_X_EXT_PROTOCOL],
>                           headers[SADB_X_EXT_FLOW_TYPE]))) {
> -                             tdb_free(freeme);
> +                             tdb_unref(freeme);
>                               freeme = NULL;
>                               NET_UNLOCK();
>                               goto ret;
> @@ -1564,7 +1565,7 @@ pfkeyv2_send(struct socket *so, void *me
>                       rval = tdb_init(newsa, alg, &ii);
>                       if (rval) {
>                               rval = EINVAL;
> -                             tdb_free(freeme);
> +                             tdb_unref(freeme);
>                               freeme = NULL;
>                               NET_UNLOCK();
>                               goto ret;
> @@ -1596,7 +1597,6 @@ pfkeyv2_send(struct socket *so, void *me
>               tdb_delete(sa2);
>               NET_UNLOCK();
>  
> -             sa2 = NULL;
>               break;
>  
>       case SADB_X_ASKPOLICY:
> @@ -1786,6 +1786,7 @@ pfkeyv2_send(struct socket *so, void *me
>                   ssa->sadb_sa_spi, sunionp,
>                   SADB_X_GETSPROTO(sa_proto->sadb_protocol_proto));
>               if (tdb2 == NULL) {
> +                     tdb_unref(tdb1);
>                       rval = ESRCH;
>                       NET_UNLOCK();
>                       goto ret;
> @@ -1794,6 +1795,8 @@ pfkeyv2_send(struct socket *so, void *me
>               /* Detect cycles */
>               for (tdb3 = tdb2; tdb3; tdb3 = tdb3->tdb_onext)
>                       if (tdb3 == tdb1) {
> +                             tdb_unref(tdb1);
> +                             tdb_unref(tdb2);
>                               rval = ESRCH;
>                               NET_UNLOCK();
>                               goto ret;
> @@ -1801,12 +1804,16 @@ pfkeyv2_send(struct socket *so, void *me
>  
>               /* Maintenance */
>               if ((tdb1->tdb_onext) &&
> -                 (tdb1->tdb_onext->tdb_inext == tdb1))
> +                 (tdb1->tdb_onext->tdb_inext == tdb1)) {
> +                     tdb_unref(tdb1->tdb_onext->tdb_inext);
>                       tdb1->tdb_onext->tdb_inext = NULL;
> +             }
>  
>               if ((tdb2->tdb_inext) &&
> -                 (tdb2->tdb_inext->tdb_onext == tdb2))
> +                 (tdb2->tdb_inext->tdb_onext == tdb2)) {
> +                     tdb_unref(tdb2->tdb_inext->tdb_onext);
>                       tdb2->tdb_inext->tdb_onext = NULL;
> +             }
>  
>               /* Link them */
>               tdb1->tdb_onext = tdb2;
> @@ -2008,10 +2015,12 @@ pfkeyv2_send(struct socket *so, void *me
>                               (caddr_t)&ipo->ipo_mask, rnh,
>                               ipo->ipo_nodes, 0)) == NULL) {
>                               /* Remove from linked list of policies on TDB */
> -                             if (ipo->ipo_tdb)
> -                                     
> TAILQ_REMOVE(&ipo->ipo_tdb->tdb_policy_head,
> +                             if (ipo->ipo_tdb != NULL) {
> +                                     TAILQ_REMOVE(
> +                                         &ipo->ipo_tdb->tdb_policy_head,
>                                           ipo, ipo_tdb_next);
> -
> +                                     tdb_unref(ipo->ipo_tdb);
> +                             }
>                               if (ipo->ipo_ids)
>                                       ipsp_ids_free(ipo->ipo_ids);
>                               pool_put(&ipsec_policy_pool, ipo);
> @@ -2127,6 +2136,7 @@ realret:
>       free(message, M_PFKEY, len);
>  
>       free(sa1, M_PFKEY, sizeof(*sa1));
> +     tdb_unref(sa2);
>  
>       return (rval);
>  }
> Index: netinet/ip_ipsp.c
> ===================================================================
> RCS file: /data/mirror/openbsd/cvs/src/sys/netinet/ip_ipsp.c,v
> retrieving revision 1.249
> diff -u -p -r1.249 ip_ipsp.c
> --- netinet/ip_ipsp.c 27 Oct 2021 16:58:44 -0000      1.249
> +++ netinet/ip_ipsp.c 13 Nov 2021 15:59:39 -0000
> @@ -297,9 +297,10 @@ reserve_spi(u_int rdomain, u_int32_t ssp
>  
>               /* Check whether we're using this SPI already. */
>               exists = gettdb(rdomain, spi, dst, sproto);
> -             if (exists)
> +             if (exists != NULL) {
> +                     tdb_unref(exists);
>                       continue;
> -
> +             }
>  
>               tdbp->tdb_spi = spi;
>               memcpy(&tdbp->tdb_dst.sa, &dst->sa, dst->sa.sa_len);
> @@ -351,6 +352,7 @@ gettdb_dir(u_int rdomain, u_int32_t spi,
>                   !memcmp(&tdbp->tdb_dst, dst, dst->sa.sa_len))
>                       break;
>  
> +     tdb_ref(tdbp);
>       mtx_leave(&tdb_sadb_mtx);
>       return tdbp;
>  }
> @@ -383,6 +385,7 @@ gettdbbysrcdst_dir(u_int rdomain, u_int3
>                       break;
>  
>       if (tdbp != NULL) {
> +             tdb_ref(tdbp);
>               mtx_leave(&tdb_sadb_mtx);
>               return tdbp;
>       }
> @@ -402,6 +405,7 @@ gettdbbysrcdst_dir(u_int rdomain, u_int3
>                   tdbp->tdb_src.sa.sa_family == AF_UNSPEC)
>                       break;
>  
> +     tdb_ref(tdbp);
>       mtx_leave(&tdb_sadb_mtx);
>       return tdbp;
>  }
> @@ -469,6 +473,7 @@ gettdbbydst(u_int rdomain, union sockadd
>                       break;
>               }
>  
> +     tdb_ref(tdbp);
>       mtx_leave(&tdb_sadb_mtx);
>       return tdbp;
>  }
> @@ -499,6 +504,7 @@ gettdbbysrc(u_int rdomain, union sockadd
>                       break;
>               }
>  
> +     tdb_ref(tdbp);
>       mtx_leave(&tdb_sadb_mtx);
>       return tdbp;
>  }
> @@ -811,12 +817,55 @@ tdb_unlink_locked(struct tdb *tdbp)
>  }
>  
>  void
> +tdb_unbundle(struct tdb *tdbp)
> +{
> +     if (tdbp->tdb_onext) {
> +             if (tdbp->tdb_onext->tdb_inext == tdbp) {
> +                     tdb_unref(tdbp);        /* to us */
> +                     tdbp->tdb_onext->tdb_inext = NULL;
> +             }
> +             tdb_unref(tdbp->tdb_onext);     /* to other */
> +             tdbp->tdb_onext = NULL;
> +     }
> +     if (tdbp->tdb_inext) {
> +             if (tdbp->tdb_inext->tdb_onext == tdbp) {
> +                     tdb_unref(tdbp);        /* to us */
> +                     tdbp->tdb_inext->tdb_onext = NULL;
> +             }
> +             tdb_unref(tdbp->tdb_inext);     /* to other */
> +             tdbp->tdb_inext = NULL;
> +     }
> +}
> +
> +struct tdb *
> +tdb_ref(struct tdb *tdb)
> +{
> +     if (tdb == NULL)
> +             return NULL;
> +     refcnt_take(&tdb->tdb_refcnt);
> +     return tdb;
> +}
> +
> +void
> +tdb_unref(struct tdb *tdb)
> +{
> +     if (tdb == NULL)
> +             return;
> +     if (refcnt_rele(&tdb->tdb_refcnt) == 0)
> +             return;
> +     tdb_free(tdb);
> +}
> +
> +void
>  tdb_delete(struct tdb *tdbp)
>  {
>       NET_ASSERT_LOCKED();
>  
>       tdb_unlink(tdbp);
> -     tdb_free(tdbp);
> +     /* release tdb_onext/tdb_inext references */
> +     tdb_unbundle(tdbp);
> +     /* release the reference for tdb_unlink() */
> +     tdb_unref(tdbp);
>  }
>  
>  /*
> @@ -831,6 +880,7 @@ tdb_alloc(u_int rdomain)
>  
>       tdbp = pool_get(&tdb_pool, PR_WAITOK | PR_ZERO);
>  
> +     refcnt_init(&tdbp->tdb_refcnt);
>       TAILQ_INIT(&tdbp->tdb_policy_head);
>  
>       /* Record establishment time. */
> @@ -867,9 +917,9 @@ tdb_free(struct tdb *tdbp)
>  #endif
>  
>       /* Cleanup SPD references. */
> -     for (ipo = TAILQ_FIRST(&tdbp->tdb_policy_head); ipo;
> -         ipo = TAILQ_FIRST(&tdbp->tdb_policy_head))  {
> +     while ((ipo = TAILQ_FIRST(&tdbp->tdb_policy_head)) != NULL) {
>               TAILQ_REMOVE(&tdbp->tdb_policy_head, ipo, ipo_tdb_next);
> +             tdb_unref(ipo->ipo_tdb);
>               ipo->ipo_tdb = NULL;
>               ipo->ipo_last_searched = 0; /* Force a re-search. */
>       }
> @@ -885,12 +935,6 @@ tdb_free(struct tdb *tdbp)
>               tdbp->tdb_tag = 0;
>       }
>  #endif
> -
> -     if ((tdbp->tdb_onext) && (tdbp->tdb_onext->tdb_inext == tdbp))
> -             tdbp->tdb_onext->tdb_inext = NULL;
> -
> -     if ((tdbp->tdb_inext) && (tdbp->tdb_inext->tdb_onext == tdbp))
> -             tdbp->tdb_inext->tdb_onext = NULL;
>  
>       /* Remove expiration timeouts. */
>       tdbp->tdb_flags &= ~(TDBF_FIRSTUSE | TDBF_SOFT_FIRSTUSE | TDBF_TIMER |
> Index: netinet/ip_ipsp.h
> ===================================================================
> RCS file: /data/mirror/openbsd/cvs/src/sys/netinet/ip_ipsp.h,v
> retrieving revision 1.219
> diff -u -p -r1.219 ip_ipsp.h
> --- netinet/ip_ipsp.h 25 Oct 2021 18:25:01 -0000      1.219
> +++ netinet/ip_ipsp.h 13 Nov 2021 15:59:19 -0000
> @@ -322,6 +322,8 @@ struct tdb {                              /* tunnel 
> descriptor blo
>       struct tdb      *tdb_inext;
>       struct tdb      *tdb_onext;
>  
> +     struct refcnt   tdb_refcnt;
> +
>       const struct xformsw    *tdb_xform;             /* Transform to use */
>       const struct enc_xform  *tdb_encalgxform;       /* Enc algorithm */
>       const struct auth_hash  *tdb_authalgxform;      /* Auth algorithm */
> @@ -555,10 +557,13 @@ struct  tdb *gettdbbysrcdst_dir(u_int, u_
>  void puttdb(struct tdb *);
>  void tdb_delete(struct tdb *);
>  struct       tdb *tdb_alloc(u_int);
> +struct       tdb *tdb_ref(struct tdb *);
> +void tdb_unref(struct tdb *);
>  void tdb_free(struct tdb *);
>  int  tdb_init(struct tdb *, u_int16_t, struct ipsecinit *);
>  void tdb_unlink(struct tdb *);
>  void tdb_unlink_locked(struct tdb *);
> +void tdb_unbundle(struct tdb *);
>  int  tdb_walk(u_int, int (*)(struct tdb *, void *, int), void *);
>  
>  /* XF_IP4 */
> Index: netinet/ip_spd.c
> ===================================================================
> RCS file: /data/mirror/openbsd/cvs/src/sys/netinet/ip_spd.c,v
> retrieving revision 1.104
> diff -u -p -r1.104 ip_spd.c
> --- netinet/ip_spd.c  8 Jul 2021 16:39:55 -0000       1.104
> +++ netinet/ip_spd.c  11 Nov 2021 23:24:31 -0000
> @@ -368,9 +368,11 @@ ipsp_spd_lookup(struct mbuf *m, int af, 
>       }
>  
>       /* Do we have a cached entry ? If so, check if it's still valid. */
> -     if ((ipo->ipo_tdb) && (ipo->ipo_tdb->tdb_flags & TDBF_INVALID)) {
> +     if (ipo->ipo_tdb != NULL &&
> +         (ipo->ipo_tdb->tdb_flags & TDBF_INVALID)) {
>               TAILQ_REMOVE(&ipo->ipo_tdb->tdb_policy_head, ipo,
>                   ipo_tdb_next);
> +             tdb_unref(ipo->ipo_tdb);
>               ipo->ipo_tdb = NULL;
>       }
>  
> @@ -398,7 +400,7 @@ ipsp_spd_lookup(struct mbuf *m, int af, 
>                       ids = ipsp_ids_lookup(ipsecflowinfo);
>  
>               /* Check that the cached TDB (if present), is appropriate. */
> -             if (ipo->ipo_tdb) {
> +             if (ipo->ipo_tdb != NULL) {
>                       if ((ipo->ipo_last_searched <= ipsec_last_added) ||
>                           (ipo->ipo_sproto != ipo->ipo_tdb->tdb_sproto) ||
>                           memcmp(dignore ? &sdst : &ipo->ipo_dst,
> @@ -420,6 +422,7 @@ ipsp_spd_lookup(struct mbuf *m, int af, 
>                       /* Cached TDB was not good. */
>                       TAILQ_REMOVE(&ipo->ipo_tdb->tdb_policy_head, ipo,
>                           ipo_tdb_next);
> +                     tdb_unref(ipo->ipo_tdb);
>                       ipo->ipo_tdb = NULL;
>                       ipo->ipo_last_searched = 0;
>               }
> @@ -439,14 +442,14 @@ ipsp_spd_lookup(struct mbuf *m, int af, 
>                               ipo->ipo_last_searched = getuptime();
>  
>                       /* Find an appropriate SA from the existing ones. */
> -                     ipo->ipo_tdb =
> -                         gettdbbydst(rdomain,
> -                             dignore ? &sdst : &ipo->ipo_dst,
> -                             ipo->ipo_sproto,
> -                             ids ? ids: ipo->ipo_ids,
> -                             &ipo->ipo_addr, &ipo->ipo_mask);
> -                     if (ipo->ipo_tdb) {
> -                             
> TAILQ_INSERT_TAIL(&ipo->ipo_tdb->tdb_policy_head,
> +                     ipo->ipo_tdb = gettdbbydst(rdomain,
> +                         dignore ? &sdst : &ipo->ipo_dst,
> +                         ipo->ipo_sproto, ids ? ids: ipo->ipo_ids,
> +                         &ipo->ipo_addr, &ipo->ipo_mask);
> +                     if (ipo->ipo_tdb != NULL) {
> +                             /* gettdbbydst() has already refcounted tdb */
> +                             TAILQ_INSERT_TAIL(
> +                                 &ipo->ipo_tdb->tdb_policy_head,
>                                   ipo, ipo_tdb_next);
>                               *error = 0;
>                               return ipsp_spd_inp(m, af, hlen, error,
> @@ -520,10 +523,12 @@ ipsp_spd_lookup(struct mbuf *m, int af, 
>                                       goto nomatchin;
>  
>                       /* Add it to the cache. */
> -                     if (ipo->ipo_tdb)
> +                     if (ipo->ipo_tdb != NULL) {
>                               TAILQ_REMOVE(&ipo->ipo_tdb->tdb_policy_head,
>                                   ipo, ipo_tdb_next);
> -                     ipo->ipo_tdb = tdbp;
> +                             tdb_unref(ipo->ipo_tdb);
> +                     }
> +                     ipo->ipo_tdb = tdb_ref(tdbp);
>                       TAILQ_INSERT_TAIL(&tdbp->tdb_policy_head, ipo,
>                           ipo_tdb_next);
>                       *error = 0;
> @@ -535,7 +540,7 @@ ipsp_spd_lookup(struct mbuf *m, int af, 
>               }
>  
>               /* Check whether cached entry applies. */
> -             if (ipo->ipo_tdb) {
> +             if (ipo->ipo_tdb != NULL) {
>                       /*
>                        * We only need to check that the correct
>                        * security protocol and security gateway are
> @@ -551,8 +556,9 @@ ipsp_spd_lookup(struct mbuf *m, int af, 
>                       /* Not applicable, unlink. */
>                       TAILQ_REMOVE(&ipo->ipo_tdb->tdb_policy_head, ipo,
>                           ipo_tdb_next);
> -                     ipo->ipo_last_searched = 0;
> +                     tdb_unref(ipo->ipo_tdb);
>                       ipo->ipo_tdb = NULL;
> +                     ipo->ipo_last_searched = 0;
>               }
>  
>               /* Find whether there exists an appropriate SA. */
> @@ -560,14 +566,16 @@ ipsp_spd_lookup(struct mbuf *m, int af, 
>                       if (dignore == 0)
>                               ipo->ipo_last_searched = getuptime();
>  
> -                     ipo->ipo_tdb =
> -                         gettdbbysrc(rdomain,
> -                             dignore ? &ssrc : &ipo->ipo_dst,
> -                             ipo->ipo_sproto, ipo->ipo_ids,
> -                             &ipo->ipo_addr, &ipo->ipo_mask);
> -                     if (ipo->ipo_tdb)
> -                             
> TAILQ_INSERT_TAIL(&ipo->ipo_tdb->tdb_policy_head,
> +                     ipo->ipo_tdb = gettdbbysrc(rdomain,
> +                         dignore ? &ssrc : &ipo->ipo_dst,
> +                         ipo->ipo_sproto, ipo->ipo_ids,
> +                         &ipo->ipo_addr, &ipo->ipo_mask);
> +                     if (ipo->ipo_tdb != NULL) {
> +                             /* gettdbbysrc() has already refcounted tdb */
> +                             TAILQ_INSERT_TAIL(
> +                                 &ipo->ipo_tdb->tdb_policy_head,
>                                   ipo, ipo_tdb_next);
> +                     }
>               }
>    skipinputsearch:
>  
> @@ -637,9 +645,12 @@ ipsec_delete_policy(struct ipsec_policy 
>           rn_delete(&ipo->ipo_addr, &ipo->ipo_mask, rnh, rn) == NULL)
>               return (ESRCH);
>  
> -     if (ipo->ipo_tdb != NULL)
> +     if (ipo->ipo_tdb != NULL) {
>               TAILQ_REMOVE(&ipo->ipo_tdb->tdb_policy_head, ipo,
>                   ipo_tdb_next);
> +             tdb_unref(ipo->ipo_tdb);
> +             ipo->ipo_tdb = NULL;
> +     }
>  
>       while ((ipa = TAILQ_FIRST(&ipo->ipo_acquires)) != NULL)
>               ipsp_delete_acquire(ipa);
> Index: netinet/ipsec_input.c
> ===================================================================
> RCS file: /data/mirror/openbsd/cvs/src/sys/netinet/ipsec_input.c,v
> retrieving revision 1.191
> diff -u -p -r1.191 ipsec_input.c
> --- netinet/ipsec_input.c     11 Nov 2021 18:08:18 -0000      1.191
> +++ netinet/ipsec_input.c     11 Nov 2021 23:24:31 -0000
> @@ -348,6 +348,7 @@ ipsec_common_input(struct mbuf **mp, int
>               ipsecstat_inc(ipsec_idrops);
>               tdbp->tdb_idrops++;
>       }
> +     tdb_unref(tdbp);
>       return prot;
>  
>   drop:
> @@ -355,6 +356,7 @@ ipsec_common_input(struct mbuf **mp, int
>       ipsecstat_inc(ipsec_idrops);
>       if (tdbp != NULL)
>               tdbp->tdb_idrops++;
> +     tdb_unref(tdbp);
>       return IPPROTO_DONE;
>  }
>  
> @@ -937,6 +939,7 @@ ipsec_common_ctlinput(u_int rdomain, int
>               tdbp = gettdb_rev(rdomain, spi, (union sockaddr_union *)&dst,
>                   proto);
>               ipsec_set_mtu(tdbp, mtu);
> +             tdb_unref(tdbp);
>       }
>  }
>  
> @@ -944,7 +947,7 @@ void
>  udpencap_ctlinput(int cmd, struct sockaddr *sa, u_int rdomain, void *v)
>  {
>       struct ip *ip = v;
> -     struct tdb *tdbp;
> +     struct tdb *tdbp, *first;
>       struct icmp *icp;
>       u_int32_t mtu;
>       struct sockaddr_in dst, src;
> @@ -973,10 +976,9 @@ udpencap_ctlinput(int cmd, struct sockad
>       src.sin_addr.s_addr = ip->ip_src.s_addr;
>       su_src = (union sockaddr_union *)&src;
>  
> -     tdbp = gettdbbysrcdst_rev(rdomain, 0, su_src, su_dst,
> -         IPPROTO_ESP);
> +     first = gettdbbysrcdst_rev(rdomain, 0, su_src, su_dst, IPPROTO_ESP);
>  
> -     for (; tdbp != NULL; tdbp = tdbp->tdb_snext) {
> +     for (tdbp = first; tdbp != NULL; tdbp = tdbp->tdb_snext) {
>               if (tdbp->tdb_sproto == IPPROTO_ESP &&
>                   ((tdbp->tdb_flags & (TDBF_INVALID|TDBF_UDPENCAP)) ==
>                   TDBF_UDPENCAP) &&
> @@ -985,6 +987,7 @@ udpencap_ctlinput(int cmd, struct sockad
>                       ipsec_set_mtu(tdbp, mtu);
>               }
>       }
> +     tdb_unref(first);
>  }
>  
>  void
> @@ -1070,6 +1073,7 @@ ipsec_forward_check(struct mbuf *m, int 
>       } else
>               tdb = NULL;
>       ipsp_spd_lookup(m, af, hlen, &error, IPSP_DIRECTION_IN, tdb, NULL, 0);
> +     tdb_unref(tdb);
>  
>       return error;
>  }
> @@ -1142,6 +1146,7 @@ ipsec_local_check(struct mbuf *m, int hl
>               tdb = NULL;
>       ipsp_spd_lookup(m, af, hlen, &error, IPSP_DIRECTION_IN,
>           tdb, NULL, 0);
> +     tdb_unref(tdb);
>  
>       return error;
>  }
> Index: netinet/ipsec_output.c
> ===================================================================
> RCS file: /data/mirror/openbsd/cvs/src/sys/netinet/ipsec_output.c,v
> retrieving revision 1.91
> diff -u -p -r1.91 ipsec_output.c
> --- netinet/ipsec_output.c    23 Oct 2021 15:42:35 -0000      1.91
> +++ netinet/ipsec_output.c    12 Nov 2021 21:03:24 -0000
> @@ -388,6 +388,7 @@ ipsp_process_done(struct mbuf *m, struct
>  #ifdef INET6
>       struct ip6_hdr *ip6;
>  #endif /* INET6 */
> +     struct tdb *tdbo;
>       struct tdb_ident *tdbi;
>       struct m_tag *mtag;
>       int roff, error;
> @@ -501,9 +502,13 @@ ipsp_process_done(struct mbuf *m, struct
>       tdb->tdb_obytes += m->m_pkthdr.len;
>  
>       /* If there's another (bundled) TDB to apply, do so. */
> -     if (tdb->tdb_onext)
> -             return ipsp_process_packet(m, tdb->tdb_onext,
> +     tdbo = tdb_ref(tdb->tdb_onext);
> +     if (tdbo != NULL) {
> +             error = ipsp_process_packet(m, tdbo,
>                   tdb->tdb_dst.sa.sa_family, 0);
> +             tdb_unref(tdbo);
> +             return error;
> +     }
>  
>  #if NPF > 0
>       /* Add pf tag if requested. */
> @@ -615,13 +620,16 @@ ipsec_adjust_mtu(struct mbuf *m, u_int32
>               if (tdbp == NULL)
>                       break;
>  
> -             if ((adjust = ipsec_hdrsz(tdbp)) == -1)
> +             if ((adjust = ipsec_hdrsz(tdbp)) == -1) {
> +                     tdb_unref(tdbp);
>                       break;
> +             }
>  
>               mtu -= adjust;
>               tdbp->tdb_mtu = mtu;
>               tdbp->tdb_mtutimeout = gettime() + ip_mtudisc_timeout;
>               DPRINTF("spi %08x mtu %d adjust %ld mbuf %p",
>                   ntohl(tdbp->tdb_spi), tdbp->tdb_mtu, adjust, m);
> +             tdb_unref(tdbp);
>       }
>  }
> Index: netinet/tcp_input.c
> ===================================================================
> RCS file: /data/mirror/openbsd/cvs/src/sys/netinet/tcp_input.c,v
> retrieving revision 1.370
> diff -u -p -r1.370 tcp_input.c
> --- netinet/tcp_input.c       9 Aug 2021 17:03:08 -0000       1.370
> +++ netinet/tcp_input.c       11 Nov 2021 23:24:31 -0000
> @@ -380,12 +380,6 @@ tcp_input(struct mbuf **mp, int *offp, i
>  #ifdef INET6
>       struct ip6_hdr *ip6 = NULL;
>  #endif /* INET6 */
> -#ifdef IPSEC
> -     struct m_tag *mtag;
> -     struct tdb_ident *tdbi;
> -     struct tdb *tdb;
> -     int error;
> -#endif /* IPSEC */
>  #ifdef TCP_ECN
>       u_char iptos;
>  #endif
> @@ -571,16 +565,22 @@ findpcb:
>       }
>  #ifdef IPSEC
>       if (ipsec_in_use) {
> +             struct m_tag *mtag;
> +             struct tdb *tdb = NULL;
> +             int error;
> +
>               /* Find most recent IPsec tag */
>               mtag = m_tag_find(m, PACKET_TAG_IPSEC_IN_DONE, NULL);
>               if (mtag != NULL) {
> +                     struct tdb_ident *tdbi;
> +
>                       tdbi = (struct tdb_ident *)(mtag + 1);
>                       tdb = gettdb(tdbi->rdomain, tdbi->spi,
>                           &tdbi->dst, tdbi->proto);
> -             } else
> -                     tdb = NULL;
> +             }
>               ipsp_spd_lookup(m, af, iphlen, &error, IPSP_DIRECTION_IN,
>                   tdb, inp, 0);
> +             tdb_unref(tdb);
>               if (error) {
>                       tcpstat_inc(tcps_rcvnosec);
>                       goto drop;
> @@ -2197,7 +2197,7 @@ tcp_dooptions(struct tcpcb *tp, u_char *
>                               continue;
>  
>                       if (sigp && timingsafe_bcmp(sigp, cp + 2, 16))
> -                             return (-1);
> +                             goto bad;
>  
>                       sigp = cp + 2;
>                       break;
> @@ -2248,7 +2248,7 @@ tcp_dooptions(struct tcpcb *tp, u_char *
>  
>       if ((sigp ? TF_SIGNATURE : 0) ^ (tp->t_flags & TF_SIGNATURE)) {
>               tcpstat_inc(tcps_rcvbadsig);
> -             return (-1);
> +             goto bad;
>       }
>  
>       if (sigp) {
> @@ -2256,22 +2256,30 @@ tcp_dooptions(struct tcpcb *tp, u_char *
>  
>               if (tdb == NULL) {
>                       tcpstat_inc(tcps_rcvbadsig);
> -                     return (-1);
> +                     goto bad;
>               }
>  
>               if (tcp_signature(tdb, tp->pf, m, th, iphlen, 1, sig) < 0)
> -                     return (-1);
> +                     goto bad;
>  
>               if (timingsafe_bcmp(sig, sigp, 16)) {
>                       tcpstat_inc(tcps_rcvbadsig);
> -                     return (-1);
> +                     goto bad;
>               }
>  
>               tcpstat_inc(tcps_rcvgoodsig);
>       }
> +
> +     tdb_unref(tdb);
>  #endif /* TCP_SIGNATURE */
>  
>       return (0);
> +
> + bad:
> +#ifdef TCP_SIGNATURE
> +     tdb_unref(tdb);
> +#endif /* TCP_SIGNATURE */
> +     return (-1);
>  }
>  
>  u_long
> @@ -4056,8 +4064,10 @@ syn_cache_respond(struct syn_cache *sc, 
>               if (tcp_signature(tdb, sc->sc_src.sa.sa_family, m, th,
>                   hlen, 0, optp) < 0) {
>                       m_freem(m);
> +                     tdb_unref(tdb);
>                       return (EINVAL);
>               }
> +             tdb_unref(tdb);
>               optp += 16;
>  
>               /* Pad options list to the next 32 bit boundary and
> Index: netinet/tcp_output.c
> ===================================================================
> RCS file: /data/mirror/openbsd/cvs/src/sys/netinet/tcp_output.c,v
> retrieving revision 1.130
> diff -u -p -r1.130 tcp_output.c
> --- netinet/tcp_output.c      8 Feb 2021 19:37:15 -0000       1.130
> +++ netinet/tcp_output.c      12 Nov 2021 21:01:58 -0000
> @@ -879,8 +879,10 @@ send:
>               if (tcp_signature(tdb, tp->pf, m, th, iphlen, 0,
>                   mtod(m, caddr_t) + hdrlen - optlen + sigoff) < 0) {
>                       m_freem(m);
> +                     tdb_unref(tdb);
>                       return (EINVAL);
>               }
> +             tdb_unref(tdb);
>       }
>  #endif /* TCP_SIGNATURE */
>  
> Index: netinet/udp_usrreq.c
> ===================================================================
> RCS file: /data/mirror/openbsd/cvs/src/sys/netinet/udp_usrreq.c,v
> retrieving revision 1.264
> diff -u -p -r1.264 udp_usrreq.c
> --- netinet/udp_usrreq.c      11 Nov 2021 18:08:18 -0000      1.264
> +++ netinet/udp_usrreq.c      12 Nov 2021 21:01:58 -0000
> @@ -514,11 +514,13 @@ udp_input(struct mbuf **mp, int *offp, i
>                   IPSP_DIRECTION_IN, tdb, inp, 0);
>               if (error) {
>                       udpstat_inc(udps_nosec);
> +                     tdb_unref(tdb);
>                       goto bad;
>               }
>               /* create ipsec options while we know that tdb cannot be 
> modified */
>               if (tdb && tdb->tdb_ids)
>                       ipsecflowinfo = tdb->tdb_ids->id_flow;
> +             tdb_unref(tdb);
>       }
>  #endif /*IPSEC */
>  
> 

Reply via email to