2013-06-21 01:27:28 +00:00
|
|
|
/* echld_dispatcher.c
|
|
|
|
* epan working child API internals
|
|
|
|
* Dispatcher process routines and definitions
|
|
|
|
*
|
|
|
|
* $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 "echld-int.h"
|
|
|
|
/**
|
|
|
|
DISPATCHER
|
|
|
|
**/
|
|
|
|
|
|
|
|
struct dispatcher_child {
|
2013-06-22 20:26:13 +00:00
|
|
|
echld_chld_id_t chld_id;
|
2013-06-21 01:27:28 +00:00
|
|
|
child_state_t state;
|
|
|
|
echld_reader_t reader;
|
|
|
|
int write_fd;
|
|
|
|
int pid;
|
|
|
|
gboolean closing;
|
|
|
|
};
|
|
|
|
|
|
|
|
struct dispatcher {
|
|
|
|
int parent_out;
|
|
|
|
echld_reader_t parent_in;
|
|
|
|
struct dispatcher_child* children;
|
|
|
|
int max_children;
|
|
|
|
int nchildren;
|
|
|
|
int reqh_id;
|
|
|
|
int pid;
|
|
|
|
int ppid;
|
|
|
|
struct _encs {
|
|
|
|
child_encoder_t* to_parent;
|
|
|
|
echld_parent_encoder_t* to_child;
|
|
|
|
} enc;
|
|
|
|
struct _decs {
|
|
|
|
child_decoder_t* from_parent;
|
|
|
|
parent_decoder_t* from_child;
|
|
|
|
} dec;
|
|
|
|
|
|
|
|
gboolean closing;
|
|
|
|
};
|
|
|
|
|
|
|
|
struct dispatcher* dispatcher;
|
|
|
|
|
|
|
|
#define DISP_RESP(B,T) (echld_write_frame( dispatcher->parent_out, (B), 0, (T), dispatcher->reqh_id, NULL))
|
|
|
|
|
|
|
|
#ifdef DEBUG_DISPATCHER
|
|
|
|
static int dbg_level = 0;
|
|
|
|
|
2013-06-22 20:26:13 +00:00
|
|
|
void dispatcher_debug(int level, const char* fmt, ...) {
|
2013-06-21 01:27:28 +00:00
|
|
|
va_list ap;
|
|
|
|
char* str;
|
|
|
|
|
|
|
|
if (dbg_level<level) return;
|
|
|
|
|
|
|
|
va_start(ap, fmt);
|
|
|
|
str = g_strdup_vprintf(fmt,ap);
|
|
|
|
va_end(ap);
|
|
|
|
|
|
|
|
fprintf(stderr, "dispatcher[%d]: reqh_id=%d dbg_level=%d message='%s'", dispatcher->pid, dispatcher->reqh_id, level, str);
|
|
|
|
g_free(str);
|
|
|
|
}
|
|
|
|
|
2013-06-22 20:26:13 +00:00
|
|
|
static char* param_get_dbg_level(char** err _U_) {
|
2013-06-21 01:27:28 +00:00
|
|
|
return g_strdup_printf("%d",dbg_level);
|
|
|
|
}
|
|
|
|
|
|
|
|
static echld_bool_t param_set_dbg_level(char* val , char** err ) {
|
|
|
|
char* p;
|
2013-06-22 20:26:13 +00:00
|
|
|
int lvl = (int)strtol(val, &p, 10);
|
2013-06-21 01:27:28 +00:00
|
|
|
|
|
|
|
if (p<=val) {
|
|
|
|
*err = g_strdup("not an integer");
|
|
|
|
return FALSE;
|
|
|
|
} else if (lvl < 0 || lvl > 5) {
|
|
|
|
*err = g_strdup_printf("invalid level=%d (min=0 max=5)",lvl);
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
|
|
|
|
dbg_level = lvl;
|
|
|
|
return TRUE;
|
|
|
|
}
|
|
|
|
|
|
|
|
#define DISP_DBG(attrs) ( dispatcher_debug attrs )
|
|
|
|
#else
|
|
|
|
#define DISP_DBG(attrs)
|
|
|
|
#endif
|
|
|
|
|
2013-06-22 20:26:13 +00:00
|
|
|
static void dispatcher_err(int errnum, const char* fmt, ...) {
|
|
|
|
size_t len= 1024;
|
|
|
|
gchar err_str[len];
|
|
|
|
va_list ap;
|
|
|
|
static GByteArray* ba;
|
2013-06-21 01:27:28 +00:00
|
|
|
|
2013-06-22 20:26:13 +00:00
|
|
|
va_start(ap, fmt);
|
|
|
|
g_vsnprintf(err_str,len,fmt,ap);
|
|
|
|
va_end(ap);
|
2013-06-21 01:27:28 +00:00
|
|
|
|
2013-06-22 20:26:13 +00:00
|
|
|
DISP_DBG((0,"error=\"%s\"",err_str));
|
2013-06-21 01:27:28 +00:00
|
|
|
|
2013-06-22 20:26:13 +00:00
|
|
|
ba = dispatcher->enc.to_parent->error(errnum, err_str);
|
|
|
|
DISP_RESP(ba,ECHLD_ERROR);
|
|
|
|
g_byte_array_free(ba,TRUE);
|
2013-06-21 01:27:28 +00:00
|
|
|
}
|
|
|
|
|
2013-06-22 20:26:13 +00:00
|
|
|
/* parameters */
|
2013-06-21 01:27:28 +00:00
|
|
|
|
2013-06-22 20:26:13 +00:00
|
|
|
/* interface listing */
|
|
|
|
|
|
|
|
static char* intflist2json(GList* if_list) {
|
|
|
|
/* blatantly stolen from print_machine_readable_interfaces in dumpcap.c */
|
|
|
|
#define ADDRSTRLEN 46 /* Covers IPv4 & IPv6 */
|
|
|
|
|
|
|
|
int i;
|
|
|
|
GList *if_entry;
|
|
|
|
if_info_t *if_info;
|
|
|
|
GSList *addr;
|
|
|
|
if_addr_t *if_addr;
|
|
|
|
char addr_str[ADDRSTRLEN];
|
|
|
|
GString *str = g_string_new("={ ");
|
|
|
|
char* s;
|
|
|
|
|
|
|
|
i = 1; /* Interface id number */
|
|
|
|
for (if_entry = g_list_first(if_list); if_entry != NULL;
|
|
|
|
if_entry = g_list_next(if_entry)) {
|
|
|
|
if_info = (if_info_t *)if_entry->data;
|
|
|
|
g_string_append_printf(str,"%d={ intf='%s',", i++, if_info->name);
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Print the contents of the if_entry struct in a parseable format.
|
|
|
|
* Each if_entry element is tab-separated. Addresses are comma-
|
|
|
|
* separated.
|
|
|
|
*/
|
|
|
|
/* XXX - Make sure our description doesn't contain a tab */
|
|
|
|
if (if_info->vendor_description != NULL)
|
|
|
|
g_string_append_printf(str," vnd_desc='%s',", if_info->vendor_description);
|
|
|
|
|
|
|
|
/* XXX - Make sure our friendly name doesn't contain a tab */
|
|
|
|
if (if_info->friendly_name != NULL)
|
|
|
|
g_string_append_printf(str," name='%s', addrs=[ ", if_info->friendly_name);
|
|
|
|
|
|
|
|
for (addr = g_slist_nth(if_info->addrs, 0); addr != NULL;
|
|
|
|
addr = g_slist_next(addr)) {
|
|
|
|
|
|
|
|
if_addr = (if_addr_t *)addr->data;
|
|
|
|
switch(if_addr->ifat_type) {
|
|
|
|
case IF_AT_IPv4:
|
|
|
|
if (inet_ntop(AF_INET, &if_addr->addr.ip4_addr, addr_str,
|
|
|
|
ADDRSTRLEN)) {
|
|
|
|
g_string_append_printf(str,"%s", addr_str);
|
|
|
|
} else {
|
|
|
|
g_string_append(str,"<unknown IPv4>");
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
case IF_AT_IPv6:
|
|
|
|
if (inet_ntop(AF_INET6, &if_addr->addr.ip6_addr,
|
|
|
|
addr_str, ADDRSTRLEN)) {
|
|
|
|
g_string_append_printf(str,"%s", addr_str);
|
|
|
|
} else {
|
|
|
|
g_string_append(str,"<unknown IPv6>");
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
g_string_append_printf(str,"<type unknown %u>", if_addr->ifat_type);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
g_string_append(str," ]"); /* addrs */
|
|
|
|
|
|
|
|
|
|
|
|
if (if_info->loopback)
|
|
|
|
g_string_append(str,", loopback=1");
|
|
|
|
else
|
|
|
|
g_string_append(str,", loopback=0");
|
|
|
|
|
|
|
|
g_string_append(str,"}, ");
|
|
|
|
}
|
|
|
|
|
|
|
|
g_string_truncate(str,str->len - 2); /* the comma and space */
|
|
|
|
g_string_append(str,"}");
|
|
|
|
|
|
|
|
s=str->str;
|
|
|
|
g_string_free(str,FALSE);
|
|
|
|
return s;
|
|
|
|
}
|
|
|
|
|
|
|
|
static char* param_get_interfaces(char** err) {
|
|
|
|
int err_no = 0;
|
|
|
|
GList* if_list;
|
|
|
|
char* s;
|
|
|
|
*err = NULL;
|
|
|
|
if_list = capture_interface_list(&err_no, err);
|
|
|
|
|
|
|
|
if (*err) {
|
|
|
|
return NULL;
|
2013-06-21 01:27:28 +00:00
|
|
|
}
|
|
|
|
|
2013-06-22 20:26:13 +00:00
|
|
|
s = intflist2json(if_list);
|
|
|
|
|
|
|
|
free_interface_list(if_list);
|
|
|
|
|
|
|
|
return s;
|
2013-06-21 01:27:28 +00:00
|
|
|
}
|
|
|
|
|
2013-06-22 20:26:13 +00:00
|
|
|
|
|
|
|
|
2013-06-21 01:27:28 +00:00
|
|
|
static param_t disp_params[] = {
|
|
|
|
#ifdef DEBUG_DISPATCHER
|
|
|
|
{"dbg_level", param_get_dbg_level, param_set_dbg_level},
|
|
|
|
# endif
|
2013-06-22 20:26:13 +00:00
|
|
|
{"interfaces",param_get_interfaces,NULL},
|
2013-06-21 01:27:28 +00:00
|
|
|
{NULL,NULL,NULL} };
|
|
|
|
|
|
|
|
static param_t* get_paramset(char* name) {
|
|
|
|
int i;
|
|
|
|
for (i = 0; disp_params[i].name != NULL;i++) {
|
|
|
|
if (strcmp(name,disp_params[i].name) == 0 ) return &(disp_params[i]);
|
|
|
|
}
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static struct dispatcher_child* dispatcher_get_child(struct dispatcher* d, guint16 chld_id) {
|
|
|
|
int i;
|
|
|
|
struct dispatcher_child* cc = d->children;
|
|
|
|
int max_children = d->max_children;
|
|
|
|
|
|
|
|
for(i = 0; i < max_children; i++) {
|
2013-06-22 20:26:13 +00:00
|
|
|
struct dispatcher_child* c = &(cc[i]);
|
2013-06-21 01:27:28 +00:00
|
|
|
if (c->chld_id == chld_id) return c;
|
|
|
|
}
|
|
|
|
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static void dispatcher_clear_child(struct dispatcher_child* c) {
|
|
|
|
echld_reset_reader(&(c->reader), -1, 4096);
|
|
|
|
c->chld_id = 0;
|
|
|
|
c->write_fd = 0;
|
|
|
|
c->pid = 0;
|
|
|
|
c->closing = 0;
|
|
|
|
}
|
|
|
|
|
2013-06-22 20:26:13 +00:00
|
|
|
static void preinit_epan(void) {
|
2013-06-21 01:27:28 +00:00
|
|
|
/* Here we do initialization of parts of epan that will be the same for every child we fork */
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2013-06-22 20:26:13 +00:00
|
|
|
static void dispatcher_clear(void) {
|
2013-06-21 01:27:28 +00:00
|
|
|
/* remove unnecessary stuff for the working child */
|
|
|
|
}
|
|
|
|
|
|
|
|
void dispatcher_reaper(int sig) {
|
|
|
|
int status;
|
|
|
|
int i;
|
|
|
|
struct dispatcher_child* cc = dispatcher->children;
|
|
|
|
int max_children = dispatcher->max_children;
|
|
|
|
int pid = waitpid(-1, &status, WNOHANG);
|
|
|
|
GByteArray* em;
|
|
|
|
|
2013-06-22 20:26:13 +00:00
|
|
|
if (sig != SIGCHLD) {
|
|
|
|
DISP_DBG((1,"Reaper got wrong signal=%d",sig));
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
DISP_DBG((2,"Child dead pid=%d",pid));
|
2013-06-21 01:27:28 +00:00
|
|
|
|
|
|
|
for(i = 0; i < max_children; i++) {
|
|
|
|
struct dispatcher_child* c = &(cc[i]);
|
|
|
|
if ( c->pid == pid ) {
|
|
|
|
if (c->closing || dispatcher->closing) {
|
2013-06-22 20:26:13 +00:00
|
|
|
em = dispatcher->enc.to_parent->child_dead("OK");
|
2013-06-21 01:27:28 +00:00
|
|
|
} else {
|
2013-06-22 20:26:13 +00:00
|
|
|
char* s = NULL;
|
|
|
|
|
|
|
|
if (WIFEXITED(status)) {
|
|
|
|
s = g_strdup_printf(
|
|
|
|
"Unexpected dead: reason='exited' pid=%d status=%d",
|
|
|
|
pid, WEXITSTATUS(status));
|
|
|
|
} else if ( WIFSIGNALED(status) ) {
|
|
|
|
s = g_strdup_printf(
|
|
|
|
"Unexpected dead: reason='signaled' pid=%d termsig=%d coredump=%s",
|
|
|
|
pid, WTERMSIG(status), WCOREDUMP(status) ? "yes":"no");
|
|
|
|
|
|
|
|
/*if (WCOREDUMP(status)) { system("analyze_coredump.sh pid=%d") } */
|
|
|
|
|
|
|
|
} else if (WIFSTOPPED(status)) {
|
|
|
|
s = g_strdup_printf(
|
|
|
|
"Unexpected dead: reason='stopped' pid=%d stopsig=%d",
|
|
|
|
pid, WSTOPSIG(status));
|
|
|
|
}
|
2013-06-21 01:27:28 +00:00
|
|
|
|
2013-06-22 20:26:13 +00:00
|
|
|
em = dispatcher->enc.to_parent->child_dead(s);
|
|
|
|
dispatcher_err(ECHLD_ERR_CRASHED_CHILD, s);
|
|
|
|
if (s) g_free(s);
|
2013-06-21 01:27:28 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
echld_write_frame(dispatcher->parent_out, em, c->chld_id, ECHLD_CHILD_DEAD, 0, NULL);
|
|
|
|
dispatcher_clear_child(c);
|
|
|
|
g_byte_array_free(em,TRUE);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
dispatcher_err(ECHLD_ERR_UNKNOWN_PID, "Unkown child pid: %d", pid);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
2013-06-22 20:26:13 +00:00
|
|
|
static void dispatcher_destroy(void) {
|
2013-06-21 01:27:28 +00:00
|
|
|
int i;
|
|
|
|
int max_children = dispatcher->max_children;
|
|
|
|
struct dispatcher_child* cc = dispatcher->children;
|
|
|
|
/* destroy the dispatcher stuff at closing */
|
|
|
|
|
|
|
|
dispatcher->closing = TRUE;
|
|
|
|
|
|
|
|
/* kill all alive children */
|
|
|
|
for(i = 0; i < max_children; i++) {
|
|
|
|
struct dispatcher_child* c = &(cc[i]);
|
|
|
|
if ( c->chld_id ) {
|
|
|
|
kill(c->pid,SIGTERM);
|
|
|
|
DISP_DBG((1,"Killing chld_id=%d pid=%d"));
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-06-22 20:26:13 +00:00
|
|
|
exit(0);
|
2013-06-21 01:27:28 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* stuff coming from child going to parent */
|
2013-06-22 20:26:13 +00:00
|
|
|
static long dispatch_to_parent(guint8* b, size_t len, echld_chld_id_t chld_id, echld_msg_type_t type, echld_reqh_id_t reqh_id, void* data) {
|
2013-06-21 01:27:28 +00:00
|
|
|
/* TODO: timeouts, clear them */
|
|
|
|
/* TODO: keep stats */
|
2013-06-22 20:26:13 +00:00
|
|
|
|
2013-06-21 01:27:28 +00:00
|
|
|
GByteArray in_ba;
|
|
|
|
|
2013-06-22 20:26:13 +00:00
|
|
|
struct dispatcher_child* c = (struct dispatcher_child*)data;
|
|
|
|
dispatcher->reqh_id = reqh_id;
|
|
|
|
|
2013-06-21 01:27:28 +00:00
|
|
|
in_ba.data = b;
|
2013-06-22 20:26:13 +00:00
|
|
|
in_ba.len = (guint)len;
|
2013-06-21 01:27:28 +00:00
|
|
|
|
|
|
|
if (chld_id != c->chld_id) {
|
|
|
|
goto misbehabing;
|
|
|
|
}
|
|
|
|
|
|
|
|
switch(type) {
|
|
|
|
case ECHLD_ERROR: break;
|
|
|
|
case ECHLD_TIMED_OUT: break;
|
|
|
|
case ECHLD_HELLO: c->state = IDLE; break;
|
|
|
|
case ECHLD_CLOSING: c->closing = TRUE; c->state = CLOSED; break;
|
|
|
|
case ECHLD_PARAM: break;
|
|
|
|
case ECHLD_PONG: break;
|
|
|
|
case ECHLD_FILE_OPENED: c->state = READING; break;
|
|
|
|
case ECHLD_INTERFACE_OPENED: c->state = READY; break;
|
|
|
|
case ECHLD_CAPTURE_STARTED: c->state = CAPTURING; break;
|
|
|
|
case ECHLD_NOTIFY: break; // notify(pre-encoded)
|
|
|
|
case ECHLD_PACKET_SUM: break; // packet_sum(pre-encoded)
|
|
|
|
case ECHLD_TREE: break; //tree(framenum, tree(pre-encoded) )
|
|
|
|
case ECHLD_BUFFER: break; // buffer (name,range,totlen,data)
|
|
|
|
case ECHLD_EOF: c->state = DONE; break;
|
|
|
|
case ECHLD_CAPTURE_STOPPED: c->state = DONE; break;
|
|
|
|
case ECHLD_NOTE_ADDED: break;
|
|
|
|
case ECHLD_PACKET_LIST: break; // packet_list(name,filter,range);
|
|
|
|
case ECHLD_FILE_SAVED: break;
|
|
|
|
|
|
|
|
default:
|
|
|
|
goto misbehabing;
|
|
|
|
}
|
|
|
|
|
|
|
|
return echld_write_frame(dispatcher->parent_out, &in_ba, chld_id, type, reqh_id, NULL);
|
|
|
|
|
|
|
|
misbehabing:
|
|
|
|
c->state = ERRORED;
|
|
|
|
c->closing = TRUE;
|
|
|
|
kill(c->pid,SIGTERM);
|
|
|
|
dispatcher_err(ECHLD_ERR_CRASHED_CHILD,"chld_id=%d",chld_id);
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
void dispatch_new_child(struct dispatcher* dd) {
|
|
|
|
struct dispatcher_child* c = dispatcher_get_child(dd, 0);
|
|
|
|
int reqh_id = dd->reqh_id;
|
|
|
|
int pid;
|
|
|
|
|
|
|
|
if ( c ) {
|
|
|
|
int parent_pipe_fds[2];
|
|
|
|
int child_pipe_fds[2];
|
|
|
|
|
|
|
|
int pipe_to_parent;
|
|
|
|
int pipe_from_parent;
|
|
|
|
int pipe_to_child;
|
|
|
|
int pipe_from_child;
|
|
|
|
|
|
|
|
if( pipe(parent_pipe_fds) < 0) {
|
|
|
|
dispatcher_err(ECHLD_ERR_CANNOT_FORK,"CANNOT OPEN PARENT PIPE: %s",strerror(errno));
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
pipe_from_parent = parent_pipe_fds[0];
|
|
|
|
pipe_to_child = parent_pipe_fds[1];
|
|
|
|
|
|
|
|
if( pipe(child_pipe_fds) < 0) {
|
|
|
|
close(pipe_from_parent);
|
|
|
|
close(pipe_to_child);
|
|
|
|
dispatcher_err(ECHLD_ERR_CANNOT_FORK,"CANNOT OPEN CHILD PIPE: %s",strerror(errno));
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
pipe_from_child = child_pipe_fds[0];
|
|
|
|
pipe_to_parent = child_pipe_fds[1];
|
|
|
|
|
|
|
|
switch (( pid = fork() )) {
|
|
|
|
case -1: {
|
|
|
|
close(pipe_to_child);
|
|
|
|
close(pipe_to_parent);
|
|
|
|
close(pipe_from_child);
|
|
|
|
close(pipe_from_parent);
|
|
|
|
dispatcher_err(ECHLD_ERR_CANNOT_FORK,"CANNOT FORK: %s",strerror(errno));
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
case 0: { /* I'm the child */
|
|
|
|
int i;
|
|
|
|
int fdt_len = getdtablesize();
|
|
|
|
|
2013-06-22 20:26:13 +00:00
|
|
|
dispatcher_clear();
|
2013-06-21 01:27:28 +00:00
|
|
|
|
|
|
|
for(i=0;i<fdt_len;i++) {
|
|
|
|
if ( i != pipe_from_parent
|
|
|
|
&& i != pipe_to_parent
|
|
|
|
&& i != STDERR_FILENO ) {
|
|
|
|
close(i);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
echld_child_initialize(pipe_from_parent,pipe_to_parent,reqh_id);
|
|
|
|
|
|
|
|
exit( echld_child_loop() );
|
|
|
|
|
|
|
|
/* it won't */
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
default: {
|
|
|
|
/* I'm the parent */
|
|
|
|
guint8 buf[4];
|
|
|
|
GByteArray out_ba;
|
|
|
|
|
|
|
|
out_ba.data = buf;
|
|
|
|
out_ba.len = 0;
|
|
|
|
|
|
|
|
close(pipe_to_parent);
|
|
|
|
close(pipe_from_parent);
|
|
|
|
|
|
|
|
echld_reset_reader(&(c->reader), pipe_from_child,4096);
|
|
|
|
c->write_fd = pipe_to_child;
|
|
|
|
c->pid = pid;
|
|
|
|
dispatcher->nchildren++;
|
|
|
|
|
|
|
|
/* configure child */
|
|
|
|
echld_write_frame(pipe_to_child, &out_ba, c->chld_id, ECHLD_NEW_CHILD, dispatcher->reqh_id, NULL);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
dispatcher_err(ECHLD_ERR_CANNOT_FORK, "MAX CHILDREN REACHED: max_children=%d",dispatcher->max_children);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/* process signals sent from parent */
|
2013-06-22 20:26:13 +00:00
|
|
|
static long dispatch_to_child(guint8* b, size_t len, echld_chld_id_t chld_id, echld_msg_type_t type, echld_reqh_id_t reqh_id, void* data) {
|
|
|
|
struct dispatcher* disp = (struct dispatcher*)data;
|
2013-06-21 01:27:28 +00:00
|
|
|
GByteArray in_ba;
|
|
|
|
|
2013-06-22 20:26:13 +00:00
|
|
|
disp->reqh_id = reqh_id;
|
|
|
|
|
2013-06-21 01:27:28 +00:00
|
|
|
in_ba.data = b;
|
2013-06-22 20:26:13 +00:00
|
|
|
in_ba.len = (guint)len;
|
2013-06-21 01:27:28 +00:00
|
|
|
|
|
|
|
if (chld_id == 0) { /* these are messages to the dispatcher itself */
|
|
|
|
switch(type) {
|
|
|
|
case ECHLD_CLOSE_CHILD:
|
|
|
|
dispatcher_destroy();
|
|
|
|
return 0;
|
|
|
|
case ECHLD_PING:
|
|
|
|
echld_write_frame(disp->parent_out, &in_ba, chld_id, ECHLD_PONG, reqh_id, NULL);
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
case ECHLD_NEW_CHILD:
|
|
|
|
dispatch_new_child(disp);
|
|
|
|
return 0;
|
|
|
|
case ECHLD_SET_PARAM:{
|
|
|
|
char* param;
|
|
|
|
char* value;
|
|
|
|
if ( disp->dec.from_parent->set_param(b,len,¶m,&value) ) {
|
|
|
|
GByteArray* ba;
|
|
|
|
param_t* p = get_paramset(param);
|
|
|
|
char* err;
|
|
|
|
if (!p) {
|
|
|
|
dispatcher_err(ECHLD_CANNOT_SET_PARAM,"no such param='%s'",param);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (! p->set ) {
|
|
|
|
dispatcher_err(ECHLD_CANNOT_SET_PARAM,"reason='read only'");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (! p->set(value,&err) ) {
|
|
|
|
dispatcher_err(ECHLD_CANNOT_SET_PARAM,"reason='%s'",err);
|
|
|
|
g_free(err);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2013-06-22 20:26:13 +00:00
|
|
|
ba = disp->enc.to_parent->param(param,value);
|
2013-06-21 01:27:28 +00:00
|
|
|
DISP_RESP(ba,ECHLD_PARAM);
|
|
|
|
g_byte_array_free(ba,TRUE);
|
|
|
|
DISP_DBG((1,"Set Param: param='%s' value='%s'",param,value));
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
} else {
|
|
|
|
dispatcher_err(ECHLD_CANNOT_SET_PARAM,"reason='decoder error'");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
case ECHLD_GET_PARAM: {
|
|
|
|
GByteArray* ba;
|
|
|
|
char* param;
|
|
|
|
if ( disp->dec.from_parent->get_param(b,len,¶m) ) {
|
|
|
|
char* err;
|
|
|
|
char* val;
|
|
|
|
|
|
|
|
param_t* p = get_paramset(param);
|
|
|
|
|
|
|
|
if (!p) {
|
|
|
|
dispatcher_err(ECHLD_CANNOT_GET_PARAM,"no such param='%s'",param);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (! p->get ) {
|
|
|
|
dispatcher_err(ECHLD_CANNOT_SET_PARAM,"reason='write only'");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!(val = p->get(&err))) {
|
|
|
|
dispatcher_err(ECHLD_CANNOT_GET_PARAM,"reason='%s'",err);
|
|
|
|
g_free(err);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2013-06-22 20:26:13 +00:00
|
|
|
ba = disp->enc.to_parent->param(param,val);
|
2013-06-21 01:27:28 +00:00
|
|
|
DISP_RESP(ba,ECHLD_PARAM);
|
|
|
|
g_byte_array_free(ba,TRUE);
|
|
|
|
DISP_DBG((2,"Get Param: param='%s' value='%s'",param,val));
|
|
|
|
return 0;
|
|
|
|
} else {
|
|
|
|
dispatcher_err(ECHLD_CANNOT_GET_PARAM,"reason='decoder error'");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
default:
|
|
|
|
dispatcher_err(ECHLD_ERR_WRONG_MSG, "wrong message to dispatcher type='%c'", type);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
struct dispatcher_child* c;
|
|
|
|
|
|
|
|
if (! (c = dispatcher_get_child(dispatcher, chld_id)) ) {
|
|
|
|
dispatcher_err(ECHLD_ERR_NO_SUCH_CHILD, "wrong chld_id %d", chld_id);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
switch(type) {
|
|
|
|
case ECHLD_CLOSE_CHILD:
|
|
|
|
c->closing = TRUE;
|
|
|
|
c->state = CLOSED;
|
|
|
|
goto relay_frame;
|
|
|
|
|
|
|
|
case ECHLD_OPEN_FILE:
|
|
|
|
c->state = READING;
|
|
|
|
goto relay_frame;
|
|
|
|
|
|
|
|
case ECHLD_OPEN_INTERFACE:
|
|
|
|
c->state = READY;
|
|
|
|
goto relay_frame;
|
|
|
|
|
|
|
|
case ECHLD_START_CAPTURE:
|
|
|
|
c->state = CAPTURING;
|
|
|
|
goto relay_frame;
|
|
|
|
|
|
|
|
case ECHLD_STOP_CAPTURE:
|
|
|
|
c->state = DONE;
|
|
|
|
goto relay_frame;
|
|
|
|
|
|
|
|
case ECHLD_SAVE_FILE:
|
|
|
|
case ECHLD_APPLY_FILTER:
|
|
|
|
case ECHLD_SET_PARAM:
|
|
|
|
case ECHLD_GET_PARAM:
|
|
|
|
case ECHLD_PING:
|
|
|
|
case ECHLD_GET_SUM:
|
|
|
|
case ECHLD_GET_TREE:
|
|
|
|
case ECHLD_GET_BUFFER:
|
|
|
|
case ECHLD_ADD_NOTE:
|
|
|
|
relay_frame:
|
|
|
|
return echld_write_frame(c->write_fd, &in_ba, chld_id, type, reqh_id, NULL);
|
|
|
|
|
|
|
|
default:
|
|
|
|
dispatcher_err(ECHLD_ERR_WRONG_MSG, "wrong message %d %c", reqh_id, type);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2013-06-22 20:26:13 +00:00
|
|
|
int dispatcher_loop(void) {
|
2013-06-21 01:27:28 +00:00
|
|
|
int parent_out = dispatcher->parent_out;
|
|
|
|
int parent_in = dispatcher->parent_in.fd;
|
|
|
|
|
|
|
|
struct dispatcher_child* children = dispatcher->children;
|
|
|
|
|
|
|
|
do {
|
|
|
|
fd_set rfds;
|
|
|
|
fd_set efds;
|
|
|
|
struct timeval timeout;
|
|
|
|
struct dispatcher_child* c;
|
|
|
|
int nfds;
|
|
|
|
|
|
|
|
FD_ZERO(&rfds);
|
|
|
|
FD_ZERO(&efds);
|
|
|
|
|
|
|
|
FD_SET(parent_in,&rfds);
|
|
|
|
FD_SET(parent_in,&efds);
|
|
|
|
FD_SET(parent_out,&efds);
|
|
|
|
|
|
|
|
for (c = children, nfds = 0; c->pid; c++) {
|
|
|
|
if (c->chld_id) {
|
|
|
|
FD_SET(c->reader.fd, &rfds);
|
|
|
|
FD_SET(c->reader.fd, &efds);
|
|
|
|
}
|
|
|
|
nfds++;
|
|
|
|
}
|
|
|
|
|
|
|
|
nfds = select(nfds, &rfds, NULL, &efds, &timeout);
|
|
|
|
|
|
|
|
if ( FD_ISSET(parent_in, &efds) || FD_ISSET(parent_out, &efds) ) {
|
|
|
|
/* XXX deep shit */
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (FD_ISSET(parent_in, &rfds)) {
|
2013-06-22 20:26:13 +00:00
|
|
|
long st = echld_read_frame(&(dispatcher->parent_in), dispatch_to_child, dispatcher);
|
2013-06-21 01:27:28 +00:00
|
|
|
|
|
|
|
if (st < 0) {
|
|
|
|
/* XXX */
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
for (c=children; c->pid; c++) {
|
|
|
|
if (c->chld_id) {
|
|
|
|
if ( FD_ISSET(c->reader.fd,&efds) ) {
|
|
|
|
/* XXX cleanup child and report */
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (FD_ISSET(c->reader.fd,&rfds)) {
|
2013-06-22 20:26:13 +00:00
|
|
|
long st = echld_read_frame(&(c->reader), dispatch_to_parent, c);
|
2013-06-21 01:27:28 +00:00
|
|
|
|
|
|
|
if (st < 0) {
|
|
|
|
/* XXX cleanup child and report */
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
} while(1);
|
|
|
|
|
|
|
|
/* won't */
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
void echld_dispatcher_start(int* in_pipe_fds, int* out_pipe_fds) {
|
|
|
|
static struct dispatcher d;
|
|
|
|
int fdt_len = getdtablesize();
|
|
|
|
int i;
|
|
|
|
|
|
|
|
preinit_epan();
|
|
|
|
|
|
|
|
signal(SIGCHLD,dispatcher_reaper);
|
|
|
|
|
|
|
|
dispatcher = &d;
|
|
|
|
|
|
|
|
echld_init_reader(&(d.parent_in),in_pipe_fds[0],4096);
|
|
|
|
d.parent_out = out_pipe_fds[1];
|
2013-06-22 20:26:13 +00:00
|
|
|
d.children = g_new0(struct dispatcher_child,ECHLD_MAX_CHILDREN);
|
2013-06-21 01:27:28 +00:00
|
|
|
d.max_children = ECHLD_MAX_CHILDREN;
|
|
|
|
d.nchildren = 0;
|
|
|
|
d.reqh_id = -1;
|
|
|
|
d.pid = getpid();
|
|
|
|
|
|
|
|
echld_get_all_codecs(&(d.enc.to_parent), &(d.dec.from_parent), &(d.enc.to_child), &(d.dec.from_child));
|
|
|
|
|
|
|
|
dispatcher_clear();
|
|
|
|
|
|
|
|
/* close all fds but those used */
|
|
|
|
for(i=0;i<fdt_len;i++) {
|
|
|
|
if ( i != d.parent_in.fd
|
|
|
|
&& i != d.parent_out
|
|
|
|
&& i != STDERR_FILENO ) {
|
|
|
|
close(i);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
exit(dispatcher_loop());
|
|
|
|
}
|
|
|
|
|