Check for UTF-8 strings on 1.9.2+

mhy70 · Dec 13, 2011 · d5f9811 · d5f9811
1 parent 12e693c
commit d5f9811
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/ext/redcarpet/rc_markdown.c b/ext/redcarpet/rc_markdown.c
@@ -89,13 +89,26 @@ static VALUE rb_redcarpet_md__new(int argc, VALUE *argv, VALUE klass)
 	return rb_markdown;
 }
 
+static void check_utf8_encoding(VALUE str)
+{
+#ifdef HAVE_RUBY_ENCODING_H
+	rb_encoding *enc = rb_enc_get(str);
+	if (enc != rb_utf8_encoding() && enc != rb_usascii_encoding()) {
+		rb_raise(rb_eTypeError,
+			"Input must be UTF-8 or US-ASCII, %s given", rb_enc_name(enc));
+	}
+#endif
+}
+
+
 static VALUE rb_redcarpet_md_render(VALUE self, VALUE text)
 {
 	VALUE rb_rndr;
 	struct buf *output_buf;
 	struct sd_markdown *markdown;
 
 	Check_Type(text, T_STRING);
+	check_utf8_encoding(text);
 
 	rb_rndr = rb_iv_get(self, "@renderer");
 	Data_Get_Struct(self, struct sd_markdown, markdown);