|
|
| version 1.1, 2022/06/22 13:01:55 | version 1.7, 2022/08/04 00:24:39 |
|---|---|
| Line 5 | Line 5 |
| #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) |
| { | { |
| int e, rcvflg; | int e, rcvflg = 0; |
| struct uio uio; | struct uio uio; |
| struct mbuf *m = NULL; | struct mbuf *m = NULL; |
| struct fws_proto *pkt; | |
| 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*); | |
| if (m_length(m, NULL) != sizeof(struct fws_proto)) { | |
| printf("FWSync packet length=%d isn't match expected %lu\n", | |
| m_length(m, NULL), sizeof(struct fws_proto)); | |
| return; | |
| } | |
| printf("%u...-> %s\n", m->m_len, m->m_data); | printf("%u...-> %s version=%c\n", m->m_len, m->m_data, pkt->fws_version); |
| } | |
| fws_ctx.config &= ~CTX_EDGE_ONLINE; | |
| kproc_exit(0); | switch (pkt->fws_version) { |
| case FWS_PKTVER_STATE: | |
| 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"); | |
| return; | |
| } | |
| } | |
| } | } |
| int | int |
| Line 156 fwsync_destroy(struct ip_fw_chain *ch, ip_fw3_opheader | Line 166 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); | |
| 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 233 fwsync_list(struct ip_fw_chain *ch, ip_fw3_opheader *o | Line 243 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 254 fwsync_start(struct ip_fw_chain *ch, ip_fw3_opheader * | Line 264 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; |
| fws_ctx.config |= CTX_EDGE_ONLINE; | |
| 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) { | |
| printf("fwsync fwsync_edge failed #%d\n", e); | |
| return e; | |
| } | |
| } | } |
| 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 285 fwsync_start(struct ip_fw_chain *ch, ip_fw3_opheader * | Line 291 fwsync_start(struct ip_fw_chain *ch, ip_fw3_opheader * |
| fws_ctx.config |= CTX_COLLECTOR_2_ONLINE; | fws_ctx.config |= CTX_COLLECTOR_2_ONLINE; |
| } | } |
| /* at last but not at least, activate config status of started sync service */ | |
| fws_cfg.cfg.on |= *n; | |
| return 0; | return 0; |
| } | } |
| Line 312 fwsync_stop(struct ip_fw_chain *ch, ip_fw3_opheader *o | Line 316 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); | |
| } | |
| 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 328 fwsync_stop(struct ip_fw_chain *ch, ip_fw3_opheader *o | Line 336 fwsync_stop(struct ip_fw_chain *ch, ip_fw3_opheader *o |
| ipfw_unregister_alias_hook(); | ipfw_unregister_alias_hook(); |
| } | } |
| /* at last but not at least, stop sync service */ | |
| fws_cfg.cfg.on &= ~*n; | |
| return 0; | return 0; |
| } | } |