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>