change ff to unix
This commit is contained in:
parent
34e1ad1e39
commit
fbfb32f8b1
|
|
@ -1,70 +1,70 @@
|
|||
import socket
|
||||
import struct
|
||||
import array
|
||||
import sys
|
||||
|
||||
IOCTL_SIOCGIFCONF = 0x8912
|
||||
|
||||
# /*
|
||||
# * Structure used in SIOCGIFCONF request.
|
||||
# * Used to retrieve interface configuration
|
||||
# * for machine (useful for programs which
|
||||
# * must know all networks accessible).
|
||||
# */
|
||||
#
|
||||
# struct ifconf {
|
||||
# int ifc_len; /* size of associated buffer */
|
||||
# union {
|
||||
# caddr_t ifcu_buf;
|
||||
# struct ifreq *ifcu_req;
|
||||
# } ifc_ifcu;
|
||||
#
|
||||
# /*
|
||||
# * Interface request structure used for socket
|
||||
# * ioctl's.
|
||||
# */
|
||||
#
|
||||
# struct ifreq {
|
||||
# char ifr_name[16]; /* if name, e.g. "en0" */
|
||||
# union {
|
||||
# struct sockaddr ifru_addr;
|
||||
# struct sockaddr ifru_dstaddr;
|
||||
# struct sockaddr ifru_broadaddr;
|
||||
# short ifru_flags;
|
||||
# int ifru_metric;
|
||||
# caddr_t ifru_data;
|
||||
# } ifr_ifru;
|
||||
#
|
||||
#
|
||||
# /* Socket address, DARPA Internet style */
|
||||
# struct sockaddr_in {
|
||||
# short sin_family;
|
||||
# unsigned short sin_port;
|
||||
# struct in_addr sin_addr;
|
||||
# char sin_zero[8];
|
||||
# };
|
||||
#
|
||||
#
|
||||
# ioctl (fd, SIOCGIFCONF, &ifconf)
|
||||
|
||||
|
||||
def enumIpUnix():
|
||||
import fcntl
|
||||
|
||||
struct_size = 40 if (sys.maxsize > 2**32) else 32
|
||||
|
||||
inbytes = 128 * struct_size # Maximum 128 interfaces
|
||||
ifreq = array.array('B', b'\0' * inbytes)
|
||||
|
||||
ifconf = struct.pack('iL', inbytes, ifreq.buffer_info()[0])
|
||||
|
||||
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||||
ifconf = fcntl.ioctl(s.fileno(), IOCTL_SIOCGIFCONF, ifconf)
|
||||
|
||||
outbytes = struct.unpack('iL', ifconf)[0]
|
||||
|
||||
iplist = [socket.inet_ntoa(ifreq[i+20:i+24]) for i in range(0,
|
||||
outbytes,
|
||||
struct_size)]
|
||||
|
||||
return [ip for ip in iplist if ip != '127.0.0.1']
|
||||
import socket
|
||||
import struct
|
||||
import array
|
||||
import sys
|
||||
|
||||
IOCTL_SIOCGIFCONF = 0x8912
|
||||
|
||||
# /*
|
||||
# * Structure used in SIOCGIFCONF request.
|
||||
# * Used to retrieve interface configuration
|
||||
# * for machine (useful for programs which
|
||||
# * must know all networks accessible).
|
||||
# */
|
||||
#
|
||||
# struct ifconf {
|
||||
# int ifc_len; /* size of associated buffer */
|
||||
# union {
|
||||
# caddr_t ifcu_buf;
|
||||
# struct ifreq *ifcu_req;
|
||||
# } ifc_ifcu;
|
||||
#
|
||||
# /*
|
||||
# * Interface request structure used for socket
|
||||
# * ioctl's.
|
||||
# */
|
||||
#
|
||||
# struct ifreq {
|
||||
# char ifr_name[16]; /* if name, e.g. "en0" */
|
||||
# union {
|
||||
# struct sockaddr ifru_addr;
|
||||
# struct sockaddr ifru_dstaddr;
|
||||
# struct sockaddr ifru_broadaddr;
|
||||
# short ifru_flags;
|
||||
# int ifru_metric;
|
||||
# caddr_t ifru_data;
|
||||
# } ifr_ifru;
|
||||
#
|
||||
#
|
||||
# /* Socket address, DARPA Internet style */
|
||||
# struct sockaddr_in {
|
||||
# short sin_family;
|
||||
# unsigned short sin_port;
|
||||
# struct in_addr sin_addr;
|
||||
# char sin_zero[8];
|
||||
# };
|
||||
#
|
||||
#
|
||||
# ioctl (fd, SIOCGIFCONF, &ifconf)
|
||||
|
||||
|
||||
def enumIpUnix():
|
||||
import fcntl
|
||||
|
||||
struct_size = 40 if (sys.maxsize > 2**32) else 32
|
||||
|
||||
inbytes = 128 * struct_size # Maximum 128 interfaces
|
||||
ifreq = array.array('B', b'\0' * inbytes)
|
||||
|
||||
ifconf = struct.pack('iL', inbytes, ifreq.buffer_info()[0])
|
||||
|
||||
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||||
ifconf = fcntl.ioctl(s.fileno(), IOCTL_SIOCGIFCONF, ifconf)
|
||||
|
||||
outbytes = struct.unpack('iL', ifconf)[0]
|
||||
|
||||
iplist = [socket.inet_ntoa(ifreq[i+20:i+24]) for i in range(0,
|
||||
outbytes,
|
||||
struct_size)]
|
||||
|
||||
return [ip for ip in iplist if ip != '127.0.0.1']
|
||||
|
|
|
|||
|
|
@ -1,357 +1,357 @@
|
|||
#!/usr/bin/env python3
|
||||
|
||||
import getopt
|
||||
import socket
|
||||
import sys
|
||||
import os
|
||||
import platform
|
||||
from nfutil import *
|
||||
from enumip import *
|
||||
|
||||
|
||||
#-----------------------------------------------------------------------------
|
||||
def usage():
|
||||
print
|
||||
print "Usage: nfcli --list --eth_addr=ADDR --ip_type=TYPE --ip_addr=IP, --netmask=MASK, --gateway=GW"
|
||||
print "Search and configure Prologix GPIB-ETHERNET Controllers."
|
||||
print "--help : display this help"
|
||||
print "--list : search for controllers"
|
||||
print "--eth_addr=ADDR : configure controller with Ethernet address ADDR"
|
||||
print "--ip_type=TYPE : set controller ip address type to TYPE (\"static\" or \"dhcp\")"
|
||||
print "--ip_addr=IP : set controller address to IP"
|
||||
print "--netmask=MASK : set controller network mask to MASK"
|
||||
print "--gateway=GW : set controller default gateway to GW"
|
||||
|
||||
|
||||
#-----------------------------------------------------------------------------
|
||||
def enumIp():
|
||||
if platform.system() in ('Windows', 'Microsoft'):
|
||||
return socket.gethostbyname_ex(socket.gethostname())[2];
|
||||
|
||||
return enumIpUnix()
|
||||
|
||||
|
||||
#-----------------------------------------------------------------------------
|
||||
def ValidateNetParams(ip_str, mask_str, gw_str):
|
||||
|
||||
try:
|
||||
ip = socket.inet_aton(ip_str)
|
||||
except:
|
||||
print "IP address is invalid."
|
||||
return False
|
||||
|
||||
try:
|
||||
mask = socket.inet_aton(mask_str)
|
||||
except:
|
||||
print "Network mask is invalid."
|
||||
return False
|
||||
|
||||
try:
|
||||
gw = socket.inet_aton(gw_str)
|
||||
except:
|
||||
print "Gateway address is invalid."
|
||||
return False
|
||||
|
||||
# Validate network mask
|
||||
|
||||
# Convert to integer from byte array
|
||||
mask = struct.unpack("!L", mask)[0]
|
||||
|
||||
# Exclude restricted masks
|
||||
if (mask == 0) or (mask == 0xFFFFFFFF):
|
||||
print "Network mask is invalid."
|
||||
return False
|
||||
|
||||
# Exclude non-left-contiguous masks
|
||||
if (((mask + (mask & -mask)) & 0xFFFFFFFF) != 0):
|
||||
print "Network mask is not contiguous."
|
||||
return False
|
||||
|
||||
|
||||
# Validate gateway address
|
||||
|
||||
octet1 = ord(gw[0])
|
||||
|
||||
# Convert to integer from byte array
|
||||
gw = struct.unpack("!L", gw)[0]
|
||||
|
||||
# Exclude restricted addresses
|
||||
# 0.0.0.0 is valid
|
||||
if ((gw != 0) and ((octet1 == 0) or (octet1 == 127) or (octet1 > 223))):
|
||||
print "Gateway address is invalid."
|
||||
return False
|
||||
|
||||
# Validate IP address
|
||||
|
||||
octet1 = ord(ip[0])
|
||||
|
||||
# Convert to integer from byte array
|
||||
ip = struct.unpack("!L", ip)[0]
|
||||
|
||||
# Exclude restricted addresses
|
||||
if ((octet1 == 0) or (octet1 == 127) or (octet1 > 223)):
|
||||
print "IP address is invalid."
|
||||
return False
|
||||
|
||||
# Exclude subnet network address
|
||||
if ((ip & ~mask) == 0):
|
||||
print "IP address is invalid."
|
||||
return False
|
||||
|
||||
# Exclude subnet broadcast address
|
||||
if ((ip & ~mask) == (0xFFFFFFFF & ~mask)):
|
||||
print "IP address is invalid."
|
||||
return False
|
||||
|
||||
return True
|
||||
|
||||
#-----------------------------------------------------------------------------
|
||||
#def ValidateAddress(address):
|
||||
|
||||
# if address is None:
|
||||
# return False
|
||||
|
||||
# parts = address.split(".")
|
||||
|
||||
# if len(parts) != 4:
|
||||
# return False
|
||||
|
||||
# try:
|
||||
# for item in parts:
|
||||
# if not 0 <= int(item) <= 255:
|
||||
# return False
|
||||
# except:
|
||||
# return False
|
||||
|
||||
# return True
|
||||
|
||||
|
||||
#-----------------------------------------------------------------------------
|
||||
def main():
|
||||
|
||||
invalid_args = False
|
||||
showhelp = False
|
||||
search = False
|
||||
ip_type = None
|
||||
ip_addr = None
|
||||
netmask = None
|
||||
gateway = None
|
||||
eth_addr = None
|
||||
|
||||
try:
|
||||
opts, args = getopt.getopt(sys.argv[1:], '', ['help', 'list', 'eth_addr=', 'ip_type=', 'ip_addr=', 'netmask=', 'gateway='])
|
||||
except getopt.GetoptError, err:
|
||||
print str(err)
|
||||
sys.exit(1)
|
||||
|
||||
# Check for unparsed parameters
|
||||
if len(args) != 0:
|
||||
usage()
|
||||
sys.exit(1)
|
||||
|
||||
for o, a in opts:
|
||||
if o == "--help":
|
||||
showhelp = True
|
||||
elif o == "--list":
|
||||
search = True
|
||||
elif o == "--eth_addr":
|
||||
eth_addr = a
|
||||
elif o == "--ip_type":
|
||||
ip_type = a
|
||||
elif o == "--ip_addr":
|
||||
ip_addr = a
|
||||
elif o == "--netmask":
|
||||
netmask = a
|
||||
elif o == "--gateway":
|
||||
gateway = a
|
||||
|
||||
if (len(opts) == 0) or (showhelp):
|
||||
usage()
|
||||
sys.exit(1)
|
||||
|
||||
if search:
|
||||
if not eth_addr is None:
|
||||
print "--list and --eth_addr are not compatible."
|
||||
invalid_args = True
|
||||
if not ip_type is None:
|
||||
print "--list and --ip_type are not compatible."
|
||||
invalid_args = True
|
||||
if not ip_addr is None:
|
||||
print "--list and --ip_addr are not compatible."
|
||||
invalid_args = True
|
||||
if not netmask is None:
|
||||
print "--list and --netmask are not compatible."
|
||||
invalid_args = True
|
||||
if not gateway is None:
|
||||
print "--list and --gateway are not compatible."
|
||||
invalid_args = True
|
||||
else:
|
||||
|
||||
try:
|
||||
eth_addr = eth_addr.strip().replace(":", "").replace("-", "")
|
||||
eth_addr = eth_addr.decode('hex')
|
||||
except:
|
||||
print "Invalid Ethernet address."
|
||||
sys.exit(1)
|
||||
|
||||
if len(eth_addr) != 6:
|
||||
print "Invalid Ethernet address."
|
||||
sys.exit(1)
|
||||
|
||||
if ip_type in ["Static", "static"]:
|
||||
ip_type = NF_IP_STATIC
|
||||
elif ip_type in ["Dynamic", "dynamic", "Dhcp", "dhcp"]:
|
||||
ip_type = NF_IP_DYNAMIC
|
||||
else:
|
||||
print "--ip_type must be 'static' or 'dhcp'."
|
||||
sys.exit(1)
|
||||
|
||||
if ip_type == NF_IP_STATIC:
|
||||
|
||||
if not ValidateNetParams(ip_addr, netmask, gateway):
|
||||
invalid_args = True
|
||||
|
||||
|
||||
# if not ValidateIP(ip_addr):
|
||||
# print "Invalid, or no, IP address specified."
|
||||
# invalid_args = True
|
||||
|
||||
# if not ValidateIP(netmask):
|
||||
# print "Invalid, or no, netmask specified."
|
||||
# invalid_args = True
|
||||
|
||||
# if not ValidateIP(gateway):
|
||||
# print "Invalid, or no, gateway address specified."
|
||||
# invalid_args = True
|
||||
else:
|
||||
if ip_addr is None:
|
||||
ip_addr = "0.0.0.0"
|
||||
else:
|
||||
print "--ip_addr not allowed when --ip_type=dhcp."
|
||||
invalid_args = True
|
||||
if netmask is None:
|
||||
netmask = "0.0.0.0"
|
||||
else:
|
||||
print "--netmask not allowed when --ip_type=dhcp."
|
||||
invalid_args = True
|
||||
if gateway is None:
|
||||
gateway = "0.0.0.0"
|
||||
else:
|
||||
print "--gateway not allowed when --ip_type=dhcp."
|
||||
invalid_args = True
|
||||
|
||||
if invalid_args:
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
|
||||
global seq
|
||||
sys.stdout = os.fdopen(sys.stdout.fileno(), 'w', 0)
|
||||
|
||||
iplist = enumIp();
|
||||
if len(iplist) == 0:
|
||||
print "Host has no IP address."
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
devices = {}
|
||||
|
||||
for ip in iplist:
|
||||
print "Searching through network interface:", ip
|
||||
|
||||
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||||
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||||
s.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
|
||||
|
||||
port = 0
|
||||
|
||||
try:
|
||||
s.bind((ip, port))
|
||||
except socket.error, e:
|
||||
print "Bind error on send socket:", e
|
||||
sys.exit(1)
|
||||
|
||||
port = s.getsockname()[1]
|
||||
|
||||
r = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||||
r.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||||
|
||||
r.setblocking(1)
|
||||
r.settimeout(0.100)
|
||||
|
||||
try:
|
||||
r.bind(('', port))
|
||||
except socket.error, e:
|
||||
print "Bind error on receive socket:", e
|
||||
sys.exit(1)
|
||||
|
||||
d = Discover(s, r)
|
||||
print
|
||||
|
||||
for k in d:
|
||||
d[k]['host_ip'] = ip
|
||||
devices[k] = d[k]
|
||||
|
||||
s.close()
|
||||
r.close()
|
||||
|
||||
if search:
|
||||
print
|
||||
print "Found", len(devices), "Prologix GPIB-ETHERNET Controller(s)."
|
||||
for key in devices:
|
||||
PrintDetails(devices[key])
|
||||
print
|
||||
else:
|
||||
if eth_addr in devices:
|
||||
|
||||
print "Updating network settings of Prologix GPIB-ETHERNET Controller", FormatEthAddr(eth_addr)
|
||||
|
||||
device = devices[eth_addr]
|
||||
|
||||
if (device['ip_type'] == NF_IP_STATIC) or (ip_type == NF_IP_STATIC):
|
||||
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||||
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||||
s.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
|
||||
|
||||
port = 0
|
||||
|
||||
try:
|
||||
s.bind((device['host_ip'], port))
|
||||
except socket.error, e:
|
||||
print "Bind error on send socket:", e
|
||||
sys.exit(1)
|
||||
|
||||
port = s.getsockname()[1]
|
||||
|
||||
r = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||||
r.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||||
|
||||
r.setblocking(1)
|
||||
r.settimeout(0.100)
|
||||
|
||||
try:
|
||||
r.bind(('', port))
|
||||
except socket.error, e:
|
||||
print "Bind error on receive socket:", e
|
||||
sys.exit(1)
|
||||
|
||||
result = Assignment(s, r, eth_addr, ip_type, ip_addr, netmask, gateway)
|
||||
print
|
||||
|
||||
if len(result) == 0:
|
||||
print "Network settings update failed."
|
||||
else:
|
||||
if result['result'] == NF_SUCCESS:
|
||||
print "Network settings updated successfully."
|
||||
else:
|
||||
print "Network settings update failed."
|
||||
else:
|
||||
print "Prologix GPIB-ETHERNET Controller", FormatEthAddr(eth_addr), "already configured for DHCP."
|
||||
else:
|
||||
print "Prologix GPIB-ETHERNET Controller", FormatEthAddr(eth_addr), "not found."
|
||||
|
||||
|
||||
|
||||
#-----------------------------------------------------------------------------
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
#!/usr/bin/env python3
|
||||
|
||||
import getopt
|
||||
import socket
|
||||
import sys
|
||||
import os
|
||||
import platform
|
||||
from nfutil import *
|
||||
from enumip import *
|
||||
|
||||
|
||||
#-----------------------------------------------------------------------------
|
||||
def usage():
|
||||
print
|
||||
print "Usage: nfcli --list --eth_addr=ADDR --ip_type=TYPE --ip_addr=IP, --netmask=MASK, --gateway=GW"
|
||||
print "Search and configure Prologix GPIB-ETHERNET Controllers."
|
||||
print "--help : display this help"
|
||||
print "--list : search for controllers"
|
||||
print "--eth_addr=ADDR : configure controller with Ethernet address ADDR"
|
||||
print "--ip_type=TYPE : set controller ip address type to TYPE (\"static\" or \"dhcp\")"
|
||||
print "--ip_addr=IP : set controller address to IP"
|
||||
print "--netmask=MASK : set controller network mask to MASK"
|
||||
print "--gateway=GW : set controller default gateway to GW"
|
||||
|
||||
|
||||
#-----------------------------------------------------------------------------
|
||||
def enumIp():
|
||||
if platform.system() in ('Windows', 'Microsoft'):
|
||||
return socket.gethostbyname_ex(socket.gethostname())[2];
|
||||
|
||||
return enumIpUnix()
|
||||
|
||||
|
||||
#-----------------------------------------------------------------------------
|
||||
def ValidateNetParams(ip_str, mask_str, gw_str):
|
||||
|
||||
try:
|
||||
ip = socket.inet_aton(ip_str)
|
||||
except:
|
||||
print "IP address is invalid."
|
||||
return False
|
||||
|
||||
try:
|
||||
mask = socket.inet_aton(mask_str)
|
||||
except:
|
||||
print "Network mask is invalid."
|
||||
return False
|
||||
|
||||
try:
|
||||
gw = socket.inet_aton(gw_str)
|
||||
except:
|
||||
print "Gateway address is invalid."
|
||||
return False
|
||||
|
||||
# Validate network mask
|
||||
|
||||
# Convert to integer from byte array
|
||||
mask = struct.unpack("!L", mask)[0]
|
||||
|
||||
# Exclude restricted masks
|
||||
if (mask == 0) or (mask == 0xFFFFFFFF):
|
||||
print "Network mask is invalid."
|
||||
return False
|
||||
|
||||
# Exclude non-left-contiguous masks
|
||||
if (((mask + (mask & -mask)) & 0xFFFFFFFF) != 0):
|
||||
print "Network mask is not contiguous."
|
||||
return False
|
||||
|
||||
|
||||
# Validate gateway address
|
||||
|
||||
octet1 = ord(gw[0])
|
||||
|
||||
# Convert to integer from byte array
|
||||
gw = struct.unpack("!L", gw)[0]
|
||||
|
||||
# Exclude restricted addresses
|
||||
# 0.0.0.0 is valid
|
||||
if ((gw != 0) and ((octet1 == 0) or (octet1 == 127) or (octet1 > 223))):
|
||||
print "Gateway address is invalid."
|
||||
return False
|
||||
|
||||
# Validate IP address
|
||||
|
||||
octet1 = ord(ip[0])
|
||||
|
||||
# Convert to integer from byte array
|
||||
ip = struct.unpack("!L", ip)[0]
|
||||
|
||||
# Exclude restricted addresses
|
||||
if ((octet1 == 0) or (octet1 == 127) or (octet1 > 223)):
|
||||
print "IP address is invalid."
|
||||
return False
|
||||
|
||||
# Exclude subnet network address
|
||||
if ((ip & ~mask) == 0):
|
||||
print "IP address is invalid."
|
||||
return False
|
||||
|
||||
# Exclude subnet broadcast address
|
||||
if ((ip & ~mask) == (0xFFFFFFFF & ~mask)):
|
||||
print "IP address is invalid."
|
||||
return False
|
||||
|
||||
return True
|
||||
|
||||
#-----------------------------------------------------------------------------
|
||||
#def ValidateAddress(address):
|
||||
|
||||
# if address is None:
|
||||
# return False
|
||||
|
||||
# parts = address.split(".")
|
||||
|
||||
# if len(parts) != 4:
|
||||
# return False
|
||||
|
||||
# try:
|
||||
# for item in parts:
|
||||
# if not 0 <= int(item) <= 255:
|
||||
# return False
|
||||
# except:
|
||||
# return False
|
||||
|
||||
# return True
|
||||
|
||||
|
||||
#-----------------------------------------------------------------------------
|
||||
def main():
|
||||
|
||||
invalid_args = False
|
||||
showhelp = False
|
||||
search = False
|
||||
ip_type = None
|
||||
ip_addr = None
|
||||
netmask = None
|
||||
gateway = None
|
||||
eth_addr = None
|
||||
|
||||
try:
|
||||
opts, args = getopt.getopt(sys.argv[1:], '', ['help', 'list', 'eth_addr=', 'ip_type=', 'ip_addr=', 'netmask=', 'gateway='])
|
||||
except getopt.GetoptError, err:
|
||||
print str(err)
|
||||
sys.exit(1)
|
||||
|
||||
# Check for unparsed parameters
|
||||
if len(args) != 0:
|
||||
usage()
|
||||
sys.exit(1)
|
||||
|
||||
for o, a in opts:
|
||||
if o == "--help":
|
||||
showhelp = True
|
||||
elif o == "--list":
|
||||
search = True
|
||||
elif o == "--eth_addr":
|
||||
eth_addr = a
|
||||
elif o == "--ip_type":
|
||||
ip_type = a
|
||||
elif o == "--ip_addr":
|
||||
ip_addr = a
|
||||
elif o == "--netmask":
|
||||
netmask = a
|
||||
elif o == "--gateway":
|
||||
gateway = a
|
||||
|
||||
if (len(opts) == 0) or (showhelp):
|
||||
usage()
|
||||
sys.exit(1)
|
||||
|
||||
if search:
|
||||
if not eth_addr is None:
|
||||
print "--list and --eth_addr are not compatible."
|
||||
invalid_args = True
|
||||
if not ip_type is None:
|
||||
print "--list and --ip_type are not compatible."
|
||||
invalid_args = True
|
||||
if not ip_addr is None:
|
||||
print "--list and --ip_addr are not compatible."
|
||||
invalid_args = True
|
||||
if not netmask is None:
|
||||
print "--list and --netmask are not compatible."
|
||||
invalid_args = True
|
||||
if not gateway is None:
|
||||
print "--list and --gateway are not compatible."
|
||||
invalid_args = True
|
||||
else:
|
||||
|
||||
try:
|
||||
eth_addr = eth_addr.strip().replace(":", "").replace("-", "")
|
||||
eth_addr = eth_addr.decode('hex')
|
||||
except:
|
||||
print "Invalid Ethernet address."
|
||||
sys.exit(1)
|
||||
|
||||
if len(eth_addr) != 6:
|
||||
print "Invalid Ethernet address."
|
||||
sys.exit(1)
|
||||
|
||||
if ip_type in ["Static", "static"]:
|
||||
ip_type = NF_IP_STATIC
|
||||
elif ip_type in ["Dynamic", "dynamic", "Dhcp", "dhcp"]:
|
||||
ip_type = NF_IP_DYNAMIC
|
||||
else:
|
||||
print "--ip_type must be 'static' or 'dhcp'."
|
||||
sys.exit(1)
|
||||
|
||||
if ip_type == NF_IP_STATIC:
|
||||
|
||||
if not ValidateNetParams(ip_addr, netmask, gateway):
|
||||
invalid_args = True
|
||||
|
||||
|
||||
# if not ValidateIP(ip_addr):
|
||||
# print "Invalid, or no, IP address specified."
|
||||
# invalid_args = True
|
||||
|
||||
# if not ValidateIP(netmask):
|
||||
# print "Invalid, or no, netmask specified."
|
||||
# invalid_args = True
|
||||
|
||||
# if not ValidateIP(gateway):
|
||||
# print "Invalid, or no, gateway address specified."
|
||||
# invalid_args = True
|
||||
else:
|
||||
if ip_addr is None:
|
||||
ip_addr = "0.0.0.0"
|
||||
else:
|
||||
print "--ip_addr not allowed when --ip_type=dhcp."
|
||||
invalid_args = True
|
||||
if netmask is None:
|
||||
netmask = "0.0.0.0"
|
||||
else:
|
||||
print "--netmask not allowed when --ip_type=dhcp."
|
||||
invalid_args = True
|
||||
if gateway is None:
|
||||
gateway = "0.0.0.0"
|
||||
else:
|
||||
print "--gateway not allowed when --ip_type=dhcp."
|
||||
invalid_args = True
|
||||
|
||||
if invalid_args:
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
|
||||
global seq
|
||||
sys.stdout = os.fdopen(sys.stdout.fileno(), 'w', 0)
|
||||
|
||||
iplist = enumIp();
|
||||
if len(iplist) == 0:
|
||||
print "Host has no IP address."
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
devices = {}
|
||||
|
||||
for ip in iplist:
|
||||
print "Searching through network interface:", ip
|
||||
|
||||
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||||
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||||
s.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
|
||||
|
||||
port = 0
|
||||
|
||||
try:
|
||||
s.bind((ip, port))
|
||||
except socket.error, e:
|
||||
print "Bind error on send socket:", e
|
||||
sys.exit(1)
|
||||
|
||||
port = s.getsockname()[1]
|
||||
|
||||
r = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||||
r.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||||
|
||||
r.setblocking(1)
|
||||
r.settimeout(0.100)
|
||||
|
||||
try:
|
||||
r.bind(('', port))
|
||||
except socket.error, e:
|
||||
print "Bind error on receive socket:", e
|
||||
sys.exit(1)
|
||||
|
||||
d = Discover(s, r)
|
||||
print
|
||||
|
||||
for k in d:
|
||||
d[k]['host_ip'] = ip
|
||||
devices[k] = d[k]
|
||||
|
||||
s.close()
|
||||
r.close()
|
||||
|
||||
if search:
|
||||
print
|
||||
print "Found", len(devices), "Prologix GPIB-ETHERNET Controller(s)."
|
||||
for key in devices:
|
||||
PrintDetails(devices[key])
|
||||
print
|
||||
else:
|
||||
if eth_addr in devices:
|
||||
|
||||
print "Updating network settings of Prologix GPIB-ETHERNET Controller", FormatEthAddr(eth_addr)
|
||||
|
||||
device = devices[eth_addr]
|
||||
|
||||
if (device['ip_type'] == NF_IP_STATIC) or (ip_type == NF_IP_STATIC):
|
||||
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||||
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||||
s.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
|
||||
|
||||
port = 0
|
||||
|
||||
try:
|
||||
s.bind((device['host_ip'], port))
|
||||
except socket.error, e:
|
||||
print "Bind error on send socket:", e
|
||||
sys.exit(1)
|
||||
|
||||
port = s.getsockname()[1]
|
||||
|
||||
r = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||||
r.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||||
|
||||
r.setblocking(1)
|
||||
r.settimeout(0.100)
|
||||
|
||||
try:
|
||||
r.bind(('', port))
|
||||
except socket.error, e:
|
||||
print "Bind error on receive socket:", e
|
||||
sys.exit(1)
|
||||
|
||||
result = Assignment(s, r, eth_addr, ip_type, ip_addr, netmask, gateway)
|
||||
print
|
||||
|
||||
if len(result) == 0:
|
||||
print "Network settings update failed."
|
||||
else:
|
||||
if result['result'] == NF_SUCCESS:
|
||||
print "Network settings updated successfully."
|
||||
else:
|
||||
print "Network settings update failed."
|
||||
else:
|
||||
print "Prologix GPIB-ETHERNET Controller", FormatEthAddr(eth_addr), "already configured for DHCP."
|
||||
else:
|
||||
print "Prologix GPIB-ETHERNET Controller", FormatEthAddr(eth_addr), "not found."
|
||||
|
||||
|
||||
|
||||
#-----------------------------------------------------------------------------
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
|
|
|
|||
Loading…
Reference in New Issue