static int sc_stream_add(struct sc_stream* s, void* vi_or_set, ef_driver_handle dh, ef_filter_cookie *cookie_out) { ef_filter_spec spec; ef_filter_spec_init(&spec, EF_FILTER_FLAG_NONE); switch( s->fields ) { case SC_SF_ALL: TRY(ef_filter_spec_set_unicast_all(&spec)); TRY(ef_vi_set_filter_add(vi_or_set, dh, &spec, cookie_out)); ef_filter_spec_init(&spec, EF_FILTER_FLAG_NONE); TRY(ef_filter_spec_set_multicast_all(&spec)); TRY(ef_vi_set_filter_add(vi_or_set, dh, &spec, cookie_out)); break; case SC_SF_SNIFF: TRY(ef_filter_spec_set_port_sniff(&spec, s->promiscuous)); TRY(ef_vi_set_filter_add(vi_or_set, dh, &spec, cookie_out)); break; case SC_SF_ETH_DHOST: TRY(ef_filter_spec_set_eth_local(&spec, EF_FILTER_VLAN_ID_ANY, s->eth_dhost)); TRY(ef_vi_set_filter_add(vi_or_set, dh, &spec, cookie_out)); break; case SC_SF_ETH_DHOST | SC_SF_ETH_VLAN_ID: TRY(ef_filter_spec_set_eth_local(&spec, s->eth_vlan_id, s->eth_dhost)); TRY(ef_vi_set_filter_add(vi_or_set, dh, &spec, cookie_out)); break; case SC_SF_ETH_TYPE | SC_SF_IP4_PROTOCOL | SC_SF_IP4_DEST_ADDR | SC_SF_IP4_DEST_PORT: TRY(ef_filter_spec_set_ip4_local(&spec, s->ip4_protocol, s->ip4_dest_addr, s->ip4_dest_port)); TRY(ef_vi_set_filter_add(vi_or_set, dh, &spec, cookie_out)); break; case SC_SF_ETH_TYPE | SC_SF_IP4_PROTOCOL | SC_SF_IP4_DEST_ADDR | SC_SF_IP4_DEST_PORT | SC_SF_IP4_SOURCE_ADDR | SC_SF_IP4_SOURCE_PORT: TRY(ef_filter_spec_set_ip4_full(&spec, s->ip4_protocol, s->ip4_dest_addr, s->ip4_dest_port, s->ip4_source_addr, s->ip4_source_port)); TRY(ef_vi_set_filter_add(vi_or_set, dh, &spec, cookie_out)); break; default: fprintf(stderr, "ERROR: sc_vi[_set]_add_stream_string: " "unsupported combination of fields (0x%x)\n", s->fields); return -EINVAL; } return 0; }
int filter_parse(ef_filter_spec* fs, const char* s_in) { struct sockaddr_in lsin, rsin; const char* type; const char* hostport; char* vlan; char* remainder; char *s; int rc = -EINVAL; int protocol; int i; ef_filter_spec_init(fs, EF_FILTER_FLAG_NONE); s = strdup(s_in); if( (type = strtok(s, ":")) == NULL ) goto out; if( ! strcmp("udp", type) || ! strcmp("tcp", type) ) { protocol = strcasecmp(type, "tcp") ? IPPROTO_UDP : IPPROTO_TCP; remainder = strtok(NULL, ""); if( ! strncmp("mcastloop-rx,", remainder, strlen("mcastloop-rx,")) ) { ef_filter_spec_init(fs, EF_FILTER_FLAG_MCAST_LOOP_RECEIVE); strtok(remainder, ","); remainder = strtok(NULL, ""); } if( ! strncmp("vid=", remainder, strlen("vid=")) ) { vlan = strtok(remainder, ","); remainder = strtok(NULL, ""); if( ! vlan ) goto out; vlan = strchr(vlan, '='); ++vlan; TRY(ef_filter_spec_set_vlan(fs, atoi(vlan))); } if( strchr(remainder, ',') ) { hostport = strtok(remainder, ","); remainder = strtok(NULL, ""); TRY(hostport_parse(&lsin, hostport)); TRY(hostport_parse(&rsin, remainder)); TRY(ef_filter_spec_set_ip4_full(fs, protocol, lsin.sin_addr.s_addr, lsin.sin_port, rsin.sin_addr.s_addr, rsin.sin_port)); rc = 0; } else { TRY(hostport_parse(&lsin, strtok(remainder, ","))); TRY(ef_filter_spec_set_ip4_local(fs, protocol, lsin.sin_addr.s_addr, lsin.sin_port)); rc = 0; } } else if( ! strcmp("eth", type) ) { uint8_t mac[6]; int vlan_id = EF_FILTER_VLAN_ID_ANY; vlan = strtok(NULL, ","); remainder = strtok(NULL, ""); if( remainder == '\0' ) /* No vlan */ remainder = vlan; else { if( strncmp("vid=", vlan, strlen("vid=")) ) goto out; vlan = strchr(vlan, '='); ++vlan; vlan_id = atoi(vlan); } for( i = 0; i < 6; ++i ) { mac[i] = strtol(remainder, &remainder, 16); if( i != 5 ) { if( *remainder != ':' ) goto out; ++remainder; if( ! strlen(remainder) ) goto out; } } if( strlen(remainder) ) goto out; TRY(ef_filter_spec_set_eth_local(fs, vlan_id, mac)); rc = 0; } else if( ! strcmp("multicast-all", type) ) { if( strlen(type) != strlen(s_in) ) goto out; TRY(ef_filter_spec_set_multicast_all(fs)); rc = 0; } else if( ! strcmp("unicast-all", type) ) { if( strlen(type) != strlen(s_in) ) goto out; TRY(ef_filter_spec_set_unicast_all(fs)); rc = 0; } else if( ! strcmp("multicast-mis", type) ) { TRY(ef_filter_spec_set_multicast_mismatch(fs)); if( strlen(type) != strlen(s_in) ) { remainder = strtok(NULL, ""); if( ! (vlan = strchr(remainder, '=')) ) goto out; ++vlan; TRY(ef_filter_spec_set_vlan(fs, atoi(vlan))); } rc = 0; } else if( ! strcmp("unicast-mis", type) ) { TRY(ef_filter_spec_set_unicast_mismatch(fs)); if( strlen(type) != strlen(s_in) ) { remainder = strtok(NULL, ""); if( ! (vlan = strchr(remainder, '=')) ) goto out; ++vlan; TRY(ef_filter_spec_set_vlan(fs, atoi(vlan))); } rc = 0; } else if( ! strcmp("sniff", type) ) { if( strlen(type) == strlen(s_in) ) { TRY(ef_filter_spec_set_port_sniff(fs, 1)); } else { remainder = strtok(NULL, ""); if( ! strcmp("promisc", remainder) ) TRY(ef_filter_spec_set_port_sniff(fs, 1)); else if( ! strcmp("no-promisc", remainder) ) TRY(ef_filter_spec_set_port_sniff(fs, 0)); else TRY(-EINVAL); } rc = 0; } else if( ! strcmp("tx-sniff", type) ) { TRY(ef_filter_spec_set_tx_port_sniff(fs)); rc = 0; } else if( ! strcmp("block-kernel", type) ) { TRY(ef_filter_spec_set_block_kernel(fs)); rc = 0; } else if( ! strcmp("block-kernel-unicast", type) ) { TRY(ef_filter_spec_set_block_kernel_unicast(fs)); rc = 0; } else if( ! strcmp("block-kernel-multicast", type) ) { TRY(ef_filter_spec_set_block_kernel_multicast(fs)); rc = 0; } out: free(s); return rc; }