Go语言黑帽子学习1

代码清单2-4 端口扫描器执行速度太快

package main

import (

"fmt"

"net"

)

func main() {

for i := 0; i <= 1024; i++ {

go func(j int) {

address := fmt.Sprintf("172.23.190.244:%d", j)

conn, err := net.Dial("tcp", address)

if err != nil {

return

}

conn.Close()

fmt.Printf("%d open\n", j)

}(i)

}

}

代码清单2-5 使用WaitGroup进行同步扫描的端口扫描器

package main

import (

    "fmt"

    "net"

    "sync"

)

func main() {

    var wg sync.WaitGroup

    for i := 1; i <= 1024; i++ {

        wg.Add(1)

        go func(j int) {

            defer wg.Done()

            address := fmt.Sprintf("172.23.190.244:%d", j)

            conn, err := net.Dial("tcp", address)

            if err != nil {

                return

            }

            conn.Close()

            fmt.Printf("%d open\n", j)

        }(i)

    }

}