How to use ForeachArg method of prog Package

Best Syzkaller code snippet using prog.ForeachArg

analysis.go

Source:analysis.go

1// Copyright 2015 syzkaller project authors. All rights reserved.2// Use of this source code is governed by Apache 2 LICENSE that can be found in the LICENSE file.3// Conservative resource-related analysis of programs.4// The analysis figures out what files descriptors are [potentially] opened5// at a particular point in program, what pages are [potentially] mapped,6// what files were already referenced in calls, etc.7package prog8import (9	"fmt"10)11const (12	maxPages = 4 << 1013)14type state struct {15	target    *Target16	ct        *ChoiceTable17	files     map[string]bool18	resources map[string][]Arg19	strings   map[string]bool20	pages     [maxPages]bool21}22// analyze analyzes the program p up to but not including call c.23func analyze(ct *ChoiceTable, p *Prog, c *Call) *state {24	s := newState(p.Target, ct)25	for _, c1 := range p.Calls {26		if c1 == c {27			break28		}29		s.analyze(c1)30	}31	return s32}33func newState(target *Target, ct *ChoiceTable) *state {34	s := &state{35		target:    target,36		ct:        ct,37		files:     make(map[string]bool),38		resources: make(map[string][]Arg),39		strings:   make(map[string]bool),40	}41	return s42}43func (s *state) analyze(c *Call) {44	foreachArgArray(&c.Args, c.Ret, func(arg, base Arg, _ *[]Arg) {45		switch typ := arg.Type().(type) {46		case *ResourceType:47			if typ.Dir() != DirIn {48				s.resources[typ.Desc.Name] = append(s.resources[typ.Desc.Name], arg)49				// TODO: negative PIDs and add them as well (that's process groups).50			}51		case *BufferType:52			a := arg.(*DataArg)53			if typ.Dir() != DirOut && len(a.Data) != 0 {54				switch typ.Kind {55				case BufferString:56					s.strings[string(a.Data)] = true57				case BufferFilename:58					s.files[string(a.Data)] = true59				}60			}61		}62	})63	start, npages, mapped := s.target.AnalyzeMmap(c)64	if npages != 0 {65		if start+npages > uint64(len(s.pages)) {66			panic(fmt.Sprintf("address is out of bounds: page=%v len=%v bound=%v",67				start, npages, len(s.pages)))68		}69		for i := uint64(0); i < npages; i++ {70			s.pages[start+i] = mapped71		}72	}73}74func foreachSubargImpl(arg Arg, parent *[]Arg, f func(arg, base Arg, parent *[]Arg)) {75	var rec func(arg, base Arg, parent *[]Arg)76	rec = func(arg, base Arg, parent *[]Arg) {77		f(arg, base, parent)78		switch a := arg.(type) {79		case *GroupArg:80			for _, arg1 := range a.Inner {81				parent1 := parent82				if _, ok := arg.Type().(*StructType); ok {83					parent1 = &a.Inner84				}85				rec(arg1, base, parent1)86			}87		case *PointerArg:88			if a.Res != nil {89				rec(a.Res, arg, parent)90			}91		case *UnionArg:92			rec(a.Option, base, parent)93		}94	}95	rec(arg, nil, parent)96}97func foreachSubarg(arg Arg, f func(arg, base Arg, parent *[]Arg)) {98	foreachSubargImpl(arg, nil, f)99}100func foreachArgArray(args *[]Arg, ret Arg, f func(arg, base Arg, parent *[]Arg)) {101	for _, arg := range *args {102		foreachSubargImpl(arg, args, f)103	}104	if ret != nil {105		foreachSubargImpl(ret, nil, f)106	}107}108func foreachArg(c *Call, f func(arg, base Arg, parent *[]Arg)) {109	foreachArgArray(&c.Args, nil, f)110}111func foreachSubargOffset(arg Arg, f func(arg Arg, offset uint64)) {112	var rec func(Arg, uint64) uint64113	rec = func(arg1 Arg, offset uint64) uint64 {114		switch a := arg1.(type) {115		case *GroupArg:116			f(arg1, offset)117			var totalSize uint64118			for _, arg2 := range a.Inner {119				size := rec(arg2, offset)120				if !arg2.Type().BitfieldMiddle() {121					offset += size122					totalSize += size123				}124			}125			if totalSize > arg1.Size() {126				panic(fmt.Sprintf("bad group arg size %v, should be <= %v for %+v", totalSize, arg1.Size(), arg1))127			}128		case *UnionArg:129			f(arg1, offset)130			size := rec(a.Option, offset)131			offset += size132			if size > arg1.Size() {133				panic(fmt.Sprintf("bad union arg size %v, should be <= %v for arg %+v with type %+v", size, arg1.Size(), arg1, arg1.Type()))134			}135		default:136			f(arg1, offset)137		}138		return arg1.Size()139	}140	rec(arg, 0)141}142func RequiresBitmasks(p *Prog) bool {143	result := false144	for _, c := range p.Calls {145		foreachArg(c, func(arg, _ Arg, _ *[]Arg) {146			if a, ok := arg.(*ConstArg); ok {147				if a.Type().BitfieldOffset() != 0 || a.Type().BitfieldLength() != 0 {148					result = true149				}150			}151		})152	}153	return result154}155func RequiresChecksums(p *Prog) bool {156	result := false157	for _, c := range p.Calls {158		foreachArg(c, func(arg, _ Arg, _ *[]Arg) {159			if _, ok := arg.Type().(*CsumType); ok {160				result = true161			}162		})163	}164	return result165}...

