diff options
Diffstat (limited to 'ruby_1_9_3/encoding.c')
-rw-r--r-- | ruby_1_9_3/encoding.c | 1662 |
1 files changed, 0 insertions, 1662 deletions
diff --git a/ruby_1_9_3/encoding.c b/ruby_1_9_3/encoding.c deleted file mode 100644 index b8c5f6d357..0000000000 --- a/ruby_1_9_3/encoding.c +++ /dev/null @@ -1,1662 +0,0 @@ -/********************************************************************** - - encoding.c - - - $Author$ - created at: Thu May 24 17:23:27 JST 2007 - - Copyright (C) 2007 Yukihiro Matsumoto - -**********************************************************************/ - -#include "ruby/ruby.h" -#include "ruby/encoding.h" -#include "internal.h" -#include "regenc.h" -#include <ctype.h> -#ifndef NO_LOCALE_CHARMAP -#ifdef __CYGWIN__ -#include <windows.h> -#endif -#ifdef HAVE_LANGINFO_H -#include <langinfo.h> -#endif -#endif -#include "ruby/util.h" - -#if defined __GNUC__ && __GNUC__ >= 4 -#pragma GCC visibility push(default) -int rb_enc_register(const char *name, rb_encoding *encoding); -void rb_enc_set_base(const char *name, const char *orig); -void rb_encdb_declare(const char *name); -int rb_encdb_replicate(const char *name, const char *orig); -int rb_encdb_dummy(const char *name); -int rb_encdb_alias(const char *alias, const char *orig); -#pragma GCC visibility pop -#endif - -static ID id_encoding; -VALUE rb_cEncoding; -static VALUE rb_encoding_list; - -struct rb_encoding_entry { - const char *name; - rb_encoding *enc; - rb_encoding *base; -}; - -static struct { - struct rb_encoding_entry *list; - int count; - int size; - st_table *names; -} enc_table; - -void rb_enc_init(void); - -#define ENCODING_COUNT ENCINDEX_BUILTIN_MAX -#define UNSPECIFIED_ENCODING INT_MAX - -#define ENCODING_NAMELEN_MAX 63 -#define valid_encoding_name_p(name) ((name) && strlen(name) <= ENCODING_NAMELEN_MAX) - -#define enc_autoload_p(enc) (!rb_enc_mbmaxlen(enc)) - -static int load_encoding(const char *name); - -static size_t -enc_memsize(const void *p) -{ - return 0; -} - -static const rb_data_type_t encoding_data_type = { - "encoding", - {0, 0, enc_memsize,}, -}; - -#define is_data_encoding(obj) (RTYPEDDATA_P(obj) && RTYPEDDATA_TYPE(obj) == &encoding_data_type) - -static VALUE -enc_new(rb_encoding *encoding) -{ - return TypedData_Wrap_Struct(rb_cEncoding, &encoding_data_type, encoding); -} - -static VALUE -rb_enc_from_encoding_index(int idx) -{ - VALUE list, enc; - - if (!(list = rb_encoding_list)) { - rb_bug("rb_enc_from_encoding_index(%d): no rb_encoding_list", idx); - } - enc = rb_ary_entry(list, idx); - if (NIL_P(enc)) { - rb_bug("rb_enc_from_encoding_index(%d): not created yet", idx); - } - return enc; -} - -VALUE -rb_enc_from_encoding(rb_encoding *encoding) -{ - int idx; - if (!encoding) return Qnil; - idx = ENC_TO_ENCINDEX(encoding); - return rb_enc_from_encoding_index(idx); -} - -static int enc_autoload(rb_encoding *); - -static int -check_encoding(rb_encoding *enc) -{ - int index = rb_enc_to_index(enc); - if (rb_enc_from_index(index) != enc) - return -1; - if (enc_autoload_p(enc)) { - index = enc_autoload(enc); - } - return index; -} - -static int -enc_check_encoding(VALUE obj) -{ - if (SPECIAL_CONST_P(obj) || !rb_typeddata_is_kind_of(obj, &encoding_data_type)) { - return -1; - } - return check_encoding(RDATA(obj)->data); -} - -static int -must_encoding(VALUE enc) -{ - int index = enc_check_encoding(enc); - if (index < 0) { - rb_raise(rb_eTypeError, "wrong argument type %s (expected Encoding)", - rb_obj_classname(enc)); - } - return index; -} - -int -rb_to_encoding_index(VALUE enc) -{ - int idx; - - idx = enc_check_encoding(enc); - if (idx >= 0) { - return idx; - } - else if (NIL_P(enc = rb_check_string_type(enc))) { - return -1; - } - if (!rb_enc_asciicompat(rb_enc_get(enc))) { - return -1; - } - return rb_enc_find_index(StringValueCStr(enc)); -} - -static rb_encoding * -to_encoding(VALUE enc) -{ - int idx; - - StringValue(enc); - if (!rb_enc_asciicompat(rb_enc_get(enc))) { - rb_raise(rb_eArgError, "invalid name encoding (non ASCII)"); - } - idx = rb_enc_find_index(StringValueCStr(enc)); - if (idx < 0) { - rb_raise(rb_eArgError, "unknown encoding name - %s", RSTRING_PTR(enc)); - } - return rb_enc_from_index(idx); -} - -rb_encoding * -rb_to_encoding(VALUE enc) -{ - if (enc_check_encoding(enc) >= 0) return RDATA(enc)->data; - return to_encoding(enc); -} - -void -rb_gc_mark_encodings(void) -{ -} - -static int -enc_table_expand(int newsize) -{ - struct rb_encoding_entry *ent; - int count = newsize; - - if (enc_table.size >= newsize) return newsize; - newsize = (newsize + 7) / 8 * 8; - ent = realloc(enc_table.list, sizeof(*enc_table.list) * newsize); - if (!ent) return -1; - memset(ent + enc_table.size, 0, sizeof(*ent)*(newsize - enc_table.size)); - enc_table.list = ent; - enc_table.size = newsize; - return count; -} - -static int -enc_register_at(int index, const char *name, rb_encoding *encoding) -{ - struct rb_encoding_entry *ent = &enc_table.list[index]; - VALUE list; - - if (!valid_encoding_name_p(name)) return -1; - if (!ent->name) { - ent->name = name = strdup(name); - } - else if (STRCASECMP(name, ent->name)) { - return -1; - } - if (!ent->enc) { - ent->enc = xmalloc(sizeof(rb_encoding)); - } - if (encoding) { - *ent->enc = *encoding; - } - else { - memset(ent->enc, 0, sizeof(*ent->enc)); - } - encoding = ent->enc; - encoding->name = name; - encoding->ruby_encoding_index = index; - st_insert(enc_table.names, (st_data_t)name, (st_data_t)index); - list = rb_encoding_list; - if (list && NIL_P(rb_ary_entry(list, index))) { - /* initialize encoding data */ - rb_ary_store(list, index, enc_new(encoding)); - } - return index; -} - -static int -enc_register(const char *name, rb_encoding *encoding) -{ - int index = enc_table.count; - - if ((index = enc_table_expand(index + 1)) < 0) return -1; - enc_table.count = index; - return enc_register_at(index - 1, name, encoding); -} - -static void set_encoding_const(const char *, rb_encoding *); -int rb_enc_registered(const char *name); - -int -rb_enc_register(const char *name, rb_encoding *encoding) -{ - int index = rb_enc_registered(name); - - if (index >= 0) { - rb_encoding *oldenc = rb_enc_from_index(index); - if (STRCASECMP(name, rb_enc_name(oldenc))) { - index = enc_register(name, encoding); - } - else if (enc_autoload_p(oldenc) || !ENC_DUMMY_P(oldenc)) { - enc_register_at(index, name, encoding); - } - else { - rb_raise(rb_eArgError, "encoding %s is already registered", name); - } - } - else { - index = enc_register(name, encoding); - set_encoding_const(name, rb_enc_from_index(index)); - } - return index; -} - -void -rb_encdb_declare(const char *name) -{ - int idx = rb_enc_registered(name); - if (idx < 0) { - idx = enc_register(name, 0); - } - set_encoding_const(name, rb_enc_from_index(idx)); -} - -static void -enc_check_duplication(const char *name) -{ - if (rb_enc_registered(name) >= 0) { - rb_raise(rb_eArgError, "encoding %s is already registered", name); - } -} - -static rb_encoding* -set_base_encoding(int index, rb_encoding *base) -{ - rb_encoding *enc = enc_table.list[index].enc; - - enc_table.list[index].base = base; - if (rb_enc_dummy_p(base)) ENC_SET_DUMMY(enc); - return enc; -} - -/* for encdb.h - * Set base encoding for encodings which are not replicas - * but not in their own files. - */ -void -rb_enc_set_base(const char *name, const char *orig) -{ - int idx = rb_enc_registered(name); - int origidx = rb_enc_registered(orig); - set_base_encoding(idx, rb_enc_from_index(origidx)); -} - -int -rb_enc_replicate(const char *name, rb_encoding *encoding) -{ - int idx; - - enc_check_duplication(name); - idx = enc_register(name, encoding); - set_base_encoding(idx, encoding); - set_encoding_const(name, rb_enc_from_index(idx)); - return idx; -} - -/* - * call-seq: - * enc.replicate(name) -> encoding - * - * Returns a replicated encoding of _enc_ whose name is _name_. - * The new encoding should have the same byte structure of _enc_. - * If _name_ is used by another encoding, raise ArgumentError. - * - */ -static VALUE -enc_replicate(VALUE encoding, VALUE name) -{ - return rb_enc_from_encoding_index( - rb_enc_replicate(StringValueCStr(name), - rb_to_encoding(encoding))); -} - -static int -enc_replicate_with_index(const char *name, rb_encoding *origenc, int idx) -{ - if (idx < 0) { - idx = enc_register(name, origenc); - } - else { - idx = enc_register_at(idx, name, origenc); - } - if (idx >= 0) { - set_base_encoding(idx, origenc); - set_encoding_const(name, rb_enc_from_index(idx)); - } - return idx; -} - -int -rb_encdb_replicate(const char *name, const char *orig) -{ - int origidx = rb_enc_registered(orig); - int idx = rb_enc_registered(name); - - if (origidx < 0) { - origidx = enc_register(orig, 0); - } - return enc_replicate_with_index(name, rb_enc_from_index(origidx), idx); -} - -int -rb_define_dummy_encoding(const char *name) -{ - int index = rb_enc_replicate(name, rb_ascii8bit_encoding()); - rb_encoding *enc = enc_table.list[index].enc; - - ENC_SET_DUMMY(enc); - return index; -} - -int -rb_encdb_dummy(const char *name) -{ - int index = enc_replicate_with_index(name, rb_ascii8bit_encoding(), - rb_enc_registered(name)); - rb_encoding *enc = enc_table.list[index].enc; - - ENC_SET_DUMMY(enc); - return index; -} - -/* - * call-seq: - * enc.dummy? -> true or false - * - * Returns true for dummy encodings. - * A dummy encoding is an encoding for which character handling is not properly - * implemented. - * It is used for stateful encodings. - * - * Encoding::ISO_2022_JP.dummy? #=> true - * Encoding::UTF_8.dummy? #=> false - * - */ -static VALUE -enc_dummy_p(VALUE enc) -{ - return ENC_DUMMY_P(enc_table.list[must_encoding(enc)].enc) ? Qtrue : Qfalse; -} - -/* - * call-seq: - * enc.ascii_compatible? -> true or false - * - * Returns whether ASCII-compatible or not. - * - * Encoding::UTF_8.ascii_compatible? #=> true - * Encoding::UTF_16BE.ascii_compatible? #=> false - * - */ -static VALUE -enc_ascii_compatible_p(VALUE enc) -{ - return rb_enc_asciicompat(enc_table.list[must_encoding(enc)].enc) ? Qtrue : Qfalse; -} - -/* - * Returns 1 when the encoding is Unicode series other than UTF-7 else 0. - */ -int -rb_enc_unicode_p(rb_encoding *enc) -{ - const char *name = rb_enc_name(enc); - return name[0] == 'U' && name[1] == 'T' && name[2] == 'F' && name[4] != '7'; -} - -/* - * Returns copied alias name when the key is added for st_table, - * else returns NULL. - */ -static int -enc_alias_internal(const char *alias, int idx) -{ - return st_insert2(enc_table.names, (st_data_t)alias, (st_data_t)idx, - (st_data_t(*)(st_data_t))strdup); -} - -static int -enc_alias(const char *alias, int idx) -{ - if (!valid_encoding_name_p(alias)) return -1; - if (!enc_alias_internal(alias, idx)) - set_encoding_const(alias, rb_enc_from_index(idx)); - return idx; -} - -int -rb_enc_alias(const char *alias, const char *orig) -{ - int idx; - - enc_check_duplication(alias); - if (!enc_table.list) { - rb_enc_init(); - } - if ((idx = rb_enc_find_index(orig)) < 0) { - return -1; - } - return enc_alias(alias, idx); -} - -int -rb_encdb_alias(const char *alias, const char *orig) -{ - int idx = rb_enc_registered(orig); - - if (idx < 0) { - idx = enc_register(orig, 0); - } - return enc_alias(alias, idx); -} - -enum { - ENCINDEX_ASCII, - ENCINDEX_UTF_8, - ENCINDEX_US_ASCII, - ENCINDEX_BUILTIN_MAX -}; - -extern rb_encoding OnigEncodingUTF_8; -extern rb_encoding OnigEncodingUS_ASCII; - -void -rb_enc_init(void) -{ - enc_table_expand(ENCODING_COUNT + 1); - if (!enc_table.names) { - enc_table.names = st_init_strcasetable(); - } -#define ENC_REGISTER(enc) enc_register_at(ENCINDEX_##enc, rb_enc_name(&OnigEncoding##enc), &OnigEncoding##enc) - ENC_REGISTER(ASCII); - ENC_REGISTER(UTF_8); - ENC_REGISTER(US_ASCII); -#undef ENC_REGISTER - enc_table.count = ENCINDEX_BUILTIN_MAX; -} - -rb_encoding * -rb_enc_from_index(int index) -{ - if (!enc_table.list) { - rb_enc_init(); - } - if (index < 0 || enc_table.count <= index) { - return 0; - } - return enc_table.list[index].enc; -} - -int -rb_enc_registered(const char *name) -{ - st_data_t idx = 0; - - if (!name) return -1; - if (!enc_table.list) return -1; - if (st_lookup(enc_table.names, (st_data_t)name, &idx)) { - return (int)idx; - } - return -1; -} - -static VALUE -require_enc(VALUE enclib) -{ - int safe = rb_safe_level(); - return rb_require_safe(enclib, safe > 3 ? 3 : safe); -} - -static int -load_encoding(const char *name) -{ - VALUE enclib = rb_sprintf("enc/%s.so", name); - VALUE verbose = ruby_verbose; - VALUE debug = ruby_debug; - VALUE loaded; - char *s = RSTRING_PTR(enclib) + 4, *e = RSTRING_END(enclib) - 3; - int idx; - - while (s < e) { - if (!ISALNUM(*s)) *s = '_'; - else if (ISUPPER(*s)) *s = TOLOWER(*s); - ++s; - } - FL_UNSET(enclib, FL_TAINT|FL_UNTRUSTED); - OBJ_FREEZE(enclib); - ruby_verbose = Qfalse; - ruby_debug = Qfalse; - loaded = rb_protect(require_enc, enclib, 0); - ruby_verbose = verbose; - ruby_debug = debug; - rb_set_errinfo(Qnil); - if (NIL_P(loaded)) return -1; - if ((idx = rb_enc_registered(name)) < 0) return -1; - if (enc_autoload_p(enc_table.list[idx].enc)) return -1; - return idx; -} - -static int -enc_autoload(rb_encoding *enc) -{ - int i; - rb_encoding *base = enc_table.list[ENC_TO_ENCINDEX(enc)].base; - - if (base) { - i = 0; - do { - if (i >= enc_table.count) return -1; - } while (enc_table.list[i].enc != base && (++i, 1)); - if (enc_autoload_p(base)) { - if (enc_autoload(base) < 0) return -1; - } - i = ENC_TO_ENCINDEX(enc); - enc_register_at(i, rb_enc_name(enc), base); - } - else { - i = load_encoding(rb_enc_name(enc)); - } - return i; -} - -int -rb_enc_find_index(const char *name) -{ - int i = rb_enc_registered(name); - rb_encoding *enc; - - if (i < 0) { - i = load_encoding(name); - } - else if (!(enc = rb_enc_from_index(i))) { - if (i != UNSPECIFIED_ENCODING) { - rb_raise(rb_eArgError, "encoding %s is not registered", name); - } - } - else if (enc_autoload_p(enc)) { - if (enc_autoload(enc) < 0) { - rb_warn("failed to load encoding (%s); use ASCII-8BIT instead", - name); - return 0; - } - } - return i; -} - -rb_encoding * -rb_enc_find(const char *name) -{ - int idx = rb_enc_find_index(name); - if (idx < 0) idx = 0; - return rb_enc_from_index(idx); -} - -static inline int -enc_capable(VALUE obj) -{ - if (SPECIAL_CONST_P(obj)) return SYMBOL_P(obj); - switch (BUILTIN_TYPE(obj)) { - case T_STRING: - case T_REGEXP: - case T_FILE: - return TRUE; - case T_DATA: - if (is_data_encoding(obj)) return TRUE; - default: - return FALSE; - } -} - -ID -rb_id_encoding(void) -{ - CONST_ID(id_encoding, "encoding"); - return id_encoding; -} - -int -rb_enc_get_index(VALUE obj) -{ - int i = -1; - VALUE tmp; - - if (SPECIAL_CONST_P(obj)) { - if (!SYMBOL_P(obj)) return -1; - obj = rb_id2str(SYM2ID(obj)); - } - switch (BUILTIN_TYPE(obj)) { - as_default: - default: - case T_STRING: - case T_REGEXP: - i = ENCODING_GET_INLINED(obj); - if (i == ENCODING_INLINE_MAX) { - VALUE iv; - - iv = rb_ivar_get(obj, rb_id_encoding()); - i = NUM2INT(iv); - } - break; - case T_FILE: - tmp = rb_funcall(obj, rb_intern("internal_encoding"), 0, 0); - if (NIL_P(tmp)) obj = rb_funcall(obj, rb_intern("external_encoding"), 0, 0); - else obj = tmp; - if (NIL_P(obj)) break; - case T_DATA: - if (is_data_encoding(obj)) { - i = enc_check_encoding(obj); - } - else { - goto as_default; - } - break; - } - return i; -} - -static void -enc_set_index(VALUE obj, int idx) -{ - if (idx < ENCODING_INLINE_MAX) { - ENCODING_SET_INLINED(obj, idx); - return; - } - ENCODING_SET_INLINED(obj, ENCODING_INLINE_MAX); - rb_ivar_set(obj, rb_id_encoding(), INT2NUM(idx)); -} - -void -rb_enc_set_index(VALUE obj, int idx) -{ - rb_check_frozen(obj); - enc_set_index(obj, idx); -} - -VALUE -rb_enc_associate_index(VALUE obj, int idx) -{ -/* enc_check_capable(obj);*/ - rb_check_frozen(obj); - if (rb_enc_get_index(obj) == idx) - return obj; - if (SPECIAL_CONST_P(obj)) { - rb_raise(rb_eArgError, "cannot set encoding"); - } - if (!ENC_CODERANGE_ASCIIONLY(obj) || - !rb_enc_asciicompat(rb_enc_from_index(idx))) { - ENC_CODERANGE_CLEAR(obj); - } - enc_set_index(obj, idx); - return obj; -} - -VALUE -rb_enc_associate(VALUE obj, rb_encoding *enc) -{ - return rb_enc_associate_index(obj, rb_enc_to_index(enc)); -} - -rb_encoding* -rb_enc_get(VALUE obj) -{ - return rb_enc_from_index(rb_enc_get_index(obj)); -} - -rb_encoding* -rb_enc_check(VALUE str1, VALUE str2) -{ - rb_encoding *enc = rb_enc_compatible(str1, str2); - if (!enc) - rb_raise(rb_eEncCompatError, "incompatible character encodings: %s and %s", - rb_enc_name(rb_enc_get(str1)), - rb_enc_name(rb_enc_get(str2))); - return enc; -} - -rb_encoding* -rb_enc_compatible(VALUE str1, VALUE str2) -{ - int idx1, idx2; - rb_encoding *enc1, *enc2; - int isstr1, isstr2; - - idx1 = rb_enc_get_index(str1); - idx2 = rb_enc_get_index(str2); - - if (idx1 < 0 || idx2 < 0) - return 0; - - if (idx1 == idx2) { - return rb_enc_from_index(idx1); - } - enc1 = rb_enc_from_index(idx1); - enc2 = rb_enc_from_index(idx2); - - isstr2 = RB_TYPE_P(str2, T_STRING); - if (isstr2 && RSTRING_LEN(str2) == 0) - return enc1; - isstr1 = RB_TYPE_P(str1, T_STRING); - if (isstr1 && RSTRING_LEN(str1) == 0) - return (rb_enc_asciicompat(enc1) && rb_enc_str_asciionly_p(str2)) ? enc1 : enc2; - if (!rb_enc_asciicompat(enc1) || !rb_enc_asciicompat(enc2)) { - return 0; - } - - /* objects whose encoding is the same of contents */ - if (!isstr2 && idx2 == ENCINDEX_US_ASCII) - return enc1; - if (!isstr1 && idx1 == ENCINDEX_US_ASCII) - return enc2; - - if (!isstr1) { - VALUE tmp = str1; - int idx0 = idx1; - str1 = str2; - str2 = tmp; - idx1 = idx2; - idx2 = idx0; - idx0 = isstr1; - isstr1 = isstr2; - isstr2 = idx0; - } - if (isstr1) { - int cr1, cr2; - - cr1 = rb_enc_str_coderange(str1); - if (isstr2) { - cr2 = rb_enc_str_coderange(str2); - if (cr1 != cr2) { - /* may need to handle ENC_CODERANGE_BROKEN */ - if (cr1 == ENC_CODERANGE_7BIT) return enc2; - if (cr2 == ENC_CODERANGE_7BIT) return enc1; - } - if (cr2 == ENC_CODERANGE_7BIT) { - return enc1; - } - } - if (cr1 == ENC_CODERANGE_7BIT) - return enc2; - } - return 0; -} - -void -rb_enc_copy(VALUE obj1, VALUE obj2) -{ - rb_enc_associate_index(obj1, rb_enc_get_index(obj2)); -} - - -/* - * call-seq: - * obj.encoding -> encoding - * - * Returns the Encoding object that represents the encoding of obj. - */ - -VALUE -rb_obj_encoding(VALUE obj) -{ - rb_encoding *enc = rb_enc_get(obj); - if (!enc) { - rb_raise(rb_eTypeError, "unknown encoding"); - } - return rb_enc_from_encoding(enc); -} - -int -rb_enc_fast_mbclen(const char *p, const char *e, rb_encoding *enc) -{ - return ONIGENC_MBC_ENC_LEN(enc, (UChar*)p, (UChar*)e); -} - -int -rb_enc_mbclen(const char *p, const char *e, rb_encoding *enc) -{ - int n = ONIGENC_PRECISE_MBC_ENC_LEN(enc, (UChar*)p, (UChar*)e); - if (MBCLEN_CHARFOUND_P(n) && MBCLEN_CHARFOUND_LEN(n) <= e-p) - return MBCLEN_CHARFOUND_LEN(n); - else { - int min = rb_enc_mbminlen(enc); - return min <= e-p ? min : (int)(e-p); - } -} - -int -rb_enc_precise_mbclen(const char *p, const char *e, rb_encoding *enc) -{ - int n; - if (e <= p) - return ONIGENC_CONSTRUCT_MBCLEN_NEEDMORE(1); - n = ONIGENC_PRECISE_MBC_ENC_LEN(enc, (UChar*)p, (UChar*)e); - if (e-p < n) - return ONIGENC_CONSTRUCT_MBCLEN_NEEDMORE(n-(int)(e-p)); - return n; -} - -int -rb_enc_ascget(const char *p, const char *e, int *len, rb_encoding *enc) -{ - unsigned int c, l; - if (e <= p) - return -1; - if (rb_enc_asciicompat(enc)) { - c = (unsigned char)*p; - if (!ISASCII(c)) - return -1; - if (len) *len = 1; - return c; - } - l = rb_enc_precise_mbclen(p, e, enc); - if (!MBCLEN_CHARFOUND_P(l)) - return -1; - c = rb_enc_mbc_to_codepoint(p, e, enc); - if (!rb_enc_isascii(c, enc)) - return -1; - if (len) *len = l; - return c; -} - -unsigned int -rb_enc_codepoint_len(const char *p, const char *e, int *len_p, rb_encoding *enc) -{ - int r; - if (e <= p) - rb_raise(rb_eArgError, "empty string"); - r = rb_enc_precise_mbclen(p, e, enc); - if (MBCLEN_CHARFOUND_P(r)) { - if (len_p) *len_p = MBCLEN_CHARFOUND_LEN(r); - return rb_enc_mbc_to_codepoint(p, e, enc); - } - else - rb_raise(rb_eArgError, "invalid byte sequence in %s", rb_enc_name(enc)); -} - -#undef rb_enc_codepoint -unsigned int -rb_enc_codepoint(const char *p, const char *e, rb_encoding *enc) -{ - return rb_enc_codepoint_len(p, e, 0, enc); -} - -int -rb_enc_codelen(int c, rb_encoding *enc) -{ - int n = ONIGENC_CODE_TO_MBCLEN(enc,c); - if (n == 0) { - rb_raise(rb_eArgError, "invalid codepoint 0x%x in %s", c, rb_enc_name(enc)); - } - return n; -} - -int -rb_enc_toupper(int c, rb_encoding *enc) -{ - return (ONIGENC_IS_ASCII_CODE(c)?ONIGENC_ASCII_CODE_TO_UPPER_CASE(c):(c)); -} - -int -rb_enc_tolower(int c, rb_encoding *enc) -{ - return (ONIGENC_IS_ASCII_CODE(c)?ONIGENC_ASCII_CODE_TO_LOWER_CASE(c):(c)); -} - -/* - * call-seq: - * enc.inspect -> string - * - * Returns a string which represents the encoding for programmers. - * - * Encoding::UTF_8.inspect #=> "#<Encoding:UTF-8>" - * Encoding::ISO_2022_JP.inspect #=> "#<Encoding:ISO-2022-JP (dummy)>" - */ -static VALUE -enc_inspect(VALUE self) -{ - VALUE str = rb_sprintf("#<%s:%s%s>", rb_obj_classname(self), - rb_enc_name((rb_encoding*)DATA_PTR(self)), - (enc_dummy_p(self) ? " (dummy)" : "")); - ENCODING_CODERANGE_SET(str, rb_usascii_encindex(), ENC_CODERANGE_7BIT); - return str; -} - -/* - * call-seq: - * enc.name -> string - * - * Returns the name of the encoding. - * - * Encoding::UTF_8.name #=> "UTF-8" - */ -static VALUE -enc_name(VALUE self) -{ - return rb_usascii_str_new2(rb_enc_name((rb_encoding*)DATA_PTR(self))); -} - -static int -enc_names_i(st_data_t name, st_data_t idx, st_data_t args) -{ - VALUE *arg = (VALUE *)args; - - if ((int)idx == (int)arg[0]) { - VALUE str = rb_usascii_str_new2((char *)name); - OBJ_FREEZE(str); - rb_ary_push(arg[1], str); - } - return ST_CONTINUE; -} - -/* - * call-seq: - * enc.names -> array - * - * Returns the list of name and aliases of the encoding. - * - * Encoding::WINDOWS_31J.names #=> ["Windows-31J", "CP932", "csWindows31J"] - */ -static VALUE -enc_names(VALUE self) -{ - VALUE args[2]; - - args[0] = (VALUE)rb_to_encoding_index(self); - args[1] = rb_ary_new2(0); - st_foreach(enc_table.names, enc_names_i, (st_data_t)args); - return args[1]; -} - -/* - * call-seq: - * Encoding.list -> [enc1, enc2, ...] - * - * Returns the list of loaded encodings. - * - * Encoding.list - * #=> [#<Encoding:ASCII-8BIT>, #<Encoding:UTF-8>, - * #<Encoding:ISO-2022-JP (dummy)>] - * - * Encoding.find("US-ASCII") - * #=> #<Encoding:US-ASCII> - * - * Encoding.list - * #=> [#<Encoding:ASCII-8BIT>, #<Encoding:UTF-8>, - * #<Encoding:US-ASCII>, #<Encoding:ISO-2022-JP (dummy)>] - * - */ -static VALUE -enc_list(VALUE klass) -{ - VALUE ary = rb_ary_new2(0); - rb_ary_replace(ary, rb_encoding_list); - return ary; -} - -/* - * call-seq: - * Encoding.find(string) -> enc - * Encoding.find(symbol) -> enc - * - * Search the encoding with specified <i>name</i>. - * <i>name</i> should be a string or symbol. - * - * Encoding.find("US-ASCII") #=> #<Encoding:US-ASCII> - * Encoding.find(:Shift_JIS) #=> #<Encoding:Shift_JIS> - * - * Names which this method accept are encoding names and aliases - * including following special aliases - * - * "external":: default external encoding - * "internal":: default internal encoding - * "locale":: locale encoding - * "filesystem":: filesystem encoding - * - * An ArgumentError is raised when no encoding with <i>name</i>. - * Only <code>Encoding.find("internal")</code> however returns nil - * when no encoding named "internal", in other words, when Ruby has no - * default internal encoding. - */ -static VALUE -enc_find(VALUE klass, VALUE enc) -{ - return rb_enc_from_encoding(rb_to_encoding(enc)); -} - -/* - * call-seq: - * Encoding.compatible?(obj1, obj2) -> enc or nil - * - * Checks the compatibility of two objects. - * - * If the objects are both strings they are compatible when they are - * concatenatable. The encoding of the concatenated string will be returned - * if they are compatible, nil if they are not. - * - * Encoding.compatible?("\xa1".force_encoding("iso-8859-1"), "b") - * #=> #<Encoding:ISO-8859-1> - * - * Encoding.compatible?( - * "\xa1".force_encoding("iso-8859-1"), - * "\xa1\xa1".force_encoding("euc-jp")) - * #=> nil - * - * If the objects are non-strings their encodings are compatible when they - * have an encoding and: - * * Either encoding is US-ASCII compatible - * * One of the encodings is a 7-bit encoding - * - */ -static VALUE -enc_compatible_p(VALUE klass, VALUE str1, VALUE str2) -{ - rb_encoding *enc; - - if (!enc_capable(str1)) return Qnil; - if (!enc_capable(str2)) return Qnil; - enc = rb_enc_compatible(str1, str2); - if (!enc) return Qnil; - return rb_enc_from_encoding(enc); -} - -/* :nodoc: */ -static VALUE -enc_dump(int argc, VALUE *argv, VALUE self) -{ - rb_scan_args(argc, argv, "01", 0); - return enc_name(self); -} - -/* :nodoc: */ -static VALUE -enc_load(VALUE klass, VALUE str) -{ - return enc_find(klass, str); -} - -rb_encoding * -rb_ascii8bit_encoding(void) -{ - if (!enc_table.list) { - rb_enc_init(); - } - return enc_table.list[ENCINDEX_ASCII].enc; -} - -int -rb_ascii8bit_encindex(void) -{ - return ENCINDEX_ASCII; -} - -rb_encoding * -rb_utf8_encoding(void) -{ - if (!enc_table.list) { - rb_enc_init(); - } - return enc_table.list[ENCINDEX_UTF_8].enc; -} - -int -rb_utf8_encindex(void) -{ - return ENCINDEX_UTF_8; -} - -rb_encoding * -rb_usascii_encoding(void) -{ - if (!enc_table.list) { - rb_enc_init(); - } - return enc_table.list[ENCINDEX_US_ASCII].enc; -} - -int -rb_usascii_encindex(void) -{ - return ENCINDEX_US_ASCII; -} - -int -rb_locale_encindex(void) -{ - VALUE charmap = rb_locale_charmap(rb_cEncoding); - int idx; - - if (NIL_P(charmap)) - idx = rb_usascii_encindex(); - else if ((idx = rb_enc_find_index(StringValueCStr(charmap))) < 0) - idx = rb_ascii8bit_encindex(); - - if (rb_enc_registered("locale") < 0) enc_alias_internal("locale", idx); - - return idx; -} - -rb_encoding * -rb_locale_encoding(void) -{ - return rb_enc_from_index(rb_locale_encindex()); -} - -static int -enc_set_filesystem_encoding(void) -{ - int idx; -#if defined NO_LOCALE_CHARMAP - idx = rb_enc_to_index(rb_default_external_encoding()); -#elif defined _WIN32 || defined __CYGWIN__ - char cp[sizeof(int) * 8 / 3 + 4]; - snprintf(cp, sizeof cp, "CP%d", AreFileApisANSI() ? GetACP() : GetOEMCP()); - idx = rb_enc_find_index(cp); - if (idx < 0) idx = rb_ascii8bit_encindex(); -#else - idx = rb_enc_to_index(rb_default_external_encoding()); -#endif - - enc_alias_internal("filesystem", idx); - return idx; -} - -int -rb_filesystem_encindex(void) -{ - int idx = rb_enc_registered("filesystem"); - if (idx < 0) - idx = rb_ascii8bit_encindex(); - return idx; -} - -rb_encoding * -rb_filesystem_encoding(void) -{ - return rb_enc_from_index(rb_filesystem_encindex()); -} - -struct default_encoding { - int index; /* -2 => not yet set, -1 => nil */ - rb_encoding *enc; -}; - -static struct default_encoding default_external = {0}; - -static int -enc_set_default_encoding(struct default_encoding *def, VALUE encoding, const char *name) -{ - int overridden = FALSE; - - if (def->index != -2) - /* Already set */ - overridden = TRUE; - - if (NIL_P(encoding)) { - def->index = -1; - def->enc = 0; - st_insert(enc_table.names, (st_data_t)strdup(name), - (st_data_t)UNSPECIFIED_ENCODING); - } - else { - def->index = rb_enc_to_index(rb_to_encoding(encoding)); - def->enc = 0; - enc_alias_internal(name, def->index); - } - - if (def == &default_external) - enc_set_filesystem_encoding(); - - return overridden; -} - -rb_encoding * -rb_default_external_encoding(void) -{ - if (default_external.enc) return default_external.enc; - - if (default_external.index >= 0) { - default_external.enc = rb_enc_from_index(default_external.index); - return default_external.enc; - } - else { - return rb_locale_encoding(); - } -} - -VALUE -rb_enc_default_external(void) -{ - return rb_enc_from_encoding(rb_default_external_encoding()); -} - -/* - * call-seq: - * Encoding.default_external -> enc - * - * Returns default external encoding. - * - * The default external encoding is used by default for strings created from - * the following locations: - * - * * CSV - * * File data read from disk - * * SDBM - * * StringIO - * * Zlib::GzipReader - * * Zlib::GzipWriter - * * String#inspect - * * Regexp#inspect - * - * While strings created from these locations will have this encoding, the - * encoding may not be valid. Be sure to check String#valid_encoding?. - * - * File data written to disk will be transcoded to the default external - * encoding when written. - * - * The default external encoding is initialized by the locale or -E option. - */ -static VALUE -get_default_external(VALUE klass) -{ - return rb_enc_default_external(); -} - -void -rb_enc_set_default_external(VALUE encoding) -{ - if (NIL_P(encoding)) { - rb_raise(rb_eArgError, "default external can not be nil"); - } - enc_set_default_encoding(&default_external, encoding, - "external"); -} - -/* - * call-seq: - * Encoding.default_external = enc - * - * Sets default external encoding. You should not set - * Encoding::default_external in ruby code as strings created before changing - * the value may have a different encoding from strings created after thevalue - * was changed., instead you should use <tt>ruby -E</tt> to invoke ruby with - * the correct default_external. - * - * See Encoding::default_external for information on how the default external - * encoding is used. - */ -static VALUE -set_default_external(VALUE klass, VALUE encoding) -{ - rb_warning("setting Encoding.default_external"); - rb_enc_set_default_external(encoding); - return encoding; -} - -static struct default_encoding default_internal = {-2}; - -rb_encoding * -rb_default_internal_encoding(void) -{ - if (!default_internal.enc && default_internal.index >= 0) { - default_internal.enc = rb_enc_from_index(default_internal.index); - } - return default_internal.enc; /* can be NULL */ -} - -VALUE -rb_enc_default_internal(void) -{ - /* Note: These functions cope with default_internal not being set */ - return rb_enc_from_encoding(rb_default_internal_encoding()); -} - -/* - * call-seq: - * Encoding.default_internal -> enc - * - * Returns default internal encoding. Strings will be transcoded to the - * default internal encoding in the following places if the default internal - * encoding is not nil: - * - * * CSV - * * Etc.sysconfdir and Etc.systmpdir - * * File data read from disk - * * File names from Dir - * * Integer#chr - * * String#inspect and Regexp#inspect - * * Strings returned from Curses - * * Strings returned from Readline - * * Strings returned from SDBM - * * Time#zone - * * Values from ENV - * * Values in ARGV including $PROGRAM_NAME - * * __FILE__ - * - * Additionally String#encode and String#encode! use the default internal - * encoding if no encoding is given. - * - * The locale encoding (__ENCODING__), not default_internal, is used as the - * encoding of created strings. - * - * Encoding::default_internal is initialized by the source file's - * internal_encoding or -E option. - */ -static VALUE -get_default_internal(VALUE klass) -{ - return rb_enc_default_internal(); -} - -void -rb_enc_set_default_internal(VALUE encoding) -{ - enc_set_default_encoding(&default_internal, encoding, - "internal"); -} - -/* - * call-seq: - * Encoding.default_internal = enc or nil - * - * Sets default internal encoding or removes default internal encoding when - * passed nil. You should not set Encoding::default_internal in ruby code as - * strings created before changing the value may have a different encoding - * from strings created after the change. Instead you should use - * <tt>ruby -E</tt> to invoke ruby with the correct default_internal. - * - * See Encoding::default_internal for information on how the default internal - * encoding is used. - */ -static VALUE -set_default_internal(VALUE klass, VALUE encoding) -{ - rb_warning("setting Encoding.default_internal"); - rb_enc_set_default_internal(encoding); - return encoding; -} - -/* - * call-seq: - * Encoding.locale_charmap -> string - * - * Returns the locale charmap name. - * It returns nil if no appropriate information. - * - * Debian GNU/Linux - * LANG=C - * Encoding.locale_charmap #=> "ANSI_X3.4-1968" - * LANG=ja_JP.EUC-JP - * Encoding.locale_charmap #=> "EUC-JP" - * - * SunOS 5 - * LANG=C - * Encoding.locale_charmap #=> "646" - * LANG=ja - * Encoding.locale_charmap #=> "eucJP" - * - * The result is highly platform dependent. - * So Encoding.find(Encoding.locale_charmap) may cause an error. - * If you need some encoding object even for unknown locale, - * Encoding.find("locale") can be used. - * - */ -VALUE -rb_locale_charmap(VALUE klass) -{ -#if defined NO_LOCALE_CHARMAP - return rb_usascii_str_new2("ASCII-8BIT"); -#elif defined _WIN32 || defined __CYGWIN__ - const char *nl_langinfo_codeset(void); - const char *codeset = nl_langinfo_codeset(); - char cp[sizeof(int) * 3 + 4]; - if (!codeset) { - UINT codepage = GetConsoleCP(); - if(!codepage) codepage = GetACP(); - snprintf(cp, sizeof(cp), "CP%d", codepage); - codeset = cp; - } - return rb_usascii_str_new2(codeset); -#elif defined HAVE_LANGINFO_H - char *codeset; - codeset = nl_langinfo(CODESET); - return rb_usascii_str_new2(codeset); -#else - return Qnil; -#endif -} - -static void -set_encoding_const(const char *name, rb_encoding *enc) -{ - VALUE encoding = rb_enc_from_encoding(enc); - char *s = (char *)name; - int haslower = 0, hasupper = 0, valid = 0; - - if (ISDIGIT(*s)) return; - if (ISUPPER(*s)) { - hasupper = 1; - while (*++s && (ISALNUM(*s) || *s == '_')) { - if (ISLOWER(*s)) haslower = 1; - } - } - if (!*s) { - if (s - name > ENCODING_NAMELEN_MAX) return; - valid = 1; - rb_define_const(rb_cEncoding, name, encoding); - } - if (!valid || haslower) { - size_t len = s - name; - if (len > ENCODING_NAMELEN_MAX) return; - if (!haslower || !hasupper) { - do { - if (ISLOWER(*s)) haslower = 1; - if (ISUPPER(*s)) hasupper = 1; - } while (*++s && (!haslower || !hasupper)); - len = s - name; - } - len += strlen(s); - if (len++ > ENCODING_NAMELEN_MAX) return; - MEMCPY(s = ALLOCA_N(char, len), name, char, len); - name = s; - if (!valid) { - if (ISLOWER(*s)) *s = ONIGENC_ASCII_CODE_TO_UPPER_CASE((int)*s); - for (; *s; ++s) { - if (!ISALNUM(*s)) *s = '_'; - } - if (hasupper) { - rb_define_const(rb_cEncoding, name, encoding); - } - } - if (haslower) { - for (s = (char *)name; *s; ++s) { - if (ISLOWER(*s)) *s = ONIGENC_ASCII_CODE_TO_UPPER_CASE((int)*s); - } - rb_define_const(rb_cEncoding, name, encoding); - } - } -} - -static int -rb_enc_name_list_i(st_data_t name, st_data_t idx, st_data_t arg) -{ - VALUE ary = (VALUE)arg; - VALUE str = rb_usascii_str_new2((char *)name); - OBJ_FREEZE(str); - rb_ary_push(ary, str); - return ST_CONTINUE; -} - -/* - * call-seq: - * Encoding.name_list -> ["enc1", "enc2", ...] - * - * Returns the list of available encoding names. - * - * Encoding.name_list - * #=> ["US-ASCII", "ASCII-8BIT", "UTF-8", - * "ISO-8859-1", "Shift_JIS", "EUC-JP", - * "Windows-31J", - * "BINARY", "CP932", "eucJP"] - * - */ - -static VALUE -rb_enc_name_list(VALUE klass) -{ - VALUE ary = rb_ary_new2(enc_table.names->num_entries); - st_foreach(enc_table.names, rb_enc_name_list_i, (st_data_t)ary); - return ary; -} - -static int -rb_enc_aliases_enc_i(st_data_t name, st_data_t orig, st_data_t arg) -{ - VALUE *p = (VALUE *)arg; - VALUE aliases = p[0], ary = p[1]; - int idx = (int)orig; - VALUE key, str = rb_ary_entry(ary, idx); - - if (NIL_P(str)) { - rb_encoding *enc = rb_enc_from_index(idx); - - if (!enc) return ST_CONTINUE; - if (STRCASECMP((char*)name, rb_enc_name(enc)) == 0) { - return ST_CONTINUE; - } - str = rb_usascii_str_new2(rb_enc_name(enc)); - OBJ_FREEZE(str); - rb_ary_store(ary, idx, str); - } - key = rb_usascii_str_new2((char *)name); - OBJ_FREEZE(key); - rb_hash_aset(aliases, key, str); - return ST_CONTINUE; -} - -/* - * call-seq: - * Encoding.aliases -> {"alias1" => "orig1", "alias2" => "orig2", ...} - * - * Returns the hash of available encoding alias and original encoding name. - * - * Encoding.aliases - * #=> {"BINARY"=>"ASCII-8BIT", "ASCII"=>"US-ASCII", "ANSI_X3.4-1986"=>"US-ASCII", - * "SJIS"=>"Shift_JIS", "eucJP"=>"EUC-JP", "CP932"=>"Windows-31J"} - * - */ - -static VALUE -rb_enc_aliases(VALUE klass) -{ - VALUE aliases[2]; - aliases[0] = rb_hash_new(); - aliases[1] = rb_ary_new(); - st_foreach(enc_table.names, rb_enc_aliases_enc_i, (st_data_t)aliases); - return aliases[0]; -} - -void -Init_Encoding(void) -{ -#undef rb_intern -#define rb_intern(str) rb_intern_const(str) - VALUE list; - int i; - - rb_cEncoding = rb_define_class("Encoding", rb_cObject); - rb_undef_alloc_func(rb_cEncoding); - rb_undef_method(CLASS_OF(rb_cEncoding), "new"); - rb_define_method(rb_cEncoding, "to_s", enc_name, 0); - rb_define_method(rb_cEncoding, "inspect", enc_inspect, 0); - rb_define_method(rb_cEncoding, "name", enc_name, 0); - rb_define_method(rb_cEncoding, "names", enc_names, 0); - rb_define_method(rb_cEncoding, "dummy?", enc_dummy_p, 0); - rb_define_method(rb_cEncoding, "ascii_compatible?", enc_ascii_compatible_p, 0); - rb_define_method(rb_cEncoding, "replicate", enc_replicate, 1); - rb_define_singleton_method(rb_cEncoding, "list", enc_list, 0); - rb_define_singleton_method(rb_cEncoding, "name_list", rb_enc_name_list, 0); - rb_define_singleton_method(rb_cEncoding, "aliases", rb_enc_aliases, 0); - rb_define_singleton_method(rb_cEncoding, "find", enc_find, 1); - rb_define_singleton_method(rb_cEncoding, "compatible?", enc_compatible_p, 2); - - rb_define_method(rb_cEncoding, "_dump", enc_dump, -1); - rb_define_singleton_method(rb_cEncoding, "_load", enc_load, 1); - - rb_define_singleton_method(rb_cEncoding, "default_external", get_default_external, 0); - rb_define_singleton_method(rb_cEncoding, "default_external=", set_default_external, 1); - rb_define_singleton_method(rb_cEncoding, "default_internal", get_default_internal, 0); - rb_define_singleton_method(rb_cEncoding, "default_internal=", set_default_internal, 1); - rb_define_singleton_method(rb_cEncoding, "locale_charmap", rb_locale_charmap, 0); - - list = rb_ary_new2(enc_table.count); - RBASIC(list)->klass = 0; - rb_encoding_list = list; - rb_gc_register_mark_object(list); - - for (i = 0; i < enc_table.count; ++i) { - rb_ary_push(list, enc_new(enc_table.list[i].enc)); - } -} - -/* locale insensitive ctype functions */ - -#define ctype_test(c, ctype) \ - (rb_isascii(c) && ONIGENC_IS_ASCII_CODE_CTYPE((c), (ctype))) - -int rb_isalnum(int c) { return ctype_test(c, ONIGENC_CTYPE_ALNUM); } -int rb_isalpha(int c) { return ctype_test(c, ONIGENC_CTYPE_ALPHA); } -int rb_isblank(int c) { return ctype_test(c, ONIGENC_CTYPE_BLANK); } -int rb_iscntrl(int c) { return ctype_test(c, ONIGENC_CTYPE_CNTRL); } -int rb_isdigit(int c) { return ctype_test(c, ONIGENC_CTYPE_DIGIT); } -int rb_isgraph(int c) { return ctype_test(c, ONIGENC_CTYPE_GRAPH); } -int rb_islower(int c) { return ctype_test(c, ONIGENC_CTYPE_LOWER); } -int rb_isprint(int c) { return ctype_test(c, ONIGENC_CTYPE_PRINT); } -int rb_ispunct(int c) { return ctype_test(c, ONIGENC_CTYPE_PUNCT); } -int rb_isspace(int c) { return ctype_test(c, ONIGENC_CTYPE_SPACE); } -int rb_isupper(int c) { return ctype_test(c, ONIGENC_CTYPE_UPPER); } -int rb_isxdigit(int c) { return ctype_test(c, ONIGENC_CTYPE_XDIGIT); } - -int -rb_tolower(int c) -{ - return rb_isascii(c) ? ONIGENC_ASCII_CODE_TO_LOWER_CASE(c) : c; -} - -int -rb_toupper(int c) -{ - return rb_isascii(c) ? ONIGENC_ASCII_CODE_TO_UPPER_CASE(c) : c; -} - |