M7350/wlan/utils/sigma-dut/traffic.c
2024-09-09 08:57:42 +00:00

246 lines
5.3 KiB
C

/*
* Sigma Control API DUT (station/AP)
* Copyright (c) 2010, Atheros Communications, Inc.
* Copyright (c) 2011-2013, Qualcomm Atheros, Inc.
* All Rights Reserved.
* Licensed under the Clear BSD license. See README for more details.
*/
#include "sigma_dut.h"
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <signal.h>
#include <ctype.h>
#ifdef ANDROID
#define SHELL "/system/bin/sh"
#else /* ANDROID */
#define SHELL "/bin/sh"
#endif /* ANDROID */
static int is_ipv6_addr(const char *str)
{
const char *pos = str;
while (*pos) {
if (*pos != ':' && (*pos < '0' || *pos > '9') &&
(*pos < 'a' || *pos > 'f') &&
(*pos < 'A' || *pos > 'F'))
return 0;
pos++;
}
return 1;
}
static int cmd_traffic_send_ping(struct sigma_dut *dut,
struct sigma_conn *conn,
struct sigma_cmd *cmd)
{
const char *dst, *val;
int size, dur, pkts;
int id;
char resp[100];
float interval;
double rate;
FILE *f;
char buf[100];
int type = 1;
int dscp = 0, use_dscp = 0;
char extra[100], int_arg[100];
val = get_param(cmd, "Type");
if (!val)
val = get_param(cmd, "IPType");
if (val)
type = atoi(val);
if (type != 1 && type != 2) {
send_resp(dut, conn, SIGMA_ERROR,
"ErrorCode,Unsupported address type");
return 0;
}
dst = get_param(cmd, "destination");
if (dst == NULL || (type == 1 && !is_ip_addr(dst)) ||
(type == 2 && !is_ipv6_addr(dst)))
return -1;
val = get_param(cmd, "frameSize");
if (val == NULL)
return -1;
size = atoi(val);
val = get_param(cmd, "frameRate");
if (val == NULL)
return -1;
rate = atof(val);
if (rate <= 0)
return -1;
val = get_param(cmd, "duration");
if (val == NULL)
return -1;
dur = atoi(val);
if (dur <= 0)
dur = 3600;
pkts = dur * rate;
interval = (float) 1 / rate;
if (interval > 100000)
return -1;
val = get_param(cmd, "DSCP");
if (val) {
dscp = atoi(val);
if (dscp < 0 || dscp > 63) {
send_resp(dut, conn, SIGMA_ERROR,
"ErrorCode,Invalid DSCP value");
return 0;
}
use_dscp = 1;
}
id = dut->next_streamid++;
snprintf(buf, sizeof(buf), SIGMA_TMPDIR "/sigma_dut-ping.%d", id);
unlink(buf);
snprintf(buf, sizeof(buf), SIGMA_TMPDIR "/sigma_dut-ping-pid.%d", id);
unlink(buf);
sigma_dut_print(dut, DUT_MSG_DEBUG, "Send ping: pkts=%d interval=%f "
"streamid=%d",
pkts, interval, id);
f = fopen(SIGMA_TMPDIR "/sigma_dut-ping.sh", "w");
if (f == NULL)
return -2;
extra[0] = '\0';
if (use_dscp) {
snprintf(extra, sizeof(extra), " -Q 0x%02x",
dscp << 2);
}
int_arg[0] = '\0';
if (rate != 1)
snprintf(int_arg, sizeof(int_arg), " -i %f", interval);
fprintf(f, "#!" SHELL "\n"
"ping%s -c %d%s -s %d%s -q %s > " SIGMA_TMPDIR
"/sigma_dut-ping.%d &\n"
"echo $! > " SIGMA_TMPDIR "/sigma_dut-ping-pid.%d\n",
type == 2 ? "6" : "", pkts, int_arg, size, extra, dst, id, id);
fclose(f);
if (chmod(SIGMA_TMPDIR "/sigma_dut-ping.sh",
S_IRUSR | S_IWUSR | S_IXUSR) < 0)
return -2;
if (system(SIGMA_TMPDIR "/sigma_dut-ping.sh") != 0) {
sigma_dut_print(dut, DUT_MSG_ERROR, "Failed to start ping");
return -2;
}
unlink(SIGMA_TMPDIR "/sigma_dut-ping.sh");
snprintf(resp, sizeof(resp), "streamID,%d", id);
send_resp(dut, conn, SIGMA_COMPLETE, resp);
return 0;
}
static int cmd_traffic_stop_ping(struct sigma_dut *dut,
struct sigma_conn *conn,
struct sigma_cmd *cmd)
{
const char *val;
int id, pid;
FILE *f;
char buf[100];
int res_found = 0, sent = 0, received = 0;
val = get_param(cmd, "streamID");
if (val == NULL)
return -1;
id = atoi(val);
snprintf(buf, sizeof(buf), SIGMA_TMPDIR "/sigma_dut-ping-pid.%d", id);
f = fopen(buf, "r");
if (f == NULL) {
send_resp(dut, conn, SIGMA_ERROR,
"ErrorCode,Unknown streamID");
return 0;
}
if (fscanf(f, "%d", &pid) != 1 || pid <= 0) {
sigma_dut_print(dut, DUT_MSG_ERROR, "No PID for ping process");
fclose(f);
unlink(buf);
return -2;
}
fclose(f);
unlink(buf);
sigma_dut_print(dut, DUT_MSG_DEBUG, "Ping process pid %d", pid);
if (kill(pid, SIGINT) < 0 && errno != ESRCH) {
sigma_dut_print(dut, DUT_MSG_DEBUG, "kill failed: %s",
strerror(errno));
}
usleep(250000);
snprintf(buf, sizeof(buf), SIGMA_TMPDIR "/sigma_dut-ping.%d", id);
f = fopen(buf, "r");
if (f == NULL) {
sigma_dut_print(dut, DUT_MSG_DEBUG,
"No ping result file found");
send_resp(dut, conn, SIGMA_COMPLETE, "sent,0,replies,0");
return 0;
}
while (fgets(buf, sizeof(buf), f)) {
char *pos;
pos = strstr(buf, " packets transmitted");
if (pos) {
pos--;
while (pos > buf && isdigit(pos[-1]))
pos--;
sent = atoi(pos);
res_found = 1;
}
pos = strstr(buf, " packets received");
if (pos == NULL)
pos = strstr(buf, " received");
if (pos) {
pos--;
while (pos > buf && isdigit(pos[-1]))
pos--;
received = atoi(pos);
res_found = 1;
}
}
fclose(f);
snprintf(buf, sizeof(buf), SIGMA_TMPDIR "/sigma_dut-ping.%d", id);
unlink(buf);
if (!res_found) {
sigma_dut_print(dut, DUT_MSG_DEBUG,
"No ping results found");
send_resp(dut, conn, SIGMA_COMPLETE, "sent,0,replies,0");
return 0;
}
snprintf(buf, sizeof(buf), "sent,%d,replies,%d", sent, received);
send_resp(dut, conn, SIGMA_COMPLETE, buf);
return 0;
}
void traffic_register_cmds(void)
{
sigma_dut_reg_cmd("traffic_send_ping", NULL, cmd_traffic_send_ping);
sigma_dut_reg_cmd("traffic_stop_ping", NULL, cmd_traffic_stop_ping);
}