(Fixed #5) bug fix

This commit is contained in:
hahwul 2020-04-11 02:28:57 +09:00
parent 6530777f27
commit 475930cdf7
2 changed files with 9 additions and 1 deletions

BIN
add-tool

Binary file not shown.

View File

@ -99,7 +99,15 @@ func main() {
defer resp.Body.Close() defer resp.Body.Close()
if title, ok := GetHtmlTitle(resp.Body); ok { if title, ok := GetHtmlTitle(resp.Body); ok {
if strings.Contains(string(title), ": ") {
desc = strings.Split(string(title), ": ")[1] desc = strings.Split(string(title), ": ")[1]
} else {
reader1 := bufio.NewReader(os.Stdin)
fmt.Println("[+] Don't have descriptions / input description")
a, _ := reader1.ReadString('\n')
desc = strings.TrimRight(a, "\r\n")
}
} else { } else {
println("Fail to get HTML title") println("Fail to get HTML title")
} }