mao-service-discovery 1.6.0-commit-2023-07-24-04-37-15.0 → 1.6.0-commit-2023-07-24-06-26-38.0

package/cmd/lib/Soap/soap-tplink-util.go

@@ -1,186 +0,0 @@
-package Soap
-
-import (
-	"bytes"
-	"errors"
-	"io/ioutil"
-	"log"
-	"net/http"
-	"strconv"
-	"strings"
-	"time"
-)
-
-const (
-	SOAP_HEADER_KEY = "SOAPAction"
-
-
-	/* ============================== ipc ============================== */
-	SOAP_URL_WANIPConnection   = "http://192.168.1.1:1900/ipc"
-
-	// ========== GetUptime ==========
-	SOAP_HEADER_VALUE_GetUptime 		= "urn:schemas-upnp-org:service:WANIPConnection:1#GetStatusInfo"
-	SOAP_MSG_GetUptime          		= "<s:Body><u:GetStatusInfo></u:GetStatusInfo></s:Body>"
-	SOAP_KEYWORD_GetUptime				= "NewUptime"
-
-	// ========== GetExternalIPAddress ==========
-	SOAP_HEADER_VALUE_GetExternalIPAddress	= "urn:schemas-upnp-org:service:WANCommonInterfaceConfig:1#GetExternalIPAddress"
-	SOAP_MSG_GetExternalIPAddress			= "<s:Body><u:GetExternalIPAddress></u:GetExternalIPAddress></s:Body>"
-	SOAP_KEYWORD_GetExternalIPAddress		= "NewExternalIPAddress"
-
-
-	/* ============================== ifc ============================== */
-	SOAP_URL_WANCommonInterfaceConfig   = "http://192.168.1.1:1900/ifc"
-
-	// ========== GetTotalBytesSent ==========
-	SOAP_HEADER_VALUE_GetTotalBytesSent = "urn:schemas-upnp-org:service:WANCommonInterfaceConfig:1#GetTotalBytesSent"
-	SOAP_MSG_GetTotalBytesSent          = "<s:Body><u:GetTotalBytesSent></u:GetTotalBytesSent></s:Body>"
-	SOAP_KEYWORD_GetTotalBytesSent		= "NewTotalBytesSent"
-
-	// ========== GetTotalBytesReceived ==========
-	SOAP_HEADER_VALUE_GetTotalBytesReceived = "urn:schemas-upnp-org:service:WANCommonInterfaceConfig:1#GetTotalBytesReceived"
-	SOAP_MSG_GetTotalBytesReceived          = "<s:Body><u:GetTotalBytesReceived></u:GetTotalBytesReceived></s:Body>"
-	SOAP_KEYWORD_GetTotalBytesReceived		= "NewTotalBytesReceived"
-
-	// ========== GetTotalPacketsSent ==========
-	SOAP_HEADER_VALUE_GetTotalPacketsSent = "urn:schemas-upnp-org:service:WANCommonInterfaceConfig:1#GetTotalPacketsSent"
-	SOAP_MSG_GetTotalPacketsSent          = "<s:Body><u:GetTotalPacketsSent></u:GetTotalPacketsSent></s:Body>"
-	SOAP_KEYWORD_GetTotalPacketsSent		= "NewTotalPacketsSent"
-
-	// ========== GetTotalPacketsReceived ==========
-	SOAP_HEADER_VALUE_GetTotalPacketsReceived = "urn:schemas-upnp-org:service:WANCommonInterfaceConfig:1#GetTotalPacketsReceived"
-	SOAP_MSG_GetTotalPacketsReceived          = "<s:Body><u:GetTotalPacketsReceived></u:GetTotalPacketsReceived></s:Body>"
-	SOAP_KEYWORD_GetTotalPacketsReceived		= "NewTotalPacketsReceived"
-)
-
-func requestSoapData(soapUrl, soapHeader, soapBody string) (*[]byte, error) {
-
-	postData := bytes.NewReader([]byte(soapBody))
-	req, err := http.NewRequest("POST", soapUrl, postData)
-	if err != nil {
-		return nil, err
-	}
-
-	req.Header.Set(SOAP_HEADER_KEY, soapHeader)
-
-	client := http.Client{
-		Timeout: 1 * time.Second,
-	}
-	resp, err := client.Do(req)
-	if err != nil {
-		return nil, err
-	}
-	defer resp.Body.Close()
-
-	body, err := ioutil.ReadAll(resp.Body)
-	if err != nil {
-		log.Println(err.Error())
-		return nil, err
-	}
-	return &body, nil
-}
-
-func getDataString(data string, keyword string) (string, error) {
-	_, after, found := strings.Cut(data, "<" + keyword + ">")
-	if !found {
-		return "", errors.New("Can't find the keyword at before: " + keyword)
-	}
-	before, _, found := strings.Cut(after, "</" + keyword + ">")
-	if !found {
-		return "", errors.New("Can't find the keyword at after: " + keyword)
-	}
-	return before, nil
-}
-
-func GetTotalBytesSent() (uint64, error) {
-	body, err := requestSoapData(SOAP_URL_WANCommonInterfaceConfig, SOAP_HEADER_VALUE_GetTotalBytesSent, SOAP_MSG_GetTotalBytesSent)
-	if err != nil {
-		return 0, err
-	}
-	data, err := getDataString(string(*body), SOAP_KEYWORD_GetTotalBytesSent)
-	if err != nil {
-		return 0, err
-	}
-	uintData, err := strconv.ParseUint(data, 10, 64)
-	if err != nil {
-		return 0, err
-	}
-	return uintData, nil
-}
-
-func GetTotalBytesReceived() (uint64, error) {
-	body, err := requestSoapData(SOAP_URL_WANCommonInterfaceConfig, SOAP_HEADER_VALUE_GetTotalBytesReceived, SOAP_MSG_GetTotalBytesReceived)
-	if err != nil {
-		return 0, err
-	}
-	data, err := getDataString(string(*body), SOAP_KEYWORD_GetTotalBytesReceived)
-	if err != nil {
-		return 0, err
-	}
-	uintData, err := strconv.ParseUint(data, 10, 64)
-	if err != nil {
-		return 0, err
-	}
-	return uintData, nil
-}
-
-func GetTotalPacketsSent() (uint64, error) {
-	body, err := requestSoapData(SOAP_URL_WANCommonInterfaceConfig, SOAP_HEADER_VALUE_GetTotalPacketsSent, SOAP_MSG_GetTotalPacketsSent)
-	if err != nil {
-		return 0, err
-	}
-	data, err := getDataString(string(*body), SOAP_KEYWORD_GetTotalPacketsSent)
-	if err != nil {
-		return 0, err
-	}
-	uintData, err := strconv.ParseUint(data, 10, 64)
-	if err != nil {
-		return 0, err
-	}
-	return uintData, nil
-}
-
-func GetTotalPacketsReceived() (uint64, error) {
-	body, err := requestSoapData(SOAP_URL_WANCommonInterfaceConfig, SOAP_HEADER_VALUE_GetTotalPacketsReceived, SOAP_MSG_GetTotalPacketsReceived)
-	if err != nil {
-		return 0, err
-	}
-	data, err := getDataString(string(*body), SOAP_KEYWORD_GetTotalPacketsReceived)
-	if err != nil {
-		return 0, err
-	}
-	uintData, err := strconv.ParseUint(data, 10, 64)
-	if err != nil {
-		return 0, err
-	}
-	return uintData, nil
-}
-
-func GetUptime() (uint64, error) {
-	body, err := requestSoapData(SOAP_URL_WANIPConnection, SOAP_HEADER_VALUE_GetUptime, SOAP_MSG_GetUptime)
-	if err != nil {
-		return 0, err
-	}
-	data, err := getDataString(string(*body), SOAP_KEYWORD_GetUptime)
-	if err != nil {
-		return 0, err
-	}
-	uintData, err := strconv.ParseUint(data[:len(data)-1], 10, 64)
-	if err != nil {
-		return 0, err
-	}
-	return uintData, nil
-}
-
-func GetExternalIPAddress() (string, error) {
-	body, err := requestSoapData(SOAP_URL_WANIPConnection, SOAP_HEADER_VALUE_GetExternalIPAddress, SOAP_MSG_GetExternalIPAddress)
-	if err != nil {
-		return "", err
-	}
-	data, err := getDataString(string(*body), SOAP_KEYWORD_GetExternalIPAddress)
-	if err != nil {
-		return "", err
-	}
-	return data, nil
-}
-

