/* Print out the top line on the screen */ void display_topline(const char * msg) { scrn_printf(1, 20, "%s", msg); pktgen_display_set_color("top.copyright"); scrn_puts(" %s", wr_copyright_msg()); pktgen_display_set_color("top.poweredby"); scrn_puts(" %s", wr_powered_by()); pktgen_display_set_color(NULL); }
int main(int argc, char **argv) { uint32_t i; int32_t ret; printf("\n%s %s\n", wr_copyright_msg(), wr_powered_by()); fflush(stdout); wr_scrn_setw(1);/* Reset the window size */ /* call before the rte_eal_init() */ (void)rte_set_application_usage_hook(pktgen_usage); memset(&pktgen, 0, sizeof(pktgen)); pktgen.flags = PRINT_LABELS_FLAG; pktgen.ident = 0x1234; pktgen.nb_rxd = DEFAULT_RX_DESC; pktgen.nb_txd = DEFAULT_TX_DESC; pktgen.nb_ports_per_page = DEFAULT_PORTS_PER_PAGE; if ( (pktgen.l2p = wr_l2p_create()) == NULL) pktgen_log_panic("Unable to create l2p"); pktgen.portdesc_cnt = wr_get_portdesc(pktgen.portlist, pktgen.portdesc, RTE_MAX_ETHPORTS, 0); /* Initialize the screen and logging */ pktgen_init_log(); pktgen_cpu_init(); /* initialize EAL */ ret = rte_eal_init(argc, argv); if (ret < 0) return -1; argc -= ret; argv += ret; pktgen.hz = rte_get_timer_hz(); /* Get the starting HZ value. */ /* parse application arguments (after the EAL ones) */ ret = pktgen_parse_args(argc, argv); if (ret < 0) return -1; pktgen_init_screen((pktgen.flags & ENABLE_THEME_FLAG) ? THEME_ON : THEME_OFF); rte_delay_ms(100); /* Wait a bit for things to settle. */ wr_print_copyright(PKTGEN_APP_NAME, PKTGEN_CREATED_BY); lua_newlib_add(_lua_openlib); /* Open the Lua script handler. */ if ( (pktgen.L = lua_create_instance()) == NULL) { pktgen_log_error("Failed to open Lua pktgen support library"); return -1; } pktgen_log_info(">>> Packet Burst %d, RX Desc %d, TX Desc %d, mbufs/port %d, mbuf cache %d", DEFAULT_PKT_BURST, DEFAULT_RX_DESC, DEFAULT_TX_DESC, MAX_MBUFS_PER_PORT, MBUF_CACHE_SIZE); /* Configure and initialize the ports */ pktgen_config_ports(); pktgen_log_info(""); pktgen_log_info("=== Display processing on lcore %d", rte_lcore_id()); /* launch per-lcore init on every lcore except master and master + 1 lcores */ for (i = 0; i < RTE_MAX_LCORE; i++) { if ( (i == rte_get_master_lcore()) || !rte_lcore_is_enabled(i) ) continue; ret = rte_eal_remote_launch(pktgen_launch_one_lcore, NULL, i); if (ret != 0) pktgen_log_error("Failed to start lcore %d, return %d", i, ret); } rte_delay_ms(1000); /* Wait for the lcores to start up. */ /* Disable printing log messages of level info and below to screen, */ /* erase the screen and start updating the screen again. */ pktgen_log_set_screen_level(LOG_LEVEL_WARNING); wr_scrn_erase(pktgen.scrn->nrows); wr_logo(3, 16, PKTGEN_APP_NAME); wr_splash_screen(3, 16, PKTGEN_APP_NAME, PKTGEN_CREATED_BY); wr_scrn_resume(); pktgen_redisplay(1); rte_timer_setup(); if (pktgen.flags & ENABLE_GUI_FLAG) { if (!wr_scrn_is_paused() ) { wr_scrn_pause(); wr_scrn_cls(); wr_scrn_setw(1); wr_scrn_pos(pktgen.scrn->nrows, 1); } lua_init_socket(pktgen.L, &pktgen.thread, pktgen.hostname, pktgen.socket_port); } pktgen_cmdline_start(); execute_lua_close(pktgen.L); pktgen_stop_running(); wr_scrn_pause(); wr_scrn_setw(1); wr_scrn_printf(100, 1, "\n"); /* Put the cursor on the last row and do a newline. */ /* Wait for all of the cores to stop running and exit. */ rte_eal_mp_wait_lcore(); return 0; }
int pktgen_save(char * path) { port_info_t * info; pkt_seq_t * pkt; range_info_t * range; uint32_t flags; char buff[64]; FILE * fd; int i, j; uint32_t lcore; struct ether_addr eaddr; fd = fopen(path, "w"); if ( fd == NULL ) { return -1; } for(i=0, lcore=0; i<RTE_MAX_LCORE; i++) if ( rte_lcore_is_enabled(i) ) lcore |= (1 << i); fprintf(fd, "#\n# Pktgen - %s\n", pktgen_version()); fprintf(fd, "# %s, %s\n\n", wr_copyright_msg(), wr_powered_by()); // TODO: Determine DPDK arguments for rank and memory, default for now. fprintf(fd, "# Command line arguments: (DPDK args are defaults)\n"); fprintf(fd, "# %s -c %x -n 3 -m 512 --proc-type %s -- ", pktgen.argv[0], lcore, (rte_eal_process_type() == RTE_PROC_PRIMARY)? "primary" : "secondary"); for(i=1; i < pktgen.argc; i++) fprintf(fd, "%s ", pktgen.argv[i]); fprintf(fd, "\n\n"); fprintf(fd, "#######################################################################\n"); fprintf(fd, "# Pktgen Configuration script information:\n"); fprintf(fd, "# GUI socket is %s\n", (pktgen.flags & ENABLE_GUI_FLAG)? "Enabled" : "Not Enabled"); fprintf(fd, "# Enabled Port mask: %08x\n", pktgen.enabled_port_mask); fprintf(fd, "# Flags %08x\n", pktgen.flags); fprintf(fd, "# Number of ports: %d\n", pktgen.nb_ports); fprintf(fd, "# Number ports per page: %d\n", pktgen.nb_ports_per_page); fprintf(fd, "# Coremask 0x%08x\n", pktgen.coremask); fprintf(fd, "# Number descriptors: RX %d TX: %d\n", pktgen.nb_rxd, pktgen.nb_txd); fprintf(fd, "# Promiscuous mode is %s\n\n", (pktgen.flags & PROMISCUOUS_ON_FLAG)? "Enabled" : "Disabled"); fprintf(fd, "# Port Descriptions (-- = blacklisted port):\n"); for(i=0; i < RTE_MAX_ETHPORTS; i++) { if ( strlen(pktgen.portdesc[i]) ) { if ( (pktgen.enabled_port_mask & (1 << i)) == 0 ) strcpy(buff, "--"); else strcpy(buff, "++"); fprintf(fd, "# %s %s\n", buff, pktgen.portdesc[i]); } } fprintf(fd, "\n#######################################################################\n"); fprintf(fd, "# Global configuration:\n"); fprintf(fd, "geometry %dx%d\n", pktgen.scrn->ncols, pktgen.scrn->nrows); fprintf(fd, "mac_from_arp %s\n\n", (pktgen.flags & MAC_FROM_ARP_FLAG)? "enable" : "disable"); for(i=0; i < RTE_MAX_ETHPORTS; i++) { info = &pktgen.info[i]; pkt = &info->seq_pkt[SINGLE_PKT]; range = &info->range; if ( info->tx_burst == 0 ) continue; fprintf(fd, "######################### Port %2d ##################################\n", i); if ( info->transmit_count == 0 ) strcpy(buff, "Forever"); else snprintf(buff, sizeof(buff), "%ld", info->transmit_count); fprintf(fd, "#\n"); flags = rte_atomic32_read(&info->port_flags); fprintf(fd, "# Port: %2d, Burst:%3d, Rate:%3d%%, Flags:%08x, TX Count:%s\n", info->pid, info->tx_burst, info->tx_rate, flags, buff); fprintf(fd, "# SeqCnt:%d, Prime:%d VLAN ID:%04x, ", info->seqCnt, info->prime_cnt, info->vlanid); pktgen_link_state(info->pid, buff, sizeof(buff)); fprintf(fd, "Link: %s\n", buff); fprintf(fd, "#\n# Set up the primary port information:\n"); fprintf(fd, "set %d count %ld\n", info->pid, info->transmit_count); fprintf(fd, "set %d size %d\n", info->pid, pkt->pktSize+FCS_SIZE); fprintf(fd, "set %d rate %d\n", info->pid, info->tx_rate); fprintf(fd, "set %d burst %d\n", info->pid, info->tx_burst); fprintf(fd, "set %d sport %d\n", info->pid, pkt->sport); fprintf(fd, "set %d dport %d\n", info->pid, pkt->dport); fprintf(fd, "set %d prime %d\n", info->pid, info->prime_cnt); fprintf(fd, "set %s %d\n", (pkt->ethType == ETHER_TYPE_IPv4)? "ipv4" : (pkt->ethType == ETHER_TYPE_IPv6)? "ipv6" : (pkt->ethType == ETHER_TYPE_VLAN)? "vlan" : "unknown", i); fprintf(fd, "set %s %d\n", (pkt->ipProto == PG_IPPROTO_TCP)? "tcp" : (pkt->ipProto == PG_IPPROTO_ICMP)? "icmp" : "udp", i); fprintf(fd, "set ip dst %d %s\n", i, inet_ntop4(buff, sizeof(buff), ntohl(pkt->ip_dst_addr), 0xFFFFFFFF)); fprintf(fd, "set ip src %d %s\n", i, inet_ntop4(buff, sizeof(buff), ntohl(pkt->ip_src_addr), pkt->ip_mask)); fprintf(fd, "set mac %d %s\n", info->pid, inet_mtoa(buff, sizeof(buff), &pkt->eth_dst_addr)); fprintf(fd, "vlanid %d %d\n\n", i, pkt->vlanid); fprintf(fd, "#\n# Port flag values:\n"); fprintf(fd, "icmp.echo %d %sable\n", i, (flags & ICMP_ECHO_ENABLE_FLAG)? "en" : "dis"); fprintf(fd, "pcap %d %sable\n", i, (flags & SEND_PCAP_PKTS)? "en" : "dis"); fprintf(fd, "range %d %sable\n", i, (flags & SEND_RANGE_PKTS)? "en" : "dis"); fprintf(fd, "process %d %sable\n", i, (flags & PROCESS_INPUT_PKTS)? "en" : "dis"); fprintf(fd, "tap %d %sable\n", i, (flags & PROCESS_TAP_PKTS)? "en" : "dis"); fprintf(fd, "vlan %d %sable\n\n", i, (flags & SEND_VLAN_ID)? "en" : "dis"); fprintf(fd, "#\n# Range packet information:\n"); fprintf(fd, "src.mac start %d %s\n", i, inet_mtoa(buff, sizeof(buff), inet_h64tom(range->src_mac, &eaddr))); fprintf(fd, "src.mac min %d %s\n", i, inet_mtoa(buff, sizeof(buff), inet_h64tom(range->src_mac_min, &eaddr))); fprintf(fd, "src.mac max %d %s\n", i, inet_mtoa(buff, sizeof(buff), inet_h64tom(range->src_mac_max, &eaddr))); fprintf(fd, "src.mac inc %d %s\n", i, inet_mtoa(buff, sizeof(buff), inet_h64tom(range->src_mac_inc, &eaddr))); fprintf(fd, "dst.mac start %d %s\n", i, inet_mtoa(buff, sizeof(buff), inet_h64tom(range->dst_mac, &eaddr))); fprintf(fd, "dst.mac min %d %s\n", i, inet_mtoa(buff, sizeof(buff), inet_h64tom(range->dst_mac_min, &eaddr))); fprintf(fd, "dst.mac max %d %s\n", i, inet_mtoa(buff, sizeof(buff), inet_h64tom(range->dst_mac_max, &eaddr))); fprintf(fd, "dst.mac inc %d %s\n", i, inet_mtoa(buff, sizeof(buff), inet_h64tom(range->dst_mac_inc, &eaddr))); fprintf(fd, "\n"); fprintf(fd, "src.ip start %d %s\n", i, inet_ntop4(buff, sizeof(buff), ntohl(range->src_ip), 0xFFFFFFFF)); fprintf(fd, "src.ip min %d %s\n", i, inet_ntop4(buff, sizeof(buff), ntohl(range->src_ip_min), 0xFFFFFFFF)); fprintf(fd, "src.ip max %d %s\n", i, inet_ntop4(buff, sizeof(buff), ntohl(range->src_ip_max), 0xFFFFFFFF)); fprintf(fd, "src.ip inc %d %s\n", i, inet_ntop4(buff, sizeof(buff), ntohl(range->src_ip_inc), 0xFFFFFFFF)); fprintf(fd, "\n"); fprintf(fd, "dst.ip start %d %s\n", i, inet_ntop4(buff, sizeof(buff), ntohl(range->dst_ip), 0xFFFFFFFF)); fprintf(fd, "dst.ip min %d %s\n", i, inet_ntop4(buff, sizeof(buff), ntohl(range->dst_ip_min), 0xFFFFFFFF)); fprintf(fd, "dst.ip max %d %s\n", i, inet_ntop4(buff, sizeof(buff), ntohl(range->dst_ip_max), 0xFFFFFFFF)); fprintf(fd, "dst.ip inc %d %s\n", i, inet_ntop4(buff, sizeof(buff), ntohl(range->dst_ip_inc), 0xFFFFFFFF)); fprintf(fd, "\n"); fprintf(fd, "src.port start %d %d\n", i, range->src_port); fprintf(fd, "src.port min %d %d\n", i, range->src_port_min); fprintf(fd, "src.port max %d %d\n", i, range->src_port_max); fprintf(fd, "src.port inc %d %d\n", i, range->src_port_inc); fprintf(fd, "\n"); fprintf(fd, "dst.port start %d %d\n", i, range->dst_port); fprintf(fd, "dst.port min %d %d\n", i, range->dst_port_min); fprintf(fd, "dst.port max %d %d\n", i, range->dst_port_max); fprintf(fd, "dst.port inc %d %d\n", i, range->dst_port_inc); fprintf(fd, "\n"); fprintf(fd, "vlan.id start %d %d\n", i, range->vlan_id); fprintf(fd, "vlan.id min %d %d\n", i, range->vlan_id_min); fprintf(fd, "vlan.id max %d %d\n", i, range->vlan_id_max); fprintf(fd, "vlan.id inc %d %d\n", i, range->vlan_id_inc); fprintf(fd, "\n"); fprintf(fd, "pkt.size start %d %d\n", i, range->pkt_size + FCS_SIZE); fprintf(fd, "pkt.size min %d %d\n", i, range->pkt_size_min + FCS_SIZE); fprintf(fd, "pkt.size max %d %d\n", i, range->pkt_size_max + FCS_SIZE); fprintf(fd, "pkt.size inc %d %d\n\n", i, range->pkt_size_inc); fprintf(fd, "#\n# Set up the sequence data for the port.\n"); fprintf(fd, "set %d seqCnt %d\n", info->pid, info->seqCnt); for(j=0; j<info->seqCnt; j++) { pkt = &info->seq_pkt[j]; fprintf(fd, "seq %d %d %s ", j, i, inet_mtoa(buff, sizeof(buff), &pkt->eth_dst_addr)); fprintf(fd, "%s ", inet_mtoa(buff, sizeof(buff), &pkt->eth_src_addr)); fprintf(fd, "%s ", inet_ntop4(buff, sizeof(buff), htonl(pkt->ip_dst_addr), 0xFFFFFFFF)); fprintf(fd, "%s ", inet_ntop4(buff, sizeof(buff), htonl(pkt->ip_src_addr), pkt->ip_mask)); fprintf(fd, "%d %d %s %s %d %d\n", pkt->sport, pkt->dport, (pkt->ethType == ETHER_TYPE_IPv4)? "ipv4" : (pkt->ethType == ETHER_TYPE_IPv6)? "ipv6" : (pkt->ethType == ETHER_TYPE_VLAN)? "vlan" : "Other", (pkt->ipProto == PG_IPPROTO_TCP)? "tcp" : (pkt->ipProto == PG_IPPROTO_ICMP)? "icmp" : "udp", pkt->vlanid, pkt->pktSize+FCS_SIZE); } if ( pktgen.info[i].pcap ) { fprintf(fd, "#\n# PCAP port %d\n", i); fprintf(fd, "# Packet count: %d\n", pktgen.info[i].pcap->pkt_count); fprintf(fd, "# Filename : %s\n", pktgen.info[i].pcap->filename); } fprintf(fd, "\n"); } fprintf(fd, "################################ Done #################################\n"); fclose(fd); return 0; }