main.go

package main

import (
	"crypto/md5"
	"flag"
	"fmt"
	"github.com/gin-gonic/gin"
	"os"
	"strconv"
	"time"
)

var (
	VERSION = "1.6.58"
)

var (
	ConfigSource string
	WebMode      string
	WebPort      int
	WebEnable    bool
	ApiKey       string
)

var (
	vkBakDict = make(map[string]int64)
)

func MD5(str string) string {
	data := []byte(str)
	has := md5.Sum(data)
	md5str := fmt.Sprintf("%x", has)
	return md5str
}

func getEnvString(name string, value string) string {
	ret := os.Getenv(name)
	if ret == "" {
		return value
	} else {
		return ret
	}
}

func getEnvInt(name string, value int) int {
	env := os.Getenv(name)
	if ret, err := strconv.Atoi(env); env == "" || err != nil {
		return value
	} else {
		return ret
	}
}

func init() {
	flag.StringVar(&ConfigSource, "c", "default", "config source default or env.")
	flag.StringVar(&WebMode, "mode", gin.ReleaseMode, "wol web mode: debug, release, test.")
	flag.IntVar(&WebPort, "port", 9090, "wol web port: 0-65535")
	flag.BoolVar(&WebEnable, "web", true, "wol web page switch: true or false.")
	flag.StringVar(&ApiKey, "key", "false", "wol web api key, key length greater than 6.")
}

func main() {
	flag.Parse()

	fmt.Printf("Start Run WolGoWeb...\n\n")
	fmt.Printf("Version: %s\n\n", VERSION)

	if ConfigSource == "env" {
		WebMode = getEnvString("MODE", WebMode)
		WebPort = getEnvInt("PORT", WebPort)
		WebEnable = getEnvString("WEB", strconv.FormatBool(WebEnable)) == "true"
		ApiKey = getEnvString("KEY", ApiKey)
	}

	gin.SetMode(WebMode)

	r := gin.Default()

	if WebEnable {
		r.GET("/", GetIndex)
		r.GET("/index", GetIndex)
	}
	r.GET("/wol", GetWol)

	fmt.Printf("WolGoWeb Runing [port:%d, key:%s, web:%s]\n", WebPort, ApiKey, strconv.FormatBool(WebEnable))

	r.Run(fmt.Sprintf(":%d", WebPort))
}

func GetIndex(c *gin.Context) {
	c.String(200, `
WOL唤醒工具

API: %s/wol

Params:
  mac  : 需要唤醒的MAC地址（必须）,
  ip   : 指定IP地址（默认：255.255.255.255）,
  port : 唤醒端口（默认：9）,
  time : 请求时间戳（配合授权验证使用）,
  token: 授权Token = MD5(key + mac + time)（必须存在key的情况下才有效，否则忽略。）,

Version: %s
`, c.Request.Host, VERSION)
}

func VerifyAuth(key string, mac string, vk int64, token string) (int, string) {
	err := 0
	message := "OK"
	if len(key) >= 6 {
		timeUnix := time.Now().Unix()
		fmt.Printf("now=%d, vk=%d\n", timeUnix, vk)
		if len(token) != 32 {
			err = 101
			message = "No authority."
		} else if timeUnix-vk > 30 || vk-timeUnix > 1 {
			err = 102
			message = "The value of Time is no longer in the valid range."
		} else if bakVK, ok := vkBakDict[mac]; ok && bakVK == vk {
			err = 103
			message = "Time value repetition."
		} else if MD5(ApiKey+mac+fmt.Sprintf("%d", vk)) != token {
			err = 104
			message = "No authority token."
		} else {
			vkBakDict[mac] = vk
		}
	}
	return err, message
}

func GetWol(c *gin.Context) {
	mac := c.Query("mac")
	ip := c.DefaultQuery("ip", "255.255.255.255")
	port := c.DefaultQuery("port", "9")
	token := c.DefaultQuery("token", "")
	vk, _ := strconv.ParseInt(c.DefaultQuery("time", "0"), 10, 64)
	if errAuth, messageAuth := VerifyAuth(ApiKey, mac, vk, token); errAuth == 0 {
		err := Wake(mac, ip, port)
		if err != nil {
			c.JSON(200, gin.H{
				"error":   100,
				"message": fmt.Sprintf("%s", err),
			})
		} else {
			c.JSON(200, gin.H{
				"error":   0,
				"message": fmt.Sprintf("Wake Success Mac:%s", mac),
			})
		}
	} else {
		c.JSON(200, gin.H{
			"error":   errAuth,
			"message": messageAuth,
		})
	}

}