int status_client_ncurses(struct conf **confs)
{
int ret=-1;
int csin=-1;
int csout=-1;
pid_t childpid=-1;
struct async *as=NULL;
const char *monitor_logfile=get_string(confs[OPT_MONITOR_LOGFILE]);
struct asfd *so_asfd=NULL;
struct sel *sel=NULL;
if(!(sel=sel_alloc()))
goto end;
setup_signals();
// Fork a burp child process that will contact the server over SSL.
// We will read and write from and to its stdout and stdin.
if((childpid=fork_monitor(&csin, &csout, confs))<0)
goto end;
//printf("childpid: %d\n", childpid);
if(!(as=async_alloc())
|| as->init(as, 0)
|| !setup_asfd_linebuf_write(as, "monitor stdin", &csin)
|| !setup_asfd_linebuf_read(as, "monitor stdout", &csout))
goto end;
//printf("ml: %s\n", monitor_logfile);
#ifdef HAVE_NCURSES
if(actg==ACTION_STATUS)
{
if(!setup_asfd_ncurses_stdin(as))
goto end;
ncurses_init();
}
#endif
if(!(so_asfd=setup_asfd_stdout(as)))
goto end;
if(monitor_logfile
&& !(lfzp=fzp_open(monitor_logfile, "wb")))
goto end;
log_fzp_set_direct(lfzp);
ret=status_client_ncurses_main_loop(as, so_asfd, sel,
get_string(confs[OPT_ORIG_CLIENT]));
end:
#ifdef HAVE_NCURSES
if(actg==ACTION_STATUS)
ncurses_free();
#endif
if(ret) logp("%s exiting with error: %d\n", __func__, ret);
fzp_close(&lfzp);
async_asfd_free_all(&as);
close_fd(&csin);
close_fd(&csout);
sel_free(&sel);
return ret;
}
static void do_snapshot_test(
const char *orig_client,
int expected_ret,
void setup_callback(
struct asfd *csin,
struct asfd *csout,
struct asfd *so_asfd
)
)
{
struct asfd *csin;
struct asfd *csout;
struct asfd *so_asfd;
struct async *as;
struct sel *sel;
as=setup_async();
csin=asfd_mock_setup(&reads_csin, &writes_csin);
csout=asfd_mock_setup(&reads_csout, &writes_csout);
so_asfd=asfd_mock_setup(&reads_so, &writes_so);
fail_unless((csin->desc=strdup_w("csin", __func__))!=NULL);
fail_unless((csout->desc=strdup_w("csout", __func__))!=NULL);
fail_unless((so_asfd->desc=strdup_w("so_asfd", __func__))!=NULL);
as->asfd_add(as, csin);
as->asfd_add(as, csout);
as->asfd_add(as, so_asfd);
as->read_write=async_rw_simple;
as->write=async_write_simple;
setup_callback(csin, csout, so_asfd);
fail_unless((sel=sel_alloc())!=NULL);
fail_unless(!status_client_ncurses_init(ACTION_STATUS_SNAPSHOT));
fail_unless(status_client_ncurses_main_loop(
as,
so_asfd,
sel,
orig_client
)==expected_ret);
sel_free(&sel);
asfd_free(&csin);
asfd_free(&csout);
asfd_free(&so_asfd);
asfd_mock_teardown(&reads_csin, &writes_csin);
asfd_mock_teardown(&reads_csout, &writes_csout);
asfd_mock_teardown(&reads_so, &writes_so);
async_free(&as);
alloc_check();
}
static void do_simple_test(
const char *orig_client,
int expected_ret,
void setup_callback(struct asfd *asfd),
void setup_stdout_callback(struct asfd *asfd)
)
{
struct asfd *asfd;
struct asfd *so_asfd;
struct async *as;
as=setup_async();
asfd=asfd_mock_setup(&reads, &writes);
so_asfd=asfd_mock_setup(&reads_so, &writes_so);
as->asfd_add(as, asfd);
as->asfd_add(as, so_asfd);
asfd->as=as;
as->read_write=async_rw_simple;
as->write=async_write_simple;
setup_callback(asfd);
if(setup_stdout_callback)
setup_stdout_callback(so_asfd);
fail_unless(!status_client_ncurses_init(ACTION_STATUS_SNAPSHOT));
fail_unless(status_client_ncurses_main_loop(
as,
so_asfd,
orig_client
)==expected_ret);
asfd_free(&asfd);
asfd_free(&so_asfd);
asfd_mock_teardown(&reads, &writes);
asfd_mock_teardown(&reads_so, &writes_so);
async_free(&as);
alloc_check();
}
END_TEST
static void do_status_test(
const char *orig_client,
int expected_ret,
void setup_callback(
struct asfd *csin,
struct asfd *csout,
struct asfd *nin,
struct asfd *so_asfd
),
void check_sel_callback(
struct sel *sel
)
)
{
struct asfd *csin;
struct asfd *csout;
struct asfd *nin;
struct asfd *so_asfd;
struct async *as;
struct sel *sel;
as=setup_async();
csin=asfd_mock_setup(&reads_csin, &writes_csin);
csout=asfd_mock_setup(&reads_csout, &writes_csout);
nin=asfd_mock_setup(&reads_nin, &writes_nin);
so_asfd=asfd_mock_setup(&reads_so, &writes_so);
fail_unless((csin->desc=strdup_w("csin", __func__))!=NULL);
fail_unless((csout->desc=strdup_w("csout", __func__))!=NULL);
fail_unless((nin->desc=strdup_w("nin", __func__))!=NULL);
fail_unless((so_asfd->desc=strdup_w("so_asfd", __func__))!=NULL);
nin->streamtype=ASFD_STREAM_NCURSES_STDIN;
as->asfd_add(as, csin);
as->asfd_add(as, csout);
as->asfd_add(as, nin);
as->asfd_add(as, so_asfd);
as->read_write=async_rw_both;
as->write=async_write_simple;
setup_callback(csin, csout, nin, so_asfd);
fail_unless((sel=sel_alloc())!=NULL);
fail_unless(!status_client_ncurses_init(ACTION_STATUS));
fail_unless(status_client_ncurses_main_loop(
as,
so_asfd,
sel,
orig_client
)==expected_ret);
if(check_sel_callback)
check_sel_callback(sel);
sel_free(&sel);
asfd_free(&csin);
asfd_free(&csout);
asfd_free(&nin);
asfd_free(&so_asfd);
asfd_mock_teardown(&reads_csin, &writes_csin);
asfd_mock_teardown(&reads_csout, &writes_csout);
asfd_mock_teardown(&reads_nin, &writes_nin);
asfd_mock_teardown(&reads_so, &writes_so);
async_free(&as);
alloc_check();
}
