Added ask for lan function
This commit is contained in:
		@@ -7,22 +7,25 @@ import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io"
 | 
			
		||||
	"log"
 | 
			
		||||
	"net"
 | 
			
		||||
	"os"
 | 
			
		||||
	"packets/internal"
 | 
			
		||||
	"path/filepath"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"github.com/klauspost/compress/zstd"
 | 
			
		||||
	"github.com/pelletier/go-toml/v2"
 | 
			
		||||
	"github.com/spf13/cobra"
 | 
			
		||||
	"golang.org/x/net/ipv4"
 | 
			
		||||
	_ "modernc.org/sqlite"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// Consts
 | 
			
		||||
 | 
			
		||||
const DefaultLinux_d = "/etc/packets"
 | 
			
		||||
const DefaultLANDeadline = 2 * time.Second
 | 
			
		||||
const LANDeadline = 2 * time.Second
 | 
			
		||||
 | 
			
		||||
// Errors
 | 
			
		||||
 | 
			
		||||
@@ -39,6 +42,11 @@ type ConfigTOML struct {
 | 
			
		||||
	} `toml:"Config"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Peer struct {
 | 
			
		||||
	IP   net.IP
 | 
			
		||||
	Port int
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// init is doing some verifications
 | 
			
		||||
func init() {
 | 
			
		||||
 | 
			
		||||
@@ -183,7 +191,72 @@ func Install(file *os.File) error {
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// COBRA CMDS
 | 
			
		||||
func AskLAN(filename string) ([]Peer, error) {
 | 
			
		||||
	var peers []Peer
 | 
			
		||||
	query := []byte("Q:" + filename)
 | 
			
		||||
 | 
			
		||||
	pc, err := net.ListenPacket("udp", ":0")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return []Peer{}, err
 | 
			
		||||
	}
 | 
			
		||||
	defer pc.Close()
 | 
			
		||||
 | 
			
		||||
	if pconn := ipv4.NewPacketConn(pc); pconn != nil {
 | 
			
		||||
		_ = pconn.SetTTL(1)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	ifaces, _ := net.Interfaces()
 | 
			
		||||
	for _, ifc := range ifaces {
 | 
			
		||||
		if ifc.Flags&net.FlagUp == 0 || ifc.Flags&net.FlagLoopback != 0 {
 | 
			
		||||
			continue
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		addrs, _ := ifc.Addrs()
 | 
			
		||||
		for _, a := range addrs {
 | 
			
		||||
			ipnet, ok := a.(*net.IPNet)
 | 
			
		||||
			if !ok || ipnet.IP.To4() == nil {
 | 
			
		||||
				continue
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			bcast := internal.BroadcastAddr(ipnet.IP.To4(), ipnet.Mask)
 | 
			
		||||
			dst := &net.UDPAddr{IP: bcast, Port: 1333}
 | 
			
		||||
 | 
			
		||||
			_, err = pc.WriteTo(query, dst)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				fmt.Printf(":: (%s) can't send to  %s: %s\n", ifc.Name, bcast, err.Error())
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	_ = pc.SetDeadline(time.Now().Add(LANDeadline))
 | 
			
		||||
	buf := make([]byte, 1500)
 | 
			
		||||
 | 
			
		||||
	for {
 | 
			
		||||
		n, addr, err := pc.ReadFrom(buf)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			break
 | 
			
		||||
		}
 | 
			
		||||
		msg := string(buf[:n])
 | 
			
		||||
 | 
			
		||||
		if strings.HasPrefix(msg, "H:"+filename) {
 | 
			
		||||
			parts := strings.Split(msg, ":")
 | 
			
		||||
			port, _ := strconv.Atoi(parts[2])
 | 
			
		||||
			peers = append(peers, Peer{IP: addr.(*net.UDPAddr).IP, Port: port})
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return peers, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// COBRA CMDS
 | 
			
		||||
var rootCmd = &cobra.Command{Use: "packets"}
 | 
			
		||||
var installCmd = &cobra.Command{}
 | 
			
		||||
var installCmd = &cobra.Command{
 | 
			
		||||
	Use:   "install {package} [packages...]",
 | 
			
		||||
	Short: "Install a package",
 | 
			
		||||
	Args:  cobra.MinimumNArgs(1),
 | 
			
		||||
	Run: func(cmd *cobra.Command, args []string) {
 | 
			
		||||
 | 
			
		||||
	},
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func main() {
 | 
			
		||||
	rootCmd.AddCommand(installCmd)
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										3
									
								
								go.mod
									
									
									
									
									
								
							
							
						
						
									
										3
									
								
								go.mod
									
									
									
									
									
								
							@@ -7,6 +7,7 @@ require (
 | 
			
		||||
	github.com/pelletier/go-toml/v2 v2.2.4
 | 
			
		||||
	github.com/spf13/cobra v1.10.1
 | 
			
		||||
	github.com/yuin/gopher-lua v1.1.1
 | 
			
		||||
	golang.org/x/net v0.44.0
 | 
			
		||||
	modernc.org/sqlite v1.38.2
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
@@ -19,7 +20,7 @@ require (
 | 
			
		||||
	github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec // indirect
 | 
			
		||||
	github.com/spf13/pflag v1.0.10 // indirect
 | 
			
		||||
	golang.org/x/exp v0.0.0-20250620022241-b7579e27df2b // indirect
 | 
			
		||||
	golang.org/x/sys v0.34.0 // indirect
 | 
			
		||||
	golang.org/x/sys v0.36.0 // indirect
 | 
			
		||||
	modernc.org/libc v1.66.3 // indirect
 | 
			
		||||
	modernc.org/mathutil v1.7.1 // indirect
 | 
			
		||||
	modernc.org/memory v1.11.0 // indirect
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										6
									
								
								go.sum
									
									
									
									
									
								
							
							
						
						
									
										6
									
								
								go.sum
									
									
									
									
									
								
							@@ -29,11 +29,13 @@ golang.org/x/exp v0.0.0-20250620022241-b7579e27df2b h1:M2rDM6z3Fhozi9O7NWsxAkg/y
 | 
			
		||||
golang.org/x/exp v0.0.0-20250620022241-b7579e27df2b/go.mod h1:3//PLf8L/X+8b4vuAfHzxeRUl04Adcb341+IGKfnqS8=
 | 
			
		||||
golang.org/x/mod v0.25.0 h1:n7a+ZbQKQA/Ysbyb0/6IbB1H/X41mKgbhfv7AfG/44w=
 | 
			
		||||
golang.org/x/mod v0.25.0/go.mod h1:IXM97Txy2VM4PJ3gI61r1YEk/gAj6zAHN3AdZt6S9Ww=
 | 
			
		||||
golang.org/x/net v0.44.0 h1:evd8IRDyfNBMBTTY5XRF1vaZlD+EmWx6x8PkhR04H/I=
 | 
			
		||||
golang.org/x/net v0.44.0/go.mod h1:ECOoLqd5U3Lhyeyo/QDCEVQ4sNgYsqvCZ722XogGieY=
 | 
			
		||||
golang.org/x/sync v0.15.0 h1:KWH3jNZsfyT6xfAfKiz6MRNmd46ByHDYaZ7KSkCtdW8=
 | 
			
		||||
golang.org/x/sync v0.15.0/go.mod h1:1dzgHSNfp02xaA81J2MS99Qcpr2w7fw1gpm99rleRqA=
 | 
			
		||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 | 
			
		||||
golang.org/x/sys v0.34.0 h1:H5Y5sJ2L2JRdyv7ROF1he/lPdvFsd0mJHFw2ThKHxLA=
 | 
			
		||||
golang.org/x/sys v0.34.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=
 | 
			
		||||
golang.org/x/sys v0.36.0 h1:KVRy2GtZBrk1cBYA7MKu5bEZFxQk4NIDV6RLVcC8o0k=
 | 
			
		||||
golang.org/x/sys v0.36.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
 | 
			
		||||
golang.org/x/tools v0.34.0 h1:qIpSLOxeCYGg9TrcJokLBG4KFA6d795g0xkBkiESGlo=
 | 
			
		||||
golang.org/x/tools v0.34.0/go.mod h1:pAP9OwEaY1CAW3HOmg3hLZC5Z0CCmzjAF2UQMSqNARg=
 | 
			
		||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 | 
			
		||||
 
 | 
			
		||||
@@ -5,6 +5,7 @@ import (
 | 
			
		||||
	"errors"
 | 
			
		||||
	"io"
 | 
			
		||||
	"log"
 | 
			
		||||
	"net"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"os"
 | 
			
		||||
	"path/filepath"
 | 
			
		||||
@@ -150,3 +151,11 @@ func GetConfigTOML() (*ConfigTOML, error) {
 | 
			
		||||
 | 
			
		||||
	return &config, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func BroadcastAddr(ip net.IP, mask net.IPMask) net.IP {
 | 
			
		||||
	b := make(net.IP, len(ip))
 | 
			
		||||
	for i := range ip {
 | 
			
		||||
		b[i] = ip[i] | ^mask[i]
 | 
			
		||||
	}
 | 
			
		||||
	return b
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user