Diff for /fwsync/driver/fwsync.c between versions 1.3 and 1.11

version 1.3, 2022/07/04 13:29:10 version 1.11, 2022/08/18 13:02:13
Line 1 Line 1
/*************************************************************************/*-
* (C) 2022 CloudSigma AG - Sofia/Bulgaria * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
*  by Michael Pounov <misho@elwix.org> *
**************************************************************************/ * Copyright (c) 2022 Michael Pounov <misho@elwix.org>, CloudSigma AG
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  * 1. Redistributions of source code must retain the above copyright
  *    notice, this list of conditions and the following disclaimer.
  * 2. Redistributions in binary form must reproduce the above copyright
  *    notice, this list of conditions and the following disclaimer in the
  *    documentation and/or other materials provided with the distribution.
  *
  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  * SUCH DAMAGE.
  */
 #include "fwsync.h"  #include "fwsync.h"
   
   
 static intptr_t fwsync_wchan;  
   
   
 static void  static void
 fwsync_edge_proc(void *arg)  fwsync_edge_proc(void *arg)
 {  {
Line 18  fwsync_edge_proc(void *arg) Line 37  fwsync_edge_proc(void *arg)
   
         DTRACE();          DTRACE();
   
           callout_schedule(&fws_co, hz);
   
         memset(&uio, 0, sizeof uio);          memset(&uio, 0, sizeof uio);
         uio.uio_resid = 1000000000;          uio.uio_resid = 1000000000;
         uio.uio_td = curthread;          uio.uio_td = curthread;
   
        fws_ctx.config |= CTX_EDGE_ONLINE;        if ((fws_cfg.cfg.on & CFG_SYNC_EDGE) && (fws_ctx.config & CTX_EDGE_READY)) {
        while ((fws_cfg.cfg.on & CFG_SYNC_EDGE) && (fws_ctx.config & CTX_EDGE_READY)) { 
                if (m) 
                        m_freem(m); 
                m = NULL; 
                 rcvflg = MSG_DONTWAIT;                  rcvflg = MSG_DONTWAIT;
                 e = soreceive(fws_ctx.sockz[CFG_SYNC_ADDR_EDGE], NULL, &uio, &m, NULL, &rcvflg);                  e = soreceive(fws_ctx.sockz[CFG_SYNC_ADDR_EDGE], NULL, &uio, &m, NULL, &rcvflg);
                 if (e) {                  if (e) {
                         if (e != EAGAIN)                          if (e != EAGAIN)
                                 printf("error in edge handler #%d\n", e);                                  printf("error in edge handler #%d\n", e);
                        tsleep(&fwsync_wchan, 0, "edge_handler", hz);                        return;
                        continue; 
                 }                  }
                 pkt = mtod(m, struct fws_proto*);                  pkt = mtod(m, struct fws_proto*);
                 if (m_length(m, NULL) != sizeof(struct fws_proto)) {                  if (m_length(m, NULL) != sizeof(struct fws_proto)) {
                         printf("FWSync packet length=%d isn't match expected %lu\n",                           printf("FWSync packet length=%d isn't match expected %lu\n", 
                                         m_length(m, NULL), sizeof(struct fws_proto));                                          m_length(m, NULL), sizeof(struct fws_proto));
                        continue;                        m_freem(m);
                         return;
                 }                  }
   
                printf("%u...-> %s version=%c\n", m->m_len, m->m_data, pkt->fws_version);                switch (pkt->fws_version) {
        }                        case FWS_PKTVER_STATE:
        fws_ctx.config &= ~CTX_EDGE_ONLINE;                                fwsync_add_state(pkt);
                                 break;
                         case FWS_PKTVER_ALIAS:
                                 fwsync_add_alias(pkt);
                                 break;
                         default:
                                 printf("FWSync packet was discarded due to wrong version\n");
                                 break;
                 }
   
        kproc_exit(0);                m_freem(m);
         }
 }  }
   
 int  int
Line 163  fwsync_destroy(struct ip_fw_chain *ch, ip_fw3_opheader Line 189  fwsync_destroy(struct ip_fw_chain *ch, ip_fw3_opheader
                 fws_cfg.cfg.addrs = 0;                  fws_cfg.cfg.addrs = 0;
                 memset(fws_cfg.cfg_addr, 0, sizeof fws_cfg.cfg_addr[0]);                  memset(fws_cfg.cfg_addr, 0, sizeof fws_cfg.cfg_addr[0]);
   
                if (fws_ctx.config & CTX_EDGE_READY) {                if (fws_ctx.config & CTX_EDGE_ONLINE) {
                         callout_drain(&fws_co);
                         ipfw_unregister_state_sync();
                         ipfw_unregister_alias_sync();
                         soshutdown(fws_ctx.sockz[CFG_SYNC_ADDR_EDGE], SHUT_RD);                          soshutdown(fws_ctx.sockz[CFG_SYNC_ADDR_EDGE], SHUT_RD);
                         tsleep(&fwsync_wchan, 0, "fwsedge_kill", 2 * hz);  
                         soclose(fws_ctx.sockz[CFG_SYNC_ADDR_EDGE]);                          soclose(fws_ctx.sockz[CFG_SYNC_ADDR_EDGE]);
                 }                  }
         }          }
