revm_bytecode/eof/verification.rs
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876
use crate::{
eof::{Eof, EofDecodeError, TypesSection},
opcode::{self, OPCODE_INFO},
utils::{read_i16, read_u16},
};
use primitives::Bytes;
use specification::constants::{MAX_INITCODE_SIZE, STACK_LIMIT};
use core::{convert::identity, mem};
use std::{borrow::Cow, fmt, vec, vec::Vec};
/// Decodes `raw` into an [`Eof`] container and validates it.
pub fn validate_raw_eof(raw: Bytes) -> Result<Eof, EofError> {
validate_raw_eof_inner(raw, Some(CodeType::ReturnContract))
}
/// Decodes `raw` into an [`Eof`] container and validates it.
#[inline]
pub fn validate_raw_eof_inner(
raw: Bytes,
first_code_type: Option<CodeType>,
) -> Result<Eof, EofError> {
if raw.len() > MAX_INITCODE_SIZE {
return Err(EofError::Decode(EofDecodeError::InvalidEOFSize));
}
let eof = Eof::decode(raw)?;
validate_eof_inner(&eof, first_code_type)?;
Ok(eof)
}
/// Fully validates an [`Eof`] container.
///
/// Only place where validation happen is in Creating Transaction.
/// Because of that we are assuming CodeType is ReturnContract.
///
/// Note: If needed we can make a flag that would assume ReturnContract CodeType.
pub fn validate_eof(eof: &Eof) -> Result<(), EofError> {
validate_eof_inner(eof, Some(CodeType::ReturnContract))
}
#[inline]
pub fn validate_eof_inner(eof: &Eof, first_code_type: Option<CodeType>) -> Result<(), EofError> {
// data needs to be filled first first container.
if !eof.body.is_data_filled {
return Err(EofError::Validation(EofValidationError::DataNotFilled));
}
if eof.body.container_section.is_empty() {
validate_eof_codes(eof, first_code_type)?;
return Ok(());
}
let mut stack = Vec::with_capacity(4);
stack.push((Cow::Borrowed(eof), first_code_type));
while let Some((eof, code_type)) = stack.pop() {
// Validate the current container.
let tracker_containers = validate_eof_codes(&eof, code_type)?;
// Decode subcontainers and push them to the stack.
for (container, code_type) in eof
.body
.container_section
.iter()
.zip(tracker_containers.into_iter())
{
stack.push((Cow::Owned(Eof::decode(container.clone())?), Some(code_type)));
}
}
Ok(())
}
/// Validates an [`Eof`] structure, without recursing into containers.
///
/// Returns a list of all sub containers that are accessed.
#[inline]
pub fn validate_eof_codes(
eof: &Eof,
this_code_type: Option<CodeType>,
) -> Result<Vec<CodeType>, EofValidationError> {
if eof.body.code_section.len() != eof.body.types_section.len() {
return Err(EofValidationError::InvalidTypesSection);
}
if eof.body.code_section.is_empty() {
// no code sections. This should be already checked in decode.
return Err(EofValidationError::NoCodeSections);
}
// the first code section must have a type signature
// (0, 0x80, max_stack_height) (0 inputs non-returning function)
let first_types = &eof.body.types_section[0];
if first_types.inputs != 0 || !first_types.is_non_returning() {
return Err(EofValidationError::InvalidTypesSection);
}
// tracking access of code and sub containers.
let mut tracker: AccessTracker = AccessTracker::new(
this_code_type,
eof.body.code_section.len(),
eof.body.container_section.len(),
);
while let Some(index) = tracker.processing_stack.pop() {
// assume index is correct.
let code = &eof.body.code_section[index];
validate_eof_code(
code,
eof.header.data_size as usize,
index,
eof.body.container_section.len(),
&eof.body.types_section,
&mut tracker,
)?;
}
// iterate over accessed codes and check if all are accessed.
if !tracker.codes.into_iter().all(identity) {
return Err(EofValidationError::CodeSectionNotAccessed);
}
// iterate over all accessed subcontainers and check if all are accessed.
if !tracker.subcontainers.iter().all(|i| i.is_some()) {
return Err(EofValidationError::SubContainerNotAccessed);
}
if tracker.this_container_code_type == Some(CodeType::ReturnContract)
&& !eof.body.is_data_filled
{
return Err(EofValidationError::DataNotFilled);
}
Ok(tracker
.subcontainers
.into_iter()
.map(|i| i.unwrap())
.collect())
}
/// EOF Error.
#[derive(Debug, Hash, PartialEq, Eq, PartialOrd, Ord, Clone, Copy)]
pub enum EofError {
Decode(EofDecodeError),
Validation(EofValidationError),
}
impl From<EofDecodeError> for EofError {
fn from(err: EofDecodeError) -> Self {
EofError::Decode(err)
}
}
impl From<EofValidationError> for EofError {
fn from(err: EofValidationError) -> Self {
EofError::Validation(err)
}
}
impl fmt::Display for EofError {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
match self {
EofError::Decode(e) => write!(f, "Bytecode decode error: {}", e),
EofError::Validation(e) => write!(f, "Bytecode validation error: {}", e),
}
}
}
impl core::error::Error for EofError {}
#[derive(Debug, Hash, PartialEq, Eq, PartialOrd, Ord, Clone, Copy)]
pub enum EofValidationError {
FalsePositive,
/// Opcode is not known. It is not defined in the opcode table.
UnknownOpcode,
/// Opcode is disabled in EOF. For example JUMP, JUMPI, etc.
OpcodeDisabled,
/// Every instruction inside bytecode should be forward accessed.
/// Forward access can be a jump or sequential opcode.
/// In case after terminal opcode there should be a forward jump.
InstructionNotForwardAccessed,
/// Bytecode is too small and is missing immediate bytes for instruction.
MissingImmediateBytes,
/// Similar to [`EofValidationError::MissingImmediateBytes`] but for special case of RJUMPV immediate bytes.
MissingRJUMPVImmediateBytes,
/// Invalid jump into immediate bytes.
JumpToImmediateBytes,
/// Invalid jump into immediate bytes.
BackwardJumpToImmediateBytes,
/// MaxIndex in RJUMPV can't be zero. Zero max index makes it RJUMPI.
RJUMPVZeroMaxIndex,
/// Jump with zero offset would make a jump to next opcode, it does not make sense.
JumpZeroOffset,
/// EOFCREATE points to container out of bounds.
EOFCREATEInvalidIndex,
/// CALLF section out of bounds.
CodeSectionOutOfBounds,
/// CALLF to non returning function is not allowed.
CALLFNonReturningFunction,
/// CALLF stack overflow.
StackOverflow,
/// JUMPF needs to have enough outputs.
JUMPFEnoughOutputs,
/// JUMPF Stack
JUMPFStackHigherThanOutputs,
/// DATA load out of bounds.
DataLoadOutOfBounds,
/// RETF biggest stack num more then outputs.
RETFBiggestStackNumMoreThenOutputs,
/// Stack requirement is more than smallest stack items.
StackUnderflow,
/// Smallest stack items is more than types output.
TypesStackUnderflow,
/// Jump out of bounds.
JumpUnderflow,
/// Jump to out of bounds.
JumpOverflow,
/// Backward jump should have same smallest and biggest stack items.
BackwardJumpBiggestNumMismatch,
/// Backward jump should have same smallest and biggest stack items.
BackwardJumpSmallestNumMismatch,
/// Last instruction should be terminating.
LastInstructionNotTerminating,
/// Code section not accessed.
CodeSectionNotAccessed,
/// Types section invalid
InvalidTypesSection,
/// First types section is invalid.
/// It should have inputs 0 and outputs 0x80.
InvalidFirstTypesSection,
/// Max stack element mismatch.
MaxStackMismatch,
/// No code sections present
NoCodeSections,
/// Sub container called in two different modes.
/// Check [`CodeType`] for more information.
SubContainerCalledInTwoModes,
/// Sub container not accessed.
SubContainerNotAccessed,
/// Data size needs to be filled for ReturnContract type.
DataNotFilled,
/// Section is marked as non-returning but has either RETF or
/// JUMPF to returning section opcodes.
NonReturningSectionIsReturning,
}
#[derive(Clone, Debug, PartialEq, Eq)]
pub struct AccessTracker {
/// This code type
pub this_container_code_type: Option<CodeType>,
/// Vector of accessed codes.
pub codes: Vec<bool>,
/// Stack of codes section that needs to be processed.
pub processing_stack: Vec<usize>,
/// Code accessed by subcontainer and expected subcontainer first code type.
/// EOF code can be invoked in EOFCREATE mode or used in RETURNCONTRACT opcode.
/// if SubContainer is called from EOFCREATE it needs to be ReturnContract type.
/// If SubContainer is called from RETURNCONTRACT it needs to be ReturnOrStop type.
///
/// None means it is not accessed.
pub subcontainers: Vec<Option<CodeType>>,
}
impl AccessTracker {
/// Returns a new instance of `CodeSubContainerAccess`.
///
/// Mark first code section as accessed and push first it to the stack.
///
/// # Panics
///
/// Panics if `codes_size` is zero.
pub fn new(
this_container_code_type: Option<CodeType>,
codes_size: usize,
subcontainers_size: usize,
) -> Self {
if codes_size == 0 {
panic!("There should be at least one code section");
}
let mut this = Self {
this_container_code_type,
codes: vec![false; codes_size],
processing_stack: Vec::with_capacity(4),
subcontainers: vec![None; subcontainers_size],
};
this.codes[0] = true;
this.processing_stack.push(0);
this
}
/// Mark code as accessed.
///
/// If code was not accessed before, it will be added to the processing stack.
///
/// Assumes that index is valid.
pub fn access_code(&mut self, index: usize) {
let was_accessed = mem::replace(&mut self.codes[index], true);
if !was_accessed {
self.processing_stack.push(index);
}
}
pub fn set_subcontainer_type(
&mut self,
index: usize,
new_code_type: CodeType,
) -> Result<(), EofValidationError> {
let Some(container) = self.subcontainers.get_mut(index) else {
panic!("It should not be possible")
};
let Some(code_type) = container else {
*container = Some(new_code_type);
return Ok(());
};
if *code_type != new_code_type {
return Err(EofValidationError::SubContainerCalledInTwoModes);
}
Ok(())
}
}
/// Types of code sections. It is a error if container to contain
/// both RETURNCONTRACT and either of RETURN or STOP.
#[derive(Clone, Copy, Debug, PartialEq, Eq)]
pub enum CodeType {
/// Return contract code.
ReturnContract,
/// Return or Stop opcodes.
ReturnOrStop,
}
impl CodeType {
/// Returns true of the code is initcode.
pub fn is_initcode(&self) -> bool {
matches!(self, CodeType::ReturnContract)
}
}
impl fmt::Display for EofValidationError {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
let s = match self {
Self::FalsePositive => "False positive",
Self::UnknownOpcode => "Opcode is not known",
Self::OpcodeDisabled => "Opcode is disabled",
Self::InstructionNotForwardAccessed => "Should have forward jump",
Self::MissingImmediateBytes => "Bytecode is missing bytes",
Self::MissingRJUMPVImmediateBytes => "Bytecode is missing bytes after RJUMPV opcode",
Self::JumpToImmediateBytes => "Invalid jump",
Self::BackwardJumpToImmediateBytes => "Invalid backward jump",
Self::RJUMPVZeroMaxIndex => "Used RJUMPV with zero as MaxIndex",
Self::JumpZeroOffset => "Used JUMP with zero as offset",
Self::EOFCREATEInvalidIndex => "EOFCREATE points to out of bound index",
Self::CodeSectionOutOfBounds => "CALLF index is out of bounds",
Self::CALLFNonReturningFunction => "CALLF was used on non-returning function",
Self::StackOverflow => "CALLF stack overflow",
Self::JUMPFEnoughOutputs => "JUMPF needs more outputs",
Self::JUMPFStackHigherThanOutputs => "JUMPF stack is too high for outputs",
Self::DataLoadOutOfBounds => "DATALOAD is out of bounds",
Self::RETFBiggestStackNumMoreThenOutputs => {
"RETF biggest stack num is more than outputs"
}
Self::StackUnderflow => "Stack requirement is above smallest stack items",
Self::TypesStackUnderflow => "Smallest stack items is more than output type",
Self::JumpUnderflow => "Jump destination is too low",
Self::JumpOverflow => "Jump destination is too high",
Self::BackwardJumpBiggestNumMismatch => {
"Backward jump has different biggest stack item"
}
Self::BackwardJumpSmallestNumMismatch => {
"Backward jump has different smallest stack item"
}
Self::LastInstructionNotTerminating => {
"Last instruction of bytecode is not terminating"
}
Self::CodeSectionNotAccessed => "Code section was not accessed",
Self::InvalidTypesSection => "Invalid types section",
Self::InvalidFirstTypesSection => "Invalid first types section",
Self::MaxStackMismatch => "Max stack element mismatchs",
Self::NoCodeSections => "No code sections",
Self::SubContainerCalledInTwoModes => "Sub container called in two modes",
Self::SubContainerNotAccessed => "Sub container not accessed",
Self::DataNotFilled => "Data not filled",
Self::NonReturningSectionIsReturning => "Non returning section is returning",
};
f.write_str(s)
}
}
impl core::error::Error for EofValidationError {}
/// Validates that:
/// * All instructions are valid.
/// * It ends with a terminating instruction or RJUMP.
/// * All instructions are accessed by forward jumps or .
///
/// Validate stack requirements and if all codes sections are used.
pub fn validate_eof_code(
code: &[u8],
data_size: usize,
this_types_index: usize,
num_of_containers: usize,
types: &[TypesSection],
tracker: &mut AccessTracker,
) -> Result<(), EofValidationError> {
let this_types = &types[this_types_index];
#[derive(Debug, Copy, Clone)]
struct InstructionInfo {
/// Is immediate byte, jumps can't happen on this part of code.
is_immediate: bool,
/// Have forward jump to this opcode. Used to check if opcode
/// after termination is accessed.
is_jumpdest: bool,
/// Smallest number of stack items accessed by jumps or sequential opcodes.
smallest: i32,
/// Biggest number of stack items accessed by jumps or sequential opcodes.
biggest: i32,
}
impl InstructionInfo {
#[inline]
fn mark_as_immediate(&mut self) -> Result<(), EofValidationError> {
if self.is_jumpdest {
// Jump to immediate bytes.
return Err(EofValidationError::JumpToImmediateBytes);
}
self.is_immediate = true;
Ok(())
}
}
impl Default for InstructionInfo {
fn default() -> Self {
Self {
is_immediate: false,
is_jumpdest: false,
smallest: i32::MAX,
biggest: i32::MIN,
}
}
}
// all bytes that are intermediate.
let mut jumps = vec![InstructionInfo::default(); code.len()];
let mut is_after_termination = false;
let mut next_smallest = this_types.inputs as i32;
let mut next_biggest = this_types.inputs as i32;
let mut is_returning = false;
let mut i = 0;
// We can check validity and jump destinations in one pass.
while i < code.len() {
let op = code[i];
let opcode = &OPCODE_INFO[op as usize];
let Some(opcode) = opcode else {
// err unknown opcode.
return Err(EofValidationError::UnknownOpcode);
};
if opcode.is_disabled_in_eof() {
// Opcode is disabled in EOF
return Err(EofValidationError::OpcodeDisabled);
}
let this_instruction = &mut jumps[i];
// update biggest/smallest values for next instruction only if it is not after termination.
if !is_after_termination {
this_instruction.smallest = core::cmp::min(this_instruction.smallest, next_smallest);
this_instruction.biggest = core::cmp::max(this_instruction.biggest, next_biggest);
}
let this_instruction = *this_instruction;
// Opcodes after termination should be accessed by forward jumps.
if is_after_termination && !this_instruction.is_jumpdest {
// opcode after termination was not accessed.
return Err(EofValidationError::InstructionNotForwardAccessed);
}
is_after_termination = opcode.is_terminating();
// mark immediate as non-jumpable. RJUMPV is special case covered later.
if opcode.immediate_size() != 0 {
// check if the opcode immediate are within the bounds of the code
if i + opcode.immediate_size() as usize >= code.len() {
// Malfunctional code
return Err(EofValidationError::MissingImmediateBytes);
}
// mark immediate bytes as non-jumpable.
for imm in 1..opcode.immediate_size() as usize + 1 {
// SAFETY: immediate size is checked above.
jumps[i + imm].mark_as_immediate()?;
}
}
// IO diff used to generate next instruction smallest/biggest value.
let mut stack_io_diff = opcode.io_diff() as i32;
// how many stack items are required for this opcode.
let mut stack_requirement = opcode.inputs() as i32;
// additional immediate bytes for RJUMPV, it has dynamic vtable.
let mut rjumpv_additional_immediates = 0;
// If opcodes is RJUMP, RJUMPI or RJUMPV then this will have absolute jumpdest.
let mut absolute_jumpdest = vec![];
match op {
opcode::RJUMP | opcode::RJUMPI => {
let offset = unsafe { read_i16(code.as_ptr().add(i + 1)) } as isize;
absolute_jumpdest = vec![offset + 3 + i as isize];
// RJUMP is considered a terminating opcode.
}
opcode::RJUMPV => {
// code length for RJUMPV is checked with immediate size.
let max_index = code[i + 1] as usize;
let len = max_index + 1;
// and max_index+1 is to get size of vtable as index starts from 0.
rjumpv_additional_immediates = len * 2;
// +1 is for max_index byte
if i + 1 + rjumpv_additional_immediates >= code.len() {
// Malfunctional code RJUMPV vtable is not complete
return Err(EofValidationError::MissingRJUMPVImmediateBytes);
}
// Mark vtable as immediate, max_index was already marked.
for imm in 0..rjumpv_additional_immediates {
// SAFETY: immediate size is checked above.
jumps[i + 2 + imm].mark_as_immediate()?;
}
let mut jumps = Vec::with_capacity(len);
for vtablei in 0..len {
let offset =
unsafe { read_i16(code.as_ptr().add(i + 2 + 2 * vtablei)) } as isize;
jumps.push(offset + i as isize + 2 + rjumpv_additional_immediates as isize);
}
absolute_jumpdest = jumps
}
opcode::CALLF => {
let section_i: usize = unsafe { read_u16(code.as_ptr().add(i + 1)) } as usize;
let Some(target_types) = types.get(section_i) else {
// code section out of bounds.
return Err(EofValidationError::CodeSectionOutOfBounds);
};
// CALLF operand must not point to to a section with 0x80 as outputs (non-returning)
if target_types.is_non_returning() {
return Err(EofValidationError::CALLFNonReturningFunction);
}
// stack input for this opcode is the input of the called code.
stack_requirement = target_types.inputs as i32;
// stack diff depends on input/output of the called code.
stack_io_diff = target_types.io_diff();
// mark called code as accessed.
tracker.access_code(section_i);
// we decrement by `types.inputs` as they are considered as send
// to the called code and included in types.max_stack_size.
if this_instruction.biggest - stack_requirement + target_types.max_stack_size as i32
> STACK_LIMIT as i32
{
// if stack max items + called code max stack size
return Err(EofValidationError::StackOverflow);
}
}
opcode::JUMPF => {
let target_index = unsafe { read_u16(code.as_ptr().add(i + 1)) } as usize;
// targeted code needs to have zero outputs (be non returning).
let Some(target_types) = types.get(target_index) else {
// code section out of bounds.
return Err(EofValidationError::CodeSectionOutOfBounds);
};
// we decrement types.inputs as they are considered send to the called code.
// and included in types.max_stack_size.
if this_instruction.biggest - target_types.inputs as i32
+ target_types.max_stack_size as i32
> STACK_LIMIT as i32
{
// stack overflow
return Err(EofValidationError::StackOverflow);
}
tracker.access_code(target_index);
if target_types.is_non_returning() {
// if it is not returning
stack_requirement = target_types.inputs as i32;
} else {
is_returning = true;
// check if target code produces enough outputs.
if this_types.outputs < target_types.outputs {
return Err(EofValidationError::JUMPFEnoughOutputs);
}
stack_requirement = this_types.outputs as i32 + target_types.inputs as i32
- target_types.outputs as i32;
// Stack requirement needs to more than this instruction biggest stack number.
if this_instruction.biggest > stack_requirement {
return Err(EofValidationError::JUMPFStackHigherThanOutputs);
}
// if this instruction max + target_types max is more then stack limit.
if this_instruction.biggest + stack_requirement > STACK_LIMIT as i32 {
return Err(EofValidationError::StackOverflow);
}
}
}
opcode::EOFCREATE => {
let index = code[i + 1] as usize;
if index >= num_of_containers {
// code section out of bounds.
return Err(EofValidationError::EOFCREATEInvalidIndex);
}
tracker.set_subcontainer_type(index, CodeType::ReturnContract)?;
}
opcode::RETURNCONTRACT => {
let index = code[i + 1] as usize;
if index >= num_of_containers {
// code section out of bounds.
// TODO custom error
return Err(EofValidationError::EOFCREATEInvalidIndex);
}
if *tracker
.this_container_code_type
.get_or_insert(CodeType::ReturnContract)
!= CodeType::ReturnContract
{
// TODO make custom error
return Err(EofValidationError::SubContainerCalledInTwoModes);
}
tracker.set_subcontainer_type(index, CodeType::ReturnOrStop)?;
}
opcode::RETURN | opcode::STOP => {
if *tracker
.this_container_code_type
.get_or_insert(CodeType::ReturnOrStop)
!= CodeType::ReturnOrStop
{
return Err(EofValidationError::SubContainerCalledInTwoModes);
}
}
opcode::DATALOADN => {
let index = unsafe { read_u16(code.as_ptr().add(i + 1)) } as isize;
if data_size < 32 || index > data_size as isize - 32 {
// data load out of bounds.
return Err(EofValidationError::DataLoadOutOfBounds);
}
}
opcode::RETF => {
stack_requirement = this_types.outputs as i32;
// mark section as returning.
is_returning = true;
if this_instruction.biggest > stack_requirement {
return Err(EofValidationError::RETFBiggestStackNumMoreThenOutputs);
}
}
opcode::DUPN => {
stack_requirement = code[i + 1] as i32 + 1;
}
opcode::SWAPN => {
stack_requirement = code[i + 1] as i32 + 2;
}
opcode::EXCHANGE => {
let imm = code[i + 1];
let n = (imm >> 4) + 1;
let m = (imm & 0x0F) + 1;
stack_requirement = n as i32 + m as i32 + 1;
}
_ => {}
}
// check if stack requirement is more than smallest stack items.
if stack_requirement > this_instruction.smallest {
// opcode requirement is more than smallest stack items.
return Err(EofValidationError::StackUnderflow);
}
next_smallest = this_instruction.smallest + stack_io_diff;
next_biggest = this_instruction.biggest + stack_io_diff;
// check if jumpdest are correct and mark forward jumps.
for absolute_jump in absolute_jumpdest {
if absolute_jump < 0 {
// jump out of bounds.
return Err(EofValidationError::JumpUnderflow);
}
if absolute_jump >= code.len() as isize {
// jump to out of bounds
return Err(EofValidationError::JumpOverflow);
}
// fine to cast as bounds are checked.
let absolute_jump = absolute_jump as usize;
let target_jump = &mut jumps[absolute_jump];
if target_jump.is_immediate {
// Jump target is immediate byte.
return Err(EofValidationError::BackwardJumpToImmediateBytes);
}
// needed to mark forward jumps. It does not do anything for backward jumps.
target_jump.is_jumpdest = true;
if absolute_jump <= i {
// backward jumps should have same smallest and biggest stack items.
if target_jump.biggest != next_biggest {
// wrong jumpdest.
return Err(EofValidationError::BackwardJumpBiggestNumMismatch);
}
if target_jump.smallest != next_smallest {
// wrong jumpdest.
return Err(EofValidationError::BackwardJumpSmallestNumMismatch);
}
} else {
// forward jumps can make min even smallest size
// while biggest num is needed to check stack overflow
target_jump.smallest = core::cmp::min(target_jump.smallest, next_smallest);
target_jump.biggest = core::cmp::max(target_jump.biggest, next_biggest);
}
}
// additional immediate are from RJUMPV vtable.
i += 1 + opcode.immediate_size() as usize + rjumpv_additional_immediates;
}
// error if section is returning but marked as non-returning.
if is_returning == this_types.is_non_returning() {
// wrong termination.
return Err(EofValidationError::NonReturningSectionIsReturning);
}
// last opcode should be terminating
if !is_after_termination {
// wrong termination.
return Err(EofValidationError::LastInstructionNotTerminating);
}
// TODO integrate max so we dont need to iterate again
let mut max_stack_requirement = 0;
for opcode in jumps {
max_stack_requirement = core::cmp::max(opcode.biggest, max_stack_requirement);
}
if max_stack_requirement != types[this_types_index].max_stack_size as i32 {
// stack overflow
return Err(EofValidationError::MaxStackMismatch);
}
Ok(())
}
#[cfg(test)]
mod test {
use super::*;
use primitives::hex;
#[test]
fn test1() {
// result:Result { result: false, exception: Some("EOF_ConflictingStackHeight") }
let err =
validate_raw_eof(hex!("ef0001010004020001000704000000008000016000e200fffc00").into());
assert!(err.is_err(), "{err:#?}");
}
#[test]
fn test2() {
// result:Result { result: false, exception: Some("EOF_InvalidNumberOfOutputs") }
let err =
validate_raw_eof_inner(hex!("ef000101000c02000300040004000204000000008000020002000100010001e30001005fe500025fe4").into(),None);
assert!(err.is_ok(), "{err:#?}");
}
#[test]
fn test3() {
// result:Result { result: false, exception: Some("EOF_InvalidNumberOfOutputs") }
let err =
validate_raw_eof_inner(hex!("ef000101000c02000300040008000304000000008000020002000503010003e30001005f5f5f5f5fe500025050e4").into(),None);
assert_eq!(
err,
Err(EofError::Validation(
EofValidationError::JUMPFStackHigherThanOutputs
))
);
}
#[test]
fn test4() {
//0xef0001010004020001000e04000000008000045f6000e100025f5f6000e1fffd00
// result:Result { result: false, exception: Some("EOF_InvalidNumberOfOutputs") }
let err = validate_raw_eof(
hex!("ef0001010004020001000e04000000008000045f6000e100025f5f6000e1fffd00").into(),
);
assert_eq!(
err,
Err(EofError::Validation(
EofValidationError::BackwardJumpBiggestNumMismatch
))
);
}
#[test]
fn test5() {
let err = validate_raw_eof(hex!("ef000101000402000100030400000000800000e5ffff").into());
assert_eq!(
err,
Err(EofError::Validation(
EofValidationError::CodeSectionOutOfBounds
))
);
}
#[test]
fn size_limit() {
let eof = validate_raw_eof_inner(
hex!("ef00010100040200010003040001000080000130500000").into(),
Some(CodeType::ReturnOrStop),
);
assert!(eof.is_ok());
}
#[test]
fn test() {
let eof = validate_raw_eof_inner(
hex!("ef0001010004020001000504ff0300008000023a60cbee1800").into(),
None,
);
assert_eq!(
eof,
Err(EofError::Validation(EofValidationError::DataNotFilled))
);
}
#[test]
fn unreachable_code_section() {
let eof = validate_raw_eof_inner(
hex!("ef000101000c02000300030001000304000000008000000080000000800000e50001fee50002")
.into(),
None,
);
assert_eq!(
eof,
Err(EofError::Validation(
EofValidationError::CodeSectionNotAccessed
))
);
}
#[test]
fn non_returning_sections() {
let eof = validate_raw_eof_inner(
hex!("ef000101000c02000300040001000304000000008000000080000000000000e300020000e50001")
.into(),
Some(CodeType::ReturnOrStop),
);
assert_eq!(
eof,
Err(EofError::Validation(
EofValidationError::NonReturningSectionIsReturning
))
);
}
#[test]
fn incompatible_container_kind() {
let eof = validate_raw_eof_inner(
hex!("ef000101000402000100060300010014040000000080000260006000ee00ef00010100040200010001040000000080000000")
.into(),
Some(CodeType::ReturnOrStop),
);
assert_eq!(
eof,
Err(EofError::Validation(
EofValidationError::SubContainerCalledInTwoModes
))
);
}
}