2018-01-15 18:42:40 +00:00
|
|
|
import
|
2018-01-31 12:57:05 +00:00
|
|
|
strformat, strutils, sequtils, sets, macros,
|
2018-01-15 18:42:40 +00:00
|
|
|
../logging, ../constants, ../opcode_values
|
|
|
|
|
|
|
|
type
|
|
|
|
CodeStream* = ref object
|
|
|
|
bytes: seq[byte]
|
|
|
|
depthProcessed: int
|
|
|
|
invalidPositions: HashSet[int]
|
|
|
|
pc*: int
|
|
|
|
logger: Logger
|
|
|
|
|
|
|
|
proc `$`*(b: byte): string =
|
|
|
|
$(b.int)
|
|
|
|
|
2018-01-31 12:57:05 +00:00
|
|
|
proc newCodeStream*(codeBytes: seq[byte]): CodeStream =
|
2018-01-15 18:42:40 +00:00
|
|
|
new(result)
|
2018-01-31 12:57:05 +00:00
|
|
|
result.bytes = codeBytes
|
2018-01-15 18:42:40 +00:00
|
|
|
result.pc = 0
|
|
|
|
result.invalidPositions = initSet[int]()
|
|
|
|
result.depthProcessed = 0
|
2018-01-31 12:57:05 +00:00
|
|
|
result.logger = logging.getLogger("vm.code_stream")
|
|
|
|
|
|
|
|
proc newCodeStream*(codeBytes: string): CodeStream =
|
|
|
|
newCodeStream(codeBytes.mapIt(it.byte))
|
2018-01-15 18:42:40 +00:00
|
|
|
|
|
|
|
proc read*(c: var CodeStream, size: int): seq[byte] =
|
2018-01-31 12:57:05 +00:00
|
|
|
if c.pc + size - 1 < c.bytes.len:
|
|
|
|
result = c.bytes[c.pc .. c.pc + size - 1]
|
|
|
|
c.pc += size
|
|
|
|
else:
|
|
|
|
result = @[]
|
|
|
|
c.pc = c.bytes.len
|
2018-01-15 18:42:40 +00:00
|
|
|
|
|
|
|
proc len*(c: CodeStream): int =
|
|
|
|
len(c.bytes)
|
|
|
|
|
2018-01-23 18:36:23 +00:00
|
|
|
proc next*(c: var CodeStream): Op =
|
2018-01-15 18:42:40 +00:00
|
|
|
var nextOpcode = c.read(1)
|
2018-01-31 12:57:05 +00:00
|
|
|
if nextOpcode.len != 0:
|
2018-01-23 18:36:23 +00:00
|
|
|
return Op(nextOpcode[0])
|
2018-01-15 18:42:40 +00:00
|
|
|
else:
|
2018-01-23 18:36:23 +00:00
|
|
|
return Op.STOP
|
2018-01-15 18:42:40 +00:00
|
|
|
|
2018-01-23 18:36:23 +00:00
|
|
|
iterator items*(c: var CodeStream): Op =
|
2018-01-15 18:42:40 +00:00
|
|
|
var nextOpcode = c.next()
|
2018-01-23 18:36:23 +00:00
|
|
|
while nextOpcode != Op.STOP:
|
2018-01-15 18:42:40 +00:00
|
|
|
yield nextOpcode
|
|
|
|
nextOpcode = c.next()
|
|
|
|
|
2018-01-31 12:57:05 +00:00
|
|
|
proc `[]`*(c: CodeStream, offset: int): Op =
|
|
|
|
Op(c.bytes[offset])
|
2018-01-15 18:42:40 +00:00
|
|
|
|
2018-01-23 18:36:23 +00:00
|
|
|
proc peek*(c: var CodeStream): Op =
|
2018-01-15 18:42:40 +00:00
|
|
|
var currentPc = c.pc
|
|
|
|
result = c.next()
|
|
|
|
c.pc = currentPc
|
|
|
|
|
|
|
|
proc updatePc*(c: var CodeStream, value: int) =
|
|
|
|
c.pc = min(value, len(c))
|
|
|
|
|
2018-01-31 12:57:05 +00:00
|
|
|
macro seek*(c: var CodeStream, pc: int, handler: untyped): untyped =
|
|
|
|
let c2 = ident("c")
|
|
|
|
result = quote:
|
|
|
|
var anchorPc = `c`.pc
|
|
|
|
`c`.pc = `pc`
|
|
|
|
try:
|
|
|
|
var `c2` = `c`
|
|
|
|
`handler`
|
|
|
|
finally:
|
|
|
|
`c`.pc = anchorPc
|
2018-01-15 18:42:40 +00:00
|
|
|
|
|
|
|
proc isValidOpcode*(c: var CodeStream, position: int): bool =
|
|
|
|
if position >= len(c):
|
|
|
|
return false
|
|
|
|
if position in c.invalidPositions:
|
|
|
|
return false
|
|
|
|
if position <= c.depthProcessed:
|
|
|
|
return true
|
|
|
|
else:
|
|
|
|
var i = c.depthProcessed
|
|
|
|
while i <= position:
|
2018-01-23 18:36:23 +00:00
|
|
|
var opcode = Op(c[i])
|
|
|
|
if opcode >= Op.PUSH1 and opcode <= Op.PUSH32:
|
2018-01-15 18:42:40 +00:00
|
|
|
var leftBound = (i + 1)
|
|
|
|
var rightBound = leftBound + (opcode.int - 95)
|
|
|
|
for z in leftBound ..< rightBound:
|
|
|
|
c.invalidPositions.incl(z)
|
|
|
|
i = rightBound
|
|
|
|
else:
|
|
|
|
c.depthProcessed = i
|
|
|
|
i += 1
|
|
|
|
if position in c.invalidPositions:
|
|
|
|
return false
|
|
|
|
else:
|
|
|
|
return true
|