upstream/mercurial-mirror Commit - r38266:99ab72b8

1

from __future__ import absolute_import

1

from __future__ import absolute_import, print_function

2

3

import argparse

3

import argparse

4

import struct

4

import struct

5

import zipfile

5

import zipfile

6

7

from mercurial import (

8

hg,

9

ui as uimod,

10

)

11

7

ap = argparse.ArgumentParser()

12

ap = argparse.ArgumentParser()

8

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

13

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

9

args = ap.parse_args()

14

args = ap.parse_args()

10

15

11

class deltafrag(object):

16

class deltafrag(object):

12

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

17

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

13

self.start = start

18

self.start = start

14

self.end = end

19

self.end = end

15

self.data = data

20

self.data = data

16

21

17

def __str__(self):

22

def __str__(self):

18

return struct.pack(

23

return struct.pack(

19

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

24

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

20

25

21

class delta(object):

26

class delta(object):

22

def __init__(self, frags):

27

def __init__(self, frags):

23

self.frags = frags

28

self.frags = frags

24

29

25

def __str__(self):

30

def __str__(self):

26

return ''.join(str(f) for f in self.frags)

31

return ''.join(str(f) for f in self.frags)

27

32

28

class corpus(object):

33

class corpus(object):

29

34

30

def __init__(self, base, deltas):

35

def __init__(self, base, deltas):

31

self.base = base

36

self.base = base

32

self.deltas = deltas

37

self.deltas = deltas

33

38

34

def __str__(self):

39

def __str__(self):

35

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

40

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

36

parts = (

41

parts = (

37

[

42

[

38

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

43

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

39

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

44

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

40

]

45

]

41

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

46

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

42

+ [self.base]

47

+ [self.base]

43

+ deltas

48

+ deltas

44

)

49

)

45

return "".join(parts)

50

return "".join(parts)

46

51

47

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

52

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

48

# Manually constructed entries

53

# Manually constructed entries

49

zf.writestr(

54

zf.writestr(

50

"one_delta_applies",

55

"one_delta_applies",

51

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

56

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

52

)

57

)

53

zf.writestr(

58

zf.writestr(

54

"one_delta_starts_late",

59

"one_delta_starts_late",

55

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

60

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

56

)

61

)

57

zf.writestr(

62

zf.writestr(

58

"one_delta_ends_late",

63

"one_delta_ends_late",

59

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

64

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

60

)

65

)

61

66

67

try:

68

# Generated from repo data

69

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

70

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

71

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

72

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

73

zf.writestr('manifest_py_rev_10',

74

str(corpus(bins[0], bins[1:])))

75

except: # skip this, so no re-raises

76

print('skipping seed file from repo data')

62

# Automatically discovered by running the fuzzer

77

# Automatically discovered by running the fuzzer

63

zf.writestr(

78

zf.writestr(

64

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

79

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

65

)

80

)

66

zf.writestr(

81

zf.writestr(

67

"mpatch_apply_over_memcpy",

82

"mpatch_apply_over_memcpy",

68

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

83

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

69