mirror of
https://github.com/golang/go
synced 2024-10-05 15:51:22 -06:00
fca50820cc
This CL changes both JoinGroup and LeaveGroup methods to take an interface as an argument for enabling IPv6 group address join/leave, join a group address on a specific interface. R=rsc, dave CC=golang-dev https://golang.org/cl/4815074
58 lines
1.1 KiB
Go
58 lines
1.1 KiB
Go
// Copyright 2011 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package main
|
|
|
|
import (
|
|
"go/ast"
|
|
)
|
|
|
|
var netudpgroupFix = fix{
|
|
"netudpgroup",
|
|
netudpgroup,
|
|
`Adapt 1-argument calls of net.(*UDPConn).JoinGroup, LeaveGroup to use 2-argument form.
|
|
|
|
http://codereview.appspot.com/4815074
|
|
`,
|
|
}
|
|
|
|
func init() {
|
|
register(netudpgroupFix)
|
|
}
|
|
|
|
func netudpgroup(f *ast.File) bool {
|
|
if !imports(f, "net") {
|
|
return false
|
|
}
|
|
|
|
fixed := false
|
|
for _, d := range f.Decls {
|
|
fd, ok := d.(*ast.FuncDecl)
|
|
if !ok {
|
|
continue
|
|
}
|
|
walk(fd.Body, func(n interface{}) {
|
|
ce, ok := n.(*ast.CallExpr)
|
|
if !ok {
|
|
return
|
|
}
|
|
se, ok := ce.Fun.(*ast.SelectorExpr)
|
|
if !ok || len(ce.Args) != 1 {
|
|
return
|
|
}
|
|
switch se.Sel.String() {
|
|
case "JoinGroup", "LeaveGroup":
|
|
// c.JoinGroup(a) -> c.JoinGroup(nil, a)
|
|
// c.LeaveGroup(a) -> c.LeaveGroup(nil, a)
|
|
arg := ce.Args[0]
|
|
ce.Args = make([]ast.Expr, 2)
|
|
ce.Args[0] = ast.NewIdent("nil")
|
|
ce.Args[1] = arg
|
|
fixed = true
|
|
}
|
|
})
|
|
}
|
|
return fixed
|
|
}
|