* transcode_data.h (rb_transcoding): add fields for restartable

transcoding.
  (rb_transcoder): add max_input field.
  from_unit_length field is renamed to input_unit_length.

* tool/transcode-tblgen.rb: generate max_input field.

* enc/trans/iso2022.erb.c: follow rb_transcoder change.

* enc/trans/utf_16_32.erb.c: ditto.

* transcode.c (PARTIAL_INPUT): new constant.
  (transcode_char_start): new function.
  (transcode_result_t): new type.
  (transcode_restartable): new function.
  (more_output_buffer): new function.
  (transcode_loop): use transcode_restartable.


git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@18452 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
akr 2008-08-09 06:02:01 +00:00
Родитель 750cb61e65
Коммит 139234e1a0
6 изменённых файлов: 439 добавлений и 82 удалений

Просмотреть файл

@ -1,3 +1,23 @@
Sat Aug 9 14:39:34 2008 Tanaka Akira <akr@fsij.org>
* transcode_data.h (rb_transcoding): add fields for restartable
transcoding.
(rb_transcoder): add max_input field.
from_unit_length field is renamed to input_unit_length.
* tool/transcode-tblgen.rb: generate max_input field.
* enc/trans/iso2022.erb.c: follow rb_transcoder change.
* enc/trans/utf_16_32.erb.c: ditto.
* transcode.c (PARTIAL_INPUT): new constant.
(transcode_char_start): new function.
(transcode_result_t): new type.
(transcode_restartable): new function.
(more_output_buffer): new function.
(transcode_loop): use transcode_restartable.
Sat Aug 9 13:35:08 2008 Nobuyoshi Nakada <nobu@ruby-lang.org> Sat Aug 9 13:35:08 2008 Nobuyoshi Nakada <nobu@ruby-lang.org>
* stable/ext/socket/socket.c (NI_MAXHOST, NI_MAXSERV): fixed invalid * stable/ext/socket/socket.c (NI_MAXHOST, NI_MAXSERV): fixed invalid

Просмотреть файл

