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
|
// Copyright 2026 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 codeeditor
import (
"path/filepath"
"strings"
"github.com/google/syzkaller/pkg/aflow"
"github.com/google/syzkaller/pkg/osutil"
)
var Tool = aflow.NewFuncTool("codeeditor", codeeditor, `
The tool does one code edit to form the final patch.
The tool should be called mutiple times to do all required changes one-by-one,
but avoid changing the same lines multiple times.
Note: You will not see your edits via the codesearch tool.
Note: The current code snippet should reflect the previous changes.
`)
type state struct {
KernelScratchSrc string
}
type args struct {
SourceFile string `jsonschema:"Full source file path."`
CurrentCode string `jsonschema:"The current code to replace verbatim with new lines, but without line numbers."`
NewCode string `jsonschema:"New code to replace the current code snippet."`
}
func codeeditor(ctx *aflow.Context, state state, args args) (struct{}, error) {
if strings.Contains(filepath.Clean(args.SourceFile), "..") {
return struct{}{}, aflow.BadCallError("SourceFile %q is outside of the source tree", args.SourceFile)
}
file := filepath.Join(state.KernelScratchSrc, args.SourceFile)
if !osutil.IsExist(file) {
return struct{}{}, aflow.BadCallError("SourceFile %q does not exist", args.SourceFile)
}
if strings.TrimSpace(args.CurrentCode) == "" {
return struct{}{}, aflow.BadCallError("CurrentCode snippet is empty")
}
// If SourceFile is incorrect, or CurrentCode is not matched, return aflow.BadCallError
// with an explanation. Say that it needs to increase context if CurrentCode is not matched.
// Try to do as fuzzy match for CurrentCode as possible (strip line numbers,
// ignore white-spaces, etc).
// Should we accept a reference line number, or function name to disambiguate in the case
// of multiple matches?
return struct{}{}, nil
}
|