package/cmd/lib/Soap/soap-tplink-util_test.go

@@ -1,45 +0,0 @@
-package Soap
-
-import (
-	"log"
-	"reflect"
-	"runtime"
-	"testing"
-	"time"
-)
-
-func TestGetExternalIPAddress(t *testing.T) {
-	sss := runtime.FuncForPC(reflect.ValueOf(GetTotalBytesSent).Pointer()).Name()
-	log.Println(sss)
-
-	t1 := time.Now()
-	time.Sleep(2 * time.Second)
-	t2 := time.Now()
-	s := t2.Sub(t1).Seconds()
-	log.Println(s)
-
-	for {
-		break
-
-		totalBytesSent, err := GetTotalBytesSent()
-		log.Printf("TotalBytesSent: %d, err: %d", totalBytesSent, err)
-
-		totalBytesReceived, err := GetTotalBytesReceived()
-		log.Printf("TotalBytesReceived: %d, err: %d", totalBytesReceived, err)
-
-		totalPacketsSent, err := GetTotalPacketsSent()
-		log.Printf("TotalPacketsSent: %d, err: %d", totalPacketsSent, err)
-
-		totalPacketsReceived, err := GetTotalPacketsReceived()
-		log.Printf("TotalPacketsReceived: %d, err: %d", totalPacketsReceived, err)
-
-		uptime, err := GetUptime()
-		log.Printf("Uptime: %d, err: %d", uptime, err)
-
-		externalIPAddress, err := GetExternalIPAddress()
-		log.Printf("ExternalIPAddress: %s, err: %d", externalIPAddress, err)
-
-		time.Sleep(500 * time.Millisecond)
-		log.Printf("\n\n\n\n\n\n==============================")
-	}
-}

