forked from osmocom/wireshark
![Luis Ontanon](/assets/img/avatar_default.png)
- improve the parameter api - Add a parameter set message to the NEW child message to set parameters in bulk in the child - some more harvestig from tshark svn path=/trunk/; revision=50393
416 lines
9.6 KiB
C
416 lines
9.6 KiB
C
/* echld-test.c
|
|
* basic test framework for echld
|
|
*
|
|
* $Id$
|
|
*
|
|
* Wireshark - Network traffic analyzer
|
|
* By Gerald Combs <gerald@wireshark.org>
|
|
* Copyright 1998 Gerald Combs
|
|
*
|
|
* Copyright (c) 2013 by Luis Ontanon <luis@ontanon.org>
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License
|
|
* as published by the Free Software Foundation; either version 2
|
|
* of the License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*/
|
|
|
|
#include "config.h"
|
|
|
|
#ifdef HAVE_FCNTL_H
|
|
#include <fcntl.h>
|
|
#endif
|
|
|
|
#ifdef HAVE_SYS_TYPES_H
|
|
# include <sys/types.h>
|
|
#endif
|
|
|
|
#include <sys/time.h>
|
|
#include <sys/uio.h>
|
|
|
|
#ifdef HAVE_UNISTD_H
|
|
#include <unistd.h>
|
|
#endif
|
|
|
|
#include <signal.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include <glib.h>
|
|
#include <glib/gprintf.h>
|
|
|
|
#include "echld/echld.h"
|
|
#include "echld/echld-util.h"
|
|
|
|
#include "epan/epan.h"
|
|
#include "wsutil/str_util.h"
|
|
|
|
typedef char* (*cmd_cb_t)(char** params, char** err);
|
|
|
|
typedef struct _cmd_t {
|
|
const char* txt;
|
|
cmd_cb_t cb;
|
|
int args_taken;
|
|
const char* help;
|
|
} cmd_t;
|
|
|
|
|
|
#define MAX_PARAMSETS 16
|
|
|
|
static enc_msg_t* paramsets[MAX_PARAMSETS] = {
|
|
NULL,NULL,NULL,NULL, NULL,NULL,NULL,NULL,
|
|
NULL,NULL,NULL,NULL, NULL,NULL,NULL,NULL
|
|
};
|
|
|
|
static int nps = 0;
|
|
|
|
static char* ps_cmd(char** pars _U_, char** err _U_) {
|
|
int n_id = nps++;
|
|
|
|
if (n_id >= MAX_PARAMSETS) {
|
|
*err = g_strdup("Max Num of Paramsets reached");
|
|
return NULL;
|
|
}
|
|
|
|
paramsets[n_id] = echld_new_child_params();
|
|
|
|
return g_strdup_printf("New Paramset ps_id=%d",n_id);
|
|
}
|
|
|
|
static char* psadd_cmd(char** params _U_, char** err) {
|
|
int ps_id = (int) strtol(params[1], NULL, 10);
|
|
|
|
if (ps_id >= nps) {
|
|
*err = g_strdup_printf("No paramset pd_is=%d",ps_id);
|
|
return NULL;
|
|
}
|
|
|
|
echld_new_child_params_add_params(paramsets[ps_id], params[2], params[3], NULL);
|
|
|
|
return g_strdup_printf("PSAdd ps_id=%d %s='%s'", ps_id, params[2], params[3]);
|
|
}
|
|
|
|
static char* psmerge_cmd(char** params, char** err) {
|
|
int ps1_id = (int) strtol(params[1], NULL, 10);
|
|
int ps2_id = (int) strtol(params[2], NULL, 10);
|
|
int n_id;
|
|
|
|
if (ps1_id >= nps) {
|
|
*err = g_strdup_printf("No paramset pd_is=%d",ps1_id);
|
|
return NULL;
|
|
}
|
|
|
|
if (ps2_id >= nps) {
|
|
*err = g_strdup_printf("No paramset pd_is=%d",ps2_id);
|
|
return NULL;
|
|
}
|
|
|
|
n_id = nps++;
|
|
|
|
if (n_id >= MAX_PARAMSETS) {
|
|
*err = g_strdup_printf("Max Num of Paramsets reached");
|
|
return NULL;
|
|
}
|
|
|
|
paramsets[n_id] = echld_new_child_params_merge(paramsets[ps1_id],paramsets[ps2_id]);
|
|
|
|
return g_strdup_printf("Merged Paramset ps1_id=%d ps2_id=%d ps_id=%d",ps1_id, ps2_id, n_id);
|
|
}
|
|
|
|
static char* new_child_cmd(char** params, char** err) {
|
|
int ps_id = (int) strtol(params[1], NULL, 10);
|
|
int child;
|
|
|
|
if (ps_id >= nps) {
|
|
*err = g_strdup_printf("No paramset pd_is=%d",ps_id);
|
|
return NULL;
|
|
}
|
|
|
|
child = echld_new(paramsets[ps_id],NULL);
|
|
|
|
if (child <= 0) {
|
|
*err = g_strdup("No child\n");
|
|
return NULL;
|
|
}
|
|
|
|
return g_strdup_printf("New chld_id=%d\n",child);;
|
|
}
|
|
|
|
|
|
void ping_cb(long usec, void* data) {
|
|
int ping_id = *((int*)data);
|
|
|
|
if (usec >= 0) {
|
|
fprintf(stdout, "Ping ping_id=%d returned in %dus\n",ping_id,(int)usec);
|
|
} else {
|
|
fprintf(stdout, "Ping ping_id=%d erored\n",ping_id);
|
|
}
|
|
|
|
g_free(data);
|
|
}
|
|
|
|
|
|
static char* ping_cmd(char** params, char** err) {
|
|
int child = (int) strtol(params[1], NULL, 10);
|
|
static int ping_id = 0;
|
|
int* ping_data = g_new(int,1);
|
|
|
|
*ping_data = ping_id++;
|
|
|
|
if (!echld_ping(child,ping_cb,ping_data)) {
|
|
*err = g_strdup_printf("Could not send ping child=%d",child);
|
|
return NULL;
|
|
} else {
|
|
return g_strdup_printf("Ping sent child=%d",child);
|
|
}
|
|
}
|
|
|
|
void param_cb(const char* param, const char* value, const char* error, void* data _U_) {
|
|
if (error) {
|
|
fprintf(stdout, "Param Set Error msg=%s\n", error );
|
|
} else {
|
|
fprintf(stdout, "Param: param='%s' val='%s'\n", param, value );
|
|
}
|
|
}
|
|
|
|
static char* set_cmd(char** params, char** err) {
|
|
int child = (int) strtol(params[1], NULL, 10);
|
|
char* param = params[2];
|
|
char* value = params[3];
|
|
|
|
if ( ! echld_set_param(child,param,value,param_cb,NULL) ) {
|
|
*err = g_strdup_printf("Failed to SET child=%d param='%s' value='%s'",child,param,value);
|
|
return NULL;
|
|
} else {
|
|
return g_strdup_printf("Set command sent child=%d param='%s' value='%s'",child,param,value);
|
|
}
|
|
}
|
|
|
|
static char* get_cmd(char** params, char** err) {
|
|
int child = (int) strtol(params[1], NULL, 10);
|
|
char* param = params[2];
|
|
|
|
if ( ! echld_get_param(child,param,param_cb,NULL) ) {
|
|
*err = g_strdup_printf("Failed to GET child=%d param='%s'",child,param);
|
|
return NULL;
|
|
} else {
|
|
return g_strdup_printf("Get command sent child=%d param='%s'",child,param);
|
|
}
|
|
}
|
|
|
|
static void close_cb(const char* error, void* data) {
|
|
if (error) {
|
|
fprintf(stdout, "Close Error msg=%s\n", error );
|
|
} else {
|
|
fprintf(stdout, "Closed: child=%d\n", *((int*)data) );
|
|
}
|
|
}
|
|
|
|
static char* close_cmd(char** params, char** err) {
|
|
int child = (int) strtol(params[1], NULL, 10);
|
|
int* cmdp = g_new(int,1);
|
|
*cmdp = child;
|
|
if ( ! echld_close(child,close_cb,cmdp) ) {
|
|
*err = g_strdup_printf("Could not close child=%d",child);
|
|
return NULL;
|
|
} else {
|
|
return g_strdup_printf("CLose command sent child=%d",child);
|
|
}
|
|
}
|
|
int keep_going = 1;
|
|
|
|
static char* quit_cmd(char** params _U_, char** err _U_) {
|
|
keep_going = 0;
|
|
return g_strdup("Quitting");
|
|
}
|
|
|
|
static char* help_cmd(char**, char**);
|
|
|
|
static char* open_file_cmd(char** pars _U_, char** err _U_) {
|
|
*err = g_strdup("Not Implemented");
|
|
return NULL;
|
|
}
|
|
|
|
static char* prepare_capture_cmd(char** pars _U_, char** err _U_) {
|
|
*err = g_strdup("Not Implemented");
|
|
return NULL;
|
|
}
|
|
|
|
static char* start_capture_cmd(char** pars _U_, char** err _U_) {
|
|
*err = g_strdup("Not Implemented");
|
|
return NULL;
|
|
}
|
|
|
|
static char* get_sum_cmd(char** pars _U_, char** err _U_) {
|
|
*err = g_strdup("Not Implemented");
|
|
return NULL;
|
|
}
|
|
|
|
static char* get_tree_cmd(char** pars _U_, char** err _U_) {
|
|
*err = g_strdup("Not Implemented");
|
|
return NULL;
|
|
}
|
|
|
|
static char* get_buf_cmd(char** pars _U_, char** err _U_) {
|
|
*err = g_strdup("Not Implemented");
|
|
return NULL;
|
|
}
|
|
|
|
static char* stop_cmd(char** pars _U_, char** err _U_) {
|
|
*err = g_strdup("Not Implemented");
|
|
return NULL;
|
|
}
|
|
|
|
static char* note_cmd(char** pars _U_, char** err _U_) {
|
|
*err = g_strdup("Not Implemented");
|
|
return NULL;
|
|
}
|
|
|
|
static char* apply_cmd(char** pars _U_, char** err _U_) {
|
|
*err = g_strdup("Not Implemented");
|
|
return NULL;
|
|
}
|
|
|
|
static char* save_cmd(char** pars _U_, char** err _U_) {
|
|
*err = g_strdup("Not Implemented");
|
|
return NULL;
|
|
}
|
|
|
|
static char* run_cmd(char** pars, char** err _U_);
|
|
|
|
|
|
|
|
|
|
cmd_t commands[] = {
|
|
{ "QUIT", quit_cmd, 0, "QUIT"},
|
|
{ "HELP", help_cmd, 0, "HELP"},
|
|
{ "RUN", run_cmd, 1, "RUN filename"},
|
|
{ "PS", ps_cmd, 1, "PS [dummy]"},
|
|
{ "PSADD", psadd_cmd, 4, "PSADD ps_id param value"},
|
|
{ "PSMERGE", psmerge_cmd, 4, "PSMERGE ps_id ps_id"},
|
|
{ "NEW", new_child_cmd, 1, "NEW ps_id"},
|
|
{ "PING", ping_cmd, 1, "PING child_id"},
|
|
{ "SET", set_cmd, 3, "SET child_id param_name param_val"},
|
|
{ "GET", get_cmd, 2, "GET child_id param_name"},
|
|
{ "CLOSE", close_cmd, 1, "CLOSE child_id"},
|
|
{ "FILE", open_file_cmd, 2, "FILE child_id filename"},
|
|
{ "PREP", prepare_capture_cmd, 2, "PREP child_id intfname params"},
|
|
{ "START",start_capture_cmd,1,"START child_id"},
|
|
{ "SUM", get_sum_cmd,2,"SUM child_id packet_range"},
|
|
{ "TREE", get_tree_cmd,2,"TREE child_id packet_range"},
|
|
{ "BUFF", get_buf_cmd,2,"BUFF child_id buffer_name"},
|
|
{ "STOP", stop_cmd,1,"STOP child_id"},
|
|
{ "NOTE", note_cmd,1,"NOTE child_id framenum note..."},
|
|
{ "APPLY", apply_cmd,1,"APPLY child_id dfilter"},
|
|
{ "SAVE", save_cmd,1,"SAVE child_id filename params"},
|
|
{ NULL, NULL, 0, NULL }
|
|
};
|
|
|
|
static char* help_cmd(char** params _U_, char** err _U_) {
|
|
GString* out = g_string_new("Commands:\n");
|
|
cmd_t* c = commands;
|
|
char* s;
|
|
|
|
for (;c->txt;c++) {
|
|
g_string_append_printf(out,"%s\n",c->help);
|
|
}
|
|
s = out->str;
|
|
g_string_free(out,FALSE);
|
|
return s;
|
|
}
|
|
|
|
|
|
static int invoke_cmd(FILE* in_fp) {
|
|
size_t len;
|
|
char* cmd_line;
|
|
|
|
if(( cmd_line = fgetln(in_fp,&len) )) {
|
|
cmd_t* c = commands;
|
|
cmd_line[len] = 0;
|
|
g_strchomp(cmd_line);
|
|
|
|
for (;c->txt;c++) {
|
|
if ( strcasestr(cmd_line, c->txt) == cmd_line ) {
|
|
char** params = g_strsplit(cmd_line, " ", c->args_taken+1);
|
|
char* err = NULL;
|
|
char* str = c->cb(params,&err);
|
|
|
|
if (err) {
|
|
fprintf(stdout, "Error: %s\n", err);
|
|
g_free(err);
|
|
} else {
|
|
fprintf(stdout, "%s\n", str);
|
|
g_free(str);
|
|
}
|
|
|
|
g_strfreev(params);
|
|
return TRUE;
|
|
}
|
|
}
|
|
|
|
fprintf(stdout, "Error: no such command %s\n", cmd_line);
|
|
return TRUE;
|
|
} else {
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
static char* run_cmd(char** pars, char** err _U_) {
|
|
FILE* fp = fopen(pars[1],"r");
|
|
while(invoke_cmd(fp)) { ; }
|
|
fclose(fp);
|
|
return NULL;
|
|
}
|
|
|
|
|
|
int got_param = 0;
|
|
|
|
|
|
int main(int argc _U_, char** argv _U_) {
|
|
struct timeval tv;
|
|
int tot_cycles = 0;
|
|
echld_init_t init = {ECHLD_ENCODING_JSON,argv[0],main,NULL,NULL,NULL,NULL};
|
|
|
|
|
|
tv.tv_sec = 5;
|
|
tv.tv_usec = 0;
|
|
|
|
echld_set_parent_dbg_level(5);
|
|
|
|
echld_initialize(&init);
|
|
|
|
do {
|
|
fd_set rfds;
|
|
fd_set efds;
|
|
int nfds;
|
|
|
|
FD_ZERO(&rfds);
|
|
FD_ZERO(&efds);
|
|
FD_SET(0,&rfds);
|
|
FD_SET(0,&efds);
|
|
|
|
nfds = echld_select(FD_SETSIZE, &rfds, NULL, &efds, &tv);
|
|
|
|
if (FD_ISSET(0,&rfds)) {
|
|
invoke_cmd(stdin);
|
|
}
|
|
|
|
tot_cycles++;
|
|
} while( keep_going );
|
|
|
|
fprintf(stderr, "Done: tot_cycles=%d\n", tot_cycles );
|
|
|
|
echld_terminate();
|
|
return 0;
|
|
}
|