Skip to content

Commit

Permalink
merge revision(s) 65125:
Browse files Browse the repository at this point in the history
	infect taint flag on Array#pack and String#unpack
	with the directives "B", "b", "H" and "h".

		* pack.c (pack_pack, pack_unpack_internal): infect taint flag.

		* test/ruby/test_pack.rb: add test for above.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/branches/ruby_2_5@65128 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
  • Loading branch information
nagachika committed Oct 17, 2018
1 parent 0b1e97d commit a2958f6
Show file tree
Hide file tree
Showing 3 changed files with 26 additions and 3 deletions.
7 changes: 7 additions & 0 deletions pack.c
Expand Up @@ -749,6 +749,7 @@ pack_pack(int argc, VALUE *argv, VALUE ary)
StringValue(from);
ptr = RSTRING_PTR(from);
plen = RSTRING_LEN(from);
OBJ_INFECT(res, from);

if (len == 0 && type == 'm') {
encodes(res, ptr, plen, type, 0);
Expand Down Expand Up @@ -776,6 +777,7 @@ pack_pack(int argc, VALUE *argv, VALUE ary)

case 'M': /* quoted-printable encoded string */
from = rb_obj_as_string(NEXTFROM);
OBJ_INFECT(res, from);
if (len <= 1)
len = 72;
qpencode(res, from, len);
Expand All @@ -801,6 +803,7 @@ pack_pack(int argc, VALUE *argv, VALUE ary)
}
else {
t = StringValuePtr(from);
OBJ_INFECT(res, from);
rb_obj_taint(from);
}
if (!associates) {
Expand Down Expand Up @@ -1184,6 +1187,7 @@ pack_unpack_internal(VALUE str, VALUE fmt, int mode)
len = (send - s) * 8;
bits = 0;
bitstr = rb_usascii_str_new(0, len);
OBJ_INFECT(bitstr, str);
t = RSTRING_PTR(bitstr);
for (i=0; i<len; i++) {
if (i & 7) bits >>= 1;
Expand All @@ -1205,6 +1209,7 @@ pack_unpack_internal(VALUE str, VALUE fmt, int mode)
len = (send - s) * 8;
bits = 0;
bitstr = rb_usascii_str_new(0, len);
OBJ_INFECT(bitstr, str);
t = RSTRING_PTR(bitstr);
for (i=0; i<len; i++) {
if (i & 7) bits <<= 1;
Expand All @@ -1226,6 +1231,7 @@ pack_unpack_internal(VALUE str, VALUE fmt, int mode)
len = (send - s) * 2;
bits = 0;
bitstr = rb_usascii_str_new(0, len);
OBJ_INFECT(bitstr, str);
t = RSTRING_PTR(bitstr);
for (i=0; i<len; i++) {
if (i & 1)
Expand All @@ -1249,6 +1255,7 @@ pack_unpack_internal(VALUE str, VALUE fmt, int mode)
len = (send - s) * 2;
bits = 0;
bitstr = rb_usascii_str_new(0, len);
OBJ_INFECT(bitstr, str);
t = RSTRING_PTR(bitstr);
for (i=0; i<len; i++) {
if (i & 1)
Expand Down
16 changes: 16 additions & 0 deletions test/ruby/test_pack.rb
Expand Up @@ -860,4 +860,20 @@ def test_unpack1
assert_equal "hogefuga", "aG9nZWZ1Z2E=".unpack1("m")
assert_equal "01000001", "A".unpack1("B*")
end

def test_pack_infection
tainted_array_string = ["123456"]
tainted_array_string.first.taint
['a', 'A', 'Z', 'B', 'b', 'H', 'h', 'u', 'M', 'm', 'P', 'p'].each do |f|
assert_predicate(tainted_array_string.pack(f), :tainted?)
end
end

def test_unpack_infection
tainted_string = "123456"
tainted_string.taint
['a', 'A', 'Z', 'B', 'b', 'H', 'h', 'u', 'M', 'm'].each do |f|
assert_predicate(tainted_string.unpack(f).first, :tainted?)
end
end
end
6 changes: 3 additions & 3 deletions version.h
@@ -1,10 +1,10 @@
#define RUBY_VERSION "2.5.2"
#define RUBY_RELEASE_DATE "2018-10-17"
#define RUBY_PATCHLEVEL 102
#define RUBY_RELEASE_DATE "2018-10-18"
#define RUBY_PATCHLEVEL 103

#define RUBY_RELEASE_YEAR 2018
#define RUBY_RELEASE_MONTH 10
#define RUBY_RELEASE_DAY 17
#define RUBY_RELEASE_DAY 18

#include "ruby/version.h"

Expand Down

0 comments on commit a2958f6

Please sign in to comment.