[BACK]Return to fwsync_workers.c CVS log [TXT] [DIR] Up to [ELWIX - Embedded LightWeight unIX -] / fwsync / driver

Diff for /fwsync/driver/fwsync_workers.c between versions 1.1 and 1.14

version 1.1, 2022/06/22 13:01:55 version 1.14, 2022/08/18 11:30:27
Line 5 Line 5
 #include "fwsync.h"  #include "fwsync.h"
   
   
static voidint
fwsync_sndpkt(void *context, int pending)fwsync_add_state(const struct fws_proto *pkt)
 {  {
        struct mbuf *m2, *m = (struct mbuf*) context;        struct fws_sndpkt *p;
 
         DTRACE();
 
         if (!pkt || pkt->fws_addrtype == 1)
                 return 0;       /* skip ethernet packet */
 
         p = malloc(sizeof(struct fws_sndpkt), M_FWSYNC, M_NOWAIT | M_ZERO);
         if (!p) {
                 return ENOMEM;
         } else
                 memcpy(&p->sp_proto, pkt, sizeof(struct fws_proto));
 
         mtx_lock(&fws_mtx_u);
         TAILQ_INSERT_TAIL(&fwsync_updpkt, p, sp_next);
         mtx_unlock(&fws_mtx_u);
         return 0;
 }
 
 
 int
 fwsync_add_alias(const struct fws_proto *pkt)
 {
         struct fws_sndpkt *p;
 
         DTRACE();
 
         if (!pkt || pkt->fws_addrtype == 1)
                 return 0;       /* skip ethernet packet */
 
         p = malloc(sizeof(struct fws_sndpkt), M_FWSYNC, M_NOWAIT | M_ZERO);
         if (!p) {
                 return ENOMEM;
         } else
                 memcpy(&p->sp_proto, pkt, sizeof(struct fws_proto));
 
         mtx_lock(&fws_mtx_n);
         TAILQ_INSERT_TAIL(&fwsync_natpkt, p, sp_next);
         mtx_unlock(&fws_mtx_n);
         return 0;
 }
 
 
 void
 fwsync_sndpkt_handler(void *context, int pending)
 {
         struct fws_sndpkt *pkt;
         struct mbuf *m2, *m;
         int e;          int e;
   
         DTRACE();          DTRACE();
   
        if (fws_cfg.cfg.on & CFG_SYNC_COLLECTOR) {        do {
                 mtx_lock(&fws_mtx_c);
                 pkt = TAILQ_FIRST(&fwsync_sndpkt);
                 if (pkt)
                         TAILQ_REMOVE(&fwsync_sndpkt, pkt, sp_next);
                 mtx_unlock(&fws_mtx_c);
 
                 if (!pkt)       /* sending queue is empty */
                         break;
 
                 if (!(fws_cfg.cfg.on & CFG_SYNC_COLLECTOR)) {
                         free(pkt, M_FWSYNC);
                         continue;
                 }
 
                 m = m_gethdr(M_NOWAIT, MT_DATA);
                 if (!m) {
                         free(pkt, M_FWSYNC);
                         continue;
                 }
 
                 memcpy(mtod(m, struct fws_proto *), &pkt->sp_proto, sizeof pkt->sp_proto);
                 m->m_len = sizeof pkt->sp_proto;
                 m_fixhdr(m);
 
                 if ((fws_ctx.config & (CTX_COLLECTOR_2_ONLINE | CTX_COLLECTOR_2_READY)) ==                   if ((fws_ctx.config & (CTX_COLLECTOR_2_ONLINE | CTX_COLLECTOR_2_READY)) == 
                                 (CTX_COLLECTOR_2_ONLINE | CTX_COLLECTOR_2_READY)) {                                  (CTX_COLLECTOR_2_ONLINE | CTX_COLLECTOR_2_READY)) {
                         m2 = m_copypacket(m, M_NOWAIT);                          m2 = m_copypacket(m, M_NOWAIT);
Line 24  fwsync_sndpkt(void *context, int pending) Line 95  fwsync_sndpkt(void *context, int pending)
                         e = sosend(fws_ctx.sockz[CFG_SYNC_ADDR_COLLECTOR_2],                           e = sosend(fws_ctx.sockz[CFG_SYNC_ADDR_COLLECTOR_2], 
                                         &fws_cfg.cfg_addr[CFG_SYNC_ADDR_COLLECTOR_2].addr, NULL,                                           &fws_cfg.cfg_addr[CFG_SYNC_ADDR_COLLECTOR_2].addr, NULL, 
                                         m2, NULL, 0, curthread);                                          m2, NULL, 0, curthread);
                        if (e != EAGAIN)                        if (e && e != EAGAIN)
                                 printf("error in collector %d handler #%d\n", CFG_SYNC_ADDR_COLLECTOR_2, e);                                  printf("error in collector %d handler #%d\n", CFG_SYNC_ADDR_COLLECTOR_2, e);
                 }                  }
   
Line 38  fwsync_sndpkt(void *context, int pending) Line 109  fwsync_sndpkt(void *context, int pending)
                         e = sosend(fws_ctx.sockz[CFG_SYNC_ADDR_COLLECTOR_1],                           e = sosend(fws_ctx.sockz[CFG_SYNC_ADDR_COLLECTOR_1], 
                                         &fws_cfg.cfg_addr[CFG_SYNC_ADDR_COLLECTOR_1].addr, NULL,                                           &fws_cfg.cfg_addr[CFG_SYNC_ADDR_COLLECTOR_1].addr, NULL, 
                                         m2, NULL, 0, curthread);                                          m2, NULL, 0, curthread);
                        if (e != EAGAIN)                        if (e && e != EAGAIN)
                                 printf("error in collector %d handler #%d\n", CFG_SYNC_ADDR_COLLECTOR_1, e);                                  printf("error in collector %d handler #%d\n", CFG_SYNC_ADDR_COLLECTOR_1, e);
                 }                  }
         }  
 out:  out:
        m_freem(m);                m_freem(m);
                free(pkt, M_FWSYNC);
         } while (--pending);
 }  }
   
 int  int
