KaneAI - World’s First E2E Software Testing Agent.Book a Demo Join TestMu Conference

How to use genUnionArg method of proggen Package

Best Syzkaller code snippet using proggen.genUnionArg

proggen.go

Source:proggen.go

...162		return ctx.genStruct(a, dir, traceArg)163	case *prog.ArrayType:164		return ctx.genArray(a, dir, traceArg)165	case *prog.UnionType:166		return ctx.genUnionArg(a, dir, traceArg)167	case *prog.VmaType:168		return ctx.genVma(a, dir, traceArg)169	default:170		log.Fatalf("unsupported type: %#v", syzType)171	}172	return nil173}174func (ctx *context) genVma(syzType *prog.VmaType, dir prog.Dir, _ parser.IrType) prog.Arg {175	npages := uint64(1)176	if syzType.RangeBegin != 0 || syzType.RangeEnd != 0 {177		npages = syzType.RangeEnd178	}179	return prog.MakeVmaPointerArg(syzType, dir, ctx.builder.AllocateVMA(npages), npages)180}181func (ctx *context) genArray(syzType *prog.ArrayType, dir prog.Dir, traceType parser.IrType) prog.Arg {182	var args []prog.Arg183	switch a := traceType.(type) {184	case *parser.GroupType:185		for i := 0; i < len(a.Elems); i++ {186			args = append(args, ctx.genArg(syzType.Elem, dir, a.Elems[i]))187		}188	default:189		log.Fatalf("unsupported type for array: %#v", traceType)190	}191	return prog.MakeGroupArg(syzType, dir, args)192}193func (ctx *context) genStruct(syzType *prog.StructType, dir prog.Dir, traceType parser.IrType) prog.Arg {194	var args []prog.Arg195	switch a := traceType.(type) {196	case *parser.GroupType:197		j := 0198		if ret, recursed := ctx.recurseStructs(syzType, dir, a); recursed {199			return ret200		}201		for i := range syzType.Fields {202			if prog.IsPad(syzType.Fields[i].Type) {203				args = append(args, syzType.Fields[i].DefaultArg(dir))204				continue205			}206			// If the last n fields of a struct are zero or NULL, strace will occasionally omit those values207			// this creates a mismatch in the number of elements in the ir type and in208			// our descriptions. We generate default values for omitted fields209			if j >= len(a.Elems) {210				args = append(args, syzType.Fields[i].DefaultArg(dir))211			} else {212				args = append(args, ctx.genArg(syzType.Fields[i].Type, dir, a.Elems[j]))213			}214			j++215		}216	case *parser.BufferType:217		// We could have a case like the following:218		// ioctl(3, 35111, {ifr_name="\x6c\x6f", ifr_hwaddr=00:00:00:00:00:00}) = 0219		// if_hwaddr gets parsed as a BufferType but our syscall descriptions have it as a struct type220		return syzType.DefaultArg(dir)221	default:222		log.Fatalf("unsupported type for struct: %#v", a)223	}224	return prog.MakeGroupArg(syzType, dir, args)225}226// recurseStructs handles cases where syzType corresponds to struct descriptions like227// sockaddr_storage_in6 {228//        addr    sockaddr_in6229// } [size[SOCKADDR_STORAGE_SIZE], align_ptr]230// which need to be recursively generated. It returns true if we needed to recurse231// along with the generated argument and false otherwise.232func (ctx *context) recurseStructs(syzType *prog.StructType, dir prog.Dir, traceType *parser.GroupType) (prog.Arg, bool) {233	// only consider structs with one non-padded field234	numFields := 0235	for _, field := range syzType.Fields {236		if prog.IsPad(field.Type) {237			continue238		}239		numFields++240	}241	if numFields != 1 {242		return nil, false243	}244	// the strace group type needs to have more one field (a mismatch)245	if len(traceType.Elems) == 1 {246		return nil, false247	}248	// first field needs to be a struct249	switch t := syzType.Fields[0].Type.(type) {250	case *prog.StructType:251		var args []prog.Arg252		// first element and traceType should have the same number of elements253		if len(t.Fields) != len(traceType.Elems) {254			return nil, false255		}256		args = append(args, ctx.genStruct(t, dir, traceType))257		for _, field := range syzType.Fields[1:] {258			args = append(args, field.DefaultArg(dir))259		}260		return prog.MakeGroupArg(syzType, dir, args), true261	}262	return nil, false263}264func (ctx *context) genUnionArg(syzType *prog.UnionType, dir prog.Dir, straceType parser.IrType) prog.Arg {265	if straceType == nil {266		log.Logf(1, "generating union arg. straceType is nil")267		return syzType.DefaultArg(dir)268	}269	log.Logf(4, "generating union arg: %s %#v", syzType.TypeName, straceType)270	// Unions are super annoying because they sometimes need to be handled case by case271	// We might need to lookinto a matching algorithm to identify the union type that most closely272	// matches our strace type.273	switch syzType.TypeName {274	case "sockaddr_storage":275		return ctx.genSockaddrStorage(syzType, dir, straceType)276	case "sockaddr_nl":277		return ctx.genSockaddrNetlink(syzType, dir, straceType)278	case "ifr_ifru":...

genUnionArg

Using AI Code Generation

1import (2func main() {3	target, err := prog.GetTarget("linux", "amd64")4	if err != nil {5		panic(err)6	}7	p := target.MakeProg()8	arg := proggen.GenUnionArg(p, target.SyscallMap["syz_open_dev"].Args[0].Type().(*prog.TypeUnion), nil)9	fmt.Println(arg)10}

genUnionArg

Using AI Code Generation

1import (2func main() {3	target, err := prog.GetTarget("linux", "amd64")4	if err != nil {5		fmt.Printf("prog.GetTarget failed: %v", err)6	}7	p := target.MakeProg()8	gen := proggen.Get(target)9	arg := gen.GenUnionArg(p, target.SyscallMap["syz_open_dev"].Args[0], nil)10	fmt.Printf("Argument: %v11}12import (13func main() {14	target, err := prog.GetTarget("linux", "amd64")15	if err != nil {16		fmt.Printf("prog.GetTarget failed: %v", err)17	}18	p := target.MakeProg()19	gen := proggen.Get(target)20	unionArg := gen.GenUnionArg(p, target.SyscallMap["syz_open_dev"].Args[0], nil)21	unionType := unionArg.Type().(*prog.UnionType)22	arg := gen.GenUnionArg(p, unionType.Fields[0], nil)23	fmt.Printf("Argument: %v24}

genUnionArg

Using AI Code Generation

1import (2func main() {3	p := new(proggen.ProgGen)4	p.GenUnionArg()5}6import (7type ProgGen struct {8}9func (p *ProgGen) GenUnionArg() {10	fmt.Println("union arguments generated")11}12import (13func main() {14	p := new(proggen.ProgGen)15	p.GenUnionArg()16}17import (18type ProgGen struct {19}20func (p *ProgGen) GenUnionArg() {21	fmt.Println("union arguments generated")22}23import (24func main() {25	p := new(proggen.ProgGen)26	p.GenUnionArg()27}28import (29type ProgGen struct {30}31func (p *ProgGen) GenUnionArg() {32	fmt.Println("union arguments generated")33}34import (35func main() {36	p := new(proggen.ProgGen)37	p.GenUnionArg()38}39import (40type ProgGen struct {41}42func (p *ProgGen) GenUnionArg() {43	fmt.Println("union arguments generated")44}

genUnionArg

Using AI Code Generation

1import (2func main() {3	pg.GenUnionArg()4	fmt.Println("Hello, playground")5}6import (7func main() {8	pg.GenUnionArg()9	fmt.Println("Hello, playground")10}11import (12func main() {13	pg.GenUnionArg()14	fmt.Println("Hello, playground")15}16import (17func main() {18	pg.GenUnionArg()19	fmt.Println("Hello, playground")20}21import (22func main() {23	pg.GenUnionArg()24	fmt.Println("Hello, playground")25}26import (27func main() {28	pg.GenUnionArg()29	fmt.Println("Hello, playground")30}31import (32func main() {33	pg.GenUnionArg()34	fmt.Println("Hello, playground")35}36import (37func main() {38	pg.GenUnionArg()39	fmt.Println("Hello, playground")40}

genUnionArg

Using AI Code Generation

1import (2func main() {3    p := proggen.New(1, 1)4    p.Target = targets.Get("test", "64")5    p.Target.MakeMmap = func(vma prog.Vma, typ prog.Type, data []byte) (res prog.Arg, calls []*prog.Call) {6    }7    p.Target.MakeVma = func(typ prog.Type, data []byte) (res prog.Arg, calls []*prog.Call) {8    }9    p.Target.MakeInt64 = func(typ prog.Type) (res prog.Arg, calls []*prog.Call) {10    }11    p.Target.MakeInt32 = func(typ prog.Type) (res prog.Arg, calls []*prog.Call) {12    }13    p.Target.MakeInt16 = func(typ prog.Type) (res prog.Arg, calls []*prog.Call) {14    }15    p.Target.MakeInt8 = func(typ prog.Type) (res prog.Arg, calls []*prog.Call) {16    }17    p.Target.MakeConstArg = func(typ prog.Type, v uint64) (res prog.Arg, calls []*prog.Call) {18    }19    p.Target.MakeResultArg = func(typ prog.Type, res []prog.Arg, csum prog.Checksum) (prog.Arg, []*prog.Call) {20    }21    p.Target.MakeReturnArg = func(typ prog.Type) (prog.Arg, []*prog.Call) {22    }23    p.Target.MakeUnionArg = func(typ prog.Type, fields []prog.UnionField, values []prog.Arg) (prog.Arg, []*prog.Call) {24    }25    p.Target.MakeStructArg = func(typ prog.Type, fields []prog.StructField, values []prog.Arg) (prog.Arg, []*prog.Call) {26    }27    p.Target.MakeArrayArg = func(typ prog.Type, values []prog.Arg) (prog.Arg, []*prog.Call) {28    }29    p.Target.MakeGroupArg = func(typ prog.Type, values []prog.Arg) (prog.Arg, []*prog.Call) {30    }

genUnionArg

Using AI Code Generation

1import (2var (3	flagOut = flag.String("out", "", "output directory")4func main() {5	flag.Parse()6	p.Syscalls = make(map[int64]*prog.Syscall)7	p.SyscallMap = make(map[string]*prog.Syscall)8	for i := 0; i < 3; i++ {9		s := prog.Syscall{10			Name:       "syz_" + strconv.Itoa(i),11			Num:        int64(i),12			NR:         int64(i),13			CallName:   "syz_" + strconv.Itoa(i),14			Args:       make([]prog.Arg, 0),15		}16		for j := 0; j < 3; j++ {17			a.Type = &prog.UnionType{18				Fields: []prog.Type{19					&prog.IntType{20						TypeCommon: prog.TypeCommon{21						},22					},23					&prog.BufferType{24						TypeCommon: prog.TypeCommon{25						},26						Range: prog.IntRange{27						},28					},29				},30			}31			a.Option = prog.UnionOption(j)32			s.Args = append(s.Args, &a)33		}34		p.Syscalls[int64(i)] = &s35		p.SyscallMap["syz_"+strconv.Itoa(i)] = &s36	}37	g := gen.MakeProgGen(&p)38	g.GenUnionArg()39}40import (41func (g *ProgGen) GenUnionArg() {42	data := struct {43	}{44	}

genUnionArg

Using AI Code Generation

1import (2func main() {3	p := proggen.NewTarget("linux", "amd64")4	unionArg := p.GenUnionArg(p.Target.Syscalls[0].Args[1])5	syscall := p.GenSyscall(p.Target.Syscalls[0], unionArg)6	fmt.Printf("%v", syscall)7}8import (9func main() {10	p := proggen.NewTarget("linux", "amd64")11	unionArg := p.GenUnionArg(p.Target.Syscalls[0].Args[1])12	syscall := p.GenSyscall(p.Target.Syscalls[0], unionArg)13	fmt.Printf("%v", syscall)14}15import (16func main() {17	p := proggen.NewTarget("linux", "amd64")18	unionArg := p.GenUnionArg(p.Target.Syscalls[0].Args[1])19	syscall := p.GenSyscall(p.Target.Syscalls[0], unionArg)20	fmt.Printf("%v", syscall)21}

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.