How to use ForeachCallType method of prog Package

Best Syzkaller code snippet using prog.ForeachCallType

refine.go

Source:refine.go

...72		// Generate a testcase based on our template73		rnd := rand.New(rand.NewSource(time.Now().UnixNano()))74		choiceTable := target.BuildChoiceTable(corpus, calls)75		var resources []string76		prog.ForeachCallType(targetCall, func(typ prog.Type, ctx prog.TypeCtx) {77			if ctx.Dir == prog.DirOut {78				return79			}80			switch typ1 := typ.(type) {81			case *prog.ResourceType:82				log.Logf(0, "find resource %s %s", typ1.TypeName, typ1.Desc.Name)83				resources = append(resources, typ1.Desc.Name)84			default:85				// Some special dependences are encoded as const/union/flags, etc.86				if strings.Contains(typ.Name(), "connection") {87					log.Logf(0, "find special dependence %s", typ.Name())88					resources = append(resources, typ.Name())89				}90			}91		})92		found := false93		// Find the most similar syscall in the reversed order94		for _, p := range corpus {95			// for idx := len(p.Calls)-1; idx >= 0; idx-- {96				// c := p.Calls[idx]97			for idx, c := range p.Calls {98				results := make(map[string]bool)99				prog.ForeachCallType(c.Meta, func(typ prog.Type, ctx prog.TypeCtx) {100					switch typ1 := typ.(type) {101					case *prog.ResourceType:102						if ctx.Dir == prog.DirIn {103							results[typ1.Desc.Name] = true104						}105					default:106						if strings.Contains(typ.Name(), "connection") {107							log.Logf(0, "match special dependence %s", typ.Name())108							results[typ.Name()] = true109						}110					}111				})112				succeed := true113				for _, res := range resources {114					if _, ok := results[res]; !ok {115						succeed = false116						break117					}118				}119				if succeed {120					found = true121					log.Logf(0, "%d: find the prog with similar syscall: %s", idx, p.Serialize())122					p := p.Clone()123					new_prog := target.GenerateValidProgram(rnd, targetCall, p, idx+1, choiceTable, corpus)124					candidates = append(candidates, new_prog)125					break126				}127			}128			if found {129				break130			}131		}132		if !found {133			for _, p := range corpus {134				pos := -1135				succeed := true136				for _, resource := range resources {137					// Find the first use of the resource138					var indices []int139					for idx, c := range p.Calls {140						prog.ForeachCallType(c.Meta, func(typ prog.Type, ctx prog.TypeCtx) {141							if ctx.Dir == prog.DirOut {142								return143							}144							switch typ1 := typ.(type) {145							case *prog.ResourceType:146								if resource == typ1.Desc.Name {147									indices = append(indices, idx)148								}149							default:150								if resource == typ.Name() {151									indices = append(indices, idx)152								}153							}154						})...

resources.go

Source:resources.go

...114	return true115}116func (target *Target) getInputResources(c *Syscall) []*ResourceDesc {117	var resources []*ResourceDesc118	ForeachCallType(c, func(typ Type, ctx TypeCtx) {119		if ctx.Dir == DirOut {120			return121		}122		switch typ1 := typ.(type) {123		case *ResourceType:124			if !typ1.IsOptional {125				resources = append(resources, typ1.Desc)126			}127		case *StructType:128			if target.OS == "linux" && (typ1.Name() == "timespec" || typ1.Name() == "timeval") {129				resources = append(resources, timespecRes)130			}131		}132	})133	return resources134}135func (target *Target) getOutputResources(c *Syscall) []*ResourceDesc {136	var resources []*ResourceDesc137	ForeachCallType(c, func(typ Type, ctx TypeCtx) {138		switch typ1 := typ.(type) {139		case *ResourceType:140			if ctx.Dir != DirIn {141				resources = append(resources, typ1.Desc)142			}143		}144	})145	if c.CallName == "clock_gettime" {146		resources = append(resources, timespecRes)147	}148	return resources149}150func (target *Target) transitivelyEnabled(enabled map[*Syscall]bool) (map[*Syscall]bool, map[string]bool) {151	supported := make(map[*Syscall]bool, len(enabled))...

ForeachCallType

Using AI Code Generation

1import (2func main() {3	f, err := parser.ParseFile(fset, "2.go", nil, parser.ImportsOnly)4	if err != nil {5		panic(err)6	}7	fmt.Println("Import declarations in", f.Name.Name)8	for _, s := range f.Imports {9		fmt.Println(s.Path.Value)10	}11}

ForeachCallType

Using AI Code Generation

1import (2func main() {3	f, err := parser.ParseFile(fset, "2.go", nil, parser.AllErrors)4	if err != nil {5	}6	ast.Inspect(f, func(n ast.Node) bool {7		switch x := n.(type) {8			fmt.Println(fset.Position(n.Pos()), ":", fset.Position(n.End()), ":", x.Fun)9		}10	})11}

ForeachCallType

Using AI Code Generation

1import (2func main() {3	prog, err := elf.Open("1")4	if err != nil {5		fmt.Println(err)6	}7	symbols, err := prog.Symbols()8	if err != nil {9		fmt.Println(err)10	}11	for _, symbol := range symbols {12		fmt.Printf("%s %s %s %s %s %s13	}14}

ForeachCallType

Using AI Code Generation

1import (2func main() {3	fset := token.NewFileSet()4	f, err := parser.ParseFile(fset, "1.go", nil, parser.ParseComments)5	if err != nil {6		fmt.Println(err)7	}8	ast.ForeachCallType(f, func(fun ast.Expr, args []ast.Expr) {9		fmt.Println(fun)10	})11}

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.