/*

* Sigma Control API DUT (sniffer)

* Copyright (c) 2013-2014, Qualcomm Atheros, Inc.

* All Rights Reserved.

* Licensed under the Clear BSD license. See README for more details.

*/

#include "sigma_dut.h"

#include <signal.h>

#include <sys/stat.h>

#include <sys/wait.h>

static void capture_process(const char *ifname, const char *filename)

{

char *env[] = { NULL };

char *argv[] = { "sigma_dut[capture]", "-i", strdup(ifname),

"-w", strdup(filename), NULL };

execve("/usr/bin/dumpcap", argv, env);

perror("execve");

exit(EXIT_FAILURE);

}

static int cmd_sniffer_control_start(struct sigma_dut *dut,

struct sigma_conn *conn,

struct sigma_cmd *cmd)

{

const char *filename = get_param(cmd, "filename");

int res;

pid_t pid;

if (dut->sniffer_pid) {

sigma_dut_print(dut, DUT_MSG_INFO, "Sniffer was already capturing - restart based on new parameters");

sniffer_close(dut);

}

if (filename == NULL) {

send_resp(dut, conn, SIGMA_ERROR,

"errorCode,Missing filename argument");

return 0;

}

if (strchr(filename, '/')) {

send_resp(dut, conn, SIGMA_ERROR, "errorCode,Invalid filename");

return 0;

}

res = cmd_wlantest_set_channel(dut, conn, cmd);

if (res != 1)

return res;

mkdir("Captures", S_IRWXU | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH);

sigma_dut_print(dut, DUT_MSG_INFO, "Starting sniffer process");

snprintf(dut->sniffer_filename, sizeof(dut->sniffer_filename),

"Captures/%s", filename);

pid = fork();

if (pid < 0) {

send_resp(dut, conn, SIGMA_ERROR,

"errorCode,Failed to fork sniffer process");

return 0;

}

if (pid == 0) {

capture_process(dut->sniffer_ifname, dut->sniffer_filename);

return 0;

}

dut->sniffer_pid = pid;

return 1;

}

void sniffer_close(struct sigma_dut *dut)

{

if (!dut->sniffer_pid)

return;

if (kill(dut->sniffer_pid, SIGTERM) < 0) {

printf("Failed to kill sniffer process: %s\n", strerror(errno));

dut->sniffer_pid = 0;

return;

}

waitpid(dut->sniffer_pid, NULL, 0);

if (dut->sniffer_filename[0]) {

chmod(dut->sniffer_filename,

S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH);

dut->sniffer_filename[0] = '\0';

}

dut->sniffer_pid = 0;

}

static int cmd_sniffer_control_stop(struct sigma_dut *dut,

struct sigma_conn *conn,

struct sigma_cmd *cmd)

{

if (!dut->sniffer_pid) {

send_resp(dut, conn, SIGMA_ERROR,

"errorCode,Sniffer was not capturing");

return 0;

}

sniffer_close(dut);

return 1;

}

static int cmd_sniffer_control_field_check(struct sigma_dut *dut,

struct sigma_conn *conn,

struct sigma_cmd *cmd)

{

const char *filename = get_param(cmd, "filename");

const char *framename = get_param(cmd, "framename");

const char *srcmac = get_param(cmd, "srcmac");

const char *wsc_state = get_param(cmd, "WSC_State");

const char *pvb_bit = get_param(cmd, "pvb_bit");

const char *moredata_bit = get_param(cmd, "MoreData_bit");

const char *eosp_bit = get_param(cmd, "EOSP_bit");

char buf[2000], *pos;

FILE *f;

if (filename == NULL || srcmac == NULL)

return -1;

if (strchr(filename, '/')) {

send_resp(dut, conn, SIGMA_ERROR, "errorCode,Invalid filename");

return 0;

}

if (!file_exists("sniffer-control-field-check.py")) {

send_resp(dut, conn, SIGMA_ERROR, "errorCode,sniffer-control-field-check.py not found");

return 0;

}

snprintf(buf, sizeof(buf),

"./sniffer-control-field-check.py FileName=Captures/%s SrcMac=%s%s%s%s%s%s%s%s%s%s%s",

filename, srcmac,

framename ? " FrameName=" : "", framename ? framename : "",

wsc_state ? " WSC_State=" : "", wsc_state ? wsc_state : "",

moredata_bit ? " MoreData_bit=" : "",

moredata_bit ? moredata_bit : "",

eosp_bit ? " EOSP_bit=" : "", eosp_bit ? eosp_bit : "",

pvb_bit ? " pvb_bit=" : "", pvb_bit ? pvb_bit : "");

sigma_dut_print(dut, DUT_MSG_INFO, "Run: %s", buf);

f = popen(buf, "r");

if (f == NULL) {

send_resp(dut, conn, SIGMA_ERROR,

"errorCode,Failed to run sniffer helper");

return 0;

}

if (!fgets(buf, sizeof(buf), f)) {

pclose(f);

send_resp(dut, conn, SIGMA_ERROR,

"errorCode,Failed extract response from sniffer helper");

return 0;

}

pos = strchr(buf, '\n');

if (pos)

*pos = '\0';

pclose(f);

send_resp(dut, conn, SIGMA_COMPLETE, buf);

return 0;

}

