зеркало из https://github.com/github/ruby.git
191 строка
3.9 KiB
C
191 строка
3.9 KiB
C
#include "ruby.h"
|
|
#include "ruby/encoding.h"
|
|
|
|
RUBY_EXTERN const char ruby_hexdigits[];
|
|
#define lower_hexdigits (ruby_hexdigits+0)
|
|
#define upper_hexdigits (ruby_hexdigits+16)
|
|
|
|
static VALUE rb_cCGI, rb_mUtil, rb_mEscape;
|
|
|
|
static void
|
|
html_escaped_cat(VALUE str, char c)
|
|
{
|
|
switch (c) {
|
|
case '\'':
|
|
rb_str_cat_cstr(str, "'");
|
|
break;
|
|
case '&':
|
|
rb_str_cat_cstr(str, "&");
|
|
break;
|
|
case '"':
|
|
rb_str_cat_cstr(str, """);
|
|
break;
|
|
case '<':
|
|
rb_str_cat_cstr(str, "<");
|
|
break;
|
|
case '>':
|
|
rb_str_cat_cstr(str, ">");
|
|
break;
|
|
}
|
|
}
|
|
|
|
static inline void
|
|
preserve_original_state(VALUE orig, VALUE dest)
|
|
{
|
|
rb_enc_associate(dest, rb_enc_get(orig));
|
|
|
|
RB_OBJ_INFECT_RAW(dest, orig);
|
|
}
|
|
|
|
static VALUE
|
|
optimized_escape_html(VALUE str)
|
|
{
|
|
long i, len, beg = 0;
|
|
VALUE dest = 0;
|
|
const char *cstr;
|
|
|
|
len = RSTRING_LEN(str);
|
|
cstr = RSTRING_PTR(str);
|
|
|
|
for (i = 0; i < len; i++) {
|
|
switch (cstr[i]) {
|
|
case '\'':
|
|
case '&':
|
|
case '"':
|
|
case '<':
|
|
case '>':
|
|
if (!dest) {
|
|
dest = rb_str_buf_new(len);
|
|
}
|
|
|
|
rb_str_cat(dest, cstr + beg, i - beg);
|
|
beg = i + 1;
|
|
|
|
html_escaped_cat(dest, cstr[i]);
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (dest) {
|
|
rb_str_cat(dest, cstr + beg, len - beg);
|
|
preserve_original_state(str, dest);
|
|
return dest;
|
|
}
|
|
else {
|
|
return rb_str_dup(str);
|
|
}
|
|
}
|
|
|
|
static int
|
|
url_unreserved_char(unsigned char c)
|
|
{
|
|
switch (c) {
|
|
case '0': case '1': case '2': case '3': case '4': case '5': case '6': case '7': case '8': case '9':
|
|
case 'a': case 'b': case 'c': case 'd': case 'e': case 'f': case 'g': case 'h': case 'i': case 'j':
|
|
case 'k': case 'l': case 'm': case 'n': case 'o': case 'p': case 'q': case 'r': case 's': case 't':
|
|
case 'u': case 'v': case 'w': case 'x': case 'y': case 'z':
|
|
case 'A': case 'B': case 'C': case 'D': case 'E': case 'F': case 'G': case 'H': case 'I': case 'J':
|
|
case 'K': case 'L': case 'M': case 'N': case 'O': case 'P': case 'Q': case 'R': case 'S': case 'T':
|
|
case 'U': case 'V': case 'W': case 'X': case 'Y': case 'Z':
|
|
case '-': case '.': case '_':
|
|
return 1;
|
|
default:
|
|
break;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static VALUE
|
|
optimized_escape(VALUE str)
|
|
{
|
|
long i, len, beg = 0;
|
|
VALUE dest = 0;
|
|
const char *cstr;
|
|
char buf[4] = {'%'};
|
|
|
|
len = RSTRING_LEN(str);
|
|
cstr = RSTRING_PTR(str);
|
|
|
|
for (i = 0; i < len; i++) {
|
|
if (!url_unreserved_char(cstr[i])) {
|
|
if (!dest) {
|
|
dest = rb_str_buf_new(len);
|
|
}
|
|
|
|
rb_str_cat(dest, cstr + beg, i - beg);
|
|
beg = i + 1;
|
|
|
|
if (cstr[i] == ' ') {
|
|
rb_str_cat_cstr(dest, "+");
|
|
}
|
|
else {
|
|
unsigned char c = (unsigned char)cstr[i];
|
|
buf[1] = upper_hexdigits[c >> 4];
|
|
buf[2] = upper_hexdigits[c & 0xf];
|
|
rb_str_cat(dest, buf, 3);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (dest) {
|
|
rb_str_cat(dest, cstr + beg, len - beg);
|
|
preserve_original_state(str, dest);
|
|
return dest;
|
|
}
|
|
else {
|
|
return rb_str_dup(str);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* CGI.escapeHTML(string) -> string
|
|
*
|
|
* Returns HTML-escaped string.
|
|
*
|
|
*/
|
|
static VALUE
|
|
cgiesc_escape_html(VALUE self, VALUE str)
|
|
{
|
|
StringValue(str);
|
|
|
|
if (rb_enc_str_asciicompat_p(str)) {
|
|
return optimized_escape_html(str);
|
|
}
|
|
else {
|
|
return rb_call_super(1, &str);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* CGI.escape(string) -> string
|
|
*
|
|
* Returns URL-escaped string.
|
|
*
|
|
*/
|
|
static VALUE
|
|
cgiesc_escape(VALUE self, VALUE str)
|
|
{
|
|
StringValue(str);
|
|
|
|
if (rb_enc_str_asciicompat_p(str)) {
|
|
return optimized_escape(str);
|
|
}
|
|
else {
|
|
return rb_call_super(1, &str);
|
|
}
|
|
}
|
|
|
|
void
|
|
Init_escape(void)
|
|
{
|
|
rb_cCGI = rb_define_class("CGI", rb_cObject);
|
|
rb_mEscape = rb_define_module_under(rb_cCGI, "Escape");
|
|
rb_mUtil = rb_define_module_under(rb_cCGI, "Util");
|
|
rb_define_method(rb_mEscape, "escapeHTML", cgiesc_escape_html, 1);
|
|
rb_define_method(rb_mEscape, "escape", cgiesc_escape, 1);
|
|
rb_prepend_module(rb_mUtil, rb_mEscape);
|
|
rb_extend_object(rb_cCGI, rb_mEscape);
|
|
}
|