fwsync_state_handler(const void *arg)fwsync_state_handler(const void *arg, const void *extdata)
 {  {
         const struct ipfw_flow_id *pkt = arg;          const struct ipfw_flow_id *pkt = arg;
           const struct ipfw_dyn_hook_extdata *edata = extdata;
         struct fws_proto *spkt;          struct fws_proto *spkt;
        struct mbuf *m;        struct fws_sndpkt *p;
   
         DTRACE();          DTRACE();
   
        if (pkt->addr_type == 1)        if (!pkt || pkt->addr_type == 1)
                 return 0;       /* skip ethernet packet */                  return 0;       /* skip ethernet packet */
   
        m = m_gethdr(M_WAITOK, MT_DATA);        mtx_lock(&fws_mtx_c);
        if (!m) 
                return ENOMEM; 
        else { 
                spkt = mtod(m, struct fws_proto *); 
                m->m_len = sizeof(struct fws_proto); 
                m_fixhdr(m); 
        } 
   
           p = malloc(sizeof(struct fws_sndpkt), M_FWSYNC, M_NOWAIT | M_ZERO);
           if (!p) {
                   mtx_unlock(&fws_mtx_c);
                   return 0;
           } else
                   spkt = &p->sp_proto;
   
         spkt->fws_version = FWS_PKTVER_STATE;          spkt->fws_version = FWS_PKTVER_STATE;
         spkt->fws_fib = pkt->fib;          spkt->fws_fib = pkt->fib;
           spkt->fws_cmdtype = edata->cmdtype;
           spkt->fws_kidx = edata->kidx;
           spkt->fws_ruleid = edata->ruleid;
           spkt->fws_rulenum = edata->rulenum;
         spkt->fws_proto = pkt->proto;          spkt->fws_proto = pkt->proto;
         spkt->fws_addrtype = pkt->addr_type;          spkt->fws_addrtype = pkt->addr_type;
         spkt->fws_sport = pkt->src_port;          spkt->fws_sport = pkt->src_port;
Line 87  fwsync_state_handler(const void *arg) Line 163  fwsync_state_handler(const void *arg)
                         spkt->fws_scopeid = pkt->flow_id6;                          spkt->fws_scopeid = pkt->flow_id6;
                         break;                          break;
                 default:                  default:
                        m_free(m);                        free(p, M_FWSYNC);
                        return 0;                        mtx_unlock(&fws_mtx_c);
                         return EINVAL;
         }          }
   
        TASK_INIT(&fws_sndpkt_task, 0, fwsync_sndpkt, m);        TAILQ_INSERT_TAIL(&fwsync_sndpkt, p, sp_next);
        taskqueue_enqueue(taskqueue_thread, &fws_sndpkt_task);
         mtx_unlock(&fws_mtx_c);
 
         taskqueue_enqueue(fws_tq, &fws_sndpkt_task);
         return 0;          return 0;
 }  }
   
 int  int