static int cmd_sniffer_get_info(struct sigma_dut *dut, struct sigma_conn *conn,

struct sigma_cmd *cmd)

{

char buf[200];

snprintf(buf, sizeof(buf), "WfaSnifferVersion,SigmaSniffer-foo,SnifferSTA,foo,DeviceSwInfo,foo,WiresharkVersion,foo");

send_resp(dut, conn, SIGMA_COMPLETE, buf);

return 0;

}

static int cmd_sniffer_control_filter_capture(struct sigma_dut *dut,

struct sigma_conn *conn,

struct sigma_cmd *cmd)

{

const char *infile = get_param(cmd, "InFile");

const char *outfile = get_param(cmd, "OutFile");

const char *srcmac = get_param(cmd, "SrcMac");

const char *framename = get_param(cmd, "FrameName");

const char *nframes = get_param(cmd, "Nframes");

const char *hasfield = get_param(cmd, "HasField");

const char *datalen = get_param(cmd, "Datalen");

char buf[500], *pos;

FILE *f;

if (infile == NULL || outfile == NULL || srcmac == NULL ||

nframes == NULL)

return -1;

if (strchr(infile, '/') || strchr(outfile, '/'))

return -1;

if (!file_exists("sniffer-control-filter-capture.py")) {

send_resp(dut, conn, SIGMA_ERROR, "errorCode,sniffer-control-filter-capture.py not found");

return 0;

}

snprintf(buf, sizeof(buf),

"./sniffer-control-filter-capture.py InFile=Captures/%s OutFile=Captures/%s SrcMac=%s%s%s Nframes=%s%s%s%s%s",

infile, outfile, srcmac,

framename ? " FrameName=" : "", framename ? framename : "",

nframes,

hasfield ? " HasField=" : "", hasfield ? hasfield : "",

datalen ? " Datalen=" : "", datalen ? datalen : "");

sigma_dut_print(dut, DUT_MSG_INFO, "Run: %s", buf);

f = popen(buf, "r");

if (f == NULL) {

send_resp(dut, conn, SIGMA_ERROR,

"errorCode,Failed to run sniffer helper");

return 0;

}

if (!fgets(buf, sizeof(buf), f)) {

pclose(f);

send_resp(dut, conn, SIGMA_ERROR,

"errorCode,Failed extract response from sniffer helper");

return 0;

}

pos = strchr(buf, '\n');

if (pos)

*pos = '\0';

pclose(f);

send_resp(dut, conn, SIGMA_COMPLETE, buf);

return 0;

}

static int cmd_sniffer_get_field_value(struct sigma_dut *dut,

struct sigma_conn *conn,

struct sigma_cmd *cmd)

