CasaOS/pkg/upnp/device.go

package upnp

import (
	"encoding/xml"
	"io/ioutil"
	"net/http"
	"strings"
)

func GetCtrlUrl(host, device string) string {
	request := ctrlUrlRequest(host, device)
	response, _ := http.DefaultClient.Do(request)
	resultBody, _ := ioutil.ReadAll(response.Body)
	defer response.Body.Close()
	if response.StatusCode == 200 {
		return resolve(string(resultBody))

	}
	return ""
}

func ctrlUrlRequest(host string, deviceDescUrl string) *http.Request {
	//请求头
	header := http.Header{}
	header.Set("Accept", "text/html, image/gif, image/jpeg, *; q=.2, */*; q=.2")
	header.Set("User-Agent", "preston")
	header.Set("Host", host)
	header.Set("Connection", "keep-alive")
	request, _ := http.NewRequest("GET", "http://"+host+deviceDescUrl, nil)
	request.Header = header
	return request
}

func resolve(resultStr string) string {
	inputReader := strings.NewReader(resultStr)

	// 从文件读取，如可以如下：
	// content, err := ioutil.ReadFile("studygolang.xml")
	// decoder := xml.NewDecoder(bytes.NewBuffer(content))

	lastLabel := ""

	ISUpnpServer := false

	IScontrolURL := false
	var controlURL string //`controlURL`
	// var eventSubURL string //`eventSubURL`
	// var SCPDURL string     //`SCPDURL`

	decoder := xml.NewDecoder(inputReader)
	for t, err := decoder.Token(); err == nil && !IScontrolURL; t, err = decoder.Token() {
		switch token := t.(type) {
		// 处理元素开始（标签）
		case xml.StartElement:
			if ISUpnpServer {
				name := token.Name.Local
				lastLabel = name
			}

		// 处理元素结束（标签）
		case xml.EndElement:
			// log.Println("结束标记：", token.Name.Local)
		// 处理字符数据（这里就是元素的文本）
		case xml.CharData:
			//得到url后其他标记就不处理了
			content := string([]byte(token))
			//找到提供端口映射的服务
			if content == "urn:schemas-upnp-org:service:WANIPConnection:1" {
				ISUpnpServer = true
				continue
			}

			if ISUpnpServer {
				switch lastLabel {
				case "controlURL":
					controlURL = content
					IScontrolURL = true
				case "eventSubURL":
					// eventSubURL = content
				case "SCPDURL":
					// SCPDURL = content
				}
			}
		default:
			// ...
		}
	}
	return controlURL
}