@ -57,7 +57,10 @@ fun_so_iso2022jp_to_eucjp(rb_transcoding* t, const unsigned char* s, size_t l, u
static const rb_transcoder static const rb_transcoder
rb_ISO_2022_JP_to_EUC_JP = { rb_ISO_2022_JP_to_EUC_JP = {
"ISO-2022-JP", "EUC-JP", &iso2022jp_to_eucjp, 1, 3, "ISO-2022-JP", "EUC-JP", &iso2022jp_to_eucjp,
1, /* input_unit_length */
3, /* max_input */
3, /* max_output */
NULL, fun_si_iso2022jp_to_eucjp, NULL, fun_so_iso2022jp_to_eucjp NULL, fun_si_iso2022jp_to_eucjp, NULL, fun_so_iso2022jp_to_eucjp
}; };
@ -129,7 +132,10 @@ finish_eucjp_to_iso2022jp(rb_transcoding *t, unsigned char *o)
static const rb_transcoder static const rb_transcoder
rb_EUC_JP_to_ISO_2022_JP = { rb_EUC_JP_to_ISO_2022_JP = {
"EUC-JP", "ISO-2022-JP", &eucjp_to_iso2022jp, 1, 5, "EUC-JP", "ISO-2022-JP", &eucjp_to_iso2022jp,
1, /* input_unit_length */
3, /* max_input */
5, /* max_output */
NULL, NULL, NULL, fun_so_eucjp_to_iso2022jp, finish_eucjp_to_iso2022jp NULL, NULL, NULL, fun_so_eucjp_to_iso2022jp, finish_eucjp_to_iso2022jp
}; };

Просмотреть файл

@ -231,7 +231,10 @@ fun_so_to_utf_32le(rb_transcoding* t, const unsigned char* s, size_t l, unsigned
static const rb_transcoder static const rb_transcoder
rb_from_UTF_16BE = { rb_from_UTF_16BE = {
"UTF-16BE", "UTF-8", &from_UTF_16BE, 2, 4, "UTF-16BE", "UTF-8", &from_UTF_16BE,
2, /* input_unit_length */
4, /* max_input */
4, /* max_output */
NULL, NULL, NULL, &fun_so_from_utf_16be NULL, NULL, NULL, &fun_so_from_utf_16be
}; };
@ -252,7 +255,10 @@ rb_from_UTF_16BE = {
static const rb_transcoder static const rb_transcoder
rb_to_UTF_16BE = { rb_to_UTF_16BE = {
"UTF-8", "UTF-16BE", &to_UTF_16BE, 1, 4, "UTF-8", "UTF-16BE", &to_UTF_16BE,
1, /* input_unit_length */
4, /* max_input */
4, /* max_output */
NULL, NULL, NULL, &fun_so_to_utf_16be NULL, NULL, NULL, &fun_so_to_utf_16be
}; };
@ -265,13 +271,19 @@ rb_to_UTF_16BE = {
static const rb_transcoder static const rb_transcoder
rb_from_UTF_16LE = { rb_from_UTF_16LE = {
"UTF-16LE", "UTF-8", &from_UTF_16LE, 2, 4, "UTF-16LE", "UTF-8", &from_UTF_16LE,
2, /* input_unit_length */
4, /* max_input */
4, /* max_output */
NULL, NULL, NULL, &fun_so_from_utf_16le NULL, NULL, NULL, &fun_so_from_utf_16le
}; };
static const rb_transcoder static const rb_transcoder
rb_to_UTF_16LE = { rb_to_UTF_16LE = {
"UTF-8", "UTF-16LE", &to_UTF_16BE, 1, 4, "UTF-8", "UTF-16LE", &to_UTF_16BE,
1, /* input_unit_length */
4, /* max_input */
4, /* max_output */
NULL, NULL, NULL, &fun_so_to_utf_16le NULL, NULL, NULL, &fun_so_to_utf_16le
}; };
@ -284,13 +296,19 @@ rb_to_UTF_16LE = {
static const rb_transcoder static const rb_transcoder
rb_from_UTF_32BE = { rb_from_UTF_32BE = {
"UTF-32BE", "UTF-8", &from_UTF_32BE, 4, 4, "UTF-32BE", "UTF-8", &from_UTF_32BE,
4, /* input_unit_length */
4, /* max_input */
4, /* max_output */
NULL, NULL, NULL, &fun_so_from_utf_32be NULL, NULL, NULL, &fun_so_from_utf_32be
}; };
static const rb_transcoder static const rb_transcoder
rb_to_UTF_32BE = { rb_to_UTF_32BE = {
"UTF-8", "UTF-32BE", &to_UTF_16BE, 1, 4, "UTF-8", "UTF-32BE", &to_UTF_16BE,
1, /* input_unit_length */
4, /* max_input */
4, /* max_output */
NULL, NULL, NULL, &fun_so_to_utf_32be NULL, NULL, NULL, &fun_so_to_utf_32be
}; };
@ -303,13 +321,19 @@ rb_to_UTF_32BE = {
static const rb_transcoder static const rb_transcoder
rb_from_UTF_32LE = { rb_from_UTF_32LE = {
"UTF-32LE", "UTF-8", &from_UTF_32LE, 4, 4, "UTF-32LE", "UTF-8", &from_UTF_32LE,
4, /* input_unit_length */
4, /* max_input */
4, /* max_output */
NULL, NULL, NULL, &fun_so_from_utf_32le NULL, NULL, NULL, &fun_so_from_utf_32le
}; };
static const rb_transcoder static const rb_transcoder
rb_to_UTF_32LE = { rb_to_UTF_32LE = {
"UTF-8", "UTF-32LE", &to_UTF_16BE, 1, 4, "UTF-8", "UTF-32LE", &to_UTF_16BE,
1, /* input_unit_length */
4, /* max_input */
4, /* max_output */
NULL, NULL, NULL, &fun_so_to_utf_32le NULL, NULL, NULL, &fun_so_to_utf_32le
}; };

Просмотреть файл

@ -101,6 +101,22 @@ class StrSet
"\#<#{self.class}: #{self.to_s}>" "\#<#{self.class}: #{self.to_s}>"
end end
def min_length
if @pat.empty?
nil
else
@pat.map {|seq| seq.length }.min
end
end
def max_length
if @pat.empty?
nil
else
@pat.map {|seq| seq.length }.max
end
end
def emptyable? def emptyable?
@pat.any? {|seq| @pat.any? {|seq|
seq.empty? seq.empty?
@ -170,6 +186,10 @@ class ActionMap
">" ">"
end end
def max_input_length
@map.keys.map {|k| k.max_length }.max
end
def empty_action def empty_action
@map.each {|ss, action| @map.each {|ss, action|
return action if ss.emptyable? return action if ss.emptyable?
@ -386,6 +406,8 @@ def transcode_compile_tree(name, from, map)
} }
am = ActionMap.parse(h) am = ActionMap.parse(h)
max_input = am.max_input_length
if ValidEncoding[from] if ValidEncoding[from]
valid_encoding = StrSet.parse(ValidEncoding[from]) valid_encoding = StrSet.parse(ValidEncoding[from])
else else
@ -394,7 +416,7 @@ def transcode_compile_tree(name, from, map)
code = '' code = ''
defined_name = am.generate_node(code, name, valid_encoding) defined_name = am.generate_node(code, name, valid_encoding)
return defined_name, code return defined_name, code, max_input
end end
TRANSCODERS = [] TRANSCODERS = []
@ -411,16 +433,19 @@ def transcode_tblgen(from, to, map)
tree_name = "from_#{id_from}_to_#{id_to}" tree_name = "from_#{id_from}_to_#{id_to}"
end end
map = encode_utf8(map) map = encode_utf8(map)
real_tree_name, tree_code = transcode_compile_tree(tree_name, from, map) real_tree_name, tree_code, max_input = transcode_compile_tree(tree_name, from, map)
transcoder_name = "rb_#{tree_name}" transcoder_name = "rb_#{tree_name}"
TRANSCODERS << transcoder_name TRANSCODERS << transcoder_name
from_unit_length = UnitLength[from] input_unit_length = UnitLength[from]
max_output = map.map {|k,v| String === v ? v.length/2 : 1 }.max max_output = map.map {|k,v| String === v ? v.length/2 : 1 }.max
transcoder_code = <<"End" transcoder_code = <<"End"
static const rb_transcoder static const rb_transcoder
#{transcoder_name} = { #{transcoder_name} = {
#{c_esc from}, #{c_esc to}, &#{real_tree_name}, #{from_unit_length}, #{max_output}, #{c_esc from}, #{c_esc to}, &#{real_tree_name},
NULL, NULL, #{input_unit_length}, /* input_unit_length */
#{max_input}, /* max_input */
#{max_output}, /* max_output */
NULL, NULL, NULL, NULL, NULL
}; };
End End
tree_code + "\n" + transcoder_code tree_code + "\n" + transcoder_code

Просмотреть файл

@ -20,6 +20,7 @@ static VALUE sym_invalid, sym_undef, sym_ignore, sym_replace;
#define INVALID_REPLACE 0x2 #define INVALID_REPLACE 0x2
#define UNDEF_IGNORE 0x10 #define UNDEF_IGNORE 0x10
#define UNDEF_REPLACE 0x20 #define UNDEF_REPLACE 0x20
#define PARTIAL_INPUT 0x100
/* /*
* Dispatch data and logic * Dispatch data and logic
@ -324,34 +325,117 @@ output_replacement_character(unsigned char **out_pp, rb_encoding *enc)
/* /*
* Transcoding engine logic * Transcoding engine logic
*/ */
static void
transcode_loop(const unsigned char **in_pos, unsigned char **out_pos,
const unsigned char *in_stop, unsigned char *out_stop,
const rb_transcoder *my_transcoder,
rb_transcoding *my_transcoding,
const int opt)
{
const unsigned char *in_p = *in_pos;
unsigned char *out_p = *out_pos;
const BYTE_LOOKUP *conv_tree_start = my_transcoder->conv_tree_start;
const BYTE_LOOKUP *next_table;
const unsigned char *char_start;
VALUE next_info;
unsigned char next_byte;
unsigned char *out_s = out_stop - my_transcoder->max_output + 1;
rb_encoding *to_encoding = rb_enc_find(my_transcoder->to_encoding);
while (in_p < in_stop) { static const unsigned char *
char_start = in_p; transcode_char_start(rb_transcoding *my_transcoding,
next_table = conv_tree_start; const unsigned char **in_pos,
if (out_p >= out_s) { const unsigned char *in_p,
int len = (out_p - *out_pos); int readlen)
int new_len = (len + my_transcoder->max_output) * 2; {
*out_pos = (*my_transcoding->flush_func)(my_transcoding, len, new_len); const unsigned char *ptr;
out_p = *out_pos + len; if (in_p - *in_pos < readlen) {
out_s = *out_pos + new_len - my_transcoder->max_output; int restlen = readlen - my_transcoding->readlen;
} MEMCPY(TRANSCODING_READBUF(my_transcoding) + my_transcoding->readlen,
in_p - restlen, unsigned char, restlen);
my_transcoding->readlen = readlen;
ptr = TRANSCODING_READBUF(my_transcoding);
}
else {
ptr = in_p - readlen;
}
return ptr;
}
typedef enum {
transcode_invalid_input,
transcode_undefined_conversion,
transcode_obuf_full,
transcode_ibuf_empty,
transcode_finished,
} transcode_result_t;
static transcode_result_t
transcode_restartable(const unsigned char **in_pos, unsigned char **out_pos,
const unsigned char *in_stop, unsigned char *out_stop,
const rb_transcoder *my_transcoder,
rb_transcoding *my_transcoding,
const int opt)
{
int unitlen = my_transcoder->input_unit_length;
const unsigned char *in_p;
unsigned char *out_p;
int readlen;
const BYTE_LOOKUP *next_table;
unsigned char empty_buf;
unsigned char *empty_ptr = &empty_buf;
if (!in_pos) {
in_pos = (const unsigned char **)&empty_ptr;
in_stop = empty_ptr;
}
if (!out_pos) {
out_pos = &empty_ptr;
out_stop = empty_ptr;
}
in_p = *in_pos;
out_p = *out_pos;
readlen = my_transcoding->readlen;
next_table = my_transcoding->next_table;
#define SUSPEND(ret, num) \
do { \
my_transcoding->resume_position = (num); \
if (my_transcoding->readlen < readlen) \
MEMCPY(TRANSCODING_READBUF(my_transcoding)+my_transcoding->readlen, \
in_p - (readlen-my_transcoding->readlen), \
unsigned char, \
readlen-my_transcoding->readlen); \
*in_pos = in_p; \
*out_pos = out_p; \
my_transcoding->readlen = readlen; \
my_transcoding->next_table = next_table; \
return ret; \
resume_label ## num:; \
} while (0)
switch (my_transcoding->resume_position) {
case 0: break;
case 1: goto resume_label1;
case 2: goto resume_label2;
case 3: goto resume_label3;
case 4: goto resume_label4;
case 5: goto resume_label5;
case 6: goto resume_label6;
case 7: goto resume_label7;
case 8: goto resume_label8;
case 9: goto resume_label9;
case 10: goto resume_label10;
case 11: goto resume_label11;
case 12: goto resume_label12;
case 13: goto resume_label13;
case 14: goto resume_label14;
}
while (1) {
unsigned char next_byte;
VALUE next_info;
if (in_stop <= in_p) {
if (!(opt & PARTIAL_INPUT))
break;
SUSPEND(transcode_ibuf_empty, 7);
continue;
}
my_transcoding->readlen = readlen = 0;
next_table = my_transcoder->conv_tree_start;
next_byte = (unsigned char)*in_p++; next_byte = (unsigned char)*in_p++;
readlen++;
follow_byte: follow_byte:
if (next_byte < next_table->base[0] || next_table->base[1] < next_byte) if (next_byte < next_table->base[0] || next_table->base[1] < next_byte)
next_info = INVALID; next_info = INVALID;
@ -361,32 +445,42 @@ transcode_loop(const unsigned char **in_pos, unsigned char **out_pos,
} }
follow_info: follow_info:
switch (next_info & 0x1F) { switch (next_info & 0x1F) {
case NOMAP: case NOMAP: /* xxx: copy last byte only? */
while (out_stop - out_p < 1) { SUSPEND(transcode_obuf_full, 3); }
*out_p++ = next_byte; *out_p++ = next_byte;
continue; continue;
case 0x00: case 0x04: case 0x08: case 0x0C: case 0x00: case 0x04: case 0x08: case 0x0C:
case 0x10: case 0x14: case 0x18: case 0x1C: case 0x10: case 0x14: case 0x18: case 0x1C:
if (in_p >= in_stop) { while (in_p >= in_stop) {
/* todo: deal with the case of backtracking */ if (!(opt & PARTIAL_INPUT))
/* todo: deal with incomplete input (streaming) */ goto invalid;
goto invalid; SUSPEND(transcode_ibuf_empty, 5);
} }
next_byte = (unsigned char)*in_p++; next_byte = (unsigned char)*in_p++;
readlen++;
next_table = (const BYTE_LOOKUP *)next_info; next_table = (const BYTE_LOOKUP *)next_info;
goto follow_byte; goto follow_byte;
/* maybe rewrite the following cases to use fallthrough???? */ /* maybe rewrite the following cases to use fallthrough???? */
case ZERObt: /* drop input */ case ZERObt: /* drop input */
continue; continue;
case ONEbt: case ONEbt:
while (out_stop - out_p < 1) { SUSPEND(transcode_obuf_full, 9); }
*out_p++ = getBT1(next_info); *out_p++ = getBT1(next_info);
continue; continue;
case TWObt: case TWObt:
while (out_stop - out_p < 2) { SUSPEND(transcode_obuf_full, 10); }
*out_p++ = getBT1(next_info); *out_p++ = getBT1(next_info);
*out_p++ = getBT2(next_info); *out_p++ = getBT2(next_info);
continue; continue;
case THREEbt:
while (out_stop - out_p < 3) { SUSPEND(transcode_obuf_full, 11); }
*out_p++ = getBT1(next_info);
*out_p++ = getBT2(next_info);
*out_p++ = getBT3(next_info);
continue;
case FOURbt: case FOURbt:
while (out_stop - out_p < 4) { SUSPEND(transcode_obuf_full, 12); }
*out_p++ = getBT0(next_info); *out_p++ = getBT0(next_info);
case THREEbt: /* fall through */
*out_p++ = getBT1(next_info); *out_p++ = getBT1(next_info);
*out_p++ = getBT2(next_info); *out_p++ = getBT2(next_info);
*out_p++ = getBT3(next_info); *out_p++ = getBT3(next_info);
@ -395,70 +489,245 @@ transcode_loop(const unsigned char **in_pos, unsigned char **out_pos,
next_info = (VALUE)(*my_transcoder->func_ii)(my_transcoding, next_info); next_info = (VALUE)(*my_transcoder->func_ii)(my_transcoding, next_info);
goto follow_info; goto follow_info;
case FUNsi: case FUNsi:
next_info = (VALUE)(*my_transcoder->func_si)(my_transcoding, char_start, (size_t)(in_p-char_start)); {
goto follow_info; const unsigned char *char_start;
break; char_start = transcode_char_start(my_transcoding, in_pos, in_p, readlen);
next_info = (VALUE)(*my_transcoder->func_si)(my_transcoding, char_start, (size_t)readlen);
break;
}
case FUNio: case FUNio:
while (out_stop - out_p < my_transcoder->max_output) { SUSPEND(transcode_obuf_full, 13); }
out_p += (VALUE)(*my_transcoder->func_io)(my_transcoding, next_info, out_p); out_p += (VALUE)(*my_transcoder->func_io)(my_transcoding, next_info, out_p);
break; break;
case FUNso: case FUNso:
out_p += (VALUE)(*my_transcoder->func_so)(my_transcoding, char_start, (size_t)(in_p-char_start), out_p); {
break; const unsigned char *char_start;
while (out_stop - out_p < my_transcoder->max_output) { SUSPEND(transcode_obuf_full, 14); }
char_start = transcode_char_start(my_transcoding, in_pos, in_p, readlen);
out_p += (VALUE)(*my_transcoder->func_so)(my_transcoding, char_start, (size_t)readlen, out_p);
break;
}
case INVALID: case INVALID:
{ {
int unitlen = my_transcoder->from_unit_length; if (readlen <= unitlen) {
if (in_stop - char_start <= unitlen) while ((opt & PARTIAL_INPUT) && readlen + (in_stop - in_p) < unitlen) {
in_p = in_stop; readlen += in_stop - in_p;
else if (in_p - char_start <= unitlen) in_p = in_stop;
in_p = char_start + unitlen; SUSPEND(transcode_ibuf_empty, 8);
else }
in_p = char_start + ((in_p - char_start - 1) / unitlen) * unitlen; if (readlen + (in_stop - in_p) <= unitlen)
in_p = in_stop;
else
in_p += unitlen - readlen;
}
else {
/* xxx: possibly in_p is lesser than *in_pos
* caller may want to access readbuf. */
in_p += ((readlen - 1) / unitlen) * unitlen - readlen;
}
goto invalid; goto invalid;
} }
case UNDEF: case UNDEF:
goto undef; goto undef;
} }
continue; continue;
invalid: invalid:
SUSPEND(transcode_invalid_input, 1);
continue;
undef:
SUSPEND(transcode_undefined_conversion, 2);
continue;
}
/* cleanup */
if (my_transcoder->finish_func) {
while (out_stop - out_p < my_transcoder->max_output) {
SUSPEND(transcode_obuf_full, 4);
}
out_p += my_transcoder->finish_func(my_transcoding, out_p);
}
while (1)
SUSPEND(transcode_finished, 6);
#undef SUSPEND
}
static void
more_output_buffer(
rb_transcoding *my_transcoding,
unsigned char **out_start_ptr,
unsigned char **out_pos,
unsigned char **out_stop_ptr)
{
size_t len = (*out_pos - *out_start_ptr);
size_t new_len = (len + my_transcoding->transcoder->max_output) * 2;
*out_start_ptr = (*my_transcoding->flush_func)(my_transcoding, len, new_len);
*out_pos = *out_start_ptr + len;
*out_stop_ptr = *out_start_ptr + new_len;
}
#if 1
static void
transcode_loop(const unsigned char **in_pos, unsigned char **out_pos,
const unsigned char *in_stop, unsigned char *out_stop,
const rb_transcoder *my_transcoder,
rb_transcoding *my_transcoding,
const int opt)
{
transcode_result_t ret;
unsigned char *out_start = *out_pos;
my_transcoding->resume_position = 0;
my_transcoding->readlen = 0;
if (sizeof(my_transcoding->readbuf.ary) < my_transcoder->max_input) {
my_transcoding->readbuf.ptr = xmalloc(my_transcoder->max_input);
}
#define CLEANUP \
do { \
if (sizeof(my_transcoding->readbuf.ary) < my_transcoder->max_input) \
xfree(my_transcoding->readbuf.ptr); \
} while(0)
resume:
ret = transcode_restartable(in_pos, out_pos, in_stop, out_stop, my_transcoder, my_transcoding, opt);
if (ret == transcode_invalid_input) {
/* deal with invalid byte sequence */ /* deal with invalid byte sequence */
/* todo: add more alternative behaviors */ /* todo: add more alternative behaviors */
if (opt&INVALID_IGNORE) { if (opt&INVALID_IGNORE) {
continue; goto resume;
} }
else if (opt&INVALID_REPLACE) { else if (opt&INVALID_REPLACE) {
output_replacement_character(&out_p, to_encoding); if (out_stop - *out_pos < my_transcoder->max_output)
continue; more_output_buffer(my_transcoding, &out_start, out_pos, &out_stop);
output_replacement_character(out_pos, rb_enc_find(my_transcoder->to_encoding));
goto resume;
} }
CLEANUP;
rb_raise(TRANSCODE_ERROR, "invalid byte sequence"); rb_raise(TRANSCODE_ERROR, "invalid byte sequence");
continue; }
undef: if (ret == transcode_undefined_conversion) {
/* valid character in from encoding /* valid character in from encoding
* but no related character(s) in to encoding */ * but no related character(s) in to encoding */
/* todo: add more alternative behaviors */ /* todo: add more alternative behaviors */
if (opt&UNDEF_IGNORE) { if (opt&UNDEF_IGNORE) {
continue; goto resume;
} }
else if (opt&UNDEF_REPLACE) { else if (opt&UNDEF_REPLACE) {
output_replacement_character(&out_p, to_encoding); if (out_stop - *out_pos < my_transcoder->max_output)
continue; more_output_buffer(my_transcoding, &out_start, out_pos, &out_stop);
output_replacement_character(out_pos, rb_enc_find(my_transcoder->to_encoding));
goto resume;
} }
rb_raise(TRANSCODE_ERROR, "conversion undefined for byte sequence (maybe invalid byte sequence)"); CLEANUP;
continue; rb_raise(TRANSCODE_ERROR, "conversion undefined for byte sequence (maybe invalid byte sequence)");
} }
/* cleanup */ if (ret == transcode_obuf_full) {
if (my_transcoder->finish_func) { more_output_buffer(my_transcoding, &out_start, out_pos, &out_stop);
if (out_p >= out_s) { goto resume;
int len = (out_p - *out_pos);
int new_len = (len + my_transcoder->max_output) * 2;
*out_pos = (*my_transcoding->flush_func)(my_transcoding, len, new_len);
out_p = *out_pos + len;
out_s = *out_pos + new_len - my_transcoder->max_output;
}
out_p += my_transcoder->finish_func(my_transcoding, out_p);
} }
*in_pos = in_p;
*out_pos = out_p; CLEANUP;
return;
#undef CLEANUP
} }
#else
/* sample transcode_loop implementation in byte-by-byte stream style */
static void
transcode_loop(const unsigned char **in_pos, unsigned char **out_pos,
const unsigned char *in_stop, unsigned char *out_stop,
const rb_transcoder *my_transcoder,
rb_transcoding *my_transcoding,
const int opt)
{
transcode_result_t ret;
unsigned char *out_start = *out_pos;
const unsigned char *ptr;
my_transcoding->resume_position = 0;
my_transcoding->readlen = 0;
if (sizeof(my_transcoding->readbuf.ary) < my_transcoder->max_input) {
my_transcoding->readbuf.ptr = xmalloc(my_transcoder->max_input);
}
#define CLEANUP \
do { \
if (sizeof(my_transcoding->readbuf.ary) < my_transcoder->max_input) \
xfree(my_transcoding->readbuf.ptr); \
} while(0)
ret = transcode_ibuf_empty;
ptr = *in_pos;
while (ret != transcode_finished) {
unsigned char input_byte;
const unsigned char *p = &input_byte;
if (ret == transcode_ibuf_empty) {
if (ptr < in_stop) {
input_byte = *ptr;
ret = transcode_restartable(&p, out_pos, p+1, out_stop, my_transcoder, my_transcoding, opt|PARTIAL_INPUT);
}
else {
ret = transcode_restartable(NULL, out_pos, NULL, out_stop, my_transcoder, my_transcoding, opt);
}
}
else {
ret = transcode_restartable(NULL, out_pos, NULL, out_stop, my_transcoder, my_transcoding, opt|PARTIAL_INPUT);
}
if (&input_byte != p)
ptr += p - &input_byte;
switch (ret) {
case transcode_invalid_input:
/* deal with invalid byte sequence */
/* todo: add more alternative behaviors */
if (opt&INVALID_IGNORE) {
break;
}
else if (opt&INVALID_REPLACE) {
if (out_stop - *out_pos < my_transcoder->max_output)
more_output_buffer(my_transcoding, &out_start, out_pos, &out_stop);
output_replacement_character(out_pos, rb_enc_find(my_transcoder->to_encoding));
break;
}
CLEANUP;
rb_raise(TRANSCODE_ERROR, "invalid byte sequence");
break;
case transcode_undefined_conversion:
/* valid character in from encoding
* but no related character(s) in to encoding */
/* todo: add more alternative behaviors */
if (opt&UNDEF_IGNORE) {
break;
}
else if (opt&UNDEF_REPLACE) {
if (out_stop - *out_pos < my_transcoder->max_output)
more_output_buffer(my_transcoding, &out_start, out_pos, &out_stop);
output_replacement_character(out_pos, rb_enc_find(my_transcoder->to_encoding));
break;
}
CLEANUP;
rb_raise(TRANSCODE_ERROR, "conversion undefined for byte sequence (maybe invalid byte sequence)");
break;
case transcode_obuf_full:
more_output_buffer(my_transcoding, &out_start, out_pos, &out_stop);
break;
case transcode_ibuf_empty:
break;
case transcode_finished:
break;
}
}
CLEANUP;
*in_pos = in_stop;
return;
#undef CLEANUP
}
#endif
/* /*

Просмотреть файл

@ -64,15 +64,28 @@ typedef struct rb_transcoding {
or NULL if something else is being converted */ or NULL if something else is being converted */
unsigned char *(*flush_func)(struct rb_transcoding*, int, int); unsigned char *(*flush_func)(struct rb_transcoding*, int, int);
int resume_position;
const BYTE_LOOKUP *next_table;
int readlen;
union {
unsigned char ary[8]; /* max_input <= sizeof(ary) */
unsigned char *ptr; /* length is max_input */
} readbuf;
unsigned char stateful[256]; /* opaque data for stateful encoding */ unsigned char stateful[256]; /* opaque data for stateful encoding */
} rb_transcoding; } rb_transcoding;
#define TRANSCODING_READBUF(tc) \
((tc)->transcoder->max_input <= sizeof((tc)->readbuf.ary) ? \
(tc)->readbuf.ary : \
(tc)->readbuf.ptr)
/* static structure, one per supported encoding pair */ /* static structure, one per supported encoding pair */
typedef struct rb_transcoder { typedef struct rb_transcoder {
const char *from_encoding; const char *from_encoding;
const char *to_encoding; const char *to_encoding;
const BYTE_LOOKUP *conv_tree_start; const BYTE_LOOKUP *conv_tree_start;
int from_unit_length; int input_unit_length;
int max_input;
int max_output; int max_output;
VALUE (*func_ii)(rb_transcoding*, VALUE); /* info -> info */ VALUE (*func_ii)(rb_transcoding*, VALUE); /* info -> info */
VALUE (*func_si)(rb_transcoding*, const unsigned char*, size_t); /* start -> info */ VALUE (*func_si)(rb_transcoding*, const unsigned char*, size_t); /* start -> info */