{

const char *infile = get_param(cmd, "FileName");

const char *srcmac = get_param(cmd, "SrcMac");

const char *framename = get_param(cmd, "FrameName");

const char *fieldname = get_param(cmd, "FieldName");

char buf[500], *pos;

FILE *f;

if (infile == NULL || srcmac == NULL || framename == NULL ||

fieldname == NULL)

return -1;

if (!file_exists("sniffer-get-field-value.py")) {

send_resp(dut, conn, SIGMA_ERROR, "errorCode,sniffer-get-field-value.py not found");

return 0;

}

snprintf(buf, sizeof(buf),

"./sniffer-get-field-value.py FileName=Captures/%s SrcMac=%s FrameName=%s FieldName=%s",

infile, srcmac, framename, fieldname);

sigma_dut_print(dut, DUT_MSG_INFO, "Run: %s", buf);

f = popen(buf, "r");

if (f == NULL) {

send_resp(dut, conn, SIGMA_ERROR,

"errorCode,Failed to run sniffer helper");

return 0;

}

if (!fgets(buf, sizeof(buf), f)) {

pclose(f);

send_resp(dut, conn, SIGMA_ERROR,

"errorCode,Failed extract response from sniffer helper");

return 0;

}

pos = strchr(buf, '\n');

if (pos)

*pos = '\0';

pclose(f);

send_resp(dut, conn, SIGMA_COMPLETE, buf);

return 0;

}

static int cmd_sniffer_check_p2p_noa_duration(struct sigma_dut *dut,

struct sigma_conn *conn,

struct sigma_cmd *cmd)

{

FILE *f;

char buf[200], *pos;

const char *infile = get_param(cmd, "FileName");

const char *bssid = get_param(cmd, "bssid");

const char *srcmac = get_param(cmd, "srcmac");

const char *destmac = get_param(cmd, "destmac");

if (infile == NULL || bssid == NULL || srcmac == NULL ||

destmac == NULL)

return -1;

if (!file_exists("sniffer-check-p2p-noa-duration.py")) {

send_resp(dut, conn, SIGMA_ERROR, "errorCode,sniffer-check-p2p-noa-duration.py not found");

return 0;

}

snprintf(buf, sizeof(buf),

"./sniffer-check-p2p-noa-duration.py Captures/%s %s %s %s",

infile, bssid, srcmac, destmac);

sigma_dut_print(dut, DUT_MSG_INFO, "Run: %s", buf);

f = popen(buf, "r");

if (f == NULL) {

send_resp(dut, conn, SIGMA_ERROR,

"errorCode,Failed to run sniffer check");

return 0;

}

if (!fgets(buf, sizeof(buf), f)) {

pclose(f);

send_resp(dut, conn, SIGMA_ERROR,

"errorCode,Failed extract response from sniffer check");

return 0;

}

pos = strchr(buf, '\n');

if (pos)

*pos = '\0';

pclose(f);

send_resp(dut, conn, SIGMA_COMPLETE, buf);

return 0;

}

static int cmd_sniffer_check_p2p_opps_client(struct sigma_dut *dut,

struct sigma_conn *conn,

struct sigma_cmd *cmd)

{

char buf[200];

/* TODO */

snprintf(buf, sizeof(buf), "FilterStatus,SUCCESS");

send_resp(dut, conn, SIGMA_COMPLETE, buf);

return 0;

}

static int cmd_sniffer_check_frame_field(struct sigma_dut *dut,

struct sigma_conn *conn,

struct sigma_cmd *cmd)

{

char buf[200];

/* TODO */

snprintf(buf, sizeof(buf), "FilterStatus,SUCCESS");

send_resp(dut, conn, SIGMA_COMPLETE, buf);

return 0;

}

void sniffer_register_cmds(void)

{

sigma_dut_reg_cmd("sniffer_control_start", NULL,

cmd_sniffer_control_start);

sigma_dut_reg_cmd("sniffer_control_stop", NULL,

cmd_sniffer_control_stop);

sigma_dut_reg_cmd("sniffer_control_field_check", NULL,

cmd_sniffer_control_field_check);

sigma_dut_reg_cmd("sniffer_get_info", NULL, cmd_sniffer_get_info);

sigma_dut_reg_cmd("sniffer_control_filter_capture", NULL,

cmd_sniffer_control_filter_capture);

sigma_dut_reg_cmd("wfa_sniffer_control_filter_capture", NULL,

cmd_sniffer_control_filter_capture);

sigma_dut_reg_cmd("sniffer_get_field_value", NULL,

cmd_sniffer_get_field_value);

sigma_dut_reg_cmd("sniffer_check_p2p_NoA_duration", NULL,

cmd_sniffer_check_p2p_noa_duration);

sigma_dut_reg_cmd("sniffer_check_p2p_opps_client", NULL,

cmd_sniffer_check_p2p_opps_client);

sigma_dut_reg_cmd("sniffer_check_frame_field", NULL,

cmd_sniffer_check_frame_field);

}