Line 175  fwsync_destroy(struct ip_fw_chain *ch, ip_fw3_opheader Line 203  fwsync_destroy(struct ip_fw_chain *ch, ip_fw3_opheader
                         ipfw_unregister_alias_hook();                          ipfw_unregister_alias_hook();
                 }                  }
   
                   taskqueue_drain(fws_tq, &fws_sndpkt_task);
   
                 fws_cfg.cfg.on &= ~CFG_SYNC_COLLECTOR;                  fws_cfg.cfg.on &= ~CFG_SYNC_COLLECTOR;
                 fws_cfg.cfg.collector = 0;                  fws_cfg.cfg.collector = 0;
                 fws_cfg.cfg.addrs = 0;                  fws_cfg.cfg.addrs = 0;
Line 240  fwsync_list(struct ip_fw_chain *ch, ip_fw3_opheader *o Line 270  fwsync_list(struct ip_fw_chain *ch, ip_fw3_opheader *o
 int  int
 fwsync_start(struct ip_fw_chain *ch, ip_fw3_opheader *op3, struct sockopt_data *sd)  fwsync_start(struct ip_fw_chain *ch, ip_fw3_opheader *op3, struct sockopt_data *sd)
 {  {
        int *n, e;        int *n;
         ipfw_obj_header *oh;          ipfw_obj_header *oh;
         size_t sz;          size_t sz;
   
Line 262  fwsync_start(struct ip_fw_chain *ch, ip_fw3_opheader * Line 292  fwsync_start(struct ip_fw_chain *ch, ip_fw3_opheader *
         if ((*n & CFG_SYNC_EDGE) && (fws_ctx.config & CTX_EDGE_READY)) {          if ((*n & CFG_SYNC_EDGE) && (fws_ctx.config & CTX_EDGE_READY)) {
                 fws_cfg.cfg.on |= CFG_SYNC_EDGE;                  fws_cfg.cfg.on |= CFG_SYNC_EDGE;
   
                e = kproc_create(fwsync_edge_proc, NULL, &fws_ctx.procz[CFG_SYNC_ADDR_EDGE],                 callout_reset(&fws_co, hz, fwsync_edge_proc, NULL);
                                0, 0, "fwsync_edge");
                if (e) {                if (!(fws_ctx.config & CTX_EDGE_ONLINE)) {
                        printf("fwsync fwsync_edge failed #%d\n", e);                        ipfw_register_state_sync(fwsync_state_sync);
                        return e;                        ipfw_register_alias_sync(fwsync_alias_sync);
                 }                  }
   
                   fws_ctx.config |= CTX_EDGE_ONLINE;
         }          }
   
         if ((*n & CFG_SYNC_COLLECTOR) && (fws_ctx.config & CTX_COLLECTOR_1_READY)) {          if ((*n & CFG_SYNC_COLLECTOR) && (fws_ctx.config & CTX_COLLECTOR_1_READY)) {
Line 317  fwsync_stop(struct ip_fw_chain *ch, ip_fw3_opheader *o Line 349  fwsync_stop(struct ip_fw_chain *ch, ip_fw3_opheader *o
   
         n = (int*) (oh + 1);          n = (int*) (oh + 1);
   
        if ((*n & CFG_SYNC_EDGE) && (fws_ctx.config & CTX_CFG_EDGE))        if ((*n & CFG_SYNC_EDGE) && (fws_ctx.config & CTX_CFG_EDGE)) {
                 fws_cfg.cfg.on &= ~CFG_SYNC_EDGE;                  fws_cfg.cfg.on &= ~CFG_SYNC_EDGE;
                   fws_ctx.config &= ~CTX_EDGE_ONLINE;
   
                   callout_drain(&fws_co);
   
                   ipfw_unregister_state_sync();
                   ipfw_unregister_alias_sync();
           }
   
         if ((*n & CFG_SYNC_COLLECTOR) && (fws_ctx.config & CTX_COLLECTOR_2_ONLINE))          if ((*n & CFG_SYNC_COLLECTOR) && (fws_ctx.config & CTX_COLLECTOR_2_ONLINE))
                 fws_ctx.config &= ~CTX_COLLECTOR_2_ONLINE;                  fws_ctx.config &= ~CTX_COLLECTOR_2_ONLINE;
         if ((*n & CFG_SYNC_COLLECTOR) && (fws_ctx.config & CTX_COLLECTOR_1_ONLINE))          if ((*n & CFG_SYNC_COLLECTOR) && (fws_ctx.config & CTX_COLLECTOR_1_ONLINE))
Line 331  fwsync_stop(struct ip_fw_chain *ch, ip_fw3_opheader *o Line 370  fwsync_stop(struct ip_fw_chain *ch, ip_fw3_opheader *o
   
                 ipfw_unregister_state_hook();                  ipfw_unregister_state_hook();
                 ipfw_unregister_alias_hook();                  ipfw_unregister_alias_hook();
   
                   taskqueue_drain(fws_tq, &fws_sndpkt_task);
         }          }
   
         return 0;          return 0;

Removed from v.1.3  
changed lines
  Added in v.1.11


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