FilesExpand file tree

 develop

/

binpatch.lua

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

129 lines (108 loc) · 3.59 KB

 develop

/

binpatch.lua

Copy path

Top

File metadata and controls

• Code
• Blame

129 lines (108 loc) · 3.59 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

-- Simple binary patch with IDA dif file support.

local _ENV = mkmodule('binpatch')

local function load_patch(name)

local filename = name

if not string.match(filename, '[./\\]') then

filename = dfhack.getHackPath()..'/patches/'..dfhack.getDFVersion()..'/'..name..'.dif'

end

local file, err = io.open(filename, 'r')

if not file then

if string.match(err, ': No such file or directory') then

return nil, 'patch not found'

end

end

local old_bytes = {}

local new_bytes = {}

local has_bytes = false

for line in file:lines() do

if string.match(line, '^%x+:') then

local offset, oldv, newv = string.match(line, '^(%x+):%s*(%x+)%s+(%x+)%s*$')

if not offset then

file:close()

return nil, 'could not parse: '..line

end

offset, oldv, newv = tonumber(offset,16), tonumber(oldv,16), tonumber(newv,16)

if oldv > 255 or newv > 255 then

file:close()

return nil, 'invalid byte values: '..line

end

old_bytes[offset] = oldv

new_bytes[offset] = newv

has_bytes = true

end

end

file:close()

if not has_bytes then

return nil, 'no patch bytes found'

end

return { name = name, old_bytes = old_bytes, new_bytes = new_bytes }

end

local function rebase_table(input)

local output = {}

local base = dfhack.internal.getImageBase()

for k,v in pairs(input) do

local offset = dfhack.internal.adjustOffset(k)

if not offset then

return nil, string.format('invalid offset: %x', k)

end

output[base + offset] = v

end

return output

end

local function rebase_patch(patch)

local nold, err = rebase_table(patch.old_bytes)

if not nold then return nil, err end

local nnew, err = rebase_table(patch.new_bytes)

if not nnew then return nil, err end

return { name = patch.name, old_bytes = nold, new_bytes = nnew }

end

BinaryPatch = defclass(BinaryPatch)

BinaryPatch.ATTRS {

name = DEFAULT_NIL,

old_bytes = DEFAULT_NIL,

new_bytes = DEFAULT_NIL,

}

function load_dif_file(name)

local patch, err = load_patch(name)

if not patch then return nil, err end

local rpatch, err = rebase_patch(patch)

if not rpatch then return nil, err end

return BinaryPatch(rpatch)

end

function BinaryPatch:status()

local old_ok, err, addr = dfhack.internal.patchBytes({}, self.old_bytes)

if old_ok then

return 'removed'

elseif dfhack.internal.patchBytes({}, self.new_bytes) then

return 'applied'

else

return 'conflict', addr

end

end

function BinaryPatch:isApplied()

return dfhack.internal.patchBytes({}, self.new_bytes)

end

function BinaryPatch:apply()

local ok, err, addr = dfhack.internal.patchBytes(self.new_bytes, self.old_bytes)

if ok then

return true, 'applied the patch'

elseif dfhack.internal.patchBytes({}, self.new_bytes) then

return true, 'patch is already applied'

else

return false, string.format('conflict at address %x', addr)

end

end

function BinaryPatch:isRemoved()

return dfhack.internal.patchBytes({}, self.old_bytes)

end

function BinaryPatch:remove()

local ok, err, addr = dfhack.internal.patchBytes(self.old_bytes, self.new_bytes)

if ok then

return true, 'removed the patch'

elseif dfhack.internal.patchBytes({}, self.old_bytes) then

return true, 'patch is already removed'

else

return false, string.format('conflict at address %x', addr)

end

end

return _ENV