摘要：作者簡介地址地址本章介紹代碼網(wǎng)絡(luò)中端口映射作者使用操作系統(tǒng)，其他平臺也大同小異介紹通用即插即用。端口映射將一個外部端口映射到一個內(nèi)網(wǎng)。

作者：Derek

Github地址：https://github.com/Bytom/bytom

Gitee地址：https://gitee.com/BytomBlockc...

本章介紹bytom代碼P2P網(wǎng)絡(luò)中upnp端口映射

作者使用MacOS操作系統(tǒng)，其他平臺也大同小異
Golang Version: 1.8

UPNP（Universal Plug and Play）通用即插即用。UPNP端口映射將一個外部端口映射到一個內(nèi)網(wǎng)ip:port。從而實現(xiàn)p2p網(wǎng)絡(luò)從外網(wǎng)能夠穿透網(wǎng)關(guān)訪問到內(nèi)網(wǎng)的bytomd節(jié)點。

SSDP（Simple Service Discovery Protocol 簡單服務(wù)發(fā)現(xiàn)協(xié)議）
GENA（Generic Event Notification Architecture 通用事件通知結(jié)構(gòu)）
SOAP（Simple Object Access Protocol 簡單對象訪問協(xié)議）
XML（Extensible Markup Language 可擴張標(biāo)記語言）

p2p/upnp/upnp.go

從網(wǎng)絡(luò)中發(fā)現(xiàn)支持UPNP功能的設(shè)備，并得到該設(shè)備的location和url等相關(guān)信息

type upnpNAT struct {
    serviceURL string // 設(shè)備的描述文件URL，用于得到該設(shè)備的描述信息
    ourIP      string // 節(jié)點本地ip地址
    urnDomain  string // 設(shè)備類型
}

func Discover() (nat NAT, err error) {
    ssdp, err := net.ResolveUDPAddr("udp4", "239.255.255.250:1900")
    if err != nil {
        return
    }
    conn, err := net.ListenPacket("udp4", ":0")
    if err != nil {
        return
    }
    socket := conn.(*net.UDPConn)
    defer socket.Close()

    err = socket.SetDeadline(time.Now().Add(3 * time.Second))
    if err != nil {
        return
    }

    st := "InternetGatewayDevice:1"

    // 多播請求：M-SEARCH SSDP協(xié)議定義的發(fā)現(xiàn)請求。
    buf := bytes.NewBufferString(
        "M-SEARCH * HTTP/1.1
" +
            "HOST: 239.255.255.250:1900
" +
            "ST: ssdp:all
" +
            "MAN: "ssdp:discover"
" +
            "MX: 2

")
    message := buf.Bytes()
    answerBytes := make([]byte, 1024)
    for i := 0; i < 3; i++ {
        // 向239.255.255.250:1900發(fā)送一條多播請求
        _, err = socket.WriteToUDP(message, ssdp)
        if err != nil {
            return
        }
        // 如果從網(wǎng)絡(luò)中發(fā)現(xiàn)UPNP設(shè)備則會從239.255.255.250:1900收到響應(yīng)消息
        var n int
        n, _, err = socket.ReadFromUDP(answerBytes)
        for {
            n, _, err = socket.ReadFromUDP(answerBytes)
            if err != nil {
                break
            }
            answer := string(answerBytes[0:n])
            if strings.Index(answer, st) < 0 {
                continue
            }
            // HTTP header field names are case-insensitive.
            // http://www.w3.org/Protocols/rfc2616/rfc2616-sec4.html#sec4.2
            // 獲得設(shè)備location
            locString := "
location:"
            answer = strings.ToLower(answer)
            locIndex := strings.Index(answer, locString)
            if locIndex < 0 {
                continue
            }
            loc := answer[locIndex+len(locString):]
            endIndex := strings.Index(loc, "
")
            if endIndex < 0 {
                continue
            }
            // 獲得設(shè)備的描述url和設(shè)備類型
            locURL := strings.TrimSpace(loc[0:endIndex])
            var serviceURL, urnDomain string
            serviceURL, urnDomain, err = getServiceURL(locURL)
            if err != nil {
                return
            }
            var ourIP net.IP
            ourIP, err = localIPv4()
            if err != nil {
                return
            }
            nat = &upnpNAT{serviceURL: serviceURL, ourIP: ourIP.String(), urnDomain: urnDomain}
            return
        }
    }
    err = errors.New("UPnP port discovery failed.")
    return
}

向upnp設(shè)備發(fā)送一條http post請求，將內(nèi)部網(wǎng)絡(luò)ip:port和外部網(wǎng)絡(luò)ip:port做映射

func (n *upnpNAT) AddPortMapping(protocol string, externalPort, internalPort int, description string, timeout int) (mappedExternalPort int, err error) {
    // A single concatenation would break ARM compilation.
    message := "
" +
        "" + strconv.Itoa(externalPort)
    message += "" + protocol + ""
    message += "" + strconv.Itoa(internalPort) + "" +
        "" + n.ourIP + "" +
        "1"
    message += description +
        "" + strconv.Itoa(timeout) +
        ""

    var response *http.Response
    response, err = soapRequest(n.serviceURL, "AddPortMapping", message, n.urnDomain)
    if response != nil {
        defer response.Body.Close()
    }
    if err != nil {
        return
    }

    // TODO: check response to see if the port was forwarded
    // log.Println(message, response)
    // JAE:
    // body, err := ioutil.ReadAll(response.Body)
    // fmt.Println(string(body), err)
    mappedExternalPort = externalPort
    _ = response
    return
}

向upnp設(shè)備發(fā)送一條http post請求，將內(nèi)部網(wǎng)絡(luò)ip:port和外部網(wǎng)絡(luò)ip:port刪除映射關(guān)系

func (n *upnpNAT) DeletePortMapping(protocol string, externalPort, internalPort int) (err error) {

    message := "
" +
        "" + strconv.Itoa(externalPort) +
        "" + protocol + "" +
        ""

    var response *http.Response
    response, err = soapRequest(n.serviceURL, "DeletePortMapping", message, n.urnDomain)
    if response != nil {
        defer response.Body.Close()
    }
    if err != nil {
        return
    }

    // TODO: check response to see if the port was deleted
    // log.Println(message, response)
    _ = response
    return
}

func (n *upnpNAT) GetExternalAddress() (addr net.IP, err error) {
    info, err := n.getExternalIPAddress()
    if err != nil {
        return
    }
    addr = net.ParseIP(info.externalIpAddress)
    return
}

func (n *upnpNAT) getExternalIPAddress() (info statusInfo, err error) {

    message := "
" +
        ""

    var response *http.Response
    response, err = soapRequest(n.serviceURL, "GetExternalIPAddress", message, n.urnDomain)
    if response != nil {
        defer response.Body.Close()
    }
    if err != nil {
        return
    }
    var envelope Envelope
    data, err := ioutil.ReadAll(response.Body)
    reader := bytes.NewReader(data)
    xml.NewDecoder(reader).Decode(&envelope)

    info = statusInfo{envelope.Soap.ExternalIP.IPAddress}

    if err != nil {
        return
    }

    return
}