How to use LoggingHandler method of ast Package

Best Syzkaller code snippet using ast.LoggingHandler

consts.go

Source:consts.go Github

copy

Full Screen

...27 msg = fmt.Sprintf(msg, args...)28 if eh0 != nil {29 eh0(pos, msg)30 } else {31 ast.LoggingHandler(pos, msg)32 }33 }34 info := &ConstInfo{35 Defines: make(map[string]string),36 }37 includeMap := make(map[string]bool)38 incdirMap := make(map[string]bool)39 constMap := make(map[string]bool)40 ast.Walk(desc, func(n1 ast.Node) {41 switch n := n1.(type) {42 case *ast.Include:43 file := n.File.Value44 if includeMap[file] {45 eh(n.Pos, "duplicate include %q", file)46 }47 includeMap[file] = true48 info.Includes = append(info.Includes, file)49 case *ast.Incdir:50 dir := n.Dir.Value51 if incdirMap[dir] {52 eh(n.Pos, "duplicate incdir %q", dir)53 }54 incdirMap[dir] = true55 info.Incdirs = append(info.Incdirs, dir)56 case *ast.Define:57 v := fmt.Sprint(n.Value.Value)58 switch {59 case n.Value.CExpr != "":60 v = n.Value.CExpr61 case n.Value.Ident != "":62 v = n.Value.Ident63 }64 name := n.Name.Name65 if info.Defines[name] != "" {66 eh(n.Pos, "duplicate define %v", name)67 }68 info.Defines[name] = v69 constMap[name] = true70 case *ast.Call:71 if target.SyscallNumbers && !strings.HasPrefix(n.CallName, "syz_") {72 constMap[target.SyscallPrefix+n.CallName] = true73 }74 case *ast.Type:75 if c := typeConstIdentifier(n); c != nil {76 constMap[c.Ident] = true77 constMap[c.Ident2] = true78 }79 case *ast.Int:80 constMap[n.Ident] = true81 }82 })83 if errors != 0 {84 return nil85 }86 info.Consts = toArray(constMap)87 return info88}89// assignSyscallNumbers assigns syscall numbers, discards unsupported syscalls90// and removes no longer irrelevant nodes from the tree (comments, new lines, etc).91func (comp *compiler) assignSyscallNumbers(consts map[string]uint64) {92 // Pseudo syscalls starting from syz_ are assigned numbers starting from syzbase.93 // Note: the numbers must be stable (not depend on file reading order, etc),94 // so we have to do it in 2 passes.95 const syzbase = 100000096 syzcalls := make(map[string]bool)97 for _, decl := range comp.desc.Nodes {98 c, ok := decl.(*ast.Call)99 if !ok {100 continue101 }102 if strings.HasPrefix(c.CallName, "syz_") {103 syzcalls[c.CallName] = true104 }105 }106 syznr := make(map[string]uint64)107 for i, name := range toArray(syzcalls) {108 syznr[name] = syzbase + uint64(i)109 }110 var top []ast.Node111 for _, decl := range comp.desc.Nodes {112 switch decl.(type) {113 case *ast.Call:114 c := decl.(*ast.Call)115 if strings.HasPrefix(c.CallName, "syz_") {116 c.NR = syznr[c.CallName]117 top = append(top, decl)118 continue119 }120 if !comp.target.SyscallNumbers {121 top = append(top, decl)122 continue123 }124 // Lookup in consts.125 str := comp.target.SyscallPrefix + c.CallName126 nr, ok := consts[str]127 top = append(top, decl)128 if ok {129 c.NR = nr130 continue131 }132 c.NR = ^uint64(0) // mark as unused to not generate it133 name := "syscall " + c.CallName134 if !comp.unsupported[name] {135 comp.unsupported[name] = true136 comp.warning(c.Pos, "unsupported syscall: %v due to missing const %v",137 c.CallName, str)138 }139 case *ast.IntFlags, *ast.Resource, *ast.Struct, *ast.StrFlags:140 top = append(top, decl)141 case *ast.NewLine, *ast.Comment, *ast.Include, *ast.Incdir, *ast.Define:142 // These are not needed anymore.143 default:144 panic(fmt.Sprintf("unknown node type: %#v", decl))145 }146 }147 comp.desc.Nodes = top148}149// patchConsts replaces all symbolic consts with their numeric values taken from consts map.150// Updates desc and returns set of unsupported syscalls and flags.151// After this pass consts are not needed for compilation.152func (comp *compiler) patchConsts(consts map[string]uint64) {153 var top []ast.Node154 for _, decl := range comp.desc.Nodes {155 switch decl.(type) {156 case *ast.IntFlags:157 // Unsupported flag values are dropped.158 n := decl.(*ast.IntFlags)159 var values []*ast.Int160 for _, v := range n.Values {161 if comp.patchIntConst(v.Pos, &v.Value, &v.Ident, consts, nil) {162 values = append(values, v)163 }164 }165 n.Values = values166 top = append(top, n)167 case *ast.StrFlags:168 top = append(top, decl)169 case *ast.Resource, *ast.Struct, *ast.Call:170 // Walk whole tree and replace consts in Int's and Type's.171 missing := ""172 ast.WalkNode(decl, func(n0 ast.Node) {173 switch n := n0.(type) {174 case *ast.Int:175 comp.patchIntConst(n.Pos, &n.Value, &n.Ident, consts, &missing)176 case *ast.Type:177 if c := typeConstIdentifier(n); c != nil {178 comp.patchIntConst(c.Pos, &c.Value, &c.Ident,179 consts, &missing)180 if c.HasColon {181 comp.patchIntConst(c.Pos2, &c.Value2, &c.Ident2,182 consts, &missing)183 }184 }185 }186 })187 if missing == "" {188 top = append(top, decl)189 continue190 }191 // Produce a warning about unsupported syscall/resource/struct.192 // TODO(dvyukov): we should transitively remove everything that193 // depends on unsupported things.194 // Potentially we still can get, say, a bad int range error195 // due to the 0 const value.196 pos, typ, name := decl.Info()197 if id := typ + " " + name; !comp.unsupported[id] {198 comp.unsupported[id] = true199 comp.warning(pos, "unsupported %v: %v due to missing const %v",200 typ, name, missing)201 }202 // We have to keep partially broken resources and structs,203 // because otherwise their usages will error.204 top = append(top, decl)205 if c, ok := decl.(*ast.Call); ok {206 c.NR = ^uint64(0) // mark as unused to not generate it207 }208 }209 }210 comp.desc.Nodes = top211}212func (comp *compiler) patchIntConst(pos ast.Pos, val *uint64, id *string,213 consts map[string]uint64, missing *string) bool {214 if *id == "" {215 return true216 }217 v, ok := consts[*id]218 if !ok {219 name := "const " + *id220 if !comp.unsupported[name] {221 comp.unsupported[name] = true222 comp.warning(pos, "unsupported const: %v", *id)223 }224 if missing != nil && *missing == "" {225 *missing = *id226 }227 }228 *val = v229 *id = ""230 return ok231}232// typeConstIdentifier returns type arg that is an integer constant (subject for const patching), if any.233func typeConstIdentifier(n *ast.Type) *ast.Type {234 // TODO: see if we can extract this info from typeDesc/typeArg.235 if n.Ident == "const" && len(n.Args) > 0 {236 return n.Args[0]237 }238 if n.Ident == "array" && len(n.Args) > 1 && n.Args[1].Ident != "opt" {239 return n.Args[1]240 }241 if n.Ident == "vma" && len(n.Args) > 0 && n.Args[0].Ident != "opt" {242 return n.Args[0]243 }244 if n.Ident == "string" && len(n.Args) > 1 && n.Args[1].Ident != "opt" {245 return n.Args[1]246 }247 if n.Ident == "csum" && len(n.Args) > 2 && n.Args[1].Ident == "pseudo" {248 return n.Args[2]249 }250 switch n.Ident {251 case "int8", "int16", "int16be", "int32", "int32be", "int64", "int64be", "intptr":252 if len(n.Args) > 0 && n.Args[0].Ident != "opt" {253 return n.Args[0]254 }255 }256 return nil257}258func SerializeConsts(consts map[string]uint64) []byte {259 type nameValuePair struct {260 name string261 val uint64262 }263 var nv []nameValuePair264 for k, v := range consts {265 nv = append(nv, nameValuePair{k, v})266 }267 sort.Slice(nv, func(i, j int) bool {268 return nv[i].name < nv[j].name269 })270 buf := new(bytes.Buffer)271 fmt.Fprintf(buf, "# AUTOGENERATED FILE\n")272 for _, x := range nv {273 fmt.Fprintf(buf, "%v = %v\n", x.name, x.val)274 }275 return buf.Bytes()276}277func DeserializeConsts(data []byte, file string, eh ast.ErrorHandler) map[string]uint64 {278 consts := make(map[string]uint64)279 pos := ast.Pos{280 File: file,281 Line: 1,282 }283 ok := true284 s := bufio.NewScanner(bytes.NewReader(data))285 for ; s.Scan(); pos.Line++ {286 line := s.Text()287 if line == "" || line[0] == '#' {288 continue289 }290 eq := strings.IndexByte(line, '=')291 if eq == -1 {292 eh(pos, "expect '='")293 ok = false294 continue295 }296 name := strings.TrimSpace(line[:eq])297 val, err := strconv.ParseUint(strings.TrimSpace(line[eq+1:]), 0, 64)298 if err != nil {299 eh(pos, fmt.Sprintf("failed to parse int: %v", err))300 ok = false301 continue302 }303 if _, ok := consts[name]; ok {304 eh(pos, fmt.Sprintf("duplicate const %q", name))305 ok = false306 continue307 }308 consts[name] = val309 }310 if err := s.Err(); err != nil {311 eh(pos, fmt.Sprintf("failed to parse: %v", err))312 ok = false313 }314 if !ok {315 return nil316 }317 return consts318}319func DeserializeConstsGlob(glob string, eh ast.ErrorHandler) map[string]uint64 {320 if eh == nil {321 eh = ast.LoggingHandler322 }323 files, err := filepath.Glob(glob)324 if err != nil {325 eh(ast.Pos{}, fmt.Sprintf("failed to find const files: %v", err))326 return nil327 }328 if len(files) == 0 {329 eh(ast.Pos{}, fmt.Sprintf("no const files matched by glob %q", glob))330 return nil331 }332 consts := make(map[string]uint64)333 for _, f := range files {334 data, err := ioutil.ReadFile(f)335 if err != nil {...

