arpBulkGet.go

//
//Copyright [2016] [SnapRoute Inc]
//
//Licensed under the Apache License, Version 2.0 (the "License");
//you may not use this file except in compliance with the License.
//You may obtain a copy of the License at
//
//    http://www.apache.org/licenses/LICENSE-2.0
//
//	 Unless required by applicable law or agreed to in writing, software
//	 distributed under the License is distributed on an "AS IS" BASIS,
//	 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//	 See the License for the specific language governing permissions and
//	 limitations under the License.
//
// _______  __       __________   ___      _______.____    __    ____  __  .___________.  ______  __    __
// |   ____||  |     |   ____\  \ /  /     /       |\   \  /  \  /   / |  | |           | /      ||  |  |  |
// |  |__   |  |     |  |__   \  V  /     |   (----` \   \/    \/   /  |  | `---|  |----`|  ,----'|  |__|  |
// |   __|  |  |     |   __|   >   <       \   \      \            /   |  |     |  |     |  |     |   __   |
// |  |     |  `----.|  |____ /  .  \  .----)   |      \    /\    /    |  |     |  |     |  `----.|  |  |  |
// |__|     |_______||_______/__/ \__\ |_______/        \__/  \__/     |__|     |__|      \______||__|  |__|
//

package server

import (
	"time"
)

func (server *ARPServer) GetBulkArpEntry(idx int, cnt int) (int, int, []ArpState) {
	var nextIdx int
	var count int

	ret := server.arpSliceRefreshTimer.Stop()
	if ret == false {
		server.logger.Err("Arp is busy refreshing the Arp Entry Cache")
		return nextIdx, count, nil
	}

	length := len(server.arpSlice)
	result := make([]ArpState, cnt)
	var i int
	var j int
	for i, j = 0, idx; i < cnt && j < length; j++ {
		arpSliceEnt := server.arpSlice[j]
		arpEnt, exist := server.arpCache[arpSliceEnt]
		if !exist {
			continue
		}
		result[i].IpAddr = arpSliceEnt
		if arpEnt.MacAddr != "incomplete" {
			result[i].MacAddr = arpEnt.MacAddr
			result[i].Intf = arpEnt.IfName
			result[i].VlanId = arpEnt.VlanId
			curTime := time.Now()
			expiryTime := time.Duration(server.timerGranularity*server.timeoutCounter) * time.Second
			timeElapsed := curTime.Sub(arpEnt.TimeStamp)
			timeLeft := expiryTime - timeElapsed
			result[i].ExpiryTimeLeft = timeLeft.String()
		} else {
			result[i].MacAddr = arpEnt.MacAddr
			result[i].Intf = "N/A"
			result[i].VlanId = -1
			result[i].ExpiryTimeLeft = "N/A"
		}
		i++
	}
	if j == length {
		nextIdx = 0
	}
	count = i
	server.arpSliceRefreshTimer.Reset(server.arpSliceRefreshDuration)
	server.printArpEntries()
	return nextIdx, count, result
}

func (server *ARPServer) GetBulkLinuxArpEntry(idx int, cnt int) (int, int, []ArpLinuxState) {
	var nextIdx int
	var count int

	arpCache := GetLinuxArpCache()

	length := len(arpCache)
	result := make([]ArpLinuxState, cnt)
	var i int
	var j int
	for i, j = 0, idx; i < cnt && j < length; j++ {
		arpEnt := arpCache[j]
		result[i].IpAddr = arpEnt.IpAddr
		if arpEnt.Flags == "0x0" {
			result[i].MacAddr = "incomplete"
			result[i].HWType = "N/A"
		} else {
			result[i].MacAddr = arpEnt.MacAddr
			if arpEnt.HWType == "0x1" {
				result[i].HWType = "ether"
			} else {
				result[i].HWType = "non-ether"
			}
		}
		result[i].IfName = arpEnt.IfName
		i++
	}
	if j == length {
		nextIdx = 0
	}
	count = i

	return nextIdx, count, result
}