File:  [ELWIX - Embedded LightWeight unIX -] / fwsync / driver / fwsync_workers.c
Revision 1.2: download - view: text, annotated - select for diffs - revision graph
Wed Jun 29 19:44:58 2022 UTC (23 months, 2 weeks ago) by misho
Branches: MAIN
CVS tags: HEAD
entirely redesign of collector network engine

    1: /*************************************************************************
    2: * (C) 2022 CloudSigma AG - Sofia/Bulgaria
    3: *  by Michael Pounov <misho@elwix.org>
    4: **************************************************************************/
    5: #include "fwsync.h"
    6: 
    7: 
    8: void
    9: fwsync_sndpkt_state(void *context, int pending)
   10: {
   11: 	struct fws_sndpkt *pkt;
   12: 	struct mbuf *m2, *m;
   13: 	int e;
   14: 
   15: 	DTRACE();
   16: 
   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: 
   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);
   46: 			if (e && e != EAGAIN)
   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);
   60: 			if (e && e != EAGAIN)
   61: 				printf("error in collector %d handler #%d\n", CFG_SYNC_ADDR_COLLECTOR_1, e);
   62: 		}
   63: out:
   64: 		m_freem(m);
   65: 	} while (--pending);
   66: }
   67: 
   68: int
   69: fwsync_state_handler(void *arg)
   70: {
   71: 	struct ipfw_flow_id *pkt = arg;
   72: 	struct fws_proto *spkt;
   73: 	struct fws_sndpkt *p;
   74: 
   75: 	DTRACE();
   76: 
   77: 	if (pkt->addr_type == 1)
   78: 		return 0;	/* skip ethernet packet */
   79: 
   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;
   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:
  108: 			free(p, M_FWSYNC);
  109: 			mtx_unlock(&fws_mtx_c);
  110: 			return EINVAL;
  111: 	}
  112: 
  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);
  118: 	return 0;
  119: }
  120: 
  121: int
  122: fwsync_alias_handler(void *arg)
  123: {
  124: 	DTRACE();
  125: 
  126: 	return 0;
  127: }

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