tb_char_t const* tb_string_rtrim(tb_string_ref_t string)
{
// check
tb_assert_and_check_return_val(string, tb_null);
// init
tb_char_t* s = (tb_char_t*)tb_string_cstr(string);
tb_size_t n = tb_string_size(string);
tb_check_return_val(s && n, tb_null);
// done
tb_char_t* e = s + n - 1;
while (e >= s && tb_isspace(*e)) e--;
// strip it
if (e >= s) tb_string_strip(string, e - s + 1);
// clear it
else tb_string_clear(string);
// ok?
return tb_string_cstr(string);
}
/* chunked_data
*
* head data tail
* ea5\r\n ..........\r\n e65\r\n..............\r\n 0\r\n\r\n
* ---------------------- ------------------------- ---------
* chunk0 chunk1 end
*/
static tb_long_t tb_filter_chunked_spak(tb_filter_t* filter, tb_static_stream_ref_t istream, tb_static_stream_ref_t ostream, tb_long_t sync)
{
// check
tb_filter_chunked_t* cfilter = tb_filter_chunked_cast(filter);
tb_assert_and_check_return_val(cfilter && istream && ostream, -1);
tb_assert_and_check_return_val(tb_static_stream_valid(istream) && tb_static_stream_valid(ostream), -1);
// the idata
tb_byte_t const* ip = tb_static_stream_pos(istream);
tb_byte_t const* ie = tb_static_stream_end(istream);
// trace
tb_trace_d("[%p]: isize: %lu, beof: %d", cfilter, tb_static_stream_size(istream), filter->beof);
// find the eof: '\r\n 0\r\n\r\n'
if ( !filter->beof
&& ip + 6 < ie
&& ie[-7] == '\r'
&& ie[-6] == '\n'
&& ie[-5] == '0'
&& ie[-4] == '\r'
&& ie[-3] == '\n'
&& ie[-2] == '\r'
&& ie[-1] == '\n')
{
// is eof
filter->beof = tb_true;
}
// the odata
tb_byte_t* op = (tb_byte_t*)tb_static_stream_pos(ostream);
tb_byte_t* oe = (tb_byte_t*)tb_static_stream_end(ostream);
tb_byte_t* ob = op;
// parse chunked head and chunked tail
if (!cfilter->size || cfilter->read >= cfilter->size)
{
// walk
while (ip < ie)
{
// the charactor
tb_char_t ch = *ip++;
// trace
tb_trace_d("[%p]: character: %x", cfilter, ch);
// check
tb_assert_and_check_return_val(ch, -1);
// append char to line
if (ch != '\n') tb_string_chrcat(&cfilter->line, ch);
// is line end?
else
{
// check
tb_char_t const* pb = tb_string_cstr(&cfilter->line);
tb_size_t pn = tb_string_size(&cfilter->line);
tb_assert_and_check_return_val(pb, -1);
// trace
tb_trace_d("[%p]: line: %s", cfilter, tb_string_cstr(&cfilter->line));
// strip '\r' if exists
if (pb[pn - 1] == '\r') tb_string_strip(&cfilter->line, pn - 1);
// is chunked tail? only "\r\n"
if (!tb_string_size(&cfilter->line))
{
// reset size
cfilter->read = 0;
cfilter->size = 0;
// trace
tb_trace_d("[%p]: tail", cfilter);
// continue
continue ;
}
// is chunked head? parse size
else
{
// parse size
cfilter->size = tb_s16tou32(pb);
// trace
tb_trace_d("[%p]: size: %lu", cfilter, cfilter->size);
// clear data
tb_string_clear(&cfilter->line);
// is eof? "0\r\n\r\n"
if (!cfilter->size)
{
// trace
tb_trace_d("[%p]: eof", cfilter);
// is eof
filter->beof = tb_true;
// continue to spak the end data
continue ;
}
// ok
break;
}
}
}
}
// check
tb_assert_and_check_return_val(cfilter->read <= cfilter->size, -1);
// read chunked data
tb_size_t size = tb_min3(ie - ip, oe - op, cfilter->size - cfilter->read);
if (size)
{
// copy data
tb_memcpy((tb_byte_t*)op, ip, size);
ip += size;
op += size;
// update read
cfilter->read += size;
}
// update stream
tb_static_stream_goto(istream, (tb_byte_t*)ip);
tb_static_stream_goto(ostream, (tb_byte_t*)op);
// trace
tb_trace_d("[%p]: read: %lu, size: %lu, beof: %u, ileft: %lu", cfilter, cfilter->read, cfilter->size, filter->beof, tb_static_stream_left(istream));
// ok
return (op - ob);
}