mirror of
https://github.com/rapid7/metasploit-framework
synced 2024-10-29 18:07:27 +01:00
258 lines
10 KiB
Ruby
258 lines
10 KiB
Ruby
# -*- coding: binary -*-
|
|
# Copyright (C) 2008 TOMITA Masahiro
|
|
# mailto:tommy@tmtm.org
|
|
|
|
require "#{File.dirname __FILE__}/error"
|
|
|
|
class RbMysql
|
|
class Charset
|
|
def initialize(number, name, csname)
|
|
@number, @name, @csname = number, name, csname
|
|
end
|
|
attr_reader :number, :name, :csname
|
|
|
|
# [[charset_number, charset_name, collation_name, default], ...]
|
|
CHARSETS = [
|
|
[ 1, "big5", "big5_chinese_ci", true ],
|
|
[ 2, "latin2", "latin2_czech_cs", false],
|
|
[ 3, "dec8", "dec8_swedish_ci", true ],
|
|
[ 4, "cp850", "cp850_general_ci", true ],
|
|
[ 5, "latin1", "latin1_german1_ci", false],
|
|
[ 6, "hp8", "hp8_english_ci", true ],
|
|
[ 7, "koi8r", "koi8r_general_ci", true ],
|
|
[ 8, "latin1", "latin1_swedish_ci", true ],
|
|
[ 9, "latin2", "latin2_general_ci", true ],
|
|
[ 10, "swe7", "swe7_swedish_ci", true ],
|
|
[ 11, "ascii", "ascii_general_ci", true ],
|
|
[ 12, "ujis", "ujis_japanese_ci", true ],
|
|
[ 13, "sjis", "sjis_japanese_ci", true ],
|
|
[ 14, "cp1251", "cp1251_bulgarian_ci", false],
|
|
[ 15, "latin1", "latin1_danish_ci", false],
|
|
[ 16, "hebrew", "hebrew_general_ci", true ],
|
|
[ 18, "tis620", "tis620_thai_ci", true ],
|
|
[ 19, "euckr", "euckr_korean_ci", true ],
|
|
[ 20, "latin7", "latin7_estonian_cs", false],
|
|
[ 21, "latin2", "latin2_hungarian_ci", false],
|
|
[ 22, "koi8u", "koi8u_general_ci", true ],
|
|
[ 23, "cp1251", "cp1251_ukrainian_ci", false],
|
|
[ 24, "gb2312", "gb2312_chinese_ci", true ],
|
|
[ 25, "greek", "greek_general_ci", true ],
|
|
[ 26, "cp1250", "cp1250_general_ci", true ],
|
|
[ 27, "latin2", "latin2_croatian_ci", false],
|
|
[ 28, "gbk", "gbk_chinese_ci", true ],
|
|
[ 29, "cp1257", "cp1257_lithuanian_ci", false],
|
|
[ 30, "latin5", "latin5_turkish_ci", true ],
|
|
[ 31, "latin1", "latin1_german2_ci", false],
|
|
[ 32, "armscii8", "armscii8_general_ci", true ],
|
|
[ 33, "utf8", "utf8_general_ci", true ],
|
|
[ 34, "cp1250", "cp1250_czech_cs", false],
|
|
[ 35, "ucs2", "ucs2_general_ci", true ],
|
|
[ 36, "cp866", "cp866_general_ci", true ],
|
|
[ 37, "keybcs2", "keybcs2_general_ci", true ],
|
|
[ 38, "macce", "macce_general_ci", true ],
|
|
[ 39, "macroman", "macroman_general_ci", true ],
|
|
[ 40, "cp852", "cp852_general_ci", true ],
|
|
[ 41, "latin7", "latin7_general_ci", true ],
|
|
[ 42, "latin7", "latin7_general_cs", false],
|
|
[ 43, "macce", "macce_bin", false],
|
|
[ 44, "cp1250", "cp1250_croatian_ci", false],
|
|
[ 47, "latin1", "latin1_bin", false],
|
|
[ 48, "latin1", "latin1_general_ci", false],
|
|
[ 49, "latin1", "latin1_general_cs", false],
|
|
[ 50, "cp1251", "cp1251_bin", false],
|
|
[ 51, "cp1251", "cp1251_general_ci", true ],
|
|
[ 52, "cp1251", "cp1251_general_cs", false],
|
|
[ 53, "macroman", "macroman_bin", false],
|
|
[ 57, "cp1256", "cp1256_general_ci", true ],
|
|
[ 58, "cp1257", "cp1257_bin", false],
|
|
[ 59, "cp1257", "cp1257_general_ci", true ],
|
|
[ 63, "binary", "binary", true ],
|
|
[ 64, "armscii8", "armscii8_bin", false],
|
|
[ 65, "ascii", "ascii_bin", false],
|
|
[ 66, "cp1250", "cp1250_bin", false],
|
|
[ 67, "cp1256", "cp1256_bin", false],
|
|
[ 68, "cp866", "cp866_bin", false],
|
|
[ 69, "dec8", "dec8_bin", false],
|
|
[ 70, "greek", "greek_bin", false],
|
|
[ 71, "hebrew", "hebrew_bin", false],
|
|
[ 72, "hp8", "hp8_bin", false],
|
|
[ 73, "keybcs2", "keybcs2_bin", false],
|
|
[ 74, "koi8r", "koi8r_bin", false],
|
|
[ 75, "koi8u", "koi8u_bin", false],
|
|
[ 77, "latin2", "latin2_bin", false],
|
|
[ 78, "latin5", "latin5_bin", false],
|
|
[ 79, "latin7", "latin7_bin", false],
|
|
[ 80, "cp850", "cp850_bin", false],
|
|
[ 81, "cp852", "cp852_bin", false],
|
|
[ 82, "swe7", "swe7_bin", false],
|
|
[ 83, "utf8", "utf8_bin", false],
|
|
[ 84, "big5", "big5_bin", false],
|
|
[ 85, "euckr", "euckr_bin", false],
|
|
[ 86, "gb2312", "gb2312_bin", false],
|
|
[ 87, "gbk", "gbk_bin", false],
|
|
[ 88, "sjis", "sjis_bin", false],
|
|
[ 89, "tis620", "tis620_bin", false],
|
|
[ 90, "ucs2", "ucs2_bin", false],
|
|
[ 91, "ujis", "ujis_bin", false],
|
|
[ 92, "geostd8", "geostd8_general_ci", true ],
|
|
[ 93, "geostd8", "geostd8_bin", false],
|
|
[ 94, "latin1", "latin1_spanish_ci", false],
|
|
[ 95, "cp932", "cp932_japanese_ci" , true ],
|
|
[ 96, "cp932", "cp932_bin" , false],
|
|
[ 97, "eucjpms", "eucjpms_japanese_ci", true ],
|
|
[ 98, "eucjpms", "eucjpms_bin", false],
|
|
[128, "ucs2", "ucs2_unicode_ci", false],
|
|
[129, "ucs2", "ucs2_icelandic_ci", false],
|
|
[130, "ucs2", "ucs2_latvian_ci", false],
|
|
[131, "ucs2", "ucs2_romanian_ci", false],
|
|
[132, "ucs2", "ucs2_slovenian_ci", false],
|
|
[133, "ucs2", "ucs2_polish_ci", false],
|
|
[134, "ucs2", "ucs2_estonian_ci", false],
|
|
[135, "ucs2", "ucs2_spanish_ci", false],
|
|
[136, "ucs2", "ucs2_swedish_ci", false],
|
|
[137, "ucs2", "ucs2_turkish_ci", false],
|
|
[138, "ucs2", "ucs2_czech_ci", false],
|
|
[139, "ucs2", "ucs2_danish_ci", false],
|
|
[140, "ucs2", "ucs2_lithuanian_ci", false],
|
|
[141, "ucs2", "ucs2_slovak_ci", false],
|
|
[142, "ucs2", "ucs2_spanish2_ci", false],
|
|
[143, "ucs2", "ucs2_roman_ci", false],
|
|
[144, "ucs2", "ucs2_persian_ci", false],
|
|
[145, "ucs2", "ucs2_esperanto_ci", false],
|
|
[146, "ucs2", "ucs2_hungarian_ci", false],
|
|
[192, "utf8", "utf8_unicode_ci", false],
|
|
[193, "utf8", "utf8_icelandic_ci", false],
|
|
[194, "utf8", "utf8_latvian_ci", false],
|
|
[195, "utf8", "utf8_romanian_ci", false],
|
|
[196, "utf8", "utf8_slovenian_ci", false],
|
|
[197, "utf8", "utf8_polish_ci", false],
|
|
[198, "utf8", "utf8_estonian_ci", false],
|
|
[199, "utf8", "utf8_spanish_ci", false],
|
|
[200, "utf8", "utf8_swedish_ci", false],
|
|
[201, "utf8", "utf8_turkish_ci", false],
|
|
[202, "utf8", "utf8_czech_ci", false],
|
|
[203, "utf8", "utf8_danish_ci", false],
|
|
[204, "utf8", "utf8_lithuanian_ci", false],
|
|
[205, "utf8", "utf8_slovak_ci", false],
|
|
[206, "utf8", "utf8_spanish2_ci", false],
|
|
[207, "utf8", "utf8_roman_ci", false],
|
|
[208, "utf8", "utf8_persian_ci", false],
|
|
[209, "utf8", "utf8_esperanto_ci", false],
|
|
[210, "utf8", "utf8_hungarian_ci", false],
|
|
]
|
|
|
|
NUMBER_TO_CHARSET = {}
|
|
COLLATION_TO_CHARSET = {}
|
|
CHARSET_DEFAULT = {}
|
|
CHARSETS.each do |number, csname, clname, default|
|
|
cs = Charset.new number, csname, clname
|
|
NUMBER_TO_CHARSET[number] = cs
|
|
COLLATION_TO_CHARSET[clname] = cs
|
|
CHARSET_DEFAULT[csname] = cs if default
|
|
end
|
|
|
|
def self.by_number(n)
|
|
raise ClientError, "unknown charset number: #{n}" unless NUMBER_TO_CHARSET.key? n
|
|
NUMBER_TO_CHARSET[n]
|
|
end
|
|
|
|
def self.by_name(str)
|
|
ret = COLLATION_TO_CHARSET[str] || CHARSET_DEFAULT[str]
|
|
raise ClientError, "unknown charset: #{str}" unless ret
|
|
ret
|
|
end
|
|
|
|
if defined? Encoding
|
|
|
|
# MySQL Charset -> Ruby's Encodeing
|
|
CHARSET_ENCODING = {
|
|
"armscii8" => nil,
|
|
"ascii" => Encoding::US_ASCII,
|
|
"big5" => Encoding::Big5,
|
|
"binary" => Encoding::ASCII_8BIT,
|
|
"cp1250" => Encoding::Windows_1250,
|
|
"cp1251" => Encoding::Windows_1251,
|
|
"cp1256" => Encoding::Windows_1256,
|
|
"cp1257" => Encoding::Windows_1257,
|
|
"cp850" => Encoding::CP850,
|
|
"cp852" => Encoding::CP852,
|
|
"cp866" => Encoding::IBM866,
|
|
"cp932" => Encoding::Windows_31J,
|
|
"dec8" => nil,
|
|
"eucjpms" => Encoding::EucJP_ms,
|
|
"euckr" => Encoding::EUC_KR,
|
|
"gb2312" => Encoding::EUC_CN,
|
|
"gbk" => Encoding::GBK,
|
|
"geostd8" => nil,
|
|
"greek" => Encoding::ISO_8859_7,
|
|
"hebrew" => Encoding::ISO_8859_8,
|
|
"hp8" => nil,
|
|
"keybcs2" => nil,
|
|
"koi8r" => Encoding::KOI8_R,
|
|
"koi8u" => Encoding::KOI8_U,
|
|
"latin1" => Encoding::ISO_8859_1,
|
|
"latin2" => Encoding::ISO_8859_2,
|
|
"latin5" => Encoding::ISO_8859_9,
|
|
"latin7" => Encoding::ISO_8859_13,
|
|
"macce" => Encoding::MacCentEuro,
|
|
"macroman" => Encoding::MacRoman,
|
|
"sjis" => Encoding::SHIFT_JIS,
|
|
"swe7" => nil,
|
|
"tis620" => nil,
|
|
"ucs2" => Encoding::UTF_16BE,
|
|
"ujis" => Encoding::EucJP_ms,
|
|
"utf8" => Encoding::UTF_8,
|
|
}
|
|
|
|
def self.to_binary(value)
|
|
value.dup.force_encoding Encoding::ASCII_8BIT
|
|
end
|
|
|
|
# retrun corresponding Ruby encoding
|
|
# === Return
|
|
# encoding [Encoding]
|
|
def encoding
|
|
enc = CHARSET_ENCODING[@name.downcase]
|
|
raise RbMysql::ClientError, "unsupported charset: #{@name}" unless enc
|
|
enc
|
|
end
|
|
|
|
# convert encoding to corrensponding to MySQL charset
|
|
def convert(value)
|
|
if value.is_a? String and value.encoding != Encoding::ASCII_8BIT
|
|
value = value.encode encoding
|
|
end
|
|
value
|
|
end
|
|
|
|
# convert encoding from MySQL charset to Ruby
|
|
def force_encoding(value)
|
|
if value.is_a? String
|
|
value = value.dup.force_encoding encoding
|
|
end
|
|
value
|
|
end
|
|
|
|
else
|
|
# for Ruby 1.8
|
|
|
|
def self.to_binary(value)
|
|
value
|
|
end
|
|
|
|
def encoding
|
|
nil
|
|
end
|
|
|
|
def convert(value)
|
|
value
|
|
end
|
|
|
|
def force_encoding(value)
|
|
value
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|