func main() { go sniff() if len(os.Args) == 1 { log.Fatalln("You didn't gave to me how times i have to perform this action") os.Exit(1) } toNuke := os.Args[1] myIP := net.ParseIP(os.Args[2]) dhcpServer := net.ParseIP(os.Args[3]) myMAC := os.Args[4] Times, _ := strconv.Atoi(toNuke) m, err := net.ParseMAC(myMAC) if err != nil { log.Printf("MAC Error:%v\n", err) } c, err := dhcp4client.NewInetSock(dhcp4client.SetLocalAddr(net.UDPAddr{IP: myIP, Port: 68}), dhcp4client.SetRemoteAddr(net.UDPAddr{IP: dhcpServer, Port: 67})) exampleClient, err := dhcp4client.New(dhcp4client.HardwareAddr(m), dhcp4client.Connection(c), dhcp4client.Broadcast(false)) defer func() { if err != nil { exampleClient.Close() } }() for j := 1; j <= Times; j++ { ip := RequestIP(exampleClient) log.Println(ip) ipv4 := net.ParseIP(ip.YIAddr().String()) //Nuke(ip) //Nuke(exampleClient, m, net.ParseIP("192.168.1.243")) Nuke(exampleClient, m, ipv4, ip) } }
func GetOffer() (ip net.IP, subnet net.IPNet, err error) { ip = nil subnet.IP = nil subnet.Mask = nil // Generate a random mac address for this request hwaddr, e := GenerateMacAddr() if e != nil { err = e return } // Setup DHCP socket //socket, e := dhcp4client.NewInetSock(local_addr, remote_addr) socket, e := dhcp4client.NewInetSock() if e != nil { err = e return } // Setup DHCP client client, e := dhcp4client.New( dhcp4client.HardwareAddr(hwaddr), dhcp4client.Connection(socket), ) // Send a DHCPREQUEST result, packet, e := client.Request() if !result { msg := fmt.Sprintf("DHCPREQUEST failed: %v", e) err = errors.New(msg) return } if e != nil { networkError, ok := e.(*net.OpError) if ok && networkError.Timeout() { msg := fmt.Sprintf("Cannot find DHCP server: %v", networkError) err = errors.New(msg) } else { msg := fmt.Sprintf("DHCPREQUEST failed: %v", networkError) err = errors.New(msg) } return } // Determine the netmask of the network options := packet.ParseOptions() mask := make(net.IPMask, net.IPv4len) mask = options[dhcp4.OptionCode(dhcp4.OptionSubnetMask)] // Setup the return structures ip = packet.YIAddr() subnet.IP = ip.To4().Mask(mask) subnet.Mask = mask return }
func newDHCPClient(link netlink.Link) (*dhcp4client.Client, error) { pktsock, err := dhcp4client.NewPacketSock(link.Attrs().Index) if err != nil { return nil, err } return dhcp4client.New( dhcp4client.HardwareAddr(link.Attrs().HardwareAddr), dhcp4client.Timeout(5*time.Second), dhcp4client.Broadcast(false), dhcp4client.Connection(pktsock), ) }
func (c *client) Request() error { defer trace.End(trace.Begin("")) log.Debugf("id: %+v", c.id) // send the request over a raw socket raw, err := dhcp4client.NewPacketSock(c.id.IfIndex) if err != nil { return err } rawc, err := dhcp4client.New( dhcp4client.Connection(raw), dhcp4client.Timeout(c.timeout), dhcp4client.HardwareAddr(c.id.HardwareAddr)) if err != nil { return err } defer rawc.Close() success := false var p dhcp4.Packet err = withRetry(func() error { var err error success, p, err = c.request(rawc) return err }) if err != nil { return err } if !success { return fmt.Errorf("failed dhcp request") } log.Debugf("%+v", p) c.ack = p return nil }
/* * Test Discovering a Lease That's not Within Our Lease Range. * This Happens When a devce switches network. * Example: Mobile Phone on Mobile internet Has IP 100.123.123.123 Switch To Home Wifi * The device requests 100.123.123.123 on Home Wifi which is out of range... */ func TestDiscoverOutOfRangeLease(test *testing.T) { //Setup the Server myServer, err := dhcp4server.New( net.IPv4(192, 168, 1, 201), getTestLeasePool(), dhcp4server.SetLocalAddr(net.UDPAddr{IP: net.IPv4(127, 0, 0, 1), Port: 1067}), dhcp4server.SetRemoteAddr(net.UDPAddr{IP: net.IPv4(127, 0, 0, 1), Port: 1068}), ) if err != nil { test.Error("Error: Can't Configure Server " + err.Error()) } var wg sync.WaitGroup wg.Add(1) go func() { defer wg.Done() err := myServer.ListenAndServe() if err != nil { log.Fatalln("Error Starting Server:" + err.Error()) } }() time.Sleep(time.Duration(5) * time.Second) //Generate Hardware Address HardwareMACAddress, err := hardwareaddr.GenerateEUI48() if err != nil { test.Error("Error: Can't Generate Valid MACAddress" + err.Error()) } //Lets Be A Client //We need to set the connection ports to 1068 and 1067 so we don't need root access c, err := dhcp4client.NewInetSock(dhcp4client.SetLocalAddr(net.UDPAddr{IP: net.IPv4(127, 0, 0, 1), Port: 1068}), dhcp4client.SetRemoteAddr(net.UDPAddr{IP: net.IPv4(127, 0, 0, 1), Port: 1067})) if err != nil { test.Error("Client Conection Generation:" + err.Error()) } client, err := dhcp4client.New(dhcp4client.HardwareAddr(HardwareMACAddress), dhcp4client.Connection(c)) defer client.Close() if err != nil { test.Error("Conection Error:" + err.Error()) } discoveryPacket := client.DiscoverPacket() discoveryPacket.SetCIAddr(net.IPv4(100, 102, 96, 123)) discoveryPacket.PadToMinSize() err = client.SendPacket(discoveryPacket) if err != nil { test.Error("Error: Sending Discover Packet" + err.Error()) } test.Log("--Discovery Packet--") test.Logf("Client IP : %v\n", discoveryPacket.CIAddr().String()) test.Logf("Your IP : %v\n", discoveryPacket.YIAddr().String()) test.Logf("Server IP : %v\n", discoveryPacket.SIAddr().String()) test.Logf("Gateway IP: %v\n", discoveryPacket.GIAddr().String()) test.Logf("Client Mac: %v\n", discoveryPacket.CHAddr().String()) if !bytes.Equal(discoveryPacket.CHAddr(), HardwareMACAddress) { test.Error("MACAddresses Don't Match??") } offerPacket, err := client.GetOffer(&discoveryPacket) if err != nil { test.Error("Error Getting Offer:" + err.Error()) } test.Log("--Offer Packet--") test.Logf("Client IP : %v\n", offerPacket.CIAddr().String()) test.Logf("Your IP : %v\n", offerPacket.YIAddr().String()) test.Logf("Server IP : %v\n", offerPacket.SIAddr().String()) test.Logf("Gateway IP: %v\n", offerPacket.GIAddr().String()) test.Logf("Client Mac: %v\n", offerPacket.CHAddr().String()) requestPacket, err := client.SendRequest(&offerPacket) if err != nil { test.Error("Error Sending Request:" + err.Error()) } test.Log("--Request Packet--") test.Logf("Client IP : %v\n", requestPacket.CIAddr().String()) test.Logf("Your IP : %v\n", requestPacket.YIAddr().String()) test.Logf("Server IP : %v\n", requestPacket.SIAddr().String()) test.Logf("Gateway IP: %v\n", requestPacket.GIAddr().String()) test.Logf("Client Mac: %v\n", requestPacket.CHAddr().String()) acknowledgement, err := client.GetAcknowledgement(&requestPacket) if err != nil { test.Error("Error Getting Acknowledgement:" + err.Error()) } test.Log("--Acknowledgement Packet--") test.Logf("Client IP : %v\n", acknowledgement.CIAddr().String()) test.Logf("Your IP : %v\n", acknowledgement.YIAddr().String()) test.Logf("Server IP : %v\n", acknowledgement.SIAddr().String()) test.Logf("Gateway IP: %v\n", acknowledgement.GIAddr().String()) test.Logf("Client Mac: %v\n", acknowledgement.CHAddr().String()) acknowledgementOptions := acknowledgement.ParseOptions() if dhcp4.MessageType(acknowledgementOptions[dhcp4.OptionDHCPMessageType][0]) != dhcp4.ACK { test.Error("Didn't get ACK?:" + err.Error()) } test.Log("Shutting Down Server") myServer.Shutdown() wg.Wait() }
/* * Benchmark the ServeDHCP Function */ func BenchmarkServeDHCP(test *testing.B) { //Create a Lease Pool We're going to use a memory pool //Remember the memory is cleared on restart so you will reissue the same IP Addresses. myMemoryLeasePool := memorypool.MemoryPool{} //Lets add a list of IPs to the pool these will be served to the clients so make sure they work for you. // So Create Array of IPs 192.168.1.1 to 192.168.1.30 for i := 0; i < test.N; i++ { err := myMemoryLeasePool.AddLease(leasepool.Lease{IP: dhcp4.IPAdd(net.IPv4(192, 168, 1, 1), i)}) if err != nil { log.Fatalln("Error Adding IP to pool:" + err.Error()) } } //Setup the Server myServer, err := dhcp4server.New( net.IPv4(127, 0, 0, 1), &myMemoryLeasePool, ) if err != nil { test.Error("Error: Can't Configure Server " + err.Error()) } //Setup A Client // Although We Won't send the packets over the network we'll use the client to create the requests. c, err := dhcp4client.NewInetSock(dhcp4client.SetLocalAddr(net.UDPAddr{IP: net.IPv4(127, 0, 0, 1), Port: 1068}), dhcp4client.SetRemoteAddr(net.UDPAddr{IP: net.IPv4(127, 0, 0, 1), Port: 1067})) if err != nil { test.Error("Client Conection Generation:" + err.Error()) } client, err := dhcp4client.New(dhcp4client.Connection(c)) if err != nil { test.Error("Error: Can't Configure Client " + err.Error()) } defer client.Close() test.ResetTimer() for i := 0; i < test.N; i++ { test.StopTimer() //Generate Hardware Address HardwareMACAddress, err := hardwareaddr.GenerateEUI48() if err != nil { test.Error("Error: Can't Generate Valid MACAddress" + err.Error()) } client.SetOption(dhcp4client.HardwareAddr(HardwareMACAddress)) discovery := client.DiscoverPacket() //Run the Discovery On the Server test.StartTimer() offer, err := myServer.ServeDHCP(discovery) if err != nil { test.Error("Discovery Error:" + err.Error()) } if len(offer) == 0 { test.Error("No Valid Offer") } else { request := client.RequestPacket(&offer) _, err := myServer.ServeDHCP(request) if err != nil { test.Error("Acknowledge Error:" + err.Error()) } } } }
func TestConsumeLeases(test *testing.T) { //Setup the Server myServer, err := dhcp4server.New( net.IPv4(127, 0, 0, 1), getTestLeasePool(), ) if err != nil { test.Error("Error: Can't Configure Server " + err.Error()) } // Setup A Client // Although We Won't send the packets over the network we'll use the client to create the requests. c, err := dhcp4client.NewInetSock(dhcp4client.SetLocalAddr(net.UDPAddr{IP: net.IPv4(127, 0, 0, 1), Port: 1068}), dhcp4client.SetRemoteAddr(net.UDPAddr{IP: net.IPv4(127, 0, 0, 1), Port: 1067})) if err != nil { test.Error("Client Conection Generation:" + err.Error()) } client, err := dhcp4client.New(dhcp4client.Connection(c)) if err != nil { test.Error("Error: Can't Configure Client " + err.Error()) } defer client.Close() for i := 0; i < 30; i++ { //Generate Hardware Address HardwareMACAddress, err := hardwareaddr.GenerateEUI48() if err != nil { test.Error("Error: Can't Generate Valid MACAddress" + err.Error()) } client.SetOption(dhcp4client.HardwareAddr(HardwareMACAddress)) test.Log("MAC:" + HardwareMACAddress.String()) discovery := client.DiscoverPacket() //Run the Discovery On the Server offer, err := myServer.ServeDHCP(discovery) _, err = myServer.ServeDHCP(discovery) if err != nil { test.Error("Discovery Error:" + err.Error()) } request := client.RequestPacket(&offer) acknowledgement, err := myServer.ServeDHCP(request) if err != nil { test.Error("Acknowledge Error:" + err.Error()) } test.Logf("Received Lease:%v\n", acknowledgement.YIAddr().String()) if !dhcp4.IPAdd(net.IPv4(192, 168, 1, 1), i).Equal(acknowledgement.YIAddr()) { test.Error("Expected IP:" + dhcp4.IPAdd(net.IPv4(192, 168, 1, 1), i).String() + " Received:" + acknowledgement.YIAddr().String()) } //How long the lease is for? acknowledgementOptions := acknowledgement.ParseOptions() if len(acknowledgementOptions) > 0 { test.Logf("Lease Options:%v\n", acknowledgementOptions) if acknowledgementOptions[dhcp4.OptionIPAddressLeaseTime] != nil { var result uint32 buf := bytes.NewBuffer(acknowledgementOptions[dhcp4.OptionIPAddressLeaseTime]) binary.Read(buf, binary.BigEndian, &result) test.Logf("Lease Time (Seconds):%d\n", result) } } else { test.Errorf("Lease:\"%v\" Has No Options\n", acknowledgement.YIAddr()) } } }
/* * Try Renewing A Lease From A Different Network. */ func TestRequestOutOfRangeLease(test *testing.T) { //Setup the Server myServer, err := dhcp4server.New( net.IPv4(192, 168, 1, 201), getTestLeasePool(), dhcp4server.SetLocalAddr(net.UDPAddr{IP: net.IPv4(127, 0, 0, 1), Port: 1067}), dhcp4server.SetRemoteAddr(net.UDPAddr{IP: net.IPv4(127, 0, 0, 1), Port: 1068}), ) if err != nil { test.Error("Error: Can't Configure Server " + err.Error()) } var wg sync.WaitGroup wg.Add(1) go func() { defer wg.Done() err := myServer.ListenAndServe() if err != nil { log.Fatalln("Error Starting Server:" + err.Error()) } }() //Sleep some so the server starts.... time.Sleep(time.Duration(5) * time.Second) //Generate Hardware Address HardwareMACAddress, err := hardwareaddr.GenerateEUI48() if err != nil { test.Error("Error: Can't Generate Valid MACAddress" + err.Error()) } HardwareMACAddress, err = net.ParseMAC("58-94-6B-73-57-0C") if err != nil { log.Printf("MAC Error:%v\n", err) } //Lets Be A Client c, err := dhcp4client.NewInetSock(dhcp4client.SetLocalAddr(net.UDPAddr{IP: net.IPv4(127, 0, 0, 1), Port: 1068}), dhcp4client.SetRemoteAddr(net.UDPAddr{IP: net.IPv4(127, 0, 0, 1), Port: 1067})) if err != nil { test.Error("Client Conection Generation:" + err.Error()) } client, err := dhcp4client.New(dhcp4client.HardwareAddr(HardwareMACAddress), dhcp4client.Connection(c)) defer client.Close() if err != nil { test.Error("Conection Error:" + err.Error()) } //Create a dummy offer packet offerPacket := client.DiscoverPacket() offerPacket.SetCIAddr(net.IPv4(100, 102, 96, 123)) offerPacket.SetSIAddr(net.IPv4(192, 168, 1, 201)) offerPacket.SetYIAddr(net.IPv4(100, 102, 96, 123)) offerPacket.AddOption(dhcp4.OptionDHCPMessageType, []byte{byte(dhcp4.Offer)}) requestPacket, err := client.SendRequest(&offerPacket) if err != nil { test.Error("Error Sending Request:" + err.Error()) } test.Log("--Request Packet--") test.Logf("Client IP : %v\n", requestPacket.CIAddr().String()) test.Logf("Your IP : %v\n", requestPacket.YIAddr().String()) test.Logf("Server IP : %v\n", requestPacket.SIAddr().String()) test.Logf("Gateway IP: %v\n", requestPacket.GIAddr().String()) test.Logf("Client Mac: %v\n", requestPacket.CHAddr().String()) acknowledgement, err := client.GetAcknowledgement(&requestPacket) if err != nil { test.Error("Error Getting Acknowledgement:" + err.Error()) } test.Log("--Acknowledgement Packet--") test.Logf("Client IP : %v\n", acknowledgement.CIAddr().String()) test.Logf("Your IP : %v\n", acknowledgement.YIAddr().String()) test.Logf("Server IP : %v\n", acknowledgement.SIAddr().String()) test.Logf("Gateway IP: %v\n", acknowledgement.GIAddr().String()) test.Logf("Client Mac: %v\n", acknowledgement.CHAddr().String()) acknowledgementOptions := acknowledgement.ParseOptions() if len(acknowledgementOptions[dhcp4.OptionDHCPMessageType]) <= 0 || dhcp4.MessageType(acknowledgementOptions[dhcp4.OptionDHCPMessageType][0]) != dhcp4.NAK { test.Errorf("Didn't get NAK got DHCP4 Message Type:%v\n", dhcp4.MessageType(acknowledgementOptions[dhcp4.OptionDHCPMessageType][0])) } test.Log("Shutting Down Server") myServer.Shutdown() wg.Wait() }
func (c *client) Request(id ID) (*dhcp.Packet, error) { log.Debugf("id: %+v", id) // send the request over a raw socket raw, err := dhcp4client.NewPacketSock(id.IfIndex) if err != nil { return nil, err } rawc, err := dhcp4client.New(dhcp4client.Connection(raw), dhcp4client.Timeout(c.timeout), dhcp4client.HardwareAddr(id.HardwareAddr)) if err != nil { return nil, err } defer rawc.Close() success := false var p *dhcp.Packet err = withRetry(func() error { var err error success, p, err = c.request(id, rawc) return err }) if err != nil { return nil, err } if !success { return nil, fmt.Errorf("failed dhcp request") } log.Debugf("%+v", p) return p, nil }