Add files via upload

Author: arshiacomplus

core_engine/main.go

@@ -10,22 +10,21 @@ import (
 	"net/http"
 	"os"
 	"os/exec"
-	"runtime"
 	"strings"
 	"sync"
 	"time"
+
 	"golang.org/x/net/proxy"
 )
 
 type TestJob struct {
-	Tag            string          `json:"tag"`
-	Protocol       string          `json:"protocol"`
-	Config         json.RawMessage `json:"config,omitempty"`
-	ConfigURI      string          `json:"config_uri,omitempty"`
-	ListenIP       string          `json:"listen_ip"`
-	TestPort       int             `json:"test_port"`
-	ClientPath     string          `json:"client_path,omitempty"`
-	FragmentConfig json.RawMessage `json:"fragment_config,omitempty"`
+	Tag        string          `json:"tag"`
+	Protocol   string          `json:"protocol"`
+	Config     json.RawMessage `json:"config,omitempty"`
+	ConfigURI  string          `json:"config_uri,omitempty"`
+	ListenIP   string          `json:"listen_ip"`
+	TestPort   int             `json:"test_port"`
+	ClientPath string          `json:"client_path,omitempty"`
 }
 
 type TestResult struct {
@@ -47,13 +46,7 @@ func main() {
 		wg.Add(1)
 		go func(j TestJob) {
 			defer wg.Done()
-
-			if j.ClientPath != "" {
-				runIndividualTest(j, results)
-			} else {
-				ping, status := testProxy(j.ListenIP, j.TestPort)
-				results <- TestResult{Tag: j.Tag, Ping: ping, Status: status}
-			}
+			runTest(j, results)
 		}(job)
 	}
 
@@ -66,14 +59,20 @@ func main() {
 	fmt.Println(string(outputData))
 }
 
-func runIndividualTest(j TestJob, results chan<- TestResult) {
+func runTest(j TestJob, results chan<- TestResult) {
 	var cmd *exec.Cmd
-	configFile, err := os.CreateTemp("", "client-test-*.json")
-	if err != nil { results <- TestResult{Tag: j.Tag, Ping: -1, Status: "tempfile_error"}; return }
-	defer os.Remove(configFile.Name())
+	var configFile *os.File
+	var err error
+
+	// Create a context with a timeout for the entire test
+	ctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
+	defer cancel()
 
 	if j.Protocol == "hysteria2" {
-		// Simplified Hysteria config creation
+		configFile, err = os.CreateTemp("", "hysteria-*.json")
+		if err != nil { results <- TestResult{Tag: j.Tag, Ping: -1, Status: "tempfile_error"}; return }
+		defer os.Remove(configFile.Name())
+
 		uriParts := strings.Split(strings.Split(j.ConfigURI, "://")[1], "@")
 		serverParts := strings.Split(uriParts[1], "?")[0]
 		sni := strings.Split(strings.Split(j.ConfigURI, "sni=")[1], "#")[0]
@@ -85,30 +84,46 @@ func runIndividualTest(j TestJob, results chan<- TestResult) {
 		configBytes, _ := json.Marshal(config)
 		configFile.Write(configBytes)
 		configFile.Close()
-		cmd = exec.Command(j.ClientPath, "client", "-c", configFile.Name())
+		cmd = exec.CommandContext(ctx, j.ClientPath, "client", "-c", configFile.Name()) // ! Use CommandContext
 	} else {
-		// Logic for other individual clients can be added here
-		results <- TestResult{Tag: j.Tag, Ping: -1, Status: "unsupported_individual_client"}; return
+		// Default is Xray
+		configFile, err = os.CreateTemp("", "xray-*.json")
+		if err != nil { results <- TestResult{Tag: j.Tag, Ping: -1, Status: "tempfile_error"}; return }
+		defer os.Remove(configFile.Name())
+
+		fullConfig := map[string]interface{}{
+			"log":       map[string]string{"loglevel": "warning"},
+			"inbounds":  []map[string]interface{}{{"protocol": "socks", "port": j.TestPort, "listen": j.ListenIP, "settings": map[string]interface{}{"auth": "noauth", "udp": true}}},
+			"outbounds": []json.RawMessage{j.Config},
+		}
+		configBytes, _ := json.Marshal(fullConfig)
+		configFile.Write(configBytes)
+		configFile.Close()
+		cmd = exec.CommandContext(ctx, j.ClientPath, "-c", configFile.Name()) // ! Use CommandContext
 	}
 
-	ctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
-	defer cancel()
-	if runtime.GOOS == "windows" { setHideWindow(cmd) }
+	setHideWindow(cmd) // This function is platform-specific
 
 	var clientOutput bytes.Buffer
 	cmd.Stdout, cmd.Stderr = &clientOutput, &clientOutput
 
-	if err := cmd.Start(); err != nil { results <- TestResult{Tag: j.Tag, Ping: -1, Status: "client_start_failed"}; return }
-	time.Sleep(1200 * time.Millisecond) // Hysteria might take longer to start
+	if err := cmd.Start(); err != nil {
+		results <- TestResult{Tag: j.Tag, Ping: -1, Status: "client_start_failed"}
+		return
+	}
 
+	time.Sleep(900 * time.Millisecond)
 	ping, status := testProxy(j.ListenIP, j.TestPort)
+
 	if status != "success" {
 		logStr := strings.ReplaceAll(string(clientOutput.Bytes()), "\n", " ")
-		status = fmt.Sprintf("%s | log: %s", status, logStr[:200])
+		if len(logStr) > 200 { logStr = logStr[:200] }
+		status = fmt.Sprintf("%s | log: %s", status, logStr)
 	}
 
 	cmd.Process.Kill()
 	cmd.Wait()
+
 	results <- TestResult{Tag: j.Tag, Ping: ping, Status: status}
 }
 
@@ -117,26 +132,11 @@ func testProxy(listenIP string, port int) (int64, string) {
 	timeout := 8 * time.Second
 	dialer, err := proxy.SOCKS5("tcp", fmt.Sprintf("%s:%d", listenIP, port), nil, proxy.Direct)
 	if err != nil { return -1, fmt.Sprintf("failed_dialer: %v", err) }
-
-	httpClient := &http.Client{
-		Transport: &http.Transport{
-			DialContext: func(ctx context.Context, network, addr string) (net.Conn, error) {
-				return dialer.Dial(network, addr)
-			},
-		},
-		Timeout: timeout,
-	}
-
+	httpClient := &http.Client{ Transport: &http.Transport{DialContext: func(ctx context.Context, network, addr string) (net.Conn, error) { return dialer.Dial(network, addr) }}, Timeout: timeout}
 	start := time.Now()
 	resp, err := httpClient.Get(targetURL)
-	if err != nil {
-		return -1, fmt.Sprintf("failed_http: %v", err)
-	}
+	if err != nil { return -1, fmt.Sprintf("failed_http: %v", err) }
 	defer resp.Body.Close()
-
-	if resp.StatusCode != http.StatusNoContent {
-		return -1, fmt.Sprintf("bad_status_%d", resp.StatusCode)
-	}
-
+	if resp.StatusCode != http.StatusNoContent { return -1, fmt.Sprintf("bad_status_%d", resp.StatusCode) }
 	return time.Since(start).Milliseconds(), "success"
 }