package/cmd/lib/Soap/tplink-gateway-module.go

@@ -1,200 +0,0 @@
-package Soap
-
-import (
-	MaoApi "MaoServerDiscovery/cmd/api"
-	"MaoServerDiscovery/cmd/lib/InfluxDB"
-	"MaoServerDiscovery/util"
-	influxdb2 "github.com/influxdata/influxdb-client-go/v2"
-	influxdb2Api "github.com/influxdata/influxdb-client-go/v2/api"
-	"time"
-)
-
-const (
-	p_TPLINK_MODULE_NAME = "TPLINK-Gateway-module"
-
-	FLAG_GATEWAY_BytesReceivedSpeed = 1 << 0
-	FLAG_GATEWAY_BytesSentSpeed = 1 << 1
-	FLAG_GATEWAY_PacketsReceivedSpeed = 1 << 2
-	FLAG_GATEWAY_PacketsSentSpeed = 1 << 3
-	FLAG_GATEWAY_Uptime = 1 << 4
-
-)
-
-type TplinkGatewayModule struct {
-
-	lastseen_BytesReceived uint64
-	lastseen_BytesReceived_timestamp time.Time
-
-	lastseen_BytesSent uint64
-	lastseen_BytesSent_timestamp time.Time
-
-	lastseen_PacketsReceived uint64
-	lastseen_PacketsReceived_timestamp time.Time
-
-	lastseen_PacketsSent uint64
-	lastseen_PacketsSent_timestamp time.Time
-
-	lastseen_Uptime_timestamp time.Time
-
-	BytesReceivedSpeed uint64
-	BytesSentSpeed uint64
-	PacketsReceivedSpeed uint64
-	PacketsSentSpeed uint64
-	Uptime uint64
-}
-
-func (t *TplinkGatewayModule) publishInfluxDB(writeAPI *influxdb2Api.WriteAPI, finishFlag uint) {
-	// write point asynchronously
-
-	if finishFlag & FLAG_GATEWAY_BytesReceivedSpeed != 0 {
-		(*writeAPI).WritePoint(
-			influxdb2.NewPointWithMeasurement(MaoApi.GATEWAY_MEASUREMENT).
-				AddTag(MaoApi.GATEWAY_TAG_GEO, "Beijing-HQ").
-				AddField(MaoApi.GATEWAY_FIELD_BytesReceivedSpeed, t.BytesReceivedSpeed).
-				AddField(MaoApi.GATEWAY_FIELD_BytesReceived, t.lastseen_BytesReceived).
-				SetTime(t.lastseen_BytesReceived_timestamp))
-	}
-	if finishFlag & FLAG_GATEWAY_BytesSentSpeed != 0 {
-		(*writeAPI).WritePoint(
-			influxdb2.NewPointWithMeasurement(MaoApi.GATEWAY_MEASUREMENT).
-				AddTag(MaoApi.GATEWAY_TAG_GEO, "Beijing-HQ").
-				AddField(MaoApi.GATEWAY_FIELD_BytesSentSpeed, t.BytesSentSpeed).
-				AddField(MaoApi.GATEWAY_FIELD_BytesSent, t.lastseen_BytesSent).
-				SetTime(t.lastseen_BytesSent_timestamp))
-	}
-	if finishFlag & FLAG_GATEWAY_PacketsReceivedSpeed != 0 {
-		(*writeAPI).WritePoint(
-			influxdb2.NewPointWithMeasurement(MaoApi.GATEWAY_MEASUREMENT).
-				AddTag(MaoApi.GATEWAY_TAG_GEO, "Beijing-HQ").
-				AddField(MaoApi.GATEWAY_FIELD_PacketsReceivedSpeed, t.PacketsReceivedSpeed).
-				AddField(MaoApi.GATEWAY_FIELD_PacketsReceived, t.lastseen_PacketsReceived).
-				SetTime(t.lastseen_PacketsReceived_timestamp))
-	}
-	if finishFlag & FLAG_GATEWAY_PacketsSentSpeed != 0 {
-		(*writeAPI).WritePoint(
-			influxdb2.NewPointWithMeasurement(MaoApi.GATEWAY_MEASUREMENT).
-				AddTag(MaoApi.GATEWAY_TAG_GEO, "Beijing-HQ").
-				AddField(MaoApi.GATEWAY_FIELD_PacketsSentSpeed, t.PacketsSentSpeed).
-				AddField(MaoApi.GATEWAY_FIELD_PacketsSent, t.lastseen_PacketsSent).
-				SetTime(t.lastseen_PacketsSent_timestamp))
-	}
-	if finishFlag & FLAG_GATEWAY_Uptime != 0 {
-		(*writeAPI).WritePoint(
-			influxdb2.NewPointWithMeasurement(MaoApi.GATEWAY_MEASUREMENT).
-				AddTag(MaoApi.GATEWAY_TAG_GEO, "Beijing-HQ").
-				AddField(MaoApi.GATEWAY_FIELD_Uptime, t.Uptime).
-				SetTime(t.lastseen_Uptime_timestamp))
-	}
-
-	// Not flush writes, avoid blocking my thread, then the lib's thread will block itself.
-	//(*writeAPI).Flush()
-}
-
-func (t *TplinkGatewayModule) pushLoop(triggerChannel *chan uint) {
-	var client *influxdb2.Client
-	var writeApi *influxdb2Api.WriteAPI
-
-	for {
-		// do...while, wait for being well configured.
-		client, writeApi = InfluxDB.CreateClientAndWriteAPI()
-		if writeApi != nil {
-			break
-		}
-		time.Sleep(1 * time.Second)
-	}
-	defer (*client).Close()
-
-	for finishFlag := range *triggerChannel {
-		t.publishInfluxDB(writeApi, finishFlag)
-	}
-}
-
-func (t *TplinkGatewayModule) controlLoop(triggerChannel *chan uint) {
-	for {
-		time.Sleep(2 * time.Second)
-		var finishFlag uint = 0
-
-		newBytesReceived, err := GetTotalBytesReceived()
-		newBytesReceived_timestamp := time.Now()
-		if err == nil {
-			if newBytesReceived >= t.lastseen_BytesReceived {
-				t.BytesReceivedSpeed = uint64(float64(newBytesReceived - t.lastseen_BytesReceived) / (newBytesReceived_timestamp.Sub(t.lastseen_BytesReceived_timestamp).Seconds()))
-			} else {
-				t.BytesReceivedSpeed = uint64(float64(newBytesReceived) / (newBytesReceived_timestamp.Sub(t.lastseen_BytesReceived_timestamp).Seconds())) // statistic may overflow (rollback to 0)
-			}
-			//log.Printf("%d, %f, %f", t.BytesReceivedSpeed, float64(newBytesReceived - t.lastseen_BytesReceived), newBytesReceived_timestamp.Sub(t.lastseen_BytesReceived_timestamp).Seconds())
-			t.lastseen_BytesReceived = newBytesReceived // statistic may overflow (rollback to 0)
-			t.lastseen_BytesReceived_timestamp = newBytesReceived_timestamp
-			finishFlag |= FLAG_GATEWAY_BytesReceivedSpeed
-		} else {
-			util.MaoLogM(util.WARN, p_TPLINK_MODULE_NAME, "Fail to get TotalBytesReceived, %s", err.Error())
-		}
-
-		newBytesSent, err := GetTotalBytesSent()
-		newBytesSent_timestamp := time.Now()
-		if err == nil {
-			if newBytesSent >= t.lastseen_BytesSent {
-				t.BytesSentSpeed = uint64(float64(newBytesSent - t.lastseen_BytesSent) / (newBytesSent_timestamp.Sub(t.lastseen_BytesSent_timestamp).Seconds()))
-			} else {
-				t.BytesSentSpeed = uint64(float64(newBytesSent) / (newBytesSent_timestamp.Sub(t.lastseen_BytesSent_timestamp).Seconds())) // statistic may overflow (rollback to 0)
-			}
-			t.lastseen_BytesSent = newBytesSent // statistic may overflow (rollback to 0)
-			t.lastseen_BytesSent_timestamp = newBytesSent_timestamp
-			finishFlag |= FLAG_GATEWAY_BytesSentSpeed
-		} else {
-			util.MaoLogM(util.WARN, p_TPLINK_MODULE_NAME, "Fail to get TotalBytesSent, %s", err.Error())
-		}
-
-		newPacketsReceived, err := GetTotalPacketsReceived()
-		newPacketsReceived_timestamp := time.Now()
-		if err == nil {
-			if newPacketsReceived >= t.lastseen_PacketsReceived {
-				t.PacketsReceivedSpeed = uint64(float64(newPacketsReceived - t.lastseen_PacketsReceived) / (newPacketsReceived_timestamp.Sub(t.lastseen_PacketsReceived_timestamp).Seconds()))
-			} else {
-				t.PacketsReceivedSpeed = uint64(float64(newPacketsReceived) / (newPacketsReceived_timestamp.Sub(t.lastseen_PacketsReceived_timestamp).Seconds())) // statistic may overflow (rollback to 0)
-			}
-			t.lastseen_PacketsReceived = newPacketsReceived // statistic may overflow (rollback to 0)
-			t.lastseen_PacketsReceived_timestamp = newPacketsReceived_timestamp
-			finishFlag |= FLAG_GATEWAY_PacketsReceivedSpeed
-		} else {
-			util.MaoLogM(util.WARN, p_TPLINK_MODULE_NAME, "Fail to get TotalPacketsReceived, %s", err.Error())
-		}
-
-		newPacketsSent, err := GetTotalPacketsSent()
-		newPacketsSent_timestamp := time.Now()
-		if err == nil {
-			if newPacketsSent >= t.lastseen_PacketsSent {
-				t.PacketsSentSpeed = uint64(float64(newPacketsSent - t.lastseen_PacketsSent) / (newPacketsSent_timestamp.Sub(t.lastseen_PacketsSent_timestamp).Seconds()))
-			} else {
-				t.PacketsSentSpeed = uint64(float64(newPacketsSent) / (newPacketsSent_timestamp.Sub(t.lastseen_PacketsSent_timestamp).Seconds())) // statistic may overflow (rollback to 0)
-			}
-			t.lastseen_PacketsSent = newPacketsSent // statistic may overflow (rollback to 0)
-			t.lastseen_PacketsSent_timestamp = newPacketsSent_timestamp
-			finishFlag |= FLAG_GATEWAY_PacketsSentSpeed
-		} else {
-			util.MaoLogM(util.WARN, p_TPLINK_MODULE_NAME, "Fail to get TotalPacketsSent, %s", err.Error())
-		}
-
-		newUptime, err := GetUptime()
-		newUptime_timestamp := time.Now()
-		if err == nil {
-			t.Uptime = newUptime
-			t.lastseen_Uptime_timestamp = newUptime_timestamp
-			finishFlag |= FLAG_GATEWAY_Uptime
-		} else {
-			util.MaoLogM(util.WARN, p_TPLINK_MODULE_NAME, "Fail to get Uptime, %s", err.Error())
-		}
-
-		*triggerChannel <- finishFlag
-
-		util.MaoLogM(util.DEBUG, p_TPLINK_MODULE_NAME, "BytesSentSpeed: %d, BytesReceivedSpeed: %d, PacketsSentSpeed: %d, PacketsReceivedSpeed: %d, Uptime: %d",
-			t.BytesSentSpeed, t.BytesReceivedSpeed, t.PacketsSentSpeed, t.PacketsReceivedSpeed, t.Uptime)
-	}
-}
-
-func (t *TplinkGatewayModule) InitTplinkGatewayModule() bool {
-	triggerChannel := make(chan uint, 100)
-	go t.controlLoop(&triggerChannel)
-	go t.pushLoop(&triggerChannel)
-	return true
-}