upstream/mercurial-mirror Commit - r44308:229215fc

1

from __future__ import absolute_import, print_function

1

from __future__ import absolute_import, print_function

2

3

import argparse

3

import argparse

4

import struct

4

import struct

5

import sys

5

import zipfile

6

import zipfile

6

7

from mercurial import (

8

from mercurial import (

8

hg,

9

hg,

9

ui as uimod,

10

ui as uimod,

10

)

11

)

11

12

ap = argparse.ArgumentParser()

13

ap = argparse.ArgumentParser()

13

ap.add_argument("out", metavar="some.zip", type=str, nargs=1)

14

ap.add_argument("out", metavar="some.zip", type=str, nargs=1)

14

args = ap.parse_args()

15

args = ap.parse_args()

15

16

17

class deltafrag(object):

18

if sys.version_info[0] < 3:

19

20

class py2reprhack(object):

21

def __repr__(self):

22

"""Py2 calls __repr__ for `bytes(foo)`, forward to __bytes__"""

23

return self.__bytes__()

24

25

26

else:

27

28

class py2reprhack(object):

29

"""Not needed on py3."""

30

31

32

class deltafrag(py2reprhack):

18

def __init__(self, start, end, data):

33

def __init__(self, start, end, data):

19

self.start = start

34

self.start = start

20

self.end = end

35

self.end = end

21

self.data = data

36

self.data = data

22

37

23

def __repr__(self):

24

# py2 calls __repr__ when you do `bytes(foo)`

25

return self.__bytes__()

26

27

def __bytes__(self):

38

def __bytes__(self):

28

return (

39

return (

29

struct.pack(">lll", self.start, self.end, len(self.data))

40

struct.pack(">lll", self.start, self.end, len(self.data))

30

+ self.data

41

+ self.data

31

)

42

)

32

43

33

44

34

class delta(~~object~~):

45

class delta(py2reprhack):

35

def __init__(self, frags):

46

def __init__(self, frags):

36

self.frags = frags

47

self.frags = frags

37

48

38

def __repr__(self):

39

# py2 calls __repr__ when you do `bytes(foo)`

40

return self.__bytes__()

41

42

def __bytes__(self):

49

def __bytes__(self):

43

return b''.join(bytes(f) for f in self.frags)

50

return b''.join(bytes(f) for f in self.frags)

44

51

45

52

46

class corpus(~~object~~):

53

class corpus(py2reprhack):

47

def __init__(self, base, deltas):

54

def __init__(self, base, deltas):

48

self.base = base

55

self.base = base

49

self.deltas = deltas

56

self.deltas = deltas

50

57

51

def __repr__(self):

52

# py2 calls __repr__ when you do `bytes(foo)`

53

return self.__bytes__()

54

55

def __bytes__(self):

58

def __bytes__(self):

56

deltas = [bytes(d) for d in self.deltas]

59

deltas = [bytes(d) for d in self.deltas]

57

parts = (

60

parts = (

58

[

61

[

59

struct.pack(">B", len(deltas) + 1),

62

struct.pack(">B", len(deltas) + 1),

60

struct.pack(">H", len(self.base)),

63

struct.pack(">H", len(self.base)),

61

]

64

]

62

+ [struct.pack(">H", len(d)) for d in deltas]

65

+ [struct.pack(">H", len(d)) for d in deltas]

63

+ [self.base]

66

+ [self.base]

64

+ deltas

67

+ deltas

65

)

68

)

66

return b''.join(parts)

69

return b''.join(parts)

67

70

68

71

69

with zipfile.ZipFile(args.out[0], "w", zipfile.ZIP_STORED) as zf:

72

with zipfile.ZipFile(args.out[0], "w", zipfile.ZIP_STORED) as zf:

70

# Manually constructed entries

73

# Manually constructed entries

71

zf.writestr(

74

zf.writestr(

72

"one_delta_applies",

75

"one_delta_applies",

73

bytes(corpus(b'a', [delta([deltafrag(0, 1, b'b')])])),

76

bytes(corpus(b'a', [delta([deltafrag(0, 1, b'b')])])),

74

)

77

)

75

zf.writestr(

78

zf.writestr(

76

"one_delta_starts_late",

79

"one_delta_starts_late",

77

bytes(corpus(b'a', [delta([deltafrag(3, 1, b'b')])])),

80

bytes(corpus(b'a', [delta([deltafrag(3, 1, b'b')])])),

78

)

81

)

79

zf.writestr(

82

zf.writestr(

80

"one_delta_ends_late",

83

"one_delta_ends_late",

81

bytes(corpus(b'a', [delta([deltafrag(0, 20, b'b')])])),

84

bytes(corpus(b'a', [delta([deltafrag(0, 20, b'b')])])),

82

)

85

)

83

86

84

try:

87

try:

85

# Generated from repo data

88

# Generated from repo data

86

r = hg.repository(uimod.ui(), b'../..')

89

r = hg.repository(uimod.ui(), b'../..')

87

fl = r.file(b'mercurial/manifest.py')

90

fl = r.file(b'mercurial/manifest.py')

88

rl = getattr(fl, '_revlog', fl)

91

rl = getattr(fl, '_revlog', fl)

89

bins = rl._chunks(rl._deltachain(10)[0])

92

bins = rl._chunks(rl._deltachain(10)[0])

90

zf.writestr('manifest_py_rev_10', bytes(corpus(bins[0], bins[1:])))

93

zf.writestr('manifest_py_rev_10', bytes(corpus(bins[0], bins[1:])))

91

except: # skip this, so no re-raises

94

except: # skip this, so no re-raises

92

print('skipping seed file from repo data')

95

print('skipping seed file from repo data')

93

# Automatically discovered by running the fuzzer

96

# Automatically discovered by running the fuzzer

94

zf.writestr(

97

zf.writestr(

95

"mpatch_decode_old_overread", b"\x02\x00\x00\x00\x02\x00\x00\x00"

98

"mpatch_decode_old_overread", b"\x02\x00\x00\x00\x02\x00\x00\x00"

96

)

99

)

97

# https://bugs.chromium.org/p/oss-fuzz/issues/detail?id=8876

100

# https://bugs.chromium.org/p/oss-fuzz/issues/detail?id=8876

98

zf.writestr(

101

zf.writestr(

99

"mpatch_ossfuzz_getbe32_ubsan",

102

"mpatch_ossfuzz_getbe32_ubsan",

100

b"\x02\x00\x00\x00\x0c \xff\xff\xff\xff ",

103

b"\x02\x00\x00\x00\x0c \xff\xff\xff\xff ",

101

)

104

)

102

zf.writestr(

105

zf.writestr(

103

"mpatch_apply_over_memcpy",

106

"mpatch_apply_over_memcpy",

104

b'\x13\x01\x00\x05\xd0\x00\x00\x00\x00\x00\x00\x00\x00\n \x00\x00\x00'

107

b'\x13\x01\x00\x05\xd0\x00\x00\x00\x00\x00\x00\x00\x00\n \x00\x00\x00'

105