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
|
// Copyright 2017 syzkaller project authors. All rights reserved.
// Use of this source code is governed by Apache 2 LICENSE that can be found in the LICENSE file.
package prog
import (
"fmt"
)
func (p *Prog) Clone() *Prog {
return p.cloneWithMap(make(map[*ResultArg]*ResultArg))
}
func (p *Prog) cloneWithMap(newargs map[*ResultArg]*ResultArg) *Prog {
if p.isUnsafe {
// We could clone it, but since we prohibit mutation
// of unsafe programs, it's unclear why we would clone it.
// Note: this also covers cloning of corpus programs
// during mutation, so if this is removed, we may need
// additional checks during mutation.
panic("cloning of unsafe programs is not supposed to be done")
}
p1 := &Prog{
Target: p.Target,
Calls: cloneCalls(p.Calls, newargs),
}
p1.debugValidate()
return p1
}
func cloneCalls(origCalls []*Call, newargs map[*ResultArg]*ResultArg) []*Call {
calls := make([]*Call, len(origCalls))
for ci, c := range origCalls {
calls[ci] = cloneCall(c, newargs)
}
return calls
}
func cloneCall(c *Call, newargs map[*ResultArg]*ResultArg) *Call {
c1 := new(Call)
c1.Meta = c.Meta
if c.Ret != nil {
c1.Ret = clone(c.Ret, newargs).(*ResultArg)
}
c1.Args = make([]Arg, len(c.Args))
for ai, arg := range c.Args {
c1.Args[ai] = clone(arg, newargs)
}
c1.Props = c.Props
return c1
}
func CloneArg(arg Arg) Arg {
return clone(arg, nil)
}
func clone(arg Arg, newargs map[*ResultArg]*ResultArg) Arg {
var arg1 Arg
switch a := arg.(type) {
case *ConstArg:
a1 := new(ConstArg)
*a1 = *a
arg1 = a1
case *PointerArg:
a1 := new(PointerArg)
*a1 = *a
arg1 = a1
if a.Res != nil {
a1.Res = clone(a.Res, newargs)
}
case *DataArg:
a1 := new(DataArg)
*a1 = *a
a1.data = append([]byte{}, a.data...)
arg1 = a1
case *GroupArg:
a1 := new(GroupArg)
*a1 = *a
arg1 = a1
a1.Inner = make([]Arg, len(a.Inner))
for i, arg2 := range a.Inner {
a1.Inner[i] = clone(arg2, newargs)
}
case *UnionArg:
a1 := new(UnionArg)
*a1 = *a
arg1 = a1
a1.Option = clone(a.Option, newargs)
case *ResultArg:
a1 := new(ResultArg)
*a1 = *a
arg1 = a1
if a1.Res != nil {
r := a1.Res
if newargs != nil {
r = newargs[a1.Res]
a1.Res = r
}
if r.uses == nil {
r.uses = make(map[*ResultArg]bool)
}
r.uses[a1] = true
}
a1.uses = nil // filled when we clone the referent
if newargs != nil {
newargs[a] = a1
}
default:
panic(fmt.Sprintf("bad arg kind: %#v", arg))
}
return arg1
}
|
