118 lines
3.0 KiB
Go
118 lines
3.0 KiB
Go
|
package Apiform
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"encoding/json"
|
|||
|
|
"errors"
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/garyburd/redigo/redis"
|
|||
|
|
"github.com/satori/go.uuid"
|
|||
|
|
"log"
|
|||
|
|
"ssh_manage/common"
|
|||
|
|
"ssh_manage/database"
|
|||
|
|
"ssh_manage/model"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type Login struct {
|
|||
|
|
Phone int `form:"phone" binding:"required"`
|
|||
|
|
Code string `form:"code" binding:"required"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type Send struct {
|
|||
|
|
Phone string `form:"phone" binding:"required"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type Slist struct {
|
|||
|
|
Page int `form:"page" binding:"required"`
|
|||
|
|
Limit int `form:"limit" binding:"required"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type List_resp struct {
|
|||
|
|
List []model.Server
|
|||
|
|
Count uint
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type GetTerm struct {
|
|||
|
|
ID uint `form:"id" binding:"required"`
|
|||
|
|
Password string `form:"setpass" binding:"required"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type WsAuth struct {
|
|||
|
|
Sid string `uri:"sid" binding:"required,uuid"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type Edit struct {
|
|||
|
|
ID uint `form:"id" binding:"required"`
|
|||
|
|
Nickname string `form:"nickname"`
|
|||
|
|
Ip string `form:"ip"`
|
|||
|
|
Port int `form:"port"`
|
|||
|
|
Username string `form:"username"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type EditPass struct {
|
|||
|
|
ID uint `form:"id" binding:"required"`
|
|||
|
|
Password string `form:"password" binding:"required"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type Addser struct {
|
|||
|
|
Nickname string `form:"nickname"`
|
|||
|
|
Ip string `form:"ip" binding:"required"`
|
|||
|
|
Port int `form:"port" binding:"required"`
|
|||
|
|
Username string `form:"username" binding:"required"`
|
|||
|
|
Password string `form:"password" binding:"required"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type SerInfo struct {
|
|||
|
|
ID uint
|
|||
|
|
Ip string
|
|||
|
|
Port int
|
|||
|
|
Username string
|
|||
|
|
Password string
|
|||
|
|
BindUser uint
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *Send) SendCaptcha(ip string) (err error) {
|
|||
|
|
cache := database.Cache.Get()
|
|||
|
|
defer cache.Close()
|
|||
|
|
ipexists, _ := redis.Bool(cache.Do("EXISTS", ip))
|
|||
|
|
phoneexists, _ := redis.Bool(cache.Do("EXISTS", s.Phone+"_time"))
|
|||
|
|
if ipexists || phoneexists {
|
|||
|
|
err = errors.New("请勿频繁发送验证码")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
cache.Send("MULTI") //开启事务操作
|
|||
|
|
cache.Send("SETEX", s.Phone+"_time", 60*2, nil) //记录手机号与IP，防止重复发送
|
|||
|
|
cache.Send("SETEX", ip, 60*2, nil)
|
|||
|
|
capcha, err := common.Sendsms(s.Phone)
|
|||
|
|
if err != nil {
|
|||
|
|
cache.Do("DISCARD") //发送失败则取消事务
|
|||
|
|
log.Println(err.Error())
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
cache.Send("SETEX", s.Phone, 60*5, capcha) //延长过期时间，用于校验
|
|||
|
|
cache.Do("EXEC") //提交事务
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (l *Login) Verify() (key, code string) {
|
|||
|
|
return fmt.Sprintf("%d", l.Phone), l.Code
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (t *GetTerm) Decode(server model.Server) (sid string, err error) {
|
|||
|
|
sid = uuid.Must(uuid.NewV4(), nil).String()
|
|||
|
|
//log.Println(server)
|
|||
|
|
s_pass := common.AesDecryptCBC(server.Password, []byte(t.Password))
|
|||
|
|
if s_pass == "" {
|
|||
|
|
return "", errors.New("秘钥验证失败")
|
|||
|
|
} else {
|
|||
|
|
var serinfo = SerInfo{server.ID,server.Ip,server.Port,server.Username,s_pass,server.BindUser}
|
|||
|
|
//server.Password = s_pass //用于建立连接
|
|||
|
|
cache := database.Cache.Get()
|
|||
|
|
defer cache.Close()
|
|||
|
|
s_info, _ := json.Marshal(serinfo)
|
|||
|
|
//log.Println(string(s_info))
|
|||
|
|
cache.Do("SETEX", sid, 10, s_info) //缓存10s，用于建立连接和验证权限
|
|||
|
|
//log.Println(sid)
|
|||
|
|
}
|
|||
|
|
return sid, nil
|
|||
|
|
}
|