fix: get was messed up

core/internal/cc/server/dispatcher.go

@@ -18,9 +18,10 @@ import (
 func apiDispatcher(wrt http.ResponseWriter, req *http.Request) {
 	vars := mux.Vars(req)
 	// Setup H2Conn for reverse shell and proxy.
-	var rshellConn, proxyConn def.H2Conn
-	network.RShellStream.H2x = &rshellConn
-	network.ProxyStream.H2x = &proxyConn
+	rshellConn := new(def.H2Conn)
+	proxyConn := new(def.H2Conn)
+	network.RShellStream.H2x = rshellConn
+	network.ProxyStream.H2x = proxyConn
 
 	if vars["api"] == "" || vars["token"] == "" {
 		logging.Debugf("Invalid request: %v, missing api/token", req)
@@ -58,7 +59,7 @@ func apiDispatcher(wrt http.ResponseWriter, req *http.Request) {
 	case transport.FTPAPI:
 		for _, sh := range network.FTPStreams {
 			if token == sh.Token {
-				handleFTPTransfer(wrt, req)
+				handleFTPTransfer(sh, wrt, req)
 				return
 			}
 		}
@@ -77,7 +78,7 @@ func apiDispatcher(wrt http.ResponseWriter, req *http.Request) {
 		}
 		http.ServeFile(wrt, req, local_path)
 	case transport.ProxyAPI:
-		handlePortForwarding(wrt, req)
+		handlePortForwarding(network.ProxyStream, wrt, req)
 	default:
 		wrt.WriteHeader(http.StatusBadRequest)
 	}

core/internal/cc/server/handler_ftp.go

@@ -49,17 +49,11 @@ func progressMonitor(bar *progressbar.ProgressBar, filewrite, targetFile string,
 }
 
 // handleFTPTransfer processes file transfer requests.
-func handleFTPTransfer(wrt http.ResponseWriter, req *http.Request) {
+func handleFTPTransfer(sh *network.StreamHandler, wrt http.ResponseWriter, req *http.Request) {
 	vars := mux.Vars(req)
 	token := vars["token"]
 
 	// Check connection occupancy and accept connection via H2Conn.
-	sh, ok := network.FTPStreams[token] // assume FTPStreams token mapping exists
-	if !ok {
-		logging.Errorf("handleFTPTransfer: token %s not found", token)
-		http.Error(wrt, http.StatusText(http.StatusBadRequest), http.StatusBadRequest)
-		return
-	}
 	if sh.H2x != nil && (sh.H2x.Ctx != nil || sh.H2x.Cancel != nil || sh.H2x.Conn != nil) {
 		logging.Errorf("handleFTPTransfer: connection occupied")
 		http.Error(wrt, http.StatusText(http.StatusBadRequest), http.StatusBadRequest)

core/internal/cc/server/handler_portfwd.go

@@ -16,11 +16,10 @@ import (
 )
 
 // handlePortForwarding handles proxy/port forwarding.
-func handlePortForwarding(wrt http.ResponseWriter, req *http.Request) {
+func handlePortForwarding(sh *network.StreamHandler, wrt http.ResponseWriter, req *http.Request) {
 	var err error
-	var h2x def.H2Conn
-	sh := new(network.StreamHandler)
-	sh.H2x = &h2x
+	h2x := new(def.H2Conn)
+	sh.H2x = h2x
 	sh.H2x.Conn, err = h2conn.Accept(wrt, req)
 	if err != nil {
 		logging.Errorf("handlePortForwarding: connection failed from %s: %s", req.RemoteAddr, err)