fwsync_alias_handler(const void *arg)fwsync_alias_handler(const void *arg, const void *extdata)
 {  {
           const struct alias_link *lnk = arg;
           struct fws_proto *spkt;
           struct fws_sndpkt *p;
   
         DTRACE();          DTRACE();
   
           if (!lnk || lnk->link_type >= IPPROTO_MAX)
                   return 0;
   
           mtx_lock(&fws_mtx_c);
   
           p = malloc(sizeof(struct fws_sndpkt), M_FWSYNC, M_NOWAIT | M_ZERO);
           if (!p) {
                   mtx_unlock(&fws_mtx_c);
                   return 0;
           } else
                   spkt = &p->sp_proto;
   
           spkt->fws_version = FWS_PKTVER_ALIAS;
           spkt->fws_proto = lnk->link_type;
           spkt->fws_addrtype = 4;
           spkt->fws_sport = lnk->src_port;
           spkt->fws_dport = lnk->dst_port;
           spkt->fws_aport = lnk->alias_port;
           spkt->fws_pport = lnk->proxy_port;
           spkt->fws_saddr.s_addr = lnk->src_addr.s_addr;
           spkt->fws_daddr.s_addr = lnk->dst_addr.s_addr;
           spkt->fws_aaddr.s_addr = lnk->alias_addr.s_addr;
           spkt->fws_paddr.s_addr = lnk->proxy_addr.s_addr;
   
           TAILQ_INSERT_TAIL(&fwsync_sndpkt, p, sp_next);
   
           mtx_unlock(&fws_mtx_c);
   
           taskqueue_enqueue(fws_tq, &fws_sndpkt_task);
           return 0;
   }
   
   int
   fwsync_state_sync(const void *arg, const void *extdata)
   {
           struct fws_sndpkt *pkt;
           struct ipfw_flow_id fid;
           struct ip_fw *rule;
           int f_pos;
           struct ip_fw_chain *chain = &V_layer3_chain;
   
           DTRACE();
   
           mtx_lock(&fws_mtx_u);
           pkt = TAILQ_FIRST(&fwsync_updpkt);
           if (pkt)
                   TAILQ_REMOVE(&fwsync_updpkt, pkt, sp_next);
           mtx_unlock(&fws_mtx_u);
   
           if (!pkt)       /* update queue is empty */
                   return 0;
   
           if (!(fws_cfg.cfg.on & CFG_SYNC_EDGE)) {
                   free(pkt, M_FWSYNC);
                   return EAGAIN;
           }
   
           memset(&fid, 0, sizeof fid);
           fid.fib = pkt->sp_proto.fws_fib;
           fid.proto = pkt->sp_proto.fws_proto;
           fid.addr_type = pkt->sp_proto.fws_addrtype;
           fid.src_port = pkt->sp_proto.fws_sport;
           fid.dst_port = pkt->sp_proto.fws_dport;
           switch (pkt->sp_proto.fws_addrtype) {
                   case 4:
                           fid.src_ip = pkt->sp_proto.fws_saddr.s_addr;
                           fid.dst_ip = pkt->sp_proto.fws_daddr.s_addr;
                           break;
                   case 6:
                           memcpy(&fid.src_ip6, &pkt->sp_proto.fws_saddr6, sizeof fid.src_ip6);
                           memcpy(&fid.dst_ip6, &pkt->sp_proto.fws_daddr6, sizeof fid.dst_ip6);
                           fid.flow_id6 = pkt->sp_proto.fws_scopeid;
                           break;
                   default:
                           free(pkt, M_FWSYNC);
                           return EINVAL;
           }
   
           f_pos = ipfw_find_rule(chain, pkt->sp_proto.fws_ruleid, pkt->sp_proto.fws_rulenum);
           rule = chain->map[f_pos];
           ipfw_dyn_install_sync_state(&fid, rule, 
                           pkt->sp_proto.fws_ruleid, pkt->sp_proto.fws_rulenum, 
                           pkt->sp_proto.fws_kidx, pkt->sp_proto.fws_cmdtype);
   
           free(pkt, M_FWSYNC);
           return 0;
   }
   
   int
   fwsync_alias_sync(const void *arg, const void *extdata)
   {
           struct fws_sndpkt *pkt;
           const struct cfg_nat *t = arg;
   
           DTRACE();
   
           mtx_lock(&fws_mtx_n);
           pkt = TAILQ_FIRST(&fwsync_natpkt);
           if (pkt)
                   TAILQ_REMOVE(&fwsync_natpkt, pkt, sp_next);
           mtx_unlock(&fws_mtx_n);
   
           if (!pkt)       /* update queue is empty */
                   return 0;
   
           if (!(fws_cfg.cfg.on & CFG_SYNC_EDGE)) {
                   free(pkt, M_FWSYNC);
                   return EAGAIN;
           }
   
           printf("%s: pkt=%p t=%p\n", __func__, pkt, t->if_name);
   
           AddLink(t->lib, pkt->sp_proto.fws_saddr, pkt->sp_proto.fws_daddr, pkt->sp_proto.fws_aaddr, 
                           pkt->sp_proto.fws_sport, pkt->sp_proto.fws_dport, pkt->sp_proto.fws_aport, 
                           pkt->sp_proto.fws_linktype | LINK_SYNC_MASK);
   
           free(pkt, M_FWSYNC);
         return 0;          return 0;
 }  }
Removed from v.1.1  
changed lines
  Added in v.1.14

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