tree_cell *
nasl_open_sock_tcp_bufsz (lex_ctxt * lexic, int bufsz)
{
int soc = -1;
struct arglist *script_infos = lexic->script_infos;
int to, port;
int transport = -1;
const char *priority;
tree_cell *retc;
int type;
to = get_int_local_var_by_name (lexic, "timeout", lexic->recv_timeout * 2);
if (to < 0)
to = 10;
transport = get_int_local_var_by_name (lexic, "transport", -1);
if (transport == OPENVAS_ENCAPS_TLScustom)
{
priority = get_str_local_var_by_name (lexic, "priority");
if (!priority)
priority = NULL;
type = get_local_var_type_by_name (lexic, "priority");
if (type != VAR2_STRING && type != VAR2_DATA)
priority = NULL;
}
else
priority = NULL;
if (bufsz < 0)
bufsz = get_int_local_var_by_name (lexic, "bufsz", 0);
port = get_int_var_by_num (lexic, 0, -1);
if (port < 0)
return NULL;
/* If "transport" has not been given, use auto detection if enabled
in the KB. if "transport" has been given with a value of 0 force
autodetection reagardless of what the KB tells. */
if (transport < 0)
soc = open_stream_auto_encaps_ext (script_infos, port, to, 0);
else if (transport == 0)
soc = open_stream_auto_encaps_ext (script_infos, port, to, 1);
else
soc = open_stream_connection_ext (script_infos, port, transport, to,
priority);
if (bufsz > 0 && soc >= 0)
{
if (stream_set_buffer (soc, bufsz) < 0)
nasl_perror (lexic, "stream_set_buffer: soc=%d,bufsz=%d\n", soc, bufsz);
}
retc = alloc_tree_cell (0, NULL);
retc->type = CONST_INT;
retc->x.i_val = soc < 0 ? 0 : soc;
return retc;
}
tree_cell * nasl_open_sock_tcp_bufsz(lex_ctxt * lexic, int bufsz)
{
int soc = -1;
struct arglist * script_infos = lexic->script_infos;
int to, port, transport = -1;
tree_cell * retc;
char *host;
to = get_int_local_var_by_name(lexic, "timeout", lexic->recv_timeout*2);
if(to <= 0)
to = 10;
transport = get_int_local_var_by_name(lexic, "transport", -1);
if (bufsz < 0)
bufsz = get_int_local_var_by_name(lexic, "bufsz", 0);
port = get_int_var_by_num(lexic, 0, -1);
if(port < 0)
return NULL;
if(transport < 0)
soc = open_stream_auto_encaps(script_infos, port, to);
else
soc = open_stream_connection(script_infos, port, transport, to);
if (bufsz > 0 && soc >= 0 )
{
if (stream_set_buffer(soc, bufsz) < 0)
nasl_perror(lexic, "stream_set_buffer: soc=%d,bufsz=%d\n", soc, bufsz);
}
retc = alloc_tree_cell(0, NULL);
retc->type = CONST_INT;
retc->x.i_val = soc < 0 ? 0 : soc;
return retc;
}
tree_cell * nasl_recv_line(lex_ctxt * lexic)
{
int len = get_int_local_var_by_name(lexic, "length", -1);
int soc = get_int_local_var_by_name(lexic, "socket", 0);
int timeout = get_int_local_var_by_name(lexic, "timeout", -1);
char * data;
int new_len = 0;
int n = 0;
tree_cell * retc;
time_t t1 = 0;
if(timeout < 0) timeout = lexic->recv_timeout;
if(len == -1 || soc <= 0)
{
nasl_perror(lexic, "recv_line: missing or undefined parameter length or soc\n");
return NULL;
}
if (timeout >= 0) /* sycalls are much more expensive than simple tests */
t1 = time(NULL);
if ( fd_is_stream(soc) != 0 )
{
int bufsz = stream_get_buffer_sz ( soc );
if ( bufsz <= 0 )
stream_set_buffer(soc, len + 1 );
}
data = emalloc(len+1);
for(;;)
{
int e = read_stream_connection_min(soc, data+n, 1, 1);
if(e < 0)
break;
if(e == 0)
{
if( timeout >= 0 && time(NULL) - t1 < timeout)
continue;
else
break;
}
n++;
if((data[n-1] == '\n') ||
(n >= len))break;
}
if(n <= 0)
{
efree(&data);
return NULL;
}
new_len = n;
retc = alloc_tree_cell(0, NULL);
retc->type = CONST_DATA;
retc->size = new_len;
retc->x.str_val = nasl_strndup(data, new_len);
efree(&data);
return retc;
}
