liangsheng8708
/
ruffle
mirror of https://mirror.ghproxy.com/https://github.com/ruffle-rs/ruffle.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

core: Limit tag parsing to slice of tag data

This commit is contained in:
Mike Welsh 2020-07-03 17:06:45 -07:00
parent c50c835790
commit 08569ae979
2 changed files with 290 additions and 231 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

99
core/src/display_object/movie_clip.rs
View File

@ -146,7 +146,13 @@ impl<'gc> MovieClip<'gc> {
let mut reader = data.read_from(self.0.read().tag_stream_pos);
let mut cur_frame = 1;
let mut ids = fnv::FnvHashMap::default();
let tag_callback = |reader: &mut SwfStream<&[u8]>, tag_code, tag_len| match tag_code {
let version = reader.version();
let tag_callback = |reader: &mut SwfStream<&[u8]>, tag_code, tag_len| {
let data = *reader.get_inner().get_ref();
let tag_pos = reader.get_inner().position() as usize;
let tag_slice = &data[tag_pos..tag_pos + tag_len];
let reader = &mut SwfStream::new(std::io::Cursor::new(tag_slice), version);
match tag_code {
TagCode::FileAttributes => {
let attributes = reader.read_file_attributes()?;
if attributes.is_action_script_3 {
@ -314,27 +320,30 @@ impl<'gc> MovieClip<'gc> {
.0
.write(context.gc_context)
.preload_remove_object(context, reader, &mut ids, 2),
TagCode::ShowFrame => {
self.0
.write(context.gc_context)
.preload_show_frame(context, reader, &mut cur_frame)
}
TagCode::SoundStreamHead => self.0.write(context.gc_context).preload_sound_stream_head(
TagCode::ShowFrame => self.0.write(context.gc_context).preload_show_frame(
context,
reader,
cur_frame,
&mut static_data,
1,
&mut cur_frame,
),
TagCode::SoundStreamHead => self
.0
.write(context.gc_context)
.preload_sound_stream_head(context, reader, cur_frame, &mut static_data, 1),
TagCode::SoundStreamHead2 => self
.0
.write(context.gc_context)
.preload_sound_stream_head(context, reader, cur_frame, &mut static_data, 2),
TagCode::SoundStreamBlock => self
.0
.write(context.gc_context)
.preload_sound_stream_block(context, reader, cur_frame, &mut static_data, tag_len),
TagCode::SoundStreamBlock => {
self.0.write(context.gc_context).preload_sound_stream_block(
context,
reader,
cur_frame,
&mut static_data,
tag_len,
)
}
_ => Ok(()),
}
};
let _ = tag_utils::decode_tags(&mut reader, tag_callback, TagCode::End);
self.0.write(context.gc_context).static_data =
@ -643,8 +652,14 @@ impl<'gc> MovieClip<'gc> {
let mut has_stream_block = false;
drop(mc);
let version = reader.version();
use swf::TagCode;
let tag_callback = |reader: &mut _, tag_code, tag_len| match tag_code {
let tag_callback = |reader: &mut SwfStream<&[u8]>, tag_code, tag_len| {
let data = *reader.get_inner().get_ref();
let tag_pos = reader.get_inner().position() as usize;
let tag_slice = &data[tag_pos..tag_pos + tag_len];
let reader = &mut SwfStream::new(std::io::Cursor::new(tag_slice), version);
match tag_code {
TagCode::DoAction => self.do_action(self_display_object, context, reader, tag_len),
TagCode::PlaceObject if run_display_actions => {
self.place_object(self_display_object, avm, context, reader, tag_len, 1)
@ -658,8 +673,12 @@ impl<'gc> MovieClip<'gc> {
TagCode::PlaceObject4 if run_display_actions => {
self.place_object(self_display_object, avm, context, reader, tag_len, 4)
}
TagCode::RemoveObject if run_display_actions => self.remove_object(context, reader, 1),
TagCode::RemoveObject2 if run_display_actions => self.remove_object(context, reader, 2),
TagCode::RemoveObject if run_display_actions => {
self.remove_object(context, reader, 1)
}
TagCode::RemoveObject2 if run_display_actions => {
self.remove_object(context, reader, 2)
}
TagCode::SetBackgroundColor => self.set_background_color(context, reader),
TagCode::StartSound => self.start_sound_1(context, reader),
TagCode::SoundStreamBlock => {
@ -667,6 +686,7 @@ impl<'gc> MovieClip<'gc> {
self.sound_stream_block(context, reader)
}
_ => Ok(()),
}
};
let _ = tag_utils::decode_tags(&mut reader, tag_callback, TagCode::ShowFrame);
@ -807,23 +827,57 @@ impl<'gc> MovieClip<'gc> {
frame_pos = reader.get_inner().position();
let mut mc = self.0.write(context.gc_context);
let version = reader.version();
use swf::TagCode;
let tag_callback = |reader: &mut _, tag_code, tag_len| match tag_code {
let tag_callback = |reader: &mut SwfStream<&[u8]>, tag_code, tag_len| {
let data = *reader.get_inner().get_ref();
let tag_pos = reader.get_inner().position() as usize;
let tag_slice = &data[tag_pos..tag_pos + tag_len];
let reader = &mut SwfStream::new(std::io::Cursor::new(tag_slice), version);
match tag_code {
TagCode::PlaceObject => {
index += 1;
mc.goto_place_object(reader, tag_len, 1, &mut goto_commands, is_rewind, index)
mc.goto_place_object(
reader,
tag_len,
1,
&mut goto_commands,
is_rewind,
index,
)
}
TagCode::PlaceObject2 => {
index += 1;
mc.goto_place_object(reader, tag_len, 2, &mut goto_commands, is_rewind, index)
mc.goto_place_object(
reader,
tag_len,
2,
&mut goto_commands,
is_rewind,
index,
)
}
TagCode::PlaceObject3 => {
index += 1;
mc.goto_place_object(reader, tag_len, 3, &mut goto_commands, is_rewind, index)
mc.goto_place_object(
reader,
tag_len,
3,
&mut goto_commands,
is_rewind,
index,
)
}
TagCode::PlaceObject4 => {
index += 1;
mc.goto_place_object(reader, tag_len, 4, &mut goto_commands, is_rewind, index)
mc.goto_place_object(
reader,
tag_len,
4,
&mut goto_commands,
is_rewind,
index,
)
}
TagCode::RemoveObject => {
mc.goto_remove_object(reader, 1, context, &mut goto_commands, is_rewind)
@ -832,6 +886,7 @@ impl<'gc> MovieClip<'gc> {
mc.goto_remove_object(reader, 2, context, &mut goto_commands, is_rewind)
}
_ => Ok(()),
}
};
let _ = tag_utils::decode_tags(&mut reader, tag_callback, TagCode::ShowFrame);
}

4
swf/src/read.rs
View File

@ -342,6 +342,10 @@ impl<R: Read> Reader<R> {
}
}
pub fn version(&self) -> u8 {
self.version
}
/// Returns a reference to the underlying `Reader`.
pub fn get_ref(&self) -> &R {
&self.input