msb-public
/
go-fly
mirror of https://github.com/taoshihan1991/go-fly.git
1
0
Code Issues Projects Releases Wiki Activity

增加验证邮箱密码和cookie存储判断

Browse Source
pull/30/head
unknown 5 years ago
parent 8ee456bf46
commit c105ef7f1f
5 changed files with 89 additions and 24 deletions
Show Stats Download Patch File Download Diff File

BIN
imaptool.exe
View File

Binary file not shown.

10
imaptool.go
Unescape View File

@ -104,11 +104,11 @@ func main() {
log.Println("读取最新的几封信(all全部):") log.Println("读取最新的几封信(all全部):")
inLine = readLineFromInput() inLine = readLineFromInput()
var maxNum uint32 var maxNum uint32
if inLine=="all"{ if inLine == "all" {
maxNum=mbox.Messages maxNum = mbox.Messages
}else { } else {
tempNum, _ := strconv.Atoi(inLine) tempNum, _ := strconv.Atoi(inLine)
maxNum=uint32(tempNum) maxNum = uint32(tempNum)
} }
from := uint32(1) from := uint32(1)
@ -130,7 +130,7 @@ func main() {
log.Printf("最新的 %d 封信:", maxNum) log.Printf("最新的 %d 封信:", maxNum)
for msg := range messages { for msg := range messages {
log.Printf("* %d:%s\n" ,to,msg.Envelope.Subject) log.Printf("* %d:%s\n", to, msg.Envelope.Subject)
to-- to--
} }

54
server.go
Unescape View File

@ -1,42 +1,68 @@
package main package main
import ( import (
"fmt"
"github.com/taoshihan1991/imaptool/tools"
"html/template" "html/template"
"log" "log"
"net/http" "net/http"
) )
func main(){ func main() {
log.Println("listen on 8080...") log.Println("listen on 8080...")
http.HandleFunc("/",index) http.HandleFunc("/", index)
//登陆界面 //登陆界面
http.HandleFunc("/login",login) http.HandleFunc("/login", login)
//监听端口 //监听端口
http.ListenAndServe(":8080", nil) http.ListenAndServe(":8080", nil)
} }
//输出首页 //输出首页
func index(w http.ResponseWriter, r *http.Request){ func index(w http.ResponseWriter, r *http.Request) {
auth:=getCookie(r,"auth") auth := getCookie(r, "auth")
if auth=="" { if auth == "" {
http.Redirect(w, r, "/login", 302) http.Redirect(w, r, "/login", 302)
} }
t,_:=template.ParseFiles("./tmpl/index.html") t, _ := template.ParseFiles("./tmpl/index.html")
t.Execute(w, nil) t.Execute(w, nil)
} }
//登陆界面 //登陆界面
func login(w http.ResponseWriter, r *http.Request){ func login(w http.ResponseWriter, r *http.Request) {
t,_:=template.ParseFiles("./tmpl/login.html") email := r.PostFormValue("email")
t.Execute(w, nil) server := r.PostFormValue("server")
password := r.PostFormValue("password")
var errStr string
if email != "" && server != "" && password != "" {
res := tools.CheckEmailPassword(server, email, password)
if !res {
errStr = "连接或验证失败"
t, _ := template.ParseFiles("./tmpl/login.html")
t.Execute(w, errStr)
} else {
auth := fmt.Sprintf("%s:%s:%s", server, email, password)
cookie := http.Cookie{
Name: "auth",
Value: auth,
}
http.SetCookie(w, &cookie)
http.Redirect(w, r, "/", 302)
}
} else {
t, _ := template.ParseFiles("./tmpl/login.html")
t.Execute(w, nil)
}
} }
//加密cookie //加密cookie
//func authCookie(){ //func authCookie(){
// //
//} //}
//获取cookie //获取cookie
func getCookie(r *http.Request,name string)string{ func getCookie(r *http.Request, name string) string {
cookies:=r.Cookies() cookies := r.Cookies()
for _,cookie:=range cookies{ for _, cookie := range cookies {
if cookie.Name==name{ if cookie.Name == name {
return cookie.Value return cookie.Value
} }
} }

10
tmpl/login.html
Unescape View File

@ -73,15 +73,15 @@
</head> </head>
<body class="text-center"> <body class="text-center">
<form class="form-signin"> <form class="form-signin" action="/login" method="post">
<h1 class="h3 mb-3 font-weight-normal">邮箱IMAP工具</h1> <h1 class="h3 mb-3 font-weight-normal">邮箱IMAP工具</h1>
<label for="inputServer" class="sr-only">IMAP服务器:</label> <label for="inputServer" class="sr-only">IMAP服务器:</label>
<input type="email" id="inputServer" class="form-control" placeholder="IMAP服务器" required autofocus> <input type="text" name="server" id="inputServer" class="form-control" placeholder="IMAP服务器" required autofocus>
<label for="inputEmail" class="sr-only">邮箱地址:</label> <label for="inputEmail" class="sr-only">邮箱地址:</label>
<input type="email" id="inputEmail" class="form-control" placeholder="邮箱地址" required autofocus> <input type="email" id="inputEmail" name="email" class="form-control" placeholder="邮箱地址" required autofocus>
<label for="inputPassword" class="sr-only">密码:</label> <label for="inputPassword" class="sr-only">密码:</label>
<input type="password" id="inputPassword" class="form-control" placeholder="密码" required> <input type="password" name="password" id="inputPassword" class="form-control" placeholder="密码" required>
{{if .}}<div><code>{{.}}</code></div>{{end}}
<button class="btn btn-lg btn-primary btn-block" type="submit">登陆</button> <button class="btn btn-lg btn-primary btn-block" type="submit">登陆</button>
<p class="mt-5 mb-3 text-muted">&copy; 2020</p> <p class="mt-5 mb-3 text-muted">&copy; 2020</p>
</form> </form>

39
tools/imap.go
Unescape View File

@ -0,0 +1,39 @@
package tools
import (
"fmt"
"github.com/emersion/go-imap/client"
"strconv"
"strings"
)
func CheckEmailPassword(server string, email string, password string) bool {
if !strings.Contains(server, ":") {
return false
}
var c *client.Client
var err error
serverSlice := strings.Split(server, ":")
uri := serverSlice[0]
port, _ := strconv.Atoi(serverSlice[1])
if port != 993 && port != 143 {
return false
}
if port == 993 {
c, err = client.DialTLS(fmt.Sprintf("%s:%d", uri, port), nil)
} else {
c, err = client.Dial(fmt.Sprintf("%s:%d", uri, port))
}
if err != nil {
return false
}
// 不要忘了退出
defer c.Logout()
// 登陆
if err := c.Login(email, password); err != nil {
return false
}
return true
}
Write Preview
Loading…
Cancel
Save