Annotation of fwsync/driver/fwsync_workers.c, revision 1.2

1.1       misho       1: /*************************************************************************
                      2: * (C) 2022 CloudSigma AG - Sofia/Bulgaria
                      3: *  by Michael Pounov <misho@elwix.org>
                      4: **************************************************************************/
                      5: #include "fwsync.h"
                      6: 
                      7: 
1.2     ! misho       8: void
        !             9: fwsync_sndpkt_state(void *context, int pending)
1.1       misho      10: {
1.2     ! misho      11:        struct fws_sndpkt *pkt;
        !            12:        struct mbuf *m2, *m;
1.1       misho      13:        int e;
                     14: 
                     15:        DTRACE();
                     16: 
1.2     ! misho      17:        printf("pending=%d\n", pending);
        !            18: 
        !            19:        do {
        !            20:                mtx_lock(&fws_mtx_c);
        !            21:                pkt = TAILQ_FIRST(&fwsync_sndpkt);
        !            22:                TAILQ_REMOVE(&fwsync_sndpkt, pkt, sp_next);
        !            23:                mtx_unlock(&fws_mtx_c);
        !            24: 
        !            25:                if (!(fws_cfg.cfg.on & CFG_SYNC_COLLECTOR))
        !            26:                        continue;
        !            27: 
        !            28:                m = m_gethdr(M_NOWAIT, MT_DATA);
        !            29:                if (!m)
        !            30:                        continue;
        !            31: 
        !            32:                memcpy(mtod(m, struct fws_proto *), &pkt->sp_proto, sizeof pkt->sp_proto);
        !            33:                m->m_len = sizeof pkt->sp_proto;
        !            34:                m_fixhdr(m);
        !            35: 
1.1       misho      36:                if ((fws_ctx.config & (CTX_COLLECTOR_2_ONLINE | CTX_COLLECTOR_2_READY)) == 
                     37:                                (CTX_COLLECTOR_2_ONLINE | CTX_COLLECTOR_2_READY)) {
                     38:                        m2 = m_copypacket(m, M_NOWAIT);
                     39:                        if (!m2) {
                     40:                                printf("error in copypacket for second collector\n");
                     41:                                goto out;
                     42:                        }
                     43:                        e = sosend(fws_ctx.sockz[CFG_SYNC_ADDR_COLLECTOR_2], 
                     44:                                        &fws_cfg.cfg_addr[CFG_SYNC_ADDR_COLLECTOR_2].addr, NULL, 
                     45:                                        m2, NULL, 0, curthread);
1.2     ! misho      46:                        if (e && e != EAGAIN)
1.1       misho      47:                                printf("error in collector %d handler #%d\n", CFG_SYNC_ADDR_COLLECTOR_2, e);
                     48:                }
                     49: 
                     50:                if ((fws_ctx.config & (CTX_COLLECTOR_1_ONLINE | CTX_COLLECTOR_1_READY)) == 
                     51:                                (CTX_COLLECTOR_1_ONLINE | CTX_COLLECTOR_1_READY)) {
                     52:                        m2 = m_copypacket(m, M_NOWAIT);
                     53:                        if (!m2) {
                     54:                                printf("error in copypacket for first collector\n");
                     55:                                goto out;
                     56:                        }
                     57:                        e = sosend(fws_ctx.sockz[CFG_SYNC_ADDR_COLLECTOR_1], 
                     58:                                        &fws_cfg.cfg_addr[CFG_SYNC_ADDR_COLLECTOR_1].addr, NULL, 
                     59:                                        m2, NULL, 0, curthread);
1.2     ! misho      60:                        if (e && e != EAGAIN)
1.1       misho      61:                                printf("error in collector %d handler #%d\n", CFG_SYNC_ADDR_COLLECTOR_1, e);
                     62:                }
                     63: out:
1.2     ! misho      64:                m_freem(m);
        !            65:        } while (--pending);
1.1       misho      66: }
                     67: 
                     68: int
1.2     ! misho      69: fwsync_state_handler(void *arg)
1.1       misho      70: {
1.2     ! misho      71:        struct ipfw_flow_id *pkt = arg;
1.1       misho      72:        struct fws_proto *spkt;
1.2     ! misho      73:        struct fws_sndpkt *p;
1.1       misho      74: 
                     75:        DTRACE();
                     76: 
                     77:        if (pkt->addr_type == 1)
                     78:                return 0;       /* skip ethernet packet */
                     79: 
1.2     ! misho      80:        mtx_lock(&fws_mtx_c);
        !            81: 
        !            82:        p = malloc(sizeof(struct fws_proto), M_FWSYNC, M_NOWAIT | M_ZERO);
        !            83:        if (!p) {
        !            84:                mtx_unlock(&fws_mtx_c);
        !            85:                return 0;
        !            86:        } else
        !            87:                spkt = &p->sp_proto;
1.1       misho      88: 
                     89:        spkt->fws_version = FWS_PKTVER_STATE;
                     90:        spkt->fws_fib = pkt->fib;
                     91:        spkt->fws_proto = pkt->proto;
                     92:        spkt->fws_addrtype = pkt->addr_type;
                     93:        spkt->fws_sport = pkt->src_port;
                     94:        spkt->fws_dport = pkt->dst_port;
                     95: 
                     96:        switch (pkt->addr_type) {
                     97:                case 4:
                     98:                        spkt->fws_saddr.s_addr = pkt->src_ip;
                     99:                        spkt->fws_daddr.s_addr = pkt->dst_ip;
                    100:                        spkt->fws_scopeid = 0;
                    101:                        break;
                    102:                case 6:
                    103:                        memcpy(&spkt->fws_saddr6, &pkt->src_ip6, sizeof spkt->fws_saddr6);
                    104:                        memcpy(&spkt->fws_daddr6, &pkt->dst_ip6, sizeof spkt->fws_daddr6);
                    105:                        spkt->fws_scopeid = pkt->flow_id6;
                    106:                        break;
                    107:                default:
1.2     ! misho     108:                        free(p, M_FWSYNC);
        !           109:                        mtx_unlock(&fws_mtx_c);
        !           110:                        return EINVAL;
1.1       misho     111:        }
                    112: 
1.2     ! misho     113:        TAILQ_INSERT_TAIL(&fwsync_sndpkt, p, sp_next);
        !           114: 
        !           115:        mtx_unlock(&fws_mtx_c);
        !           116: 
        !           117:        taskqueue_enqueue(fws_tq, &fws_sndpkt_task);
1.1       misho     118:        return 0;
                    119: }
                    120: 
                    121: int
1.2     ! misho     122: fwsync_alias_handler(void *arg)
1.1       misho     123: {
                    124:        DTRACE();
                    125: 
                    126:        return 0;
                    127: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>