Full Screen

Full Screen

LoggingHandler

Using AI Code Generation

copy

Full Screen

1import (2func main() {3 fset := token.NewFileSet()4 file, err := parser.ParseFile(fset, "log.go", nil, parser.ParseComments)5 if err != nil {6 log.Fatal(err)7 }8 ast.Print(fset, file)9 fmt.Println("----------------------------")10 f, err := os.Create("ast.txt")11 if err != nil {12 log.Fatal(err)13 }14 defer f.Close()15 ast.Fprint(f, fset, file, nil)16}

Full Screen

Full Screen

LoggingHandler

Using AI Code Generation

copy

Full Screen

1import (2func main() {3 fileSet := token.NewFileSet()4 file, err := parser.ParseFile(fileSet, "1.go", nil, parser.ParseComments)5 if err != nil {6 log.Fatal(err)7 }8 ast.Inspect(file, func(n ast.Node) bool {9 switch x := n.(type) {10 fmt.Println(x.Name)11 }12 })13}14import (15func main() {16 fmt.Println("Hello World!")17}

Full Screen

Full Screen

LoggingHandler

Using AI Code Generation

copy

Full Screen

1import (2func main() {3 logs.Trace("Trace message")4 logs.Debug("Debug message")5 logs.Info("Info message")6 logs.Warn("Warn message")7 logs.Error("Error message")8 logs.Critical("Critical message")9}10import (11func main() {12 logs.SetLogger(logs.AdapterConsole)13 logs.SetLogger(logs.AdapterFile, `{"filename":"test.log"}`)14 logs.Trace("Trace message")15 logs.Debug("Debug message")16 logs.Info("Info message")17 logs.Warn("Warn message")18 logs.Error("Error message")19 logs.Critical("Critical message")20}21import (22func main() {23 logs.SetLogger(logs.AdapterConsole)24 logs.SetLogger(logs.AdapterFile, `{"filename":"test.log"}`)

Full Screen

Full Screen

LoggingHandler

Using AI Code Generation

copy

Full Screen

1func main() {2 logger := log.New(os.Stdout, "logger: ", log.Lshortfile)3 handler := LoggingHandler(logger)4 server := http.Server{5 }6 server.ListenAndServe()7}8func main() {9 logger := log.New(os.Stdout, "logger: ", log.Lshortfile)10 handler := LoggingHandler(logger)11 server := http.Server{12 }13 server.ListenAndServe()14}15func main() {16 logger := log.New(os.Stdout, "logger: ", log.Lshortfile)17 handler := LoggingHandler(logger)18 server := http.Server{19 }20 server.ListenAndServe()21}22func main() {23 logger := log.New(os.Stdout, "logger: ", log.Lshortfile)24 handler := LoggingHandler(logger)25 server := http.Server{26 }27 server.ListenAndServe()28}29func main() {30 logger := log.New(os.Stdout, "logger: ", log.Lshortfile)31 handler := LoggingHandler(logger)32 server := http.Server{33 }34 server.ListenAndServe()35}36func main() {

Full Screen

Full Screen

LoggingHandler

Using AI Code Generation

copy

Full Screen

1import (2func main() {3 Ast := ast.NewAst()4 AstNode := ast.NewAstNode("AstNode")5 AstNode.AddChild(ast.NewAstNode("ChildNode"))6 AstNode.AddSibling(ast.NewAstNode("SiblingNode"))7 AstNode.AddParent(ast.NewAstNode("ParentNode"))8 AstNode.AddChild(ast.NewAstNode("ChildNode2"))9 AstNode.AddSibling(ast.NewAstNode("SiblingNode2"))10 AstNode.AddParent(ast.NewAstNode("ParentNode2"))11 AstNode.AddChild(ast.NewAstNode("ChildNode3"))12 AstNode.AddSibling(ast.NewAstNode("SiblingNode3"))13 AstNode.AddParent(ast.NewAstNode("ParentNode3"))14 AstNode.AddChild(ast.NewAstNode("ChildNode4"))15 AstNode.AddSibling(ast.NewAstNode("SiblingNode4"))16 AstNode.AddParent(ast.NewAstNode("ParentNode4"))17 AstNode.AddChild(ast.NewAstNode("ChildNode5"))18 AstNode.AddSibling(ast.NewAstNode("SiblingNode5"))19 AstNode.AddParent(ast.NewAstNode("ParentNode5"))20 AstNode.AddChild(ast.NewAstNode("ChildNode6"))21 AstNode.AddSibling(ast.NewAstNode("SiblingNode6"))22 AstNode.AddParent(ast.NewAstNode("ParentNode6"))

Full Screen

Full Screen

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.

LambdaTest Learning Hubs:

YouTube

You could also refer to video tutorials over LambdaTest YouTube channel to get step by step demonstration from industry experts.

Run Syzkaller automation tests on LambdaTest cloud grid

Perform automation testing on 3000+ real desktop and mobile devices online.

... View sample repo... View DocumentationSign up Free
›
_

Most used method in

Try LambdaTest Now !!

Get 100 minutes of automation test minutes FREE!!

...
Sign up with GoogleSign up with Email
Next-Gen App & Browser Testing Cloud

Was this article helpful?

Helpful

NotHelpful