ForeachArg

Using AI Code Generation

1import (2func main() {3	f, err := parser.ParseFile(fset, "1.go", nil, parser.ParseComments)4	if err != nil {5		fmt.Println("error parsing file", err)6	}7	for _, s := range f.Imports {8		fmt.Println(s.Path.Value)9	}10	for _, s := range f.Decls {11		fmt.Println(s)12	}13	for _, s := range f.Scope.Objects {14		fmt.Println(s.Name)15	}16	for _, s := range f.Comments {17		fmt.Println(s)18	}19	for _, s := range f.Unresolved {20		fmt.Println(s)21	}22	for _, s := range f.Scope.Objects {23		fmt.Println(s)24	}25	for _, s := range f.Scope.Objects {26		fmt.Println(s)27	}28	for _, s := range f.Scope.Objects {29		fmt.Println(s)30	}31	for _, s := range f.Scope.Objects {32		fmt.Println(s)33	}34	for _, s := range f.Scope.Objects {35		fmt.Println(s)36	}37	for _, s := range f.Scope.Objects {38		fmt.Println(s)39	}40	for _, s := range f.Scope.Objects {41		fmt.Println(s)42	}43	for _, s := range f.Scope.Objects {44		fmt.Println(s)45	}46}47import (48func main() {

ForeachArg

Using AI Code Generation

1import (2func main() {3	target, err := prog.GetTarget("linux", "amd64")4	if err != nil {5		fmt.Fprintf(os.Stderr, "failed to create target instance: %v", err)6		os.Exit(1)7	}8	prog, err := target.Deserialize([]byte(`9		#syz call: openat$fs2(0xffffffffffffff9c, 0x7f0000000000, 0x0, 0x0)10		openat$fs2(0xffffffffffffff9c, 0x7f0000000000, 0x0, 0x0)11	if err != nil {12		fmt.Fprintf(os.Stderr, "failed to deserialize the prog: %v", err)13		os.Exit(1)14	}15	fmt.Println("Program:")16	fmt.Println(prog.String())17	fmt.Println("Arg values:")18	prog.Calls[0].ForeachArg(func(arg prog.Arg, ctx prog.VisitCtx) {19		fmt.Printf("%v: %#v20	})21}22openat$fs2(0xffffffffffffff9c, 0x7f0000000000, 0x0, 0x0)23Arg{0}: 0xffffffffffffff9c24Arg{0}->Dir: 0xffffffffffffff9c25Arg{1}: 0x7f000000000026Arg{1}->Path: 0x7f000000000027Arg{2}: 0x028Arg{2}->Flags: 0x029Arg{3}: 0x030Arg{3}->Mode: 0x0

ForeachArg

Using AI Code Generation

1import (2func main() {3	flag.Parse()4	if flag.NArg() < 1 {5		fmt.Fprintln(os.Stderr, "usage: 2 [packages]")6		os.Exit(1)7	}8	conf := loader.Config{Build: &loader.FakeContext{GoPath: os.Getenv("GOPATH")}}9	_, err := conf.FromArgs(flag.Args(), true)10	if err != nil {11		log.Fatal(err)12	}13	prog, err := conf.Load()14	if err != nil {15		log.Fatal(err)16	}17	mainPkg := prog.Package("main").Pkg18	prog.BuildAll()19	fmt.Printf("Members of package %q:20", mainPkg.Path())21	for _, mem := range mainPkg.Scope().Children() {22		fmt.Println(mem.Name())23	}24	fmt.Printf("25", mainPkg.Path())26	for _, mem := range mainPkg.Scope().Children() {27		if fn, ok := mem.(*ssa.Function); ok {28			fmt.Println(fn.Name())29		}30	}31	fmt.Printf("32", mainPkg.Path())33	for _, mem := range mainPkg.Scope().Children() {34		if fn, ok := mem.(*ssa.Function); ok {35			fmt.Println(fn.Name())36		}37	}38	fmt.Printf("39", mainPkg.Path())40	for _, mem := range mainPkg.Scope().Children() {41		if fn, ok := mem.(*ssa.Function); ok {42			fmt.Println(fn.Name())43		}44	}45	fmt.Printf("46", mainPkg.Path())47	for _, mem := range mainPkg.Scope().Children() {48		if fn, ok := mem.(*ssa.Function); ok {49			fmt.Println(fn.Name

ForeachArg

Using AI Code Generation

1import (2func main() {3	var (4		name = flag.String("name", "everyone", "The greeting object.")5	flag.Parse()6	fmt.Printf("Hello, %s!7}8import (9func main() {10	var (11		name = flag.String("name", "everyone", "The greeting object.")12	flag.Parse()13	fmt.Printf("Hello, %s!14}15import (16func main() {17	var (18		name = flag.String("name", "everyone", "The greeting object.")19	flag.Parse()20	fmt.Printf("Hello, %s!21}22import (23func main() {24	var (25		name = flag.String("name", "everyone", "The greeting object.")26	flag.Parse()27	fmt.Printf("Hello, %s!28}29import (30func main() {31	var (32		name = flag.String("name", "everyone", "The greeting object.")33	flag.Parse()34	fmt.Printf("Hello, %s!35}36import (37func main() {38	var (39		name = flag.String("name", "everyone", "The greeting object.")40	flag.Parse()41	fmt.Printf("Hello, %s!42}43import (44func main() {45	var (46		name = flag.String("name", "everyone", "The greeting object.")47	flag.Parse()48	fmt.Printf("Hello, %s!49}

ForeachArg

Using AI Code Generation

1import (2func main() {3    flagset := getopt.New()4    log.SetFlags(0)5    log.SetPrefix("findcallers: ")6    flagset.Usage = func() {7        fmt.Fprintln(os.Stderr, "findcallers finds callers of a function")8        fmt.Fprintln(os.Stderr, "usage: findcallers [flags] functionname")9        flagset.PrintDefaults()10    }11    flagset.Parse(true)12    if flagset.NArg() != 1 {13        flagset.Usage()14        os.Exit(2)15    }16    funcname := flagset.Arg(0)17    prog, err := ssautil.AllPackages(nil, 0)18    if err != nil {19        log.Fatal(err)20    }21    prog.Build()22    for _, mem := range prog.Members {23        if mem.Name() == funcname {24            fmt.Printf("25            for _, b := range mem.Blocks {26                for _, instr := range b.Instrs {27                    if call, ok := instr.(*ssa.Call); ok {28                        if call.Call.IsInvoke() {29                            fmt.Printf("%s.%s30", call.Call.Value.Type(), call.Call.Method)31                        } else {32                            fmt.Printf("%s33                        }34                    }35                }36            }37        }38    }39    if n == 0 {40        fmt.Printf("%s not found41    }42}

ForeachArg

Using AI Code Generation

1import (2func main() {3	p, err := prog.Deserialize([]byte("r1 = openat$sockfs(0xffffffffffffff9c, 0x7f0000000000, 0x0, 0x1)"))4	if err != nil {5		panic(err)6	}7	p.ForeachArg(func(arg prog.Arg, ctx prog.ForeachArgCtx) {8		if arg.Type().Dir() == prog.DirOut {9			ctx.ReplaceArg(arg)10		}11	})12	fmt.Println(p.Serialize())13}14r1 = openat$sockfs(0xffffffffffffff9c, 0x7f0000000000, 0x0, 0x1)15import (16func main() {17	p, err := prog.Deserialize([]byte("r1 = openat$sockfs(0xffffffffffffff9c, 0x7f0000000000, 0x0, 0x1)"))18	if err != nil {19		panic(err)20	}21	p.ForeachCall(func(c *prog.Call) {22		if c.Meta().Name == "openat$sockfs" {23			c.Meta = prog.Syscalls[prog.GetCallID("openat")]24		}25	})26	fmt.Println(p.Serialize())27}28r1 = openat(0xffffffffffffff9c, 0x7f0000000000, 0x0, 0x1)29import (30func main() {31	p, err := prog.Deserialize([]byte("r1 = openat$sockfs(0xffffffffffffff9c, 0x7f0000000000, 0x0, 0x1)"))32	if err != nil {33		panic(err)34	}35	p.ForeachArg(func(arg prog.Arg, ctx prog.ForeachArgCtx)

ForeachArg

Using AI Code Generation

1import (2func main() {3    p := prog.InitTarget("linux", "amd64").Prog()4    s := p.AppendSyscall(p.Target.Syscalls[0])5    s.AppendArg(p.Target.MakeReturnArg(s.Call))6    fmt.Printf("%v7    for _, a := range p.ForeachArg(nil) {8        fmt.Printf("%v9    }10}11openat$dirfd$AT_FDCWD(0x0, 0x0, 0x0)12import (13func main() {14    p := prog.InitTarget("linux", "amd64").Prog()15    s := p.AppendSyscall(p.Target.Syscalls[0])16    s.AppendArg(p.Target.MakeReturnArg(s.Call))17    fmt.Printf("%v18    for _, c := range p.ForeachCall(nil) {19        fmt.Printf("%v20    }21}22openat$dirfd$AT_FDCWD(0x0, 0x0, 0x0)23openat$dirfd$AT_FDCWD(0x0, 0x0, 0x0)

ForeachArg

Using AI Code Generation

1import (2func main() {3	profile := pprof.NewProfile("test")4	profile.Add("main.main", 1)5	profile.Add("main.main", 2)6	profile.Add("main.main", 3)7	profile.ForEach(func(f *pprof.Function) {8		fmt.Println(f.Name)9	})10	profile.ForEach(func(f *pprof.Function) {11		fmt.Println(f.Name)12	})

ForeachArg

Using AI Code Generation

1import (2func main() {3	cmd := exec.Command("ls", "-l")4	cmd.ForeachArg(func(arg string) {5		fmt.Println(arg)6	})7}8Related posts: Golang | exec.Command() function

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.