| 1 | // Copyright 2021 The Go Authors. All rights reserved. |
|---|---|
| 2 | // Use of this source code is governed by a BSD-style |
| 3 | // license that can be found in the LICENSE file. |
| 4 | |
| 5 | package generator |
| 6 | |
| 7 | import ( |
| 8 | "bytes" |
| 9 | "fmt" |
| 10 | ) |
| 11 | |
| 12 | // typedefparm describes a parameter that is a typedef of some other |
| 13 | // type; it implements the "parm" interface |
| 14 | type typedefparm struct { |
| 15 | aname string |
| 16 | qname string |
| 17 | target parm |
| 18 | isBlank |
| 19 | addrTakenHow |
| 20 | isGenValFunc |
| 21 | skipCompare |
| 22 | } |
| 23 | |
| 24 | func (p typedefparm) Declare(b *bytes.Buffer, prefix string, suffix string, caller bool) { |
| 25 | n := p.aname |
| 26 | if caller { |
| 27 | n = p.qname |
| 28 | } |
| 29 | b.WriteString(fmt.Sprintf("%s %s%s", prefix, n, suffix)) |
| 30 | } |
| 31 | |
| 32 | func (p typedefparm) GenElemRef(elidx int, path string) (string, parm) { |
| 33 | _, isarr := p.target.(*arrayparm) |
| 34 | _, isstruct := p.target.(*structparm) |
| 35 | _, ismap := p.target.(*mapparm) |
| 36 | rv, rp := p.target.GenElemRef(elidx, path) |
| 37 | // this is hacky, but I don't see a nicer way to do this |
| 38 | if isarr || isstruct || ismap { |
| 39 | return rv, rp |
| 40 | } |
| 41 | rp = &p |
| 42 | return rv, rp |
| 43 | } |
| 44 | |
| 45 | func (p typedefparm) GenValue(s *genstate, f *funcdef, value int, caller bool) (string, int) { |
| 46 | n := p.aname |
| 47 | if caller { |
| 48 | n = p.qname |
| 49 | } |
| 50 | rv, v := s.GenValue(f, p.target, value, caller) |
| 51 | rv = n + "(" + rv + ")" |
| 52 | return rv, v |
| 53 | } |
| 54 | |
| 55 | func (p typedefparm) IsControl() bool { |
| 56 | return false |
| 57 | } |
| 58 | |
| 59 | func (p typedefparm) NumElements() int { |
| 60 | return p.target.NumElements() |
| 61 | } |
| 62 | |
| 63 | func (p typedefparm) String() string { |
| 64 | return fmt.Sprintf("%s typedef of %s", p.aname, p.target.String()) |
| 65 | |
| 66 | } |
| 67 | |
| 68 | func (p typedefparm) TypeName() string { |
| 69 | return p.aname |
| 70 | |
| 71 | } |
| 72 | |
| 73 | func (p typedefparm) QualName() string { |
| 74 | return p.qname |
| 75 | } |
| 76 | |
| 77 | func (p typedefparm) HasPointer() bool { |
| 78 | return p.target.HasPointer() |
| 79 | } |
| 80 | |
| 81 | func (s *genstate) makeTypedefParm(f *funcdef, target parm, pidx int) parm { |
| 82 | var tdp typedefparm |
| 83 | ns := len(f.typedefs) |
| 84 | tdp.aname = fmt.Sprintf("MyTypeF%dS%d", f.idx, ns) |
| 85 | tdp.qname = fmt.Sprintf("%s.MyTypeF%dS%d", s.checkerPkg(pidx), f.idx, ns) |
| 86 | tdp.target = target |
| 87 | tdp.SetBlank(uint8(s.wr.Intn(100)) < tunables.blankPerc) |
| 88 | f.typedefs = append(f.typedefs, tdp) |
| 89 | return &tdp |
| 90 | } |
| 91 |
Members