Annotation of fwsync/driver/fwsync_workers.c, revision 1.6
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.6 ! misho 8: static int
! 9: fwsync_add_state_4(const struct fws_proto *pkt)
! 10: {
! 11: DTRACE();
! 12:
! 13: return 0;
! 14: }
! 15:
! 16: static int
! 17: fwsync_add_state_6(const struct fws_proto *pkt)
! 18: {
! 19: DTRACE();
! 20:
! 21: return 0;
! 22: }
! 23:
! 24: int
! 25: fwsync_add_state(const struct fws_proto *pkt)
! 26: {
! 27: DTRACE();
! 28:
! 29: if (pkt->fws_addrtype == 4)
! 30: fwsync_add_state_4(pkt);
! 31: else if (pkt->fws_addrtype == 6)
! 32: fwsync_add_state_6(pkt);
! 33: else
! 34: return EINVAL;
! 35:
! 36: return 0;
! 37: }
! 38:
! 39: int
! 40: fwsync_add_alias(const struct fws_proto *pkt)
! 41: {
! 42: DTRACE();
! 43:
! 44: return 0;
! 45: }
! 46:
! 47:
1.2 misho 48: void
1.3 misho 49: fwsync_sndpkt_handler(void *context, int pending)
1.1 misho 50: {
1.2 misho 51: struct fws_sndpkt *pkt;
52: struct mbuf *m2, *m;
1.1 misho 53: int e;
54:
55: DTRACE();
56:
1.2 misho 57: printf("pending=%d\n", pending);
58:
59: do {
60: mtx_lock(&fws_mtx_c);
61: pkt = TAILQ_FIRST(&fwsync_sndpkt);
1.4 misho 62: if (pkt)
63: TAILQ_REMOVE(&fwsync_sndpkt, pkt, sp_next);
1.2 misho 64: mtx_unlock(&fws_mtx_c);
65:
1.4 misho 66: if (!pkt) /* sending queue is empty */
67: break;
68:
69: if (!(fws_cfg.cfg.on & CFG_SYNC_COLLECTOR)) {
70: free(pkt, M_FWSYNC);
1.2 misho 71: continue;
1.4 misho 72: }
1.2 misho 73:
74: m = m_gethdr(M_NOWAIT, MT_DATA);
1.4 misho 75: if (!m) {
76: free(pkt, M_FWSYNC);
1.2 misho 77: continue;
1.4 misho 78: }
1.2 misho 79:
80: memcpy(mtod(m, struct fws_proto *), &pkt->sp_proto, sizeof pkt->sp_proto);
81: m->m_len = sizeof pkt->sp_proto;
82: m_fixhdr(m);
83:
1.1 misho 84: if ((fws_ctx.config & (CTX_COLLECTOR_2_ONLINE | CTX_COLLECTOR_2_READY)) ==
85: (CTX_COLLECTOR_2_ONLINE | CTX_COLLECTOR_2_READY)) {
86: m2 = m_copypacket(m, M_NOWAIT);
87: if (!m2) {
88: printf("error in copypacket for second collector\n");
89: goto out;
90: }
91: e = sosend(fws_ctx.sockz[CFG_SYNC_ADDR_COLLECTOR_2],
92: &fws_cfg.cfg_addr[CFG_SYNC_ADDR_COLLECTOR_2].addr, NULL,
93: m2, NULL, 0, curthread);
1.2 misho 94: if (e && e != EAGAIN)
1.1 misho 95: printf("error in collector %d handler #%d\n", CFG_SYNC_ADDR_COLLECTOR_2, e);
96: }
97:
98: if ((fws_ctx.config & (CTX_COLLECTOR_1_ONLINE | CTX_COLLECTOR_1_READY)) ==
99: (CTX_COLLECTOR_1_ONLINE | CTX_COLLECTOR_1_READY)) {
100: m2 = m_copypacket(m, M_NOWAIT);
101: if (!m2) {
102: printf("error in copypacket for first collector\n");
103: goto out;
104: }
105: e = sosend(fws_ctx.sockz[CFG_SYNC_ADDR_COLLECTOR_1],
106: &fws_cfg.cfg_addr[CFG_SYNC_ADDR_COLLECTOR_1].addr, NULL,
107: m2, NULL, 0, curthread);
1.2 misho 108: if (e && e != EAGAIN)
1.1 misho 109: printf("error in collector %d handler #%d\n", CFG_SYNC_ADDR_COLLECTOR_1, e);
110: }
111: out:
1.2 misho 112: m_freem(m);
1.4 misho 113: free(pkt, M_FWSYNC);
1.2 misho 114: } while (--pending);
1.1 misho 115: }
116:
117: int
1.4 misho 118: fwsync_state_handler(const void *arg)
1.1 misho 119: {
1.4 misho 120: const struct ipfw_flow_id *pkt = arg;
1.1 misho 121: struct fws_proto *spkt;
1.2 misho 122: struct fws_sndpkt *p;
1.1 misho 123:
124: DTRACE();
125:
1.4 misho 126: if (!pkt || pkt->addr_type == 1)
1.1 misho 127: return 0; /* skip ethernet packet */
128:
1.2 misho 129: mtx_lock(&fws_mtx_c);
130:
1.5 misho 131: p = malloc(sizeof(struct fws_sndpkt), M_FWSYNC, M_NOWAIT | M_ZERO);
1.2 misho 132: if (!p) {
133: mtx_unlock(&fws_mtx_c);
134: return 0;
135: } else
136: spkt = &p->sp_proto;
1.1 misho 137:
138: spkt->fws_version = FWS_PKTVER_STATE;
139: spkt->fws_fib = pkt->fib;
140: spkt->fws_proto = pkt->proto;
141: spkt->fws_addrtype = pkt->addr_type;
142: spkt->fws_sport = pkt->src_port;
143: spkt->fws_dport = pkt->dst_port;
144:
145: switch (pkt->addr_type) {
146: case 4:
147: spkt->fws_saddr.s_addr = pkt->src_ip;
148: spkt->fws_daddr.s_addr = pkt->dst_ip;
149: spkt->fws_scopeid = 0;
150: break;
151: case 6:
152: memcpy(&spkt->fws_saddr6, &pkt->src_ip6, sizeof spkt->fws_saddr6);
153: memcpy(&spkt->fws_daddr6, &pkt->dst_ip6, sizeof spkt->fws_daddr6);
154: spkt->fws_scopeid = pkt->flow_id6;
155: break;
156: default:
1.2 misho 157: free(p, M_FWSYNC);
158: mtx_unlock(&fws_mtx_c);
159: return EINVAL;
1.1 misho 160: }
161:
1.2 misho 162: TAILQ_INSERT_TAIL(&fwsync_sndpkt, p, sp_next);
163:
164: mtx_unlock(&fws_mtx_c);
165:
166: taskqueue_enqueue(fws_tq, &fws_sndpkt_task);
1.1 misho 167: return 0;
168: }
169:
170: int
1.4 misho 171: fwsync_alias_handler(const void *arg)
1.1 misho 172: {
1.4 misho 173: const struct alias_link *lnk = arg;
174: struct fws_proto *spkt;
175: struct fws_sndpkt *p;
176:
1.1 misho 177: DTRACE();
178:
1.4 misho 179: if (!lnk || lnk->link_type >= IPPROTO_MAX)
180: return 0;
181:
182: mtx_lock(&fws_mtx_c);
183:
1.5 misho 184: p = malloc(sizeof(struct fws_sndpkt), M_FWSYNC, M_NOWAIT | M_ZERO);
1.4 misho 185: if (!p) {
186: mtx_unlock(&fws_mtx_c);
187: return 0;
188: } else
189: spkt = &p->sp_proto;
190:
191: spkt->fws_version = FWS_PKTVER_ALIAS;
192: spkt->fws_proto = lnk->link_type;
193: spkt->fws_addrtype = 4;
194: spkt->fws_sport = lnk->src_port;
195: spkt->fws_dport = lnk->dst_port;
196: spkt->fws_aport = lnk->alias_port;
197: spkt->fws_pport = lnk->proxy_port;
198: spkt->fws_saddr.s_addr = lnk->src_addr.s_addr;
199: spkt->fws_daddr.s_addr = lnk->dst_addr.s_addr;
200: spkt->fws_aaddr.s_addr = lnk->alias_addr.s_addr;
201: spkt->fws_paddr.s_addr = lnk->proxy_addr.s_addr;
202:
203: TAILQ_INSERT_TAIL(&fwsync_sndpkt, p, sp_next);
204:
205: mtx_unlock(&fws_mtx_c);
206:
207: taskqueue_enqueue(fws_tq, &fws_sndpkt_task);
1.1 misho 208: return 0;
209: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>
![[BACK]](/icons/cvsweb/back.gif){kind=icon}
Return to fwsync_workers.c CVS log ![[TXT]](/icons/cvsweb/text.gif){kind=icon}
![[DIR]](/icons/cvsweb/dir.gif){kind=icon}
Up to [ELWIX - Embedded LightWeight unIX -] / fwsync / driver