peter50216/pwntools-ruby

# encoding: ASCII-8BIT
# frozen_string_literal: true

require 'pwnlib/shellcraft/generators/i386/common/common'
require 'pwnlib/shellcraft/generators/i386/common/setregs'

module Pwnlib
  module Shellcraft
    module Generators
      module I386
        module Common
          # Like +memcpy+ in glibc.
          #
          # Copy +n+ bytes from +src+ to +dst+.
          #
          # @param [String, Symbol, Integer] dst
          #   Destination.
          # @param [String, Symbol, Integer] src
          #   Source to be copied.
          # @param [Integer] n
          #   The number of bytes to be copied.
          #
          # @see Amd64::Common#memcpy
          def memcpy(dst, src, n)
            cat "/* memcpy(#{pretty(dst)}, #{pretty(src)}, #{pretty(n)}) */"
            cat 'cld'
            cat Common.setregs({ edi: dst, esi: src, ecx: n })
            cat 'rep movsb'
          end
        end
      end
    end
  end
end